# SPDX-License-Identifier: LGPL-2.1-or-later # *************************************************************************** # * * # * Copyright (c) 2011 Yorik van Havre * # * * # * This file is part of FreeCAD. * # * * # * FreeCAD is free software: you can redistribute it and/or modify it * # * under the terms of the GNU Lesser General Public License as * # * published by the Free Software Foundation, either version 2.1 of the * # * License, or (at your option) any later version. * # * * # * FreeCAD 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 * # * Lesser General Public License for more details. * # * * # * You should have received a copy of the GNU Lesser General Public * # * License along with FreeCAD. If not, see * # * . * # * * # *************************************************************************** # Modified 2016-01-03 JAndersM __title__ = "FreeCAD Structure" __author__ = "Yorik van Havre" __url__ = "https://www.freecad.org" ## @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. import FreeCAD import ArchComponent import ArchCommands import ArchProfile import Draft import DraftVecUtils from FreeCAD import Vector from draftutils import params from draftutils import gui_utils if FreeCAD.GuiUp: from PySide import QtCore, QtGui from PySide.QtCore import QT_TRANSLATE_NOOP import FreeCADGui import ArchPrecast import draftguitools.gui_trackers as DraftTrackers from draftutils.translate import translate else: # \cond def translate(ctxt, txt): return txt def QT_TRANSLATE_NOOP(ctxt, txt): return txt # \endcond # 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 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 = params.get_param_arch("StructureWidth") if height: obj.Height = height else: if not length: obj.Height = params.get_param_arch("StructureHeight") 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 = params.get_param_arch("StructureLength") 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 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 import WorkingPlane up = WorkingPlane.get_working_plane(update=False).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", "Creates multiple BIM Structures from a selected base, using each selected edge as an extrusion path", ), } def IsActive(self): v = hasattr(FreeCADGui.getMainWindow().getActiveWindow(), "getSceneGraph") return v 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", "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", "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): self.doc = FreeCAD.ActiveDocument self.Width = params.get_param_arch("StructureWidth") if self.beammode: self.Height = params.get_param_arch("StructureLength") self.Length = params.get_param_arch("StructureHeight") else: self.Length = params.get_param_arch("StructureLength") self.Height = params.get_param_arch("StructureHeight") self.Profile = None self.bpoint = None self.bmode = False self.precastvalues = None self.wp = 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): self.doc.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)") self.doc.commitTransaction() self.doc.recompute() return # interactive mode import WorkingPlane FreeCAD.activeDraftCommand = self # register as a Draft command for auto grid on/off self.wp = WorkingPlane.get_working_plane() self.points = [] self.tracker = DraftTrackers.boxTracker() self.tracker.width(self.Width) self.tracker.height(self.Height) self.tracker.length(self.Length) self.tracker.setRotation(self.wp.get_placement().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, ) FreeCADGui.draftToolBar.continueCmd.show() 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: FreeCAD.activeDraftCommand = None FreeCADGui.Snapper.off() 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 FreeCAD.activeDraftCommand = None FreeCADGui.Snapper.off() self.tracker.off() horiz = True # determines the type of rotation to apply to the final object self.doc.openTransaction(translate("Arch", "Create Structure")) FreeCADGui.addModule("Arch") FreeCADGui.addModule("WorkingPlane") if self.bmode: self.Length = point.sub(self.bpoint).Length params.set_param_arch("StructureHeight", self.Length) 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) delta = self.wp.get_global_coords(delta, as_vector=True) 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("wp = WorkingPlane.get_working_plane()") FreeCADGui.doCommand( "s.Placement.Rotation = s.Placement.Rotation.multiply(wp.get_placement().Rotation)" ) FreeCADGui.addModule("Draft") FreeCADGui.doCommand("Draft.autogroup(s)") self.doc.commitTransaction() self.doc.recompute() # gui_utils.end_all_events() # Causes a crash on Linux. self.tracker.finalize() if FreeCADGui.draftToolBar.continueMode: 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 = params.get_param("Decimals", path="Units") 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", "Parameters of the structure") + ":") 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 value4 = QtGui.QPushButton(translate("Arch", "Switch Length/Height")) grid.addWidget(value4, 7, 0, 1, 1) value5 = QtGui.QPushButton(translate("Arch", "Switch Length/Width")) grid.addWidget(value5, 7, 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("pressed()"), self.rotateLH) QtCore.QObject.connect(value5, QtCore.SIGNAL("pressed()"), self.rotateLW) QtCore.QObject.connect(self.modeb, QtCore.SIGNAL("toggled(bool)"), self.switchLH) # restore preset stored = params.get_param_arch("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) delta = self.wp.get_global_coords(delta, as_vector=True) if self.modec.isChecked(): self.tracker.pos(point.add(delta)) self.tracker.on() else: if self.bpoint: delta = Vector(0, 0, -self.Height / 2) delta = self.wp.get_global_coords(delta, as_vector=True) 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) params.set_param_arch("StructureWidth", d) def setHeight(self, d): self.Height = d self.tracker.height(d) if self.modeb.isChecked(): params.set_param_arch("StructureLength", d) else: params.set_param_arch("StructureHeight", d) def setLength(self, d): self.Length = d self.tracker.length(d) if self.modeb.isChecked(): params.set_param_arch("StructureHeight", d) else: params.set_param_arch("StructureLength", d) 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) params.set_param_arch("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() params.set_param_arch("StructurePreset", self.Profile) else: self.vLength.setText( FreeCAD.Units.Quantity(float(elt[4]), FreeCAD.Units.Length).UserString ) self.vWidth.setText( FreeCAD.Units.Quantity(float(elt[5]), FreeCAD.Units.Length).UserString ) self.Profile = elt params.set_param_arch("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): l = self.vLength.text() h = self.vHeight.text() self.vLength.setText(h) self.vHeight.setText(l) def rotateLW(self): l = self.vLength.text() w = self.vWidth.text() self.vLength.setText(w) self.vWidth.setText(l) class _Structure(ArchComponent.Component): "The Structure object" def __init__(self, obj): ArchComponent.Component.__init__(self, obj) self.Type = "Structure" 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"), locked=True, ) if not "ComputedLength" in pl: obj.addProperty( "App::PropertyDistance", "ComputedLength", "ExtrusionPath", QT_TRANSLATE_NOOP("App::Property", "The computed length of the extrusion path"), 1, locked=True, ) 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)", ), locked=True, ) 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)", ), locked=True, ) 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", ), locked=True, ) 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)", ), locked=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)", ), locked=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)", ), locked=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)", ), locked=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" ), locked=True, ) 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" ), locked=True, ) 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", ), locked=True, ) 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)", ), locked=True, ) if not "Nodes" in pl: obj.addProperty( "App::PropertyVectorList", "Nodes", "Structure", QT_TRANSLATE_NOOP("App::Property", "The structural nodes of this element"), locked=True, ) 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", ), locked=True, ) 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" ), locked=True, ) 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" ), locked=True, ) obj.FaceMaker = ["None", "Simple", "Cheese", "Bullseye"] if not "ArchSketchData" in pl: obj.addProperty( "App::PropertyBool", "ArchSketchData", "Structure", QT_TRANSLATE_NOOP( "App::Property", "Use Base ArchSketch (if used) data (e.g. widths, aligns, offsets) instead of Wall's properties", ), locked=True, ) obj.ArchSketchData = True 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 BIM Structure (instead of using all the Base shape's edges by default). Input are index numbers of edges or groups.", ), locked=True, ) 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 BIM Structure (instead of using all the Base shape's edges by default). Input are index numbers of edges or groups.", ), locked=True, ) obj.ArchSketchEdges = newStrValue if not hasattr(obj, "ArchSketchPropertySet"): obj.addProperty( "App::PropertyEnumeration", "ArchSketchPropertySet", "Structure", QT_TRANSLATE_NOOP( "App::Property", "Select User Defined PropertySet to use in creating variant shape, with same ArchSketch ", ), locked=True, ) obj.ArchSketchPropertySet = ["Default"] if not hasattr(self, "ArchSkPropSetPickedUuid"): self.ArchSkPropSetPickedUuid = "" if not hasattr(self, "ArchSkPropSetListPrev"): self.ArchSkPropSetListPrev = [] def dumps(self): # Supercede Arch.Component.dumps() dump = super().dumps() if not isinstance(dump, tuple): dump = (dump,) # Python Tuple With One Item dump = dump + (self.ArchSkPropSetPickedUuid, self.ArchSkPropSetListPrev) return dump def loads(self, state): self.Type = "Structure" if state == None: return elif state[0] == "S": # state[1] == 't', behaviour before 2024.11.28 return elif state[0] == "Structure": self.ArchSkPropSetPickedUuid = state[1] self.ArchSkPropSetListPrev = state[2] elif state[0] != "Structure": # model before merging super.dumps/loads() self.ArchSkPropSetPickedUuid = state[0] self.ArchSkPropSetListPrev = state[1] def onDocumentRestored(self, obj): ArchComponent.Component.onDocumentRestored(self, obj) self.setProperties(obj) if ( hasattr(obj, "ArchSketchData") and obj.ArchSketchData and Draft.getType(obj.Base) == "ArchSketch" ): if hasattr(obj, "ArchSketchEdges"): obj.setEditorMode("ArchSketchEdges", ["ReadOnly"]) if hasattr(obj, "ArchSketchPropertySet"): obj.setEditorMode("ArchSketchPropertySet", 0) else: if hasattr(obj, "ArchSketchEdges"): obj.setEditorMode("ArchSketchEdges", 0) if hasattr(obj, "ArchSketchPropertySet"): obj.setEditorMode("ArchSketchPropertySet", ["ReadOnly"]) # set a flag to indicate onDocumentRestored() is run def execute(self, obj): "creates the structure shape" import Part import DraftGeomUtils if self.clone(obj): return if obj.Base and not self.ensureBase(obj): return base = None pl = obj.Placement # PropertySet support propSetPickedUuidPrev = self.ArchSkPropSetPickedUuid propSetListPrev = self.ArchSkPropSetListPrev propSetSelectedNamePrev = obj.ArchSketchPropertySet propSetSelectedNameCur = None propSetListCur = None if Draft.getType(obj.Base) == "ArchSketch": baseProxy = obj.Base.Proxy if hasattr(baseProxy, "getPropertySet"): # get full list of PropertySet propSetListCur = baseProxy.getPropertySet(obj.Base) # get updated name (if any) of the selected PropertySet propSetSelectedNameCur = baseProxy.getPropertySet( obj.Base, propSetUuid=propSetPickedUuidPrev ) if propSetSelectedNameCur: # True if selection is not deleted if propSetListPrev != propSetListCur: obj.ArchSketchPropertySet = propSetListCur obj.ArchSketchPropertySet = propSetSelectedNameCur self.ArchSkPropSetListPrev = propSetListCur # elif propSetListPrev == propSetListCur: # pass #nothing to do in this case # but if below, though (propSetListPrev == propSetListCur) elif propSetSelectedNamePrev != propSetSelectedNameCur: obj.ArchSketchPropertySet = propSetSelectedNameCur else: # True if selection is deleted if propSetListCur: if propSetListPrev != propSetListCur: obj.ArchSketchPropertySet = propSetListCur obj.ArchSketchPropertySet = "Default" # else: # Seems no need ... # obj.PropertySet = 'Default' 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 could not 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 import 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 = [] baseShapeWires = [] # baseSlabWires / baseSlabOpeningWires = None faceMaker = None if ( hasattr(obj.Base, "Proxy") and obj.ArchSketchData and hasattr(obj.Base.Proxy, "getStructureBaseShapeWires") ): propSetUuid = self.ArchSkPropSetPickedUuid # provide selected edges, or groups, in obj.ArchSketchEdges for processing in getStructureBaseShapeWires() (getSortedClusters) as override structureBaseShapeWires = obj.Base.Proxy.getStructureBaseShapeWires( obj.Base, propSetUuid=propSetUuid ) # 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") elif obj.Base.isDerivedFrom("Sketcher::SketchObject"): skGeom = obj.Base.GeometryFacadeList skGeomEdges = [] skPlacement = ( obj.Base.Placement ) # Get Sketch's placement to restore later # Get ArchSketch edges to construct ArchStructure # No need to test obj.ArchSketchData ... for ig, geom in enumerate(skGeom): # Construction mode edges should be ignored if # ArchSketchEdges, otherwise, ArchSketchEdges data # needs to take out those in Construction before # using as parameters. if (not obj.ArchSketchEdges and not geom.Construction) or str( ig ) in obj.ArchSketchEdges: # support Line, Arc, Circle, Ellipse, BSplineCurve for Sketch # as Base at the moment if isinstance( geom.Geometry, ( Part.LineSegment, Part.Circle, Part.ArcOfCircle, Part.Ellipse, Part.BSplineCurve, ), ): skGeomEdgesI = geom.Geometry.toShape() skGeomEdges.append(skGeomEdgesI) clusterTransformed = [] for cluster in Part.getSortedClusters(skGeomEdges): edgesTransformed = [] for edge in cluster: edge.Placement = edge.Placement.multiply(skPlacement) edgesTransformed.append(edge) clusterTransformed.append(edgesTransformed) for clusterT in clusterTransformed: baseShapeWires.append(Part.Wire(clusterT)) 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: # check if this was a profil based arch structure # profile-based structures should use XY plane orientation use_profile_orientation = hasattr(obj, "Profile") and obj.Profile if (length > height) and (IfcType in ["Beam", "Column"]): h2 = height / 2 or 0.5 w2 = width / 2 or 0.5 if use_profile_orientation: v1 = Vector(-w2, -h2, 0) v2 = Vector(w2, -h2, 0) v3 = Vector(w2, h2, 0) v4 = Vector(-w2, h2, 0) else: v1 = Vector(0, -w2, -h2) v2 = Vector(0, w2, -h2) v3 = Vector(0, w2, h2) v4 = 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 in ["Beam", "Column"]): 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): # check the flag indicating if we are currently in the process of # restoring document; if not, no further code is run as getExtrusionData() # below return error when some properties are not added by onDocumentRestored() if FreeCAD.ActiveDocument.Restoring: return 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 in ["Beam", "Column"]: 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) if prop == "ArchSketchPropertySet" and Draft.getType(obj.Base) == "ArchSketch": baseProxy = obj.Base.Proxy if hasattr(baseProxy, "getPropertySet"): uuid = baseProxy.getPropertySet(obj, propSetName=obj.ArchSketchPropertySet) self.ArchSkPropSetPickedUuid = uuid if ( hasattr(obj, "ArchSketchData") and obj.ArchSketchData and Draft.getType(obj.Base) == "ArchSketch" ): if hasattr(obj, "ArchSketchEdges"): obj.setEditorMode("ArchSketchEdges", ["ReadOnly"]) if hasattr(obj, "ArchSketchPropertySet"): obj.setEditorMode("ArchSketchPropertySet", 0) else: if hasattr(obj, "ArchSketchEdges"): obj.setEditorMode("ArchSketchEdges", 0) if hasattr(obj, "ArchSketchPropertySet"): obj.setEditorMode("ArchSketchPropertySet", ["ReadOnly"]) def getNodeEdges(self, obj): "returns a list of edges from structural 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"), locked=True, ).ShowNodes = False if not "NodeLine" in pl: vobj.addProperty( "App::PropertyFloat", "NodeLine", "Nodes", QT_TRANSLATE_NOOP("App::Property", "The width of the nodes line"), locked=True, ) if not "NodeSize" in pl: vobj.addProperty( "App::PropertyFloat", "NodeSize", "Nodes", QT_TRANSLATE_NOOP("App::Property", "The size of the node points"), locked=True, ) 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"), locked=True, ) 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"), locked=True, ) 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", "Selects 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"), locked=True, ) 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) self.Type = "StructuralSystem" obj.addProperty( "App::PropertyLinkList", "Axes", "Arch", QT_TRANSLATE_NOOP("App::Property", "Axes systems this structure is built on"), locked=True, ) obj.addProperty( "App::PropertyIntegerList", "Exclude", "Arch", QT_TRANSLATE_NOOP( "App::Property", "The element numbers to exclude when this structure is based on axes", ), locked=True, ) obj.addProperty( "App::PropertyBool", "Align", "Arch", QT_TRANSLATE_NOOP("App::Property", "If true the element are aligned with axes"), locked=True, ).Align = False def loads(self, state): self.Type = "StructuralSystem" def execute(self, obj): "creates the structure shape" import Part import 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", "Could not 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())