summaryrefslogtreecommitdiffstats
path: root/src/gallium/state_trackers/va/image.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/gallium/state_trackers/va/image.c')
-rw-r--r--src/gallium/state_trackers/va/image.c266
1 files changed, 258 insertions, 8 deletions
diff --git a/src/gallium/state_trackers/va/image.c b/src/gallium/state_trackers/va/image.c
index 8aaa29cee2d..cd4044a1a76 100644
--- a/src/gallium/state_trackers/va/image.c
+++ b/src/gallium/state_trackers/va/image.c
@@ -26,18 +26,66 @@
*
**************************************************************************/
+#include "pipe/p_screen.h"
+
+#include "util/u_memory.h"
+#include "util/u_handle_table.h"
+#include "util/u_surface.h"
+#include "util/u_video.h"
+
+#include "vl/vl_winsys.h"
+
#include "va_private.h"
+static const VAImageFormat formats[VL_VA_MAX_IMAGE_FORMATS] =
+{
+ {VA_FOURCC('N','V','1','2')},
+ {VA_FOURCC('I','4','2','0')},
+ {VA_FOURCC('Y','V','1','2')},
+ {VA_FOURCC('Y','U','Y','V')},
+ {VA_FOURCC('U','Y','V','Y')},
+};
+
+static void
+vlVaVideoSurfaceSize(vlVaSurface *p_surf, int component,
+ unsigned *width, unsigned *height)
+{
+ *width = p_surf->templat.width;
+ *height = p_surf->templat.height;
+
+ if (component > 0) {
+ if (p_surf->templat.chroma_format == PIPE_VIDEO_CHROMA_FORMAT_420) {
+ *width /= 2;
+ *height /= 2;
+ } else if (p_surf->templat.chroma_format == PIPE_VIDEO_CHROMA_FORMAT_422)
+ *width /= 2;
+ }
+ if (p_surf->templat.interlaced)
+ *height /= 2;
+}
+
VAStatus
vlVaQueryImageFormats(VADriverContextP ctx, VAImageFormat *format_list, int *num_formats)
{
+ struct pipe_screen *pscreen;
+ enum pipe_format format;
+ int i;
+
if (!ctx)
return VA_STATUS_ERROR_INVALID_CONTEXT;
if (!(format_list && num_formats))
- return VA_STATUS_ERROR_UNKNOWN;
+ return VA_STATUS_ERROR_INVALID_PARAMETER;
*num_formats = 0;
+ pscreen = VL_VA_PSCREEN(ctx);
+ for (i = 0; i < VL_VA_MAX_IMAGE_FORMATS; ++i) {
+ format = YCbCrToPipe(formats[i].fourcc);
+ if (pscreen->is_video_format_supported(pscreen, format,
+ PIPE_VIDEO_PROFILE_UNKNOWN,
+ PIPE_VIDEO_ENTRYPOINT_BITSTREAM))
+ format_list[(*num_formats)++] = formats[i];
+ }
return VA_STATUS_SUCCESS;
}
@@ -45,16 +93,61 @@ vlVaQueryImageFormats(VADriverContextP ctx, VAImageFormat *format_list, int *num
VAStatus
vlVaCreateImage(VADriverContextP ctx, VAImageFormat *format, int width, int height, VAImage *image)
{
+ vlVaDriver *drv;
+ int w, h;
+
if (!ctx)
return VA_STATUS_ERROR_INVALID_CONTEXT;
- if(!format)
- return VA_STATUS_ERROR_UNKNOWN;
+ if (!(format && image && width && height))
+ return VA_STATUS_ERROR_INVALID_PARAMETER;
+
+ drv = VL_VA_DRIVER(ctx);
- if (!(width && height))
+ image->image_id = handle_table_add(drv->htab, image);
+ image->format = *format;
+ image->width = width;
+ image->height = height;
+ w = align(width, 2);
+ h = align(width, 2);
+
+ switch (format->fourcc) {
+ case VA_FOURCC('N','V','1','2'):
+ image->num_planes = 2;
+ image->pitches[0] = w;
+ image->offsets[0] = 0;
+ image->pitches[1] = w;
+ image->offsets[1] = w * h;
+ image->data_size = w * h * 3 / 2;
+ break;
+
+ case VA_FOURCC('I','4','2','0'):
+ case VA_FOURCC('Y','V','1','2'):
+ image->num_planes = 3;
+ image->pitches[0] = w;
+ image->offsets[0] = 0;
+ image->pitches[1] = w / 2;
+ image->offsets[1] = w * h;
+ image->pitches[2] = w / 2;
+ image->offsets[2] = w * h * 5 / 4;
+ image->data_size = w * h * 3 / 2;
+ break;
+
+ case VA_FOURCC('U','Y','V','Y'):
+ case VA_FOURCC('Y','U','Y','V'):
+ image->num_planes = 1;
+ image->pitches[0] = w * 2;
+ image->offsets[0] = 0;
+ image->data_size = w * h * 2;
+ break;
+
+ default:
return VA_STATUS_ERROR_INVALID_IMAGE_FORMAT;
+ }
- return VA_STATUS_ERROR_UNIMPLEMENTED;
+ return vlVaCreateBuffer(ctx, 0, VAImageBufferType,
+ align(image->data_size, 16),
+ 1, NULL, &image->buf);
}
VAStatus
@@ -69,10 +162,16 @@ vlVaDeriveImage(VADriverContextP ctx, VASurfaceID surface, VAImage *image)
VAStatus
vlVaDestroyImage(VADriverContextP ctx, VAImageID image)
{
+ VAImage *vaimage;
+
if (!ctx)
return VA_STATUS_ERROR_INVALID_CONTEXT;
- return VA_STATUS_ERROR_UNIMPLEMENTED;
+ vaimage = handle_table_get(VL_VA_DRIVER(ctx)->htab, image);
+ if (!vaimage)
+ return VA_STATUS_ERROR_INVALID_IMAGE;
+
+ return vlVaDestroyBuffer(ctx, vaimage->buf);
}
VAStatus
@@ -88,10 +187,93 @@ VAStatus
vlVaGetImage(VADriverContextP ctx, VASurfaceID surface, int x, int y,
unsigned int width, unsigned int height, VAImageID image)
{
+ vlVaDriver *drv;
+ vlVaSurface *surf;
+ vlVaBuffer *img_buf;
+ VAImage *vaimage;
+ struct pipe_sampler_view **views;
+ enum pipe_format format;
+ bool convert = false;
+ void *data[3];
+ unsigned pitches[3], i, j;
+
if (!ctx)
return VA_STATUS_ERROR_INVALID_CONTEXT;
- return VA_STATUS_ERROR_UNIMPLEMENTED;
+ drv = VL_VA_DRIVER(ctx);
+
+ surf = handle_table_get(drv->htab, surface);
+ if (!surf || !surf->buffer)
+ return VA_STATUS_ERROR_INVALID_SURFACE;
+
+ vaimage = handle_table_get(drv->htab, image);
+ if (!vaimage)
+ return VA_STATUS_ERROR_INVALID_IMAGE;
+
+ img_buf = handle_table_get(drv->htab, vaimage->buf);
+ if (!img_buf)
+ return VA_STATUS_ERROR_INVALID_BUFFER;
+
+ format = YCbCrToPipe(vaimage->format.fourcc);
+ if (format == PIPE_FORMAT_NONE)
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+
+ if (format != surf->buffer->buffer_format) {
+ /* support NV12 to YV12 conversion now only */
+ if (format == PIPE_FORMAT_YV12 &&
+ surf->buffer->buffer_format == PIPE_FORMAT_NV12)
+ convert = true;
+ else
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+ }
+
+ views = surf->buffer->get_sampler_view_planes(surf->buffer);
+ if (!views)
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+
+ for (i = 0; i < vaimage->num_planes; i++) {
+ data[i] = img_buf->data + vaimage->offsets[i];
+ pitches[i] = vaimage->pitches[i];
+ }
+ if (vaimage->format.fourcc == VA_FOURCC('I','4','2','0')) {
+ void *tmp_d;
+ unsigned tmp_p;
+ tmp_d = data[1];
+ data[1] = data[2];
+ data[2] = tmp_d;
+ tmp_p = pitches[1];
+ pitches[1] = pitches[2];
+ pitches[2] = tmp_p;
+ }
+
+ for (i = 0; i < vaimage->num_planes; i++) {
+ unsigned width, height;
+ if (!views[i]) continue;
+ vlVaVideoSurfaceSize(surf, i, &width, &height);
+ for (j = 0; j < views[i]->texture->array_size; ++j) {
+ struct pipe_box box = {0, 0, j, width, height, 1};
+ struct pipe_transfer *transfer;
+ uint8_t *map;
+ map = drv->pipe->transfer_map(drv->pipe, views[i]->texture, 0,
+ PIPE_TRANSFER_READ, &box, &transfer);
+ if (!map)
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+
+ if (i == 1 && convert) {
+ u_copy_nv12_to_yv12(data, pitches, i, j,
+ transfer->stride, views[i]->texture->array_size,
+ map, box.width, box.height);
+ } else {
+ util_copy_rect(data[i] + pitches[i] * j,
+ views[i]->texture->format,
+ pitches[i] * views[i]->texture->array_size, 0, 0,
+ box.width, box.height, map, transfer->stride, 0, 0);
+ }
+ pipe_transfer_unmap(drv->pipe, transfer);
+ }
+ }
+
+ return VA_STATUS_SUCCESS;
}
VAStatus
@@ -99,8 +281,76 @@ vlVaPutImage(VADriverContextP ctx, VASurfaceID surface, VAImageID image,
int src_x, int src_y, unsigned int src_width, unsigned int src_height,
int dest_x, int dest_y, unsigned int dest_width, unsigned int dest_height)
{
+ vlVaDriver *drv;
+ vlVaSurface *surf;
+ vlVaBuffer *img_buf;
+ VAImage *vaimage;
+ struct pipe_sampler_view **views;
+ enum pipe_format format;
+ void *data[3];
+ unsigned pitches[3], i, j;
+
if (!ctx)
return VA_STATUS_ERROR_INVALID_CONTEXT;
- return VA_STATUS_ERROR_UNIMPLEMENTED;
+ drv = VL_VA_DRIVER(ctx);
+
+ surf = handle_table_get(drv->htab, surface);
+ if (!surf || !surf->buffer)
+ return VA_STATUS_ERROR_INVALID_SURFACE;
+
+ vaimage = handle_table_get(drv->htab, image);
+ if (!vaimage)
+ return VA_STATUS_ERROR_INVALID_IMAGE;
+
+ img_buf = handle_table_get(drv->htab, vaimage->buf);
+ if (!img_buf)
+ return VA_STATUS_ERROR_INVALID_BUFFER;
+
+ format = YCbCrToPipe(vaimage->format.fourcc);
+ if (format == PIPE_FORMAT_NONE)
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+
+ if (surf->buffer == NULL || format != surf->buffer->buffer_format) {
+ if (surf->buffer)
+ surf->buffer->destroy(surf->buffer);
+ surf->templat.buffer_format = format;
+ surf->buffer = drv->pipe->create_video_buffer(drv->pipe, &surf->templat);
+ if (!surf->buffer)
+ return VA_STATUS_ERROR_ALLOCATION_FAILED;
+ }
+
+ views = surf->buffer->get_sampler_view_planes(surf->buffer);
+ if (!views)
+ return VA_STATUS_ERROR_OPERATION_FAILED;
+
+ for (i = 0; i < vaimage->num_planes; i++) {
+ data[i] = img_buf->data + vaimage->offsets[i];
+ pitches[i] = vaimage->pitches[i];
+ }
+ if (vaimage->format.fourcc == VA_FOURCC('I','4','2','0')) {
+ void *tmp_d;
+ unsigned tmp_p;
+ tmp_d = data[1];
+ data[1] = data[2];
+ data[2] = tmp_d;
+ tmp_p = pitches[1];
+ pitches[1] = pitches[2];
+ pitches[2] = tmp_p;
+ }
+
+ for (i = 0; i < vaimage->num_planes; ++i) {
+ unsigned width, height;
+ if (!views[i]) continue;
+ vlVaVideoSurfaceSize(surf, i, &width, &height);
+ for (j = 0; j < views[i]->texture->array_size; ++j) {
+ struct pipe_box dst_box = {0, 0, j, width, height, 1};
+ drv->pipe->transfer_inline_write(drv->pipe, views[i]->texture, 0,
+ PIPE_TRANSFER_WRITE, &dst_box,
+ data[i] + pitches[i] * j,
+ pitches[i] * views[i]->texture->array_size, 0);
+ }
+ }
+
+ return VA_STATUS_SUCCESS;
}