go-dmr/bit/bit.go

package bit

import "fmt"

type Bit byte

func (b *Bit) Flip() {
	(*b) ^= 0x01
}

type Bits []Bit

func toBits(b byte) Bits {
	var o = make(Bits, 8)
	for bit, mask := 0, byte(128); bit < 8; bit, mask = bit+1, mask>>1 {
		if b&mask != 0 {
			o[bit] = 1
		}
	}
	return o
}

func NewBits(bytes []byte) Bits {
	var l = len(bytes)
	var o = make(Bits, 0)
	for i := 0; i < l; i++ {
		o = append(o, toBits(bytes[i])...)
	}
	return o
}

func (bits *Bits) Bytes() []byte {
	var l = len(*bits)
	var o = make([]byte, (l+7)/8)
	for i, b := range *bits {
		if b == 0x01 {
			o[i/8] |= (1 << byte(7-(i%8)))
		}
	}
	return o
}

func (bits Bits) Debits() Debits {
	var debits = make(Debits, (len(bits)+1)/2)
	for i := 0; i < len(bits); i += 2 {
		debits[i/2] = Debit((bits[i] << 1) | (bits[i+1]))
	}
	return debits
}

func (bits Bits) Dump() string {
	var (
		s     string
		bytes = bits.Bytes()
	)

	for i, b := range bytes {
		if i%7 == 0 {
			if i != 0 {
				s += "\n"
			}
			s += fmt.Sprintf("%08x  ", i)
		}
		s += fmt.Sprintf("%08b ", b)
	}
	s += "\n"
	return s
}

func (bits Bits) Equal(other Bits) bool {
	var l = bits.Len()
	if l != other.Len() {
		return false
	}
	for i := 0; i < l; i++ {
		if bits[i] != other[i] {
			return false
		}
	}
	return true
}

func (bits Bits) Len() int {
	return len(bits)
}

func (bits Bits) String() string {
	var s = ""
	for _, b := range bits {
		if b == 0x01 {
			s += "1"
		} else {
			s += "0"
		}
	}
	return s
}