/*
 * Copyright (c) 2021 Rockchip, Inc. All Rights Reserved.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

#include "rkadk_common.h"
#include "rkadk_media_comm.h"
#include "rkadk_ui.h"
#include "rkadk_disp.h"
#include "rkadk_log.h"
#include "rkadk_param.h"
#include "isp/sample_isp.h"
#include <getopt.h>
#include <signal.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

extern int optind;
extern char *optarg;

static bool is_quit = false;
static RKADK_CHAR optstr[] = "a:I:W:H:p:f:l:d:h";

#define IQ_FILE_PATH "/etc/iqfiles"

static void print_usage(const RKADK_CHAR *name) {
  printf("usage example:\n");
  printf("\t%s [-a /etc/iqfiles] [-I 0]\n", name);
  printf("\t-a: enable aiq with dirpath provided, eg:-a "
         "/oem/etc/iqfiles/, Default /etc/iqfiles,"
         "without this option aiq should run in other application\n");
  printf("\t-I: camera id, Default 0\n");
  printf("\t-p: param ini directory path, Default:/data/rkadk\n");
  printf("\t-W: display width, default rk3576 = 1080, other chip = 720\n");
  printf("\t-H: display height, default rk3576 = 1920, other chip = 1280\n");
  printf("\t-f: display pixel format, option: 0(RGB888), 1(NV12), 2(RGB565), 3(RGB444); Default: 0\n");
  printf("\t-l: ui vo layer id, Default: rk3576 = 5, other chip = 0\n");
  printf("\t-d: ui vo device id, Default: rk3576 = 1, other chip = 0\n");
}

static void sigterm_handler(int sig) {
  fprintf(stderr, "signal %d\n", sig);
  is_quit = true;
}

int main(int argc, char *argv[]) {
  int c, ret;
  RKADK_MW_PTR pUi = NULL;
  RKADK_U32 u32CamId = 0;
  RKADK_U32 u32Size = 0;
  RKADK_U32 u32Width, u32Height;
  RKADK_VOID *pData;
  RKADK_UI_ATTR_S stUiAttr;
  RKADK_UI_FRAME_INFO stUiFrmInfo;
  const char *iniPath = NULL;
  char path[RKADK_PATH_LEN];
  char sensorPath[RKADK_MAX_SENSOR_CNT][RKADK_PATH_LEN];
  int u32VoFormat = -1;

#ifdef RKAIQ
  RKADK_PARAM_FPS_S stFps;
  const char *tmp_optarg = optarg;
  SAMPLE_ISP_PARAM stIspParam;

  memset(&stIspParam, 0, sizeof(SAMPLE_ISP_PARAM));
  stIspParam.iqFileDir = IQ_FILE_PATH;
#endif

  memset(&stUiAttr, 0, sizeof(RKADK_UI_ATTR_S));

#ifdef RK3576
  u32Width = 1080;
  u32Height = 1920;
  stUiAttr.u32VoDev = 1;
  stUiAttr.u32VoLay = 5;
#else
  u32Width = 720;
  u32Height = 1280;
#endif

  while ((c = getopt(argc, argv, optstr)) != -1) {
    switch (c) {
#ifdef RKAIQ
    case 'a':
      if (!optarg && NULL != argv[optind] && '-' != argv[optind][0]) {
        tmp_optarg = argv[optind++];
      }

      if (tmp_optarg)
        stIspParam.iqFileDir = (char *)tmp_optarg;
      break;
#endif
    case 'I':
      u32CamId = atoi(optarg);
      break;
    case 'p':
      iniPath = optarg;
      RKADK_LOGP("iniPath: %s", iniPath);
      break;
    case 'W':
      u32Width = atoi(optarg);
      break;
    case 'H':
      u32Height = atoi(optarg);
      break;
    case 'f':
      u32VoFormat = atoi(optarg);
      break;
    case 'l':
      stUiAttr.u32VoLay = atoi(optarg);
      break;
    case 'd':
      stUiAttr.u32VoDev = atoi(optarg);
      break;
    case 'h':
    default:
      print_usage(argv[0]);
      optind = 0;
      return 0;
    }
  }
  optind = 0;

  RKADK_LOGP("#camera id: %d", u32CamId);

  if (u32VoFormat == 1)
    stUiAttr.enUiVoFormat = VO_FORMAT_NV12;
  else if (u32VoFormat == 2)
    stUiAttr.enUiVoFormat = VO_FORMAT_RGB565;
  else if (u32VoFormat == 3)
    stUiAttr.enUiVoFormat = VO_FORMAT_RGB444;
  else
    stUiAttr.enUiVoFormat = VO_FORMAT_RGB888;

  stUiAttr.u32VoChn = 1;
  stUiAttr.u32DispWidth = u32Width;
  stUiAttr.u32DispHeight = u32Height;
  stUiAttr.u32DispFrmRt = 30;
  stUiAttr.u32ImgWidth = u32Width;
  stUiAttr.u32ImgHeight = u32Height;
  stUiAttr.enVoSpliceMode = SPLICE_MODE_RGA;
#if defined(RV1106_1103) || defined(RV1103B)
  stUiAttr.enUiVoIntfTye = DISPLAY_TYPE_DEFAULT;
#else
  stUiAttr.enUiVoIntfTye = DISPLAY_TYPE_MIPI;
#endif

  stUiFrmInfo.Format = RKADK_FMT_RGBA8888;
  stUiFrmInfo.u32Width = u32Width;
  stUiFrmInfo.u32Height = u32Height;

  u32Size = stUiFrmInfo.u32Width * stUiFrmInfo.u32Height * 4;

  RKADK_MPI_SYS_Init();

  if (iniPath) {
    memset(path, 0, RKADK_PATH_LEN);
    memset(sensorPath, 0, RKADK_MAX_SENSOR_CNT * RKADK_PATH_LEN);
    sprintf(path, "%s/rkadk_setting.ini", iniPath);
    for (int i = 0; i < RKADK_MAX_SENSOR_CNT; i++)
      sprintf(sensorPath[i], "%s/rkadk_setting_sensor_%d.ini", iniPath, i);

    /*
    lg:
      char *sPath[] = {"/data/rkadk/rkadk_setting_sensor_0.ini",
      "/data/rkadk/rkadk_setting_sensor_1.ini", NULL};
    */
    char *sPath[] = {sensorPath[0], sensorPath[1], NULL};

    RKADK_PARAM_Init(path, sPath);
  } else {
    RKADK_PARAM_Init(NULL, NULL);
  }

  ret = RK_MPI_MMZ_Alloc(&stUiFrmInfo.pMblk, u32Size,  RK_MMZ_ALLOC_CACHEABLE);
  if (ret) {
    RKADK_LOGE("alloc failed!");
    return -1;
  }

  pData = RK_MPI_MMZ_Handle2VirAddr(stUiFrmInfo.pMblk);
  memset(pData, 0x00, u32Size);
  RK_MPI_SYS_MmzFlushCache(stUiFrmInfo.pMblk, RK_FALSE);

