Draft: split wire related tools from Draft.py
Line Polyline BezCurve BSpline . . .
This commit is contained in:
carlopav
committed by
Yorik van Havre
Yorik van Havre
parent
4eab0bb787
commit
f1eaa0b93c
251
src/Mod/Draft/draftobjects/wire.py
Normal file
251
src/Mod/Draft/draftobjects/wire.py
Normal file
@@ -0,0 +1,251 @@
|
||||
# ***************************************************************************
|
||||
# * Copyright (c) 2009, 2010 Yorik van Havre <yorik@uncreated.net> *
|
||||
# * Copyright (c) 2009, 2010 Ken Cline <cline@frii.com> *
|
||||
# * Copyright (c) 2020 FreeCAD Developers *
|
||||
# * *
|
||||
# * This program is free software; you can redistribute it and/or modify *
|
||||
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
||||
# * as published by the Free Software Foundation; either version 2 of *
|
||||
# * the License, or (at your option) any later version. *
|
||||
# * for detail see the LICENCE text file. *
|
||||
# * *
|
||||
# * 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 Library General Public License for more details. *
|
||||
# * *
|
||||
# * You should have received a copy of the GNU Library General Public *
|
||||
# * License along with this program; if not, write to the Free Software *
|
||||
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
||||
# * USA *
|
||||
# * *
|
||||
# ***************************************************************************
|
||||
"""This module provides the object code for Draft Wire.
|
||||
"""
|
||||
## @package wire
|
||||
# \ingroup DRAFT
|
||||
# \brief This module provides the object code for Draft Wire.
|
||||
|
||||
import math
|
||||
|
||||
import FreeCAD as App
|
||||
|
||||
import DraftGeomUtils
|
||||
import DraftVecUtils
|
||||
|
||||
from PySide.QtCore import QT_TRANSLATE_NOOP
|
||||
|
||||
from draftutils.utils import get_param
|
||||
|
||||
from draftobjects.base import DraftObject
|
||||
|
||||
|
||||
|
||||
class Wire(DraftObject):
|
||||
"""The Wire object"""
|
||||
|
||||
def __init__(self, obj):
|
||||
super(Wire, self).__init__(obj, "Wire")
|
||||
|
||||
_tip = "The vertices of the wire"
|
||||
obj.addProperty("App::PropertyVectorList","Points",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "If the wire is closed or not"
|
||||
obj.addProperty("App::PropertyBool","Closed",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The base object is the wire, it's formed from 2 objects"
|
||||
obj.addProperty("App::PropertyLink","Base",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The tool object is the wire, it's formed from 2 objects"
|
||||
obj.addProperty("App::PropertyLink","Tool",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The start point of this line"
|
||||
obj.addProperty("App::PropertyVectorDistance","Start",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The end point of this line"
|
||||
obj.addProperty("App::PropertyVectorDistance","End",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The length of this line"
|
||||
obj.addProperty("App::PropertyLength","Length",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "Radius to use to fillet the corners"
|
||||
obj.addProperty("App::PropertyLength","FilletRadius",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "Size of the chamfer to give to the corners"
|
||||
obj.addProperty("App::PropertyLength","ChamferSize",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "Create a face if this object is closed"
|
||||
obj.addProperty("App::PropertyBool","MakeFace",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The number of subdivisions of each edge"
|
||||
obj.addProperty("App::PropertyInteger","Subdivisions",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
_tip = "The area of this object"
|
||||
obj.addProperty("App::PropertyArea","Area",
|
||||
"Draft",QT_TRANSLATE_NOOP("App::Property", _tip))
|
||||
|
||||
obj.MakeFace = get_param("fillmode",True)
|
||||
obj.Closed = False
|
||||
|
||||
def execute(self, obj):
|
||||
import Part
|
||||
plm = obj.Placement
|
||||
if obj.Base and (not obj.Tool):
|
||||
if obj.Base.isDerivedFrom("Sketcher::SketchObject"):
|
||||
shape = obj.Base.Shape.copy()
|
||||
if obj.Base.Shape.isClosed():
|
||||
if hasattr(obj,"MakeFace"):
|
||||
if obj.MakeFace:
|
||||
shape = Part.Face(shape)
|
||||
else:
|
||||
shape = Part.Face(shape)
|
||||
obj.Shape = shape
|
||||
elif obj.Base and obj.Tool:
|
||||
if hasattr(obj.Base,'Shape') and hasattr(obj.Tool,'Shape'):
|
||||
if (not obj.Base.Shape.isNull()) and (not obj.Tool.Shape.isNull()):
|
||||
sh1 = obj.Base.Shape.copy()
|
||||
sh2 = obj.Tool.Shape.copy()
|
||||
shape = sh1.fuse(sh2)
|
||||
if DraftGeomUtils.isCoplanar(shape.Faces):
|
||||
shape = DraftGeomUtils.concatenate(shape)
|
||||
obj.Shape = shape
|
||||
p = []
|
||||
for v in shape.Vertexes: p.append(v.Point)
|
||||
if obj.Points != p: obj.Points = p
|
||||
elif obj.Points:
|
||||
if obj.Points[0] == obj.Points[-1]:
|
||||
if not obj.Closed: obj.Closed = True
|
||||
obj.Points.pop()
|
||||
if obj.Closed and (len(obj.Points) > 2):
|
||||
pts = obj.Points
|
||||
if hasattr(obj,"Subdivisions"):
|
||||
if obj.Subdivisions > 0:
|
||||
npts = []
|
||||
for i in range(len(pts)):
|
||||
p1 = pts[i]
|
||||
npts.append(pts[i])
|
||||
if i == len(pts)-1:
|
||||
p2 = pts[0]
|
||||
else:
|
||||
p2 = pts[i+1]
|
||||
v = p2.sub(p1)
|
||||
v = DraftVecUtils.scaleTo(v,v.Length/(obj.Subdivisions+1))
|
||||
for j in range(obj.Subdivisions):
|
||||
npts.append(p1.add(App.Vector(v).multiply(j+1)))
|
||||
pts = npts
|
||||
shape = Part.makePolygon(pts+[pts[0]])
|
||||
if "ChamferSize" in obj.PropertiesList:
|
||||
if obj.ChamferSize.Value != 0:
|
||||
w = DraftGeomUtils.filletWire(shape,obj.ChamferSize.Value,chamfer=True)
|
||||
if w:
|
||||
shape = w
|
||||
if "FilletRadius" in obj.PropertiesList:
|
||||
if obj.FilletRadius.Value != 0:
|
||||
w = DraftGeomUtils.filletWire(shape,obj.FilletRadius.Value)
|
||||
if w:
|
||||
shape = w
|
||||
try:
|
||||
if hasattr(obj,"MakeFace"):
|
||||
if obj.MakeFace:
|
||||
shape = Part.Face(shape)
|
||||
else:
|
||||
shape = Part.Face(shape)
|
||||
except Part.OCCError:
|
||||
pass
|
||||
else:
|
||||
edges = []
|
||||
pts = obj.Points[1:]
|
||||
lp = obj.Points[0]
|
||||
for p in pts:
|
||||
if not DraftVecUtils.equals(lp,p):
|
||||
if hasattr(obj,"Subdivisions"):
|
||||
if obj.Subdivisions > 0:
|
||||
npts = []
|
||||
v = p.sub(lp)
|
||||
v = DraftVecUtils.scaleTo(v,v.Length/(obj.Subdivisions+1))
|
||||
edges.append(Part.LineSegment(lp,lp.add(v)).toShape())
|
||||
lv = lp.add(v)
|
||||
for j in range(obj.Subdivisions):
|
||||
edges.append(Part.LineSegment(lv,lv.add(v)).toShape())
|
||||
lv = lv.add(v)
|
||||
else:
|
||||
edges.append(Part.LineSegment(lp,p).toShape())
|
||||
else:
|
||||
edges.append(Part.LineSegment(lp,p).toShape())
|
||||
lp = p
|
||||
try:
|
||||
shape = Part.Wire(edges)
|
||||
except Part.OCCError:
|
||||
print("Error wiring edges")
|
||||
shape = None
|
||||
if "ChamferSize" in obj.PropertiesList:
|
||||
if obj.ChamferSize.Value != 0:
|
||||
w = DraftGeomUtils.filletWire(shape,obj.ChamferSize.Value,chamfer=True)
|
||||
if w:
|
||||
shape = w
|
||||
if "FilletRadius" in obj.PropertiesList:
|
||||
if obj.FilletRadius.Value != 0:
|
||||
w = DraftGeomUtils.filletWire(shape,obj.FilletRadius.Value)
|
||||
if w:
|
||||
shape = w
|
||||
if shape:
|
||||
obj.Shape = shape
|
||||
if hasattr(obj,"Area") and hasattr(shape,"Area"):
|
||||
obj.Area = shape.Area
|
||||
if hasattr(obj,"Length"):
|
||||
obj.Length = shape.Length
|
||||
|
||||
obj.Placement = plm
|
||||
obj.positionBySupport()
|
||||
self.onChanged(obj,"Placement")
|
||||
|
||||
def onChanged(self, obj, prop):
|
||||
if prop == "Start":
|
||||
pts = obj.Points
|
||||
invpl = App.Placement(obj.Placement).inverse()
|
||||
realfpstart = invpl.multVec(obj.Start)
|
||||
if pts:
|
||||
if pts[0] != realfpstart:
|
||||
pts[0] = realfpstart
|
||||
obj.Points = pts
|
||||
|
||||
elif prop == "End":
|
||||
pts = obj.Points
|
||||
invpl = App.Placement(obj.Placement).inverse()
|
||||
realfpend = invpl.multVec(obj.End)
|
||||
if len(pts) > 1:
|
||||
if pts[-1] != realfpend:
|
||||
pts[-1] = realfpend
|
||||
obj.Points = pts
|
||||
|
||||
elif prop == "Length":
|
||||
if obj.Shape and not obj.Shape.isNull():
|
||||
if obj.Length.Value != obj.Shape.Length:
|
||||
if len(obj.Points) == 2:
|
||||
v = obj.Points[-1].sub(obj.Points[0])
|
||||
v = DraftVecUtils.scaleTo(v,obj.Length.Value)
|
||||
obj.Points = [obj.Points[0],obj.Points[0].add(v)]
|
||||
|
||||
elif prop == "Placement":
|
||||
pl = App.Placement(obj.Placement)
|
||||
if len(obj.Points) >= 2:
|
||||
displayfpstart = pl.multVec(obj.Points[0])
|
||||
displayfpend = pl.multVec(obj.Points[-1])
|
||||
if obj.Start != displayfpstart:
|
||||
obj.Start = displayfpstart
|
||||
if obj.End != displayfpend:
|
||||
obj.End = displayfpend
|
||||
|
||||
|
||||
_Wire = Wire
|
||||
Reference in New Issue
Block a user