Hu Chunming / vpt_ascend_arm

Blame view

src/ai_platform/stl_aiplatform.cpp 2.46 KB
Edit Raw Normal View History
09c2d08c   Hu Chunming   arm交付版 
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
 
  /*
   * @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);
  }
7c5cf8c9   Hu Chunming   优化GetTaskStatus接口 
28
29
 
  #endif
09c2d08c   Hu Chunming   arm交付版 
30
31
32
33
34
35
 
  int get_task_status(void *handle, char *task_id)
  {
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
  	return tools->GetTaskStatus(task_id);
  }
09c2d08c   Hu Chunming   arm交付版 
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
 
  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;
  }
294ef5bc   Hu Chunming   修复解码器内因线程退出顺序错误导致... 
98
99
100
101
102
103
104
105
106
107
108
 
  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();
  }
09c2d08c   Hu Chunming   arm交付版 
109
110
111
112
113
 
  
  const char* get_tsl_aiplatform_version()
  {
  	return "sy_stl_aiplatform_version:0.0.0.221202.gpu.x64";
  }