gocv.cc

// +build gocv

#include <dlib/opencv.h> 
#include <opencv2/core/core.hpp>
#include <opencv2/core/types_c.h>
#include "facerec.h"
#include "tracker.h"
#include "gocv.h"

using namespace dlib;

facesret* facerec_detect_from_mat(facerec* rec, const void *mat,int type) {
    FaceRec* cls = (FaceRec*)(rec->cls);
	facesret* ret = (facesret*)calloc(1, sizeof(facesret));
	image_t img;

	try {
        cv::Mat *mat_img = (cv::Mat*)mat;
        if (!mat_img) {
            ret->err_code = UNKNOWN_ERROR;
            return ret;
        }
        IplImage ipl_img = cvIplImage(*mat_img);
        if (mat_img->channels() > 1) {
		    cv_image<bgr_pixel> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        } else {
            cv_image<uchar> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        }
	} catch(image_load_error& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = IMAGE_LOAD_ERROR;
		return ret;
	} catch (std::exception& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = UNKNOWN_ERROR;
		return ret;
	}
    
    return cls->detect(ret, img, type);
}

facesret* start_track_from_mat(tracker* tr, const void *mat,int x1, int y1, int x2, int y2) {
    Tracker* cls = (Tracker*)(tr->cls);
	facesret* ret = (facesret*)calloc(1, sizeof(facesret));
	image_t img;

	try {
        cv::Mat *mat_img = (cv::Mat*)mat;
        if (!mat_img) {
            ret->err_code = UNKNOWN_ERROR;
            return ret;
        }
        IplImage ipl_img = cvIplImage(*mat_img);
        if (mat_img->channels() > 1) {
		    cv_image<bgr_pixel> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        } else {
            cv_image<uchar> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        }
        
        cls->StartTrack(img,rectangle(x1,y1,x2,y2));
	} catch(image_load_error& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = IMAGE_LOAD_ERROR;
		return ret;
	} catch (std::exception& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = UNKNOWN_ERROR;
		return ret;
	}
    
    return ret;
}

update_ret* update_track_from_mat(tracker* tr, const void *mat) {
    Tracker* cls = (Tracker*)(tr->cls);
	update_ret* ret = (update_ret*)calloc(1, sizeof(update_ret));
	image_t img;

	try {
        cv::Mat *mat_img = (cv::Mat*)mat;
        if (!mat_img) {
            ret->err_code = UNKNOWN_ERROR;
            return ret;
        }
        IplImage ipl_img = cvIplImage(*mat_img);
        if (mat_img->channels() > 1) {
		    cv_image<bgr_pixel> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        } else {
            cv_image<uchar> dlib_img(&ipl_img);
            assign_image(img, dlib_img);
        }
        
        ret->confidence = cls->Update(img);
	} catch(image_load_error& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = IMAGE_LOAD_ERROR;
		return ret;
	} catch (std::exception& e) {
		ret->err_str = strdup(e.what());
		ret->err_code = UNKNOWN_ERROR;
		return ret;
	}
    
    return ret;
}