#*************************************************************************** #* * #* Copyright (c) 2018 - 2020 * #* Paul Lee * #* * #* 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 * #* * #*************************************************************************** import FreeCAD, FreeCADGui, Sketcher, Part, Draft, DraftVecUtils import ArchComponent, ArchBuilding, ArchFloor, ArchAxis, ArchWall, ArchWindow import ArchSpace, ArchStairs, ArchSectionPlane import SketchArchIcon import SketchArchCommands import math, time from PySide import QtGui, QtCore from FreeCAD import Vector App = FreeCAD Gui = FreeCADGui pi = math.pi zeroMM = FreeCAD.Units.Quantity('0mm') MM = FreeCAD.Units.Quantity('1mm') tolerance = 0.000000001 #--------------------------------------------------------------------------# # Class Definition # #--------------------------------------------------------------------------# class ArchSketchObject: def __init__(self, obj): #self.Type = "ArchSketchObject" pass class ArchSketch(ArchSketchObject): ''' ArchSketch - Sketcher::SketchObjectPython for Architectual Layout ''' MasterSketchSubelementTags = [ "MasterSketchSubelementTag", "MasterSketchIntersectingSubelementTag" ] def __init__(self, obj): ArchSketchObject.__init__(self, obj) ''' call self.setProperties ''' self.setProperties(obj) self.initEditorMode(obj) obj.ViewObject.Proxy=0 return None def initEditorMode(self, obj): ''' Set DispayMode for Data Properties in Combo View Editor ''' obj.setEditorMode("MapMode",1) obj.setEditorMode("MapReversed",1) def setProperties(self, fp): ''' Add self.Properties _&_ Convert Old Properties: See__Init ''' fp.Proxy = self if not hasattr(self,"Type"): self.Type = "ArchSketch" if not hasattr(self,"clEdgeSameIndexFlat"): self.clEdgeSameIndexFlat = [] ''' Added ArchSketch Properties ''' def setPropertiesLinkCommon(self, orgFp, linkFp=None): if linkFp: fp = linkFp else: fp = orgFp prop = fp.PropertiesList for i in ArchSketch.MasterSketchSubelementTags: if linkFp: # no Proxy if i not in prop: linkFp.addProperty("App::PropertyPythonObject", i) setattr(linkFp, i, str()) else: # either ArchSketch or ArchObjects, should have Proxy if orgFp.Proxy.Type == "ArchSketch": if i not in prop: setattr(orgFp.Proxy, i, str()) else: # i.e. other ArchObjects if i not in prop: orgFp.addProperty("App::PropertyPythonObject", i) setattr(orgFp, i, str()) ''' Referenced Object ''' if "MasterSketch" not in prop: fp.addProperty("App::PropertyLink","MasterSketch","Referenced Object","Master Sketch to Attach on") if "MasterSketchSubelement" not in prop: fp.addProperty("App::PropertyString","MasterSketchSubelement","Referenced Object","Master Sketch Sub-Element to Attach on") if "MasterSketchSubelementOffset" not in prop: fp.addProperty("App::PropertyDistance","MasterSketchSubelementOffset","Referenced Object","Master Sketch Sub-Element Attached Offset from Startpoint") if "MasterSketchIntersectingSubelement" not in prop: fp.addProperty("App::PropertyString","MasterSketchIntersectingSubelement", "Referenced Object","Master Sketch Subelement Intersecting the Sub-Element Attached on") if "AttachToSubelementOrOffset" not in prop: fp.addProperty("App::PropertyEnumeration","AttachToSubelementOrOffset","Referenced Object","Select MasterSketch Subelement or Specify Offset to Attach") fp.AttachToSubelementOrOffset = [ "Attach To Edge & Alignment", "Attach to Edge", "Follow Only Offset XYZ & Rotation" ] if "AttachmentOffsetXyzAndRotation" not in prop: fp.addProperty("App::PropertyPlacement","AttachmentOffsetXyzAndRotation","Referenced Object","Specify XYZ and Rotation Offset") if "AttachmentOffsetExtraRotation" not in prop: fp.addProperty("App::PropertyEnumeration","AttachmentOffsetExtraRotation","Referenced Object","Extra Rotation about X, Y or Z Axis") fp.AttachmentOffsetExtraRotation = [ "None", "X-Axis CW90", "X-Axis CCW90", "X-Axis CW180", "Y-Axis CW90", "Y-Axis CCW90", "Y-Axis CW180","Z-Axis CW90", "Z-Axis CCW90", "Z-Axis CW180"] if "FlipOffsetOriginToOtherEnd" not in prop: fp.addProperty("App::PropertyBool","FlipOffsetOriginToOtherEnd","Referenced Object","Flip Offset Origin to Other End of Edge / Wall ") if "Flip180Degree" not in prop: fp.addProperty("App::PropertyBool","Flip180Degree","Referenced Object","Flip Orientation 180 Degree / Inside-Outside / Front-Back") if "OffsetFromIntersectingSubelement" not in prop: fp.addProperty("App::PropertyBool","OffsetFromIntersectingSubelement", "Referenced Object","Offset from the Master Sketch Subelement Intersecting the Sub-Element to Attached on") if "AttachmentAlignment" not in prop: fp.addProperty("App::PropertyEnumeration","AttachmentAlignment","Referenced Object","If AttachToEdge&Alignment, Set EdgeGroupWidthLeft/Right to alignt to EdgeGroupWidth ") fp.AttachmentAlignment = [ "Edge", "EdgeGroupWidthLeft", "EdgeGroupWidthRight" ] fp.AttachmentAlignment = "EdgeGroupWidthRight" # default for Windows which have normal 0,1,0 so somehow set to ArchWindows if "AttachmentAlignmentOffset" not in prop: fp.addProperty("App::PropertyDistance","AttachmentAlignmentOffset","Referenced Object","Set Offset from Edge / EdgeGroupWidth +ve Right / -ve Left") attachToAxisOrSketchExisting = None fpLinkedObject = fp.getLinkedObject() if "AttachToAxisOrSketch" in prop: attachToAxisOrSketchExisting = fp.AttachToAxisOrSketch else: # elif "AttachToAxisOrSketch" not in prop: fp.addProperty("App::PropertyEnumeration","AttachToAxisOrSketch","Referenced Object","Select Object Type to Attach on ") if fpLinkedObject.Proxy.Type == "ArchSketch": fp.AttachToAxisOrSketch = [ "Master Sketch", "Placement Axis" ] else: # i.e. other ArchObjects fp.AttachToAxisOrSketch = [ "None", "Hosts", "Master Sketch", "Placement Axis" ] if attachToAxisOrSketchExisting is not None: fp.AttachToAxisOrSketch = attachToAxisOrSketchExisting elif fpLinkedObject.Proxy.Type == "ArchSketch": fp.AttachToAxisOrSketch = "Master Sketch" # default option for ArchSketch else: # elif fpLinkedObject.Proxy.Type != "ArchSketch": fp.AttachToAxisOrSketch = "None" # default option for ArchObject (ArchWindow) def execute(self, fp): ''' Features to Run in Addition to Sketcher.execute() ''' ''' (IX or XI) - Update the order of edges by getSortedClusters ''' self.updateSortedClustersEdgesOrder(fp) fp.solve() fp.recompute() def updateSortedClustersEdgesOrder(self, fp): clEdgePartnerIndex, clEdgeSameIndex, clEdgeEqualIndex, clEdgePartnerIndexFlat, clEdgeSameIndexFlat, clEdgeEqualIndexFlat = getSortedClustersEdgesOrder(fp) self.clEdgePartnerIndex = clEdgePartnerIndex self.clEdgeSameIndex = clEdgeSameIndex self.clEdgeEqualIndex = clEdgeEqualIndex self.clEdgePartnerIndexFlat = clEdgePartnerIndexFlat self.clEdgeSameIndexFlat = clEdgeSameIndexFlat self.clEdgeEqualIndexFlat = clEdgeEqualIndexFlat #def getWidths(self, fp): # # ''' wrapper function for uniform format ''' # # return self.getSortedClustersEdgesWidth(fp) def getSortedClustersEdgesWidth(self, fp): ''' This method check the SortedClusters-isSame-(flat)List find the corresponding edgesWidth and make a list of (WidthX, WidthX+1 ...) ''' ''' Options of data to store width (& other) information conceived 1st Option - Use self.widths: a Dict of { EdgeX : WidthX, ...} But when Sketch is edit with some edges deleted, the edges indexes change, the widths stored become wrong 2nd Option - Use abdullah's edge geometry .... bugs found, not working 3rd Option - Use self.EdgeTagDictSync .... convoluted object sync, restoreOnLoad ''' return None #def getAligns(self, fp): # # ''' wrapper function for uniform format ''' # # return self.getSortedClustersEdgesAlign(fp) def getSortedClustersEdgesAlign(self, fp): ''' This method check the SortedClusters-isSame-(flat)List find the corresponding edgesAlign ... and make a list of (AlignX, AlignX+1 ...) ''' return None #---------------------------------------------------------------------------# # FreeCAD Commands Classes & Associated Functions # #---------------------------------------------------------------------------# class _CommandEditWallAlign(): '''Edit Wall Segment (Underlying [Arch]Sketch) Align Command Definition''' def GetResources(self): return {'Pixmap' : SketchArchIcon.getIconPath()+'/icons/Edit_Align', 'Accel' : "E, A", 'MenuText': "Edit Wall Segment Align", 'ToolTip' : "Select Wall/ArchSketch to Flip Segment Align ", 'CmdType' : "ForEdit"} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): try: sel0 = Gui.Selection.getSelection()[0] except: reply = QtGui.QMessageBox.information(None,"","Select an Arch Wall ( with underlying Base ArchSketch or Sketch ) or ArchSketch ") return targetObjectBase = None if Draft.getType(sel0) not in ["Wall","ArchSketch"]: reply = QtGui.QMessageBox.information(None,"","Select an Arch Wall ( with underlying Base ArchSketch or Sketch ) or ArchSketch ") return if hasattr(sel0, "Base"): # Wall has Base, ArchSketch does not if sel0.Base: targetObjectBase = sel0.Base else: reply = QtGui.QMessageBox.information(None,"","Arch Wall without Base is not supported - Select an Arch Wall ( with underlying Base ArchSketch or Sketch )") return else: targetObjectBase = sel0 if Draft.getType(sel0.InList[0]) in ["Wall"]: sel0 = sel0.InList[0] else: sel0 = None if Draft.getType(targetObjectBase) in ['ArchSketch', 'Sketch']: if Draft.getType(targetObjectBase) == 'Sketch': reply = QtGui.QMessageBox.information(None,"","Multi-Align support Sketch with Part Geometry Extension (abdullah's development) / ArchSketch primarily. Support on Sketch could only be done 'partially' (indexes of edges is disturbed if sketch is edited) until bug in Part Geometry Extension is fixed - currently for demonstration purpose. Procced now. ") elif Draft.getType(targetObjectBase) == 'ArchSketch': reply = QtGui.QMessageBox.information(None,"","ArchSketch features being added, fallback to treat as Sketch if particular feature not implemented yet - currently for demonstration purpose. Procced now. ") targetObjectBase.ViewObject.HideDependent = False Gui.ActiveDocument.ActiveView.setCameraType("Orthographic") Gui.ActiveDocument.setEdit(targetObjectBase) App.Console.PrintMessage("Select target Edge of the ArchSketch / Sketch to Flip the corresponding Wall Segment Align "+ "\n") FreeCADGui.Selection.clearSelection() s=GuiEditWallAlignObserver(sel0, targetObjectBase) self.observer = s FreeCADGui.Selection.addObserver(s) elif Draft.getType(targetObjectBase) == 'Wire': reply = QtGui.QMessageBox.information(None,"","Gui to edit Arch Wall with a DWire Base is not implemented yet - Please directly edit ArchWall OverrideAlign attribute for the purpose.") FreeCADGui.addCommand('EditWallAlign', _CommandEditWallAlign()) class GuiEditWallAlignObserver(SketchArchCommands.selectObjectObserver): def __init__(self, targetWall, targetWallBaseArchSketch): SketchArchCommands.selectObjectObserver.__init__(self,None,None,None,'Edge') self.targetWall = targetWall # maybe None self.targetArchSketch = targetWallBaseArchSketch # maybe None if self.targetWall: self.targetWallTransparentcy = targetWall.ViewObject.Transparency targetWall.ViewObject.Transparency = 60 if targetWallBaseArchSketch: if Draft.getType(self.targetArchSketch) in ['Sketch','ArchSketch']: if self.targetWall: tempOverrideAlign = self.targetWall.OverrideAlign wallAlign = targetWall.Align # use Wall's Align # filling OverrideAlign if entry is missing for a particular index while len(tempOverrideAlign) < len(self.targetArchSketch.Geometry): tempOverrideAlign.append(wallAlign) #('Left') self.targetWall.OverrideAlign = tempOverrideAlign def proceed(self, doc, obj, sub, pnt): self.edge = sub self.pickedEdgePlacement = App.Vector(pnt) subIndex = int( sub.lstrip('Edge'))-1 if self.targetArchSketch is not None: if Draft.getType(self.targetArchSketch) == 'Sketch': print (" It is a Sketch") curAlign = self.targetWall.OverrideAlign[subIndex] if curAlign == 'Left': curAlign = 'Right' elif curAlign == 'Right': curAlign = 'Center' elif curAlign == 'Center': curAlign = 'Left' else: # 'Center' or else? curAlign = 'Right' # Save information in ArchWall if self.targetWall: tempOverrideAlign = self.targetWall.OverrideAlign tempOverrideAlign[subIndex] = curAlign self.targetWall.OverrideAlign = tempOverrideAlign elif Draft.getType(self.targetArchSketch) == 'ArchSketch': print (" It is an ArchSketch") print (" Full Support not added currently yet !") print (" Fallback to treat as Sketch as 'partial preview' if particular feature Not implemented in ArchSketch yet !") # Test if particular ArchSketch feature has been implemented or not - Fallback to use 'Sketch workflow' if Not if not hasattr(self.targetArchSketch.Proxy, "getEdgeTagDictSyncAlign"): curAlign = self.targetWall.OverrideAlign[subIndex] if curAlign == 'Left': curAlign = 'Right' elif curAlign == 'Right': curAlign = 'Center' elif curAlign == 'Center': curAlign = 'Left' # Test if particular ArchSketch feature has been implemented or not - Fallback to use 'Sketch workflow' if Not # Save information in ArchWall if not hasattr(self.targetArchSketch.Proxy, "getEdgeTagDictSyncAlign"): if self.targetWall: tempOverrideAlign = self.targetWall.OverrideAlign tempOverrideAlign[subIndex] = curAlign if self.targetWall: self.targetWall.OverrideAlign = tempOverrideAlign self.targetArchSketch.recompute() else: # nothing implemented if self.targetArchSketch is None pass if self.targetWall: self.targetWall.recompute() def escape(self,info): k=info['Key'] if k=="ESCAPE": self.targetWall.ViewObject.Transparency = self.targetWallTransparentcy SketchArchCommands.selectObjectObserver.escape(self,info) class _CommandEditWallWidth(): '''Edit Wall Segment (Underlying [Arch]Sketch) Width Command Definition''' def GetResources(self): return {'Pixmap' : SketchArchIcon.getIconPath()+'/icons/Edit_Width', 'Accel' : "E, W", 'MenuText': "Edit Wall Segment Width", 'ToolTip' : "select Wall to Edit Wall Segment Width ", 'CmdType' : "ForEdit"} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): try: sel0 = Gui.Selection.getSelection()[0] except: reply = QtGui.QMessageBox.information(None,"","Select an Arch Wall ( with underlying Base ArchSketch or Sketch ) or ArchSketch ") return targetObjectBase = None if Draft.getType(sel0) not in ["Wall","ArchSketch"]: #if Draft.getType(sel0) != "Wall": reply = QtGui.QMessageBox.information(None,"","Select an Arch Wall ( with underlying Base ArchSketch or Sketch ) or ArchSketch ") return if hasattr(sel0, "Base"): # Wall has Base, ArchSketch does not if sel0.Base: targetObjectBase = sel0.Base else: reply = QtGui.QMessageBox.information(None,"","Arch Wall without Base is not supported - Select an Arch Wall ( with underlying Base ArchSketch or Sketch ) or ArchSketch") return else: targetObjectBase = sel0 if Draft.getType(sel0.InList[0]) in ["Wall"]: sel0 = sel0.InList[0] else: sel0 = None if Draft.getType(targetObjectBase) in ['ArchSketch', 'Sketch']: if Draft.getType(targetObjectBase) == 'Sketch': reply = QtGui.QMessageBox.information(None,"","Multi-Width support Sketch with Part Geometry Extension (abdullah's development) / ArchSketch primarily. Support on Sketch could only be done 'partially' (indexes of edges is disturbed if sketch is edited) until bug in Part Geometry Extension is fixed - currently for demonstration purpose. Procced now. ") elif Draft.getType(targetObjectBase) == 'ArchSketch': reply = QtGui.QMessageBox.information(None,"","ArchSketch features being added, fallback to treat as Sketch if particular feature not implemented yet - currently for demonstration purpose. Procced now. ") targetObjectBase.ViewObject.HideDependent = False Gui.ActiveDocument.ActiveView.setCameraType("Orthographic") Gui.ActiveDocument.setEdit(targetObjectBase) App.Console.PrintMessage("Select target Edge of the ArchSketch / Sketch to Edit the corresponding Wall Segment Width "+ "\n") FreeCADGui.Selection.clearSelection() s=GuiEditWallWidthObserver(sel0, targetObjectBase) self.observer = s FreeCADGui.Selection.addObserver(s) elif Draft.getType(targetObjectBase) == 'Wire': reply = QtGui.QMessageBox.information(None,"","Gui to edit Arch Wall with a DWire Base is not implemented yet - Please directly edit ArchWall OverrideAlign attribute for the purpose.") FreeCADGui.addCommand('EditWallWidth', _CommandEditWallWidth()) class GuiEditWallWidthObserver(SketchArchCommands.selectObjectObserver): def __init__(self, targetWall, targetWallBaseArchSketch): SketchArchCommands.selectObjectObserver.__init__(self,None,None,None,'Edge') self.targetWall = targetWall self.targetArchSketch = targetWallBaseArchSketch self.targetWallTransparentcy = targetWall.ViewObject.Transparency targetWall.ViewObject.Transparency = 60 if targetWallBaseArchSketch: if Draft.getType(self.targetArchSketch) in ['Sketch','ArchSketch']: tempOverrideWidth = self.targetWall.OverrideWidth wallWidth = targetWall.Width.Value # use Wall's Width # filling OverrideWidth if entry is missing for a particular index while len(tempOverrideWidth) < len(self.targetArchSketch.Geometry): tempOverrideWidth.append(wallWidth) #(0) self.targetWall.OverrideWidth = tempOverrideWidth def proceed(self, doc, obj, sub, pnt): self.edge = sub self.pickedEdgePlacement = App.Vector(pnt) subIndex = int( sub.lstrip('Edge'))-1 App.Console.PrintMessage("Input Width"+ "\n") reply = QtGui.QInputDialog.getText(None, "Input Width","Width of Wall Segment") if reply[1]: # user clicked OK if reply[0]: replyWidth = float(reply[0]) else: # no input return None else: # user clicked not OK, i.e. Cancel ? return None if self.targetArchSketch is not None: if Draft.getType(self.targetArchSketch) == 'Sketch': # Save information in ArchWall tempOverrideWidth = self.targetWall.OverrideWidth tempOverrideWidth[subIndex] = replyWidth self.targetWall.OverrideWidth = tempOverrideWidth elif Draft.getType(self.targetArchSketch) == 'ArchSketch': print (" It is an ArchSketch") print (" Full Support not added currently yet !") print (" Fallback to treat as Sketch as 'partial preview' if particular feature Not implemented in ArchSketch yet !") # Test if particular ArchSketch feature has been implemented or not - Fallback to use 'Sketch workflow' if Not # Save information in ArchWall if not hasattr(self.targetArchSketch.Proxy, "getEdgeTagDictSyncAlign"): tempOverrideWidth = self.targetWall.OverrideWidth tempOverrideWidth[subIndex] = replyWidth self.targetWall.OverrideWidth = tempOverrideWidth self.targetArchSketch.recompute() else: # nothing implemented if self.targetArchSketch is None pass self.targetWall.recompute() def escape(self,info): k=info['Key'] if k=="ESCAPE": self.targetWall.ViewObject.Transparency = self.targetWallTransparentcy SketchArchCommands.selectObjectObserver.escape(self,info) class _Command_ArchSketch(): ''' ArchSketch Command Definition - Gui to make an ArchSketch ''' def GetResources(self): return {'Pixmap' : SketchArchIcon.getIconPath() + '/icons/SketchArchWorkbench.svg', 'Accel' : "Alt+S", 'MenuText': "New ArchSketch", 'ToolTip' : "create an ArchSketch"} def IsActive(self): return not FreeCAD.ActiveDocument is None def Activated(self): reply = QtGui.QMessageBox.information(None,"","ArchSketch functionalities being added - none at the moment :) ") makeArchSketch() FreeCADGui.addCommand('ArchSketch', _Command_ArchSketch()) #----------------------------------------------------------------------------# # Functions # #----------------------------------------------------------------------------# def updateAttachmentOffset(fp, linkFp=None): fpOrgSelf = fp.Proxy if linkFp: fp = linkFp print (" fp is re-directed to linkFp... ") hostSketch = None hostWall = None if hasattr(fp, "Hosts"): # Arch Window if fp.Hosts: hostWall = fp.Hosts[0] # Can just take 1st Host wall if fp.Hosts[0].Base.isDerivedFrom("Sketcher::SketchObject"): hostSketch = fp.Hosts[0].Base # Host wall's base Sketch if not hostSketch and hasattr(fp, "MasterSketch"): hostSketch = fp.MasterSketch if not hostSketch: return # if found no Sketch to 'attach' to / calculate placement masterSketchSubelement = fp.MasterSketchSubelement masterSketchSubelementOffset = fp.MasterSketchSubelementOffset attachToAxisOrSketch = fp.AttachToAxisOrSketch attachToSubelementOrOffset = fp.AttachToSubelementOrOffset attachmentOffsetXyzAndRotation = fp.AttachmentOffsetXyzAndRotation flipOffsetOriginToOtherEnd = fp.FlipOffsetOriginToOtherEnd flip180Degree = fp.Flip180Degree offsetFromIntersectingSubelement = fp.OffsetFromIntersectingSubelement attachmentAlignment = fp.AttachmentAlignment attachmentAlignmentOffset = fp.AttachmentAlignmentOffset masterSketchSubelementTag = None masterSketchIntersectingSubelementTag = None masterSketchSubelementEdge = None masterSketchSubelementIndex = None if hasattr(fp, "Proxy"): # ArchSketch / Arch Objects if fp.Proxy.Type == "ArchSketch": masterSketchSubelementTag = fp.Proxy.MasterSketchSubelementTag else: # Other Arch Objects (Windows / Doors) masterSketchSubelementEdge = masterSketchSubelement if fp.Proxy.Type == "ArchSketch": masterSketchIntersectingSubelementTag = fp.Proxy.MasterSketchIntersectingSubelementTag else: # Other Arch Objects (Windows / Doors) masterSketchIntersectingSubelementTag = fp.MasterSketchIntersectingSubelementTag else: # Link objects (of ArchSketch or Arch Windows / Doors) masterSketchSubelementEdge = masterSketchSubelement masterSketchIntersectingSubelementTag = fp.MasterSketchIntersectingSubelementTag if not masterSketchSubelementEdge: masterSketchSubelementEdge = "Edge1" # default be 1 masterSketchSubelementIndex = int(masterSketchSubelementEdge.lstrip('Edge'))-1 if attachToAxisOrSketch in ["Hosts", "Master Sketch"]: # == "Master Sketch": tempAttachmentOffset = FreeCAD.Placement() if (attachToSubelementOrOffset in [ "Attach to Edge", "Attach To Edge & Alignment"] ) and masterSketchSubelementEdge is not None: masterSketchIntersectingSubelementEdge = None edgeOffsetPointVector = getSketchEdgeOffsetPointVector(fp, hostSketch, masterSketchSubelementEdge, masterSketchSubelementOffset, attachmentOffsetXyzAndRotation, flipOffsetOriginToOtherEnd, flip180Degree, attachToSubelementOrOffset, offsetFromIntersectingSubelement, masterSketchIntersectingSubelementEdge) if attachToSubelementOrOffset == "Attach To Edge & Alignment": edgeAngle = getSketchEdgeAngle(hostSketch, masterSketchSubelementEdge) tempAttachmentOffset.Rotation.Angle = edgeAngle else: tempAttachmentOffset.Rotation.Angle = attachmentOffsetXyzAndRotation.Rotation.Angle #edgeAngle = attachmentOffsetXyzAndRotation.Rotation.Angle tempAttachmentOffset.Base= edgeOffsetPointVector ''' Offset Parallel from Line Alignment ''' masterSketchSubelementEdgeVec = getSketchEdgeVec(hostSketch, masterSketchSubelementEdge) masterSketchSubelementEdgeGroupWidth = zeroMM if attachmentAlignment in ["EdgeGroupWidthLeft", "EdgeGroupWidthRight"]: if hasattr(ArchSketch, "getEdgesIndexAndWidthInEdgeGroup") and hasattr(hostSketch.Proxy,"EdgeTagDictSync"): none, masterSketchSubelementEdgeGroupWidth, none,none,align = ArchSketch.getEdgesIndexAndWidthInEdgeGroup(hostSketch.Proxy, hostSketch, None, None, masterSketchSubelementIndex, None) elif hostWall: try: masterSketchSubelementEdgeGroupWidth = hostWall.OverrideWidth[masterSketchSubelementIndex]*MM except: masterSketchSubelementEdgeGroupWidth = hostWall.Width try: align = hostWall.OverrideAlign[masterSketchSubelementIndex] except: align = hostWall.Align else: print (" something wrong ?") if True: offsetValue = 0 if (masterSketchSubelementEdgeGroupWidth is not None) and (masterSketchSubelementEdgeGroupWidth.Value != 0): offsetValue = masterSketchSubelementEdgeGroupWidth.Value/2 # + attachmentAlignmentOffset.Value elif hostSketch.ArchSketchWidth != 0: offsetValue = hostSketch.ArchSketchWidth.Value / 2 if attachmentAlignment == "EdgeGroupWidthLeft": offsetValue = -offsetValue + attachmentAlignmentOffset.Value elif attachmentAlignment == "EdgeGroupWidthRight": offsetValue = offsetValue + attachmentAlignmentOffset.Value else: offsetValue = attachmentAlignmentOffset.Value if offsetValue is not 0: vOffsetH = DraftVecUtils.scaleTo(masterSketchSubelementEdgeVec.cross(Vector(0,0,1)), offsetValue) tempAttachmentOffset.Base = tempAttachmentOffset.Base.add(DraftVecUtils.scale(vOffsetH,1)) elif attachToSubelementOrOffset == "Follow Only Offset XYZ & Rotation": tempAttachmentOffset = attachmentOffsetXyzAndRotation if linkFp or hostWall: hostSketchPl = hostSketch.Placement if hostWall: hostWallPl = hostWall.Placement tempAttachmentOffset = (hostWallPl.multiply(hostSketchPl)).multiply(tempAttachmentOffset) else: tempAttachmentOffset = hostSketchPl.multiply(tempAttachmentOffset) print (" fp.Placement (superimposed) is thus ... ", tempAttachmentOffset) #if edgeAngle: # edgeAngleRotationPl = FreeCAD.Placement() # edgeAngleRotationPl.Rotation.Angle = edgeAngle # tempAttachmentOffset = tempAttachmentOffset.multiply(edgeAngleRotationPl) extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,0,0)) #, 0) if fp.AttachmentOffsetExtraRotation == "X-Axis CCW90": # [ "X-Axis CW90", "X-Axis CCW90", "X-Axis CW180", ] extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,0,90)) #, 0) elif fp.AttachmentOffsetExtraRotation == "X-Axis CW90": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,0,-90)) elif fp.AttachmentOffsetExtraRotation == "X-Axis CW180": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,0,180)) elif fp.AttachmentOffsetExtraRotation == "Y-Axis CW90": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,90,0)) elif fp.AttachmentOffsetExtraRotation == "Y-Axis CCW90": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,-90,0)) elif fp.AttachmentOffsetExtraRotation == "Y-Axis CW180": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(0,180,0)) elif fp.AttachmentOffsetExtraRotation == "Z-Axis CCW90": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(90,0,0)) elif fp.AttachmentOffsetExtraRotation == "Z-Axis CW90": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(-90,0,0)) elif fp.AttachmentOffsetExtraRotation == "Z-Axis CW180": extraRotation = FreeCAD.Placement(App.Vector(0,0,0),App.Rotation(180,0,0)) tempAttachmentOffset = tempAttachmentOffset.multiply(extraRotation) if linkFp or not hasattr(fp, "AttachmentOffset"): ## TODO or if hostWall ... fp.Placement = tempAttachmentOffset else: fp.AttachmentOffset = tempAttachmentOffset '''------------------- Creation/Insertion Functions ---------------------''' def makeArchSketch(grp=None, label="ArchSketch__NAME", attachToAxisOrSketch=None, placementAxis_Or_masterSketch=None, copyFlag=None, visibility=None): name = "ArchSketch" if grp: archSketch = grp.newObject("Sketcher::SketchObjectPython",name) else: archSketch=App.ActiveDocument.addObject("Sketcher::SketchObjectPython",name) archSketch.Label = label archSketchInsta=ArchSketch(archSketch) archSketch.AttachToAxisOrSketch = "Master Sketch" return archSketch '''------------------------- Low Level Operation --------------------------''' def getSketchEdgeAngle(masterSketch, subelement): vec = getSketchEdgeVec(masterSketch, subelement) draftAngle = -DraftVecUtils.angle(vec) return draftAngle def getSketchEdgeVec(sketch, subelement): geoindex = int(subelement.lstrip('Edge'))-1 lp1=sketch.Geometry[geoindex].EndPoint lp2=sketch.Geometry[geoindex].StartPoint vec = lp1 - lp2 return vec def getSketchEdgeOffsetPointVector(subject, masterSketch, subelement, attachmentOffset, zOffset, flipOffsetOriginToOtherEnd=False, flip180Degree=False, attachToSubelementOrOffset=None, offsetFromIntersectingSubelement=False, masterSketchIntersectingSubelement=None): geoindex = int(subelement.lstrip('Edge'))-1 if masterSketchIntersectingSubelement and offsetFromIntersectingSubelement: geoindex2 = int(masterSketchIntersectingSubelement.lstrip('Edge'))-1 childObjectWidth = 0 def getObjectWidth(): childObjectWidth = 0 try: if Draft.getType(subject.InList[0]) in ["Window"]: subjectWin = subject.InList[0] if subjectWin.Width == 0: subjectWin = subject.InList[0].CloneOf childObjectWidth = subjectWin.Shape.BoundBox.XLength else: childWindowWidthMM = subjectWin.Width childObjectWidth = float(childWindowWidthMM) except: try: subjectObject = subject.InList[0].CloneOf childObjectWidth = subjectObject.Shape.BoundBox.XLength except: try: subjectObject = subject.InList[0] childObjectWidth = subjectObject.Shape.BoundBox.XLength except: childObjectWidth = 0 return childObjectWidth if ((not flipOffsetOriginToOtherEnd) and (not offsetFromIntersectingSubelement)) or (flipOffsetOriginToOtherEnd and offsetFromIntersectingSubelement): if not flip180Degree: EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(float(attachmentOffset)) else: if attachToSubelementOrOffset == "Attach To Edge & Alignment": childWindowWidth = getObjectWidth() EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(float(attachmentOffset) + childWindowWidth) else: EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(float(attachmentOffset)) if offsetFromIntersectingSubelement: intersectVec = getSketchEdgeIntersection(masterSketch, geoindex, geoindex2) offsetVec = intersectVec.sub(masterSketch.Geometry[geoindex].StartPoint) EdgeOffsetPoint = EdgeOffsetPoint.add(offsetVec) else: edgeLength = masterSketch.Geometry[geoindex].length() if not flip180Degree: if attachToSubelementOrOffset == "Attach To Edge & Alignment": childWindowWidth = getObjectWidth() EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(edgeLength - float(attachmentOffset) - childWindowWidth) else: EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(edgeLength - float(attachmentOffset)) else: # directly use elif:? EdgeOffsetPoint = masterSketch.Geometry[geoindex].value(edgeLength - float(attachmentOffset)) if offsetFromIntersectingSubelement: intersectVec = getSketchEdgeIntersection(masterSketch, geoindex, geoindex2) offsetVec = intersectVec.sub(masterSketch.Geometry[geoindex].EndPoint) EdgeOffsetPoint = EdgeOffsetPoint.add(offsetVec) EdgeOffsetPoint.z = zOffset.Base.z return EdgeOffsetPoint def getSortedClustersEdgesOrder(sketch): ''' Do Part.getSortedClusters() on geometry of a Sketch, check the order of edges to return lists of indexes in in the order of sorted edges return: - clEdgePartnerIndex, clEdgeSameIndex, clEdgeEqualIndex, and - clEdgePartnerIndexFlat, clEdgeSameIndexFlat, clEdgeEqualIndexFlat ''' skGeom = sketch.Geometry skGeomEdges = [] skGeomEdgesShort = [] for i in skGeom: skGeomEdge = i.toShape() skGeomEdges.append(skGeomEdge) if not i.Construction: skGeomEdgesShort.append(skGeomEdge) skGeomEdgesSorted = Part.getSortedClusters(skGeomEdgesShort) ## a list of lists (not exactly array / matrix) to contain index of found matching geometry clEdgePartnerIndex = [] clEdgeSameIndex = [] clEdgeEqualIndex = [] ## a flat list containing above information - but just flat, not a list of lists .. clEdgePartnerIndexFlat = [] clEdgeSameIndexFlat = [] clEdgeEqualIndexFlat = [] for h, c in enumerate(skGeomEdgesSorted): clEdgePartnerIndex.append([]) clEdgeSameIndex.append([]) clEdgeEqualIndex.append([]) ''' Build the full sub-list ''' for a in c: clEdgePartnerIndex[h].append(None) clEdgeSameIndex[h].append(None) clEdgeEqualIndex[h].append(None) for i, skGeomEdgesSortedI in enumerate(c): for j, skGeomEdgesI in enumerate(skGeomEdges): if skGeomEdgesI: # is not None / i.e. Construction Geometry if j not in clEdgePartnerIndexFlat: if skGeomEdgesSortedI.isPartner(skGeomEdgesI): clEdgePartnerIndex[h][i] = j clEdgePartnerIndexFlat.append(j) if j not in clEdgeSameIndexFlat: if skGeomEdgesSortedI.isSame(skGeomEdgesI): clEdgeSameIndex[h][i] = j clEdgeSameIndexFlat.append(j) if j not in clEdgeEqualIndexFlat: if skGeomEdgesSortedI.isEqual(skGeomEdgesI): clEdgeEqualIndex[h][i] = j clEdgeEqualIndexFlat.append(j) if clEdgePartnerIndex[h][i] == None: clEdgePartnerIndexFlat.append(None) if clEdgeSameIndex[h][i] == None: clEdgeSameIndexFlat.append(None) if clEdgeEqualIndex[h][i] == None: clEdgeEqualIndexFlat.append(None) return clEdgePartnerIndex, clEdgeSameIndex, clEdgeEqualIndex, clEdgePartnerIndexFlat, clEdgeSameIndexFlat, clEdgeEqualIndexFlat def sortSketchAlign(sketch,edgeAlignList): ''' This function is primarily to support Ordinary Sketch + Arch Wall to gain feature that individual edge / wall segment to have individual Align setting with OverrideAlign attribute in Arch Wall This function arrange the edgeAlignList - a list of Align in the order of Edge Indexes - into a list of Align following the order of edges sorted by Part.getSortedClusters() ''' sortedIndexes = getSortedClustersEdgesOrder(sketch) clEdgeSameIndexFlat = sortedIndexes[4] alignsList = [] for i in clEdgeSameIndexFlat: try: curAlign = edgeAlignList[i] # if edgeAlignList does not cover the edge except: curAlign = 'Left' # default alignsList.append(curAlign) return alignsList def sortSketchWidth(sketch,edgeWidthList): ''' This function is primarily to support Ordinary Sketch + Arch Wall to gain feature that individual edge / wall segment to have individual Width setting with OverrideWidth attribute in Arch Wall This function arrange the edgeWidthList - a list of Width in the order of Edge Indexes - into a list of Width following the order of edges sorted by Part.getSortedClusters() ''' sortedIndexes = getSortedClustersEdgesOrder(sketch) clEdgeSameIndexFlat = sortedIndexes[4] widthList = [] for i in clEdgeSameIndexFlat: try: curWidth = edgeWidthList[i] # if edgeWidthList does not cover the edge except: curWidth = '200' # default widthList.append(curWidth) return widthList