Refactor: Rename NanoKVM to BatchuKVM and update server URL

server/common/kvm_vision.go

@@ -0,0 +1,111 @@
+package common
+
+/*
+	#cgo CFLAGS: -I../include
+	#cgo LDFLAGS: -L../dl_lib -lkvm
+	#include "kvm_vision.h"
+*/
+import "C"
+import (
+	"sync"
+	"unsafe"
+
+	log "github.com/sirupsen/logrus"
+)
+
+var (
+	kvmVision     *KvmVision
+	kvmVisionOnce sync.Once
+)
+
+type KvmVision struct{}
+
+func GetKvmVision() *KvmVision {
+	kvmVisionOnce.Do(func() {
+		kvmVision = &KvmVision{}
+
+		logLevel := C.uint8_t(0)
+		C.kvmv_init(logLevel)
+		log.Debugf("kvm vision initialized")
+	})
+
+	return kvmVision
+}
+
+func (k *KvmVision) ReadMjpeg(width uint16, height uint16, quality uint16) (data []byte, result int) {
+	var (
+		kvmData  *C.uint8_t
+		dataSize C.uint32_t
+	)
+
+	result = int(C.kvmv_read_img(
+		C.uint16_t(width),
+		C.uint16_t(height),
+		C.uint8_t(0),
+		C.uint16_t(quality),
+		&kvmData,
+		&dataSize,
+	))
+	if result < 0 {
+		log.Errorf("failed to read kvm image: %v", result)
+		return
+	}
+	defer C.free_kvmv_data(&kvmData)
+
+	data = C.GoBytes(unsafe.Pointer(kvmData), C.int(dataSize))
+	return
+}
+
+func (k *KvmVision) ReadH264(width uint16, height uint16, bitRate uint16) (data []byte, result int) {
+	var (
+		kvmData  *C.uint8_t
+		dataSize C.uint32_t
+	)
+
+	result = int(C.kvmv_read_img(
+		C.uint16_t(width),
+		C.uint16_t(height),
+		C.uint8_t(1),
+		C.uint16_t(bitRate),
+		&kvmData,
+		&dataSize,
+	))
+	if result < 0 {
+		log.Errorf("failed to read kvm image: %v", result)
+		return
+	}
+	defer C.free_kvmv_data(&kvmData)
+
+	data = C.GoBytes(unsafe.Pointer(kvmData), C.int(dataSize))
+	return
+}
+
+func (k *KvmVision) SetHDMI(enable bool) int {
+	hdmiEnable := C.uint8_t(0)
+	if enable {
+		hdmiEnable = C.uint8_t(1)
+	}
+
+	result := int(C.kvmv_hdmi_control(hdmiEnable))
+	if result < 0 {
+		log.Errorf("failed to set hdmi to %t", enable)
+		return result
+	}
+
+	return result
+}
+
+func (k *KvmVision) SetGop(gop uint8) {
+	_gop := C.uint8_t(gop)
+	C.set_h264_gop(_gop)
+}
+
+func (k *KvmVision) SetFrameDetect(frame uint8) {
+	_frame := C.uint8_t(frame)
+	C.set_frame_detact(_frame)
+}
+
+func (k *KvmVision) Close() {
+	C.kvmv_deinit()
+	log.Debugf("stop kvm vision...")
+}

server/common/screen.go

@@ -0,0 +1,105 @@
+package common
+
+import "sync"
+
+type Screen struct {
+	Width   uint16
+	Height  uint16
+	FPS     int
+	Quality uint16
+	BitRate uint16
+	GOP     uint8
+}
+
+var (
+	screen     *Screen
+	screenOnce sync.Once
+)
+
+// ResolutionMap height to width
+var ResolutionMap = map[uint16]uint16{
+	1080: 1920,
+	720:  1280,
+	600:  800,
+	480:  640,
+	0:    0,
+}
+
+var QualityMap = map[uint16]bool{
+	100: true,
+	80:  true,
+	60:  true,
+	50:  true,
+}
+
+var BitRateMap = map[uint16]bool{
+	5000: true,
+	3000: true,
+	2000: true,
+	1000: true,
+}
+
+func GetScreen() *Screen {
+	screenOnce.Do(func() {
+		screen = &Screen{
+			Width:   0,
+			Height:  0,
+			Quality: 80,
+			FPS:     30,
+			BitRate: 3000,
+			GOP:     30,
+		}
+	})
+
+	return screen
+}
+
+func SetScreen(key string, value int) {
+	switch key {
+	case "resolution":
+		height := uint16(value)
+		if width, ok := ResolutionMap[height]; ok {
+			screen.Width = width
+			screen.Height = height
+		}
+
+	case "quality":
+		if value > 100 {
+			screen.BitRate = uint16(value)
+		} else {
+			screen.Quality = uint16(value)
+		}
+
+	case "fps":
+		screen.FPS = validateFPS(value)
+
+	case "gop":
+		screen.GOP = uint8(value)
+	}
+}
+
+func CheckScreen() {
+	if _, ok := ResolutionMap[screen.Height]; !ok {
+		screen.Width = 1920
+		screen.Height = 1080
+	}
+
+	if _, ok := QualityMap[screen.Quality]; !ok {
+		screen.Quality = 80
+	}
+
+	if _, ok := BitRateMap[screen.BitRate]; !ok {
+		screen.BitRate = 3000
+	}
+}
+
+func validateFPS(fps int) int {
+	if fps > 60 {
+		return 60
+	}
+	if fps < 10 {
+		return 10
+	}
+
+	return fps
+}