hyperstone/bit/reader.go

package bit

import (
	"bufio"
	"bytes"
	"io"
)

// Reader allows for the reading and skipping of bits and bytes.
type Reader interface {
	ReadBits(uint) uint64
	DiscardBits(int)
	ReadByte() byte
	Read([]byte) int
	DiscardBytes(int)
	Err() error
}

// NewReader creates a new bit.Reader for any arbitrary reader.
func NewReader(r io.Reader) Reader {
	br, ok := r.(io.ByteReader)
	if !ok {
		br = bufio.NewReader(r)
	}
	return &streamReader{src: br}
}

// NewByteReader creates a bit.Reader for a static slice of bytes. It's just
// using a bytes.Reader internally.
func NewBytesReader(b []byte) Reader {
	return NewReader(bytes.NewReader(b))
}