feat: Add image detection bindings

Signed-off-by: Anastassios Nanos <ananos@nubificus.co.uk>

Author: ananos

examples/detect/main.go

@@ -0,0 +1,85 @@
+// SPDX-License-Identifier: Apache-2.0
+
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/nubificus/vaccel-go/vaccel"
+)
+
+func main() {
+
+	if len(os.Args) < 2 || len(os.Args) > 3 {
+		fmt.Printf("Usage: %s <filename> [model]\n", os.Args[0])
+		return
+	}
+
+	image := os.Args[1]
+	imageBytes, e := os.ReadFile(image)
+	if e != nil {
+		fmt.Printf("Error reading file: %s\n", e)
+		os.Exit(1)
+	}
+
+	var outImgName string
+	var session vaccel.Session
+	var resource vaccel.Resource
+
+	err := session.Init(0)
+	if err != vaccel.OK {
+		fmt.Println("error initializing session")
+		os.Exit(err)
+	}
+
+	if len(os.Args) == 3 {
+		model := os.Args[2]
+		err = resource.Init(model, vaccel.ResourceModel)
+		if err != vaccel.OK {
+			fmt.Printf("error initializing resource: %d\n", err)
+			goto ReleaseSession
+		}
+
+		err = session.Register(&resource)
+		if err != vaccel.OK {
+			fmt.Printf("error while registering resource: %d\n", err)
+			goto ReleaseResource
+		}
+	}
+
+	outImgName, err = vaccel.ImageDetectionFromFile(&session, image)
+	if err != vaccel.OK {
+		fmt.Println("Image Detection failed")
+		goto UnregisterResource
+	}
+	fmt.Println("Output(1): ", outImgName)
+
+	outImgName, err = vaccel.ImageDetection(&session, imageBytes)
+	if err != vaccel.OK {
+		fmt.Println("Image Detection failed")
+		goto UnregisterResource
+	}
+	fmt.Println("Output(2): ", outImgName)
+
+UnregisterResource:
+	if len(os.Args) == 3 {
+		err = session.Unregister(&resource)
+		if err != vaccel.OK {
+			fmt.Printf("Could not unregister resource: %d\n", err)
+		}
+	}
+ReleaseResource:
+	if len(os.Args) == 3 {
+		err = resource.Release()
+		if err != vaccel.OK {
+			fmt.Printf("Could not release resource: %d\n", err)
+		}
+	}
+ReleaseSession:
+	err = session.Release()
+	if err != vaccel.OK {
+		fmt.Println("An error occurred while freeing the session")
+		os.Exit(1)
+	}
+}

vaccel/image_detection.go

@@ -0,0 +1,80 @@
+// SPDX-License-Identifier: Apache-2.0
+
+package vaccel
+
+/*
+#cgo pkg-config: vaccel
+#cgo LDFLAGS: -lvaccel -ldl
+#include <vaccel.h>
+
+*/
+import "C"
+import (
+	"fmt"
+	"os"
+	"unsafe"
+)
+
+func ImageDetectionFromFile(sess *Session, imagePath string) (string, int) {
+
+	imageBytes, err := os.ReadFile(imagePath)
+	if err != nil {
+		fmt.Printf("Error reading file: %s\n", err)
+		os.Exit(1)
+	}
+
+	cImageBytes := (*C.uchar)(&imageBytes[0])
+	cImgBuf := unsafe.Pointer(cImageBytes)
+	cImgLen := C.size_t(len(imageBytes))
+
+	cOutImageName := (*C.uchar)(C.malloc(C.size_t(256)))
+
+	/* Free the memory when done */
+	defer C.free(unsafe.Pointer(cOutImageName))
+
+	cRet := C.vaccel_image_detection(
+		sess.cSess, cImgBuf, cOutImageName,
+		cImgLen, C.size_t(256)) //nolint:gocritic
+
+	var golangOut string
+
+	if int(cRet) == 0 {
+		ptr := unsafe.Pointer(cOutImageName)
+		typeCast := (*C.char)(ptr)
+		golangOut = C.GoString(typeCast)
+	} else {
+		golangOut =
+			"A problem occurred while running the Operation"
+	}
+
+	return golangOut, int(cRet)
+}
+
+func ImageDetection(sess *Session, image []byte) (string, int) {
+
+	cImageBytes := (*C.uchar)(&image[0])
+	cImgBuf := unsafe.Pointer(cImageBytes)
+	cImgLen := C.size_t(len(image))
+
+	cOutImageName := (*C.uchar)(C.malloc(C.size_t(256)))
+
+	/* Free the memory when done */
+	defer C.free(unsafe.Pointer(cOutImageName))
+
+	cRet := C.vaccel_image_detection(
+		sess.cSess, cImgBuf, cOutImageName,
+		cImgLen, C.size_t(256)) //nolint:gocritic
+
+	var golangOut string
+
+	if int(cRet) == 0 {
+		ptr := unsafe.Pointer(cOutImageName)
+		typeCast := (*C.char)(ptr)
+		golangOut = C.GoString(typeCast)
+	} else {
+		golangOut =
+			"A problem occurred while running the Operation"
+	}
+
+	return golangOut, int(cRet)
+}