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 * src author: <mediapipe-team@google.com> 17*4882a593Smuzhiyun * new author: modified by <rimon.xu@rock-chips.com> 18*4882a593Smuzhiyun * date: 2020-03-23 19*4882a593Smuzhiyun * reference: https://github.com/google/mediapipe 20*4882a593Smuzhiyun */ 21*4882a593Smuzhiyun 22*4882a593Smuzhiyun #ifndef SRC_RT_TASK_TASK_GRAPH_RTTASKNODECONTEXTMANAGER_H_ 23*4882a593Smuzhiyun #define SRC_RT_TASK_TASK_GRAPH_RTTASKNODECONTEXTMANAGER_H_ 24*4882a593Smuzhiyun 25*4882a593Smuzhiyun #include <memory> 26*4882a593Smuzhiyun #include <vector> 27*4882a593Smuzhiyun 28*4882a593Smuzhiyun #include "rt_header.h" 29*4882a593Smuzhiyun #include "rt_metadata.h" 30*4882a593Smuzhiyun #include "RTStreamInfo.h" 31*4882a593Smuzhiyun #include "RTTaskNodeContext.h" 32*4882a593Smuzhiyun 33*4882a593Smuzhiyun class RTTaskNodeContextManager { 34*4882a593Smuzhiyun public: RTTaskNodeContextManager()35*4882a593Smuzhiyun RTTaskNodeContextManager() {} ~RTTaskNodeContextManager()36*4882a593Smuzhiyun virtual ~RTTaskNodeContextManager() {} 37*4882a593Smuzhiyun 38*4882a593Smuzhiyun RT_RET initialize( 39*4882a593Smuzhiyun std::string nodeName, 40*4882a593Smuzhiyun INT32 nodeId, 41*4882a593Smuzhiyun std::vector<RTStreamInfo *> *inputInfo, 42*4882a593Smuzhiyun std::vector<RTStreamInfo *> *outputInfo, 43*4882a593Smuzhiyun bool isParallelRunning); 44*4882a593Smuzhiyun 45*4882a593Smuzhiyun RT_RET prepareForRun(RtMetaData *options); 46*4882a593Smuzhiyun RT_RET cleanupAfterRun(); 47*4882a593Smuzhiyun 48*4882a593Smuzhiyun RTTaskNodeContext *getDefaultNodeContext() const; 49*4882a593Smuzhiyun 50*4882a593Smuzhiyun private: 51*4882a593Smuzhiyun std::unique_ptr<RTTaskNodeContext> mDefaultContext; 52*4882a593Smuzhiyun std::vector<RTStreamInfo *> *mInputInfos; 53*4882a593Smuzhiyun std::vector<RTStreamInfo *> *mOutputInfos; 54*4882a593Smuzhiyun bool mParallelRunning; 55*4882a593Smuzhiyun RtMetaData* mOptions; 56*4882a593Smuzhiyun std::string mNodeName; 57*4882a593Smuzhiyun INT32 mNodeId; 58*4882a593Smuzhiyun }; 59*4882a593Smuzhiyun 60*4882a593Smuzhiyun #endif // SRC_RT_TASK_TASK_GRAPH_RTTASKNODECONTEXTMANAGER_H_ 61*4882a593Smuzhiyun 62