// 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_DBUS_GRAPH_CONTROL_H_
|
#define _RK_DBUS_GRAPH_CONTROL_H_
|
|
#include <assert.h>
|
#include <fcntl.h>
|
#include <stdio.h>
|
#include <stdlib.h>
|
#include <unistd.h>
|
|
#include <memory>
|
|
#include "dbus_dispatcher.h"
|
#include "ai_uvc_graph.h"
|
|
#define RT_APP_UVC "app_uvc"
|
#define RT_APP_NN "app_nn"
|
#define RT_APP_AI_FEATURE "app_feature"
|
#define RT_ACTION_CONFIG_CAMERA "updateCameraParams"
|
#define RT_ACTION_CONFIG_ENCODER "updateEncoderParams"
|
#define RT_ACTION_RETRIVE_FEATURE "retriveAIFeature"
|
|
namespace rockchip {
|
namespace aiserver {
|
|
class RTGraphListener {
|
public:
|
virtual ~RTGraphListener() {}
|
virtual int32_t start(const std::string &appName) = 0;
|
virtual int32_t stop(const std::string &appName) = 0;
|
virtual int32_t observeGraphOutput(const std::string &appName, const int32_t &enable) = 0;
|
|
virtual int32_t setEPTZ(const AI_UVC_EPTZ_MODE &mode, const int32_t &enabled) = 0;
|
virtual int32_t setZoom(const double &val) = 0;
|
virtual int32_t setFaceAE(const int32_t &enabled) = 0;
|
virtual int32_t setFaceLine(const int32_t &enabled) = 0;
|
virtual int32_t enableAIAlgorithm(const std::string &type) = 0;
|
virtual int32_t disableAIAlgorithm(const std::string &type) = 0;
|
virtual int32_t updateAIAlgorithmParams(const std::string &type) = 0;
|
virtual int32_t openAIMatting() = 0;
|
virtual int32_t closeAIMatting() = 0;
|
|
virtual int32_t invoke(const std::string &appName, const std::string &actionName, void *params) = 0;
|
virtual int32_t ctrlSubGraph(const char* nnName, int32_t enable) = 0;
|
};
|
|
class DBusGraphControl : public control::graph_adaptor,
|
public DBus::IntrospectableAdaptor,
|
public DBus::ObjectAdaptor {
|
public:
|
DBusGraphControl() = delete;
|
DBusGraphControl(DBus::Connection &connection, RTGraphListener* listener);
|
virtual ~DBusGraphControl();
|
|
public:
|
int32_t Start(const std::string &appName);
|
int32_t Stop(const std::string &appName);
|
int32_t SetGraphOutputObserver(const std::string &appName, const int32_t &enabled);
|
|
// UVC
|
int32_t EnableEPTZ(const int32_t &enabled);
|
int32_t SetZoom(const double &val);
|
int32_t EnableFaceAE(const int32_t &enabled);
|
int32_t EnableFaceLine(const int32_t &enabled);
|
|
// AI
|
int32_t EnableAIAlgorithm(const std::string &type);
|
int32_t DisableAIAlgorithm(const std::string &type);
|
int32_t UpdateAIAlgorithmParams(const std::string &cmdName);
|
|
int32_t OpenAIMatting(const std::string &type);
|
int32_t CloseAIMatting(const std::string &type);
|
// Extension
|
int32_t Invoke(const std::string &appName, const std::string &actionName,
|
const int32_t &ext1, const int64_t &ext2);
|
// Old API
|
int32_t SetRockxStatus(const std::string &nnName);
|
int32_t SetNpuCtlStatus(const std::string &cmdName);
|
|
private:
|
RTGraphListener* mGraphListener;
|
};
|
|
} // namespace aiserver
|
} // namespace rockchip
|
|
#endif // _RK_DBUS_GRAPH_CONTROL_H_
|
