stl_aiplatform.cpp
2.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
* @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);
}
int get_task_status(void *handle, char *task_id)
{
CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
return tools->GetTaskStatus(task_id);
}
#endif
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";
}