// Copyright 2022, 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 h2645

import (
	"github.com/q191201771/lal/pkg/avc"
	"github.com/q191201771/lal/pkg/hevc"
	"github.com/q191201771/naza/pkg/bele"
)

// TODO(chef): 逐渐将package avc, hevc迁移到package h2645，这个package处于开发阶段，如果内容不全，请使用package avc, hevc

// 无特殊说明的函数则同时支持h264和h265两种格式

var (
	NaluStartCode4 = []byte{0x0, 0x0, 0x0, 0x1}
)

const (
	H264NaluTypeSlice    uint8 = 1
	H264NaluTypeIdrSlice uint8 = 5
	H264NaluTypeSei      uint8 = 6
	H264NaluTypeSps      uint8 = 7
	H264NaluTypePps      uint8 = 8
	H264NaluTypeAud      uint8 = 9  // Access Unit Delimiter
	H264NaluTypeFd       uint8 = 12 // Filler Data
)

// ISO_IEC_23008-2_2013.pdf
// Table 7-1 – NAL unit type codes and NAL unit type classes
const (
	H265NaluTypeSliceTrailN uint8 = 0 // 0x0
	H265NaluTypeSliceTrailR uint8 = 1 // 0x01
	H265NaluTypeSliceTsaN   uint8 = 2 // 0x02
	H265NaluTypeSliceTsaR   uint8 = 3 // 0x03
	H265NaluTypeSliceStsaN  uint8 = 4 // 0x04
	H265NaluTypeSliceStsaR  uint8 = 5 // 0x05
	H265NaluTypeSliceRadlN  uint8 = 6 // 0x06
	H265NaluTypeSliceRadlR  uint8 = 7 // 0x07
	H265NaluTypeSliceRaslN  uint8 = 8 // 0x06
	H265NaluTypeSliceRaslR  uint8 = 9 // 0x09

	H265NaluTypeSliceBlaWlp       uint8 = 16 // 0x10
	H265NaluTypeSliceBlaWradl     uint8 = 17 // 0x11
	H265NaluTypeSliceBlaNlp       uint8 = 18 // 0x12
	H265NaluTypeSliceIdr          uint8 = 19 // 0x13
	H265NaluTypeSliceIdrNlp       uint8 = 20 // 0x14
	H265NaluTypeSliceCranut       uint8 = 21 // 0x15
	H265NaluTypeSliceRsvIrapVcl22 uint8 = 22 // 0x16
	H265NaluTypeSliceRsvIrapVcl23 uint8 = 23 // 0x17

	H265NaluTypeVps       uint8 = 32 // 0x20
	H265NaluTypeSps       uint8 = 33 // 0x21
	H265NaluTypePps       uint8 = 34 // 0x22
	H265NaluTypeAud       uint8 = 35 // 0x23
	H265NaluTypeSei       uint8 = 39 // 0x27
	H265NaluTypeSeiSuffix uint8 = 40 // 0x28
)

// IterateNaluAvcc 遍历Avcc格式的nalu流
func IterateNaluAvcc(nals []byte, handler func(nal []byte)) error {
	return avc.IterateNaluAvcc(nals, handler)
}

func IterateNaluStartCode(nalu []byte, start int) (pos, length int) {
	return avc.IterateNaluStartCode(nalu, start)
}

func ParseNaluType(isH264 bool, v uint8) uint8 {
	if isH264 {
		return avc.ParseNaluType(v)
	}
	return hevc.ParseNaluType(v)
}

func SeqHeader2Annexb(isH264 bool, payload []byte) ([]byte, error) {
	if isH264 {
		return avc.SpsPpsSeqHeader2Annexb(payload)
	}
	return hevc.VpsSpsPpsSeqHeader2Annexb(payload)
}

func H265IsIrapNalu(typ uint8) bool {
	return hevc.IsIrapNalu(typ)
}

func JoinNaluAvcc(naluList ...[]byte) []byte {
	n := len(naluList)
	if n == 0 {
		return nil
	}
	n *= 4
	for _, item := range naluList {
		n += len(item)
	}
	ret := make([]byte, n)

	pos := 0
	for _, item := range naluList {
		bele.BePutUint32(ret[pos:], uint32(len(item)))
		pos += 4
		copy(ret[pos:], item)
		pos += len(item)
	}

	return ret
}