xref: /OK3568_Linux_fs/external/rockit/tgi/sdk/include/RTAIDetectResults.h (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  * Copyright 2020 Rockchip Electronics Co. LTD
3*4882a593Smuzhiyun  *
4*4882a593Smuzhiyun  * Licensed under the Apache License, Version 2.0 (the "License");
5*4882a593Smuzhiyun  * you may not use this file except in compliance with the License.
6*4882a593Smuzhiyun  * You may obtain a copy of the License at
7*4882a593Smuzhiyun  *
8*4882a593Smuzhiyun  *      http://www.apache.org/licenses/LICENSE-2.0
9*4882a593Smuzhiyun  *
10*4882a593Smuzhiyun  * Unless required by applicable law or agreed to in writing, software
11*4882a593Smuzhiyun  * distributed under the License is distributed on an "AS IS" BASIS,
12*4882a593Smuzhiyun  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*4882a593Smuzhiyun  * See the License for the specific language governing permissions and
14*4882a593Smuzhiyun  * limitations under the License.
15*4882a593Smuzhiyun  *
16*4882a593Smuzhiyun  */
17*4882a593Smuzhiyun #ifndef SRC_RT_MEDIA_AV_FILER_INCLUDE_RTAIDECTRESULTS_H_
18*4882a593Smuzhiyun #define SRC_RT_MEDIA_AV_FILER_INCLUDE_RTAIDECTRESULTS_H_
19*4882a593Smuzhiyun 
20*4882a593Smuzhiyun #include "rt_type.h"
21*4882a593Smuzhiyun #include "RTMediaBuffer.h"
22*4882a593Smuzhiyun 
23*4882a593Smuzhiyun #if defined(__cplusplus)
24*4882a593Smuzhiyun extern "C" {
25*4882a593Smuzhiyun #endif
26*4882a593Smuzhiyun 
27*4882a593Smuzhiyun #define RT_AI_MAX_LEN 16
28*4882a593Smuzhiyun 
29*4882a593Smuzhiyun typedef enum _RTAIDetectType {
30*4882a593Smuzhiyun     RT_AI_DETECT_FACE,
31*4882a593Smuzhiyun     RT_AI_DETECT_FACELANDMARK,
32*4882a593Smuzhiyun     RT_AI_DETECT_BODY,
33*4882a593Smuzhiyun     RT_AI_DETECT_HAND,
34*4882a593Smuzhiyun     RT_AI_DETECT_FINGER,
35*4882a593Smuzhiyun     // add more
36*4882a593Smuzhiyun } RTAIDetectType;
37*4882a593Smuzhiyun 
38*4882a593Smuzhiyun #define VENDOR_RK_ROCKX   "rockx"
39*4882a593Smuzhiyun #define VENDOR_SENSETIME  "st_asteria"
40*4882a593Smuzhiyun 
41*4882a593Smuzhiyun typedef struct _RTAIDetectResults {
42*4882a593Smuzhiyun     /*
43*4882a593Smuzhiyun      * the name of vendor
44*4882a593Smuzhiyun      */
45*4882a593Smuzhiyun     char vendor[RT_AI_MAX_LEN];
46*4882a593Smuzhiyun 
47*4882a593Smuzhiyun     /*
48*4882a593Smuzhiyun      * the version of vendor's ai lib
49*4882a593Smuzhiyun      */
50*4882a593Smuzhiyun     char version[RT_AI_MAX_LEN];
51*4882a593Smuzhiyun 
52*4882a593Smuzhiyun     /*
53*4882a593Smuzhiyun      * describe of private data
54*4882a593Smuzhiyun      */
55*4882a593Smuzhiyun     char describe[RT_AI_MAX_LEN];
56*4882a593Smuzhiyun 
57*4882a593Smuzhiyun     /*
58*4882a593Smuzhiyun      * the sizeof of private data
59*4882a593Smuzhiyun      */
60*4882a593Smuzhiyun     INT32 privSize;
61*4882a593Smuzhiyun 
62*4882a593Smuzhiyun     /*
63*4882a593Smuzhiyun      * the point of private data
64*4882a593Smuzhiyun      */
65*4882a593Smuzhiyun     void *privData;
66*4882a593Smuzhiyun } RTAIDetectResults;
67*4882a593Smuzhiyun 
68*4882a593Smuzhiyun /*
69*4882a593Smuzhiyun  * create an ai detect struct
70*4882a593Smuzhiyun  */
71*4882a593Smuzhiyun RTAIDetectResults* createAIDetectResults();
72*4882a593Smuzhiyun 
73*4882a593Smuzhiyun /*
74*4882a593Smuzhiyun  * destory an ai detect struct
75*4882a593Smuzhiyun  * the user must release the privData by themself
76*4882a593Smuzhiyun  */
77*4882a593Smuzhiyun void  destroyAIDetectResults(RTAIDetectResults* result);
78*4882a593Smuzhiyun 
79*4882a593Smuzhiyun void* getAIDetectResults(RTMediaBuffer* buffer);
80*4882a593Smuzhiyun 
81*4882a593Smuzhiyun #ifdef __cplusplus
82*4882a593Smuzhiyun }  // extern "C"
83*4882a593Smuzhiyun #endif
84*4882a593Smuzhiyun 
85*4882a593Smuzhiyun #endif  // SRC_RT_MEDIA_AV_FILER_INCLUDE_RTAIDECTRESULTS_H_
86*4882a593Smuzhiyun 
87