merge format and formatdecenc into formats (#222)

pkg/formats/h265.go

@@ -0,0 +1,164 @@
+package formats
+
+import (
+	"encoding/base64"
+	"fmt"
+	"strconv"
+	"sync"
+
+	"github.com/pion/rtp"
+
+	"github.com/bluenviron/gortsplib/v3/pkg/formats/rtph265"
+)
+
+// H265 is a format that uses the H265 codec.
+type H265 struct {
+	PayloadTyp uint8
+	VPS        []byte
+	SPS        []byte
+	PPS        []byte
+	MaxDONDiff int
+
+	mutex sync.RWMutex
+}
+
+// String implements Format.
+func (t *H265) String() string {
+	return "H265"
+}
+
+// ClockRate implements Format.
+func (t *H265) ClockRate() int {
+	return 90000
+}
+
+// PayloadType implements Format.
+func (t *H265) PayloadType() uint8 {
+	return t.PayloadTyp
+}
+
+func (t *H265) unmarshal(payloadType uint8, clock string, codec string, rtpmap string, fmtp map[string]string) error {
+	t.PayloadTyp = payloadType
+
+	for key, val := range fmtp {
+		switch key {
+		case "sprop-vps":
+			var err error
+			t.VPS, err = base64.StdEncoding.DecodeString(val)
+			if err != nil {
+				return fmt.Errorf("invalid sprop-vps (%v)", fmtp)
+			}
+
+		case "sprop-sps":
+			var err error
+			t.SPS, err = base64.StdEncoding.DecodeString(val)
+			if err != nil {
+				return fmt.Errorf("invalid sprop-sps (%v)", fmtp)
+			}
+
+		case "sprop-pps":
+			var err error
+			t.PPS, err = base64.StdEncoding.DecodeString(val)
+			if err != nil {
+				return fmt.Errorf("invalid sprop-pps (%v)", fmtp)
+			}
+
+		case "sprop-max-don-diff":
+			tmp, err := strconv.ParseInt(val, 10, 64)
+			if err != nil {
+				return fmt.Errorf("invalid sprop-max-don-diff (%v)", fmtp)
+			}
+			t.MaxDONDiff = int(tmp)
+		}
+	}
+
+	return nil
+}
+
+// Marshal implements Format.
+func (t *H265) Marshal() (string, map[string]string) {
+	t.mutex.RLock()
+	defer t.mutex.RUnlock()
+
+	fmtp := make(map[string]string)
+	if t.VPS != nil {
+		fmtp["sprop-vps"] = base64.StdEncoding.EncodeToString(t.VPS)
+	}
+	if t.SPS != nil {
+		fmtp["sprop-sps"] = base64.StdEncoding.EncodeToString(t.SPS)
+	}
+	if t.PPS != nil {
+		fmtp["sprop-pps"] = base64.StdEncoding.EncodeToString(t.PPS)
+	}
+	if t.MaxDONDiff != 0 {
+		fmtp["sprop-max-don-diff"] = strconv.FormatInt(int64(t.MaxDONDiff), 10)
+	}
+
+	return "H265/90000", fmtp
+}
+
+// PTSEqualsDTS implements Format.
+func (t *H265) PTSEqualsDTS(*rtp.Packet) bool {
+	return true
+}
+
+// CreateDecoder creates a decoder able to decode the content of the format.
+func (t *H265) CreateDecoder() *rtph265.Decoder {
+	d := &rtph265.Decoder{
+		MaxDONDiff: t.MaxDONDiff,
+	}
+	d.Init()
+	return d
+}
+
+// CreateEncoder creates an encoder able to encode the content of the format.
+func (t *H265) CreateEncoder() *rtph265.Encoder {
+	e := &rtph265.Encoder{
+		PayloadType: t.PayloadTyp,
+		MaxDONDiff:  t.MaxDONDiff,
+	}
+	e.Init()
+	return e
+}
+
+// SafeVPS returns the format VPS.
+func (t *H265) SafeVPS() []byte {
+	t.mutex.RLock()
+	defer t.mutex.RUnlock()
+	return t.VPS
+}
+
+// SafeSPS returns the format SPS.
+func (t *H265) SafeSPS() []byte {
+	t.mutex.RLock()
+	defer t.mutex.RUnlock()
+	return t.SPS
+}
+
+// SafePPS returns the format PPS.
+func (t *H265) SafePPS() []byte {
+	t.mutex.RLock()
+	defer t.mutex.RUnlock()
+	return t.PPS
+}
+
+// SafeSetVPS sets the format VPS.
+func (t *H265) SafeSetVPS(v []byte) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	t.VPS = v
+}
+
+// SafeSetSPS sets the format SPS.
+func (t *H265) SafeSetSPS(v []byte) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	t.SPS = v
+}
+
+// SafeSetPPS sets the format PPS.
+func (t *H265) SafeSetPPS(v []byte) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	t.PPS = v
+}