VladPolskiy/blender-addons
1
0
Fork 0
mirror of https://github.com/blender/blender-addons.git synced 2025-08-16 15:35:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6598f0a25cb0485c6e6fa66964c2d74bf80ff705
blender-addons/curve_tools/cad.py
Campbell Barton 6598f0a25c Cleanup: trailing space
2020-01-29 13:51:36 +11:00

292 lines
11 KiB
Python
Raw Blame History

# ***** 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 3 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, see <http://www.gnu.org/licenses/>.
# All rights reserved.
#
# ***** GPL LICENSE BLOCK *****
bl_info = {
'name': 'Curve CAD Tools',
'author': 'Alexander Meißner',
'version': (1, 0, 0),
'blender': (2, 80, 0),
'category': 'Curve',
'wiki_url': 'https://github.com/Lichtso/curve_cad',
'tracker_url': 'https://github.com/lichtso/curve_cad/issues'
}
import bpy
from . import internal
from . import util
class Fillet(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_fillet'
bl_description = bl_label = 'Fillet'
bl_options = {'REGISTER', 'UNDO'}
radius: bpy.props.FloatProperty(name='Radius', description='Radius of the rounded corners', unit='LENGTH', min=0.0, default=0.1)
chamfer_mode: bpy.props.BoolProperty(name='Chamfer', description='Cut off sharp without rounding', default=False)
limit_half_way: bpy.props.BoolProperty(name='Limit Half Way', description='Limits the segements to half their length in order to prevent collisions', default=False)
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
splines = internal.getSelectedSplines(True, True, True)
if len(splines) == 0:
self.report({'WARNING'}, 'Nothing selected')
return {'CANCELLED'}
for spline in splines:
internal.filletSpline(spline, self.radius, self.chamfer_mode, self.limit_half_way)
bpy.context.object.data.splines.remove(spline)
return {'FINISHED'}
class Boolean(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_boolean'
bl_description = bl_label = 'Boolean'
bl_options = {'REGISTER', 'UNDO'}
operation: bpy.props.EnumProperty(name='Type', items=[
('UNION', 'Union', 'Boolean OR', 0),
('INTERSECTION', 'Intersection', 'Boolean AND', 1),
('DIFFERENCE', 'Difference', 'Active minus Selected', 2)
])
@classmethod
def poll(cls, context):
return util.Selected1Curve()
def execute(self, context):
current_mode = bpy.context.object.mode
bpy.ops.object.mode_set(mode = 'EDIT')
if bpy.context.object.data.dimensions != '2D':
self.report({'WARNING'}, 'Can only be applied in 2D')
return {'CANCELLED'}
splines = internal.getSelectedSplines(True, True)
if len(splines) != 2:
self.report({'WARNING'}, 'Invalid selection. Only work to selected two spline.')
return {'CANCELLED'}
bpy.ops.curve.spline_type_set(type='BEZIER')
splineA = bpy.context.object.data.splines.active
splineB = splines[0] if (splines[1] == splineA) else splines[1]
if not internal.bezierBooleanGeometry(splineA, splineB, self.operation):
self.report({'WARNING'}, 'Invalid selection. Only work to selected two spline.')
return {'CANCELLED'}
bpy.ops.object.mode_set (mode = current_mode)
return {'FINISHED'}
class Intersection(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_intersection'
bl_description = bl_label = 'Intersection'
bl_options = {'REGISTER', 'UNDO'}
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
segments = internal.bezierSegments(bpy.context.object.data.splines, True)
if len(segments) < 2:
self.report({'WARNING'}, 'Invalid selection')
return {'CANCELLED'}
internal.bezierMultiIntersection(segments)
return {'FINISHED'}
class HandleProjection(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_handle_projection'
bl_description = bl_label = 'Handle Projection'
bl_options = {'REGISTER', 'UNDO'}
@classmethod
def poll(cls, context):
return internal.curveObject()
def execute(self, context):
segments = internal.bezierSegments(bpy.context.object.data.splines, True)
if len(segments) < 1:
self.report({'WARNING'}, 'Nothing selected')
return {'CANCELLED'}
internal.bezierProjectHandles(segments)
return {'FINISHED'}
class MergeEnds(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_merge_ends'
bl_description = bl_label = 'Merge Ends'
bl_options = {'REGISTER', 'UNDO'}
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
points = []
selected_splines = []
is_last_point = []
for spline in bpy.context.object.data.splines:
if spline.type != 'BEZIER' or spline.use_cyclic_u:
continue
if spline.bezier_points[0].select_control_point:
points.append(spline.bezier_points[0])
selected_splines.append(spline)
is_last_point.append(False)
if spline.bezier_points[-1].select_control_point:
points.append(spline.bezier_points[-1])
selected_splines.append(spline)
is_last_point.append(True)
if len(points) != 2:
self.report({'WARNING'}, 'Invalid selection')
return {'CANCELLED'}
points[0].handle_left_type = 'FREE'
points[0].handle_right_type = 'FREE'
new_co = (points[0].co+points[1].co)*0.5
handle = (points[1].handle_left if is_last_point[1] else points[1].handle_right)+new_co-points[1].co
if is_last_point[0]:
points[0].handle_left += new_co-points[0].co
points[0].handle_right = handle
else:
points[0].handle_right += new_co-points[0].co
points[0].handle_left = handle
points[0].co = new_co
bpy.ops.curve.select_all(action='DESELECT')
points[1].select_control_point = True
bpy.ops.curve.delete()
selected_splines[0].bezier_points[-1 if is_last_point[0] else 0].select_control_point = True
selected_splines[1].bezier_points[-1 if is_last_point[1] else 0].select_control_point = True
bpy.ops.curve.make_segment()
bpy.ops.curve.select_all(action='DESELECT')
return {'FINISHED'}
class Subdivide(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_subdivide'
bl_description = bl_label = 'Subdivide'
bl_options = {'REGISTER', 'UNDO'}
params: bpy.props.StringProperty(name='Params', default='0.25 0.5 0.75')
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
current_mode = bpy.context.object.mode
bpy.ops.object.mode_set(mode = 'EDIT')
segments = internal.bezierSegments(bpy.context.object.data.splines, True)
if len(segments) == 0:
self.report({'WARNING'}, 'Nothing selected')
return {'CANCELLED'}
cuts = []
for param in self.params.split(' '):
cuts.append({'param': max(0.0, min(float(param), 1.0))})
cuts.sort(key=(lambda cut: cut['param']))
for segment in segments:
segment['cuts'].extend(cuts)
internal.subdivideBezierSegments(segments)
bpy.ops.object.mode_set (mode = current_mode)
return {'FINISHED'}
class Array(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_array'
bl_description = bl_label = 'Array'
bl_options = {'REGISTER', 'UNDO'}
offset: bpy.props.FloatVectorProperty(name='Offset', unit='LENGTH', description='Vector between to copies', subtype='DIRECTION', default=(1.0, 0.0, 0.0), size=3)
count: bpy.props.IntProperty(name='Count', description='Number of copies', min=1, default=2)
connect: bpy.props.BoolProperty(name='Connect', description='Concatenate individual copies', default=False)
serpentine: bpy.props.BoolProperty(name='Serpentine', description='Switch direction of every second copy', default=False)
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
splines = internal.getSelectedSplines(True, True)
if len(splines) == 0:
self.report({'WARNING'}, 'Nothing selected')
return {'CANCELLED'}
internal.arrayModifier(splines, self.offset, self.count, self.connect, self.serpentine)
return {'FINISHED'}
class Circle(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_circle'
bl_description = bl_label = 'Circle'
bl_options = {'REGISTER', 'UNDO'}
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
segments = internal.bezierSegments(bpy.context.object.data.splines, True)
if len(segments) != 1:
self.report({'WARNING'}, 'Invalid selection')
return {'CANCELLED'}
segment = internal.bezierSegmentPoints(segments[0]['beginPoint'], segments[0]['endPoint'])
circle = internal.circleOfBezier(segment)
if circle == None:
self.report({'WARNING'}, 'Not a circle')
return {'CANCELLED'}
bpy.context.scene.cursor.location = circle.center
bpy.context.scene.cursor.rotation_mode = 'QUATERNION'
bpy.context.scene.cursor.rotation_quaternion = circle.orientation.to_quaternion()
return {'FINISHED'}
class Length(bpy.types.Operator):
bl_idname = 'curvetools.bezier_cad_length'
bl_description = bl_label = 'Length'
@classmethod
def poll(cls, context):
return util.Selected1OrMoreCurves()
def execute(self, context):
segments = internal.bezierSegments(bpy.context.object.data.splines, True)
if len(segments) == 0:
self.report({'WARNING'}, 'Nothing selected')
return {'CANCELLED'}
length = 0
for segment in segments:
length += internal.bezierLength(internal.bezierSegmentPoints(segment['beginPoint'], segment['endPoint']))
self.report({'INFO'}, bpy.utils.units.to_string(bpy.context.scene.unit_settings.system, 'LENGTH', length))
return {'FINISHED'}
def register():
for cls in classes:
bpy.utils.register_class(operators)
def unregister():
for cls in classes:
bpy.utils.unregister_class(operators)
if __name__ == "__main__":
register()
operators = [Fillet, Boolean, Intersection, HandleProjection, MergeEnds, Subdivide, Array, Circle, Length]
Reference in New Issue View Git Blame Copy Permalink