Пример #1
0
func TestBitReaderBits(t *testing.T) {
	buf := bytes.NewReader([]byte{9, 129, 128 + 32})
	exp := int64((9 << 11) | (129 << 3) | 5)

	br := binary.BitReader{Inner: buf}
	i, err := br.ReadBits(19)
	if err != nil {
		t.Error(err)
	} else if i != exp {
		t.Errorf("Expected %v, but got %v", exp, i)
	}
}
Пример #2
0
func TestBitReader(t *testing.T) {
	buf := bytes.NewReader([]byte{9, 129})
	exp := []bool{
		false, false, false, false, true, false, false, true,
		true, false, false, false, false, false, false, true,
	}

	br := binary.BitReader{Inner: buf}
	for i, e := range exp {
		b, err := br.ReadBit()
		if err != nil {
			t.Errorf("Test %d errored: %v", i, err)
		} else if b != e {
			t.Errorf("Test %d expected %v, but got %v", i, e, b)
		}
	}
	if _, err := br.ReadBit(); err != io.EOF {
		t.Errorf("Expected an EOF error but didn't get one")
	}

}
Пример #3
0
func (M Golomb) Read(br *binary.BitReader) (int, error) {
	q, r := 0, 0
	for {
		b, err := br.ReadBit()
		if err != nil {
			return 0, err
		}
		if b {
			q++
		} else {
			break
		}
	}
	bl := bitLen(int(M))
	cutoff := (2 << uint32(bl)) - int(M)

	for i := 0; i < bl-1; i++ {
		r <<= 1
		b, err := br.ReadBit()
		if err != nil {
			return 0, err
		}
		if b {
			r++
		}
	}
	if r >= cutoff {
		r <<= 1
		b, err := br.ReadBit()
		if err != nil {
			return 0, err
		}
		if b {
			r++
		}
	}
	log.Println(q, r)
	return q*int(M) + r, nil
}