Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
5.9 kB · 323 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324package 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", "height": 1080, "bitrate": 6000000, "profile": "h264constrainedhigh" }, { "name": "720p", "height": 720, "bitrate": 3000000, "profile": "h264constrainedhigh" }, { "name": "360p", "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "240p", "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "160p", "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 60, "fpsDen": 2 } ] `, }, { name: "2K with fractional framerate", spseg: seg(2160, 1440, 60000, 1001), lp: ` [ { "name": "1080p", "height": 1080, "bitrate": 6000000, "profile": "h264constrainedhigh" }, { "name": "720p", "height": 720, "bitrate": 3000000, "profile": "h264constrainedhigh" }, { "name": "360p", "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60000, "fpsDen": 2002 }, { "name": "240p", "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60000, "fpsDen": 2002 }, { "name": "160p", "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 60000, "fpsDen": 2002 } ] `, }, { name: "720p 50fps", spseg: seg(1280, 720, 50, 1), lp: ` [ { "name": "360p", "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 50, "fpsDen": 2 }, { "name": "240p", "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 50, "fpsDen": 2 }, { "name": "160p", "height": 160, "bitrate": 250000, "profile": "h264baseline", "fps": 50, "fpsDen": 2 } ] `, }, { name: "720p 30fps", spseg: seg(1280, 720, 30, 1), lp: ` [ { "name": "360p", "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh" }, { "name": "240p", "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh" }, { "name": "160p", "height": 160, "bitrate": 250000, "profile": "h264baseline" } ] `, }, { name: "720p 25fps", spseg: seg(1280, 720, 25, 1), lp: ` [ { "name": "360p", "height": 360, "bitrate": 1000000, "profile": "h264constrainedhigh" }, { "name": "240p", "height": 240, "bitrate": 500000, "profile": "h264constrainedhigh" }, { "name": "160p", "height": 160, "bitrate": 250000, "profile": "h264baseline" } ] `, }, { name: "Vertical video 60fps", spseg: seg(480, 640, 60, 1), lp: ` [ { "name": "360p", "width": 360, "bitrate": 1000000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "240p", "width": 240, "bitrate": 500000, "profile": "h264constrainedhigh", "fps": 60, "fpsDen": 2 }, { "name": "160p", "width": 160, "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", "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, "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, "bitrate": 6000000, "profile": "h264constrainedhigh" } `, }, { name: "Stupidly-Tall Portrait 4K 60fps", spseg: seg(2160, 5000, 60, 1), dimensions: []int{829, 1920}, lp: ` { "name": "1080p", "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)) }) }}