mirror of
https://gitee.com/Vancouver2017/luban-lite-t3e-pro.git
synced 2025-12-14 10:28:54 +00:00
526 lines
12 KiB
C
526 lines
12 KiB
C
/*
|
||
* Copyright (c) 2022-2024, ArtInChip Technology Co., Ltd
|
||
*
|
||
* SPDX-License-Identifier: Apache-2.0
|
||
*
|
||
* Authors: matteo <duanmt@artinchip.com>
|
||
*/
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <getopt.h>
|
||
#include <sys/time.h>
|
||
|
||
#include <posix/string.h>
|
||
#include <drivers/pin.h>
|
||
|
||
#include "aic_core.h"
|
||
#include "aic_log.h"
|
||
#include "aic_osal.h"
|
||
#include "aic_drv_gpio.h"
|
||
#include "include/yydecoder.h"
|
||
#include "include/video_font_data.h"
|
||
|
||
#include "mpp_vin.h"
|
||
#include "drv_camera.h"
|
||
#include "artinchip_fb.h"
|
||
#include "mpp_fb.h"
|
||
#include "mpp_ge.h"
|
||
|
||
#define BARCODE_DEMO_SUCCESS 1
|
||
#define BUFFER_SIZE 180 * 1024
|
||
#define VID_BUF_NUM 3
|
||
#define VID_BUF_PLANE_NUM 2
|
||
#define VID_SCALE_OFFSET 0
|
||
#define VIDEO_FONT_HEIGHT 32
|
||
|
||
#define ALIGN_DOWM(x, align) ((x) & ~(align - 1))
|
||
|
||
struct aic_dvp_data {
|
||
int w;
|
||
int h;
|
||
int frame_size;
|
||
int frame_cnt;
|
||
int rotation;
|
||
int dst_fmt; // output format
|
||
struct mpp_video_fmt src_fmt;
|
||
uint32_t num_buffers;
|
||
struct vin_video_buf binfo;
|
||
};
|
||
|
||
struct aic_decode_data {
|
||
bool dready;
|
||
int w;
|
||
int h;
|
||
unsigned char *in_buffer;
|
||
unsigned char out_buffer[256];
|
||
};
|
||
|
||
static struct mpp_fb *g_fb = NULL;
|
||
static struct aicfb_screeninfo g_fb_info = {0};
|
||
|
||
static struct aic_dvp_data g_vdata = {0};
|
||
static struct aic_decode_data g_ddata = {0};
|
||
static bool g_dvp_running = true;
|
||
static bool g_running = true;
|
||
unsigned char g_last_barcode[128] = {0};
|
||
|
||
static int dvp_cfg(int width, int height, int format)
|
||
{
|
||
int ret = 0;
|
||
struct dvp_out_fmt f = {0};
|
||
|
||
f.width = g_vdata.src_fmt.width;
|
||
f.height = g_vdata.src_fmt.height;
|
||
f.pixelformat = format;
|
||
f.num_planes = VID_BUF_PLANE_NUM;
|
||
|
||
ret = mpp_dvp_ioctl(DVP_OUT_S_FMT, &f);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int dvp_subdev_set_fmt(void)
|
||
{
|
||
int ret = 0;
|
||
|
||
ret = mpp_dvp_ioctl(DVP_IN_S_FMT, &g_vdata.src_fmt);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int sensor_get_fmt(void)
|
||
{
|
||
int ret = 0;
|
||
struct mpp_video_fmt f = {0};
|
||
|
||
ret = mpp_dvp_ioctl(DVP_IN_G_FMT, &f);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
// return -1;
|
||
}
|
||
|
||
g_vdata.src_fmt = f;
|
||
g_vdata.w = g_vdata.src_fmt.width;
|
||
g_vdata.h = g_vdata.src_fmt.height;
|
||
pr_info("Sensor format: w %d h %d, code 0x%x, bus 0x%x, colorspace 0x%x\n",
|
||
f.width, f.height, f.code, f.bus_type, f.colorspace);
|
||
return 0;
|
||
}
|
||
|
||
static int dvp_request_buf(struct vin_video_buf *vbuf)
|
||
{
|
||
int i;
|
||
|
||
if (mpp_dvp_ioctl(DVP_REQ_BUF, (void *)vbuf) < 0) {
|
||
pr_err("ioctl() failed!\n");
|
||
return -1;
|
||
}
|
||
|
||
pr_info("Buf Plane[0] size Plane[1] size\n");
|
||
for (i = 0; i < vbuf->num_buffers; i++) {
|
||
pr_info("%3d 0x%x %8d 0x%x %8d\n", i,
|
||
vbuf->planes[i * vbuf->num_planes].buf,
|
||
vbuf->planes[i * vbuf->num_planes].len,
|
||
vbuf->planes[i * vbuf->num_planes + 1].buf,
|
||
vbuf->planes[i * vbuf->num_planes + 1].len);
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void dvp_release_buf(int num)
|
||
{
|
||
#if 0
|
||
int i;
|
||
struct video_buf_info *binfo = NULL;
|
||
|
||
for (i = 0; i < num; i++) {
|
||
binfo = &g_vdata.binfo[i];
|
||
if (binfo->vaddr) {
|
||
munmap(binfo->vaddr, binfo->len);
|
||
binfo->vaddr = NULL;
|
||
}
|
||
}
|
||
#endif
|
||
}
|
||
|
||
static int dvp_queue_buf(int index)
|
||
{
|
||
if (mpp_dvp_ioctl(DVP_Q_BUF, (void *)(ptr_t)index) < 0) {
|
||
pr_err("ioctl() failed!\n");
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int dvp_dequeue_buf(int *index)
|
||
{
|
||
int ret = 0;
|
||
|
||
ret = mpp_dvp_ioctl(DVP_DQ_BUF, (void *)index);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int dvp_start(void)
|
||
{
|
||
int ret = 0;
|
||
|
||
ret = mpp_dvp_ioctl(DVP_STREAM_ON, NULL);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static int dvp_stop(void)
|
||
{
|
||
int ret = 0;
|
||
|
||
ret = mpp_dvp_ioctl(DVP_STREAM_OFF, NULL);
|
||
if (ret < 0) {
|
||
pr_err("ioctl() failed! err -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void barcode_copy_buffer(struct aic_dvp_data *vdata, int index)
|
||
{
|
||
struct vin_video_buf *binfo = &vdata->binfo;
|
||
unsigned long buf = binfo->planes[index * VID_BUF_PLANE_NUM].buf;
|
||
int len = binfo->planes[index * VID_BUF_PLANE_NUM].len;
|
||
|
||
aicos_memcpy(g_ddata.in_buffer, (void *)buf, len);
|
||
g_ddata.dready = true;
|
||
}
|
||
|
||
static int barcode_showvideo_tolcd(struct aic_dvp_data *vdata, int index)
|
||
{
|
||
int i;
|
||
struct aicfb_layer_data layer = {0};
|
||
struct vin_video_buf *binfo = &vdata->binfo;
|
||
|
||
layer.layer_id = AICFB_LAYER_TYPE_VIDEO;
|
||
layer.enable = 1;
|
||
|
||
layer.scale_size.width = g_fb_info.width - VID_SCALE_OFFSET * 2;
|
||
layer.scale_size.height = g_fb_info.height - VID_SCALE_OFFSET * 2;
|
||
layer.pos.x = VID_SCALE_OFFSET;
|
||
layer.pos.y = VID_SCALE_OFFSET;
|
||
|
||
layer.buf.size.width = vdata->w;
|
||
layer.buf.size.height = vdata->h;
|
||
layer.buf.format = MPP_FMT_NV12;
|
||
|
||
layer.buf.buf_type = MPP_PHY_ADDR;
|
||
|
||
for (i = 0; i < VID_BUF_PLANE_NUM; i++) {
|
||
layer.buf.stride[i] = layer.buf.size.width;
|
||
layer.buf.phy_addr[i] = binfo->planes[index * VID_BUF_PLANE_NUM + i].buf;
|
||
}
|
||
|
||
if (mpp_fb_ioctl(g_fb, AICFB_UPDATE_LAYER_CONFIG, &layer) < 0) {
|
||
pr_err("ioctl() failed!\n");
|
||
return -1;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
static void aicfb_lcd_putc(unsigned int x, unsigned int y, char ch)
|
||
{
|
||
unsigned long dcache_size, dcache_start;
|
||
int pbytes = g_fb_info.bits_per_pixel / 8;
|
||
int i, row;
|
||
void *line;
|
||
|
||
line = (unsigned char *)(g_fb_info.framebuffer + y * g_fb_info.stride + x * pbytes);
|
||
dcache_start = ALIGN_DOWM((unsigned long)line, ARCH_DMA_MINALIGN);
|
||
|
||
for (row = 0; row < VIDEO_FONT_HEIGHT; row++) {
|
||
unsigned int idx = (ch - 32) * VIDEO_FONT_HEIGHT + row;
|
||
uint32_t bits = video_fontdata[idx];
|
||
|
||
uint16_t *dst = line;
|
||
|
||
for (i = 0; i < VIDEO_FONT_WIDTH; i++) {
|
||
*dst++ = (bits & 0x80000000) ? 0xFFFF : 0x0000;
|
||
bits <<= 1;
|
||
}
|
||
|
||
line += g_fb_info.stride;
|
||
}
|
||
|
||
dcache_size = ALIGN_UP((unsigned long)line - dcache_start,
|
||
ARCH_DMA_MINALIGN);
|
||
aicos_dcache_clean_range((unsigned long *)dcache_start, dcache_size);
|
||
}
|
||
|
||
static int barcode_set_ui_layer_alpha(int val)
|
||
{
|
||
int ret = BARCODE_DEMO_SUCCESS;
|
||
struct aicfb_alpha_config alpha = {0};
|
||
|
||
alpha.layer_id = AICFB_LAYER_TYPE_UI;
|
||
alpha.enable = 1;
|
||
alpha.mode = 1;
|
||
alpha.value = val;
|
||
ret = mpp_fb_ioctl(g_fb, AICFB_UPDATE_ALPHA_CONFIG, &alpha);
|
||
if (ret < 0)
|
||
pr_err("ioctl() failed! errno: -%d\n", -ret);
|
||
return BARCODE_DEMO_SUCCESS;
|
||
}
|
||
|
||
static int show_barcode_tolcd(int len)
|
||
{
|
||
int font_width = 32;
|
||
int i,startx;
|
||
startx = 10;
|
||
if(strcmp((char *)g_last_barcode, (char *)(g_ddata.out_buffer)) == 0){
|
||
//pr_info("same, not shshow ！\n");
|
||
return -1;
|
||
}
|
||
|
||
memset(g_fb_info.framebuffer,0, g_fb_info.smem_len);
|
||
aicos_dcache_clean_range(g_fb_info.framebuffer, g_fb_info.smem_len);
|
||
|
||
for (i = 0; i < len; i ++)
|
||
aicfb_lcd_putc(startx + (i * font_width), 60, g_ddata.out_buffer[i]);
|
||
|
||
strcpy((char *)g_last_barcode, (char *)(g_ddata.out_buffer));
|
||
|
||
pr_info("GetDecoderResult %d [%s] \n", len, g_ddata.out_buffer);
|
||
return BARCODE_DEMO_SUCCESS;
|
||
}
|
||
|
||
static void show_duration_tolcd(char* buf, int len)
|
||
{
|
||
int font_width = 30;
|
||
int i,startx;
|
||
startx = 10;
|
||
|
||
for (i = 0; i < len -1; i ++)
|
||
aicfb_lcd_putc(startx + (i * font_width), 150, buf[i]);
|
||
|
||
pr_info("duration %d [%s] \n", buf);
|
||
}
|
||
|
||
static void barcode_shotting_thread()
|
||
{
|
||
int index = 0;
|
||
int i;
|
||
|
||
if (dvp_request_buf(&g_vdata.binfo) < 0)
|
||
return;
|
||
|
||
for (i = 0; i < VID_BUF_NUM; i++) {
|
||
if (dvp_queue_buf(i) < 0)
|
||
return;
|
||
}
|
||
|
||
if (dvp_start() < 0)
|
||
return;
|
||
|
||
g_dvp_running = true;
|
||
while (g_dvp_running) {
|
||
if (dvp_dequeue_buf(&index) < 0)
|
||
break;
|
||
|
||
barcode_showvideo_tolcd(&g_vdata, index);
|
||
|
||
if( !g_ddata.dready)
|
||
barcode_copy_buffer(&g_vdata, index);
|
||
|
||
dvp_queue_buf(index);
|
||
}
|
||
|
||
g_running = false;
|
||
dvp_stop();
|
||
dvp_release_buf(g_vdata.binfo.num_buffers);
|
||
mpp_vin_deinit();
|
||
}
|
||
|
||
static void barcode_decode_thread()
|
||
{
|
||
int ret;
|
||
int len = 0;
|
||
struct timespec begin, end;
|
||
char str_duration[64] = {0};
|
||
float diff_duration;
|
||
g_running = true;
|
||
|
||
while(g_running){
|
||
|
||
if( g_ddata.dready){
|
||
gettimespec(&begin);
|
||
ret = Decoding_Image(g_ddata.in_buffer, g_ddata.w, g_ddata.h);
|
||
//pr_info("Decoding_Image with %d [%d, %d] \n", ret, g_ddata.w, g_ddata.h);
|
||
|
||
len = GetResultLength();
|
||
//pr_info("GetResultLength with %d\n", len);
|
||
|
||
ret = GetDecoderResult(g_ddata.out_buffer);
|
||
//pr_info("GetDecoderResult with %d \n", ret);
|
||
if(len > 0){
|
||
gettimespec(&end);
|
||
//show_timespec_diff("decode", NULL, &begin, &end);
|
||
if(show_barcode_tolcd(len) == BARCODE_DEMO_SUCCESS){
|
||
diff_duration = timespec_diff(&begin, &end);
|
||
sprintf(str_duration, "Time: %d ms\n", (u32)(diff_duration * MS_PER_SEC) % MS_PER_SEC);
|
||
show_duration_tolcd(str_duration, strlen(str_duration));
|
||
memset(str_duration, 0, sizeof(str_duration));
|
||
}
|
||
}
|
||
}
|
||
g_ddata.dready = false;
|
||
aicos_msleep(500);
|
||
}
|
||
|
||
if (g_fb) {
|
||
mpp_fb_close(g_fb);
|
||
g_fb = NULL;
|
||
}
|
||
aicos_free(MEM_CMA, g_ddata.in_buffer);
|
||
ret = 0;
|
||
pr_info("exit with ret = %d \n", ret);
|
||
}
|
||
|
||
static int init_dvp_device()
|
||
{
|
||
memset(&g_vdata, 0, sizeof(struct aic_dvp_data));
|
||
g_vdata.dst_fmt = MPP_FMT_NV12;
|
||
g_vdata.frame_cnt = 1;
|
||
|
||
if (mpp_vin_init(CAMERA_DEV_NAME))
|
||
return -1;
|
||
|
||
if (sensor_get_fmt() < 0)
|
||
goto error_out;
|
||
|
||
if (dvp_subdev_set_fmt() < 0)
|
||
goto error_out;
|
||
|
||
g_vdata.frame_size = (g_vdata.w * g_vdata.h * 3) >> 1;
|
||
|
||
if (dvp_cfg(g_vdata.w, g_vdata.h, g_vdata.dst_fmt) < 0)
|
||
goto error_out;
|
||
|
||
g_vdata.num_buffers = VID_BUF_NUM;
|
||
|
||
return BARCODE_DEMO_SUCCESS;
|
||
|
||
error_out:
|
||
mpp_vin_deinit();
|
||
return -1;
|
||
}
|
||
|
||
static int init_screen_fb()
|
||
{
|
||
int ret = 0;
|
||
g_fb = mpp_fb_open();
|
||
|
||
ret = mpp_fb_ioctl(g_fb, AICFB_GET_SCREENINFO, &g_fb_info);
|
||
if (ret < 0){
|
||
pr_err("ioctl() failed! errno: -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
memset(g_fb_info.framebuffer,0, g_fb_info.smem_len);
|
||
aicos_dcache_clean_range(g_fb_info.framebuffer, g_fb_info.smem_len);
|
||
|
||
ret = mpp_fb_ioctl(g_fb, AICFB_POWERON, &g_fb_info);
|
||
if (ret < 0){
|
||
pr_err("ioctl() failed! errno: -%d\n", -ret);
|
||
return -1;
|
||
}
|
||
|
||
pr_info("Screen width: %d, height %d\n", g_fb_info.width, g_fb_info.height);
|
||
|
||
return BARCODE_DEMO_SUCCESS;
|
||
}
|
||
|
||
static void cmd_barcode_demo(int argc, char **argv)
|
||
{
|
||
int ret;
|
||
aicos_thread_t thid = NULL;
|
||
|
||
ret = init_screen_fb();
|
||
if(ret != BARCODE_DEMO_SUCCESS){
|
||
pr_info("init_screen_fb failed with %d \n", ret);
|
||
return;
|
||
}
|
||
|
||
ret = barcode_set_ui_layer_alpha(64);
|
||
if(ret != BARCODE_DEMO_SUCCESS){
|
||
pr_info("set_ui_layer_alpha failed with %d \n", ret);
|
||
goto error_out;
|
||
}
|
||
|
||
ret = init_dvp_device();
|
||
if(ret != BARCODE_DEMO_SUCCESS){
|
||
pr_info("init_dvp_device failed with %d \n", ret);
|
||
return;
|
||
}
|
||
|
||
g_ddata.w = g_vdata.w;
|
||
g_ddata.h = g_vdata.h;
|
||
g_ddata.dready = false;
|
||
g_running = true;
|
||
|
||
g_ddata.in_buffer = aicos_malloc(MEM_CMA, g_ddata.w * g_ddata.w * 2);
|
||
|
||
ret = Initial_Decoder();
|
||
if(ret != BARCODE_DEMO_SUCCESS){
|
||
pr_info("Initial_Decoder failed with %d \n", ret);
|
||
goto error_out;
|
||
}
|
||
|
||
for (size_t i = 0; i <= 13; i++) {
|
||
Set_Donfig_Decoder(i, 1);
|
||
}
|
||
|
||
|
||
thid = aicos_thread_create("shoting_thead", 8192, 0, barcode_shotting_thread, NULL);
|
||
if (thid == NULL) {
|
||
pr_err("Failed to create DVP thread\n");
|
||
goto error_out;
|
||
}
|
||
|
||
thid = aicos_thread_create("barcode_thread", 1024 * 32, 0, barcode_decode_thread, NULL);
|
||
if (thid == NULL) {
|
||
pr_err("Failed to create decode thread\n");
|
||
goto error_out;
|
||
}
|
||
return;
|
||
|
||
error_out:
|
||
if (g_fb) {
|
||
mpp_fb_close(g_fb);
|
||
g_fb = NULL;
|
||
}
|
||
aicos_free(MEM_CMA, g_ddata.in_buffer);
|
||
}
|
||
static int barcode_demo()
|
||
{
|
||
cmd_barcode_demo(0, NULL);
|
||
return 0;
|
||
}
|
||
INIT_LATE_APP_EXPORT (barcode_demo);
|
||
|