OpenDiablo2/d2common/stream_writer.go

package d2common

import "bytes"

// StreamWriter allows you to create a byte array by streaming in writes of various sizes
type StreamWriter struct {
	data *bytes.Buffer
}

// CreateStreamWriter creates a new StreamWriter instance
func CreateStreamWriter() *StreamWriter {
	result := &StreamWriter{
		data: new(bytes.Buffer),
	}
	return result
}

// PushByte writes a byte to the stream
func (v *StreamWriter) PushByte(val byte) {
	v.data.WriteByte(val)
}

// PushUint16 writes an uint16 word to the stream
func (v *StreamWriter) PushUint16(val uint16) {
	v.data.WriteByte(byte(val) & 0xFF)
	v.data.WriteByte(byte(val>>8) & 0xFF)
}

// PushInt16 writes a int16 word to the stream
func (v *StreamWriter) PushInt16(val int16) {
	v.data.WriteByte(byte(val) & 0xFF)
	v.data.WriteByte(byte(val>>8) & 0xFF)
}

// PushUint32 writes a uint32 dword to the stream
func (v *StreamWriter) PushUint32(val uint32) {
	v.data.WriteByte(byte(val) & 0xFF)
	v.data.WriteByte(byte(val>>8) & 0xFF)
	v.data.WriteByte(byte(val>>16) & 0xFF)
	v.data.WriteByte(byte(val>>24) & 0xFF)
}

// PushUint64 writes a uint64 qword to the stream
func (v *StreamWriter) PushUint64(val uint64) {
	v.data.WriteByte(byte(val) & 0xFF)
	v.data.WriteByte(byte(val>>8) & 0xFF)
	v.data.WriteByte(byte(val>>16) & 0xFF)
	v.data.WriteByte(byte(val>>24) & 0xFF)
	v.data.WriteByte(byte(val>>32) & 0xFF)
	v.data.WriteByte(byte(val>>40) & 0xFF)
	v.data.WriteByte(byte(val>>48) & 0xFF)
	v.data.WriteByte(byte(val>>56) & 0xFF)
}

// PushInt64 writes a uint64 qword to the stream
func (v *StreamWriter) PushInt64(val int64) {
	v.PushUint64(uint64(val))
}

// GetBytes returns the the byte slice of the underlying data
func (v *StreamWriter) GetBytes() []byte {
	return v.data.Bytes()
}