package/TrafficSpeed
1
0
Fork 0
mirror of https://github.com/jehiah/TrafficSpeed.git synced 2025-10-16 13:20:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

minor updates

Browse Source
This commit is contained in:
Jehiah Czebotar
2020-04-24 16:31:55 -04:00
parent b62fd13263
commit d85693af3a
5 changed files with 52 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/project/video_iterator.go
View File

@@ -57,7 +57,7 @@ func NewIterator(filename string) (iter *Iterator, err error) {
// astream := stream.(av.AudioCodecData)
// fmt.Println(astream.Type(), astream.SampleRate(), astream.SampleFormat(), astream.ChannelLayout())
// } else if stream.Type().IsVideo() {
fmt.Printf("stream[%d] = %s\n", i, stream.Type())
fmt.Printf("stream[%d] = %s (video:%v)\n", i, stream.Type(), stream.Type().IsVideo())
if stream.Type().IsVideo() {
vstream := stream.(av.VideoCodecData)
r := image.Rect(0, 0, vstream.Width(), vstream.Height())
@@ -138,6 +138,10 @@ func (i *Iterator) DecodeFrame() error {
// decode
decoder := i.decoders[i.packet.Idx]
var err error
if len(i.packet.Data) == 0 {
log.Printf("no packet at frame %d", i.frame)
return nil
}
i.vf, err = decoder.Decode(i.packet.Data)
if i.vf == nil {
log.Printf("no image at frame %d", i.frame)
@@ -148,6 +152,11 @@ func (i *Iterator) DecodeFrame() error {
}
func (i *Iterator) Image() *image.YCbCr {
if i.frame == -1 {
if !i.NextWithImage() {
panic("no image")
}
}
i.err = i.DecodeFrame()
if i.vf == nil {
return nil