New tool: Polar Array
Initial, but quite powerful implementation. Not thoroughly tested yet. Supports three modes: number in span, step in span, number and step. Supports manual and linked axis.
This commit is contained in:
DeepSOIC
parent
e74072d1c9
commit
82d3199d24
11
InitGui.py
11
InitGui.py
|
|
@ -36,9 +36,12 @@ __Communication__ = 'vv.titov@gmail.com; DeepSOIC on FreeCAD forum'
|
|||
class LatticeWorkbench (Workbench):
|
||||
MenuText = 'Lattice'
|
||||
def Initialize(self):
|
||||
commandslist1=[]
|
||||
commandslist2=[]
|
||||
|
||||
commandslist0 = []
|
||||
commandslist1 = []
|
||||
commandslist2 = []
|
||||
|
||||
import latticePolarArray
|
||||
commandslist0 = commandslist0 + latticePolarArray.exportedCommands
|
||||
import latticeDowngrade
|
||||
commandslist1 = commandslist1 + latticeDowngrade.exportedCommands
|
||||
import CompoundFilter
|
||||
|
|
@ -48,10 +51,12 @@ class LatticeWorkbench (Workbench):
|
|||
import latticeBoundBox
|
||||
commandslist2 = commandslist2 + latticeBoundBox.exportedCommands
|
||||
|
||||
self.appendToolbar('LatticeArrayTools', commandslist0)
|
||||
self.appendToolbar('LatticeCompoundTools', commandslist1)
|
||||
self.appendToolbar('LatticeMiscTools', commandslist2)
|
||||
#FreeCADGui.addIconPath( '' )
|
||||
#FreeCADGui.addPreferencePage( '','Lattice' )
|
||||
self.appendMenu('Lattice', commandslist0)
|
||||
self.appendMenu('Lattice', commandslist1)
|
||||
self.appendMenu('Lattice', commandslist2)
|
||||
|
||||
|
|
|
|||
|
|
@ -53,5 +53,6 @@ def getIconPath(icon_dot_svg):
|
|||
|
||||
# OCC's Precision::Confusion; should have taken this from FreeCAD but haven't found; unlikely to ever change.
|
||||
DistConfusion = 1e-7
|
||||
ParaConfusion = 1e-8
|
||||
|
||||
import lattice_rc
|
||||
224
latticePolarArray.py
Normal file
224
latticePolarArray.py
Normal file
|
|
@ -0,0 +1,224 @@
|
|||
#***************************************************************************
|
||||
#* *
|
||||
#* Copyright (c) 2015 - Victor Titov (DeepSOIC) *
|
||||
#* <vv.titov@gmail.com> *
|
||||
#* *
|
||||
#* 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 *
|
||||
#* *
|
||||
#***************************************************************************
|
||||
|
||||
__title__="Base feature module for lattice object of lattice workbench for FreeCAD"
|
||||
__author__ = "DeepSOIC"
|
||||
__url__ = ""
|
||||
|
||||
import math
|
||||
|
||||
import FreeCAD as App
|
||||
import Part
|
||||
|
||||
from latticeCommon import *
|
||||
import latticeBaseFeature
|
||||
|
||||
def makePolarArray(name):
|
||||
'''makePolarArray(name): makes a PolarArray object.'''
|
||||
return latticeBaseFeature.makeLatticeFeature(name, PolarArray,'Lattice_PolarArray.svg')
|
||||
|
||||
class PolarArray(latticeBaseFeature.LatticeFeature):
|
||||
"The Lattice PolarArray object"
|
||||
def derivedInit(self,obj):
|
||||
self.Type = "LatticePolarArray"
|
||||
obj.addProperty("App::PropertyEnumeration","Mode","Array","")
|
||||
obj.Mode = ['SpanN','StepN','SpanStep','Spreadsheet']
|
||||
obj.Mode = 'SpanN'
|
||||
|
||||
obj.addProperty("App::PropertyAngle","AngleSpanStart","Array","starting angle for angular span")
|
||||
obj.AngleSpanStart = 0
|
||||
obj.addProperty("App::PropertyAngle","AngleSpanEnd","Array","ending angle for angular span")
|
||||
obj.AngleSpanEnd = 360
|
||||
obj.addProperty("App::PropertyBool","EndInclusive","Array","Determines if the last occurence is placed exactly at the ending angle of the span, or the ending angle is super-last.")
|
||||
obj.EndInclusive = False
|
||||
|
||||
obj.addProperty("App::PropertyAngle","AngleStep","Array","")
|
||||
|
||||
obj.addProperty("App::PropertyFloat","NumberPolar","Array","Number of occurences.")
|
||||
obj.NumberPolar = 5
|
||||
|
||||
obj.addProperty("App::PropertyFloat","Offset","Array","Offset of the first item, expressed as a fraction of angular step.")
|
||||
|
||||
|
||||
obj.addProperty("App::PropertyLength","Radius","Geometric","Radius of the array (set to zero for just rotation).")
|
||||
obj.Radius = 3 #temporary, to see the array (because marker display mode is not implemented yet)
|
||||
|
||||
obj.addProperty("App::PropertyVector","AxisDir","Geometric","Vector that defines axis direction")
|
||||
obj.AxisDir = App.Vector(0,0,1)
|
||||
|
||||
obj.addProperty("App::PropertyVector","AxisPoint","Geometric","Center of rotation")
|
||||
|
||||
obj.addProperty("App::PropertyLink","AxisLink","Geometric","Link to the axis (Edge1 is used for the axis).")
|
||||
|
||||
obj.addProperty("App::PropertyBool","AxisLinkIgnoreDir","Geometric","If True, AxisDir is not updated based on the link.")
|
||||
obj.addProperty("App::PropertyBool","AxisLinkIgnorePoint","Geometric","If True, AxisPoint is not updated based on the link.")
|
||||
|
||||
|
||||
|
||||
def updateReadOnlyness(self, obj):
|
||||
m = obj.Mode
|
||||
obj.setEditorMode("AngleStep", 1 if m == "SpanN" or m == "Spreadsheet" else 0)
|
||||
obj.setEditorMode("AngleSpanEnd", 1 if m == "StepN" or m == "Spreadsheet" else 0)
|
||||
obj.setEditorMode("NumberPolar", 1 if m == "SpanStep" or m == "Spreadsheet" else 0)
|
||||
obj.setEditorMode("AxisDir", 1 if (obj.AxisLink and not obj.AxisLinkIgnoreDir) else 0)
|
||||
obj.setEditorMode("AxisPoint", 1 if (obj.AxisLink and not obj.AxisLinkIgnorePoint) else 0)
|
||||
obj.setEditorMode("AxisLinkIgnoreDir", 0 if obj.AxisLink else 1)
|
||||
obj.setEditorMode("AxisLinkIgnorePoint", 0 if obj.AxisLink else 1)
|
||||
|
||||
|
||||
def execute(self,obj):
|
||||
# Fill in (update read-only) properties that are driven by the mode.
|
||||
self.updateReadOnlyness(obj)
|
||||
if obj.Mode == 'SpanN':
|
||||
n = obj.NumberPolar
|
||||
if obj.EndInclusive:
|
||||
n -= 1
|
||||
if n == 0:
|
||||
n = 1
|
||||
obj.AngleStep = (obj.AngleSpanEnd - obj.AngleSpanStart)/n
|
||||
elif obj.Mode == 'StepN':
|
||||
n = obj.NumberPolar
|
||||
if obj.EndInclusive:
|
||||
n += 1
|
||||
obj.AngleSpanEnd = obj.AngleSpanStart + obj.AngleStep*n
|
||||
elif obj.Mode == 'SpanStep':
|
||||
nfloat = (obj.AngleSpanEnd - obj.AngleSpanStart) / obj.AngleStep
|
||||
n = math.trunc(nfloat + ParaConfusion) + 1
|
||||
if obj.EndInclusive:
|
||||
n = n - 1
|
||||
obj.NumberPolar = n
|
||||
|
||||
# Apply links
|
||||
if obj.AxisLink:
|
||||
axShape = obj.AxisLink
|
||||
edges = axShape.Shape.Edges
|
||||
if len(edges) < 1:
|
||||
raise ValueError('There are no edges in axis link shape!')
|
||||
elif len(edges) > 1:
|
||||
App.Console.PrintWarning('There is more than one edge in shape linked as an axis. The first edge will be used, but the shape link was probably added mistakenly.')
|
||||
edge = edges[0]
|
||||
|
||||
#prepare
|
||||
dir = App.Vector()
|
||||
point = App.Vector()
|
||||
if isinstance(edge.Curve, Part.Line):
|
||||
dir = edge.Curve.EndPoint - edge.Curve.StartPoint
|
||||
point = edge.Curve.StartPoint
|
||||
elif isinstance(edge.Curve, Part.Circle):
|
||||
obj.AxisDir = edge.Curve.Axis
|
||||
obj.AxisPoint = edge.Curve.Center
|
||||
else:
|
||||
raise ValueError("Edge " + repr(edge) + " can't be used to derive an axis. It must be either a line or a circle/arc.")
|
||||
|
||||
#apply
|
||||
if not obj.AxisLinkIgnoreDir:
|
||||
obj.AxisDir = dir
|
||||
if not obj.AxisLinkIgnorePoint:
|
||||
obj.AxisPoint = point
|
||||
|
||||
if obj.Mode != 'Spreadsheet':
|
||||
# Generate the actual array. We can use Step and N directly to
|
||||
# completely avoid mode logic, since we had updated them
|
||||
|
||||
# cache properties into variables
|
||||
step = float(obj.AngleStep)
|
||||
startAng = float(obj.AngleSpanStart) + step*float(obj.Offset)
|
||||
n = int(obj.NumberPolar)
|
||||
radius = float(obj.Radius)
|
||||
|
||||
# compute initial vector. It is to be perpendicular to Axis
|
||||
rot_ini = App.Rotation(App.Vector(0,0,1), obj.AxisDir)
|
||||
overallPlacement = App.Placement(obj.AxisPoint, rot_ini)
|
||||
|
||||
# Make the array
|
||||
rst = [] # Shapes for holding placements (each shape is a vertex with placement)
|
||||
for i in range(0, n):
|
||||
ang = startAng + step*i
|
||||
p = Part.Vertex()
|
||||
localrot = App.Rotation(App.Vector(0,0,1), ang)
|
||||
localtransl = localrot.multVec(App.Vector(radius,0,0))
|
||||
localplm = App.Placement(localtransl, localrot)
|
||||
p.Placement = overallPlacement.multiply(localplm)
|
||||
rst.append(p)
|
||||
|
||||
# Make final compound (or empty-set marker)
|
||||
if len(rst) == 0:
|
||||
scale = 1.0
|
||||
if not obj.Base.Shape.isNull():
|
||||
scale = abs(obj.Radius)
|
||||
if scale < DistConfusion * 100:
|
||||
scale = 1.0
|
||||
obj.Shape = markers.getNullShapeShape(scale)
|
||||
raise ValueError('Array output is null') #Feeding empty compounds to FreeCAD seems to cause rendering issues, otherwise it would have been a good idea to output nothing.
|
||||
|
||||
obj.Shape = Part.makeCompound(rst)
|
||||
else:
|
||||
raise ValueError("Spreadsheet mode not implemeted yet")
|
||||
|
||||
# -------------------------- /common stuff --------------------------------------------------
|
||||
|
||||
# -------------------------- Gui command --------------------------------------------------
|
||||
|
||||
def CreatePolarArray(name):
|
||||
sel = FreeCADGui.Selection.getSelectionEx()
|
||||
FreeCAD.ActiveDocument.openTransaction("Create PolarArray")
|
||||
FreeCADGui.addModule("latticePolarArray")
|
||||
FreeCADGui.doCommand("f = latticePolarArray.makePolarArray(name='"+name+"')")
|
||||
if len(sel) == 1:
|
||||
FreeCADGui.doCommand("f.AxisLink = App.ActiveDocument."+sel[0].ObjectName)
|
||||
FreeCADGui.doCommand("f.Proxy.execute(f)")
|
||||
FreeCADGui.doCommand("f.purgeTouched()")
|
||||
FreeCADGui.doCommand("f = None")
|
||||
FreeCAD.ActiveDocument.commitTransaction()
|
||||
|
||||
|
||||
class _CommandPolarArray:
|
||||
"Command to create PolarArray feature"
|
||||
def GetResources(self):
|
||||
return {'Pixmap' : getIconPath("Lattice_PolarArray.svg"),
|
||||
'MenuText': QtCore.QT_TRANSLATE_NOOP("Lattice_PolarArray","Generate polar array"),
|
||||
'Accel': "",
|
||||
'ToolTip': QtCore.QT_TRANSLATE_NOOP("Lattice_PolarArray","Make a polar array lattice object (array of placements)")}
|
||||
|
||||
def Activated(self):
|
||||
if len(FreeCADGui.Selection.getSelection()) < 2 :
|
||||
CreatePolarArray(name = "PolarArray")
|
||||
else:
|
||||
mb = QtGui.QMessageBox()
|
||||
mb.setIcon(mb.Icon.Warning)
|
||||
mb.setText(translate("Lattice_PolarArray", "Either don't select anything, or select an object to serve an axis. More than one object was selected, not supported.", None))
|
||||
mb.setWindowTitle(translate("Lattice_PolarArray","Bad selection", None))
|
||||
mb.exec_()
|
||||
|
||||
def IsActive(self):
|
||||
if FreeCAD.ActiveDocument:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
FreeCADGui.addCommand('Lattice_PolarArray', _CommandPolarArray())
|
||||
|
||||
exportedCommands = ['Lattice_PolarArray']
|
||||
|
||||
# -------------------------- /Gui command --------------------------------------------------
|
||||
|
||||
Loading…
Reference in New Issue
Block a user