#ifdef RKAIQ
  stFps.enStreamType = RKADK_STREAM_TYPE_SENSOR;
  ret = RKADK_PARAM_GetCamParam(u32CamId, RKADK_PARAM_TYPE_FPS, &stFps);
  if (ret) {
    RKADK_LOGE("RKADK_PARAM_GetCamParam fps failed");
    return -1;
  }

  stIspParam.WDRMode = RK_AIQ_WORKING_MODE_NORMAL;
  stIspParam.bMultiCam = false;
  stIspParam.fps = stFps.u32Framerate;
  SAMPLE_ISP_Start(u32CamId, stIspParam);
#endif
  ret = RKADK_DISP_Init(u32CamId);
  ret |= RKADK_UI_Create(&stUiAttr, &pUi);
  if (ret) {
    RKADK_LOGE("RKADK_DISP_Init failed(%d)", ret);
#ifdef RKAIQ
    SAMPLE_ISP_Stop(u32CamId);
#endif
    return -1;
  }

  signal(SIGINT, sigterm_handler);
  char cmd[64];
  printf("\n#Usage: input 'quit' to exit programe!\n"
         "peress any other key to quit\n");

  while (!is_quit) {
    fgets(cmd, sizeof(cmd), stdin);
    if (strstr(cmd, "quit") || is_quit) {
      RKADK_LOGP("#Get 'quit' cmd!");
      break;
    } else if (strstr(cmd, "all")){
      memset(pData, 0xff, u32Size);
      RKADK_UI_Update(pUi, &stUiFrmInfo);
    } else if (strstr(cmd, "few")){
      memset(pData, 0x00, u32Size);
      memset(pData, 0xaa, u32Size / 2);
      RKADK_UI_Update(pUi, &stUiFrmInfo);
    }
  }

  RKADK_UI_Destroy(pUi);
  RKADK_DISP_DeInit(u32CamId);
  RK_MPI_MMZ_Free(stUiFrmInfo.pMblk);

#ifdef RKAIQ
  SAMPLE_ISP_Stop(u32CamId);
#endif
  RKADK_MPI_SYS_Exit();
  return 0;
}