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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2006 Blender Foundation. All rights reserved. */
/** \file
* \ingroup texnodes
*/
#include "NOD_texture.h"
#include "node_texture_util.h"
static bNodeSocketTemplate outputs[] = {
{SOCK_RGBA, N_("Image")},
{-1, ""},
};
static void colorfn(
float *out, TexParams *p, bNode *node, bNodeStack **UNUSED(in), short UNUSED(thread))
{
float x = p->co[0];
float y = p->co[1];
Image *ima = (Image *)node->id;
ImageUser *iuser = (ImageUser *)node->storage;
if (ima) {
ImBuf *ibuf = BKE_image_acquire_ibuf(ima, iuser, NULL);
if (ibuf) {
float xsize, ysize;
float xoff, yoff;
int px, py;
const float *result;
xsize = ibuf->x / 2;
ysize = ibuf->y / 2;
xoff = yoff = -1;
px = (int)((x - xoff) * xsize);
py = (int)((y - yoff) * ysize);
if ((!xsize) || (!ysize)) {
return;
}
if (!ibuf->rect_float) {
BLI_thread_lock(LOCK_IMAGE);
if (!ibuf->rect_float) {
IMB_float_from_rect(ibuf);
}
BLI_thread_unlock(LOCK_IMAGE);
}
while (px < 0) {
px += ibuf->x;
}
while (py < 0) {
py += ibuf->y;
}
while (px >= ibuf->x) {
px -= ibuf->x;
}
while (py >= ibuf->y) {
py -= ibuf->y;
}
result = ibuf->rect_float + py * ibuf->x * 4 + px * 4;
copy_v4_v4(out, result);
BKE_image_release_ibuf(ima, ibuf, NULL);
}
}
}
static void exec(void *data,
int UNUSED(thread),
bNode *node,
bNodeExecData *execdata,
bNodeStack **in,
bNodeStack **out)
{
tex_output(node, execdata, in, out[0], &colorfn, data);
}
static void init(bNodeTree *UNUSED(ntree), bNode *node)
{
ImageUser *iuser = MEM_callocN(sizeof(ImageUser), "node image user");
node->storage = iuser;
iuser->sfra = 1;
iuser->flag |= IMA_ANIM_ALWAYS;
}
void register_node_type_tex_image(void)
{
static bNodeType ntype;
tex_node_type_base(&ntype, TEX_NODE_IMAGE, "Image", NODE_CLASS_INPUT);
node_type_socket_templates(&ntype, NULL, outputs);
node_type_init(&ntype, init);
node_type_storage(&ntype, "ImageUser", node_free_standard_storage, node_copy_standard_storage);
node_type_exec(&ntype, NULL, NULL, exec);
ntype.labelfunc = node_image_label;
ntype.flag |= NODE_PREVIEW;
nodeRegisterType(&ntype);
}
|
