diff options
Diffstat (limited to 'test/loadsurface.h')
-rwxr-xr-x | test/loadsurface.h | 489 |
1 files changed, 0 insertions, 489 deletions
diff --git a/test/loadsurface.h b/test/loadsurface.h deleted file mode 100755 index 82cf61c..0000000 --- a/test/loadsurface.h +++ /dev/null @@ -1,489 +0,0 @@ -/* - * Copyright (c) 2008-2009 Intel Corporation. All Rights Reserved. - * - * Permission is hereby granted, free of charge, to any person obtaining a - * copy of this software and associated documentation files (the - * "Software"), to deal in the Software without restriction, including - * without limitation the rights to use, copy, modify, merge, publish, - * distribute, sub license, and/or sell copies of the Software, and to - * permit persons to whom the Software is furnished to do so, subject to - * the following conditions: - * - * The above copyright notice and this permission notice (including the - * next paragraph) shall be included in all copies or substantial portions - * of the Software. - * - * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS - * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF - * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. - * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR - * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, - * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE - * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. - */ -#include "loadsurface_yuv.h" - -static int scale_2dimage(unsigned char *src_img, int src_imgw, int src_imgh, - unsigned char *dst_img, int dst_imgw, int dst_imgh) -{ - int row=0, col=0; - - for (row=0; row<dst_imgh; row++) { - for (col=0; col<dst_imgw; col++) { - *(dst_img + row * dst_imgw + col) = *(src_img + (row * src_imgh/dst_imgh) * src_imgw + col * src_imgw/dst_imgw); - } - } - - return 0; -} - - -static int YUV_blend_with_pic(int width, int height, - unsigned char *Y_start, int Y_pitch, - unsigned char *U_start, int U_pitch, - unsigned char *V_start, int V_pitch, - unsigned int fourcc, int fixed_alpha) -{ - /* PIC YUV format */ - unsigned char *pic_y_old = yuvga_pic; - unsigned char *pic_u_old = pic_y_old + 640*480; - unsigned char *pic_v_old = pic_u_old + 640*480/4; - unsigned char *pic_y, *pic_u, *pic_v; - - int alpha_values[] = {100,90,80,70,60,50,40,30,20,30,40,50,60,70,80,90}; - - static int alpha_idx = 0; - int alpha; - int allocated = 0; - - int row, col; - - if (fixed_alpha == 0) { - alpha = alpha_values[alpha_idx % 16 ]; - alpha_idx ++; - } else - alpha = fixed_alpha; - - //alpha = 0; - - pic_y = pic_y_old; - pic_u = pic_u_old; - pic_v = pic_v_old; - - if (width != 640 || height != 480) { /* need to scale the pic */ - pic_y = (unsigned char *)malloc(width * height); - if(pic_y == NULL) { - printf("Failed to allocate memory for pic_y\n"); - return -1; - } - - pic_u = (unsigned char *)malloc(width * height/4); - if(pic_u == NULL) { - printf("Failed to allocate memory for pic_u\n"); - free(pic_y); - return -1; - } - - pic_v = (unsigned char *)malloc(width * height/4); - if(pic_v == NULL) { - printf("Failed to allocate memory for pic_v\n"); - free(pic_y); - free(pic_u); - return -1; - } - allocated = 1; - - memset(pic_y, 0, width * height); - memset(pic_u, 0, width * height /4); - memset(pic_v, 0, width * height /4); - - scale_2dimage(pic_y_old, 640, 480, - pic_y, width, height); - scale_2dimage(pic_u_old, 320, 240, - pic_u, width/2, height/2); - scale_2dimage(pic_v_old, 320, 240, - pic_v, width/2, height/2); - } - - /* begin blend */ - - /* Y plane */ - int Y_pixel_stride = 1; - if (fourcc == VA_FOURCC_YUY2) - Y_pixel_stride = 2; - - for (row=0; row<height; row++) { - unsigned char *p = Y_start + row * Y_pitch; - unsigned char *q = pic_y + row * width; - for (col=0; col<width; col++, q++) { - *p = *p * (100 - alpha) / 100 + *q * alpha/100; - p += Y_pixel_stride; - } - } - - /* U/V plane */ - int U_pixel_stride = 0, V_pixel_stride = 0; - int v_factor_to_nv12 = 1; - switch (fourcc) { - case VA_FOURCC_YV12: - U_pixel_stride = V_pixel_stride = 1; - break; - case VA_FOURCC_NV12: - U_pixel_stride = V_pixel_stride = 2; - break; - case VA_FOURCC_YUY2: - U_pixel_stride = V_pixel_stride = 4; - v_factor_to_nv12 = 2; - break; - default: - break; - } - for (row=0; row<height/2*v_factor_to_nv12; row++) { - unsigned char *pU = U_start + row * U_pitch; - unsigned char *pV = V_start + row * V_pitch; - unsigned char *qU = pic_u + row/v_factor_to_nv12 * width/2; - unsigned char *qV = pic_v + row/v_factor_to_nv12 * width/2; - - for (col=0; col<width/2; col++, qU++, qV++) { - *pU = *pU * (100 - alpha) / 100 + *qU * alpha/100; - *pV = *pV * (100 - alpha) / 100 + *qV * alpha/100; - - pU += U_pixel_stride; - pV += V_pixel_stride; - } - } - - if (allocated) { - free(pic_y); - free(pic_u); - free(pic_v); - } - - return 0; -} - -static int yuvgen_planar(int width, int height, - unsigned char *Y_start, int Y_pitch, - unsigned char *U_start, int U_pitch, - unsigned char *V_start, int V_pitch, - unsigned int fourcc, int box_width, int row_shift, - int field) -{ - int row, alpha; - unsigned char uv_value = 0x80; - - /* copy Y plane */ - int y_factor = 1; - if (fourcc == VA_FOURCC_YUY2) y_factor = 2; - for (row=0;row<height;row++) { - unsigned char *Y_row = Y_start + row * Y_pitch; - int jj, xpos, ypos; - - ypos = (row / box_width) & 0x1; - - /* fill garbage data into the other field */ - if (((field == VA_TOP_FIELD) && (row &1)) - || ((field == VA_BOTTOM_FIELD) && ((row &1)==0))) { - memset(Y_row, 0xff, width); - continue; - } - - for (jj=0; jj<width; jj++) { - xpos = ((row_shift + jj) / box_width) & 0x1; - if (xpos == ypos) - Y_row[jj*y_factor] = 0xeb; - else - Y_row[jj*y_factor] = 0x10; - - if (fourcc == VA_FOURCC_YUY2) { - Y_row[jj*y_factor+1] = uv_value; // it is for UV - } - } - } - - /* copy UV data */ - for( row =0; row < height/2; row++) { - - /* fill garbage data into the other field */ - if (((field == VA_TOP_FIELD) && (row &1)) - || ((field == VA_BOTTOM_FIELD) && ((row &1)==0))) { - uv_value = 0xff; - } - - unsigned char *U_row = U_start + row * U_pitch; - unsigned char *V_row = V_start + row * V_pitch; - switch (fourcc) { - case VA_FOURCC_NV12: - memset(U_row, uv_value, width); - break; - case VA_FOURCC_YV12: - memset (U_row,uv_value,width/2); - memset (V_row,uv_value,width/2); - break; - case VA_FOURCC_YUY2: - // see above. it is set with Y update. - break; - default: - printf("unsupported fourcc in loadsurface.h\n"); - assert(0); - } - } - - if (getenv("AUTO_NOUV")) - return 0; - - if (getenv("AUTO_ALPHA")) - alpha = 0; - else - alpha = 70; - - YUV_blend_with_pic(width,height, - Y_start, Y_pitch, - U_start, U_pitch, - V_start, V_pitch, - fourcc, alpha); - - return 0; -} - -static int upload_surface(VADisplay va_dpy, VASurfaceID surface_id, - int box_width, int row_shift, - int field) -{ - VAImage surface_image; - void *surface_p=NULL, *U_start = NULL,*V_start = NULL; - VAStatus va_status; - unsigned int pitches[3]={0,0,0}; - - va_status = vaDeriveImage(va_dpy,surface_id,&surface_image); - CHECK_VASTATUS(va_status,"vaDeriveImage"); - - vaMapBuffer(va_dpy,surface_image.buf,&surface_p); - assert(VA_STATUS_SUCCESS == va_status); - - pitches[0] = surface_image.pitches[0]; - switch (surface_image.format.fourcc) { - case VA_FOURCC_NV12: - U_start = (char *)surface_p + surface_image.offsets[1]; - V_start = (char *)U_start + 1; - pitches[1] = surface_image.pitches[1]; - pitches[2] = surface_image.pitches[1]; - break; - case VA_FOURCC_IYUV: - U_start = (char *)surface_p + surface_image.offsets[1]; - V_start = (char *)surface_p + surface_image.offsets[2]; - pitches[1] = surface_image.pitches[1]; - pitches[2] = surface_image.pitches[2]; - break; - case VA_FOURCC_YV12: - U_start = (char *)surface_p + surface_image.offsets[2]; - V_start = (char *)surface_p + surface_image.offsets[1]; - pitches[1] = surface_image.pitches[2]; - pitches[2] = surface_image.pitches[1]; - break; - case VA_FOURCC_YUY2: - U_start = (char *)surface_p + 1; - V_start = (char *)surface_p + 3; - pitches[1] = surface_image.pitches[0]; - pitches[2] = surface_image.pitches[0]; - break; - default: - assert(0); - } - - /* assume surface is planar format */ - yuvgen_planar(surface_image.width, surface_image.height, - (unsigned char *)surface_p, pitches[0], - (unsigned char *)U_start, pitches[1], - (unsigned char *)V_start, pitches[2], - surface_image.format.fourcc, - box_width, row_shift, field); - - vaUnmapBuffer(va_dpy,surface_image.buf); - - vaDestroyImage(va_dpy,surface_image.image_id); - - return 0; -} - -/* - * Upload YUV data from memory into a surface - * if src_fourcc == NV12, assume the buffer pointed by src_U - * is UV interleaved (src_V is ignored) - */ -static int upload_surface_yuv(VADisplay va_dpy, VASurfaceID surface_id, - int src_fourcc, int src_width, int src_height, - unsigned char *src_Y, unsigned char *src_U, unsigned char *src_V) -{ - VAImage surface_image; - unsigned char *surface_p=NULL, *Y_start=NULL, *U_start=NULL; - int Y_pitch=0, U_pitch=0, row; - VAStatus va_status; - - va_status = vaDeriveImage(va_dpy,surface_id, &surface_image); - CHECK_VASTATUS(va_status,"vaDeriveImage"); - - vaMapBuffer(va_dpy,surface_image.buf,(void **)&surface_p); - assert(VA_STATUS_SUCCESS == va_status); - - Y_start = surface_p; - Y_pitch = surface_image.pitches[0]; - switch (surface_image.format.fourcc) { - case VA_FOURCC_NV12: - U_start = (unsigned char *)surface_p + surface_image.offsets[1]; - U_pitch = surface_image.pitches[1]; - break; - case VA_FOURCC_IYUV: - U_start = (unsigned char *)surface_p + surface_image.offsets[1]; - U_pitch = surface_image.pitches[1]; - break; - case VA_FOURCC_YV12: - U_start = (unsigned char *)surface_p + surface_image.offsets[2]; - U_pitch = surface_image.pitches[2]; - break; - case VA_FOURCC_YUY2: - U_start = surface_p + 1; - U_pitch = surface_image.pitches[0]; - break; - default: - assert(0); - } - - /* copy Y plane */ - for (row=0;row<src_height;row++) { - unsigned char *Y_row = Y_start + row * Y_pitch; - memcpy(Y_row, src_Y + row*src_width, src_width); - } - - for (row =0; row < src_height/2; row++) { - unsigned char *U_row = U_start + row * U_pitch; - unsigned char *u_ptr = NULL, *v_ptr=NULL; - int j; - switch (surface_image.format.fourcc) { - case VA_FOURCC_NV12: - if (src_fourcc == VA_FOURCC_NV12) { - memcpy(U_row, src_U + row * src_width, src_width); - break; - } else if (src_fourcc == VA_FOURCC_IYUV) { - u_ptr = src_U + row * (src_width/2); - v_ptr = src_V + row * (src_width/2); - } else if (src_fourcc == VA_FOURCC_YV12) { - v_ptr = src_U + row * (src_width/2); - u_ptr = src_V + row * (src_width/2); - } - if ((src_fourcc == VA_FOURCC_IYUV) || - (src_fourcc == VA_FOURCC_YV12)) { - for(j = 0; j < src_width/2; j++) { - U_row[2*j] = u_ptr[j]; - U_row[2*j+1] = v_ptr[j]; - } - } - break; - case VA_FOURCC_IYUV: - case VA_FOURCC_YV12: - case VA_FOURCC_YUY2: - default: - printf("unsupported fourcc in load_surface_yuv\n"); - assert(0); - } - } - - vaUnmapBuffer(va_dpy,surface_image.buf); - - vaDestroyImage(va_dpy,surface_image.image_id); - - return 0; -} - -/* - * Download YUV data from a surface into memory - * Some hardward doesn't have a aperture for linear access of - * tiled surface, thus use vaGetImage to expect the implemnetion - * to do tile to linear convert - * - * if dst_fourcc == NV12, assume the buffer pointed by dst_U - * is UV interleaved (src_V is ignored) - */ -static int download_surface_yuv(VADisplay va_dpy, VASurfaceID surface_id, - int dst_fourcc, int dst_width, int dst_height, - unsigned char *dst_Y, unsigned char *dst_U, unsigned char *dst_V) -{ - VAImage surface_image; - unsigned char *surface_p=NULL, *Y_start=NULL, *U_start=NULL; - int Y_pitch=0, U_pitch=0, row; - VAStatus va_status; - - va_status = vaDeriveImage(va_dpy,surface_id, &surface_image); - CHECK_VASTATUS(va_status,"vaDeriveImage"); - - vaMapBuffer(va_dpy,surface_image.buf,(void **)&surface_p); - assert(VA_STATUS_SUCCESS == va_status); - - Y_start = surface_p; - Y_pitch = surface_image.pitches[0]; - switch (surface_image.format.fourcc) { - case VA_FOURCC_NV12: - U_start = (unsigned char *)surface_p + surface_image.offsets[1]; - U_pitch = surface_image.pitches[1]; - break; - case VA_FOURCC_IYUV: - U_start = (unsigned char *)surface_p + surface_image.offsets[1]; - U_pitch = surface_image.pitches[1]; - break; - case VA_FOURCC_YV12: - U_start = (unsigned char *)surface_p + surface_image.offsets[2]; - U_pitch = surface_image.pitches[2]; - break; - case VA_FOURCC_YUY2: - U_start = surface_p + 1; - U_pitch = surface_image.pitches[0]; - break; - default: - assert(0); - } - - /* copy Y plane */ - for (row=0;row<dst_height;row++) { - unsigned char *Y_row = Y_start + row * Y_pitch; - memcpy(dst_Y + row*dst_width, Y_row, dst_width); - } - - for (row =0; row < dst_height/2; row++) { - unsigned char *U_row = U_start + row * U_pitch; - unsigned char *u_ptr = NULL, *v_ptr = NULL; - int j; - switch (surface_image.format.fourcc) { - case VA_FOURCC_NV12: - if (dst_fourcc == VA_FOURCC_NV12) { - memcpy(dst_U + row * dst_width, U_row, dst_width); - break; - } else if (dst_fourcc == VA_FOURCC_IYUV) { - u_ptr = dst_U + row * (dst_width/2); - v_ptr = dst_V + row * (dst_width/2); - } else if (dst_fourcc == VA_FOURCC_YV12) { - v_ptr = dst_U + row * (dst_width/2); - u_ptr = dst_V + row * (dst_width/2); - } - if ((dst_fourcc == VA_FOURCC_IYUV) || - (dst_fourcc == VA_FOURCC_YV12)) { - for(j = 0; j < dst_width/2; j++) { - u_ptr[j] = U_row[2*j]; - v_ptr[j] = U_row[2*j+1]; - } - } - break; - case VA_FOURCC_IYUV: - case VA_FOURCC_YV12: - case VA_FOURCC_YUY2: - default: - printf("unsupported fourcc in load_surface_yuv\n"); - assert(0); - } - } - - vaUnmapBuffer(va_dpy,surface_image.buf); - - vaDestroyImage(va_dpy,surface_image.image_id); - - return 0; -} |