#ifndef _TOOL_RKAIQ_MEDIA_H_
#define _TOOL_RKAIQ_MEDIA_H_

#include <errno.h>
#include <linux/videodev2.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <string>

#include "logger/log.h"
#include "mediactl-priv.h"
#include "mediactl.h"
#include "v4l2subdev.h"

#define MAX_MEDIA_NUM 20
#define MAX_CAM_NUM 20
#define SENSOR_ATTACHED_FLASH_MAX_NUM 2

#define VICAP_COMPACT_TEST_ON "echo 1 > /sys/devices/platform/rkcif_mipi_lvds/compact_test"
#define VICAP2_COMPACT_TEST_ON "echo 1 > /sys/devices/platform/rkcif_lite_mipi_lvds/compact_test"

typedef struct {
    int model_idx;
    int linked_sensor;
    std::string sensor_name;
    std::string sensor_subdev_path;
    std::string media_dev_path;
    std::string isp_dev_path;
    std::string csi_dev_path;
    std::string mpfbc_dev_path;
    std::string main_path;
    std::string self_path;
    std::string fbc_path;
    std::string rawwr0_path;
    std::string rawwr1_path;
    std::string rawwr2_path;
    std::string rawwr3_path;
    std::string dma_path;
    std::string rawrd0_m_path;
    std::string rawrd1_l_path;
    std::string rawrd2_s_path;
    std::string stats_path;
    std::string input_params_path;
    std::string mipi_luma_path;
    std::string mipi_dphy_rx_path;
} isp_info_t;

typedef struct {
    int model_idx;
    std::string media_dev_path;
    std::string pp_input_image_path;
    std::string pp_m_bypass_path;
    std::string pp_scale0_path;
    std::string pp_scale1_path;
    std::string pp_scale2_path;
    std::string pp_input_params_path;
    std::string pp_stats_path;
    std::string pp_dev_path;
} ispp_info_t;

typedef struct {
    int model_idx;
    int linked_sensor;
    std::string sensor_name;
    std::string sensor_subdev_path;
    std::string media_dev_path;
    std::string mipi_id0;
    std::string mipi_id1;
    std::string mipi_id2;
    std::string mipi_id3;
    std::string mipi_dphy_rx_path;
    std::string mipi_csi2_sd_path;
    std::string lvds_sd_path;
    std::string mipi_luma_path;
} cif_info_t;

typedef struct {
    int model_idx;
    int linked_sensor;
    std::string sensor_name;
    std::string sensor_subdev_path;
    std::string media_dev_path;
    std::string dvp_id0;
    std::string dvp_id1;
    std::string dvp_id2;
    std::string dvp_id3;
} dvp_info_t;

typedef struct {
    int model_idx;
    std::string sensor_name;
    std::string sensor_subdev_path;
    std::string media_dev_path;
    std::string module_lens_dev_name;
    std::string module_ircut_dev_name;
    int flash_num;
    std::string module_flash_dev_name[SENSOR_ATTACHED_FLASH_MAX_NUM];
    bool fl_strth_adj_sup;
    int flash_ir_num;
    std::string module_flash_ir_dev_name[SENSOR_ATTACHED_FLASH_MAX_NUM];
    bool fl_ir_strth_adj_sup;
} lens_info_t;

typedef struct {
    isp_info_t isp;
    ispp_info_t ispp;
    cif_info_t cif;
    dvp_info_t dvp;
    lens_info_t lens;
} media_info_t;

class RKAiqMedia
{
  public:
    RKAiqMedia();
    virtual ~RKAiqMedia() = default;
    int IsLinkSensor(struct media_device* device, int cam_index);
    std::string GetLinkSensorSubDev(struct media_device* device, int cam_index);
    std::string GetSensorName(struct media_device* device, int cam_index);
    void GetIsppSubDevs(int id, struct media_device* device, const char* devpath);
    void GetIspSubDevs(int id, struct media_device* device, const char* devpath);
    void GetCifSubDevs(int id, struct media_device* device, const char* devpath);
    void GetDvpSubDevs(int id, struct media_device* device, const char* devpath);
    void GetLensSubDevs(int id, struct media_device* device, const char* devpath, int count);
    int GetMediaInfo();
    int GetIspVer();
    int DumpMediaInfo();
    int LinkToIsp(bool enable = false);
    int LinkToSensor(int cam_index);

  public:
    std::string GetSensorName(int id)
    {
        if (media_info[id].isp.linked_sensor) {
            return media_info[id].isp.sensor_name;
        } else if (media_info[id].cif.linked_sensor) {
            return media_info[id].cif.sensor_name;
        } else if (media_info[id].dvp.linked_sensor) {
            return media_info[id].dvp.sensor_name;
        }
        return "";
    }

    media_info_t GetMediaInfoT(int id)
    {
        return media_info[id];
    }

  private:
    media_info_t media_info[MAX_CAM_NUM];
};

#endif // _TOOL_RKAIQ_MEDIA_H_