lal/pkg/rtmp/metadata.go

// Copyright 2020, Chef.  All rights reserved.
// https://github.com/q191201771/lal
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package rtmp

import (
	"bytes"

	"github.com/q191201771/lal/pkg/base"
)

func ParseMetadata(b []byte) (ObjectPairArray, error) {
	pos := 0
	v, l, err := AMF0.ReadString(b[pos:])
	if err != nil {
		return nil, err
	}
	pos += l
	if v == "@setDataFrame" {
		_, l, err = AMF0.ReadString(b[pos:])
		if err != nil {
			return nil, err
		}
		pos += l
	}
	opa, _, err := AMF0.ReadObjectOrArray(b[pos:])
	return opa, err
}

// spec-video_file_format_spec_v10.pdf
// onMetaData
// - duration        DOUBLE, seconds
// - width           DOUBLE
// - height          DOUBLE
// - videodatarate   DOUBLE
// - framerate       DOUBLE
// - videocodecid    DOUBLE
// - audiosamplerate DOUBLE
// - audiosamplesize DOUBLE
// - stereo          BOOL
// - audiocodecid    DOUBLE
// - filesize        DOUBLE, bytes
//
// 目前包含的字段：
// - width
// - height
// - audiocodecid
// - videocodecid
// - version
//
// @param width        如果为-1，则metadata中不写入该字段
// @param height       如果为-1，则metadata中不写入该字段
// @param audiocodecid 如果为-1，则metadata中不写入该字段
//                     AAC 10
// @param videocodecid 如果为-1，则metadata中不写入该字段
//                     H264 7
//                     H265 12
// @return 返回的内存块为新申请的独立内存块
func BuildMetadata(width int, height int, audiocodecid int, videocodecid int) ([]byte, error) {
	buf := &bytes.Buffer{}
	if err := AMF0.WriteString(buf, "onMetaData"); err != nil {
		return nil, err
	}

	var opa ObjectPairArray
	if width != -1 {
		opa = append(opa, ObjectPair{
			Key:   "width",
			Value: width,
		})
	}
	if height != -1 {
		opa = append(opa, ObjectPair{
			Key:   "height",
			Value: height,
		})
	}
	if audiocodecid != -1 {
		opa = append(opa, ObjectPair{
			Key:   "audiocodecid",
			Value: audiocodecid,
		})
	}
	if videocodecid != -1 {
		opa = append(opa, ObjectPair{
			Key:   "videocodecid",
			Value: videocodecid,
		})
	}
	opa = append(opa, ObjectPair{
		Key:   "version",
		Value: base.LALRTMPBuildMetadataEncoder,
	})

	if err := AMF0.WriteObject(buf, opa); err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}