77 lines
2.0 KiB
Go
77 lines
2.0 KiB
Go
|
package aep
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/binary"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// chunk builds a RIFX chunk: 4-byte id + BE32 size + data (+ pad to even).
|
||
|
|
func chunk(id string, data []byte) []byte {
|
||
|
|
b := new(bytes.Buffer)
|
||
|
|
b.WriteString(id)
|
||
|
|
_ = binary.Write(b, binary.BigEndian, uint32(len(data)))
|
||
|
|
b.Write(data)
|
||
|
|
if len(data)%2 == 1 {
|
||
|
|
b.WriteByte(0)
|
||
|
|
}
|
||
|
|
return b.Bytes()
|
||
|
|
}
|
||
|
|
|
||
|
|
// list builds a LIST chunk of the given form type wrapping the body chunks.
|
||
|
|
func list(formType string, body []byte) []byte {
|
||
|
|
return chunk("LIST", append([]byte(formType), body...))
|
||
|
|
}
|
||
|
|
|
||
|
|
func cdtaWithDuration(durFrames, scale uint32) []byte {
|
||
|
|
b := make([]byte, 0x28)
|
||
|
|
binary.BigEndian.PutUint32(b[0x20:0x24], durFrames)
|
||
|
|
binary.BigEndian.PutUint32(b[0x24:0x28], scale)
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseComps(t *testing.T) {
|
||
|
|
// One composition (has cdta), one footage item (no cdta), inside a folder.
|
||
|
|
comp := list("Item", bytes.Join([][]byte{
|
||
|
|
chunk("cdta", cdtaWithDuration(150, 30)), // 150 frames @ 30 → 5.0s
|
||
|
|
chunk("Utf8", []byte("scene_intro")),
|
||
|
|
}, nil))
|
||
|
|
footage := list("Item", bytes.Join([][]byte{
|
||
|
|
chunk("Utf8", []byte("clip.mp4")),
|
||
|
|
chunk("sspc", make([]byte, 8)),
|
||
|
|
}, nil))
|
||
|
|
folder := list("Item", bytes.Join([][]byte{
|
||
|
|
chunk("Utf8", []byte("My Folder")),
|
||
|
|
comp,
|
||
|
|
footage,
|
||
|
|
}, nil))
|
||
|
|
|
||
|
|
body := bytes.Join([][]byte{folder}, nil)
|
||
|
|
// RIFX header: "RIFX" + size + "Egg!" + body
|
||
|
|
file := new(bytes.Buffer)
|
||
|
|
file.WriteString("RIFX")
|
||
|
|
_ = binary.Write(file, binary.BigEndian, uint32(len(body)+4))
|
||
|
|
file.WriteString("Egg!")
|
||
|
|
file.Write(body)
|
||
|
|
|
||
|
|
comps, err := ParseComps(file.Bytes())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("ParseComps: %v", err)
|
||
|
|
}
|
||
|
|
if len(comps) != 1 {
|
||
|
|
t.Fatalf("expected 1 comp, got %d: %+v", len(comps), comps)
|
||
|
|
}
|
||
|
|
if comps[0].Name != "scene_intro" {
|
||
|
|
t.Errorf("name = %q, want scene_intro", comps[0].Name)
|
||
|
|
}
|
||
|
|
if comps[0].DurationSec != 5.0 {
|
||
|
|
t.Errorf("duration = %v, want 5.0", comps[0].DurationSec)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseCompsRejectsNonRifx(t *testing.T) {
|
||
|
|
if _, err := ParseComps([]byte("not an aep file at all")); err == nil {
|
||
|
|
t.Error("expected error for non-RIFX input")
|
||
|
|
}
|
||
|
|
}
|