• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**************************************************************************
2  *
3  * Copyright 2011 Christian König.
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the
8  * "Software"), to deal in the Software without restriction, including
9  * without limitation the rights to use, copy, modify, merge, publish,
10  * distribute, sub license, and/or sell copies of the Software, and to
11  * permit persons to whom the Software is furnished to do so, subject to
12  * the following conditions:
13  *
14  * The above copyright notice and this permission notice (including the
15  * next paragraph) shall be included in all copies or substantial portions
16  * of the Software.
17  *
18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25  *
26  **************************************************************************/
27 
28 #include <assert.h>
29 
30 #include "pipe/p_screen.h"
31 #include "pipe/p_context.h"
32 #include "pipe/p_state.h"
33 
34 #include "util/u_format.h"
35 #include "util/u_inlines.h"
36 #include "util/u_sampler.h"
37 #include "util/u_memory.h"
38 
39 #include "vl_video_buffer.h"
40 
41 const enum pipe_format const_resource_formats_YV12[3] = {
42    PIPE_FORMAT_R8_UNORM,
43    PIPE_FORMAT_R8_UNORM,
44    PIPE_FORMAT_R8_UNORM
45 };
46 
47 const enum pipe_format const_resource_formats_NV12[3] = {
48    PIPE_FORMAT_R8_UNORM,
49    PIPE_FORMAT_R8G8_UNORM,
50    PIPE_FORMAT_NONE
51 };
52 
53 const enum pipe_format const_resource_formats_YUVA[3] = {
54    PIPE_FORMAT_R8G8B8A8_UNORM,
55    PIPE_FORMAT_NONE,
56    PIPE_FORMAT_NONE
57 };
58 
59 const enum pipe_format const_resource_formats_VUYA[3] = {
60    PIPE_FORMAT_B8G8R8A8_UNORM,
61    PIPE_FORMAT_NONE,
62    PIPE_FORMAT_NONE
63 };
64 
65 const enum pipe_format const_resource_formats_YUVX[3] = {
66    PIPE_FORMAT_R8G8B8X8_UNORM,
67    PIPE_FORMAT_NONE,
68    PIPE_FORMAT_NONE
69 };
70 
71 const enum pipe_format const_resource_formats_VUYX[3] = {
72    PIPE_FORMAT_B8G8R8X8_UNORM,
73    PIPE_FORMAT_NONE,
74    PIPE_FORMAT_NONE
75 };
76 
77 const enum pipe_format const_resource_formats_YUYV[3] = {
78    PIPE_FORMAT_R8G8_R8B8_UNORM,
79    PIPE_FORMAT_NONE,
80    PIPE_FORMAT_NONE
81 };
82 
83 const enum pipe_format const_resource_formats_UYVY[3] = {
84    PIPE_FORMAT_G8R8_B8R8_UNORM,
85    PIPE_FORMAT_NONE,
86    PIPE_FORMAT_NONE
87 };
88 
89 const enum pipe_format const_resource_formats_P016[3] = {
90    PIPE_FORMAT_R16_UNORM,
91    PIPE_FORMAT_R16G16_UNORM,
92    PIPE_FORMAT_NONE
93 };
94 
95 const unsigned const_resource_plane_order_YUV[3] = {
96    0,
97    1,
98    2
99 };
100 
101 const unsigned const_resource_plane_order_YVU[3] = {
102    0,
103    2,
104    1
105 };
106 
107 const enum pipe_format *
vl_video_buffer_formats(struct pipe_screen * screen,enum pipe_format format)108 vl_video_buffer_formats(struct pipe_screen *screen, enum pipe_format format)
109 {
110    switch(format) {
111    case PIPE_FORMAT_YV12:
112       return const_resource_formats_YV12;
113 
114    case PIPE_FORMAT_NV12:
115       return const_resource_formats_NV12;
116 
117    case PIPE_FORMAT_R8G8B8A8_UNORM:
118       return const_resource_formats_YUVA;
119 
120    case PIPE_FORMAT_B8G8R8A8_UNORM:
121       return const_resource_formats_VUYA;
122 
123    case PIPE_FORMAT_R8G8B8X8_UNORM:
124       return const_resource_formats_YUVX;
125 
126    case PIPE_FORMAT_B8G8R8X8_UNORM:
127       return const_resource_formats_VUYX;
128 
129    case PIPE_FORMAT_YUYV:
130       return const_resource_formats_YUYV;
131 
132    case PIPE_FORMAT_UYVY:
133       return const_resource_formats_UYVY;
134 
135    case PIPE_FORMAT_P016:
136       return const_resource_formats_P016;
137 
138    default:
139       return NULL;
140    }
141 }
142 
143 const unsigned *
vl_video_buffer_plane_order(enum pipe_format format)144 vl_video_buffer_plane_order(enum pipe_format format)
145 {
146    switch(format) {
147    case PIPE_FORMAT_YV12:
148       return const_resource_plane_order_YVU;
149 
150    case PIPE_FORMAT_NV12:
151    case PIPE_FORMAT_R8G8B8A8_UNORM:
152    case PIPE_FORMAT_B8G8R8A8_UNORM:
153    case PIPE_FORMAT_YUYV:
154    case PIPE_FORMAT_UYVY:
155    case PIPE_FORMAT_P016:
156       return const_resource_plane_order_YUV;
157 
158    default:
159       return NULL;
160    }
161 }
162 
163 static enum pipe_format
vl_video_buffer_surface_format(enum pipe_format format)164 vl_video_buffer_surface_format(enum pipe_format format)
165 {
166    const struct util_format_description *desc = util_format_description(format);
167 
168    /* a subsampled formats can't work as surface use RGBA instead */
169    if (desc->layout == UTIL_FORMAT_LAYOUT_SUBSAMPLED)
170       return PIPE_FORMAT_R8G8B8A8_UNORM;
171 
172    return format;
173 }
174 
175 boolean
vl_video_buffer_is_format_supported(struct pipe_screen * screen,enum pipe_format format,enum pipe_video_profile profile,enum pipe_video_entrypoint entrypoint)176 vl_video_buffer_is_format_supported(struct pipe_screen *screen,
177                                     enum pipe_format format,
178                                     enum pipe_video_profile profile,
179                                     enum pipe_video_entrypoint entrypoint)
180 {
181    const enum pipe_format *resource_formats;
182    unsigned i;
183 
184    resource_formats = vl_video_buffer_formats(screen, format);
185    if (!resource_formats)
186       return false;
187 
188    for (i = 0; i < VL_NUM_COMPONENTS; ++i) {
189       enum pipe_format format = resource_formats[i];
190 
191       if (format == PIPE_FORMAT_NONE)
192          continue;
193 
194       /* we at least need to sample from it */
195       if (!screen->is_format_supported(screen, format, PIPE_TEXTURE_2D, 0, PIPE_BIND_SAMPLER_VIEW))
196          return false;
197 
198       format = vl_video_buffer_surface_format(format);
199       if (!screen->is_format_supported(screen, format, PIPE_TEXTURE_2D, 0, PIPE_BIND_RENDER_TARGET))
200          return false;
201    }
202 
203    return true;
204 }
205 
206 unsigned
vl_video_buffer_max_size(struct pipe_screen * screen)207 vl_video_buffer_max_size(struct pipe_screen *screen)
208 {
209    uint32_t max_2d_texture_level;
210 
211    max_2d_texture_level = screen->get_param(screen, PIPE_CAP_MAX_TEXTURE_2D_LEVELS);
212 
213    return 1 << (max_2d_texture_level-1);
214 }
215 
216 void
vl_video_buffer_set_associated_data(struct pipe_video_buffer * vbuf,struct pipe_video_codec * vcodec,void * associated_data,void (* destroy_associated_data)(void *))217 vl_video_buffer_set_associated_data(struct pipe_video_buffer *vbuf,
218                                     struct pipe_video_codec *vcodec,
219                                     void *associated_data,
220                                     void (*destroy_associated_data)(void *))
221 {
222    vbuf->codec = vcodec;
223 
224    if (vbuf->associated_data == associated_data)
225       return;
226 
227    if (vbuf->associated_data)
228       vbuf->destroy_associated_data(vbuf->associated_data);
229 
230    vbuf->associated_data = associated_data;
231    vbuf->destroy_associated_data = destroy_associated_data;
232 }
233 
234 void *
vl_video_buffer_get_associated_data(struct pipe_video_buffer * vbuf,struct pipe_video_codec * vcodec)235 vl_video_buffer_get_associated_data(struct pipe_video_buffer *vbuf,
236                                     struct pipe_video_codec *vcodec)
237 {
238    if (vbuf->codec == vcodec)
239       return vbuf->associated_data;
240    else
241       return NULL;
242 }
243 
244 void
vl_video_buffer_template(struct pipe_resource * templ,const struct pipe_video_buffer * tmpl,enum pipe_format resource_format,unsigned depth,unsigned array_size,unsigned usage,unsigned plane)245 vl_video_buffer_template(struct pipe_resource *templ,
246                          const struct pipe_video_buffer *tmpl,
247                          enum pipe_format resource_format,
248                          unsigned depth, unsigned array_size,
249                          unsigned usage, unsigned plane)
250 {
251    unsigned height = tmpl->height;
252 
253    memset(templ, 0, sizeof(*templ));
254    if (depth > 1)
255       templ->target = PIPE_TEXTURE_3D;
256    else if (array_size > 1)
257       templ->target = PIPE_TEXTURE_2D_ARRAY;
258    else
259       templ->target = PIPE_TEXTURE_2D;
260    templ->format = resource_format;
261    templ->width0 = tmpl->width;
262    templ->depth0 = depth;
263    templ->array_size = array_size;
264    templ->bind = PIPE_BIND_SAMPLER_VIEW | PIPE_BIND_RENDER_TARGET | tmpl->bind;
265    templ->usage = usage;
266 
267    vl_video_buffer_adjust_size(&templ->width0, &height, plane,
268                                tmpl->chroma_format, false);
269    templ->height0 = height;
270 }
271 
272 static void
vl_video_buffer_destroy(struct pipe_video_buffer * buffer)273 vl_video_buffer_destroy(struct pipe_video_buffer *buffer)
274 {
275    struct vl_video_buffer *buf = (struct vl_video_buffer *)buffer;
276    unsigned i;
277 
278    assert(buf);
279 
280    for (i = 0; i < VL_NUM_COMPONENTS; ++i) {
281       pipe_sampler_view_reference(&buf->sampler_view_planes[i], NULL);
282       pipe_sampler_view_reference(&buf->sampler_view_components[i], NULL);
283       pipe_resource_reference(&buf->resources[i], NULL);
284    }
285 
286    for (i = 0; i < VL_MAX_SURFACES; ++i)
287       pipe_surface_reference(&buf->surfaces[i], NULL);
288 
289    vl_video_buffer_set_associated_data(buffer, NULL, NULL, NULL);
290 
291    FREE(buffer);
292 }
293 
294 static struct pipe_sampler_view **
vl_video_buffer_sampler_view_planes(struct pipe_video_buffer * buffer)295 vl_video_buffer_sampler_view_planes(struct pipe_video_buffer *buffer)
296 {
297    struct vl_video_buffer *buf = (struct vl_video_buffer *)buffer;
298    struct pipe_sampler_view sv_templ;
299    struct pipe_context *pipe;
300    unsigned i;
301 
302    assert(buf);
303 
304    pipe = buf->base.context;
305 
306    for (i = 0; i < buf->num_planes; ++i ) {
307       if (!buf->sampler_view_planes[i]) {
308          memset(&sv_templ, 0, sizeof(sv_templ));
309          u_sampler_view_default_template(&sv_templ, buf->resources[i], buf->resources[i]->format);
310 
311          if (util_format_get_nr_components(buf->resources[i]->format) == 1)
312             sv_templ.swizzle_r = sv_templ.swizzle_g = sv_templ.swizzle_b = sv_templ.swizzle_a = PIPE_SWIZZLE_X;
313 
314          buf->sampler_view_planes[i] = pipe->create_sampler_view(pipe, buf->resources[i], &sv_templ);
315          if (!buf->sampler_view_planes[i])
316             goto error;
317       }
318    }
319 
320    return buf->sampler_view_planes;
321 
322 error:
323    for (i = 0; i < buf->num_planes; ++i )
324       pipe_sampler_view_reference(&buf->sampler_view_planes[i], NULL);
325 
326    return NULL;
327 }
328 
329 static struct pipe_sampler_view **
vl_video_buffer_sampler_view_components(struct pipe_video_buffer * buffer)330 vl_video_buffer_sampler_view_components(struct pipe_video_buffer *buffer)
331 {
332    struct vl_video_buffer *buf = (struct vl_video_buffer *)buffer;
333    struct pipe_sampler_view sv_templ;
334    struct pipe_context *pipe;
335    const enum pipe_format *sampler_format;
336    const unsigned *plane_order;
337    unsigned i, j, component;
338 
339    assert(buf);
340 
341    pipe = buf->base.context;
342 
343    sampler_format = vl_video_buffer_formats(pipe->screen, buf->base.buffer_format);
344    plane_order = vl_video_buffer_plane_order(buf->base.buffer_format);
345 
346    for (component = 0, i = 0; i < buf->num_planes; ++i ) {
347       struct pipe_resource *res = buf->resources[plane_order[i]];
348       const struct util_format_description *desc = util_format_description(res->format);
349       unsigned nr_components = util_format_get_nr_components(res->format);
350       if (desc->layout == UTIL_FORMAT_LAYOUT_SUBSAMPLED)
351          nr_components = 3;
352 
353       for (j = 0; j < nr_components && component < VL_NUM_COMPONENTS; ++j, ++component) {
354          if (buf->sampler_view_components[component])
355             continue;
356 
357          memset(&sv_templ, 0, sizeof(sv_templ));
358          u_sampler_view_default_template(&sv_templ, res, sampler_format[plane_order[i]]);
359          sv_templ.swizzle_r = sv_templ.swizzle_g = sv_templ.swizzle_b = PIPE_SWIZZLE_X + j;
360          sv_templ.swizzle_a = PIPE_SWIZZLE_1;
361          buf->sampler_view_components[component] = pipe->create_sampler_view(pipe, res, &sv_templ);
362          if (!buf->sampler_view_components[component])
363             goto error;
364       }
365    }
366    assert(component == VL_NUM_COMPONENTS);
367 
368    return buf->sampler_view_components;
369 
370 error:
371    for (i = 0; i < VL_NUM_COMPONENTS; ++i )
372       pipe_sampler_view_reference(&buf->sampler_view_components[i], NULL);
373 
374    return NULL;
375 }
376 
377 static struct pipe_surface **
vl_video_buffer_surfaces(struct pipe_video_buffer * buffer)378 vl_video_buffer_surfaces(struct pipe_video_buffer *buffer)
379 {
380    struct vl_video_buffer *buf = (struct vl_video_buffer *)buffer;
381    struct pipe_surface surf_templ;
382    struct pipe_context *pipe;
383    unsigned i, j, array_size, surf;
384 
385    assert(buf);
386 
387    pipe = buf->base.context;
388 
389    array_size = buffer->interlaced ? 2 : 1;
390    for (i = 0, surf = 0; i < VL_NUM_COMPONENTS; ++i) {
391       for (j = 0; j < array_size; ++j, ++surf) {
392          assert(surf < VL_MAX_SURFACES);
393 
394          if (!buf->resources[i]) {
395             pipe_surface_reference(&buf->surfaces[surf], NULL);
396             continue;
397          }
398 
399          if (!buf->surfaces[surf]) {
400             memset(&surf_templ, 0, sizeof(surf_templ));
401             surf_templ.format = vl_video_buffer_surface_format(buf->resources[i]->format);
402             surf_templ.u.tex.first_layer = surf_templ.u.tex.last_layer = j;
403             buf->surfaces[surf] = pipe->create_surface(pipe, buf->resources[i], &surf_templ);
404             if (!buf->surfaces[surf])
405                goto error;
406          }
407       }
408    }
409 
410    return buf->surfaces;
411 
412 error:
413    for (i = 0; i < VL_MAX_SURFACES; ++i )
414       pipe_surface_reference(&buf->surfaces[i], NULL);
415 
416    return NULL;
417 }
418 
419 struct pipe_video_buffer *
vl_video_buffer_create(struct pipe_context * pipe,const struct pipe_video_buffer * tmpl)420 vl_video_buffer_create(struct pipe_context *pipe,
421                        const struct pipe_video_buffer *tmpl)
422 {
423    const enum pipe_format *resource_formats;
424    struct pipe_video_buffer templat, *result;
425    bool pot_buffers;
426 
427    assert(pipe);
428    assert(tmpl->width > 0 && tmpl->height > 0);
429 
430    pot_buffers = !pipe->screen->get_video_param
431    (
432       pipe->screen,
433       PIPE_VIDEO_PROFILE_UNKNOWN,
434       PIPE_VIDEO_ENTRYPOINT_UNKNOWN,
435       PIPE_VIDEO_CAP_NPOT_TEXTURES
436    );
437 
438    resource_formats = vl_video_buffer_formats(pipe->screen, tmpl->buffer_format);
439    if (!resource_formats)
440       return NULL;
441 
442    templat = *tmpl;
443    templat.width = pot_buffers ? util_next_power_of_two(tmpl->width)
444                  : align(tmpl->width, VL_MACROBLOCK_WIDTH);
445    templat.height = pot_buffers ? util_next_power_of_two(tmpl->height)
446                   : align(tmpl->height, VL_MACROBLOCK_HEIGHT);
447 
448    if (tmpl->interlaced)
449       templat.height /= 2;
450 
451    result = vl_video_buffer_create_ex
452    (
453       pipe, &templat, resource_formats,
454       1, tmpl->interlaced ? 2 : 1, PIPE_USAGE_DEFAULT
455    );
456 
457 
458    if (result && tmpl->interlaced)
459       result->height *= 2;
460 
461    return result;
462 }
463 
464 struct pipe_video_buffer *
vl_video_buffer_create_ex(struct pipe_context * pipe,const struct pipe_video_buffer * tmpl,const enum pipe_format resource_formats[VL_NUM_COMPONENTS],unsigned depth,unsigned array_size,unsigned usage)465 vl_video_buffer_create_ex(struct pipe_context *pipe,
466                           const struct pipe_video_buffer *tmpl,
467                           const enum pipe_format resource_formats[VL_NUM_COMPONENTS],
468                           unsigned depth, unsigned array_size, unsigned usage)
469 {
470    struct pipe_resource res_tmpl;
471    struct pipe_resource *resources[VL_NUM_COMPONENTS];
472    unsigned i;
473 
474    assert(pipe);
475 
476    memset(resources, 0, sizeof resources);
477 
478    vl_video_buffer_template(&res_tmpl, tmpl, resource_formats[0], depth, array_size, usage, 0);
479    resources[0] = pipe->screen->resource_create(pipe->screen, &res_tmpl);
480    if (!resources[0])
481       goto error;
482 
483    if (resource_formats[1] == PIPE_FORMAT_NONE) {
484       assert(resource_formats[2] == PIPE_FORMAT_NONE);
485       return vl_video_buffer_create_ex2(pipe, tmpl, resources);
486    }
487 
488    vl_video_buffer_template(&res_tmpl, tmpl, resource_formats[1], depth, array_size, usage, 1);
489    resources[1] = pipe->screen->resource_create(pipe->screen, &res_tmpl);
490    if (!resources[1])
491       goto error;
492 
493    if (resource_formats[2] == PIPE_FORMAT_NONE)
494       return vl_video_buffer_create_ex2(pipe, tmpl, resources);
495 
496    vl_video_buffer_template(&res_tmpl, tmpl, resource_formats[2], depth, array_size, usage, 2);
497    resources[2] = pipe->screen->resource_create(pipe->screen, &res_tmpl);
498    if (!resources[2])
499       goto error;
500 
501    return vl_video_buffer_create_ex2(pipe, tmpl, resources);
502 
503 error:
504    for (i = 0; i < VL_NUM_COMPONENTS; ++i)
505       pipe_resource_reference(&resources[i], NULL);
506 
507    return NULL;
508 }
509 
510 struct pipe_video_buffer *
vl_video_buffer_create_ex2(struct pipe_context * pipe,const struct pipe_video_buffer * tmpl,struct pipe_resource * resources[VL_NUM_COMPONENTS])511 vl_video_buffer_create_ex2(struct pipe_context *pipe,
512                            const struct pipe_video_buffer *tmpl,
513                            struct pipe_resource *resources[VL_NUM_COMPONENTS])
514 {
515    struct vl_video_buffer *buffer;
516    unsigned i;
517 
518    buffer = CALLOC_STRUCT(vl_video_buffer);
519    if (!buffer)
520       return NULL;
521 
522    buffer->base = *tmpl;
523    buffer->base.context = pipe;
524    buffer->base.destroy = vl_video_buffer_destroy;
525    buffer->base.get_sampler_view_planes = vl_video_buffer_sampler_view_planes;
526    buffer->base.get_sampler_view_components = vl_video_buffer_sampler_view_components;
527    buffer->base.get_surfaces = vl_video_buffer_surfaces;
528    buffer->num_planes = 0;
529 
530    for (i = 0; i < VL_NUM_COMPONENTS; ++i) {
531       buffer->resources[i] = resources[i];
532       if (resources[i])
533          buffer->num_planes++;
534    }
535 
536    return &buffer->base;
537 }
538