obs-studio/libobs/graphics/effect.c

325 lines
7.6 KiB
C

/******************************************************************************
Copyright (C) 2013 by Hugh Bailey <obs.jim@gmail.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/
#include "effect.h"
#include "graphics-internal.h"
#include "vec2.h"
#include "vec3.h"
#include "vec4.h"
void effect_destroy(effect_t effect)
{
if (effect) {
effect_free(effect);
bfree(effect);
}
}
technique_t effect_gettechnique(effect_t effect, const char *name)
{
size_t i;
struct effect_technique *array = effect->techniques.array;
for (i = 0; i < effect->techniques.num; i++) {
struct effect_technique *tech = array+i;
if (strcmp(tech->name, name) == 0)
return tech;
}
return NULL;
}
size_t technique_begin(technique_t tech)
{
tech->effect->cur_technique = tech;
tech->effect->graphics->cur_effect = tech->effect;
return tech->passes.num;
}
void technique_end(technique_t tech)
{
struct gs_effect *effect = tech->effect;
struct effect_param *params = effect->params.array;
size_t i;
gs_load_vertexshader(NULL);
gs_load_pixelshader(NULL);
tech->effect->cur_technique = NULL;
tech->effect->graphics->cur_effect = NULL;
for (i = 0; i < effect->params.num; i++) {
struct effect_param *param = params+i;
da_free(param->cur_val);
param->changed = false;
}
}
static inline void reset_params(struct darray *shaderparams)
{
struct pass_shaderparam *params = shaderparams->array;
size_t i;
for (i = 0; i < shaderparams->num; i++)
params[i].eparam->changed = false;
}
static void upload_shader_params(shader_t shader, struct darray *pass_params,
bool changed_only)
{
struct pass_shaderparam *params = pass_params->array;
size_t i;
for (i = 0; i < pass_params->num; i++) {
struct pass_shaderparam *param = params+i;
struct effect_param *eparam = param->eparam;
sparam_t sparam = param->sparam;
if (changed_only && !eparam->changed)
continue;
if (!eparam->cur_val.num) {
if (eparam->default_val.num)
da_copy(eparam->cur_val, eparam->default_val);
else
continue;
}
shader_setval(shader, sparam, eparam->cur_val.array,
eparam->cur_val.num);
}
}
static inline void upload_parameters(struct gs_effect *effect,
bool changed_only)
{
struct darray *vshader_params, *pshader_params;
if (!effect->cur_pass)
return;
vshader_params = &effect->cur_pass->vertshader_params.da;
pshader_params = &effect->cur_pass->pixelshader_params.da;
upload_shader_params(effect->cur_pass->vertshader, vshader_params,
changed_only);
upload_shader_params(effect->cur_pass->pixelshader, pshader_params,
changed_only);
reset_params(vshader_params);
reset_params(pshader_params);
}
void effect_updateparams(effect_t effect)
{
upload_parameters(effect, true);
}
bool technique_beginpass(technique_t tech, size_t idx)
{
struct effect_pass *passes;
struct effect_pass *cur_pass;
if (idx >= tech->passes.num)
return false;
passes = tech->passes.array;
cur_pass = passes+idx;
tech->effect->cur_pass = cur_pass;
gs_load_vertexshader(cur_pass->vertshader);
gs_load_pixelshader(cur_pass->pixelshader);
upload_parameters(tech->effect, false);
return true;
}
bool technique_beginpassbyname(technique_t tech,
const char *name)
{
size_t i;
for (i = 0; i < tech->passes.num; i++) {
struct effect_pass *pass = tech->passes.array+i;
if (strcmp(pass->name, name) == 0) {
technique_beginpass(tech, i);
return true;
}
}
return false;
}
static inline void clear_tex_params(shader_t shader, struct darray *in_params)
{
struct pass_shaderparam *params = in_params->array;
size_t i;
for (i = 0; i < in_params->num; i++) {
struct pass_shaderparam *param = params+i;
struct shader_param_info info;
shader_getparaminfo(shader, param->sparam, &info);
if (info.type == SHADER_PARAM_TEXTURE)
shader_settexture(shader, param->sparam, NULL);
}
}
void technique_endpass(technique_t tech)
{
struct effect_pass *pass = tech->effect->cur_pass;
if (!pass)
return;
clear_tex_params(pass->vertshader, &pass->vertshader_params.da);
clear_tex_params(pass->pixelshader, &pass->pixelshader_params.da);
tech->effect->cur_pass = NULL;
}
size_t effect_numparams(effect_t effect)
{
return effect->params.num;
}
eparam_t effect_getparambyidx(effect_t effect, size_t param)
{
struct effect_param *params = effect->params.array;
if (param >= effect->params.num)
return NULL;
return params+param;
}
eparam_t effect_getparambyname(effect_t effect, const char *name)
{
struct effect_param *params = effect->params.array;
size_t i;
for (i = 0; i < effect->params.num; i++) {
struct effect_param *param = params+i;
if (strcmp(param->name, name) == 0)
return param;
}
return NULL;
}
static inline bool matching_effect(effect_t effect, eparam_t param)
{
if (effect != param->effect) {
blog(LOG_ERROR, "Effect and effect parameter do not match");
return false;
}
return true;
}
void effect_getparaminfo(effect_t effect, eparam_t param,
struct effect_param_info *info)
{
if (!matching_effect(effect, param))
return;
info->name = param->name;
info->type = param->type;
}
eparam_t effect_getviewprojmatrix(effect_t effect)
{
return effect->view_proj;
}
eparam_t effect_getworldmatrix(effect_t effect)
{
return effect->world;
}
static inline void effect_setval_inline(effect_t effect, eparam_t param,
const void *data, size_t size)
{
bool size_changed = param->cur_val.num != size;
if (!matching_effect(effect, param))
return;
if (size_changed)
da_resize(param->cur_val, size);
if (size_changed || memcmp(param->cur_val.array, data, size) != 0) {
memcpy(param->cur_val.array, data, size);
param->changed = true;
}
}
void effect_setbool(effect_t effect, eparam_t param, bool val)
{
effect_setval_inline(effect, param, &val, sizeof(bool));
}
void effect_setfloat(effect_t effect, eparam_t param, float val)
{
effect_setval_inline(effect, param, &val, sizeof(float));
}
void effect_setint(effect_t effect, eparam_t param, int val)
{
effect_setval_inline(effect, param, &val, sizeof(int));
}
void effect_setmatrix4(effect_t effect, eparam_t param,
const struct matrix4 *val)
{
effect_setval_inline(effect, param, val, sizeof(struct matrix4));
}
void effect_setvec2(effect_t effect, eparam_t param,
const struct vec2 *val)
{
effect_setval_inline(effect, param, val, sizeof(struct vec2));
}
void effect_setvec3(effect_t effect, eparam_t param,
const struct vec3 *val)
{
effect_setval_inline(effect, param, val, sizeof(float) * 3);
}
void effect_setvec4(effect_t effect, eparam_t param,
const struct vec4 *val)
{
effect_setval_inline(effect, param, val, sizeof(struct vec4));
}
void effect_settexture(effect_t effect, eparam_t param, texture_t val)
{
effect_setval_inline(effect, param, &val, sizeof(texture_t));
}
void effect_setval(effect_t effect, eparam_t param, const void *val,
size_t size)
{
effect_setval_inline(effect, param, val, size);
}
void effect_setdefault(effect_t effect, eparam_t param)
{
effect_setval_inline(effect, param, param->default_val.array,
param->default_val.num);
}