#*************************************************************************** #* * #* 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 DraftTools import translate from PySide.QtCore import QT_TRANSLATE_NOOP 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__ = "http://www.freecadweb.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="Structure"): '''makeStructure([obj],[length],[width],[height],[swap]): 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") obj.Label = translate("Arch","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" elif obj.Length > obj.Height: obj.IfcType = "Beam" obj.Label = translate("Arch","Beam") elif obj.Height > obj.Length: obj.IfcType = "Column" obj.Label = translate("Arch","Column") return obj def makeStructuralSystem(objects=[],axes=[],name="StructuralSystem"): '''makeStructuralSystem(objects,axes): 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",name) obj.Label = translate("Arch",name) _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 _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 object 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: 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() 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() import DraftTrackers,ArchPrecast 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 == None: self.tracker.finalize() return if self.bmode and (self.bpoint == 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: 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: 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::PropertyLink","Tool","Structure",QT_TRANSLATE_NOOP("App::Property","An optional extrusion path for this element")) 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"] 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 import Part, DraftGeomUtils 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 = [] for i in range(len(sh)): shi = sh[i] if i < len(ev): evi = ev[i] else: evi = FreeCAD.Vector(ev[-1]) if i < len(pla): pli = pla[i] else: pli = pla[-1].copy() shi.Placement = pli.multiply(shi.Placement) extv = pla[0].Rotation.multVec(evi) if obj.Tool: try: shi = obj.Tool.Shape.copy().makePipe(obj.Base.Shape.copy()) except Part.OCCError: FreeCAD.Console.PrintError(translate("Arch","Error: The base shape couldn't be extruded along this tool object")+"\n") return else: shi = shi.extrude(extv) base.append(shi) if len(base) == 1: base = base[0] else: base = Part.makeCompound(base) if obj.Base: if obj.Base.isDerivedFrom("Part::Feature"): 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,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 normal = None if not height: for p in obj.InList: if Draft.getType(p) in ["Floor","BuildingPart"]: if p.Height.Value: height = p.Height.Value base = None placement = None if obj.Base: if obj.Base.isDerivedFrom("Part::Feature"): if obj.Base.Shape: if obj.Base.Shape.Solids: return None elif obj.Base.Shape.Faces: if not DraftGeomUtils.isCoplanar(obj.Base.Shape.Faces,tolerance=0.01): return None else: base,placement = self.rebase(obj.Base.Shape) normal = obj.Base.Shape.Faces[0].normalAt(0,0) normal = placement.inverse().Rotation.multVec(normal) if (len(obj.Shape.Solids) > 1) and (len(obj.Shape.Solids) == len(obj.Base.Shape.Faces)): # multiple extrusions b = [] p = [] hint = obj.Base.Shape.Faces[0].normalAt(0,0) for f in obj.Base.Shape.Faces: bf,pf = self.rebase(f,hint) b.append(bf) p.append(pf) base = b placement = p elif obj.Base.Shape.Wires: baseface = None if hasattr(obj,"FaceMaker"): if obj.FaceMaker != "None": try: baseface = Part.makeFace(obj.Base.Shape.Wires,"Part::FaceMaker"+str(obj.FaceMaker)) except: FreeCAD.Console.PrintError(translate("Arch","Facemaker returned an error")+"\n") return None if len(baseface.Faces) > 1: baseface = baseface.Faces[0] normal = baseface.normalAt(0,0) if not baseface: for w in obj.Base.Shape.Wires: if not w.isClosed(): p0 = w.OrderedVertexes[0].Point p1 = w.OrderedVertexes[-1].Point if p0 != p1: e = Part.Line(p0,p1).toShape() w.add(e) w.fix(0.1,0,1) # fixes self-intersecting wires f = Part.Face(w) if baseface: baseface = baseface.fuse(f) else: baseface = f normal = f.normalAt(0,0) base,placement = self.rebase(baseface) normal = placement.inverse().Rotation.multVec(normal) elif (len(obj.Base.Shape.Edges) == 1) and (len(obj.Base.Shape.Vertexes) == 1): # closed edge w = Part.Wire(obj.Base.Shape.Edges[0]) baseface = Part.Face(w) base,placement = self.rebase(baseface) 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])) base,placement = self.rebase(baseface) if base and placement: if obj.Normal.Length: normal = Vector(obj.Normal) if isinstance(placement,list): normal = placement[0].inverse().Rotation.multVec(normal) else: normal = placement.inverse().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) 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] nodes.Placement = nodes.Placement.multiply(extdata[2]) if IfcType not in ["Slab"]: if obj.Tool: nodes = obj.Tool.Shape elif extdata[1].Length > 0: if hasattr(nodes,"CenterOfMass"): import Part nodes = Part.LineSegment(nodes.CenterOfMass,nodes.CenterOfMass.add(extdata[1])).toShape() 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) 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,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,range(len(p)+1)+[-1]) self.faceset.coordIndex.setValues(0,len(p)+1,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: taskd = StructureTaskPanel(vobj.Object) taskd.obj = self.Object taskd.update() FreeCADGui.Control.showDialog(taskd) return True return False class StructureTaskPanel(ArchComponent.ComponentTaskPanel): def __init__(self,obj): ArchComponent.ComponentTaskPanel.__init__(self) self.optwid = QtGui.QWidget() self.optwid.setWindowTitle(QtGui.QApplication.translate("Arch", "Node Tools", None)) lay = QtGui.QVBoxLayout(self.optwid) self.resetButton = QtGui.QPushButton(self.optwid) 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.optwid) 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.optwid) 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.optwid) 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.optwid) 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.form = [self.form,self.optwid] 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 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 obj.Base.isDerivedFrom("Part::Feature"): 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 == True : 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 == True : 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())