remove register/unregister from templates
[blender.git] / release / scripts / templates / operator_modal_draw.py
1 import bpy
2 import bgl
3 import blf
4
5 def draw_callback_px(self, context):
6     print("mouse points", len(self.mouse_path))
7
8     font_id = 0 # XXX, need to find out how best to get this.
9
10     # draw some text
11     blf.position(font_id, 15, 30, 0)
12     blf.size(font_id, 20, 72)
13     blf.draw(font_id, "Hello Word " + str(len(self.mouse_path)))
14
15     # 50% alpha, 2 pixel width line
16     bgl.glEnable(bgl.GL_BLEND)
17     bgl.glColor4f(0.0, 0.0, 0.0, 0.5)
18     bgl.glLineWidth(2)
19
20     bgl.glBegin(bgl.GL_LINE_STRIP)
21     for x, y in self.mouse_path:
22         bgl.glVertex2i(x, y)
23
24     bgl.glEnd()
25
26     # restore opengl defaults
27     bgl.glLineWidth(1)
28     bgl.glDisable(bgl.GL_BLEND)
29     bgl.glColor4f(0.0, 0.0, 0.0, 1.0)
30
31
32 class ModalDrawOperator(bpy.types.Operator):
33     '''Draw a line with the mouse'''
34     bl_idname = "view3d.modal_operator"
35     bl_label = "Simple Modal View3D Operator"
36
37     def modal(self, context, event):
38         context.area.tag_redraw()
39
40         if event.type == 'MOUSEMOVE':
41             self.mouse_path.append((event.mouse_region_x, event.mouse_region_y))
42
43         elif event.type == 'LEFTMOUSE':
44             context.region.callback_remove(self._handle)
45             return {'FINISHED'}
46
47         elif event.type in ('RIGHTMOUSE', 'ESC'):
48             context.region.callback_remove(self._handle)
49             return {'CANCELLED'}
50
51         return {'RUNNING_MODAL'}
52
53     def invoke(self, context, event):
54         if context.area.type == 'VIEW_3D':
55             context.manager.add_modal_handler(self)
56
57             # Add the region OpenGL drawing callback
58             # draw in view space with 'POST_VIEW' and 'PRE_VIEW'
59             self._handle = context.region.callback_add(draw_callback_px, (self, context), 'POST_PIXEL')
60
61             self.mouse_path = []
62
63             return {'RUNNING_MODAL'}
64         else:
65             self.report({'WARNING'}, "View3D not found, cannot run operator")
66             return {'CANCELLED'}