Add wav backchannel (not used yet)

This commit is contained in:
Alex X
2025-04-21 20:09:04 +03:00
parent d0c3cb066c
commit 7fe23c7bc5
2 changed files with 117 additions and 0 deletions

67
pkg/wav/backchannel.go Normal file
View File

@@ -0,0 +1,67 @@
package wav
import (
"github.com/AlexxIT/go2rtc/pkg/core"
"github.com/AlexxIT/go2rtc/pkg/shell"
"github.com/pion/rtp"
)
type Backchannel struct {
core.Connection
cmd *shell.Command
}
func NewBackchannel(cmd *shell.Command) (core.Producer, error) {
medias := []*core.Media{
{
Kind: core.KindAudio,
Direction: core.DirectionSendonly,
Codecs: []*core.Codec{
//{Name: core.CodecPCML},
{Name: core.CodecPCMA},
{Name: core.CodecPCMU},
},
},
}
return &Backchannel{
Connection: core.Connection{
ID: core.NewID(),
FormatName: "wav",
Protocol: "pipe",
Medias: medias,
Transport: cmd,
},
cmd: cmd,
}, nil
}
func (c *Backchannel) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver, error) {
return nil, core.ErrCantGetTrack
}
func (c *Backchannel) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiver) error {
wr, err := c.cmd.StdinPipe()
if err != nil {
return err
}
b := Header(track.Codec)
if _, err = wr.Write(b); err != nil {
return err
}
sender := core.NewSender(media, track.Codec)
sender.Handler = func(packet *rtp.Packet) {
if n, err := wr.Write(packet.Payload); err != nil {
c.Send += n
}
}
sender.HandleRTP(track)
c.Senders = append(c.Senders, sender)
return nil
}
func (c *Backchannel) Start() error {
return c.cmd.Run()
}

50
pkg/wav/wav.go Normal file
View File

@@ -0,0 +1,50 @@
package wav
import (
"encoding/binary"
"github.com/AlexxIT/go2rtc/pkg/core"
)
func Header(codec *core.Codec) []byte {
var fmt, size, extra byte
switch codec.Name {
case core.CodecPCML:
fmt = 1
size = 2
case core.CodecPCMA:
fmt = 6
size = 1
extra = 2
case core.CodecPCMU:
fmt = 7
size = 1
extra = 2
default:
return nil
}
channels := byte(codec.Channels)
if channels == 0 {
channels = 1
}
b := make([]byte, 0, 46) // cap with extra
b = append(b, "RIFF\xFF\xFF\xFF\xFFWAVEfmt "...)
b = append(b, 0x10+extra, 0, 0, 0)
b = append(b, fmt, 0)
b = append(b, channels, 0)
b = binary.LittleEndian.AppendUint32(b, codec.ClockRate)
b = binary.LittleEndian.AppendUint32(b, uint32(size*channels)*codec.ClockRate)
b = append(b, size*channels, 0)
b = append(b, size*8, 0)
if extra > 0 {
b = append(b, 0, 0) // ExtraParamSize (if PCM, then doesn't exist)
}
b = append(b, "data\xFF\xFF\xFF\xFF"...)
return b
}