141 lines
3.9 KiB
Python
141 lines
3.9 KiB
Python
# ##### 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.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
import bpy
|
|
|
|
|
|
class NODE_HT_header(bpy.types.Header):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
snode = context.space_data
|
|
|
|
row = layout.row(align=True)
|
|
row.template_header()
|
|
|
|
if context.area.show_menus:
|
|
sub = row.row(align=True)
|
|
sub.itemM("NODE_MT_view")
|
|
sub.itemM("NODE_MT_select")
|
|
sub.itemM("NODE_MT_add")
|
|
sub.itemM("NODE_MT_node")
|
|
|
|
row = layout.row()
|
|
row.itemR(snode, "tree_type", text="", expand=True)
|
|
|
|
if snode.tree_type == 'MATERIAL':
|
|
ob = snode.id_from
|
|
snode_id = snode.id
|
|
if ob:
|
|
layout.template_ID(ob, "active_material", new="material.new")
|
|
if snode_id:
|
|
layout.itemR(snode_id, "use_nodes")
|
|
|
|
elif snode.tree_type == 'TEXTURE':
|
|
row.itemR(snode, "texture_type", text="", expand=True)
|
|
|
|
snode_id = snode.id
|
|
id_from = snode.id_from
|
|
if id_from:
|
|
layout.template_ID(id_from, "active_texture", new="texture.new")
|
|
if snode_id:
|
|
layout.itemR(snode_id, "use_nodes")
|
|
|
|
elif snode.tree_type == 'COMPOSITING':
|
|
snode_id = snode.id
|
|
|
|
layout.itemR(snode_id, "use_nodes")
|
|
layout.itemR(snode_id.render_data, "free_unused_nodes", text="Free Unused")
|
|
layout.itemR(snode, "backdrop")
|
|
|
|
|
|
class NODE_MT_view(bpy.types.Menu):
|
|
bl_label = "View"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
layout.itemO("node.properties", icon='ICON_MENU_PANEL')
|
|
layout.itemS()
|
|
|
|
# layout.itemO("grease_pencil..")
|
|
# layout.itemS()
|
|
|
|
layout.itemO("view2d.zoom_in")
|
|
layout.itemO("view2d.zoom_out")
|
|
|
|
layout.itemS()
|
|
|
|
layout.itemO("node.view_all")
|
|
layout.itemO("screen.screen_full_area")
|
|
|
|
|
|
class NODE_MT_select(bpy.types.Menu):
|
|
bl_label = "Select"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
layout.itemO("node.select_border")
|
|
|
|
layout.itemS()
|
|
layout.itemO("node.select_all")
|
|
layout.itemO("node.select_linked_from")
|
|
layout.itemO("node.select_linked_to")
|
|
|
|
|
|
class NODE_MT_node(bpy.types.Menu):
|
|
bl_label = "Node"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
layout.itemO("tfm.translate")
|
|
layout.itemO("tfm.resize")
|
|
layout.itemO("tfm.rotate")
|
|
|
|
layout.itemS()
|
|
|
|
layout.itemO("node.duplicate")
|
|
layout.itemO("node.delete")
|
|
|
|
# XXX
|
|
# layout.itemS()
|
|
# layout.itemO("node.make_link")
|
|
layout.itemS()
|
|
layout.itemO("node.group_edit")
|
|
layout.itemO("node.group_ungroup")
|
|
layout.itemO("node.group_make")
|
|
|
|
layout.itemS()
|
|
|
|
layout.itemO("node.visibility_toggle")
|
|
|
|
# XXX
|
|
# layout.itemO("node.rename")
|
|
# layout.itemS()
|
|
# layout.itemO("node.show_cyclic_dependencies")
|
|
|
|
bpy.types.register(NODE_HT_header)
|
|
bpy.types.register(NODE_MT_view)
|
|
bpy.types.register(NODE_MT_select)
|
|
bpy.types.register(NODE_MT_node)
|