TTA
/
tta-ai-streamer
18
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '2731c1ed44'
${ noResults }
tta-ai-streamer/common/nc_video_codec/component/helper/yuv/yuvLoaderfeeder.c

1073 lines
40 KiB
Raw Normal View History Unescape

first commit
7 months ago
//-----------------------------------------------------------------------------
// COPYRIGHT (C) 2020 CHIPS&MEDIA INC. ALL RIGHTS RESERVED
//
// This file is distributed under BSD 3 clause and LGPL2.1 (dual license)
// SPDX License Identifier: BSD-3-Clause
// SPDX License Identifier: LGPL-2.1-only
//
// The entire notice above must be reproduced on all authorized copies.
//
// Description :
//-----------------------------------------------------------------------------
#include <string.h>
#include "main_helper.h"
//#define SAVE_WHOLE_LOADED_YUV
//#define BIG_ENDIAN_YUV //LITTLE_ENDIAN YUV : ref-c output, cnm YUV style
typedef enum {
PIXEL_FORMAT_8BIT = 0,
PIXEL_FORMAT_16BIT_MSB = 1,
PIXEL_FORMAT_32BIT_MSB = 2,
PIXEL_FORMAT_16BIT_LSB = 5,
PIXEL_FORMAT_32BIT_LSB = 6,
} PixelFormat;
#ifdef SAVE_WHOLE_LOADED_YUV
FILE *save_fp;
int save_frame_cnt=0;
int save_frame_max=10;
#endif
typedef struct {
osal_file_t fp;
Uint8* pYuv;
Uint32 cbcrInterleave;
Uint32 nv21;
Uint32 i422;
Uint32 packedFormat;
} yuvLoaderContext;
enum {
OUTPUT_PLANAR = 0,
OUTPUT_NV12 = 2,
OUTPUT_NV21 = 3,
OUTPUT_PACKED = 4,
OUTPUT_I422 = 5,
OUTPUT_NV16 = 6,
OUTPUT_NV61 = 7,
};
typedef unsigned short pel; // the size of pel should be 16.
typedef struct{
pel* src_buf[3];
Int32 rot_mode;
Int32 src_stride;
Int32 src_width;
Int32 src_height;
Int32 padded_y_stride;
Int32 padded_c_stride;
Int32 padded_y_height;
Int32 padded_c_height;
Int32 enc_stride;
Int32 enc_width;
Int32 enc_height;
Int32 enc_c_stride;
pel* p_out_buf[3];
pel* out_buf;
pel* pad_buf;
}prp_t;
void conv_image (unsigned short *outbuf, unsigned short *inbuf,
int pic_width, int pic_height, int out_width, int out_height, int frm_format, int packedFormat)
{
int x, y;
int iaddr, oaddr;
int temp=0, tbase;
if ( packedFormat == NOT_PACKED )
{
if (frm_format == OUTPUT_PLANAR || frm_format == OUTPUT_I422) { // planar, I422
// Y
iaddr = 0;
oaddr = 0;
for (y=0; y<out_height; y++) {
for (x=0; x<out_width; x++) {
if (y >= pic_height) {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[x + pic_width*(pic_height-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
oaddr++;
}
}
// U
tbase = pic_width*pic_height;
iaddr = tbase;
oaddr = out_width*out_height;
for (y=0; y<out_height/2; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height/2) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
if (frm_format == OUTPUT_I422) {
outbuf[oaddr+(out_width/2)] = (unsigned short)temp;
}
oaddr++;
}
if (frm_format == OUTPUT_I422) {
oaddr+=out_width/2;
}
}
// V
tbase = pic_width*pic_height*5/4;
iaddr = tbase;
oaddr = (frm_format == OUTPUT_I422) ? out_width*out_height*3/2 : out_width*out_height*5/4;
for (y=0; y<out_height/2; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height/2) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
if (frm_format == OUTPUT_I422) {
outbuf[oaddr+(out_width/2)] = (unsigned short)temp;
}
oaddr++;
}
if (frm_format == OUTPUT_I422) {
oaddr+=out_width/2;
}
}
} else if (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV21 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) { // NV12, NV21, NV16, NV61
// Y
iaddr = 0;
oaddr = 0;
for (y=0; y<out_height; y++) {
for (x=0; x<out_width; x++) {
if (y >= pic_height) {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[x + pic_width*(pic_height-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
oaddr++;
}
}
// U
tbase = (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV16) ? pic_width*pic_height : pic_width*pic_height*5/4;
iaddr = tbase;
oaddr = out_width*out_height;
for (y=0; y<out_height/2; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height/2) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
if (frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) {
outbuf[oaddr+out_width] = (unsigned short)temp;
}
oaddr+=2;
}
if (frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) {
oaddr+=out_width;
}
}
// V
tbase = (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV16) ? pic_width*pic_height*5/4 : pic_width*pic_height;
iaddr = tbase;
oaddr = out_width*out_height + 1;
for (y=0; y<out_height/2; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height/2) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
if (frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) {
outbuf[oaddr+out_width] = (unsigned short)temp;
}
oaddr+=2;
}
if (frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) {
oaddr+=out_width;
}
}
}
else {
printf("Error) Unknown frame format: %d\n", frm_format);
}
} else {/* packdformat */
if ( packedFormat == PACKED_YUYV || packedFormat == PACKED_YVYU){
oaddr = 0;
} else {
oaddr = 1;
}
// Y
iaddr = 0;
for (y=0; y<out_height; y++) {
for (x=0; x<out_width; x++) {
if (y >= pic_height) {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[x + pic_width*(pic_height-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
oaddr+=2;
}
}
// U
switch (packedFormat)
{
case PACKED_YUYV: oaddr = 1;break;
case PACKED_YVYU: oaddr = 3;break;
case PACKED_UYVY: oaddr = 0;break;
case PACKED_VYUY: oaddr = 2;break;
default:
VLOG(ERR, "%s:%d Not supported format(%d)\n", __FILE__, __LINE__, packedFormat);
break;
}
tbase = pic_width*pic_height;
iaddr = tbase;
for (y=0; y<out_height; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
oaddr+=4;
}
if (y%2 == 0)
iaddr = iaddr - out_width/2;
}
// V
switch (packedFormat)
{
case PACKED_YUYV: oaddr = 3;break;
case PACKED_YVYU: oaddr = 1;break;
case PACKED_UYVY: oaddr = 2;break;
case PACKED_VYUY: oaddr = 0;break;
default:
VLOG(ERR, "%s:%d Not supported format(%d)\n", __FILE__, __LINE__, packedFormat);
break;
}
tbase = pic_width*pic_height*5/4;
iaddr = tbase;
for (y=0; y<out_height; y++) {
for (x=0; x<out_width/2; x++) {
if (y >= pic_height) {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[tbase + x + (pic_width/2)*(pic_height/2-1)];
outbuf[oaddr] = (unsigned short)temp;
}
} else {
if (x >= pic_width/2)
outbuf[oaddr] = (unsigned short)temp;
else {
temp = inbuf[iaddr];
outbuf[oaddr] = (unsigned short)temp;
iaddr++;
}
}
oaddr+=4;
}
if (y%2 == 0)
iaddr = iaddr - out_width/2;
}
}
}
int write_image_to_file (FILE *fp, unsigned short *outbuf,
int out_width, int out_height, int pxl_format, int frm_format, int packedFormat)
{
int i, x, y;
unsigned short tmp16;
unsigned int tmp32 = 0;
int width, height;
int size, cb_start, cr_start;
unsigned char *temp = (unsigned char*)outbuf;
size = (packedFormat || frm_format == OUTPUT_I422 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_width*out_height*2 : out_width*out_height*3/2;
//***** 8bit
if (pxl_format == PIXEL_FORMAT_8BIT) {
for (i=0; i<size; i++)
fputc(temp[i]&0xff, fp);
}
//***** 16bit
else if (pxl_format == PIXEL_FORMAT_16BIT_LSB) {
for (i=0; i<size; i++) {
//tmp16 = outbuf[i]&0x3ff;
tmp16 = outbuf[i];
#ifdef BIG_ENDIAN_YUV
byte_swap((unsigned char *)&tmp16, 2);
#endif
fputc(tmp16&0xff, fp);
fputc((tmp16>>8)&0xff, fp);
}
} else if (pxl_format == PIXEL_FORMAT_16BIT_MSB) {
for (i=0; i<size; i++) {
//tmp16 = (outbuf[i]&0x3ff)<<6;
tmp16 = outbuf[i];
#ifdef BIG_ENDIAN_YUV
byte_swap((unsigned char *)&tmp16, 2);
#endif
fputc(tmp16&0xff, fp);
fputc((tmp16>>8)&0xff, fp);
}
}
//***** 32bit
else {
// Y
width = (packedFormat) ? out_width*2 : out_width;
height = out_height;
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((outbuf[x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((outbuf[x+0+width*y]&0x3ff)<<22) |
((outbuf[x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((outbuf[x+0+width*y]&0x3ff)<<22) |
((outbuf[x+1+width*y]&0x3ff)<<12) |
((outbuf[x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (outbuf[x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (outbuf[x+0+width*y]&0x3ff) |
((outbuf[x+1+width*y]&0x3ff)<<10);
else
tmp32 = (outbuf[x+0+width*y]&0x3ff) |
((outbuf[x+1+width*y]&0x3ff)<<10) |
((outbuf[x+2+width*y]&0x3ff)<<20);
}
fputc((tmp32>>24)&0xff, fp);
fputc((tmp32>>16)&0xff, fp);
fputc((tmp32>>8)&0xff, fp);
fputc(tmp32&0xff, fp);
}
}
if (packedFormat) // packed
return 1;
width = (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV21 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_width : out_width/2;
height = (frm_format == OUTPUT_I422 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_height : out_height/2;
cb_start = out_width*out_height;
// Cb
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((outbuf[cb_start+x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((outbuf[cb_start+x+0+width*y]&0x3ff)<<22) |
((outbuf[cb_start+x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((outbuf[cb_start+x+0+width*y]&0x3ff)<<22) |
((outbuf[cb_start+x+1+width*y]&0x3ff)<<12) |
((outbuf[cb_start+x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (outbuf[cb_start+x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (outbuf[cb_start+x+0+width*y]&0x3ff) |
((outbuf[cb_start+x+1+width*y]&0x3ff)<<10);
else
tmp32 = (outbuf[cb_start+x+0+width*y]&0x3ff) |
((outbuf[cb_start+x+1+width*y]&0x3ff)<<10) |
((outbuf[cb_start+x+2+width*y]&0x3ff)<<20);
}
fputc((tmp32>>24)&0xff, fp);
fputc((tmp32>>16)&0xff, fp);
fputc((tmp32>>8)&0xff, fp);
fputc(tmp32&0xff, fp);
}
}
if (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV21 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) // interleave
return 1;
width = out_width/2;
height = (frm_format == OUTPUT_I422) ? out_height : out_height/2;
cr_start = (frm_format == OUTPUT_I422) ? out_width*out_height*3/2 : out_width*out_height*5/4;
// Cr
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((outbuf[cr_start+x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((outbuf[cr_start+x+0+width*y]&0x3ff)<<22) |
((outbuf[cr_start+x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((outbuf[cr_start+x+0+width*y]&0x3ff)<<22) |
((outbuf[cr_start+x+1+width*y]&0x3ff)<<12) |
((outbuf[cr_start+x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (outbuf[cr_start+x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (outbuf[cr_start+x+0+width*y]&0x3ff) |
((outbuf[cr_start+x+1+width*y]&0x3ff)<<10);
else
tmp32 = (outbuf[cr_start+x+0+width*y]&0x3ff) |
((outbuf[cr_start+x+1+width*y]&0x3ff)<<10) |
((outbuf[cr_start+x+2+width*y]&0x3ff)<<20);
}
fputc((tmp32>>24)&0xff, fp);
fputc((tmp32>>16)&0xff, fp);
fputc((tmp32>>8)&0xff, fp);
fputc(tmp32&0xff, fp);
}
}
}
return 1;
}
int write_image_to_mem (unsigned short *out_mem, unsigned short *inbuf,
int out_width, int out_height, int pxl_format, int frm_format, int packedFormat)
{
int i, x, y;
unsigned int tmp32 = 0;
int width, height;
int size, cb_start, cr_start;
int oaddr=0;
unsigned char *temp = (unsigned char *)out_mem;
size = (packedFormat || frm_format == OUTPUT_I422 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_width*out_height*2 : out_width*out_height*3/2;
// 8BIT
if (pxl_format == PIXEL_FORMAT_8BIT) {
for (i=0; i<size; i++){
temp[oaddr++] = (unsigned char)(inbuf[i]&0xff);
}
}
// 16BIT MSB -> 16BIT_LSB
else if ( pxl_format == PIXEL_FORMAT_16BIT_LSB) {
#ifdef BIG_ENDIAN_YUV
#else
for (i=0; i<size; i++) {
out_mem[i] = (unsigned short)(((inbuf[i]&0xff)<<8) | ((inbuf[i]>>8)&0xff));
}
memcpy(inbuf, out_mem, size*2);
#endif
for (i=0; i<size; i++) {
out_mem[oaddr++] = inbuf[i]<<6;
}
#ifdef BIG_ENDIAN_YUV
#else
for (i=0; i<size; i++) {
inbuf[i] = (unsigned short)(((out_mem[i]&0xff)<<8) | ((out_mem[i]>>8)&0xff));
}
memcpy(out_mem, inbuf, size*2);
#endif
}
// 16BIT
else if (pxl_format == PIXEL_FORMAT_16BIT_MSB) {
for (i=0; i<size; i++) {
out_mem[oaddr++] = inbuf[i];
}
}
// 32BIT
else {
// Y
width = (packedFormat) ? out_width*2 : out_width;
height = out_height;
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((inbuf[x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((inbuf[x+0+width*y]&0x3ff)<<22) |
((inbuf[x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((inbuf[x+0+width*y]&0x3ff)<<22) |
((inbuf[x+1+width*y]&0x3ff)<<12) |
((inbuf[x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (inbuf[x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (inbuf[x+0+width*y]&0x3ff) |
((inbuf[x+1+width*y]&0x3ff)<<10);
else
tmp32 = (inbuf[x+0+width*y]&0x3ff) |
((inbuf[x+1+width*y]&0x3ff)<<10) |
((inbuf[x+2+width*y]&0x3ff)<<20);
}
temp[oaddr++] = (unsigned char)((tmp32>>24)&0xff);
temp[oaddr++] = (unsigned char)((tmp32>>16)&0xff);
temp[oaddr++] = (unsigned char)((tmp32>>8)&0xff);
temp[oaddr++] = tmp32&0xff;
}
}
if (packedFormat) // packed
return 1;
width = (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV21 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_width : out_width/2;
height = (frm_format == OUTPUT_I422 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) ? out_height : out_height/2;
// Cb
cb_start = out_width*out_height;
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((inbuf[cb_start+x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((inbuf[cb_start+x+0+width*y]&0x3ff)<<22) |
((inbuf[cb_start+x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((inbuf[cb_start+x+0+width*y]&0x3ff)<<22) |
((inbuf[cb_start+x+1+width*y]&0x3ff)<<12) |
((inbuf[cb_start+x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (inbuf[cb_start+x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (inbuf[cb_start+x+0+width*y]&0x3ff) |
((inbuf[cb_start+x+1+width*y]&0x3ff)<<10);
else
tmp32 = (inbuf[cb_start+x+0+width*y]&0x3ff) |
((inbuf[cb_start+x+1+width*y]&0x3ff)<<10) |
((inbuf[cb_start+x+2+width*y]&0x3ff)<<20);
}
temp[oaddr++] = (unsigned char)((tmp32>>24)&0xff);
temp[oaddr++] = (unsigned char)((tmp32>>16)&0xff);
temp[oaddr++] = (unsigned char)((tmp32>>8)&0xff);
temp[oaddr++] = tmp32&0xff;
}
}
if (frm_format == OUTPUT_NV12 || frm_format == OUTPUT_NV21 || frm_format == OUTPUT_NV16 || frm_format == OUTPUT_NV61) // interleave
return 1;
width = out_width/2;
height = (frm_format == OUTPUT_I422) ? out_height : out_height/2;
cr_start = (frm_format == OUTPUT_I422) ? out_width*out_height*3/2 : out_width*out_height*5/4;
// Cr
for (y=0; y<height; y++) {
for (x=0; x<width; x=x+3) {
if (pxl_format == PIXEL_FORMAT_32BIT_MSB) {
if (x+1 >= width)
tmp32 = ((inbuf[cr_start+x+0+width*y]&0x3ff)<<22);
else if (x+2 >= width)
tmp32 = ((inbuf[cr_start+x+0+width*y]&0x3ff)<<22) |
((inbuf[cr_start+x+1+width*y]&0x3ff)<<12);
else
tmp32 = ((inbuf[cr_start+x+0+width*y]&0x3ff)<<22) |
((inbuf[cr_start+x+1+width*y]&0x3ff)<<12) |
((inbuf[cr_start+x+2+width*y]&0x3ff)<<2);
} else if (pxl_format == PIXEL_FORMAT_32BIT_LSB) {
if (x+1 >= width)
tmp32 = (inbuf[cr_start+x+0+width*y]&0x3ff);
else if (x+2 >= width)
tmp32 = (inbuf[cr_start+x+0+width*y]&0x3ff) |
((inbuf[cr_start+x+1+width*y]&0x3ff)<<10);
else
tmp32 = (inbuf[cr_start+x+0+width*y]&0x3ff) |
((inbuf[cr_start+x+1+width*y]&0x3ff)<<10) |
((inbuf[cr_start+x+2+width*y]&0x3ff)<<20);
}
temp[oaddr++] = (unsigned char)(tmp32>>24)&0xff;
temp[oaddr++] = (unsigned char)(tmp32>>16)&0xff;
temp[oaddr++] = (unsigned char)(tmp32>>8)&0xff;
temp[oaddr++] = tmp32&0xff;
}
}
}
return 1;
}
BOOL loaderYuvFeeder_Create(
YuvFeederImpl* impl,
const char* path,
Uint32 packed,
Uint32 fbStride,
Uint32 fbHeight)
{
yuvLoaderContext* ctx;
osal_file_t fp;
Uint8* pYuv;
if ((fp=osal_fopen(path, "rb")) == NULL) {
VLOG(ERR, "%s:%d failed to open yuv file: %s\n", __FUNCTION__, __LINE__, path);
return FALSE;
}
if ( packed == 1 )
pYuv = (Uint8*)osal_malloc(fbStride*fbHeight*3*2);//packed, unsigned short
else
pYuv = (Uint8*)osal_malloc(fbStride*fbHeight*3);//unsigned short
if ((ctx=(yuvLoaderContext*)osal_malloc(sizeof(yuvLoaderContext))) == NULL) {
osal_free(pYuv);
osal_fclose(fp);
return FALSE;
}
osal_memset(ctx, 0, sizeof(yuvLoaderContext));
ctx->fp = fp;
ctx->pYuv = pYuv;
impl->context = ctx;
return TRUE;
}
BOOL loaderYuvFeeder_Destory(
YuvFeederImpl *impl
)
{
yuvLoaderContext* ctx = (yuvLoaderContext*)impl->context;
osal_fclose(ctx->fp);
osal_free(ctx->pYuv);
osal_free(ctx);
return TRUE;
}
BOOL loaderYuvFeeder_Feed(
YuvFeederImpl* impl,
Int32 coreIdx,
FrameBuffer* fb,
size_t picWidth,
size_t picHeight,
Uint32 srcFbIndex,
ENC_subFrameSyncCfg* subFrameSyncConfig,
void* arg
)
{
size_t srcFrameSize, lumaSize, chromaSize;
size_t fileSize;
Int32 ystride;
Int32 out_frm_format;
yuvLoaderContext* ctx = (yuvLoaderContext*)impl->context;
Uint8* pYuv_int8 = ctx->pYuv;
osal_file_t yuvFp = ctx->fp;
Uint16* pYuv_int16 = NULL, *pYuv2_int16 = NULL;
Int32 outsize, out_width, out_height, pxl_format = 0, yuv3p4b = 0;
Int32 bitdepth = 8;
Uint32 i;
Int32 productId = PRODUCT_ID_NONE;
Uint32 subFrameSyncSrcWriteMode = subFrameSyncConfig->subFrameSyncSrcWriteMode;
BOOL subFrameSyncEn = subFrameSyncConfig->subFrameSyncOn;
Uint32 writeSrcLine = subFrameSyncSrcWriteMode & ~REMAIN_SRC_DATA_WRITE;
if ( subFrameSyncEn == TRUE && subFrameSyncSrcWriteMode == SRC_0LINE_WRITE ) {
return TRUE;
}
UNREFERENCED_PARAMETER(arg);
productId = VPU_GetProductId(coreIdx);
if ( fb->cbcrInterleave == TRUE ) {
if ( ctx->nv21 ) {
out_frm_format = (ctx->i422) ? OUTPUT_NV61 : OUTPUT_NV21;
} else {
out_frm_format = (ctx->i422) ? OUTPUT_NV16 : OUTPUT_NV12;
}
} else if ( ctx->i422 == TRUE) {
out_frm_format = OUTPUT_I422 ;
} else {
out_frm_format = OUTPUT_PLANAR;
}
switch (fb->format) {
case FORMAT_400:
case FORMAT_420:
case FORMAT_422:
case FORMAT_YUYV:
case FORMAT_YVYU:
case FORMAT_UYVY:
case FORMAT_VYUY:
bitdepth = 8;
pxl_format = PIXEL_FORMAT_8BIT;
break;
case FORMAT_420_P10_16BIT_LSB:
case FORMAT_422_P10_16BIT_LSB:
case FORMAT_YUYV_P10_16BIT_LSB:
case FORMAT_YVYU_P10_16BIT_LSB:
case FORMAT_UYVY_P10_16BIT_LSB:
case FORMAT_VYUY_P10_16BIT_LSB:
bitdepth = 10;
pxl_format = PIXEL_FORMAT_16BIT_LSB;
break;
case FORMAT_420_P10_16BIT_MSB:
case FORMAT_422_P10_16BIT_MSB:
case FORMAT_YUYV_P10_16BIT_MSB:
case FORMAT_YVYU_P10_16BIT_MSB:
case FORMAT_UYVY_P10_16BIT_MSB:
case FORMAT_VYUY_P10_16BIT_MSB:
bitdepth = 10;
pxl_format = PIXEL_FORMAT_16BIT_MSB;
break;
case FORMAT_420_P10_32BIT_LSB:
case FORMAT_422_P10_32BIT_LSB:
case FORMAT_YUYV_P10_32BIT_LSB:
case FORMAT_YVYU_P10_32BIT_LSB:
case FORMAT_UYVY_P10_32BIT_LSB:
case FORMAT_VYUY_P10_32BIT_LSB:
bitdepth = 10;
pxl_format = PIXEL_FORMAT_32BIT_LSB;
yuv3p4b = 1;
break;
case FORMAT_420_P10_32BIT_MSB:
case FORMAT_422_P10_32BIT_MSB:
case FORMAT_YUYV_P10_32BIT_MSB:
case FORMAT_YVYU_P10_32BIT_MSB:
case FORMAT_UYVY_P10_32BIT_MSB:
case FORMAT_VYUY_P10_32BIT_MSB:
bitdepth = 10;
pxl_format = PIXEL_FORMAT_32BIT_MSB;
yuv3p4b = 1;
break;
default:
VLOG(ERR, "%s:%d Not supported format(%d)\n", __FILE__, __LINE__, fb->format);
break;
}
//calculate input YUV(8bit, 16BIT) size
if (ctx->i422) {
if (bitdepth == 10) {
CalcYuvSize(fb->format, (Int32)picWidth, (Int32)picHeight, fb->cbcrInterleave, &lumaSize, &chromaSize, &srcFrameSize, NULL, NULL, NULL);
fileSize = srcFrameSize;
}
else {
CalcYuvSize(fb->format, (Int32)picWidth, (Int32)picHeight, fb->cbcrInterleave, &lumaSize, &chromaSize, &srcFrameSize, NULL, NULL, NULL);
fileSize = srcFrameSize;
}
} else {
if (bitdepth == 10) {
CalcYuvSize(FORMAT_420_P10_16BIT_LSB, (Int32)picWidth, (Int32)picHeight, fb->cbcrInterleave, &lumaSize, &chromaSize, &srcFrameSize, NULL, NULL, NULL);
fileSize = srcFrameSize;
}
else {
CalcYuvSize(FORMAT_420, (Int32)picWidth, (Int32)picHeight, fb->cbcrInterleave, &lumaSize, &chromaSize, &srcFrameSize, NULL, NULL, NULL);
fileSize = srcFrameSize;
}
}
out_width = (Int32)((yuv3p4b&&ctx->packedFormat==0) ? ((picWidth+31)/32)*32 : picWidth);
out_height = (Int32)((yuv3p4b) ? ((picHeight+3)/4)*4 : picHeight);
outsize = (ctx->packedFormat || out_frm_format == OUTPUT_I422 || out_frm_format == OUTPUT_NV16 || out_frm_format == OUTPUT_NV61) ? out_width*out_height*2 : out_width*out_height*3/2;
outsize*=2;//align margin
if ((pYuv_int16 = (Uint16*)osal_malloc(outsize * sizeof(Uint16))) == NULL) {
printf("malloc error: outbuf\n");
return 0;
}
if ((pYuv2_int16 = (Uint16*)osal_malloc(outsize * sizeof(Uint16))) == NULL) {
osal_free(pYuv_int16);
printf("malloc error: outbuf\n");
return 0;
}
if ( bitdepth == 8 ) {
if( !osal_fread(pYuv_int8, 1, (Int32)fileSize, yuvFp) ) {
if( !osal_feof( yuvFp ) )
VLOG(ERR, "YUV Data osal_fread failed file handle is 0x%x \n", yuvFp );
if ( pYuv_int16 )
osal_free(pYuv_int16);
if ( pYuv2_int16 )
osal_free(pYuv2_int16);
return FALSE;
}
if ( subFrameSyncEn && (subFrameSyncSrcWriteMode & REMAIN_SRC_DATA_WRITE) == 0)
osal_fseek(yuvFp, -((long)fileSize), SEEK_CUR);
//Already wrote in SRC_XXXLINE_WRITE mode
if ( subFrameSyncEn && (subFrameSyncSrcWriteMode & REMAIN_SRC_DATA_WRITE) && writeSrcLine == picHeight) {
if ( pYuv_int16 )
osal_free(pYuv_int16);
if ( pYuv2_int16 )
osal_free(pYuv2_int16);
return TRUE;
}
for (i=0; i<fileSize; i++)
pYuv_int16[i] = (unsigned short) pYuv_int8[i];
}
else {//10bit
if( !osal_fread(pYuv_int16, 1, (Int32)fileSize, yuvFp) ) {
if( !osal_feof( yuvFp ) )
VLOG(ERR, "YUV Data osal_fread failed file handle is 0x%x \n", yuvFp );
if ( pYuv_int16 )
osal_free(pYuv_int16);
if ( pYuv2_int16 )
osal_free(pYuv2_int16);
return 0;
}
if ( subFrameSyncEn && (subFrameSyncSrcWriteMode & REMAIN_SRC_DATA_WRITE) == 0)
osal_fseek(yuvFp, -((long)fileSize), SEEK_CUR);
//Already wrote in SRC_XXXLINE_WRITE mode
if ( subFrameSyncEn && (subFrameSyncSrcWriteMode & REMAIN_SRC_DATA_WRITE) && writeSrcLine == picHeight) {
if ( pYuv_int16 )
osal_free(pYuv_int16);
if ( pYuv2_int16 )
osal_free(pYuv2_int16);
return TRUE;
}
if ( yuv3p4b == 0 ) {
/**************************************************************
* when 10bit source YUV == LITTLE ENDIAN, 16LSB, alway do byte-swapping
**************************************************************/
for (i=0; i<fileSize/2; i++) {
pYuv_int8[i*2] = (Uint8)(pYuv_int16[i]>>8)&0xff;
pYuv_int8[i*2 + 1] = (Uint8)(pYuv_int16[i])&0xff;
}
#ifdef BIG_ENDIAN_YUV //big endian yuv
/***************************************************************
* when 10bit source YUV == BIG ENDIAN, 16LSB
**************************************************************/
osal_memcpy(pYuv_int8, pYuv_int16, fileSize);
#endif
}
else {
/***************************************************************
* when 10bit source YUV == LITTLE ENDIAN, 16LSB
**************************************************************/
osal_memcpy(pYuv_int8, pYuv_int16, (Int32)fileSize);
#ifdef BIG_ENDIAN_YUV
/***************************************************************
* when 10bit source YUV == BIG ENDIAN, 16LSB
**************************************************************/
for (i=0; i<fileSize/2; i++) {
pYuv_int8[i*2] = (Uint8)(pYuv_int16[i]>>8)&0xff;
pYuv_int8[i*2 + 1] = (Uint8)(pYuv_int16[i])&0xff;
}
#endif
}
}
if (yuv3p4b==1 && ctx->packedFormat) {
ystride = (Int32)picWidth;
} else if (yuv3p4b == 1) {
ystride = VPU_ALIGN32((Int32)picWidth);
} else {
ystride = VPU_ALIGN16((Int32)picWidth);
}
if ( out_frm_format != OUTPUT_PLANAR || ctx->packedFormat || yuv3p4b || (pxl_format == PIXEL_FORMAT_16BIT_LSB)) {
conv_image((unsigned short *)pYuv2_int16, ((bitdepth==8) ? ((unsigned short*)pYuv_int16) : ((unsigned short *)pYuv_int8)), (Int32)picWidth, (Int32)picHeight, out_width, out_height, out_frm_format, ctx->packedFormat);
write_image_to_mem((unsigned short *)pYuv_int8, (unsigned short *)pYuv2_int16, ystride, out_height, pxl_format, out_frm_format, ctx->packedFormat);
}
#if 0
{
char fileName[MAX_FILE_PATH];
#ifdef SAVE_WHOLE_LOADED_YUV
#else
FILE *save_fp;
#endif
char *frm_string;
char *pixel_string;
if ( ctx->packedFormat == 1)
frm_string = "YUYV";
else if ( ctx->packedFormat == 2)
frm_string = "YVYU";
else if ( ctx->packedFormat == 3)
frm_string = "UYVY";
else if ( ctx->packedFormat == 4)
frm_string = "VYUY";
else if ( out_frm_format == OUTPUT_NV12)
frm_string = "NV12";
else if ( out_frm_format == OUTPUT_NV21)
frm_string = "NV21";
else if ( out_frm_format == OUTPUT_NV16)
frm_string = "NV16";
else if ( out_frm_format == OUTPUT_NV61)
frm_string = "NV61";
else if ( out_frm_format == OUTPUT_I422)
frm_string = "I422";
else
frm_string = "I420";
if ( pxl_format == PIXEL_FORMAT_32BIT_MSB )
pixel_string = "3P4B_MSB";
else if ( pxl_format == PIXEL_FORMAT_32BIT_LSB )
pixel_string = "3P4B_LSB";
else if ( pxl_format == PIXEL_FORMAT_16BIT_MSB )
pixel_string = "1P2B_MSB";
else if ( pxl_format == PIXEL_FORMAT_16BIT_LSB )
pixel_string = "1P2B_LSB";
else
pixel_string = "8b";
sprintf(fileName, "loaded_%s_%dx%d_%s.yuv", pixel_string, picWidth, picHeight, frm_string);
VLOG(INFO, "save %s\n", fileName);
#ifdef SAVE_WHOLE_LOADED_YUV
if ( save_frame_cnt == 0)
save_fp = osal_fopen(fileName, "wb");
#else
save_fp = osal_fopen(fileName, "wb");
#endif
if ( yuv3p4b == 0 )
write_image_to_file(save_fp, (unsigned short *)pYuv_int8, out_width, out_height, pxl_format, out_frm_format, ctx->packedFormat);
else
write_image_to_file(save_fp, (unsigned short *)pYuv2_int16, out_width, out_height, pxl_format, out_frm_format, ctx->packedFormat);
osal_fflush(save_fp);
#ifdef SAVE_WHOLE_LOADED_YUV
if ( ++save_frame_cnt == save_frame_max )
osal_fclose(save_fp);
#else
osal_fclose(save_fp);
#endif
}
#endif
if ( yuv3p4b && ctx->packedFormat) {
out_width = (Int32)(((picWidth*2)+2)/3*4);
}
else if(ctx->packedFormat) {
out_width *= 2; // 8bit packed mode(YUYV) only. (need to add 10bit cases later)
if (bitdepth == 10) // 10bit packed
out_width *= 2;
}
if (productId == PRODUCT_ID_960 || productId == PRODUCT_ID_980) {
//CODA series
if (LINEAR_FRAME_MAP != fb->mapType) {
LoadTiledImageYuvBurst(impl->handle, coreIdx, pYuv_int8, out_width, out_height, fb, *((TiledMapConfig*)arg));
} else {
LoadYuvImageByYCbCrLine(impl->handle, coreIdx, pYuv_int8, out_width, out_height, fb, arg, subFrameSyncConfig, srcFbIndex);
}
} else {
//WAVE series
#ifdef CNM_FPGA_VU440_INTERFACE
LoadYuvImageBurstFormat(coreIdx, pYuv_int8, out_width, out_height, fb, TRUE);
#else
LoadYuvImageByYCbCrLine(impl->handle, coreIdx, pYuv_int8, out_width, out_height, fb, arg, subFrameSyncConfig, srcFbIndex);
#endif
}
if ( pYuv_int16 )
osal_free(pYuv_int16);
if ( pYuv2_int16 )
osal_free(pYuv2_int16);
return TRUE;
}
BOOL loaderYuvFeeder_Configure(
YuvFeederImpl* impl,
Uint32 cmd,
YuvInfo yuv
)
{
yuvLoaderContext* ctx = (yuvLoaderContext*)impl->context;
UNREFERENCED_PARAMETER(cmd);
ctx->cbcrInterleave = yuv.cbcrInterleave;
ctx->nv21 = yuv.nv21;
ctx->i422 = yuv.i422;
ctx->packedFormat = yuv.packedFormat;
return TRUE;
}
YuvFeederImpl loaderYuvFeederImpl = {
NULL,
loaderYuvFeeder_Create,
loaderYuvFeeder_Feed,
NULL,
loaderYuvFeeder_Destory,
loaderYuvFeeder_Configure,
NULL
};