Newer
Older
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
# ##### 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110 - 1301, USA.
#
# ##### END GPL LICENSE BLOCK #####
#
# LOAD MODUL #
import bpy
from bpy import *
from bpy.props import *
from bpy.types import AddonPreferences, PropertyGroup
import bgl
import blf
import gpu
from gpu_extras.batch import batch_for_shader
import math
import mathutils
from mathutils import Vector
from mathutils.geometry import interpolate_bezier
import bpy_extras
from bpy_extras.view3d_utils import location_3d_to_region_2d as loc3d2d
def get_points(spline, matrix_world):
bezier_points = spline.bezier_points
if len(bezier_points) < 2:
return []
r = spline.resolution_u + 1
segments = len(bezier_points)
if not spline.use_cyclic_u:
segments -= 1
point_list = []
for i in range(segments):
inext = (i + 1) % len(bezier_points)
bezier_points1 = matrix_world @ bezier_points[i].co
handle1 = matrix_world @ bezier_points[i].handle_right
handle2 = matrix_world @ bezier_points[inext].handle_left
bezier_points2 = matrix_world @ bezier_points[inext].co
bezier = bezier_points1, handle1, handle2, bezier_points2, r
points = interpolate_bezier(*bezier)
point_list.extend(points)
return point_list
def draw(self, context, spline, curve_vertcolor, matrix_world):
points = get_points(spline, matrix_world)
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
shader = gpu.shader.from_builtin('3D_UNIFORM_COLOR')
batch = batch_for_shader(shader, 'POINTS', {"pos": points})
shader.bind()
shader.uniform_float("color", curve_vertcolor)
batch.draw(shader)
class ShowCurveResolution(bpy.types.Operator):
bl_idname = "curve.show_resolution"
bl_label = "Show Curve Resolution"
bl_description = "Show curve Resolution / [ESC] - remove"
handlers = []
def modal(self, context, event):
context.area.tag_redraw()
if event.type in {'ESC'}:
for handler in self.handlers:
try:
bpy.types.SpaceView3D.draw_handler_remove(handler, 'WINDOW')
except:
pass
for handler in self.handlers:
self.handlers.remove(handler)
return {'CANCELLED'}
return {'PASS_THROUGH'}
def invoke(self, context, event):
if context.area.type == 'VIEW_3D':
# color change in the panel
curve_vertcolor = bpy.context.scene.curvetools.curve_vertcolor
# the arguments we pass the the callback
splines = context.active_object.data.splines
matrix_world = context.active_object.matrix_world
for spline in splines:
args = (self, context, spline, curve_vertcolor, matrix_world)
# Add the region OpenGL drawing callback
# draw in view space with 'POST_VIEW' and 'PRE_VIEW'
self.handlers.append(bpy.types.SpaceView3D.draw_handler_add(draw, args, 'WINDOW', 'POST_VIEW'))
context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'}
else:
self.report({'WARNING'},
"View3D not found, cannot run operator")
return {'CANCELLED'}