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
|
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
import bpy
from mathutils import Vector
class AMTH_NODE_OT_AddTemplateVectorBlur(bpy.types.Operator):
bl_idname = "node.template_add_vectorblur"
bl_label = "Add Vector Blur"
bl_description = "Add a vector blur filter"
bl_options = {"REGISTER", "UNDO"}
@classmethod
def poll(cls, context):
space = context.space_data
tree = context.scene.node_tree
return space.type == "NODE_EDITOR" \
and space.node_tree is not None \
and space.tree_type == "CompositorNodeTree" \
and tree \
and tree.nodes.active \
and tree.nodes.active.type == "R_LAYERS"
def _setupNodes(self, context):
scene = context.scene
space = context.space_data
tree = scene.node_tree
bpy.ops.node.select_all(action="DESELECT")
act_node = tree.nodes.active
#rlayer = act_node.scene.render.layers[act_node.layer]
#if not rlayer.use_pass_vector:
#rlayer.use_pass_vector = True
vblur = tree.nodes.new(type="CompositorNodeVecBlur")
vblur.use_curved = True
vblur.factor = 0.5
tree.links.new(act_node.outputs["Image"], vblur.inputs["Image"])
tree.links.new(act_node.outputs["Depth"], vblur.inputs["Z"])
tree.links.new(act_node.outputs["Vector"], vblur.inputs["Speed"])
if tree.nodes.active:
vblur.location = tree.nodes.active.location
vblur.location += Vector((250.0, 0.0))
else:
vblur.location += Vector(
(space.cursor_location[0], space.cursor_location[1]))
vblur.select = True
def execute(self, context):
self._setupNodes(context)
return {"FINISHED"}
|