/*
 * 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_log.h"
#include "rkadk_param.h"
#include "rkadk_photo.h"
#include "rkadk_osd.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:p:m:o:W:H:i:t: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] [-t NV12]\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-o: osd file, ARGB8888 fmt, Default:NULL\n");
  printf("\t-W: osd width or input file data width, Default: 1920\n");
  printf("\t-H: osd height or input file data height, Default: 1080\n");
  printf("\t-i: input file, Default:null\n");
  printf("\t-t: input file data type, default NV12, options: NV12, RGB565, RGBA8888, BGRA8888\n");
  printf("\t-m: multiple sensors, Default:0, options: 1(all isp sensors), 2(isp+ahd sensors)\n");
}

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

static void PhotoDataRecv(RKADK_PHOTO_RECV_DATA_S *pstData) {
  static FILE *file = NULL;
  static RKADK_U64 write_len = 0;
  static RKADK_U32 photoId = 0;
  static char jpegPath[128];

  if (!pstData || !pstData->pu8DataBuf) {
    RKADK_LOGE("Invalid photo data");
    return;
  }

  if (file == NULL) {
    memset(jpegPath, 0, 128);
    sprintf(jpegPath, "/tmp/PhotoTest_%d.jpeg", photoId);
    file = fopen(jpegPath, "w");
    if (!file) {
      RKADK_LOGE("Create jpeg file(%s) failed", jpegPath);
      return;
    }

    RKADK_LOGP("%s: save u32CamId[%d] jpeg to %s", __func__, pstData->u32CamId, jpegPath);
  }

  fwrite(pstData->pu8DataBuf, 1, pstData->u32DataLen, file);
  write_len += pstData->u32DataLen;

  if (pstData->bStreamEnd) {
    RKADK_LOGP("%s: Close file(%s), write len: %lld", __func__, jpegPath, write_len);
    fflush(file);
    fclose(file);
    file = NULL;
    write_len = 0;

    photoId++;
    if (photoId > 10)
      photoId = 0;
  }
}

static int PostIspCallback(RK_VOID *pParam, RK_VOID *pPrivateData) {
  int ret = 0;

#ifdef RKAIQ
  RKADK_U32 u32CamId = (RKADK_U32)pPrivateData;
  rk_ainr_param *pAinrParam = (rk_ainr_param *)pParam;

  if (pAinrParam == RK_NULL) {
    RKADK_LOGE("pAinrParam is nullptr!");
    return -1;
  }

  memset(pAinrParam, 0, sizeof(rk_ainr_param));
  ret = SAMPLE_ISP_GetAINrParams(u32CamId, pAinrParam);
  if (ret) {
    RKADK_LOGE("u32CamId[%d] can't get ainr param!", u32CamId);
    return ret;
  }

  RKADK_LOGD("aiisp cam %d enable %d", u32CamId, ((rk_ainr_param *)pAinrParam)->enable);
#else
  RKADK_LOGW("Don't enable aiq");
#endif

  return ret;
}

int main(int argc, char *argv[]) {
  int c, ret, inCmd = 0;
  RKADK_U32 u32CamId = 0;
  RKADK_PHOTO_ATTR_S stPhotoAttr;
  RKADK_TAKE_PHOTO_ATTR_S stTakePhotoAttr;
  const char *iniPath = NULL;
  char path[RKADK_PATH_LEN];
  char sensorPath[RKADK_MAX_SENSOR_CNT][RKADK_PATH_LEN];
  RKADK_MW_PTR pHandle = NULL, pHandle1 = NULL;
  RKADK_BOOL bMultiCam = RKADK_FALSE;
  RKADK_BOOL bMultiSensor = RK_FALSE;
  char *osdfile = NULL;
  RKADK_U32 u32OsdWidth = 1920, u32OsdHeight = 1080;
  RKADK_OSD_ATTR_S OsdAttr;
  RKADK_OSD_STREAM_ATTR_S OsdStreamAttr;
  RKADK_U32 u32OsdId = 0;
  RKADK_U32 u32SliceHeight;
  RKADK_PARAM_INPUT_FMT_S stInputFmt;

  //aiisp
  bool bAiispEnable = true;
  RKADK_POST_ISP_ATTR_S stPostIspAttr;

  RKADK_CHAR *pInuptPath = NULL;
  const char *postfix = "nv12";
  char filePath[RKADK_MAX_FILE_PATH_LEN];
  RKADK_PHOTO_DATA_ATTR_S stDataAttr;

  memset(&stDataAttr, 0, sizeof(RKADK_PHOTO_DATA_ATTR_S));
  stDataAttr.enType = RKADK_THUMB_TYPE_NV12;
  stDataAttr.u32Width = 1920;
  stDataAttr.u32Height = 1080;
  stDataAttr.u32VirWidth = 1920;
  stDataAttr.u32VirHeight = 1080;
  stDataAttr.s32VdecChn = -1;
  stDataAttr.s32VpssGrp = -1;
  stDataAttr.s32VpssChn = -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

  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 'm':
      inCmd = atoi(optarg);
      if (inCmd == 1) {
        bMultiCam = RKADK_TRUE;
        bMultiSensor = RKADK_TRUE;
      } else if (inCmd == 2)
        bMultiSensor = RKADK_TRUE;
      break;
    case 'o':
      osdfile = optarg;
      break;
    case 'W':
      u32OsdWidth = atoi(optarg);
      stDataAttr.u32Width = u32OsdWidth;
      stDataAttr.u32VirWidth = u32OsdWidth;
      break;
    case 'H':
      u32OsdHeight = atoi(optarg);
      stDataAttr.u32Height = u32OsdHeight;
      stDataAttr.u32VirHeight = u32OsdHeight;
      break;
    case 't':
      if (strstr(optarg, "NV12")) {
        stDataAttr.enType = RKADK_THUMB_TYPE_NV12;
        postfix = "yuv";
      } else if (strstr(optarg, "RGB565")) {
        stDataAttr.enType = RKADK_THUMB_TYPE_RGB565;
        postfix = "rgb565";
      } else if (strstr(optarg, "RGBA8888")) {
        stDataAttr.enType = RKADK_THUMB_TYPE_RGBA8888;
        postfix = "rgba8888";
      } else if (strstr(optarg, "BGRA8888")) {
        stDataAttr.enType = RKADK_THUMB_TYPE_BGRA8888;
        postfix = "bgra8888";
      }
      break;
    case 'i':
      pInuptPath = optarg;
      break;
    case 'h':
    default:
      print_usage(argv[0]);
      optind = 0;
      return 0;
    }
  }
  optind = 0;

  if (bMultiSensor)
    u32CamId = 0;

  RKADK_LOGP("#camera id: %d, bMultiCam: %d, bMultiSensor: %d", u32CamId, bMultiCam, bMultiSensor);
  RKADK_LOGP("osdfile: %s, width: %d, height: %d", osdfile, u32OsdWidth, u32OsdHeight);

  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);
  }

  if (pInuptPath) {
    memset(filePath, 0, RKADK_MAX_FILE_PATH_LEN);
    sprintf(filePath, "/tmp/photo_data.%s", postfix);

    if (!RKADK_PHOTO_GetData(pInuptPath, &stDataAttr)) {
      RKADK_LOGP("[%d, %d, %d, %d], u32BufSize: %d", stDataAttr.u32Width,
                 stDataAttr.u32Height, stDataAttr.u32VirWidth,
                 stDataAttr.u32VirHeight, stDataAttr.u32BufSize);

      FILE *file = fopen(filePath, "w");
      if (!file) {
        RKADK_LOGE("Create file(%s) failed", filePath);
      } else {
        fwrite(stDataAttr.pu8Buf, 1, stDataAttr.u32BufSize, file);
        fclose(file);
        RKADK_LOGP("Save %s done", filePath);
      }

      RKADK_PHOTO_FreeData(&stDataAttr);
    }

    RKADK_MPI_SYS_Exit();
    return 0;
  }

  //aiisp param init
  memset(&stPostIspAttr, 0, sizeof(RKADK_POST_ISP_ATTR_S));
  stPostIspAttr.pModelFilePath = "/oem/usr/lib/";
  stPostIspAttr.stAiIspCallback.pPrivateData = (void *)u32CamId;
  stPostIspAttr.stAiIspCallback.pfUpdateCallback = PostIspCallback;
  stPostIspAttr.u32FrameBufCnt = 2;

photo:
#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 u32CamId[%d] fps failed", u32CamId);
    return -1;
  }

  stIspParam.WDRMode = RK_AIQ_WORKING_MODE_NORMAL;
  stIspParam.bMultiCam = bMultiCam;
  stIspParam.fps = stFps.u32Framerate;
  SAMPLE_ISP_Start(u32CamId, stIspParam);
  RKADK_BUFINFO("isp[%d] init", u32CamId);

  if (bMultiCam) {
    ret = RKADK_PARAM_GetCamParam(1, RKADK_PARAM_TYPE_FPS, &stFps);
    if (ret) {
      RKADK_LOGE("RKADK_PARAM_GetCamParam u32CamId[1] fps failed");
      SAMPLE_ISP_Stop(u32CamId);
      return -1;
    }

    SAMPLE_ISP_Start(1, stIspParam);
    RKADK_BUFINFO("isp[1] init");
  }
#endif

  memset(&stTakePhotoAttr, 0, sizeof(RKADK_TAKE_PHOTO_ATTR_S));
  stTakePhotoAttr.enPhotoType = RKADK_PHOTO_TYPE_SINGLE;

  memset(&stPhotoAttr, 0, sizeof(RKADK_PHOTO_ATTR_S));

  stPhotoAttr.u32CamId = u32CamId;
  stPhotoAttr.pfnPhotoDataProc = PhotoDataRecv;
  stPhotoAttr.stThumbAttr.bSupportDCF = RKADK_FALSE;
  stPhotoAttr.stThumbAttr.stMPFAttr.eMode = RKADK_PHOTO_MPF_SINGLE;
  stPhotoAttr.stThumbAttr.stMPFAttr.sCfg.u8LargeThumbNum = 1;
  stPhotoAttr.stThumbAttr.stMPFAttr.sCfg.astLargeThumbSize[0].u32Width = 320;
  stPhotoAttr.stThumbAttr.stMPFAttr.sCfg.astLargeThumbSize[0].u32Height = 180;
  stPhotoAttr.pstPostIspAttr = &stPostIspAttr;

  //fbc0 format change
  stPhotoAttr.stFmtChange.u32VencChn = 12;
  stPhotoAttr.stFmtChange.u32VdecChn = 2;

  ret = RKADK_PHOTO_Init(&stPhotoAttr, &pHandle);
  if (ret) {
    RKADK_LOGE("RKADK_PHOTO_Init u32CamId[%d] failed[%d]", u32CamId, ret);
#ifdef RKAIQ
    SAMPLE_ISP_Stop(u32CamId);

    if (bMultiCam)
      SAMPLE_ISP_Stop(1);
#endif
    return -1;
  }

  if (bMultiSensor) {
    stPhotoAttr.u32CamId = 1;
    ret = RKADK_PHOTO_Init(&stPhotoAttr, &pHandle1);
    if (ret) {
      RKADK_LOGE("RKADK_PHOTO_Init u32CamId[1] failed[%d]", ret);
      RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
      SAMPLE_ISP_Stop(u32CamId);
      if (bMultiCam)
        SAMPLE_ISP_Stop(1);
#endif
    return -1;
    }
  }

  if (osdfile && u32OsdWidth && u32OsdHeight) {
    memset(&OsdAttr, 0, sizeof(RKADK_OSD_ATTR_S));
    memset(&OsdStreamAttr, 0, sizeof(RKADK_OSD_STREAM_ATTR_S));
    OsdAttr.Format = RKADK_FMT_ARGB8888;
    OsdAttr.Width = u32OsdWidth;
    OsdAttr.Height = u32OsdHeight;
    OsdAttr.pData = malloc(OsdAttr.Height * OsdAttr.Width * 4);

#if defined(RV1106_1103) || defined(RV1103B)
    OsdAttr.enOsdType = RKADK_OSD_TYPE_NORMAL;
#else
    OsdAttr.enOsdType = RKADK_OSD_TYPE_EXTRA;
#endif

    OsdStreamAttr.Origin_X = 0;
    OsdStreamAttr.Origin_Y = 0;
    OsdStreamAttr.bEnableShow = RKADK_TRUE;
    OsdStreamAttr.enOsdType = OsdAttr.enOsdType;

    RKADK_OSD_Init(u32OsdId, &OsdAttr);
    RKADK_OSD_AttachToStream(u32OsdId, u32CamId, RKADK_STREAM_TYPE_SNAP, &OsdStreamAttr);

    FILE *fp = fopen(osdfile, "rw");
    if (!fp) {
      RKADK_LOGP("open osd file fail");
    } else {
      RKADK_LOGP("open osd file success");
      fread((RKADK_U8 *)OsdAttr.pData, OsdAttr.Width * OsdAttr.Height * 4, 1, fp);
      fclose(fp);
      RKADK_OSD_UpdateBitMap(u32OsdId, &OsdAttr);
    }
  }

  signal(SIGINT, sigterm_handler);

  char cmd[64];
  printf("\n#Usage: input 'quit' to exit programe!\n"
         "peress any other key to capture one picture to file\n");

  RKADK_PARAM_RES_E type;
  while (!is_quit) {
    fgets(cmd, sizeof(cmd), stdin);
    if (strstr(cmd, "quit") || is_quit) {
      RKADK_LOGP("#Get 'quit' cmd!");
      break;
    } else if (strstr(cmd, "1080")) {
      type = RKADK_RES_1080P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "720")) {
      type = RKADK_RES_720P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "1620")) {
      type = RKADK_RES_1620P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "1296")) {
      type = RKADK_RES_1296P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "4480")) {
      u32SliceHeight = 320;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_SLICE_HEIGHT, &u32SliceHeight);

      type = RKADK_RES_4480P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "4275")) {
      u32SliceHeight = 304;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_SLICE_HEIGHT, &u32SliceHeight);

      type = RKADK_RES_4275P;
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_PHOTO_RES, &type);
      ret = RKADK_PHOTO_Reset(&pHandle);
      if (ret < 0) {
#if !defined(RV1106_1103) && !defined(RV1103B)
        RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
        SAMPLE_ISP_Stop(u32CamId);
#endif
        pHandle = NULL;
        goto photo;
#endif
      }
    } else if (strstr(cmd, "aiisp")) {
      if (bAiispEnable)
        bAiispEnable = false;
      else
        bAiispEnable = true;

      ret = RKADK_MEDIA_SetPostIspAttr(u32CamId, RKADK_STREAM_TYPE_SNAP, bAiispEnable, &stPostIspAttr);
      if (ret)
        RKADK_LOGE("RKADK_MEDIA_SetPostIspAttr failed");
    } else if (strstr(cmd, "fbc0")) {
      RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
      SAMPLE_ISP_Stop(u32CamId);
#endif
      pHandle = NULL;

      stInputFmt.enStreamType = RKADK_STREAM_TYPE_SNAP;
      stInputFmt.format = "FBC0";
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_INPUT_FMT, &stInputFmt);
      goto photo;
    } else if (strstr(cmd, "nv16")) {
      RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
      SAMPLE_ISP_Stop(u32CamId);
#endif
      pHandle = NULL;

      stInputFmt.enStreamType = RKADK_STREAM_TYPE_SNAP;
      stInputFmt.format = "NV16";
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_INPUT_FMT, &stInputFmt);
      goto photo;
    } else if (strstr(cmd, "nv12")) {
      RKADK_PHOTO_DeInit(pHandle);
#ifdef RKAIQ
      SAMPLE_ISP_Stop(u32CamId);
#endif
      pHandle = NULL;

      stInputFmt.enStreamType = RKADK_STREAM_TYPE_SNAP;
      stInputFmt.format = "NV12";
      RKADK_PARAM_SetCamParam(u32CamId, RKADK_PARAM_TYPE_INPUT_FMT, &stInputFmt);
      goto photo;
    }

    if (RKADK_PHOTO_TakePhoto(pHandle, &stTakePhotoAttr)) {
      RKADK_LOGE("RKADK_PHOTO_TakePhoto u32CamId[%d] failed", u32CamId);
      break;
    }

    if (bMultiSensor) {
      if (RKADK_PHOTO_TakePhoto(pHandle1, &stTakePhotoAttr)) {
        RKADK_LOGE("RKADK_PHOTO_TakePhoto u32CamId[1] failed");
        break;
      }
    }

    usleep(500000);
  }

  if (osdfile) {
    RKADK_OSD_DettachFromStream(u32OsdId, u32CamId, RKADK_STREAM_TYPE_SNAP);
    RKADK_OSD_Deinit(u32OsdId);
    free(OsdAttr.pData);
  }

  RKADK_PHOTO_DeInit(pHandle);

  if (bMultiSensor)
    RKADK_PHOTO_DeInit(pHandle1);

#ifdef RKAIQ
  SAMPLE_ISP_Stop(u32CamId);

  if (bMultiCam)
    SAMPLE_ISP_Stop(1);
#endif
  RKADK_MPI_SYS_Exit();
  return 0;
}