linuxOS_AP06/app/rkadk/examples/rkadk_photo_test.c
2025-06-03 12:28:32 +08:00

587 lines
17 KiB
C

/*
* 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;
}