#include <stdio.h>
#include <unistd.h>
#include <cerrno>
#include <cstring>
#include <stdint.h>
#include <cstdlib>
#include <RTTaskNodeContext.h>
#include <RTTaskNode.h>
#include <RTMediaBuffer.h>

#define kStubRockitDemo  MKTAG('r', 'k', 'd', 'm')

// 创建外部节点， 外部节点需要继承RTTaskNode
// 基础接口需要完成open/process/close
class RTRockitDemoNode : public RTTaskNode {
 public:
    RTRockitDemoNode() {}
    virtual ~RTRockitDemoNode() {}

    virtual RT_RET open(RTTaskNodeContext *context) { return RT_OK; }
    virtual RT_RET process(RTTaskNodeContext *context);
    virtual RT_RET close(RTTaskNodeContext *context) { return RT_OK; }

 private:
    RT_RET doAIProcess(RTMediaBuffer *src, RTMediaBuffer *dst);
};

// 用于节点创建， 该函数指针将存于RTNodeStub.mCreateObj中
static RTTaskNode* createRockitDemoNode() {
    return new RTRockitDemoNode();
}

RT_RET RTRockitDemoNode::doAIProcess(RTMediaBuffer *src, RTMediaBuffer *dst) {
    return RT_OK;
}

// 节点处理函数，将输入的数据处理， 然后输出到下级节点
RT_RET RTRockitDemoNode::process(RTTaskNodeContext *context) {
    RTMediaBuffer *inputBuffer  = RT_NULL;
    RTMediaBuffer *outputBuffer = RT_NULL;

    // 判断输入是否为空
    while (!context->inputIsEmpty()) {
        // 取出输入的buffer
        inputBuffer = context->dequeInputBuffer();
        // 取出一块未被使用的输出Buffer
        outputBuffer = context->dequeOutputBuffer(RT_TRUE, inputBuffer->getLength());
        // 执行处理函数
        doAIProcess(inputBuffer, outputBuffer);
        // 设置输出buffer的范围
        outputBuffer->setRange(0, inputBuffer->getLength());
        // 标记EOS
        outputBuffer->getMetaData()->setInt32(kKeyFrameEOS, 1);
        // 输入Buffer使用完成，调用释放
        inputBuffer->release();
        // 将输出buffer带出，完成处理流程
        context->queueOutputBuffer(outputBuffer);
    }

    return RT_OK;
}

//节点信息存根， 用于完成节点注册
RTNodeStub node_stub_rockit_demo {
    // 节点uid， 节点的唯一标识符 (0~1000)
    .mUid          = kStubRockitDemo,
    // 节点名， 主要用于节点查找、创建
    // corp_role_name，命名保证唯一
    .mName         = "rockit_demo",
    // 版本号
    .mVersion      = "v1.0",
    // 节点创建方法; 改成宏定义
    .mCreateObj    = createRockitDemoNode,
    .mCapsSrc      = { "video/x-raw", RT_PAD_SRC, RT_MB_TYPE_VFRAME, {RT_NULL, RT_NULL} },
    .mCapsSink     = { "video/x-raw", RT_PAD_SINK, RT_MB_TYPE_VFRAME, {RT_NULL, RT_NULL} },
};

RT_NODE_FACTORY_REGISTER_STUB(node_stub_rockit_demo);