1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
/*
* This file is part of libplacebo.
*
* libplacebo is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* libplacebo is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with libplacebo. If not, see <http://www.gnu.org/licenses/>.
*/
#include "shaders.h"
#include <libplacebo/shaders/custom.h>
bool pl_shader_custom(pl_shader sh, const struct pl_custom_shader *params)
{
if (params->compute) {
int bw = PL_DEF(params->compute_group_size[0], 16);
int bh = PL_DEF(params->compute_group_size[1], 16);
bool flex = !params->compute_group_size[0] ||
!params->compute_group_size[1];
if (!sh_try_compute(sh, bw, bh, flex, params->compute_shmem))
return false;
}
if (!sh_require(sh, params->input, params->output_w, params->output_h))
return false;
sh->output = params->output;
for (int i = 0; i < params->num_variables; i++) {
struct pl_shader_var sv = params->variables[i];
GLSLP("#define %s "$"\n", sv.var.name, sh_var(sh, sv));
}
for (int i = 0; i < params->num_descriptors; i++) {
struct pl_shader_desc sd = params->descriptors[i];
GLSLP("#define %s "$"\n", sd.desc.name, sh_desc(sh, sd));
}
for (int i = 0; i < params->num_vertex_attribs; i++) {
struct pl_shader_va sva = params->vertex_attribs[i];
GLSLP("#define %s "$"\n", sva.attr.name, sh_attr(sh, sva));
}
for (int i = 0; i < params->num_constants; i++) {
struct pl_shader_const sc = params->constants[i];
GLSLP("#define %s "$"\n", sc.name, sh_const(sh, sc));
}
if (params->prelude)
GLSLP("// pl_shader_custom prelude: \n%s\n", params->prelude);
if (params->header)
GLSLH("// pl_shader_custom header: \n%s\n", params->header);
if (params->description)
sh_describef(sh, "%s", params->description);
if (params->body) {
const char *output_decl = "";
if (params->output != params->input) {
switch (params->output) {
case PL_SHADER_SIG_NONE: break;
case PL_SHADER_SIG_COLOR:
output_decl = "vec4 color = vec4(0.0);";
break;
case PL_SHADER_SIG_SAMPLER:
pl_unreachable();
}
}
GLSL("// pl_shader_custom \n"
"%s \n"
"{ \n"
"%s \n"
"} \n",
output_decl, params->body);
}
return true;
}
|