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. */
41struct pp_queue_t *
42pp_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. */
121void
122pp_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. */
145void
146pp_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. */
182void
183pp_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. */
196void
197pp_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