// 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_FEATURE_RETRIVER_H_
#define _RK_AI_FEATURE_RETRIVER_H_

#include <mutex>
#include "ai_task_manager.h"
#include "RTAIGraph.h"

#define PRELOAD_HANDLE_FEATURE   1
#define PRELOAD_HANDLE_AI        1

namespace rockchip {
namespace aiserver {

class AIFeatureRetriver {
 public:
    AIFeatureRetriver();
    ~AIFeatureRetriver();

 public:
    INT32  setup(AITaskManager *taskManager);
    INT32  start();
    INT32  stop();
    INT32  runTaskOnce(void *params);

 private:
    INT32  preload();
    RT_RET ai_feature_output_callback(RTMediaBuffer *buffer);

 private:
    std::mutex     mOpMutex;
    RTAIGraph     *mAIGraph;
    AITaskManager *mAITaskManager;
};

} // namespace aiserver
} // namespace rockchip


#endif // _RK_AI_FEATURE_RETRIVER_H_