#*************************************************************************** #* Copyright (c) 2011 Yorik van Havre * #* * #* 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 * #* * #*************************************************************************** #Modified 2016-01-03 JAndersM import FreeCAD,Draft,ArchComponent,DraftVecUtils,ArchCommands from FreeCAD import Vector import ArchProfile if FreeCAD.GuiUp: import FreeCADGui from PySide import QtCore, QtGui from draftutils.translate import translate from PySide.QtCore import QT_TRANSLATE_NOOP import ArchPrecast import draftguitools.gui_trackers as DraftTrackers else: # \cond def translate(ctxt,txt): return txt def QT_TRANSLATE_NOOP(ctxt,txt): return txt # \endcond ## @package ArchStructure # \ingroup ARCH # \brief The Structure object and tools # # This module provides tools to build Structure objects. # Structure elements are beams, columns, slabs, and other # elements that have a structural function, that is, that # support other parts of the building. __title__= "FreeCAD Structure" __author__ = "Yorik van Havre" __url__ = "https://www.freecad.org" #Reads preset profiles and categorizes them Categories=[] Presets=ArchProfile.readPresets() for pre in Presets: if pre[1] not in Categories: Categories.append(pre[1]) def makeStructure(baseobj=None,length=None,width=None,height=None,name=None): '''makeStructure([baseobj],[length],[width],[height],[name]): creates a structure element based on the given profile object and the given extrusion height. If no base object is given, you can also specify length and width for a cubic object.''' if not FreeCAD.ActiveDocument: FreeCAD.Console.PrintError("No active document. Aborting\n") return p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch") obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython","Structure") _Structure(obj) if FreeCAD.GuiUp: _ViewProviderStructure(obj.ViewObject) if baseobj: obj.Base = baseobj if FreeCAD.GuiUp: obj.Base.ViewObject.hide() if width: obj.Width = width else: obj.Width = p.GetFloat("StructureWidth",100) if height: obj.Height = height else: if not length: obj.Height = p.GetFloat("StructureHeight",1000) if length: obj.Length = length else: if not baseobj: # don't set the length if we have a base object, otherwise the length X height calc # gets wrong obj.Length = p.GetFloat("StructureLength",100) if baseobj: w = 0 h = 0 if hasattr(baseobj,"Width") and hasattr(baseobj,"Height"): w = baseobj.Width.Value h = baseobj.Height.Value elif hasattr(baseobj,"Length") and hasattr(baseobj,"Width"): w = baseobj.Length.Value h = baseobj.Width.Value elif hasattr(baseobj,"Length") and hasattr(baseobj,"Height"): w = baseobj.Length.Value h = baseobj.Height.Value if w and h: if length and not height: obj.Width = w obj.Height = h elif height and not length: obj.Width = w obj.Length = h if not height and not length: obj.IfcType = "Undefined" obj.Label = name if name else translate("Arch","Structure") elif obj.Length > obj.Height: obj.IfcType = "Beam" obj.Label = name if name else translate("Arch","Beam") elif obj.Height > obj.Length: obj.IfcType = "Column" obj.Label = name if name else translate("Arch","Column") return obj def makeStructuralSystem(objects=[],axes=[],name=None): '''makeStructuralSystem([objects],[axes],[name]): makes a structural system based on the given objects and axes''' if not FreeCAD.ActiveDocument: FreeCAD.Console.PrintError("No active document. Aborting\n") return result = [] if not axes: print("At least one axis must be given") return if objects: if not isinstance(objects,list): objects = [objects] else: objects = [None] for o in objects: obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython","StructuralSystem") obj.Label = name if name else translate("Arch","StructuralSystem") _StructuralSystem(obj) if FreeCAD.GuiUp: _ViewProviderStructuralSystem(obj.ViewObject) if o: obj.Base = o obj.Axes = axes result.append(obj) if FreeCAD.GuiUp and o: o.ViewObject.hide() Draft.formatObject(obj,o) FreeCAD.ActiveDocument.recompute() if len(result) == 1: return result[0] else: return result def placeAlongEdge(p1,p2,horizontal=False): """placeAlongEdge(p1,p2,[horizontal]): returns a Placement positioned at p1, with Z axis oriented towards p2. If horizontal is True, then the X axis is oriented towards p2, not the Z axis""" pl = FreeCAD.Placement() pl.Base = p1 up = FreeCAD.Vector(0,0,1) if hasattr(FreeCAD,"DraftWorkingPlane"): up = FreeCAD.DraftWorkingPlane.axis zaxis = p2.sub(p1) yaxis = up.cross(zaxis) if yaxis.Length > 0: xaxis = zaxis.cross(yaxis) if horizontal: pl.Rotation = FreeCAD.Rotation(zaxis,yaxis,xaxis,"ZXY") else: pl.Rotation = FreeCAD.Rotation(xaxis,yaxis,zaxis,"ZXY") pl.Rotation = FreeCAD.Rotation(pl.Rotation.multVec(FreeCAD.Vector(0,0,1)),90).multiply(pl.Rotation) return pl class CommandStructuresFromSelection: """ The Arch Structures from selection command definition. """ def __init__(self): pass def GetResources(self): return {'Pixmap': 'Arch_MultipleStructures', 'MenuText': QT_TRANSLATE_NOOP("Arch_StructuresFromSelection", "Multiple Structures"), 'ToolTip': QT_TRANSLATE_NOOP("Arch_StructuresFromSelection", "Create multiple Arch Structures from a selected base, using each selected edge as an extrusion path")} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): selex = FreeCADGui.Selection.getSelectionEx() if len(selex) >= 2: FreeCAD.ActiveDocument.openTransaction(translate("Arch", "Create Structures From Selection")) FreeCADGui.addModule("Arch") FreeCADGui.addModule("Draft") base = selex[0].Object # The first selected object is the base for the Structure objects for selexi in selex[1:]: # All the edges from the other objects are used as a Tool (extrusion paths) if len(selexi.SubElementNames) == 0: subelement_names = ["Edge" + str(i) for i in range(1, len(selexi.Object.Shape.Edges) + 1)] else: subelement_names = [sub for sub in selexi.SubElementNames if sub.startswith("Edge")] for sub in subelement_names: FreeCADGui.doCommand("structure = Arch.makeStructure(FreeCAD.ActiveDocument." + base.Name + ")") FreeCADGui.doCommand("structure.Tool = (FreeCAD.ActiveDocument." + selexi.Object.Name + ", '" + sub + "')") FreeCADGui.doCommand("structure.BasePerpendicularToTool = True") FreeCADGui.doCommand("Draft.autogroup(structure)") FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() else: FreeCAD.Console.PrintError(translate("Arch", "Please select the base object first and then the edges to use as extrusion paths") + "\n") class CommandStructuralSystem: """ The Arch Structural System command definition. """ def __init__(self): pass def GetResources(self): return {'Pixmap': 'Arch_StructuralSystem', 'MenuText': QT_TRANSLATE_NOOP("Arch_StructuralSystem", "Structural System"), 'ToolTip': QT_TRANSLATE_NOOP("Arch_StructuralSystem", "Create a structural system from a selected structure and axis")} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): sel = FreeCADGui.Selection.getSelection() if sel: st = Draft.getObjectsOfType(sel, "Structure") ax = Draft.getObjectsOfType(sel, "Axis") if ax: FreeCAD.ActiveDocument.openTransaction(translate("Arch", "Create Structural System")) FreeCADGui.addModule("Arch") if st: FreeCADGui.doCommand("obj = Arch.makeStructuralSystem(" + ArchCommands.getStringList(st) + ", " + ArchCommands.getStringList(ax) + ")") else: FreeCADGui.doCommand("obj = Arch.makeStructuralSystem(axes = " + ArchCommands.getStringList(ax) + ")") FreeCADGui.addModule("Draft") FreeCADGui.doCommand("Draft.autogroup(obj)") FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() else: FreeCAD.Console.PrintError(translate("Arch", "Please select at least an axis object") + "\n") class _CommandStructure: "the Arch Structure command definition" def __init__(self): self.beammode = False def GetResources(self): return {'Pixmap' : 'Arch_Structure', 'MenuText': QT_TRANSLATE_NOOP("Arch_Structure","Structure"), 'Accel': "S, T", 'ToolTip': QT_TRANSLATE_NOOP("Arch_Structure","Creates a structure from scratch or from a selected object (sketch, wire, face or solid)")} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch") self.Width = p.GetFloat("StructureWidth",100) if self.beammode: self.Height = p.GetFloat("StructureLength",100) self.Length = p.GetFloat("StructureHeight",1000) else: self.Length = p.GetFloat("StructureLength",100) self.Height = p.GetFloat("StructureHeight",1000) self.Profile = None self.continueCmd = False self.bpoint = None self.bmode = False self.precastvalues = None sel = FreeCADGui.Selection.getSelection() if sel: st = Draft.getObjectsOfType(sel,"Structure") ax = Draft.getObjectsOfType(sel,"Axis") if ax: FreeCADGui.runCommand("Arch_StructuralSystem") return elif not(ax) and not(st): FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Structure")) FreeCADGui.addModule("Arch") for obj in sel: FreeCADGui.doCommand("obj = Arch.makeStructure(FreeCAD.ActiveDocument." + obj.Name + ")") FreeCADGui.addModule("Draft") FreeCADGui.doCommand("Draft.autogroup(obj)") FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() return # interactive mode if hasattr(FreeCAD,"DraftWorkingPlane"): FreeCAD.DraftWorkingPlane.setup() self.points = [] self.tracker = DraftTrackers.boxTracker() self.tracker.width(self.Width) self.tracker.height(self.Height) self.tracker.length(self.Length) self.tracker.setRotation(FreeCAD.DraftWorkingPlane.getRotation().Rotation) self.tracker.on() self.precast = ArchPrecast._PrecastTaskPanel() self.dents = ArchPrecast._DentsTaskPanel() self.precast.Dents = self.dents if self.beammode: title=translate("Arch","First point of the beam")+":" else: title=translate("Arch","Base point of column")+":" FreeCADGui.Snapper.getPoint(callback=self.getPoint,movecallback=self.update,extradlg=[self.taskbox(),self.precast.form,self.dents.form],title=title) def getPoint(self,point=None,obj=None): "this function is called by the snapper when it has a 3D point" self.bmode = self.modeb.isChecked() if point is None: self.tracker.finalize() return if self.bmode and (self.bpoint is None): self.bpoint = point FreeCADGui.Snapper.getPoint(last=point,callback=self.getPoint,movecallback=self.update,extradlg=[self.taskbox(),self.precast.form,self.dents.form],title=translate("Arch","Next point")+":",mode="line") return self.tracker.finalize() horiz = True # determines the type of rotation to apply to the final object FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Structure")) FreeCADGui.addModule("Arch") if self.Profile is not None: try: # try to update latest precast values - fails if dialog has been destroyed already self.precastvalues = self.precast.getValues() except Exception: pass if ("Precast" in self.Profile) and self.precastvalues: # precast concrete self.precastvalues["PrecastType"] = self.Profile.split("_")[1] self.precastvalues["Length"] = self.Length self.precastvalues["Width"] = self.Width self.precastvalues["Height"] = self.Height argstring = "" # fix for precast placement, since their (0,0) point is the lower left corner if self.bmode: delta = FreeCAD.Vector(0,0-self.Width/2,0) else: delta = FreeCAD.Vector(-self.Length/2,-self.Width/2,0) if hasattr(FreeCAD,"DraftWorkingPlane"): delta = FreeCAD.DraftWorkingPlane.getRotation().multVec(delta) point = point.add(delta) if self.bpoint: self.bpoint = self.bpoint.add(delta) # build the string definition for pair in self.precastvalues.items(): argstring += pair[0].lower() + "=" if isinstance(pair[1],str): argstring += '"' + pair[1] + '",' else: argstring += str(pair[1]) + "," FreeCADGui.addModule("ArchPrecast") FreeCADGui.doCommand("s = ArchPrecast.makePrecast("+argstring+")") else: # metal profile FreeCADGui.doCommand('p = Arch.makeProfile('+str(self.Profile)+')') if (abs(self.Length - self.Profile[4]) >= 0.1) or self.bmode: # forgive rounding errors # horizontal FreeCADGui.doCommand('s = Arch.makeStructure(p,length='+str(self.Length)+')') horiz = False else: # vertical FreeCADGui.doCommand('s = Arch.makeStructure(p,height='+str(self.Height)+')') #if not self.bmode: # FreeCADGui.doCommand('s.Placement.Rotation = FreeCAD.Rotation(-0.5,0.5,-0.5,0.5)') FreeCADGui.doCommand('s.Profile = "'+self.Profile[2]+'"') else : FreeCADGui.doCommand('s = Arch.makeStructure(length='+str(self.Length)+',width='+str(self.Width)+',height='+str(self.Height)+')') # calculate rotation if self.bmode and self.bpoint: FreeCADGui.doCommand('s.Placement = Arch.placeAlongEdge('+DraftVecUtils.toString(self.bpoint)+","+DraftVecUtils.toString(point)+","+str(horiz)+")") else: FreeCADGui.doCommand('s.Placement.Base = '+DraftVecUtils.toString(point)) FreeCADGui.doCommand('s.Placement.Rotation = s.Placement.Rotation.multiply(FreeCAD.DraftWorkingPlane.getRotation().Rotation)') FreeCADGui.addModule("Draft") FreeCADGui.doCommand("Draft.autogroup(s)") FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() if self.continueCmd: self.Activated() def _createItemlist(self, baselist): "create nice labels for presets in the task panel" ilist=[] for p in baselist: f = FreeCAD.Units.Quantity(p[4],FreeCAD.Units.Length).getUserPreferred() d = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units").GetInt("Decimals",2) s1 = str(round(p[4]/f[1],d)) s2 = str(round(p[5]/f[1],d)) s3 = str(f[2]) ilist.append(p[2]+" ("+s1+"x"+s2+s3+")") return ilist def taskbox(self): "sets up a taskbox widget" w = QtGui.QWidget() ui = FreeCADGui.UiLoader() w.setWindowTitle(translate("Arch","Structure options")) grid = QtGui.QGridLayout(w) # mode box labelmode = QtGui.QLabel(translate("Arch","Drawing mode")+":") self.modeb = QtGui.QRadioButton(translate("Arch","Beam")) self.modec = QtGui.QRadioButton(translate("Arch","Column")) if self.bpoint or self.beammode: self.modeb.setChecked(True) else: self.modec.setChecked(True) grid.addWidget(labelmode,0,0,1,2) grid.addWidget(self.modeb,1,0,1,1) grid.addWidget(self.modec,1,1,1,1) # categories box labelc = QtGui.QLabel(translate("Arch","Category")) self.valuec = QtGui.QComboBox() self.valuec.addItems([" ","Precast concrete"]+Categories) grid.addWidget(labelc,2,0,1,1) grid.addWidget(self.valuec,2,1,1,1) # presets box labelp = QtGui.QLabel(translate("Arch","Preset")) self.vPresets = QtGui.QComboBox() self.pSelect = [None] fpresets = [" "] self.vPresets.addItems(fpresets) grid.addWidget(labelp,3,0,1,1) grid.addWidget(self.vPresets,3,1,1,1) # length label1 = QtGui.QLabel(translate("Arch","Length")) self.vLength = ui.createWidget("Gui::InputField") if self.modeb.isChecked(): self.vLength.setText(FreeCAD.Units.Quantity(self.Height,FreeCAD.Units.Length).UserString) else: self.vLength.setText(FreeCAD.Units.Quantity(self.Length,FreeCAD.Units.Length).UserString) grid.addWidget(label1,4,0,1,1) grid.addWidget(self.vLength,4,1,1,1) # width label2 = QtGui.QLabel(translate("Arch","Width")) self.vWidth = ui.createWidget("Gui::InputField") self.vWidth.setText(FreeCAD.Units.Quantity(self.Width,FreeCAD.Units.Length).UserString) grid.addWidget(label2,5,0,1,1) grid.addWidget(self.vWidth,5,1,1,1) # height label3 = QtGui.QLabel(translate("Arch","Height")) self.vHeight = ui.createWidget("Gui::InputField") if self.modeb.isChecked(): self.vHeight.setText(FreeCAD.Units.Quantity(self.Length,FreeCAD.Units.Length).UserString) else: self.vHeight.setText(FreeCAD.Units.Quantity(self.Height,FreeCAD.Units.Length).UserString) grid.addWidget(label3,6,0,1,1) grid.addWidget(self.vHeight,6,1,1,1) # horizontal button value5 = QtGui.QPushButton(translate("Arch","Switch L/H")) grid.addWidget(value5,7,0,1,1) value6 = QtGui.QPushButton(translate("Arch","Switch L/W")) grid.addWidget(value6,7,1,1,1) # continue button label4 = QtGui.QLabel(translate("Arch","Con&tinue")) value4 = QtGui.QCheckBox() value4.setObjectName("ContinueCmd") value4.setLayoutDirection(QtCore.Qt.RightToLeft) label4.setBuddy(value4) if hasattr(FreeCADGui,"draftToolBar"): value4.setChecked(FreeCADGui.draftToolBar.continueMode) self.continueCmd = FreeCADGui.draftToolBar.continueMode grid.addWidget(label4,8,0,1,1) grid.addWidget(value4,8,1,1,1) # connect slots QtCore.QObject.connect(self.valuec,QtCore.SIGNAL("currentIndexChanged(int)"),self.setCategory) QtCore.QObject.connect(self.vPresets,QtCore.SIGNAL("currentIndexChanged(int)"),self.setPreset) QtCore.QObject.connect(self.vLength,QtCore.SIGNAL("valueChanged(double)"),self.setLength) QtCore.QObject.connect(self.vWidth,QtCore.SIGNAL("valueChanged(double)"),self.setWidth) QtCore.QObject.connect(self.vHeight,QtCore.SIGNAL("valueChanged(double)"),self.setHeight) QtCore.QObject.connect(value4,QtCore.SIGNAL("stateChanged(int)"),self.setContinue) QtCore.QObject.connect(value5,QtCore.SIGNAL("pressed()"),self.rotateLH) QtCore.QObject.connect(value6,QtCore.SIGNAL("pressed()"),self.rotateLW) QtCore.QObject.connect(self.modeb,QtCore.SIGNAL("toggled(bool)"),self.switchLH) # restore preset stored = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").GetString("StructurePreset","") if stored: if stored.lower().startswith("precast_"): self.valuec.setCurrentIndex(1) tp = stored.split("_")[1] if tp and (tp in self.precast.PrecastTypes): self.vPresets.setCurrentIndex(self.precast.PrecastTypes.index(tp)) elif ";" in stored: stored = stored.split(";") if len(stored) >= 3: if stored[1] in Categories: self.valuec.setCurrentIndex(2+Categories.index(stored[1])) ps = [p[2] for p in self.pSelect] if stored[2] in ps: self.vPresets.setCurrentIndex(ps.index(stored[2])) return w def update(self,point,info): "this function is called by the Snapper when the mouse is moved" if FreeCADGui.Control.activeDialog(): try: # try to update latest precast values - fails if dialog has been destroyed already self.precastvalues = self.precast.getValues() except Exception: pass if self.Height >= self.Length: delta = Vector(0,0,self.Height/2) else: delta = Vector(self.Length/2,0,0) if hasattr(FreeCAD,"DraftWorkingPlane"): delta = FreeCAD.DraftWorkingPlane.getRotation().multVec(delta) if self.modec.isChecked(): self.tracker.pos(point.add(delta)) self.tracker.on() else: if self.bpoint: delta = Vector(0,0,-self.Height/2) if hasattr(FreeCAD,"DraftWorkingPlane"): delta = FreeCAD.DraftWorkingPlane.getRotation().multVec(delta) self.tracker.update([self.bpoint.add(delta),point.add(delta)]) self.tracker.on() l = (point.sub(self.bpoint)).Length self.vLength.setText(FreeCAD.Units.Quantity(l,FreeCAD.Units.Length).UserString) else: self.tracker.off() def setWidth(self,d): self.Width = d self.tracker.width(d) FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetFloat("StructureWidth",d) def setHeight(self,d): self.Height = d self.tracker.height(d) if self.modeb.isChecked(): FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetFloat("StructureLength",d) else: FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetFloat("StructureHeight",d) def setLength(self,d): self.Length = d self.tracker.length(d) if self.modeb.isChecked(): FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetFloat("StructureHeight",d) else: FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetFloat("StructureLength",d) def setContinue(self,i): self.continueCmd = bool(i) if hasattr(FreeCADGui,"draftToolBar"): FreeCADGui.draftToolBar.continueMode = bool(i) def setCategory(self,i): self.vPresets.clear() if i > 1: self.precast.form.hide() self.pSelect = [p for p in Presets if p[1] == Categories[i-2]] fpresets = self._createItemlist(self.pSelect) self.vPresets.addItems(fpresets) self.setPreset(0) elif i == 1: self.precast.form.show() self.pSelect = self.precast.PrecastTypes fpresets = self.precast.PrecastTypes self.vPresets.addItems(fpresets) self.setPreset(0) else: self.precast.form.hide() self.pSelect = [None] fpresets = [" "] self.vPresets.addItems(fpresets) FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetString("StructurePreset","") def setPreset(self,i): self.Profile = None elt = self.pSelect[i] if elt: if elt in self.precast.PrecastTypes: self.precast.setPreset(elt) self.Profile = "Precast_" + elt if elt in ["Pillar","Beam"]: self.dents.form.show() else: self.dents.form.hide() FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetString("StructurePreset",self.Profile) else: p=elt[0]-1 # Presets indexes are 1-based self.vLength.setText(FreeCAD.Units.Quantity(float(Presets[p][4]),FreeCAD.Units.Length).UserString) self.vWidth.setText(FreeCAD.Units.Quantity(float(Presets[p][5]),FreeCAD.Units.Length).UserString) self.Profile = Presets[p] FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetString("StructurePreset",";".join([str(i) for i in self.Profile])) def switchLH(self,bmode): if bmode: self.bmode = True if self.Height > self.Length: self.rotateLH() else: self.bmode = False if self.Length > self.Height: self.rotateLH() self.tracker.setRotation(FreeCAD.Rotation()) def rotateLH(self): h = self.Height l = self.Length self.vLength.setText(FreeCAD.Units.Quantity(h,FreeCAD.Units.Length).UserString) self.vHeight.setText(FreeCAD.Units.Quantity(l,FreeCAD.Units.Length).UserString) def rotateLW(self): w = self.Width l = self.Length self.vLength.setText(FreeCAD.Units.Quantity(w,FreeCAD.Units.Length).UserString) self.vWidth.setText(FreeCAD.Units.Quantity(l,FreeCAD.Units.Length).UserString) class _Structure(ArchComponent.Component): "The Structure object" def __init__(self,obj): ArchComponent.Component.__init__(self,obj) self.setProperties(obj) obj.IfcType = "Beam" def setProperties(self,obj): pl = obj.PropertiesList if not "Tool" in pl: obj.addProperty("App::PropertyLinkSubList", "Tool", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "An optional extrusion path for this element")) if not "ComputedLength" in pl: obj.addProperty("App::PropertyDistance", "ComputedLength", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "The computed length of the extrusion path"), 1) if not "ToolOffsetFirst" in pl: obj.addProperty("App::PropertyDistance", "ToolOffsetFirst", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "Start offset distance along the extrusion path (positive: extend, negative: trim)")) if not "ToolOffsetLast" in pl: obj.addProperty("App::PropertyDistance", "ToolOffsetLast", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "End offset distance along the extrusion path (positive: extend, negative: trim)")) if not "BasePerpendicularToTool" in pl: obj.addProperty("App::PropertyBool", "BasePerpendicularToTool", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "Automatically align the Base of the Structure perpendicular to the Tool axis")) if not "BaseOffsetX" in pl: obj.addProperty("App::PropertyDistance", "BaseOffsetX", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "X offset between the Base origin and the Tool axis (only used if BasePerpendicularToTool is True)")) if not "BaseOffsetY" in pl: obj.addProperty("App::PropertyDistance", "BaseOffsetY", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "Y offset between the Base origin and the Tool axis (only used if BasePerpendicularToTool is True)")) if not "BaseMirror" in pl: obj.addProperty("App::PropertyBool", "BaseMirror", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "Mirror the Base along its Y axis (only used if BasePerpendicularToTool is True)")) if not "BaseRotation" in pl: obj.addProperty("App::PropertyAngle", "BaseRotation", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "Base rotation around the Tool axis (only used if BasePerpendicularToTool is True)")) if not "Length" in pl: obj.addProperty("App::PropertyLength","Length","Structure",QT_TRANSLATE_NOOP("App::Property","The length of this element, if not based on a profile")) if not "Width" in pl: obj.addProperty("App::PropertyLength","Width","Structure",QT_TRANSLATE_NOOP("App::Property","The width of this element, if not based on a profile")) if not "Height" in pl: obj.addProperty("App::PropertyLength","Height","Structure",QT_TRANSLATE_NOOP("App::Property","The height or extrusion depth of this element. Keep 0 for automatic")) if not "Normal" in pl: obj.addProperty("App::PropertyVector","Normal","Structure",QT_TRANSLATE_NOOP("App::Property","The normal extrusion direction of this object (keep (0,0,0) for automatic normal)")) if not "Nodes" in pl: obj.addProperty("App::PropertyVectorList","Nodes","Structure",QT_TRANSLATE_NOOP("App::Property","The structural nodes of this element")) if not "Profile" in pl: obj.addProperty("App::PropertyString","Profile","Structure",QT_TRANSLATE_NOOP("App::Property","A description of the standard profile this element is based upon")) if not "NodesOffset" in pl: obj.addProperty("App::PropertyDistance","NodesOffset","Structure",QT_TRANSLATE_NOOP("App::Property","Offset distance between the centerline and the nodes line")) if not "FaceMaker" in pl: obj.addProperty("App::PropertyEnumeration","FaceMaker","Structure",QT_TRANSLATE_NOOP("App::Property","The facemaker type to use to build the profile of this object")) obj.FaceMaker = ["None","Simple","Cheese","Bullseye"] if not "ArchSketchEdges" in pl: # PropertyStringList obj.addProperty("App::PropertyStringList","ArchSketchEdges","Structure",QT_TRANSLATE_NOOP("App::Property","Selected edges (or group of edges) of the base ArchSketch, to use in creating the shape of this Arch Structure (instead of using all the Base shape's edges by default). Input are index numbers of edges or groups.")) else: # test if the property was added but as IntegerList, then update; type = obj.getTypeIdOfProperty('ArchSketchEdges') if type == "App::PropertyIntegerList": oldIntValue = obj.ArchSketchEdges newStrValue = [str(x) for x in oldIntValue] obj.removeProperty("ArchSketchEdges") obj.addProperty("App::PropertyStringList","ArchSketchEdges","Structure",QT_TRANSLATE_NOOP("App::Property","Selected edges (or group of edges) of the base ArchSketch, to use in creating the shape of this Arch Structure (instead of using all the Base shape's edges by default). Input are index numbers of edges or groups.")) obj.ArchSketchEdges = newStrValue self.Type = "Structure" def onDocumentRestored(self,obj): ArchComponent.Component.onDocumentRestored(self,obj) self.setProperties(obj) def execute(self,obj): "creates the structure shape" import Part, DraftGeomUtils if self.clone(obj): return base = None pl = obj.Placement extdata = self.getExtrusionData(obj) if extdata: sh = extdata[0] if not isinstance(sh,list): sh = [sh] ev = extdata[1] if not isinstance(ev,list): ev = [ev] pla = extdata[2] if not isinstance(pla,list): pla = [pla] base = [] extrusion_length = 0.0 for i in range(len(sh)): shi = sh[i] if i < len(ev): evi = ev[i] else: evi = ev[-1] if isinstance(evi, FreeCAD.Vector): evi = FreeCAD.Vector(evi) else: evi = evi.copy() if i < len(pla): pli = pla[i] else: pli = pla[-1].copy() shi.Placement = pli.multiply(shi.Placement) if isinstance(evi, FreeCAD.Vector): extv = pla[0].Rotation.multVec(evi) shi = shi.extrude(extv) else: try: shi = evi.makePipe(shi) except Part.OCCError: FreeCAD.Console.PrintError(translate("Arch","Error: The base shape couldn't be extruded along this tool object")+"\n") return base.append(shi) extrusion_length += evi.Length if len(base) == 1: base = base[0] else: base = Part.makeCompound(base) obj.ComputedLength = FreeCAD.Units.Quantity(extrusion_length, FreeCAD.Units.Length) if obj.Base: if hasattr(obj.Base,'Shape'): if obj.Base.Shape.isNull(): return if not obj.Base.Shape.isValid(): if not obj.Base.Shape.Solids: # let pass invalid objects if they have solids... return elif obj.Base.Shape.Solids: base = obj.Base.Shape.copy() elif obj.Base.isDerivedFrom("Mesh::Feature"): if obj.Base.Mesh.isSolid(): if obj.Base.Mesh.countComponents() == 1: sh = ArchCommands.getShapeFromMesh(obj.Base.Mesh) if sh.isClosed() and sh.isValid() and sh.Solids and (not sh.isNull()): base = sh else: FreeCAD.Console.PrintWarning(translate("Arch","This mesh is an invalid solid")+"\n") obj.Base.ViewObject.show() if (not base) and (not obj.Additions): #FreeCAD.Console.PrintError(translate("Arch","Error: Invalid base object")+"\n") return base = self.processSubShapes(obj,base,pl) self.applyShape(obj,base,pl) def getExtrusionData(self,obj): """returns (shape,extrusion vector or path,placement) or None""" if hasattr(obj,"IfcType"): IfcType = obj.IfcType else: IfcType = None import Part,DraftGeomUtils data = ArchComponent.Component.getExtrusionData(self,obj) if data: if not isinstance(data[0],list): # multifuses not considered here return data length = obj.Length.Value width = obj.Width.Value height = obj.Height.Value if not height: height = self.getParentHeight(obj) baseface = None extrusion = None normal = None if obj.Base: if hasattr(obj.Base,'Shape'): if obj.Base.Shape: if obj.Base.Shape.Solids: return None elif obj.Base.Shape.Faces: if not DraftGeomUtils.isCoplanar(obj.Base.Shape.Faces,tol=0.01): return None else: baseface = obj.Base.Shape.copy() elif obj.Base.Shape.Wires: # ArchSketch feature : # Get base shape wires, and faceMaker, for Structure (slab. etc.) from Base Objects if they store and provide by getStructureBaseShapeWires() # (thickness, normal/extrusion, length, width, baseface maybe for later) of structure (slab etc.) structureBaseShapeWires = None baseShapeWires = None #baseSlabWires / baseSlabOpeningWires = None faceMaker = None if hasattr(obj.Base, 'Proxy'): if hasattr(obj.Base.Proxy, 'getStructureBaseShapeWires'): structureBaseShapeWires = obj.Base.Proxy.getStructureBaseShapeWires(obj.Base, archsketchEdges=obj.ArchSketchEdges) # provide selected edges, or groups, in obj.ArchSketchEdges for processing in getStructureBaseShapeWires() (getSortedClusters) as override # returned a {dict} ( or a [list] ) # get slab wires; use original wires if structureBaseShapeWires() provided none if structureBaseShapeWires: # would be false (none) if both base ArchSketch and obj do not have the edges stored / inputted by user # if structureBaseShapeWires is {dict} baseShapeWires = structureBaseShapeWires.get('slabWires') faceMaker = structureBaseShapeWires.get('faceMaker') if not baseShapeWires: baseShapeWires = obj.Base.Shape.Wires if faceMaker or (obj.FaceMaker != "None"): if not faceMaker: faceMaker = obj.FaceMaker try: baseface = Part.makeFace(baseShapeWires,"Part::FaceMaker"+str(faceMaker)) except Exception: FreeCAD.Console.PrintError(translate("Arch","Facemaker returned an error")+"\n") # Not returning even Part.makeFace fails, fall back to 'non-Part.makeFace' method if not baseface: for w in baseShapeWires: if not w.isClosed(): p0 = w.OrderedVertexes[0].Point p1 = w.OrderedVertexes[-1].Point if p0 != p1: e = Part.LineSegment(p0,p1).toShape() w.add(e) w.fix(0.1,0,1) # fixes self-intersecting wires f = Part.Face(w) # check if it is 1st face (f) created from w in baseShapeWires; if not, fuse() if baseface: baseface = baseface.fuse(f) else: # TODO use Part.Shape() rather than shape.copy() ... ? baseface = f.copy() elif length and width and height: if (length > height) and (IfcType != "Slab"): h2 = height/2 or 0.5 w2 = width/2 or 0.5 v1 = Vector(0,-w2,-h2) v4 = Vector(0,-w2,h2) v3 = Vector(0,w2,h2) v2 = Vector(0,w2,-h2) else: l2 = length/2 or 0.5 w2 = width/2 or 0.5 v1 = Vector(-l2,-w2,0) v2 = Vector(l2,-w2,0) v3 = Vector(l2,w2,0) v4 = Vector(-l2,w2,0) import Part baseface = Part.Face(Part.makePolygon([v1,v2,v3,v4,v1])) if baseface: if hasattr(obj, "Tool") and obj.Tool: tool = obj.Tool edges = DraftGeomUtils.get_referenced_edges(tool) if len(edges) > 0: extrusion = Part.Wire(Part.__sortEdges__(edges)) if hasattr(obj, "ToolOffsetFirst"): offset_start = float(obj.ToolOffsetFirst.getValueAs("mm")) else: offset_start = 0.0 if hasattr(obj, "ToolOffsetLast"): offset_end = float(obj.ToolOffsetLast.getValueAs("mm")) else: offset_end = 0.0 if offset_start != 0.0 or offset_end != 0.0: extrusion = DraftGeomUtils.get_extended_wire(extrusion, offset_start, offset_end) if hasattr(obj, "BasePerpendicularToTool") and obj.BasePerpendicularToTool: pl = FreeCAD.Placement() if hasattr(obj, "BaseRotation"): pl.rotate(FreeCAD.Vector(0, 0, 0), FreeCAD.Vector(0, 0, 1), -obj.BaseRotation) if hasattr(obj, "BaseOffsetX") and hasattr(obj, "BaseOffsetY"): pl.translate(FreeCAD.Vector(obj.BaseOffsetX, obj.BaseOffsetY, 0)) if hasattr(obj, "BaseMirror") and obj.BaseMirror: pl.rotate(FreeCAD.Vector(0, 0, 0), FreeCAD.Vector(0, 1, 0), 180) baseface.Placement = DraftGeomUtils.get_placement_perpendicular_to_wire(extrusion).multiply(pl) else: if obj.Normal.Length: normal = Vector(obj.Normal).normalize() else: normal = baseface.Faces[0].normalAt(0, 0) ## TODO to use ArchSketch's 'normal' for consistency base = None placement = None inverse_placement = None if len(baseface.Faces) > 1: base = [] placement = [] hint = baseface.Faces[0].normalAt(0, 0) ## TODO anything to do ? for f in baseface.Faces: bf, pf = self.rebase(f, hint) base.append(bf) placement.append(pf) inverse_placement = placement[0].inverse() else: base, placement = self.rebase(baseface) inverse_placement = placement.inverse() if extrusion: if len(extrusion.Edges) == 1 and DraftGeomUtils.geomType(extrusion.Edges[0]) == "Line": extrusion = DraftGeomUtils.vec(extrusion.Edges[0], True) if isinstance(extrusion, FreeCAD.Vector): extrusion = inverse_placement.Rotation.multVec(extrusion) elif normal: normal = inverse_placement.Rotation.multVec(normal) if not normal: normal = Vector(0,0,1) if not normal.Length: normal = Vector(0,0,1) extrusion = normal if (length > height) and (IfcType != "Slab"): if length: extrusion = normal.multiply(length) else: if height: extrusion = normal.multiply(height) if extrusion: return (base, extrusion, placement) return None def onChanged(self,obj,prop): if hasattr(obj,"IfcType"): IfcType = obj.IfcType else: IfcType = None self.hideSubobjects(obj,prop) if prop in ["Shape","ResetNodes","NodesOffset"]: # ResetNodes is not a property but it allows us to use this function to force reset the nodes nodes = None extdata = self.getExtrusionData(obj) if extdata and not isinstance(extdata[0],list): nodes = extdata[0] if IfcType not in ["Slab"]: if not isinstance(extdata[1], FreeCAD.Vector): nodes = extdata[1] elif extdata[1].Length > 0: if hasattr(nodes,"CenterOfMass"): import Part nodes = Part.LineSegment(nodes.CenterOfMass,nodes.CenterOfMass.add(extdata[1])).toShape() if isinstance(extdata[1], FreeCAD.Vector): nodes.Placement = nodes.Placement.multiply(extdata[2]) offset = FreeCAD.Vector() if hasattr(obj,"NodesOffset"): offset = FreeCAD.Vector(0,0,obj.NodesOffset.Value) if obj.Nodes and (prop != "ResetNodes"): if hasattr(self,"nodes"): if self.nodes: if obj.Nodes != self.nodes: # nodes are set manually: don't touch them return else: # nodes haven't been calculated yet, but are set (file load) # we set the nodes now but don't change the property if nodes: self.nodes = [v.Point.add(offset) for v in nodes.Vertexes] return # we set the nodes if nodes: self.nodes = [v.Point.add(offset) for v in nodes.Vertexes] obj.Nodes = self.nodes ArchComponent.Component.onChanged(self,obj,prop) def getNodeEdges(self,obj): "returns a list of edges from stuctural nodes" edges = [] if obj.Nodes: import Part for i in range(len(obj.Nodes)-1): edges.append(Part.LineSegment(obj.Placement.multVec(obj.Nodes[i]),obj.Placement.multVec(obj.Nodes[i+1])).toShape()) if hasattr(obj.ViewObject,"NodeType"): if (obj.ViewObject.NodeType == "Area") and (len(obj.Nodes) > 2): edges.append(Part.LineSegment(obj.Placement.multVec(obj.Nodes[-1]),obj.Placement.multVec(obj.Nodes[0])).toShape()) return edges class _ViewProviderStructure(ArchComponent.ViewProviderComponent): "A View Provider for the Structure object" def __init__(self,vobj): ArchComponent.ViewProviderComponent.__init__(self,vobj) # setProperties of ArchComponent will be overwritten # thus setProperties from ArchComponent will be explicit called to get the properties ArchComponent.ViewProviderComponent.setProperties(self, vobj) self.setProperties(vobj) vobj.ShapeColor = ArchCommands.getDefaultColor("Structure") def setProperties(self,vobj): pl = vobj.PropertiesList if not "ShowNodes" in pl: vobj.addProperty("App::PropertyBool","ShowNodes","Nodes",QT_TRANSLATE_NOOP("App::Property","If the nodes are visible or not")).ShowNodes = False if not "NodeLine" in pl: vobj.addProperty("App::PropertyFloat","NodeLine","Nodes",QT_TRANSLATE_NOOP("App::Property","The width of the nodes line")) if not "NodeSize" in pl: vobj.addProperty("App::PropertyFloat","NodeSize","Nodes",QT_TRANSLATE_NOOP("App::Property","The size of the node points")) vobj.NodeSize = 6 if not "NodeColor" in pl: vobj.addProperty("App::PropertyColor","NodeColor","Nodes",QT_TRANSLATE_NOOP("App::Property","The color of the nodes line")) vobj.NodeColor = (1.0,1.0,1.0,1.0) if not "NodeType" in pl: vobj.addProperty("App::PropertyEnumeration","NodeType","Nodes",QT_TRANSLATE_NOOP("App::Property","The type of structural node")) vobj.NodeType = ["Linear","Area"] def onDocumentRestored(self,vobj): self.setProperties(vobj) def getIcon(self): import Arch_rc if hasattr(self,"Object"): if hasattr(self.Object,"CloneOf"): if self.Object.CloneOf: return ":/icons/Arch_Structure_Clone.svg" return ":/icons/Arch_Structure_Tree.svg" def updateData(self,obj,prop): if prop == "Nodes": if obj.Nodes: if hasattr(self,"nodes"): p = [] self.pointset.numPoints.setValue(0) self.lineset.coordIndex.deleteValues(0) self.faceset.coordIndex.deleteValues(0) for n in obj.Nodes: p.append([n.x,n.y,n.z]) self.coords.point.setValues(0,len(p),p) self.pointset.numPoints.setValue(len(p)) self.lineset.coordIndex.setValues(0,len(p)+1,list(range(len(p)))+[-1]) if hasattr(obj.ViewObject,"NodeType"): if (obj.ViewObject.NodeType == "Area") and (len(p) > 2): self.coords.point.set1Value(len(p),p[0][0],p[0][1],p[0][2]) self.lineset.coordIndex.setValues(0,len(p)+2,list(range(len(p)+1))+[-1]) self.faceset.coordIndex.setValues(0,len(p)+1,list(range(len(p)))+[-1]) elif prop in ["IfcType"]: if hasattr(obj.ViewObject,"NodeType"): if hasattr(obj,"IfcType"): IfcType = obj.IfcType else: IfcType = None if IfcType == "Slab": obj.ViewObject.NodeType = "Area" else: obj.ViewObject.NodeType = "Linear" else: ArchComponent.ViewProviderComponent.updateData(self,obj,prop) def onChanged(self,vobj,prop): if prop == "ShowNodes": if hasattr(self,"nodes"): vobj.Annotation.removeChild(self.nodes) del self.nodes if vobj.ShowNodes: from pivy import coin self.nodes = coin.SoAnnotation() self.coords = coin.SoCoordinate3() self.mat = coin.SoMaterial() self.pointstyle = coin.SoDrawStyle() self.pointstyle.style = coin.SoDrawStyle.POINTS self.pointset = coin.SoType.fromName("SoBrepPointSet").createInstance() self.linestyle = coin.SoDrawStyle() self.linestyle.style = coin.SoDrawStyle.LINES self.lineset = coin.SoType.fromName("SoBrepEdgeSet").createInstance() self.facestyle = coin.SoDrawStyle() self.facestyle.style = coin.SoDrawStyle.FILLED self.shapehints = coin.SoShapeHints() self.shapehints.faceType = coin.SoShapeHints.UNKNOWN_FACE_TYPE self.fmat = coin.SoMaterial() self.fmat.transparency.setValue(0.75) self.faceset = coin.SoIndexedFaceSet() self.nodes.addChild(self.coords) self.nodes.addChild(self.mat) self.nodes.addChild(self.pointstyle) self.nodes.addChild(self.pointset) self.nodes.addChild(self.linestyle) self.nodes.addChild(self.lineset) self.nodes.addChild(self.facestyle) self.nodes.addChild(self.shapehints) self.nodes.addChild(self.fmat) self.nodes.addChild(self.faceset) vobj.Annotation.addChild(self.nodes) self.updateData(vobj.Object,"Nodes") self.onChanged(vobj,"NodeColor") self.onChanged(vobj,"NodeLine") self.onChanged(vobj,"NodeSize") elif prop == "NodeColor": if hasattr(self,"mat"): l = vobj.NodeColor self.mat.diffuseColor.setValue([l[0],l[1],l[2]]) self.fmat.diffuseColor.setValue([l[0],l[1],l[2]]) elif prop == "NodeLine": if hasattr(self,"linestyle"): self.linestyle.lineWidth = vobj.NodeLine elif prop == "NodeSize": if hasattr(self,"pointstyle"): self.pointstyle.pointSize = vobj.NodeSize elif prop == "NodeType": self.updateData(vobj.Object,"Nodes") else: ArchComponent.ViewProviderComponent.onChanged(self,vobj,prop) def setEdit(self,vobj,mode): if mode != 0: return None taskd = StructureTaskPanel(vobj.Object) taskd.obj = self.Object taskd.update() FreeCADGui.Control.showDialog(taskd) return True class StructureTaskPanel(ArchComponent.ComponentTaskPanel): def __init__(self,obj): ArchComponent.ComponentTaskPanel.__init__(self) self.nodes_widget = QtGui.QWidget() self.nodes_widget.setWindowTitle(QtGui.QApplication.translate("Arch", "Node Tools", None)) lay = QtGui.QVBoxLayout(self.nodes_widget) self.resetButton = QtGui.QPushButton(self.nodes_widget) self.resetButton.setIcon(QtGui.QIcon(":/icons/edit-undo.svg")) self.resetButton.setText(QtGui.QApplication.translate("Arch", "Reset nodes", None)) lay.addWidget(self.resetButton) QtCore.QObject.connect(self.resetButton, QtCore.SIGNAL("clicked()"), self.resetNodes) self.editButton = QtGui.QPushButton(self.nodes_widget) self.editButton.setIcon(QtGui.QIcon(":/icons/Draft_Edit.svg")) self.editButton.setText(QtGui.QApplication.translate("Arch", "Edit nodes", None)) lay.addWidget(self.editButton) QtCore.QObject.connect(self.editButton, QtCore.SIGNAL("clicked()"), self.editNodes) self.extendButton = QtGui.QPushButton(self.nodes_widget) self.extendButton.setIcon(QtGui.QIcon(":/icons/Snap_Perpendicular.svg")) self.extendButton.setText(QtGui.QApplication.translate("Arch", "Extend nodes", None)) self.extendButton.setToolTip(QtGui.QApplication.translate("Arch", "Extends the nodes of this element to reach the nodes of another element", None)) lay.addWidget(self.extendButton) QtCore.QObject.connect(self.extendButton, QtCore.SIGNAL("clicked()"), self.extendNodes) self.connectButton = QtGui.QPushButton(self.nodes_widget) self.connectButton.setIcon(QtGui.QIcon(":/icons/Snap_Intersection.svg")) self.connectButton.setText(QtGui.QApplication.translate("Arch", "Connect nodes", None)) self.connectButton.setToolTip(QtGui.QApplication.translate("Arch", "Connects nodes of this element with the nodes of another element", None)) lay.addWidget(self.connectButton) QtCore.QObject.connect(self.connectButton, QtCore.SIGNAL("clicked()"), self.connectNodes) self.toggleButton = QtGui.QPushButton(self.nodes_widget) self.toggleButton.setIcon(QtGui.QIcon(":/icons/dagViewVisible.svg")) self.toggleButton.setText(QtGui.QApplication.translate("Arch", "Toggle all nodes", None)) self.toggleButton.setToolTip(QtGui.QApplication.translate("Arch", "Toggles all structural nodes of the document on/off", None)) lay.addWidget(self.toggleButton) QtCore.QObject.connect(self.toggleButton, QtCore.SIGNAL("clicked()"), self.toggleNodes) self.extrusion_widget = QtGui.QWidget() self.extrusion_widget.setWindowTitle(QtGui.QApplication.translate("Arch", "Extrusion Tools", None)) lay = QtGui.QVBoxLayout(self.extrusion_widget) self.selectToolButton = QtGui.QPushButton(self.extrusion_widget) self.selectToolButton.setIcon(QtGui.QIcon()) self.selectToolButton.setText(QtGui.QApplication.translate("Arch", "Select tool...", None)) self.selectToolButton.setToolTip(QtGui.QApplication.translate("Arch", "Select object or edges to be used as a Tool (extrusion path)", None)) lay.addWidget(self.selectToolButton) QtCore.QObject.connect(self.selectToolButton, QtCore.SIGNAL("clicked()"), self.setSelectionFromTool) self.form = [self.form, self.nodes_widget, self.extrusion_widget] self.Object = obj self.observer = None self.nodevis = None def editNodes(self): FreeCADGui.Control.closeDialog() FreeCADGui.runCommand("Draft_Edit") def resetNodes(self): self.Object.Proxy.onChanged(self.Object,"ResetNodes") def extendNodes(self,other=None): if not other: self.observer = StructSelectionObserver(self.extendNodes) FreeCADGui.Selection.addObserver(self.observer) FreeCAD.Console.PrintMessage(translate("Arch","Choose another Structure object:")) else: FreeCADGui.Selection.removeObserver(self.observer) self.observer = None if Draft.getType(other) != "Structure": FreeCAD.Console.PrintError(translate("Arch","The chosen object is not a Structure")+"\n") else: if not other.Nodes: FreeCAD.Console.PrintError(translate("Arch","The chosen object has no structural nodes")+"\n") else: if (len(self.Object.Nodes) != 2) or (len(other.Nodes) != 2): FreeCAD.Console.PrintError(translate("Arch","One of these objects has more than 2 nodes")+"\n") else: import DraftGeomUtils nodes1 = [self.Object.Placement.multVec(v) for v in self.Object.Nodes] nodes2 = [other.Placement.multVec(v) for v in other.Nodes] intersect = DraftGeomUtils.findIntersection(nodes1[0],nodes1[1],nodes2[0],nodes2[1],True,True) if not intersect: FreeCAD.Console.PrintError(translate("Arch","Unable to find a suitable intersection point")+"\n") else: intersect = intersect[0] FreeCAD.Console.PrintMessage(translate("Arch","Intersection found.\n")) if DraftGeomUtils.findClosest(intersect,nodes1) == 0: self.Object.Nodes = [self.Object.Placement.inverse().multVec(intersect),self.Object.Nodes[1]] else: self.Object.Nodes = [self.Object.Nodes[0],self.Object.Placement.inverse().multVec(intersect)] def connectNodes(self,other=None): if not other: self.observer = StructSelectionObserver(self.connectNodes) FreeCADGui.Selection.addObserver(self.observer) FreeCAD.Console.PrintMessage(translate("Arch","Choose another Structure object:")) else: FreeCADGui.Selection.removeObserver(self.observer) self.observer = None if Draft.getType(other) != "Structure": FreeCAD.Console.PrintError(translate("Arch","The chosen object is not a Structure")+"\n") else: if not other.Nodes: FreeCAD.Console.PrintError(translate("Arch","The chosen object has no structural nodes")+"\n") else: if (len(self.Object.Nodes) != 2) or (len(other.Nodes) != 2): FreeCAD.Console.PrintError(translate("Arch","One of these objects has more than 2 nodes")+"\n") else: import DraftGeomUtils nodes1 = [self.Object.Placement.multVec(v) for v in self.Object.Nodes] nodes2 = [other.Placement.multVec(v) for v in other.Nodes] intersect = DraftGeomUtils.findIntersection(nodes1[0],nodes1[1],nodes2[0],nodes2[1],True,True) if not intersect: FreeCAD.Console.PrintError(translate("Arch","Unable to find a suitable intersection point")+"\n") else: intersect = intersect[0] FreeCAD.Console.PrintMessage(translate("Arch","Intersection found.")+"\n") if DraftGeomUtils.findClosest(intersect,nodes1) == 0: self.Object.Nodes = [self.Object.Placement.inverse().multVec(intersect),self.Object.Nodes[1]] else: self.Object.Nodes = [self.Object.Nodes[0],self.Object.Placement.inverse().multVec(intersect)] if DraftGeomUtils.findClosest(intersect,nodes2) == 0: other.Nodes = [other.Placement.inverse().multVec(intersect),other.Nodes[1]] else: other.Nodes = [other.Nodes[0],other.Placement.inverse().multVec(intersect)] def toggleNodes(self): if self.nodevis: for obj in self.nodevis: obj[0].ViewObject.ShowNodes = obj[1] self.nodevis = None else: self.nodevis = [] for obj in FreeCAD.ActiveDocument.Objects: if hasattr(obj.ViewObject,"ShowNodes"): self.nodevis.append([obj,obj.ViewObject.ShowNodes]) obj.ViewObject.ShowNodes = True def setSelectionFromTool(self): FreeCADGui.Selection.clearSelection() if hasattr(self.Object, "Tool"): tool = self.Object.Tool if hasattr(tool, "Shape") and tool.Shape: FreeCADGui.Selection.addSelection(tool) else: if not isinstance(tool, list): tool = [tool] for o, subs in tool: FreeCADGui.Selection.addSelection(o, subs) QtCore.QObject.disconnect(self.selectToolButton, QtCore.SIGNAL("clicked()"), self.setSelectionFromTool) QtCore.QObject.connect(self.selectToolButton, QtCore.SIGNAL("clicked()"), self.setToolFromSelection) self.selectToolButton.setText(QtGui.QApplication.translate("Arch", "Done", None)) def setToolFromSelection(self): objectList = [] selEx = FreeCADGui.Selection.getSelectionEx() for selExi in selEx: if len(selExi.SubElementNames) == 0: # Add entirely selected objects objectList.append(selExi.Object) else: subElementsNames = [subElementName for subElementName in selExi.SubElementNames if subElementName.startswith("Edge")] # Check that at least an edge is selected from the object's shape if len(subElementsNames) > 0: objectList.append((selExi.Object, subElementsNames)) if self.Object.getTypeIdOfProperty("Tool") != "App::PropertyLinkSubList": # Upgrade property Tool from App::PropertyLink to App::PropertyLinkSubList (note: Undo/Redo fails) self.Object.removeProperty("Tool") self.Object.addProperty("App::PropertyLinkSubList", "Tool", "Structure", QT_TRANSLATE_NOOP("App::Property", "An optional extrusion path for this element")) self.Object.Tool = objectList QtCore.QObject.disconnect(self.selectToolButton, QtCore.SIGNAL("clicked()"), self.setToolFromSelection) QtCore.QObject.connect(self.selectToolButton, QtCore.SIGNAL("clicked()"), self.setSelectionFromTool) self.selectToolButton.setText(QtGui.QApplication.translate("Arch", "Select tool...", None)) def accept(self): if self.observer: FreeCADGui.Selection.removeObserver(self.observer) if self.nodevis: self.toggleNodes() FreeCAD.ActiveDocument.recompute() FreeCADGui.ActiveDocument.resetEdit() return True class StructSelectionObserver: def __init__(self,callback): self.callback = callback def addSelection(self, docName, objName, sub, pos): print("got ",objName) obj = FreeCAD.getDocument(docName).getObject(objName) self.callback(obj) class _StructuralSystem(ArchComponent.Component): # OBSOLETE - All Arch objects can now be based on axes "The Structural System object" def __init__(self,obj): ArchComponent.Component.__init__(self,obj) obj.addProperty("App::PropertyLinkList","Axes","Arch",QT_TRANSLATE_NOOP("App::Property","Axes systems this structure is built on")) obj.addProperty("App::PropertyIntegerList","Exclude","Arch",QT_TRANSLATE_NOOP("App::Property","The element numbers to exclude when this structure is based on axes")) obj.addProperty("App::PropertyBool","Align","Arch",QT_TRANSLATE_NOOP("App::Property","If true the element are aligned with axes")).Align = False self.Type = "StructuralSystem" def execute(self,obj): "creates the structure shape" import Part, DraftGeomUtils # creating base shape pl = obj.Placement if obj.Base: if hasattr(obj.Base,'Shape'): if obj.Base.Shape.isNull(): return if not obj.Base.Shape.Solids: return base = None # applying axes pts = self.getAxisPoints(obj) if hasattr(obj,"Align"): if obj.Align == False : apl = self.getAxisPlacement(obj) if obj.Align: apl = None else : apl = self.getAxisPlacement(obj) if pts: fsh = [] for i in range(len(pts)): sh = obj.Base.Shape.copy() if hasattr(obj,"Exclude"): if i in obj.Exclude: continue if apl: sh.Placement.Rotation = sh.Placement.Rotation.multiply(apl.Rotation) sh.translate(pts[i]) fsh.append(sh) if fsh: base = Part.makeCompound(fsh) base = self.processSubShapes(obj,base,pl) if base: if not base.isNull(): if base.isValid() and base.Solids: if base.Volume < 0: base.reverse() if base.Volume < 0: FreeCAD.Console.PrintError(translate("Arch","Couldn't compute a shape")) return base = base.removeSplitter() obj.Shape = base if not pl.isNull(): obj.Placement = pl def getAxisPoints(self,obj): "returns the gridpoints of linked axes" import DraftGeomUtils pts = [] if len(obj.Axes) == 1: if hasattr(obj,"Align"): if obj.Align: p0 = obj.Axes[0].Shape.Edges[0].Vertexes[1].Point for e in obj.Axes[0].Shape.Edges: p = e.Vertexes[1].Point p = p.sub(p0) pts.append(p) else: for e in obj.Axes[0].Shape.Edges: pts.append(e.Vertexes[0].Point) else: for e in obj.Axes[0].Shape.Edges: pts.append(e.Vertexes[0].Point) elif len(obj.Axes) >= 2: set1 = obj.Axes[0].Shape.Edges set2 = obj.Axes[1].Shape.Edges for e1 in set1: for e2 in set2: pts.extend(DraftGeomUtils.findIntersection(e1,e2)) return pts def getAxisPlacement(self,obj): "returns an axis placement" if obj.Axes: return obj.Axes[0].Placement return None class _ViewProviderStructuralSystem(ArchComponent.ViewProviderComponent): "A View Provider for the Structural System object" def getIcon(self): import Arch_rc return ":/icons/Arch_StructuralSystem_Tree.svg" if FreeCAD.GuiUp: FreeCADGui.addCommand("Arch_Structure", _CommandStructure()) FreeCADGui.addCommand("Arch_StructuralSystem", CommandStructuralSystem()) FreeCADGui.addCommand("Arch_StructuresFromSelection", CommandStructuresFromSelection()) class _ArchStructureGroupCommand: def GetCommands(self): return ("Arch_Structure", "Arch_StructuralSystem", "Arch_StructuresFromSelection") def GetResources(self): return { "MenuText": QT_TRANSLATE_NOOP("Arch_StructureTools", "Structure tools"), "ToolTip": QT_TRANSLATE_NOOP("Arch_StructureTools", "Structure tools") } def IsActive(self): return not FreeCAD.ActiveDocument is None FreeCADGui.addCommand("Arch_StructureTools", _ArchStructureGroupCommand())