230 lines
8.3 KiB
C
230 lines
8.3 KiB
C
|
/*
|
||
|
|
* Copyright 2020 Rockchip Electronics Co. LTD
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*
|
||
|
|
* author: <rimon.xu@rock-chips.com> and <martin.cheng@rock-chips.com>
|
||
|
|
* date: 2020-04-03
|
||
|
|
* module: RTTaskGraph
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef SRC_RT_TASK_APP_GRAPH_RTTASKGRAPH_H_
|
||
|
|
#define SRC_RT_TASK_APP_GRAPH_RTTASKGRAPH_H_
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <memory>
|
||
|
|
#include <map>
|
||
|
|
#include <functional>
|
||
|
|
#include <stdarg.h>
|
||
|
|
|
||
|
|
#include "rt_header.h"
|
||
|
|
#include "rt_metadata.h"
|
||
|
|
#include "RTGraphCommon.h"
|
||
|
|
|
||
|
|
typedef enum ExecutorMode {
|
||
|
|
EXECUTOR_THREAD_POOL,
|
||
|
|
EXECUTOR_THREAD_LOCAL,
|
||
|
|
} RTExecutorMode;
|
||
|
|
|
||
|
|
typedef enum GraphCmd {
|
||
|
|
GRAPH_CMD_PREPARE = 100,
|
||
|
|
GRAPH_CMD_START,
|
||
|
|
GRAPH_CMD_STOP,
|
||
|
|
GRAPH_CMD_RESUME,
|
||
|
|
GRAPH_CMD_PAUSE,
|
||
|
|
GRAPH_CMD_FLUSH,
|
||
|
|
// for graph private command
|
||
|
|
GRAPH_CMD_PRIVATE_CMD,
|
||
|
|
// for task node command
|
||
|
|
GRAPH_CMD_TASK_NODE_PRIVATE_CMD,
|
||
|
|
GRAPH_CMD_MAX,
|
||
|
|
} RTGraphCmd;
|
||
|
|
|
||
|
|
class RTInputStreamManager;
|
||
|
|
class RTOutputStreamManager;
|
||
|
|
class RTTaskNode;
|
||
|
|
class RTGraphInputStream;
|
||
|
|
class RTGraphOutputStream;
|
||
|
|
class RTExecutor;
|
||
|
|
class RTScheduler;
|
||
|
|
class RTMediaBuffer;
|
||
|
|
class RTGraphParser;
|
||
|
|
class RTTaskGraphConfig;
|
||
|
|
class RTTaskGraphStat;
|
||
|
|
class RTTaskGraph {
|
||
|
|
public:
|
||
|
|
explicit RTTaskGraph(const char* tagName);
|
||
|
|
explicit RTTaskGraph(UINT64 uid, const char* tagName);
|
||
|
|
virtual ~RTTaskGraph();
|
||
|
|
|
||
|
|
public:
|
||
|
|
// external common api
|
||
|
|
UINT64 getUid();
|
||
|
|
std::string getName();
|
||
|
|
RT_RET autoBuild(const char* config, RT_BOOL isFileType = RT_TRUE);
|
||
|
|
RT_RET prepare(RtMetaData *params = RT_NULL);
|
||
|
|
RT_RET start();
|
||
|
|
RT_RET stop(RT_BOOL block = RT_TRUE);
|
||
|
|
RT_RET resume();
|
||
|
|
RT_RET pause(RT_BOOL block = RT_FALSE);
|
||
|
|
RT_RET flush();
|
||
|
|
RT_RET reset();
|
||
|
|
RT_RET release();
|
||
|
|
RT_RET invoke(INT32 cmd, RtMetaData *params);
|
||
|
|
RT_RET waitForObservedOutput(INT64 timeoutUs = -1);
|
||
|
|
RT_RET waitUntilDone(INT64 timeoutUs = -1);
|
||
|
|
RT_RET waitUntilEos(INT64 timeoutUs = -1);
|
||
|
|
RT_RET queryStat(RTTaskGraphStat *stat);
|
||
|
|
RT_RET dump();
|
||
|
|
|
||
|
|
RTCBHandle observeOutputStream(
|
||
|
|
const std::string streamName,
|
||
|
|
std::function<RT_RET(RTMediaBuffer *)> streamCallback);
|
||
|
|
RT_RET cancelObserveOutputStream(RTCBHandle handle);
|
||
|
|
|
||
|
|
public:
|
||
|
|
// deprecated api
|
||
|
|
RT_RET observeOutputStream(const std::string& streamName,
|
||
|
|
INT32 streamId,
|
||
|
|
std::function<RT_RET(RTMediaBuffer *)> streamCallback);
|
||
|
|
RT_RET cancelObserveOutputStream(INT32 streamId);
|
||
|
|
RT_RET addSubGraph(const char *graphConfigFile);
|
||
|
|
RT_RET removeSubGraph(const char *graphConfigFile);
|
||
|
|
RT_BOOL hasMirror(INT32 nodeId, RT_BOOL nodeOnly = RT_FALSE);
|
||
|
|
RT_RET setExternalExecutor(RTExecutor *executor);
|
||
|
|
|
||
|
|
public:
|
||
|
|
// external link graph api
|
||
|
|
RTLinkHandle addDownGraph(RTTaskGraph *graph, std::string srcName, std::string dstName);
|
||
|
|
RT_RET removeDownGraph(RTLinkHandle handle);
|
||
|
|
RT_RET setGraphOutputStreamAddMode(RTGraphIOStreamMode mode);
|
||
|
|
|
||
|
|
public:
|
||
|
|
// external link node api
|
||
|
|
RTTaskNode* createNode(std::string nodeConfig, std::string streamConfig);
|
||
|
|
RT_RET linkNode(RTTaskNode *srcNode, RTTaskNode *dstNode);
|
||
|
|
template <class T, class... Args>
|
||
|
|
RT_RET linkNode(T src, T dst, Args... rest) {
|
||
|
|
return linkMultiNode(src, dst, rest...);
|
||
|
|
}
|
||
|
|
RT_RET unlinkNode(RTTaskNode *srcNode, RTTaskNode *dstNode);
|
||
|
|
|
||
|
|
RT_BOOL hasLinkMode(std::string mode);
|
||
|
|
RT_RET selectLinkMode(std::string mode);
|
||
|
|
template <class T, class... Args>
|
||
|
|
RT_RET selectLinkMode(T arg1, T arg2, Args... rest) {
|
||
|
|
return selectMultiMode(arg1, arg2, rest...);
|
||
|
|
}
|
||
|
|
RT_RET clearLinkShips();
|
||
|
|
|
||
|
|
public:
|
||
|
|
// internal api
|
||
|
|
void sendInterrupt(std::string reason);
|
||
|
|
RT_BOOL unthrottleSources();
|
||
|
|
|
||
|
|
protected:
|
||
|
|
RT_RET addPacketToInputStream(std::string streamName, INT64 timeoutUs, RTMediaBuffer *packet);
|
||
|
|
RTCBHandle addOutputStreamPoller(const std::string streamName);
|
||
|
|
RT_RET removeOutputStreamPoller(RTCBHandle handle);
|
||
|
|
RT_RET checkInputStreamFull(std::string streamName);
|
||
|
|
virtual RT_RET onReceiveOutputStreams(RTGraphOutputStream *graphOutputStream) { return RT_OK; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
template <class T, class... Args>
|
||
|
|
RT_RET linkMultiNode(T src, T dst, Args... rest) {
|
||
|
|
RT_RET ret = linkNode(src, dst);
|
||
|
|
if (ret != RT_OK) {
|
||
|
|
RT_LOGE("src node %d dst node %d link failed", src->getID(), dst->getID());
|
||
|
|
} else {
|
||
|
|
linkMultiNode(dst, rest...);
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class T>
|
||
|
|
RT_RET linkMultiNode(T end) {
|
||
|
|
return RT_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class T, class... Args>
|
||
|
|
RT_RET selectMultiMode(T arg1, T arg2, Args... rest) {
|
||
|
|
RT_RET ret = selectLinkMode(arg1);
|
||
|
|
if (ret != RT_OK) {
|
||
|
|
RT_LOGE("select mode %s failed", arg1);
|
||
|
|
} else {
|
||
|
|
selectMultiMode(arg2, rest...);
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class T>
|
||
|
|
RT_RET selectMultiMode(T end) {
|
||
|
|
RT_RET ret = selectLinkMode(end);
|
||
|
|
if (ret != RT_OK) {
|
||
|
|
RT_LOGE("select mode %s failed", end);
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
RT_RET autoLinkSource();
|
||
|
|
RT_RET autoUnlinkSource(INT32 nodeId);
|
||
|
|
RT_RET prepareForRun(RtMetaData *options);
|
||
|
|
RT_RET prepareNodeForRun(RTTaskNode *node, RtMetaData *options);
|
||
|
|
RT_RET cleanupAfterRun();
|
||
|
|
void updateThrottledNodes(RTInputStreamManager* stream, bool *streamWasFull);
|
||
|
|
void updateThrottledNodesNoLock(RTInputStreamManager* stream, bool *streamWasFull);
|
||
|
|
RT_RET buildTaskNode(INT32 pipeId, INT32 nodeId, RTGraphParser* nodeParser);
|
||
|
|
RT_RET buildExecutors(RTGraphParser *parser);
|
||
|
|
RT_RET buildNodes(RTGraphParser *parser);
|
||
|
|
RT_RET buildLinkModes(RTGraphParser *parser);
|
||
|
|
RT_RET buildSubGraph(RTGraphParser *graphParser);
|
||
|
|
RT_RET notifyOutputStreams(RTGraphOutputStream *graphOutputStream);
|
||
|
|
RTTaskNode *createNodeByText(const char *graphConfig);
|
||
|
|
std::vector<std::vector<INT32>> parseLinkShip(std::string linkShip);
|
||
|
|
RT_RET linkNode(INT32 srcNodeId, INT32 dstNodeId);
|
||
|
|
RT_RET unlinkNode(INT32 srcNodeId, INT32 dstNodeId);
|
||
|
|
|
||
|
|
RT_RET setupGraphInputStream(std::string streamName);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
UINT64 mGraphUid;
|
||
|
|
std::string mTagName;
|
||
|
|
RT_BOOL mHasError;
|
||
|
|
RtMutex mFullInputStreamsMutex;
|
||
|
|
RtMutex mErrorMutex;
|
||
|
|
RTScheduler *mScheduler;
|
||
|
|
RtMutex *mLock;
|
||
|
|
RTTaskGraphConfig *mGraphConfig;
|
||
|
|
RTGraphIOStreamMode mIoStreamMode;
|
||
|
|
INT32 mThreadNum;
|
||
|
|
std::map<INT32, RTExecutor *> mExecutors;
|
||
|
|
RTExecutor *mExtExecutor;
|
||
|
|
std::map<INT32/* node id */, RTTaskNode *> mNodes;
|
||
|
|
std::map<INT32, std::vector<RTInputStreamManager *>> mFullInputStreams;
|
||
|
|
std::map<INT32/* node id */, RTInputStreamManager *> mInputManagers;
|
||
|
|
std::map<INT32/* node id */, RTOutputStreamManager *> mOutputManagers;
|
||
|
|
// The graph output streams.
|
||
|
|
std::map<INT32, RTGraphOutputStream *> mGraphOutputStreams;
|
||
|
|
std::map<INT32, RTGraphInputStream *> mGraphInputStreams;
|
||
|
|
// Maps graph input streams to their virtual node ids.
|
||
|
|
std::map<std::string, INT32> mGraphInputStreamIds;
|
||
|
|
std::map<std::string, std::string> mLinkShips;
|
||
|
|
std::vector<std::string> mLinkModes;
|
||
|
|
// Compatible with old interfaces
|
||
|
|
std::map<INT32/* old stream id */, std::string> mOutputStreamCompMaps;
|
||
|
|
std::map<INT32/* old stream id */, RTCBHandle> mGraphOutCompMaps;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // SRC_RT_TASK_APP_GRAPH_RTTASKGRAPH_H_
|