2010-07-08 16:24:24 +00:00
|
|
|
import bpy
|
2010-02-28 14:57:26 +00:00
|
|
|
import blf
|
2019-05-21 11:53:17 +00:00
|
|
|
import gpu
|
|
|
|
from gpu_extras.batch import batch_for_shader
|
2011-01-26 07:54:27 +00:00
|
|
|
|
2019-10-15 04:20:15 +00:00
|
|
|
|
2010-02-28 09:36:02 +00:00
|
|
|
def draw_callback_px(self, context):
|
2010-02-28 11:18:54 +00:00
|
|
|
print("mouse points", len(self.mouse_path))
|
|
|
|
|
2011-01-26 07:54:27 +00:00
|
|
|
font_id = 0 # XXX, need to find out how best to get this.
|
2010-04-27 19:46:15 +00:00
|
|
|
|
2010-02-28 11:18:54 +00:00
|
|
|
# draw some text
|
2010-04-27 19:46:15 +00:00
|
|
|
blf.position(font_id, 15, 30, 0)
|
|
|
|
blf.size(font_id, 20, 72)
|
|
|
|
blf.draw(font_id, "Hello Word " + str(len(self.mouse_path)))
|
2010-02-28 09:36:02 +00:00
|
|
|
|
|
|
|
# 50% alpha, 2 pixel width line
|
2022-09-02 12:33:29 +00:00
|
|
|
shader = gpu.shader.from_builtin('UNIFORM_COLOR')
|
2021-05-01 14:59:54 +00:00
|
|
|
gpu.state.blend_set('ALPHA')
|
|
|
|
gpu.state.line_width_set(2.0)
|
2019-05-21 11:53:17 +00:00
|
|
|
batch = batch_for_shader(shader, 'LINE_STRIP', {"pos": self.mouse_path})
|
|
|
|
shader.uniform_float("color", (0.0, 0.0, 0.0, 0.5))
|
|
|
|
batch.draw(shader)
|
2010-02-28 09:36:02 +00:00
|
|
|
|
2010-03-06 01:40:29 +00:00
|
|
|
# restore opengl defaults
|
2021-05-01 14:59:54 +00:00
|
|
|
gpu.state.line_width_set(1.0)
|
|
|
|
gpu.state.blend_set('NONE')
|
2010-02-28 09:36:02 +00:00
|
|
|
|
|
|
|
|
|
|
|
class ModalDrawOperator(bpy.types.Operator):
|
2012-07-01 07:55:44 +00:00
|
|
|
"""Draw a line with the mouse"""
|
2021-11-16 09:45:23 +00:00
|
|
|
bl_idname = "view3d.modal_draw_operator"
|
2010-02-28 11:18:54 +00:00
|
|
|
bl_label = "Simple Modal View3D Operator"
|
2010-02-28 09:36:02 +00:00
|
|
|
|
|
|
|
def modal(self, context, event):
|
|
|
|
context.area.tag_redraw()
|
|
|
|
|
|
|
|
if event.type == 'MOUSEMOVE':
|
|
|
|
self.mouse_path.append((event.mouse_region_x, event.mouse_region_y))
|
|
|
|
|
|
|
|
elif event.type == 'LEFTMOUSE':
|
2012-12-20 13:29:58 +00:00
|
|
|
bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW')
|
2010-02-28 09:36:02 +00:00
|
|
|
return {'FINISHED'}
|
|
|
|
|
2011-08-08 05:21:37 +00:00
|
|
|
elif event.type in {'RIGHTMOUSE', 'ESC'}:
|
2012-12-20 13:29:58 +00:00
|
|
|
bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW')
|
2010-02-28 09:36:02 +00:00
|
|
|
return {'CANCELLED'}
|
|
|
|
|
|
|
|
return {'RUNNING_MODAL'}
|
|
|
|
|
|
|
|
def invoke(self, context, event):
|
|
|
|
if context.area.type == 'VIEW_3D':
|
2012-12-20 13:29:58 +00:00
|
|
|
# the arguments we pass the the callback
|
|
|
|
args = (self, context)
|
2010-02-28 09:36:02 +00:00
|
|
|
# Add the region OpenGL drawing callback
|
|
|
|
# draw in view space with 'POST_VIEW' and 'PRE_VIEW'
|
2012-12-20 13:29:58 +00:00
|
|
|
self._handle = bpy.types.SpaceView3D.draw_handler_add(draw_callback_px, args, 'WINDOW', 'POST_PIXEL')
|
2010-02-28 09:36:02 +00:00
|
|
|
|
|
|
|
self.mouse_path = []
|
|
|
|
|
2012-09-05 00:11:39 +00:00
|
|
|
context.window_manager.modal_handler_add(self)
|
2010-02-28 09:36:02 +00:00
|
|
|
return {'RUNNING_MODAL'}
|
|
|
|
else:
|
|
|
|
self.report({'WARNING'}, "View3D not found, cannot run operator")
|
|
|
|
return {'CANCELLED'}
|
2011-02-12 08:04:32 +00:00
|
|
|
|
2022-04-20 05:11:22 +00:00
|
|
|
|
2021-11-16 09:45:23 +00:00
|
|
|
def menu_func(self, context):
|
2022-04-20 05:11:22 +00:00
|
|
|
self.layout.operator(ModalDrawOperator.bl_idname, text="Modal Draw Operator")
|
|
|
|
|
2011-02-12 08:04:32 +00:00
|
|
|
|
2022-04-20 05:11:22 +00:00
|
|
|
# Register and add to the "view" menu (required to also use F3 search "Modal Draw Operator" for quick access).
|
2011-02-12 08:04:32 +00:00
|
|
|
def register():
|
|
|
|
bpy.utils.register_class(ModalDrawOperator)
|
2021-11-16 09:45:23 +00:00
|
|
|
bpy.types.VIEW3D_MT_view.append(menu_func)
|
2011-02-12 08:04:32 +00:00
|
|
|
|
|
|
|
|
|
|
|
def unregister():
|
|
|
|
bpy.utils.unregister_class(ModalDrawOperator)
|
2021-11-16 09:45:23 +00:00
|
|
|
bpy.types.VIEW3D_MT_view.remove(menu_func)
|
2011-02-12 08:04:32 +00:00
|
|
|
|
2018-06-26 17:41:37 +00:00
|
|
|
|
2011-02-12 08:04:32 +00:00
|
|
|
if __name__ == "__main__":
|
|
|
|
register()
|