xref: /OK3568_Linux_fs/external/rockit/tgi/sdk/include/RTAIGraph.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  *     author: <fxw@rock-chips.com>
17*4882a593Smuzhiyun  *       date: 2020-09-03
18*4882a593Smuzhiyun  *     module: RTAIGraph
19*4882a593Smuzhiyun  */
20*4882a593Smuzhiyun 
21*4882a593Smuzhiyun #ifndef SRC_RT_TASK_APP_GRAPH_RTAIGRAPH_H_
22*4882a593Smuzhiyun #define SRC_RT_TASK_APP_GRAPH_RTAIGRAPH_H_
23*4882a593Smuzhiyun 
24*4882a593Smuzhiyun #include "rt_header.h"
25*4882a593Smuzhiyun #include "RTTaskGraph.h"
26*4882a593Smuzhiyun 
27*4882a593Smuzhiyun class RTAIGraph : public RTTaskGraph {
28*4882a593Smuzhiyun  public:
29*4882a593Smuzhiyun     explicit RTAIGraph(const char* tagName);
30*4882a593Smuzhiyun     ~RTAIGraph();
31*4882a593Smuzhiyun 
32*4882a593Smuzhiyun     RT_RET prepare();
33*4882a593Smuzhiyun     RT_RET start();
34*4882a593Smuzhiyun     RT_RET stop();
35*4882a593Smuzhiyun 
36*4882a593Smuzhiyun     RT_RET preload(RtMetaData *meta);
37*4882a593Smuzhiyun     RT_RET recognize(RTMediaBuffer *buffer);
38*4882a593Smuzhiyun     RT_RET observeOutputStream(std::function<RT_RET(RTMediaBuffer *)> streamCallback);
39*4882a593Smuzhiyun 
40*4882a593Smuzhiyun  private:
41*4882a593Smuzhiyun     RT_RET initialize(const char* tagName);
42*4882a593Smuzhiyun     RT_RET deinitialize();
43*4882a593Smuzhiyun };
44*4882a593Smuzhiyun 
45*4882a593Smuzhiyun #endif  // SRC_RT_TASK_APP_GRAPH_RTAIGRAPH_H_
46