mirror of
https://github.com/asticode/go-astiav.git
synced 2025-10-05 00:02:45 +08:00
90 lines
3.2 KiB
Go
90 lines
3.2 KiB
Go
package astiav_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/asticode/go-astiav"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCodecParameters(t *testing.T) {
|
|
_, s1, s2, err := videoInputStreams()
|
|
require.NoError(t, err)
|
|
|
|
cp1 := s1.CodecParameters()
|
|
require.Equal(t, int64(441324), cp1.BitRate())
|
|
require.Equal(t, astiav.ChromaLocationLeft, cp1.ChromaLocation())
|
|
require.Equal(t, astiav.CodecIDH264, cp1.CodecID())
|
|
require.Equal(t, astiav.CodecTag(0x31637661), cp1.CodecTag())
|
|
require.Equal(t, astiav.ColorPrimariesUnspecified, cp1.ColorPrimaries())
|
|
require.Equal(t, astiav.ColorRangeUnspecified, cp1.ColorRange())
|
|
require.Equal(t, astiav.ColorSpaceUnspecified, cp1.ColorSpace())
|
|
require.Equal(t, astiav.ColorTransferCharacteristicUnspecified, cp1.ColorTransferCharacteristic())
|
|
require.Equal(t, 180, cp1.Height())
|
|
require.Equal(t, astiav.Level(13), cp1.Level())
|
|
require.Equal(t, astiav.MediaTypeVideo, cp1.MediaType())
|
|
require.Equal(t, astiav.PixelFormatYuv420P, cp1.PixelFormat())
|
|
require.Equal(t, astiav.ProfileH264ConstrainedBaseline, cp1.Profile())
|
|
require.Equal(t, astiav.NewRational(1, 1), cp1.SampleAspectRatio())
|
|
require.Equal(t, 320, cp1.Width())
|
|
|
|
cp2 := s2.CodecParameters()
|
|
require.Equal(t, int64(161052), cp2.BitRate())
|
|
require.Equal(t, 2, cp2.Channels())
|
|
require.Equal(t, astiav.ChannelLayoutStereo, cp2.ChannelLayout())
|
|
require.Equal(t, astiav.CodecIDAac, cp2.CodecID())
|
|
require.Equal(t, astiav.CodecTag(0x6134706d), cp2.CodecTag())
|
|
require.Equal(t, 1024, cp2.FrameSize())
|
|
require.Equal(t, astiav.MediaTypeAudio, cp2.MediaType())
|
|
require.Equal(t, astiav.SampleFormatFltp, cp2.SampleFormat())
|
|
require.Equal(t, 48000, cp2.SampleRate())
|
|
|
|
cp3 := astiav.AllocCodecParameters()
|
|
require.NotNil(t, cp3)
|
|
defer cp3.Free()
|
|
err = cp2.Copy(cp3)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, cp3.Channels())
|
|
|
|
cc4 := astiav.AllocCodecContext(nil)
|
|
require.NotNil(t, cc4)
|
|
defer cc4.Free()
|
|
err = cp2.ToCodecContext(cc4)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, cc4.Channels())
|
|
|
|
cp5 := astiav.AllocCodecParameters()
|
|
require.NotNil(t, cp5)
|
|
defer cp5.Free()
|
|
err = cp5.FromCodecContext(cc4)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, cp5.Channels())
|
|
|
|
cp6 := astiav.AllocCodecParameters()
|
|
require.NotNil(t, cp6)
|
|
defer cp6.Free()
|
|
cp6.SetChannelLayout(astiav.ChannelLayout21)
|
|
require.Equal(t, astiav.ChannelLayout21, cp6.ChannelLayout())
|
|
defer cp6.Free()
|
|
cp6.SetChannels(3)
|
|
require.Equal(t, 3, cp6.Channels())
|
|
cp6.SetCodecID(astiav.CodecIDRawvideo)
|
|
require.Equal(t, astiav.CodecIDRawvideo, cp6.CodecID())
|
|
cp6.SetCodecTag(astiav.CodecTag(2))
|
|
require.Equal(t, astiav.CodecTag(2), cp6.CodecTag())
|
|
cp6.SetCodecType(astiav.MediaTypeAudio)
|
|
require.Equal(t, astiav.MediaTypeAudio, cp6.CodecType())
|
|
cp6.SetHeight(1)
|
|
require.Equal(t, 1, cp6.Height())
|
|
cp6.SetPixelFormat(astiav.PixelFormat0Bgr)
|
|
require.Equal(t, astiav.PixelFormat0Bgr, cp6.PixelFormat())
|
|
cp6.SetSampleAspectRatio(astiav.NewRational(1, 2))
|
|
require.Equal(t, astiav.NewRational(1, 2), cp6.SampleAspectRatio())
|
|
cp6.SetSampleFormat(astiav.SampleFormatDbl)
|
|
require.Equal(t, astiav.SampleFormatDbl, cp6.SampleFormat())
|
|
cp6.SetSampleRate(4)
|
|
require.Equal(t, 4, cp6.SampleRate())
|
|
cp6.SetWidth(2)
|
|
require.Equal(t, 2, cp6.Width())
|
|
}
|