diff options
-rw-r--r-- | source/blender/blenkernel/BKE_node.h | 1 | ||||
-rw-r--r-- | source/blender/blenkernel/intern/node.c | 1 | ||||
-rw-r--r-- | source/blender/nodes/CMP_node.h | 1 | ||||
-rw-r--r-- | source/blender/nodes/intern/CMP_nodes/CMP_normalize.c | 113 |
4 files changed, 116 insertions, 0 deletions
diff --git a/source/blender/blenkernel/BKE_node.h b/source/blender/blenkernel/BKE_node.h index 44266a1c854..5e5377c8969 100644 --- a/source/blender/blenkernel/BKE_node.h +++ b/source/blender/blenkernel/BKE_node.h @@ -298,6 +298,7 @@ void set_node_shader_lamp_loop(void (*lamp_loop_func)(struct ShadeInput *, str #define CMP_NODE_BRIGHTCONTRAST 249 #define CMP_NODE_GAMMA 250 #define CMP_NODE_INVERT 251 +#define CMP_NODE_NORMALIZE 252 /* channel toggles */ #define CMP_CHAN_RGB 1 diff --git a/source/blender/blenkernel/intern/node.c b/source/blender/blenkernel/intern/node.c index e7b7b36aaa4..5c4bb81ba22 100644 --- a/source/blender/blenkernel/intern/node.c +++ b/source/blender/blenkernel/intern/node.c @@ -2331,6 +2331,7 @@ static void registerCompositNodes(ListBase *ntypelist) nodeRegisterType(ntypelist, &cmp_node_normal); nodeRegisterType(ntypelist, &cmp_node_curve_vec); nodeRegisterType(ntypelist, &cmp_node_map_value); + nodeRegisterType(ntypeList, &cmp_node_normalize); nodeRegisterType(ntypelist, &cmp_node_filter); nodeRegisterType(ntypelist, &cmp_node_blur); diff --git a/source/blender/nodes/CMP_node.h b/source/blender/nodes/CMP_node.h index 3410238c0a2..6e800303887 100644 --- a/source/blender/nodes/CMP_node.h +++ b/source/blender/nodes/CMP_node.h @@ -62,6 +62,7 @@ extern bNodeType cmp_node_zcombine; extern bNodeType cmp_node_normal; extern bNodeType cmp_node_curve_vec; extern bNodeType cmp_node_map_value; +extern bNodeType cmp_node_normalize; extern bNodeType cmp_node_filter; extern bNodeType cmp_node_blur; diff --git a/source/blender/nodes/intern/CMP_nodes/CMP_normalize.c b/source/blender/nodes/intern/CMP_nodes/CMP_normalize.c new file mode 100644 index 00000000000..4b6b8958fea --- /dev/null +++ b/source/blender/nodes/intern/CMP_nodes/CMP_normalize.c @@ -0,0 +1,113 @@ +/** + * $Id: CMP_normalize.c,v 1.0 2007/03/24 06:57:29 scourage Exp $ + * + * ***** BEGIN GPL LICENSE BLOCK ***** + * + * 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, write to the Free Software Foundation, + * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + * + * The Original Code is Copyright (C) 2006 Blender Foundation. + * All rights reserved. + * + * The Original Code is: all of this file. + * + * Contributor(s): none yet. + * + * ***** END GPL LICENSE BLOCK ***** + */ + +#include "../CMP_util.h" + + +/* **************** NORMALIZE single channel, useful for Z buffer ******************** */ +static bNodeSocketType cmp_node_normalize_in[]= { + { SOCK_VALUE, 1, "Value", 1.0f, 0.8f, 0.8f, 1.0f, 0.0f, 1.0f}, + { -1, 0, "" } +}; +static bNodeSocketType cmp_node_normalize_out[]= { + { SOCK_VALUE, 0, "Value", 1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 1.0f}, + { -1, 0, "" } +}; + +static void do_normalize(bNode *node, float *out, float *src, float *min, float *mult) +{ + float res; + res = (src[0] - min[0]) * mult[0]; + if (res > 1.0f) { + out[0] = 1.0f; + } + else if (res < 0.0f) { + out[0] = 0.0f; + } + else { + out[0] = res; + } +} + +#define BLENDER_ZMAX 10000.0f + +static void node_composit_exec_normalize(void *data, bNode *node, bNodeStack **in, bNodeStack **out) +{ + /* stack order in: valbuf */ + /* stack order out: valbuf */ + if(out[0]->hasoutput==0) return; + + /* input no image? then only value operation */ + if(in[0]->data==NULL) { + QUATCOPY(out[0]->vec, in[0]->vec); + } + else { + float min = 1.0f+BLENDER_ZMAX; + float max = -1.0f-BLENDER_ZMAX; + float mult = 1.0f; + float *val; + /* make output size of input image */ + CompBuf *cbuf= in[0]->data; + int tot= cbuf->x*cbuf->y; + CompBuf *stackbuf= alloc_compbuf(cbuf->x, cbuf->y, CB_VAL, 1); /* allocs */ + + for (val = cbuf->rect; tot; tot--, val++) { + if ((*val > max) && (*val < BLENDER_ZMAX)) { + max = *val; + } + if (*val < min) { + min = *val; + } + } + mult = 1.0f/(max-min); + + printf("min %f max %f\n", min, max); + + composit3_pixel_processor(node, stackbuf, in[0]->data, in[0]->vec, NULL, &min, NULL, &mult, do_normalize, CB_VAL, CB_VAL, CB_VAL); + + out[0]->data= stackbuf; + } +} + +bNodeType cmp_node_normalize= { + /* *next, *prev*/ NULL, NULL, + /* type code */ CMP_NODE_NORMALIZE, + /* name */ "Normalize", + /* width+range */ 100, 60, 150, + /* class+opts */ NODE_CLASS_OP_VECTOR, NODE_OPTIONS, + /* input sock */ cmp_node_normalize_in, + /* output sock */ cmp_node_normalize_out, + /* storage */ "TexMapping", + /* execfunc */ node_composit_exec_normalize, + /* butfunc */ NULL, + /* initfunc */ NULL, + /* freestoragefunc */ NULL, + /* copystoragefunc */ NULL, + /* id */ NULL +}; |