package gosdk

/*
#cgo CFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -w -g
#cgo CXXFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -w -g -std=c++11
#cgo LDFLAGS: -L/usr/local/cuda/lib64 -L${SRCDIR}/sdk/face/lib/gpu -L${SRCDIR}/sdk/darknet/lib
#cgo LDFLAGS: -Wl,-rpath,${SRCDIR}/sdk/face/lib/gpu:${SRCDIR}/sdk/darknet/lib
#cgo LDFLAGS: -ldarknet -lTHFaceImage -lTHFeature -lTHFaceProperty -lTHFaceTracking -lcudart -lcublas -lcurand -lrt -ldl -lpthread
#include <stdlib.h>
#include "csdk.h"
*/
import "C"
import (
	"unsafe"
)

// YoloHandle wrap C
type YoloHandle struct {
	handle C.YoloHandle
}

// SDKImage sdk image
type SDKImage struct {
	Data   []byte
	Width  int
	Height int
}

// InitYolo init yolo sdk
func InitYolo(fc, fw, fn string, gi int) *YoloHandle {

	c := C.CString(fc)
	defer C.free(unsafe.Pointer(c))
	w := C.CString(fw)
	defer C.free(unsafe.Pointer(w))
	n := C.CString(fn)
	defer C.free(unsafe.Pointer(n))

	g := C.int(gi)

	p := C.c_api_yolo_init(c, w, n, g)
	return &YoloHandle{p}
}

// InitFaceDetector init face detector
func InitFaceDetector(tm, gi, minFaces, rollAngle int) {

	C.c_api_face_detector_init(C.int(tm), C.int(gi), C.int(minFaces), C.int(rollAngle))
}

// InitFaceProperty init face property
func InitFaceProperty(tm int) {

	C.c_api_face_property_init(C.int(tm))
}

// InitFaceExtractor init face extractor
func InitFaceExtractor(tm, gi int) {
	C.c_api_face_extractor_init(C.int(tm), C.int(gi))
}

// InitFaceTracker init face tracker
func InitFaceTracker(tm, gi, w, h, maxFaces, interval, sample int) {

	C.c_api_face_tracker_init(C.int(tm), C.int(gi), C.int(w), C.int(h), C.int(maxFaces), C.int(interval), C.int(sample))
}

// Free free sdk
func Free() {
	C.c_api_release()
}

// FaceDetect face detect
func FaceDetect(img SDKImage, ch int) []CFacePos {

	data := img.Data
	w := img.Width
	h := img.Height

	var count C.int
	cfpos := C.c_api_face_detect(&count, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
	if cfpos != nil {
		defer C.free(unsafe.Pointer(cfpos))
		return CFacePosArrayToGoArray(unsafe.Pointer(cfpos), int(count))
	}
	return nil
}

// FaceProperty face property
func FaceProperty(fpos CFacePos, img SDKImage, ch int) CThftResult {
	data := img.Data
	w := img.Width
	h := img.Height

	pos := (*C.cFacePos)(unsafe.Pointer(&fpos))
	result := C.c_api_face_property(pos, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))

	return *((*CThftResult)(unsafe.Pointer(&result)))
}

// FaceExtract face extract feature
func FaceExtract(fpos CFacePos, img SDKImage, ch int) []byte {

	data := img.Data
	w := img.Width
	h := img.Height

	var featLen C.int
	pos := (*C.cFacePos)(unsafe.Pointer(&fpos))

	p := C.c_api_face_extract(&featLen, pos, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
	defer C.free(unsafe.Pointer(p))
	b := C.GoBytes(unsafe.Pointer(p), featLen)
	return b
}

// FaceCompare face compare
func FaceCompare(feat1 []byte, feat2 []byte) float32 {
	res := C.c_api_face_compare((*C.uchar)(unsafe.Pointer(&feat1[0])), (*C.uchar)(unsafe.Pointer(&feat2[0])))
	return float32(res)
}

// FaceTrackOnly face tracker face only
func FaceTrackOnly(img SDKImage, ch int) []CRECT {
	data := img.Data
	w := img.Width
	h := img.Height

	var fCount C.int
	rect := C.c_api_face_track_only(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
	if rect != nil {
		defer C.free(unsafe.Pointer(rect))
		return CRECTArrayToGoArray(unsafe.Pointer(rect), int(fCount))
	}
	return nil
}

// FaceTrackDetect face tracker face detect
func FaceTrackDetect(img SDKImage, ch int) []CFaceInfo {
	data := img.Data
	w := img.Width
	h := img.Height

	var fCount C.int
	finfo := C.c_api_face_track_only(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
	if finfo != nil {
		defer C.free(unsafe.Pointer(finfo))
		return CFaceInfoArrayToGoArray(unsafe.Pointer(finfo), int(fCount))
	}
	return nil
}

// FaceTrackingInfo face track info
type FaceTrackingInfo struct {
	tracking bool
	faces    []CFaceInfo
}

// FaceTrack face tracking info
func FaceTrack(res map[int64]FaceTrackingInfo, img SDKImage, ch int) {
	data := img.Data
	w := img.Width
	h := img.Height

	var fCount C.int
	cFinfo := C.c_api_face_track(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
	if cFinfo == nil {
		return
	}
	defer C.free(unsafe.Pointer(cFinfo))
	goFinfo := CFaceInfoArrayToGoArray(unsafe.Pointer(cFinfo), int(fCount))

	// 空，添加
	if len(res) == 0 {
		for _, v := range goFinfo {
			var faces []CFaceInfo
			faces = append(faces, v)
			tInfo := FaceTrackingInfo{true, faces}
			res[v.NFaceID] = tInfo
		}
		// 非空
	} else {
		// 默认所有faces没有tracking
		for _, v := range res {
			v.tracking = false
		}
		// 查找是否存在tracking
		for _, v := range goFinfo {
			// 存在已有信息
			if info, ok := res[v.NFaceID]; ok {
				// tracking到face
				info.tracking = true
				// 如果tracking长度大于100，删除一些
				if len(info.faces) > 100 {
					info.faces = info.faces[1:]
				}
				info.faces = append(info.faces, v)
				// 不存在，新face，加入
			} else {
				var faces []CFaceInfo
				faces = append(faces, v)
				tInfo := FaceTrackingInfo{true, faces}
				res[v.NFaceID] = tInfo
			}
		}
		// 删除不再tracking的faces
		for k, v := range res {
			if !v.tracking {
				delete(res, k)
			}
		}
	}
}

// YoloDetect yolo detect
func YoloDetect(handle *YoloHandle, img SDKImage, thrsh float32, umns int) []CObjInfo {

	data := img.Data
	w := img.Width
	h := img.Height

	var count C.int

	cobjinfo := C.c_api_yolo_detect(handle.handle, &count, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.float(thrsh), C.int(umns))

	if cobjinfo != nil {
		defer C.free(unsafe.Pointer(cobjinfo))
		return CYoloObjInfoArrayToGoArray(unsafe.Pointer(cobjinfo), int(count))
	}
	return nil
}

// YoloObjName obj name by type
func YoloObjName(typ int) string {
	p := C.c_api_yolo_obj_name(C.int(typ))

	return C.GoString(p)
}