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 *
pp_init(struct pipe_screen * pscreen,const unsigned int * enabled)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
pp_free_fbos(struct pp_queue_t * ppq)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
pp_free(struct pp_queue_t * ppq)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
pp_debug(const char * fmt,...)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
pp_init_fbos(struct pp_queue_t * ppq,unsigned int w,unsigned int h)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