mirror of
https://github.com/amitbet/vnc2video.git
synced 2025-05-04 21:36:21 +00:00
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package vnc2video
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"image"
|
|
"image/draw"
|
|
"vnc2video/logger"
|
|
)
|
|
|
|
type CopyRectEncoding struct {
|
|
SX, SY uint16
|
|
Image draw.Image
|
|
}
|
|
|
|
func (*CopyRectEncoding) Supported(Conn) bool {
|
|
return true
|
|
}
|
|
func (*CopyRectEncoding) Reset() error {
|
|
return nil
|
|
}
|
|
func (*CopyRectEncoding) Type() EncodingType { return EncCopyRect }
|
|
|
|
func (enc *CopyRectEncoding) Read(c Conn, rect *Rectangle) error {
|
|
logger.Debugf("Reading: CopyRect%v", rect)
|
|
if err := binary.Read(c, binary.BigEndian, &enc.SX); err != nil {
|
|
return err
|
|
}
|
|
if err := binary.Read(c, binary.BigEndian, &enc.SY); err != nil {
|
|
return err
|
|
}
|
|
cpyIm := image.NewRGBA(image.Rectangle{Min: image.Point{0, 0}, Max: image.Point{int(rect.Width), int(rect.Height)}})
|
|
for x := 0; x < int(rect.Width); x++ {
|
|
for y := 0; x < int(rect.Height); y++ {
|
|
col := enc.Image.At(x+int(enc.SX), y+int(enc.SY))
|
|
cpyIm.Set(x, y, col)
|
|
}
|
|
}
|
|
|
|
draw.Draw(enc.Image, enc.Image.Bounds(), cpyIm, image.Point{int(rect.X), int(rect.Y)}, draw.Src)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (enc *CopyRectEncoding) Write(c Conn, rect *Rectangle) error {
|
|
if err := binary.Write(c, binary.BigEndian, enc.SX); err != nil {
|
|
return err
|
|
}
|
|
if err := binary.Write(c, binary.BigEndian, enc.SY); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|