#include "cuda.h"

GpuMat GpuMat_New() {
    return new cv::cuda::GpuMat();
}

GpuMat GpuMat_NewFromMat(Mat mat) {
    return new cv::cuda::GpuMat(*mat);
}

void GpuMat_Upload(GpuMat m, Mat data, Stream s){
    if (s == NULL) {
        m->upload(*data);
        return;
    }
    m->upload(*data, *s);
}

void GpuMat_Download(GpuMat m, Mat dst, Stream s){
    if (s == NULL) {
        m->download(*dst);
        return;
    }
    m->download(*dst, *s);
}

int GpuMat_Empty(GpuMat m){
    return m->empty();
}

void GpuMat_Close(GpuMat m){
    delete m;
}

void PrintCudaDeviceInfo(int device){
    cv::cuda::printCudaDeviceInfo(device);
}

void PrintShortCudaDeviceInfo(int device){
    cv::cuda::printShortCudaDeviceInfo(device);
}

int GetCudaEnabledDeviceCount(){
    return cv::cuda::getCudaEnabledDeviceCount();
}

void GpuMat_ConvertTo(GpuMat m, GpuMat dst, int type, Stream s) {
    if (s == NULL) {
        m->convertTo(*dst, type);
        return;
    }
    m->convertTo(*dst, type, *s);
}

void GpuMat_CopyTo(GpuMat m, GpuMat dst, Stream s) {
    if (s == NULL) {
        m->copyTo(*dst);
        return;
    }
    m->copyTo(*dst, *s);
}

void GpuMat_Reshape(GpuMat m, int cn, int rows) {
    m->reshape(cn, rows);
}

int GpuMat_Cols(GpuMat m) {
    return m->cols;
}

int GpuMat_Rows(GpuMat m) {
    return m->rows;
}

int GpuMat_Channels(GpuMat m) {
    return m->channels();
}

int GpuMat_Type(GpuMat m) {
    return m->type();
}

Stream Stream_New() {
    return new cv::cuda::Stream();
}

void Stream_Close(Stream s){
    delete s;
}

bool Stream_QueryIfComplete(Stream s) {
    return s->queryIfComplete();
}

void Stream_WaitForCompletion(Stream s) {
    s->waitForCompletion();
}