Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
7.0 kB · 367 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368package renditions
import ( "encoding/json" "testing"
"github.com/stretchr/testify/require" "stream.place/streamplace/pkg/placestream")
func seg(width int, height int, fpsNum int, fpsDen int) *placestream.Segment { return &placestream.Segment{ Video: []placestream.Segment_Video{ { Width: int64(width), Height: int64(height), Framerate: &placestream.Segment_Framerate{ Num: int64(fpsNum), Den: int64(fpsDen), }, }, }, }}
var cases = []struct { name string spseg *placestream.Segment lp string}{ { name: "4K 60fps", spseg: seg(3840, 2160, 60, 1), lp: ` [ { "name": "1080p", "width": 1920, "height": 1080, "bitrate": 6000000, "profile": "h264constrainedhigh" }, { "name": "720p", "width": 1280, "height": 720, "bitrate": 3000000, "profile": "h264constrainedhigh" }, { "name": "360p", "width": 640, "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "240p", "width": 426, "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "160p", "width": 284, "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 60, "fpsDen": 2 } ] `, }, { name: "2K with fractional framerate", spseg: seg(2160, 1440, 60000, 1001), lp: ` [ { "name": "1080p", "width": 1620, "height": 1080, "bitrate": 6000000, "profile": "h264constrainedhigh" }, { "name": "720p", "width": 1080, "height": 720, "bitrate": 3000000, "profile": "h264constrainedhigh" }, { "name": "360p", "width": 540, "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60000, "fpsDen": 2002 }, { "name": "240p", "width": 360, "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60000, "fpsDen": 2002 }, { "name": "160p", "width": 240, "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 60000, "fpsDen": 2002 } ] `, }, { name: "720p 50fps", spseg: seg(1280, 720, 50, 1), lp: ` [ { "name": "360p", "width": 640, "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 50, "fpsDen": 2 }, { "name": "240p", "width": 426, "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 50, "fpsDen": 2 }, { "name": "160p", "width": 284, "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 50, "fpsDen": 2 } ] `, }, { name: "720p 30fps", spseg: seg(1280, 720, 30, 1), lp: ` [ { "name": "360p", "width": 640, "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh" }, { "name": "240p", "width": 426, "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh" }, { "name": "160p", "width": 284, "height": 160, "bitrate": 250000, "profile": "h264baseline" } ] `, }, { name: "720p 25fps", spseg: seg(1280, 720, 25, 1), lp: ` [ { "name": "360p", "width": 640, "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh" }, { "name": "240p", "width": 426, "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh" }, { "name": "160p", "width": 284, "height": 160, "bitrate": 250000, "profile": "h264baseline" } ] `, }, { name: "Vertical video 60fps", spseg: seg(480, 640, 60, 1), lp: ` [ { "name": "360p", "width": 360, "height": 480, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "240p", "width": 240, "height": 320, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "160p", "width": 160, "height": 214, "bitrate": 250000, "profile": "h264baseline", "fps": 60, "fpsDen": 2 } ] `, },}
func TestRenditions(t *testing.T) { for _, c := range cases { t.Run(c.name, func(t *testing.T) { rends, err := GenerateRenditions(c.spseg) require.NoError(t, err) lp := rends.ToLivepeerProfiles() bs, err := json.Marshal(lp) require.NoError(t, err) require.JSONEq(t, c.lp, string(bs)) }) }}
var singleCases = []struct { name string spseg *placestream.Segment lp string dimensions []int}{ { name: "Nearly-Square Landscape 4K 60fps", spseg: seg(3840, 3830, 60, 1), dimensions: []int{1083, 1080}, lp: ` { "name": "1080p", "width": 1084, "height": 1080, "bitrate": 6000000, "profile": "h264constrainedhigh" } `, }, { name: "Nearly-Square Portrait 4K 60fps", spseg: seg(3830, 3840, 60, 1), dimensions: []int{1080, 1083}, lp: ` { "name": "1080p", "width": 1080, "height": 1084, "bitrate": 6000000, "profile": "h264constrainedhigh" } `, }, { name: "Stupidly-Wide Landscape 4K 60fps", spseg: seg(5000, 2160, 60, 1), dimensions: []int{1920, 829}, lp: ` { "name": "1080p", "width": 1920, "height": 830, "bitrate": 6000000, "profile": "h264constrainedhigh" } `, }, { name: "Stupidly-Tall Portrait 4K 60fps", spseg: seg(2160, 5000, 60, 1), dimensions: []int{829, 1920}, lp: ` { "name": "1080p", "width": 830, "height": 1920, "bitrate": 6000000, "profile": "h264constrainedhigh" } `, },}
func TestSingleRendition(t *testing.T) { for _, c := range singleCases { t.Run(c.name, func(t *testing.T) { rends, err := GenerateRenditions(c.spseg) require.NoError(t, err) first := rends[0] require.Equal(t, c.dimensions, []int{int(first.Width), int(first.Height)}) lp := rends.ToLivepeerProfiles() bs, err := json.Marshal(lp[0]) require.NoError(t, err) require.JSONEq(t, c.lp, string(bs)) }) }}
func TestLadder(t *testing.T) { all, err := Ladder("") require.NoError(t, err) require.Equal(t, DesiredRenditions, all) two, err := Ladder("160p, 240p") require.NoError(t, err) require.Equal(t, []string{"240p", "160p"}, []string{two[0].Name, two[1].Name}, "ladder order, whatever the order given") _, err = Ladder("240p,4k") require.ErrorContains(t, err, `unknown rendition "4k"`)
src := seg(1920, 1080, 30, 1) rs, err := GenerateRenditionsFrom(src, two) require.NoError(t, err) require.Len(t, rs, 2) require.Equal(t, "240p", rs[0].Name) require.Equal(t, "160p", rs[1].Name)}