// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef _RK_AI_SCENE_DIRECTOR_H_
#define _RK_AI_SCENE_DIRECTOR_H_

#include "dbus_graph_control.h"
#include "ai_feature_retriver.h"
#include "ai_task_manager.h"
#include "shmc/shm_control_uvc.h"
#include "ai_uvc_graph.h"

namespace rockchip {
namespace aiserver {

typedef enum _RockxTaskMode {
    ROCKX_TASK_MODE_SINGLE = 0,
    ROCKX_TASK_MODE_COMPLEX,
    ROCKX_TASK_MODE_MAX,
} RockxTaskMode;

/*
 * 1. run task graph for generic ai scene.
 * 2. output NN vision result to SHMC(ipc)
 * 3. provide a minimalist interface to aiserver
 */
class AISceneDirector : public RTGraphListener {
 public:
    AISceneDirector();
    ~AISceneDirector();

 public:
    int32_t setup();
    int32_t prepareUVCGraph();
    int32_t interrupt();
    int32_t waitUntilDone();

 public:
    // overide RTGraphListener control task graph
    virtual int32_t start(const std::string &appName);
    virtual int32_t stop(const std::string &appName);
    virtual int32_t observeGraphOutput(const std::string &appName, const int32_t &enable);

    virtual int32_t setEPTZ(const AI_UVC_EPTZ_MODE &mode, const int32_t &enabled);
    virtual int32_t setZoom(const double &val);
    virtual int32_t setFaceAE(const int32_t &enabled);
    virtual int32_t setFaceLine(const int32_t &enabled);

    virtual int32_t enableAIAlgorithm(const std::string &type);
    virtual int32_t disableAIAlgorithm(const std::string &type);
    virtual int32_t updateAIAlgorithmParams(const std::string &type);

    virtual int32_t openAIMatting();
    virtual int32_t closeAIMatting();

    virtual int32_t invoke(const std::string &appName, const std::string &actionName, void *params);
    virtual int32_t ctrlSubGraph(const char* nnName, int32_t enable);

 private:
    int32_t invokeFeature(const std::string &actionName, void *params);
    int32_t invokeUVC(const std::string &actionName, void *params);

    RT_RET  nn_data_output_callback(RTMediaBuffer *buffer);
    RT_RET  ai_matting_output_callback(RTMediaBuffer *buffer);
    RT_RET  uvc_data_output_callback(RTMediaBuffer *buffer);

 public:
    ShmUVCController *mUVCController;
    AITaskManager    *mAITaskManager;

 private:
    std::mutex   mOpMutex;
    AIUVCGraph  *mUVCGraph;
    AIFeatureRetriver *mAIFeatureRetriver;
    int32_t      mUVCGraphRef         = 0;
    int32_t      mEnableUVC           = 0;
    int32_t      mEnableNN            = 0;
};

} // namespace aiserver
} // namespace rockchip


#endif // _RK_AI_SCENE_DIRECTOR_H_