srs/trunk/3rdparty/srs-bench/vendor/github.com/yapingcat/gomedia/codec/vp8.go

package codec

import "errors"

type VP8FrameTag struct {
    FrameType     uint32 //0: I frame , 1: P frame
    Version       uint32
    Display       uint32
    FirstPartSize uint32
}

type VP8KeyFrameHead struct {
    Width      int
    Height     int
    HorizScale int
    VertScale  int
}

func DecodeFrameTag(frame []byte) (*VP8FrameTag, error) {
    if len(frame) < 3 {
        return nil, errors.New("frame bytes < 3")
    }
    var tmp uint32 = (uint32(frame[2]) << 16) | (uint32(frame[1]) << 8) | uint32(frame[0])
    tag := &VP8FrameTag{}
    tag.FrameType = tmp & 0x01
    tag.Version = (tmp >> 1) & 0x07
    tag.Display = (tmp >> 4) & 0x01
    tag.FirstPartSize = (tmp >> 5) & 0x7FFFF
    return tag, nil
}

func DecodeKeyFrameHead(frame []byte) (*VP8KeyFrameHead, error) {
    if len(frame) < 7 {
        return nil, errors.New("frame bytes < 3")
    }

    if frame[0] != 0x9d || frame[1] != 0x01 || frame[2] != 0x2a {
        return nil, errors.New("not find Start code")
    }

    head := &VP8KeyFrameHead{}
    head.Width = int(uint16(frame[4]&0x3f)<<8 | uint16(frame[3]))
    head.HorizScale = int(frame[4] >> 6)
    head.Height = int(uint16(frame[6]&0x3f)<<8 | uint16(frame[5]))
    head.VertScale = int(frame[6] >> 6)
    return head, nil
}

func IsKeyFrame(frame []byte) bool {
    tag, err := DecodeFrameTag(frame)
    if err != nil {
        return false
    }

    if tag.FrameType == 0 {
        return true
    } else {
        return false
    }
}

func GetResloution(frame []byte) (width int, height int, err error) {
    if !IsKeyFrame(frame) {
        return 0, 0, errors.New("the frame is not Key frame")
    }

    head, err := DecodeKeyFrameHead(frame[3:])
    if err != nil {
        return 0, 0, err
    }
    return head.Width, head.Height, nil
}