342 lines
11 KiB
C
342 lines
11 KiB
C
/**
|
|
* Copyright (C) 2010 Jorge Jimenez (jorge@iryoku.com)
|
|
* Copyright (C) 2010 Belen Masia (bmasia@unizar.es)
|
|
* Copyright (C) 2010 Jose I. Echevarria (joseignacioechevarria@gmail.com)
|
|
* Copyright (C) 2010 Fernando Navarro (fernandn@microsoft.com)
|
|
* Copyright (C) 2010 Diego Gutierrez (diegog@unizar.es)
|
|
* Copyright (C) 2011 Lauri Kasanen (cand@gmx.com)
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the following statement:
|
|
*
|
|
* "Uses Jimenez's MLAA. Copyright (C) 2010 by Jorge Jimenez, Belen Masia,
|
|
* Jose I. Echevarria, Fernando Navarro and Diego Gutierrez."
|
|
*
|
|
* Only for use in the Mesa project, this point 2 is filled by naming the
|
|
* technique Jimenez's MLAA in the Mesa config options.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS
|
|
* IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
|
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL COPYRIGHT HOLDERS OR CONTRIBUTORS
|
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
* The views and conclusions contained in the software and documentation are
|
|
* those of the authors and should not be interpreted as representing official
|
|
* policies, either expressed or implied, of the copyright holders.
|
|
*/
|
|
|
|
#include "pipe/p_compiler.h"
|
|
|
|
#include "postprocess/postprocess.h"
|
|
#include "postprocess/pp_mlaa.h"
|
|
#include "postprocess/pp_filters.h"
|
|
#include "postprocess/pp_private.h"
|
|
|
|
#include "util/u_box.h"
|
|
#include "util/u_sampler.h"
|
|
#include "util/u_inlines.h"
|
|
#include "util/u_memory.h"
|
|
#include "util/u_string.h"
|
|
#include "pipe/p_screen.h"
|
|
|
|
#define IMM_SPACE 80
|
|
|
|
static float constants[] = { 1, 1, 0, 0 };
|
|
static unsigned int dimensions[2] = { 0, 0 };
|
|
|
|
/** Run function of the MLAA filter. */
|
|
static void
|
|
pp_jimenezmlaa_run(struct pp_queue_t *ppq, struct pipe_resource *in,
|
|
struct pipe_resource *out, unsigned int n, bool iscolor)
|
|
{
|
|
|
|
struct pp_program *p = ppq->p;
|
|
|
|
struct pipe_depth_stencil_alpha_state mstencil;
|
|
struct pipe_sampler_view v_tmp, *arr[3];
|
|
|
|
unsigned int w = 0;
|
|
unsigned int h = 0;
|
|
|
|
const struct pipe_stencil_ref ref = { {1} };
|
|
|
|
/* Insufficient initialization checks. */
|
|
assert(p);
|
|
assert(ppq);
|
|
assert(ppq->areamaptex);
|
|
assert(ppq->inner_tmp);
|
|
assert(ppq->shaders[n]);
|
|
|
|
w = p->framebuffer.width;
|
|
h = p->framebuffer.height;
|
|
|
|
memset(&mstencil, 0, sizeof(mstencil));
|
|
|
|
cso_set_stencil_ref(p->cso, ref);
|
|
|
|
/* Init the pixel size constant */
|
|
if (dimensions[0] != p->framebuffer.width ||
|
|
dimensions[1] != p->framebuffer.height) {
|
|
constants[0] = 1.0f / p->framebuffer.width;
|
|
constants[1] = 1.0f / p->framebuffer.height;
|
|
|
|
dimensions[0] = p->framebuffer.width;
|
|
dimensions[1] = p->framebuffer.height;
|
|
}
|
|
|
|
struct pipe_constant_buffer cb;
|
|
cb.buffer = NULL;
|
|
cb.buffer_offset = 0;
|
|
cb.buffer_size = sizeof(constants);
|
|
cb.user_buffer = constants;
|
|
|
|
struct pipe_context *pipe = ppq->p->pipe;
|
|
pipe->set_constant_buffer(pipe, PIPE_SHADER_VERTEX, 0, false, &cb);
|
|
pipe->set_constant_buffer(pipe, PIPE_SHADER_FRAGMENT, 0, false, &cb);
|
|
|
|
mstencil.stencil[0].enabled = 1;
|
|
mstencil.stencil[0].valuemask = mstencil.stencil[0].writemask = ~0;
|
|
mstencil.stencil[0].func = PIPE_FUNC_ALWAYS;
|
|
mstencil.stencil[0].fail_op = PIPE_STENCIL_OP_KEEP;
|
|
mstencil.stencil[0].zfail_op = PIPE_STENCIL_OP_KEEP;
|
|
mstencil.stencil[0].zpass_op = PIPE_STENCIL_OP_REPLACE;
|
|
|
|
p->framebuffer.zsbuf = ppq->stencils;
|
|
|
|
/* First pass: depth edge detection */
|
|
if (iscolor)
|
|
pp_filter_setup_in(p, in);
|
|
else
|
|
pp_filter_setup_in(p, ppq->depth);
|
|
|
|
pp_filter_setup_out(p, ppq->inner_tmp[0]);
|
|
|
|
pp_filter_set_fb(p);
|
|
pp_filter_misc_state(p);
|
|
cso_set_depth_stencil_alpha(p->cso, &mstencil);
|
|
p->pipe->clear(p->pipe, PIPE_CLEAR_STENCIL | PIPE_CLEAR_COLOR0, NULL,
|
|
&p->clear_color, 0, 0);
|
|
|
|
{
|
|
const struct pipe_sampler_state *samplers[] = {&p->sampler_point};
|
|
cso_set_samplers(p->cso, PIPE_SHADER_FRAGMENT, 1, samplers);
|
|
}
|
|
pipe->set_sampler_views(pipe, PIPE_SHADER_FRAGMENT, 0, 1, 0, false, &p->view);
|
|
|
|
cso_set_vertex_shader_handle(p->cso, ppq->shaders[n][1]); /* offsetvs */
|
|
cso_set_fragment_shader_handle(p->cso, ppq->shaders[n][2]);
|
|
|
|
pp_filter_draw(p);
|
|
pp_filter_end_pass(p);
|
|
|
|
|
|
/* Second pass: blend weights */
|
|
/* Sampler order: areamap, edgesmap, edgesmapL (reversed, thx compiler) */
|
|
mstencil.stencil[0].func = PIPE_FUNC_EQUAL;
|
|
mstencil.stencil[0].zpass_op = PIPE_STENCIL_OP_KEEP;
|
|
cso_set_depth_stencil_alpha(p->cso, &mstencil);
|
|
|
|
pp_filter_setup_in(p, ppq->areamaptex);
|
|
pp_filter_setup_out(p, ppq->inner_tmp[1]);
|
|
|
|
u_sampler_view_default_template(&v_tmp, ppq->inner_tmp[0],
|
|
ppq->inner_tmp[0]->format);
|
|
arr[1] = arr[2] = p->pipe->create_sampler_view(p->pipe,
|
|
ppq->inner_tmp[0], &v_tmp);
|
|
|
|
pp_filter_set_clear_fb(p);
|
|
|
|
{
|
|
const struct pipe_sampler_state *samplers[] =
|
|
{&p->sampler_point, &p->sampler_point, &p->sampler};
|
|
cso_set_samplers(p->cso, PIPE_SHADER_FRAGMENT, 3, samplers);
|
|
}
|
|
|
|
arr[0] = p->view;
|
|
pipe->set_sampler_views(pipe, PIPE_SHADER_FRAGMENT, 0, 3, 0, false, arr);
|
|
|
|
cso_set_vertex_shader_handle(p->cso, ppq->shaders[n][0]); /* passvs */
|
|
cso_set_fragment_shader_handle(p->cso, ppq->shaders[n][3]);
|
|
|
|
pp_filter_draw(p);
|
|
pp_filter_end_pass(p);
|
|
pipe_sampler_view_reference(&arr[1], NULL);
|
|
|
|
|
|
/* Third pass: smoothed edges */
|
|
/* Sampler order: colormap, blendmap (wtf compiler) */
|
|
pp_filter_setup_in(p, ppq->inner_tmp[1]);
|
|
pp_filter_setup_out(p, out);
|
|
|
|
pp_filter_set_fb(p);
|
|
|
|
/* Blit the input to the output */
|
|
pp_blit(p->pipe, in, 0, 0,
|
|
w, h, 0, p->framebuffer.cbufs[0],
|
|
0, 0, w, h);
|
|
|
|
u_sampler_view_default_template(&v_tmp, in, in->format);
|
|
arr[0] = p->pipe->create_sampler_view(p->pipe, in, &v_tmp);
|
|
|
|
{
|
|
const struct pipe_sampler_state *samplers[] =
|
|
{&p->sampler_point, &p->sampler_point};
|
|
cso_set_samplers(p->cso, PIPE_SHADER_FRAGMENT, 2, samplers);
|
|
}
|
|
|
|
arr[1] = p->view;
|
|
pipe->set_sampler_views(pipe, PIPE_SHADER_FRAGMENT, 0, 2, 0, false, arr);
|
|
|
|
cso_set_vertex_shader_handle(p->cso, ppq->shaders[n][1]); /* offsetvs */
|
|
cso_set_fragment_shader_handle(p->cso, ppq->shaders[n][4]);
|
|
|
|
p->blend.rt[0].blend_enable = 1;
|
|
cso_set_blend(p->cso, &p->blend);
|
|
|
|
pp_filter_draw(p);
|
|
pp_filter_end_pass(p);
|
|
pipe_sampler_view_reference(&arr[0], NULL);
|
|
|
|
p->blend.rt[0].blend_enable = 0;
|
|
p->framebuffer.zsbuf = NULL;
|
|
}
|
|
|
|
/** The init function of the MLAA filter. */
|
|
static bool
|
|
pp_jimenezmlaa_init_run(struct pp_queue_t *ppq, unsigned int n,
|
|
unsigned int val, bool iscolor)
|
|
{
|
|
|
|
struct pipe_box box;
|
|
struct pipe_resource res;
|
|
char *tmp_text = NULL;
|
|
|
|
tmp_text = CALLOC(sizeof(blend2fs_1) + sizeof(blend2fs_2) +
|
|
IMM_SPACE, sizeof(char));
|
|
|
|
if (!tmp_text) {
|
|
pp_debug("Failed to allocate shader space\n");
|
|
return FALSE;
|
|
}
|
|
|
|
pp_debug("mlaa: using %u max search steps\n", val);
|
|
|
|
sprintf(tmp_text, "%s"
|
|
"IMM FLT32 { %.8f, 0.0000, 0.0000, 0.0000}\n"
|
|
"%s\n", blend2fs_1, (float) val, blend2fs_2);
|
|
|
|
memset(&res, 0, sizeof(res));
|
|
|
|
res.target = PIPE_TEXTURE_2D;
|
|
res.format = PIPE_FORMAT_R8G8_UNORM;
|
|
res.width0 = res.height0 = 165;
|
|
res.bind = PIPE_BIND_SAMPLER_VIEW;
|
|
res.usage = PIPE_USAGE_DEFAULT;
|
|
res.depth0 = res.array_size = res.nr_samples = res.nr_storage_samples = 1;
|
|
|
|
if (!ppq->p->screen->is_format_supported(ppq->p->screen, res.format,
|
|
res.target, 1, 1, res.bind))
|
|
pp_debug("Areamap format not supported\n");
|
|
|
|
ppq->areamaptex = ppq->p->screen->resource_create(ppq->p->screen, &res);
|
|
|
|
if (ppq->areamaptex == NULL) {
|
|
pp_debug("Failed to allocate area map texture\n");
|
|
goto fail;
|
|
}
|
|
|
|
u_box_2d(0, 0, 165, 165, &box);
|
|
|
|
ppq->p->pipe->texture_subdata(ppq->p->pipe, ppq->areamaptex, 0,
|
|
PIPE_MAP_WRITE, &box,
|
|
areamap, 165 * 2, sizeof(areamap));
|
|
|
|
ppq->shaders[n][1] = pp_tgsi_to_state(ppq->p->pipe, offsetvs, true,
|
|
"offsetvs");
|
|
if (iscolor)
|
|
ppq->shaders[n][2] = pp_tgsi_to_state(ppq->p->pipe, color1fs,
|
|
false, "color1fs");
|
|
else
|
|
ppq->shaders[n][2] = pp_tgsi_to_state(ppq->p->pipe, depth1fs,
|
|
false, "depth1fs");
|
|
ppq->shaders[n][3] = pp_tgsi_to_state(ppq->p->pipe, tmp_text, false,
|
|
"blend2fs");
|
|
ppq->shaders[n][4] = pp_tgsi_to_state(ppq->p->pipe, neigh3fs, false,
|
|
"neigh3fs");
|
|
|
|
FREE(tmp_text);
|
|
|
|
return TRUE;
|
|
|
|
fail:
|
|
|
|
FREE(tmp_text);
|
|
|
|
/*
|
|
* Call the common free function for destruction of partially initialized
|
|
* resources.
|
|
*/
|
|
pp_jimenezmlaa_free(ppq, n);
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/** Short wrapper to init the depth version. */
|
|
bool
|
|
pp_jimenezmlaa_init(struct pp_queue_t *ppq, unsigned int n, unsigned int val)
|
|
{
|
|
return pp_jimenezmlaa_init_run(ppq, n, val, false);
|
|
}
|
|
|
|
/** Short wrapper to init the color version. */
|
|
bool
|
|
pp_jimenezmlaa_init_color(struct pp_queue_t *ppq, unsigned int n,
|
|
unsigned int val)
|
|
{
|
|
return pp_jimenezmlaa_init_run(ppq, n, val, true);
|
|
}
|
|
|
|
/** Short wrapper to run the depth version. */
|
|
void
|
|
pp_jimenezmlaa(struct pp_queue_t *ppq, struct pipe_resource *in,
|
|
struct pipe_resource *out, unsigned int n)
|
|
{
|
|
if (!ppq->depth) {
|
|
return;
|
|
}
|
|
pp_jimenezmlaa_run(ppq, in, out, n, false);
|
|
}
|
|
|
|
/** Short wrapper to run the color version. */
|
|
void
|
|
pp_jimenezmlaa_color(struct pp_queue_t *ppq, struct pipe_resource *in,
|
|
struct pipe_resource *out, unsigned int n)
|
|
{
|
|
pp_jimenezmlaa_run(ppq, in, out, n, true);
|
|
}
|
|
|
|
|
|
/**
|
|
* Short wrapper to free the mlaa filter resources. Shaders are freed in
|
|
* the common code in pp_free.
|
|
*/
|
|
void
|
|
pp_jimenezmlaa_free(struct pp_queue_t *ppq, unsigned int n)
|
|
{
|
|
pipe_resource_reference(&ppq->areamaptex, NULL);
|
|
}
|
|
|