| /* |
| * 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; |
| } |
| |
| #ifdef LIBVA_UTILS_UPLOAD_DOWNLOAD_YUV_SURFACE |
| |
| /* |
| * 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; |
| } |
| |
| #endif /* LIBVA_UTILS_UPLOAD_DOWNLOAD_YUV_SURFACE */ |