Amaranth Addon
[blender-addons-contrib.git] / amaranth / node_editor / templates / vectorblur.py
1 #  This program is free software; you can redistribute it and/or
2 #  modify it under the terms of the GNU General Public License
3 #  as published by the Free Software Foundation; either version 2
4 #  of the License, or (at your option) any later version.
5 #
6 #  This program is distributed in the hope that it will be useful,
7 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
8 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
9 #  GNU General Public License for more details.
10 #
11 #  You should have received a copy of the GNU General Public License
12 #  along with this program; if not, write to the Free Software Foundation,
13 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
14
15 import bpy
16 from mathutils import Vector
17
18
19 class AMTH_NODE_OT_AddTemplateVectorBlur(bpy.types.Operator):
20     bl_idname = "node.template_add_vectorblur"
21     bl_label = "Add Vector Blur"
22     bl_description = "Add a vector blur filter"
23     bl_options = {"REGISTER", "UNDO"}
24
25     @classmethod
26     def poll(cls, context):
27         space = context.space_data
28         tree = context.scene.node_tree
29         return space.type == "NODE_EDITOR" \
30             and space.node_tree is not None \
31             and space.tree_type == "CompositorNodeTree" \
32             and tree \
33             and tree.nodes.active \
34             and tree.nodes.active.type == "R_LAYERS"
35
36     def _setupNodes(self, context):
37         scene = context.scene
38         space = context.space_data
39         tree = scene.node_tree
40
41         bpy.ops.node.select_all(action="DESELECT")
42
43         act_node = tree.nodes.active
44         rlayer = act_node.scene.render.layers[act_node.layer]
45
46         if not rlayer.use_pass_vector:
47             rlayer.use_pass_vector = True
48
49         vblur = tree.nodes.new(type="CompositorNodeVecBlur")
50         vblur.use_curved = True
51         vblur.factor = 0.5
52
53         tree.links.new(act_node.outputs["Image"], vblur.inputs["Image"])
54         tree.links.new(act_node.outputs["Z"], vblur.inputs["Z"])
55         tree.links.new(act_node.outputs["Speed"], vblur.inputs["Speed"])
56
57         if tree.nodes.active:
58             vblur.location = tree.nodes.active.location
59             vblur.location += Vector((250.0, 0.0))
60         else:
61             vblur.location += Vector(
62                 (space.cursor_location[0], space.cursor_location[1]))
63
64         vblur.select = True
65
66     def execute(self, context):
67         self._setupNodes(context)
68
69         return {"FINISHED"}