vnc2video/encoding_corre.go

package vnc2video

import (
	"encoding/binary"
	"io"
)

type CoRREEncoding struct {
	numSubRects     uint32
	backgroundColor []byte
	subRectData     []byte
}

func (z *CoRREEncoding) Type() int32 {
	return 4
}

func (z *CoRREEncoding) WriteTo(w io.Writer) (n int, err error) {
	binary.Write(w, binary.BigEndian, z.numSubRects)
	if err != nil {
		return 0, err
	}

	w.Write(z.backgroundColor)
	if err != nil {
		return 0, err
	}

	w.Write(z.subRectData)

	if err != nil {
		return 0, err
	}
	b := len(z.backgroundColor) + len(z.subRectData) + 4
	return b, nil
}
func (z *CoRREEncoding) Read(r Conn, rect *Rectangle) error {
	//func (z *CoRREEncoding) Read(pixelFmt *PixelFormat, rect *Rectangle, r io.Reader) (Encoding, error) {
	bytesPerPixel := int(r.PixelFormat().BPP / 8)
	var numOfSubrectangles uint32
	if err := binary.Read(r, binary.BigEndian, &numOfSubrectangles); err != nil {
		return err
	}

	z.numSubRects = numOfSubrectangles
	var err error
	//read whole-rect background color
	z.backgroundColor, err = ReadBytes(bytesPerPixel, r)
	if err != nil {
		return err
	}

	//read all individual rects (color=BPP + x=16b + y=16b + w=16b + h=16b)
	z.subRectData, err = ReadBytes(int(numOfSubrectangles)*(bytesPerPixel+4), r)
	if err != nil {
		return err
	}

	return nil
}