mirror of
https://github.com/jwetzell/artnet-go.git
synced 2026-09-11 08:19:28 +00:00
add ArtNzs
This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
package artnet
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
"slices"
|
||||
)
|
||||
|
||||
type ArtNzs struct {
|
||||
ID [8]uint8
|
||||
OpCode uint16
|
||||
ProtVerHi uint8
|
||||
ProtVerLo uint8
|
||||
Sequence uint8
|
||||
StartCode uint8
|
||||
SubUni uint8
|
||||
Net uint8
|
||||
Data []uint8
|
||||
}
|
||||
|
||||
func (an *ArtNzs) GetOpCode() uint16 {
|
||||
return an.OpCode
|
||||
}
|
||||
|
||||
func (an *ArtNzs) GetProtVer() uint16 {
|
||||
return uint16(an.ProtVerHi)<<8 + uint16(an.ProtVerLo)
|
||||
}
|
||||
|
||||
func (an *ArtNzs) GetID() [8]uint8 {
|
||||
return an.ID
|
||||
}
|
||||
|
||||
func (an *ArtNzs) Length() uint16 {
|
||||
return uint16(len(an.Data))
|
||||
}
|
||||
|
||||
func (an *ArtNzs) UnmarshalBinary(data []byte) error {
|
||||
|
||||
if len(data) < 18 {
|
||||
return errors.New("ArtNzs packet must be at least 18 bytes long")
|
||||
}
|
||||
|
||||
copy(an.ID[:], data[0:8])
|
||||
|
||||
if !slices.Equal(ArtNetID[:], an.ID[:]) {
|
||||
return errors.New("ID does not match Art-Net ID")
|
||||
}
|
||||
|
||||
an.OpCode = binary.LittleEndian.Uint16(data[8:10])
|
||||
an.ProtVerHi = data[10]
|
||||
an.ProtVerLo = data[11]
|
||||
|
||||
offset := 12
|
||||
an.Sequence = data[offset]
|
||||
an.StartCode = data[offset+1]
|
||||
an.SubUni = data[offset+2]
|
||||
an.Net = data[offset+3]
|
||||
|
||||
dataLength := int(binary.LittleEndian.Uint16(data[offset+4 : offset+6]))
|
||||
|
||||
if len(data[offset+6:]) < dataLength {
|
||||
return errors.New("[]byte length not long enough to contain data length specified in packet")
|
||||
}
|
||||
an.Data = make([]uint8, dataLength)
|
||||
copy(an.Data, data[offset+6:offset+6+dataLength])
|
||||
return nil
|
||||
}
|
||||
|
||||
func (an *ArtNzs) MarshalBinary() ([]byte, error) {
|
||||
data := make([]byte, 8+8)
|
||||
copy(data[0:8], an.ID[:])
|
||||
binary.LittleEndian.PutUint16(data[8:10], an.OpCode)
|
||||
data[10] = an.ProtVerHi
|
||||
data[11] = an.ProtVerLo
|
||||
data[12] = an.Sequence
|
||||
data[13] = an.StartCode
|
||||
data[14] = an.SubUni
|
||||
data[15] = an.Net
|
||||
dataLength := uint16(len(an.Data))
|
||||
if dataLength > 512 {
|
||||
return nil, errors.New("data length must be less than or equal to 512 bytes")
|
||||
}
|
||||
binary.LittleEndian.PutUint16(data[16:18], dataLength)
|
||||
copy(data[18:], an.Data)
|
||||
return data, nil
|
||||
}
|
||||
Reference in New Issue
Block a user