Files
hisi_SS524V100_SDK/back/sample/bitrate_auto/sample_rate.c
T

143 lines
4.0 KiB
C

/*
Copyright (c), 2001-2022, Shenshu Tech. Co., Ltd.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <signal.h>
#include "sample_comm.h"
#define PIC_SIZE PIC_1080P
#define BIG_STREAM_SIZE PIC_1080P
#define SMALL_STREAM_SIZE PIC_D1_NTSC
#define SAMPLE_CHN_NUM 2
#define SAMPLE_CHN_0 0
#define SAMPLE_CHN_1 1
#define SAMPLE_VPSS_0 0
#define SAMPLE_VPSS_1 1
static td_s32 g_sample_venc_exit = 0;
/******************************************************************************
* function : show usage
******************************************************************************/
td_void sample_rate_auto_usage(td_char *s_prg_nm)
{
printf("usage : %s [index] \n", s_prg_nm);
printf("index:\n");
printf("\t 0) rate_auto : bitrate_auto sample.\n");
return;
}
/******************************************************************************
* function : to process abnormal case
******************************************************************************/
td_void sample_rate_auto_handle_sig(td_s32 signo)
{
if (signo == SIGINT || signo == SIGTERM) {
g_sample_venc_exit = 1;
}
}
td_void sample_rate_getchar()
{
if (g_sample_venc_exit == 1) {
sample_comm_sys_exit();
printf("\033[0;31mprogram termination abnormally!\033[0;39m\n");
exit(-1);
}
getchar();
if (g_sample_venc_exit == 1) {
sample_comm_sys_exit();
printf("\033[0;31mprogram termination abnormally!\033[0;39m\n");
exit(-1);
}
}
/******************************************************************************
* function : H.265e@1080P@30fps + h264e@D1@30fps
******************************************************************************/
td_s32 sample_rate_auto_normal(td_void)
{
td_s32 ret;
const ot_vpss_grp vpss_grp = 0;
const td_s32 chn_num = SAMPLE_CHN_NUM;
ot_venc_chn venc_chn[SAMPLE_CHN_NUM] = {SAMPLE_CHN_0, SAMPLE_CHN_1};
ot_vpss_chn vpss_chn[SAMPLE_CHN_NUM] = {SAMPLE_VPSS_0, SAMPLE_VPSS_1};
/******************************************
stream save process
******************************************/
ret = sample_comm_venc_rateauto_start(venc_chn, chn_num, vpss_grp, vpss_chn);
if (ret != TD_SUCCESS) {
sample_print("Start Venc failed!\n");
}
printf("please press ENTER to exit this sample\n");
sample_rate_getchar();
/******************************************
exit process
******************************************/
sample_comm_venc_stop_rateauto(venc_chn, chn_num);
return ret;
}
/******************************************************************************
* function : main()
* description : video venc sample
******************************************************************************/
#ifdef __LITEOS__
td_s32 app_main(td_s32 argc, td_char *argv[])
#else
td_s32 main(td_s32 argc, td_char *argv[])
#endif
{
td_s32 ret;
td_u32 index;
struct sigaction sa;
if ((argc != 2) || !strncmp(argv[1], "-h", 2)) { /* 2:arg num */
sample_rate_auto_usage(argv[0]);
return TD_FAILURE;
}
if (strlen(argv[1]) != 1) {
sample_rate_auto_usage(argv[0]);
return TD_FAILURE;
}
if (argv[1][0] != '0') {
sample_rate_auto_usage(argv[0]);
return TD_FAILURE;
}
index = atoi(argv[1]);
#ifndef __LITEOS__
(td_void)memset_s(&sa, sizeof(struct sigaction), 0, sizeof(struct sigaction));
sa.sa_handler = sample_rate_auto_handle_sig;
sa.sa_flags = 0;
sigaction(SIGINT, &sa, NULL);
sigaction(SIGTERM, &sa, NULL);
#endif
switch (index) {
case 0: // 0 : case 0
ret = sample_rate_auto_normal();
break;
default:
printf("the index is invalid!\n");
sample_rate_auto_usage(argv[0]);
return TD_FAILURE;
}
if (ret == TD_SUCCESS) {
printf("program exit normally!\n");
} else {
printf("program exit abnormally!\n");
}
#ifdef __LITEOS__
return ret;
#else
exit(ret);
#endif
}