Color management refactoiring and some extra options
[blender.git] / release / scripts / startup / bl_ui / space_node.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20 import bpy
21 from bpy.types import Header, Menu, Panel
22
23
24 class NODE_HT_header(Header):
25     bl_space_type = 'NODE_EDITOR'
26
27     def draw(self, context):
28         layout = self.layout
29
30         scene = context.scene
31         ob = context.object
32         snode = context.space_data
33         snode_id = snode.id
34         id_from = snode.id_from
35         toolsettings = context.tool_settings
36
37         row = layout.row(align=True)
38         row.template_header()
39
40         if context.area.show_menus:
41             row.menu("NODE_MT_view")
42             row.menu("NODE_MT_select")
43             row.menu("NODE_MT_add")
44             row.menu("NODE_MT_node")
45
46         layout.prop(snode, "tree_type", text="", expand=True)
47
48         if snode.tree_type == 'SHADER':
49             if scene.render.use_shading_nodes:
50                 layout.prop(snode, "shader_type", text="", expand=True)
51
52             if (not scene.render.use_shading_nodes or snode.shader_type == 'OBJECT') and ob:
53                 # Show material.new when no active ID/slot exists
54                 if not id_from and ob.type in {'MESH', 'CURVE', 'SURFACE', 'FONT', 'METABALL'}:
55                     layout.template_ID(ob, "active_material", new="material.new")
56                 # Material ID, but not for Lamps
57                 if id_from and ob.type != 'LAMP':
58                     layout.template_ID(id_from, "active_material", new="material.new")
59                 # Don't show "Use Nodes" Button when Engine is BI for Lamps
60                 if snode_id and not (scene.render.use_shading_nodes == 0 and ob.type == 'LAMP'):
61                     layout.prop(snode_id, "use_nodes")
62
63             if snode.shader_type == 'WORLD':
64                 layout.template_ID(scene, "world", new="world.new")
65                 if snode_id:
66                     layout.prop(snode_id, "use_nodes")
67
68         elif snode.tree_type == 'TEXTURE':
69             layout.prop(snode, "texture_type", text="", expand=True)
70
71             if id_from:
72                 if snode.texture_type == 'BRUSH':
73                     layout.template_ID(id_from, "texture", new="texture.new")
74                 else:
75                     layout.template_ID(id_from, "active_texture", new="texture.new")
76             if snode_id:
77                 layout.prop(snode_id, "use_nodes")
78
79         elif snode.tree_type == 'COMPOSITING':
80             layout.prop(snode_id, "use_nodes")
81             layout.prop(snode_id.render, "use_free_unused_nodes", text="Free Unused")
82             layout.prop(snode, "show_backdrop")
83             if snode.show_backdrop:
84                 row = layout.row(align=True)
85                 row.prop(snode, "backdrop_channels", text="", expand=True)
86             layout.prop(snode, "use_auto_render")
87
88         layout.separator()
89
90         # Snap
91         row = layout.row(align=True)
92         row.prop(toolsettings, "use_snap", text="")
93         row.prop(toolsettings, "snap_node_element", text="", icon_only=True)
94         if toolsettings.snap_node_element != 'INCREMENT':
95             row.prop(toolsettings, "snap_target", text="")
96
97         layout.template_running_jobs()
98
99
100 class NODE_MT_view(Menu):
101     bl_label = "View"
102
103     def draw(self, context):
104         layout = self.layout
105
106         layout.operator("node.properties", icon='MENU_PANEL')
107         layout.separator()
108
109         layout.operator("view2d.zoom_in")
110         layout.operator("view2d.zoom_out")
111
112         layout.separator()
113
114         layout.operator("node.view_all")
115
116         if context.space_data.show_backdrop:
117             layout.separator()
118
119             layout.operator("node.backimage_move", text="Backdrop move")
120             layout.operator("node.backimage_zoom", text="Backdrop zoom in").factor = 1.2
121             layout.operator("node.backimage_zoom", text="Backdrop zoom out").factor = 0.833
122
123         layout.separator()
124
125         layout.operator("screen.area_dupli")
126         layout.operator("screen.screen_full_area")
127
128
129 class NODE_MT_select(Menu):
130     bl_label = "Select"
131
132     def draw(self, context):
133         layout = self.layout
134
135         layout.operator("node.select_border")
136
137         layout.separator()
138         layout.operator("node.select_all")
139         layout.operator("node.select_linked_from")
140         layout.operator("node.select_linked_to")
141         layout.operator("node.select_same_type")
142         layout.operator("node.select_same_type_next")
143         layout.operator("node.select_same_type_prev")
144
145
146 class NODE_MT_node(Menu):
147     bl_label = "Node"
148
149     def draw(self, context):
150         layout = self.layout
151
152         layout.operator("transform.translate")
153         layout.operator("transform.rotate")
154         layout.operator("transform.resize")
155
156         layout.separator()
157
158         layout.operator("node.duplicate_move")
159         layout.operator("node.delete")
160         layout.operator("node.delete_reconnect")
161
162         layout.separator()
163         layout.operator("node.link_make")
164         layout.operator("node.link_make", text="Make and Replace Links").replace = True
165         layout.operator("node.links_cut")
166
167         layout.separator()
168         layout.operator("node.group_edit")
169         layout.operator("node.group_ungroup")
170         layout.operator("node.group_make")
171
172         layout.separator()
173
174         layout.operator("node.hide_toggle")
175         layout.operator("node.mute_toggle")
176         layout.operator("node.preview_toggle")
177         layout.operator("node.hide_socket_toggle")
178         layout.operator("node.options_toggle")
179
180         layout.separator()
181
182         layout.operator("node.show_cyclic_dependencies")
183         layout.operator("node.read_renderlayers")
184         layout.operator("node.read_fullsamplelayers")
185
186
187 class NODE_PT_display_properties(Panel):
188     bl_space_type = 'NODE_EDITOR'
189     bl_region_type = 'UI'
190     bl_label = "Display Properties"
191
192     def draw(self, context):
193         layout = self.layout
194
195         snode = context.space_data
196         window = context.window
197         view_settings = snode.view_settings
198
199         # OCIO_TODO: de-duplicate this between different spaces
200         col = layout.column()
201         col.prop(window, "display_device", text="Display")
202         col.prop(view_settings, "view_transform", text="View")
203
204         col = layout.column()
205         col.active = view_settings.view_transform not in {'ACES ODT Tonecurve', 'NONE'}
206         col.prop(view_settings, "exposure")
207         col.prop(view_settings, "gamma")
208
209
210 # Node Backdrop options
211 class NODE_PT_properties(Panel):
212     bl_space_type = 'NODE_EDITOR'
213     bl_region_type = 'UI'
214     bl_label = "Backdrop"
215
216     @classmethod
217     def poll(cls, context):
218         snode = context.space_data
219         return snode.tree_type == 'COMPOSITING'
220
221     def draw_header(self, context):
222         snode = context.space_data
223         self.layout.prop(snode, "show_backdrop", text="")
224
225     def draw(self, context):
226         layout = self.layout
227
228         snode = context.space_data
229         layout.active = snode.show_backdrop
230         layout.prop(snode, "backdrop_channels", text="")
231         layout.prop(snode, "backdrop_zoom", text="Zoom")
232
233         col = layout.column(align=True)
234         col.label(text="Offset:")
235         col.prop(snode, "backdrop_x", text="X")
236         col.prop(snode, "backdrop_y", text="Y")
237         col.operator("node.backimage_move", text="Move")
238
239
240 class NODE_PT_quality(bpy.types.Panel):
241     bl_space_type = 'NODE_EDITOR'
242     bl_region_type = 'UI'
243     bl_label = "Quality"
244
245     @classmethod
246     def poll(cls, context):
247         snode = context.space_data
248         return snode.tree_type == 'COMPOSITING' and snode.node_tree is not None
249
250     def draw(self, context):
251         layout = self.layout
252         snode = context.space_data
253         tree = snode.node_tree
254
255         layout.prop(tree, "render_quality", text="Render")
256         layout.prop(tree, "edit_quality", text="Edit")
257         layout.prop(tree, "chunk_size")
258         layout.prop(tree, "use_opencl")
259
260
261 class NODE_MT_node_color_presets(Menu):
262     """Predefined node color"""
263     bl_label = "Color Presets"
264     preset_subdir = "node_color"
265     preset_operator = "script.execute_preset"
266     draw = Menu.draw_preset
267
268
269 class NODE_MT_node_color_specials(Menu):
270     bl_label = "Node Color Specials"
271
272     def draw(self, context):
273         layout = self.layout
274
275         layout.operator("node.node_copy_color", icon='COPY_ID')
276
277
278 if __name__ == "__main__":  # only for live edit.
279     bpy.utils.register_module(__name__)