108 lines
4.5 KiB
Python
108 lines
4.5 KiB
Python
# ***************************************************************************
|
|
# * 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 the Draft PointArray object.
|
|
"""
|
|
## @package pointarray
|
|
# \ingroup DRAFT
|
|
# \brief This module provides the object code for the Draft PointArray object.
|
|
|
|
import math
|
|
|
|
from PySide.QtCore import QT_TRANSLATE_NOOP
|
|
|
|
import FreeCAD as App
|
|
import DraftVecUtils
|
|
|
|
import draftutils.utils as utils
|
|
|
|
from draftobjects.base import DraftObject
|
|
|
|
|
|
class PointArray(DraftObject):
|
|
"""The Draft Point Array object"""
|
|
|
|
def __init__(self, obj, bobj, ptlst):
|
|
super(PointArray, self).__init__(obj, "PointArray")
|
|
|
|
_tip = "Base object"
|
|
obj.addProperty("App::PropertyLink", "Base",
|
|
"Objects", QT_TRANSLATE_NOOP("App::Property", _tip))
|
|
|
|
_tip = "List of points used to distribute the base object"
|
|
obj.addProperty("App::PropertyLink", "PointList",
|
|
"Objects", QT_TRANSLATE_NOOP("App::Property", _tip))
|
|
|
|
_tip = "Number of copies" # TODO: verify description of the tooltip
|
|
obj.addProperty("App::PropertyInteger", "Count",
|
|
"Parameters", QT_TRANSLATE_NOOP("App::Property", _tip))
|
|
|
|
obj.Base = bobj
|
|
obj.PointList = ptlst
|
|
obj.Count = 0
|
|
|
|
obj.setEditorMode("Count", 1)
|
|
|
|
def execute(self, obj):
|
|
import Part
|
|
pls = []
|
|
opl = obj.PointList
|
|
while utils.get_type(opl) == 'Clone':
|
|
opl = opl.Objects[0]
|
|
if hasattr(opl, 'Geometry'):
|
|
place = opl.Placement
|
|
for pts in opl.Geometry:
|
|
if hasattr(pts, 'X') and hasattr(pts, 'Y') and hasattr(pts, 'Z'):
|
|
pn = pts.copy()
|
|
pn.translate(place.Base)
|
|
pn.rotate(place)
|
|
pls.append(pn)
|
|
elif hasattr(opl, 'Links'):
|
|
pls = opl.Links
|
|
elif hasattr(opl, 'Components'):
|
|
pls = opl.Components
|
|
|
|
base = []
|
|
i = 0
|
|
if hasattr(obj.Base, 'Shape'):
|
|
for pts in pls:
|
|
#print pts # inspect the objects
|
|
if hasattr(pts, 'X') and hasattr(pts, 'Y') and hasattr(pts, 'Z'):
|
|
nshape = obj.Base.Shape.copy()
|
|
if hasattr(pts, 'Placement'):
|
|
place = pts.Placement
|
|
nshape.translate(place.Base)
|
|
nshape.rotate(place.Base, place.Rotation.Axis, place.Rotation.Angle * 180 / math.pi )
|
|
else:
|
|
nshape.translate(App.Vector(pts.X,pts.Y,pts.Z))
|
|
i += 1
|
|
base.append(nshape)
|
|
obj.Count = i
|
|
if i > 0:
|
|
obj.Shape = Part.makeCompound(base)
|
|
else:
|
|
App.Console.PrintError(QT_TRANSLATE_NOOP("draft","No point found\n"))
|
|
obj.Shape = obj.Base.Shape.copy()
|
|
|
|
|
|
_PointArray = PointArray
|