style cleanup: follow style guide for formatting of if/for/while loops, and else...
[blender.git] / source / blender / nodes / texture / nodes / node_texture_image.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version. 
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2006 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Robin Allen
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/nodes/texture/nodes/node_texture_image.c
29  *  \ingroup texnodes
30  */
31
32
33 #include "node_texture_util.h"
34 #include "NOD_texture.h"
35
36 static bNodeSocketTemplate outputs[]= {
37         { SOCK_RGBA, 0, "Image"},
38         { -1, 0, "" }
39 };
40
41 static void colorfn(float *out, TexParams *p, bNode *node, bNodeStack **UNUSED(in), short UNUSED(thread))
42 {
43         float x = p->co[0];
44         float y = p->co[1];
45         Image *ima= (Image *)node->id;
46         ImageUser *iuser= (ImageUser *)node->storage;
47         
48         if ( ima ) {
49                 ImBuf *ibuf = BKE_image_get_ibuf(ima, iuser);
50                 if ( ibuf ) {
51                         float xsize, ysize;
52                         float xoff, yoff;
53                         int px, py;
54                         
55                         float *result;
56
57                         xsize = ibuf->x / 2;
58                         ysize = ibuf->y / 2;
59                         xoff = yoff = -1;
60                                         
61                         px = (int)( (x-xoff) * xsize );
62                         py = (int)( (y-yoff) * ysize );
63                 
64                         if ( (!xsize) || (!ysize) ) return;
65                         
66                         if ( !ibuf->rect_float ) {
67                                 BLI_lock_thread(LOCK_IMAGE);
68                                 if ( !ibuf->rect_float )
69                                         IMB_float_from_rect(ibuf);
70                                 BLI_unlock_thread(LOCK_IMAGE);
71                         }
72                         
73                         while ( px < 0 ) px += ibuf->x;
74                         while ( py < 0 ) py += ibuf->y;
75                         while ( px >= ibuf->x ) px -= ibuf->x;
76                         while ( py >= ibuf->y ) py -= ibuf->y;
77                         
78                         result = ibuf->rect_float + py*ibuf->x*4 + px*4;
79                         copy_v4_v4( out, result );
80                 }
81         }
82 }
83
84 static void exec(void *data, bNode *node, bNodeStack **in, bNodeStack **out)
85 {
86         tex_output(node, in, out[0], &colorfn, data);
87 }
88
89 static void init(bNodeTree *UNUSED(ntree), bNode* node, bNodeTemplate *UNUSED(ntemp))
90 {
91         ImageUser *iuser= MEM_callocN(sizeof(ImageUser), "node image user");
92         node->storage= iuser;
93         iuser->sfra= 1;
94         iuser->fie_ima= 2;
95         iuser->ok= 1;
96 }
97
98 void register_node_type_tex_image(bNodeTreeType *ttype)
99 {
100         static bNodeType ntype;
101         
102         node_type_base(ttype, &ntype, TEX_NODE_IMAGE, "Image", NODE_CLASS_INPUT, NODE_PREVIEW|NODE_OPTIONS);
103         node_type_socket_templates(&ntype, NULL, outputs);
104         node_type_size(&ntype, 120, 80, 300);
105         node_type_init(&ntype, init);
106         node_type_storage(&ntype, "ImageUser", node_free_standard_storage, node_copy_standard_storage);
107         node_type_exec(&ntype, exec);
108         
109         nodeRegisterType(ttype, &ntype);
110 }