/*
 * @Author: yangzilong
 * @Date: 2021-12-15 17:59:50
 * @Last Modified by: yangzilong
 * @Email: yangzilong@objecteye.com
 * @Description:
 */
#include "stl_aiplatform.h"
#include "MultiSourceProcess.h"
#include "ErrorInfo.h"


int tsl_aiplatform_init(void **handle, tsl_aiplatform_param param)
{
	*handle = new CMultiSourceProcess();
	CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
	return tools->InitAlgorthim(param);
}


#ifdef POST_USE_RABBITMQ
int add_mq_conn(void *handle, mq_type_t tstatus, rabbitmq_conn_params_t mq_conn_param)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	return tools->AddMqConn(tstatus, mq_conn_param);
}

#endif

int get_task_status(void *handle, char *task_id)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	return tools->GetTaskStatus(task_id);
}

int add_task(void *handle, task_param param)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	int error_code = FAILED;
	if (tools->AddTask(param)){
		error_code = SUCCESS;
	}
	return error_code;
}


int pause_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	if (tools->PauseTask(task_id)){
		error_code = SUCCESS;
	}
	return error_code;
}


int restart_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	if (tools->RestartTask(task_id)){
		error_code = SUCCESS;
	}
	return error_code;
}

int finish_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	if (tools->FinishTask(task_id)){
		error_code = SUCCESS;
	}
	return error_code;
}

// added by zsh 220801
int screenshot_task(void *handle, task_param param)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	return tools->SnapShot(param);
}

int tsl_aiplatform_release(void **handle)
{
	if (*handle)
	{
		CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
		tools->CloseAllTask();

		delete tools;
		tools = NULL;
	}
	return SUCCESS;
}

int count_running_task(void *handle)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	return tools->CountRunningTask();
}

void close_all_task(void *handle) 
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	tools->CloseAllTask2();
}

const char* get_tsl_aiplatform_version()
{
	return "sy_stl_aiplatform_version:0.0.0.221202.gpu.x64";
}