Home | History | Annotate | Download | only in postprocess
      1 /**************************************************************************
      2  *
      3  * Copyright 2011 Lauri Kasanen
      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 THE AUTHORS 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 "pipe/p_compiler.h"
     29 
     30 #include "postprocess/filters.h"
     31 
     32 #include "pipe/p_screen.h"
     33 #include "util/u_inlines.h"
     34 #include "util/u_blit.h"
     35 #include "util/u_math.h"
     36 #include "util/u_debug.h"
     37 #include "util/u_memory.h"
     38 #include "cso_cache/cso_context.h"
     39 
     40 /** Initialize the post-processing queue. */
     41 struct pp_queue_t *
     42 pp_init(struct pipe_screen *pscreen, const unsigned int *enabled)
     43 {
     44 
     45    unsigned int curpos = 0, i, tmp_req = 0;
     46    struct pp_queue_t *ppq;
     47    pp_func *tmp_q;
     48 
     49    pp_debug("Initializing the post-processing queue.\n");
     50 
     51    /* How many filters were requested? */
     52    for (i = 0; i < PP_FILTERS; i++) {
     53       if (enabled[i])
     54          curpos++;
     55    }
     56    if (!curpos)
     57       return NULL;
     58 
     59    ppq = CALLOC(1, sizeof(struct pp_queue_t));
     60    tmp_q = CALLOC(curpos, sizeof(pp_func));
     61    ppq->shaders = CALLOC(curpos, sizeof(void *));
     62    ppq->verts = CALLOC(curpos, sizeof(unsigned int));
     63 
     64    if (!tmp_q || !ppq || !ppq->shaders || !ppq->verts)
     65       goto error;
     66 
     67    ppq->p = pp_init_prog(ppq, pscreen);
     68    if (!ppq->p)
     69       goto error;
     70 
     71    /* Add the enabled filters to the queue, in order */
     72    curpos = 0;
     73    ppq->pp_queue = tmp_q;
     74    for (i = 0; i < PP_FILTERS; i++) {
     75       if (enabled[i]) {
     76          ppq->pp_queue[curpos] = pp_filters[i].main;
     77          tmp_req = MAX2(tmp_req, pp_filters[i].inner_tmps);
     78 
     79          if (pp_filters[i].shaders) {
     80             ppq->shaders[curpos] =
     81                CALLOC(pp_filters[i].shaders + 1, sizeof(void *));
     82             ppq->verts[curpos] = pp_filters[i].verts;
     83             if (!ppq->shaders[curpos])
     84                goto error;
     85          }
     86          pp_filters[i].init(ppq, curpos, enabled[i]);
     87 
     88          curpos++;
     89       }
     90    }
     91 
     92    ppq->p->blitctx = util_create_blit(ppq->p->pipe, ppq->p->cso);
     93    if (!ppq->p->blitctx)
     94       goto error;
     95 
     96    ppq->n_filters = curpos;
     97    ppq->n_tmp = (curpos > 2 ? 2 : 1);
     98    ppq->n_inner_tmp = tmp_req;
     99 
    100    ppq->fbos_init = false;
    101 
    102    for (i = 0; i < curpos; i++)
    103       ppq->shaders[i][0] = ppq->p->passvs;
    104 
    105    pp_debug("Queue successfully allocated. %u filter(s).\n", curpos);
    106 
    107    return ppq;
    108 
    109  error:
    110    pp_debug("Error setting up pp\n");
    111 
    112    if (ppq)
    113       FREE(ppq->p);
    114    FREE(ppq);
    115    FREE(tmp_q);
    116 
    117    return NULL;
    118 }
    119 
    120 /** Free any allocated FBOs (temp buffers). Called after resizing for example. */
    121 void
    122 pp_free_fbos(struct pp_queue_t *ppq)
    123 {
    124 
    125    unsigned int i;
    126 
    127    if (!ppq->fbos_init)
    128       return;
    129 
    130    for (i = 0; i < ppq->n_tmp; i++) {
    131       pipe_surface_reference(&ppq->tmps[i], NULL);
    132       pipe_resource_reference(&ppq->tmp[i], NULL);
    133    }
    134    for (i = 0; i < ppq->n_inner_tmp; i++) {
    135       pipe_surface_reference(&ppq->inner_tmps[i], NULL);
    136       pipe_resource_reference(&ppq->inner_tmp[i], NULL);
    137    }
    138    pipe_surface_reference(&ppq->stencils, NULL);
    139    pipe_resource_reference(&ppq->stencil, NULL);
    140 
    141    ppq->fbos_init = false;
    142 }
    143 
    144 /** Free the pp queue. Called on context termination. */
    145 void
    146 pp_free(struct pp_queue_t *ppq)
    147 {
    148 
    149    unsigned int i, j;
    150 
    151    pp_free_fbos(ppq);
    152 
    153    util_destroy_blit(ppq->p->blitctx);
    154 
    155    cso_set_sampler_views(ppq->p->cso, PIPE_SHADER_FRAGMENT, 0, NULL);
    156    cso_release_all(ppq->p->cso);
    157 
    158    for (i = 0; i < ppq->n_filters; i++) {
    159       for (j = 0; j < PP_MAX_PASSES && ppq->shaders[i][j]; j++) {
    160          if (j >= ppq->verts[i]) {
    161             ppq->p->pipe->delete_fs_state(ppq->p->pipe, ppq->shaders[i][j]);
    162             ppq->shaders[i][j] = NULL;
    163          }
    164          else if (ppq->shaders[i][j] != ppq->p->passvs) {
    165             ppq->p->pipe->delete_vs_state(ppq->p->pipe, ppq->shaders[i][j]);
    166             ppq->shaders[i][j] = NULL;
    167          }
    168       }
    169    }
    170 
    171    cso_destroy_context(ppq->p->cso);
    172    ppq->p->pipe->destroy(ppq->p->pipe);
    173 
    174    FREE(ppq->p);
    175    FREE(ppq->pp_queue);
    176    FREE(ppq);
    177 
    178    pp_debug("Queue taken down.\n");
    179 }
    180 
    181 /** Internal debug function. Should be available to final users. */
    182 void
    183 pp_debug(const char *fmt, ...)
    184 {
    185    va_list ap;
    186 
    187    if (!debug_get_bool_option("PP_DEBUG", FALSE))
    188       return;
    189 
    190    va_start(ap, fmt);
    191    _debug_vprintf(fmt, ap);
    192    va_end(ap);
    193 }
    194 
    195 /** Allocate the temp FBOs. Called on makecurrent and resize. */
    196 void
    197 pp_init_fbos(struct pp_queue_t *ppq, unsigned int w,
    198              unsigned int h)
    199 {
    200 
    201    struct program *p = ppq->p;  /* The lazy will inherit the earth */
    202 
    203    unsigned int i;
    204    struct pipe_resource tmp_res;
    205 
    206    if (ppq->fbos_init)
    207       return;
    208 
    209    pp_debug("Initializing FBOs, size %ux%u\n", w, h);
    210    pp_debug("Requesting %u temps and %u inner temps\n", ppq->n_tmp,
    211             ppq->n_inner_tmp);
    212 
    213    memset(&tmp_res, 0, sizeof(tmp_res));
    214    tmp_res.target = PIPE_TEXTURE_2D;
    215    tmp_res.format = p->surf.format = PIPE_FORMAT_B8G8R8A8_UNORM;
    216    tmp_res.width0 = w;
    217    tmp_res.height0 = h;
    218    tmp_res.depth0 = 1;
    219    tmp_res.array_size = 1;
    220    tmp_res.last_level = 0;
    221    tmp_res.bind = p->surf.usage = PIPE_BIND_RENDER_TARGET;
    222 
    223    if (!p->screen->is_format_supported(p->screen, tmp_res.format,
    224                                        tmp_res.target, 1, tmp_res.bind))
    225       pp_debug("Temp buffers' format fail\n");
    226 
    227    for (i = 0; i < ppq->n_tmp; i++) {
    228       ppq->tmp[i] = p->screen->resource_create(p->screen, &tmp_res);
    229       ppq->tmps[i] = p->pipe->create_surface(p->pipe, ppq->tmp[i], &p->surf);
    230 
    231       if (!ppq->tmp[i] || !ppq->tmps[i])
    232          goto error;
    233    }
    234 
    235    for (i = 0; i < ppq->n_inner_tmp; i++) {
    236       ppq->inner_tmp[i] = p->screen->resource_create(p->screen, &tmp_res);
    237       ppq->inner_tmps[i] = p->pipe->create_surface(p->pipe,
    238                                                    ppq->inner_tmp[i],
    239                                                    &p->surf);
    240 
    241       if (!ppq->inner_tmp[i] || !ppq->inner_tmps[i])
    242          goto error;
    243    }
    244 
    245    tmp_res.bind = p->surf.usage = PIPE_BIND_DEPTH_STENCIL;
    246 
    247    tmp_res.format = p->surf.format = PIPE_FORMAT_S8_UINT_Z24_UNORM;
    248 
    249    if (!p->screen->is_format_supported(p->screen, tmp_res.format,
    250                                        tmp_res.target, 1, tmp_res.bind)) {
    251 
    252       tmp_res.format = p->surf.format = PIPE_FORMAT_Z24_UNORM_S8_UINT;
    253 
    254       if (!p->screen->is_format_supported(p->screen, tmp_res.format,
    255                                           tmp_res.target, 1, tmp_res.bind))
    256          pp_debug("Temp Sbuffer format fail\n");
    257    }
    258 
    259    ppq->stencil = p->screen->resource_create(p->screen, &tmp_res);
    260    ppq->stencils = p->pipe->create_surface(p->pipe, ppq->stencil, &p->surf);
    261    if (!ppq->stencil || !ppq->stencils)
    262       goto error;
    263 
    264 
    265    p->framebuffer.width = w;
    266    p->framebuffer.height = h;
    267 
    268    p->viewport.scale[0] = p->viewport.translate[0] = (float) w / 2.0;
    269    p->viewport.scale[1] = p->viewport.translate[1] = (float) h / 2.0;
    270    p->viewport.scale[3] = 1.0f;
    271    p->viewport.translate[3] = 0.0f;
    272 
    273    ppq->fbos_init = true;
    274 
    275    return;
    276 
    277  error:
    278    pp_debug("Failed to allocate temp buffers!\n");
    279 }
    280