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
105
106
107
|
# SPDX-License-Identifier: GPL-2.0-or-later
"""
Display Active Image Node on Image Editor
When selecting an Image node, it will show it on the Image editor (if
there is any available). If you don't like this behavior, you can
disable it from the Amaranth Toolset panel on the Scene properties.
Coded by the awesome Sergey Sharybin. This feature only works on Blender
2.68 and newer. Select an Image Node in the Compositor or Cycles nodes
editor, there must be at least one image editor available.
"""
import bpy
KEYMAPS = list()
image_nodes = ("CompositorNodeRLayers",
"CompositorNodeImage",
"CompositorNodeViewer",
"CompositorNodeComposite",
"ShaderNodeTexImage",
"ShaderNodeTexEnvironment",
"GeometryNodeImageTexture")
class AMTH_NODE_OT_show_active_node_image(bpy.types.Operator):
"""Show active image node image in the image editor"""
bl_idname = "node.show_active_node_image"
bl_label = "Preview Image from Node"
bl_options = {"UNDO"}
@classmethod
def poll(cls, context):
return context.space_data == 'NODE_EDITOR' and context.active_node is not None
def execute(self, context):
return {'FINISHED'}
def invoke(self, context, event):
mlocx = event.mouse_region_x
mlocy = event.mouse_region_y
select_node = bpy.ops.node.select(location=(mlocx, mlocy), extend=False)
if 'FINISHED' in select_node: # Only run if we're clicking on a node
get_addon = "amaranth" in context.preferences.addons.keys()
if not get_addon:
return {"CANCELLED"}
preferences = context.preferences.addons["amaranth"].preferences
if preferences.use_image_node_display:
if context.active_node:
active_node = context.active_node
if active_node.bl_idname in image_nodes:
# Use largest image editor
area = None
area_size = 0
for a in context.screen.areas:
if a.type == "IMAGE_EDITOR":
size = a.width * a.height
if size > area_size:
area_size = size
area = a
if area:
for space in area.spaces:
if space.type == "IMAGE_EDITOR":
if active_node.bl_idname == "CompositorNodeViewer":
space.image = bpy.data.images[
"Viewer Node"]
elif active_node.bl_idname in ["CompositorNodeComposite", "CompositorNodeRLayers"]:
space.image = bpy.data.images[
"Render Result"]
elif active_node.bl_idname == "GeometryNodeImageTexture":
if active_node.inputs['Image'].is_linked:
self.report({'INFO'}, "Previewing linked sockets is not supported yet")
break
if active_node.inputs['Image'].default_value:
space.image = active_node.inputs['Image'].default_value
elif active_node.image:
space.image = active_node.image
else:
self.report({'INFO'}, "No image detected")
break
break
else:
return {'CANCELLED'}
return {"FINISHED"}
else:
return {"PASS_THROUGH"}
def register():
bpy.utils.register_class(AMTH_NODE_OT_show_active_node_image)
kc = bpy.context.window_manager.keyconfigs.addon
km = kc.keymaps.new(name="Node Editor", space_type="NODE_EDITOR")
kmi = km.keymap_items.new("node.show_active_node_image",
"LEFTMOUSE", "DOUBLE_CLICK")
KEYMAPS.append((km, kmi))
def unregister():
bpy.utils.unregister_class(AMTH_NODE_OT_show_active_node_image)
for km, kmi in KEYMAPS:
km.keymap_items.remove(kmi)
KEYMAPS.clear()
|
