#*************************************************************************** #* * #* Copyright (c) 2011 * #* Yorik van Havre * #* * #* This program is free software; you can redistribute it and/or modify * #* it under the terms of the GNU Lesser General Public License (LGPL) * #* as published by the Free Software Foundation; either version 2 of * #* the License, or (at your option) any later version. * #* for detail see the LICENCE text file. * #* * #* This program is distributed in the hope that it will be useful, * #* but WITHOUT ANY WARRANTY; without even the implied warranty of * #* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * #* GNU Library General Public License for more details. * #* * #* You should have received a copy of the GNU Library General Public * #* License along with this program; if not, write to the Free Software * #* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * #* USA * #* * #*************************************************************************** import FreeCAD,FreeCADGui,Draft,ArchComponent,DraftVecUtils from FreeCAD import Vector from PyQt4 import QtCore from DraftTools import translate __title__="FreeCAD Wall" __author__ = "Yorik van Havre" __url__ = "http://free-cad.sourceforge.net" def makeWall(baseobj=None,width=None,height=None,align="Center",name=str(translate("Arch","Wall"))): '''makeWall(obj,[width],[height],[align],[name]): creates a wall based on the given object, which can be a sketch, a draft object, a face or a solid. align can be "Center","Left" or "Right"''' obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython",name) _Wall(obj) _ViewProviderWall(obj.ViewObject) if baseobj: obj.Base = baseobj if width: obj.Width = width if height: obj.Height = height obj.Align = align if obj.Base: obj.Base.ViewObject.hide() p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch") c = p.GetUnsigned("WallColor") r = float((c>>24)&0xFF)/255.0 g = float((c>>16)&0xFF)/255.0 b = float((c>>8)&0xFF)/255.0 obj.ViewObject.ShapeColor = (r,g,b,1.0) return obj def joinWalls(walls): "joins the given list of walls into one sketch-based wall" if not walls: return None if not isinstance(walls,list): walls = [walls] if not areSameWallTypes(walls): return None base = walls.pop() if base.Base: if base.Base.Shape.Faces: return None if Draft.getType(base.Base) == "Sketch": sk = base.Base else: sk = Draft.makeSketch(base.Base,autoconstraints=True) if sk: base.Base = sk for w in walls: if w.Base: if not base.Base.Shape.Faces: for e in base.Base.Shape.Edges: sk.addGeometry(e) FreeCAD.ActiveDocument.recompute() return base def areSameWallTypes(walls): "returns True is all the walls in the given list have same height, width, and alignment" for att in ["Width","Height","Align"]: value = None for w in walls: if not hasattr(w,att): return False if not value: value = getattr(w,att) else: if type(value) == float: if round(value,Draft.precision()) != round(getattr(w,att),Draft.precision()): return False else: if value != getattr(w,att): return False return True class _CommandWall: "the Arch Wall command definition" def GetResources(self): return {'Pixmap' : 'Arch_Wall', 'MenuText': QtCore.QT_TRANSLATE_NOOP("Arch_Wall","Wall"), 'Accel': "W, A", 'ToolTip': QtCore.QT_TRANSLATE_NOOP("Arch_Wall","Creates a wall object from scratch or from a selected object (wire, face or solid)")} def Activated(self): global QtGui, QtCore from PyQt4 import QtGui, QtCore self.Width = 0.1 self.Height = 1 self.Align = "Center" self.continueCmd = False sel = FreeCADGui.Selection.getSelection() done = False self.existing = [] if sel: import Draft if Draft.getType(sel[0]) != "Wall": FreeCAD.ActiveDocument.openTransaction(str(translate("Arch","Create Wall"))) FreeCADGui.doCommand('import Arch') for obj in sel: FreeCADGui.doCommand('Arch.makeWall(FreeCAD.ActiveDocument.'+obj.Name+')') FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() done = True if not done: import DraftTrackers self.points = [] self.tracker = DraftTrackers.boxTracker() FreeCADGui.Snapper.getPoint(callback=self.getPoint,extradlg=self.taskbox()) def getPoint(self,point=None,obj=None): "this function is called by the snapper when it has a 3D point" if obj: if Draft.getType(obj) == "Wall": if not obj in self.existing: self.existing.append(obj) if point == None: self.tracker.finalize() return self.points.append(point) if len(self.points) == 1: self.tracker.on() FreeCADGui.Snapper.getPoint(last=self.points[0],callback=self.getPoint,movecallback=self.update,extradlg=self.taskbox()) elif len(self.points) == 2: import Part add = False l = Part.Line(self.points[0],self.points[1]) self.tracker.finalize() FreeCAD.ActiveDocument.openTransaction(str(translate("Arch","Create Wall"))) FreeCADGui.doCommand('import Arch') FreeCADGui.doCommand('import Part') FreeCADGui.doCommand('trace=Part.Line(FreeCAD.'+str(l.StartPoint)+',FreeCAD.'+str(l.EndPoint)+')') if not self.existing: self.addDefault(l) else: w = joinWalls(self.existing) if w: if areSameWallTypes([w,self]): FreeCADGui.doCommand('FreeCAD.ActiveDocument.'+w.Name+'.Base.addGeometry(trace)') else: self.addDefault(l) add = True else: self.addDefault(l) if add: FreeCADGui.doCommand('Arch.addComponents(FreeCAD.ActiveDocument.'+FreeCAD.ActiveDocument.Objects[-1].Name+',FreeCAD.ActiveDocument.'+w.Name+')') FreeCAD.ActiveDocument.commitTransaction() FreeCAD.ActiveDocument.recompute() if self.continueCmd: self.Activated() def addDefault(self,l): FreeCADGui.doCommand('base=FreeCAD.ActiveDocument.addObject("Sketcher::SketchObject","'+str(translate('Arch','WallTrace'))+'")') FreeCADGui.doCommand('base.addGeometry(trace)') FreeCADGui.doCommand('Arch.makeWall(base,width='+str(self.Width)+',height='+str(self.Height)+',align="'+str(self.Align)+'")') def update(self,point): "this function is called by the Snapper when the mouse is moved" b = self.points[0] n = FreeCAD.DraftWorkingPlane.axis bv = point.sub(b) dv = bv.cross(n) dv = DraftVecUtils.scaleTo(dv,self.Width/2) if self.Align == "Center": self.tracker.update([b,point]) elif self.Align == "Left": self.tracker.update([b.add(dv),point.add(dv)]) else: dv = DraftVecUtils.neg(dv) self.tracker.update([b.add(dv),point.add(dv)]) def taskbox(self): "sets up a taskbox widget" w = QtGui.QWidget() w.setWindowTitle(str(translate("Arch","Wall options"))) lay0 = QtGui.QVBoxLayout(w) lay1 = QtGui.QHBoxLayout() lay0.addLayout(lay1) label1 = QtGui.QLabel(str(translate("Arch","Width"))) lay1.addWidget(label1) value1 = QtGui.QDoubleSpinBox() value1.setDecimals(2) value1.setValue(self.Width) lay1.addWidget(value1) lay2 = QtGui.QHBoxLayout() lay0.addLayout(lay2) label2 = QtGui.QLabel(str(translate("Arch","Height"))) lay2.addWidget(label2) value2 = QtGui.QDoubleSpinBox() value2.setDecimals(2) value2.setValue(self.Height) lay2.addWidget(value2) lay3 = QtGui.QHBoxLayout() lay0.addLayout(lay3) label3 = QtGui.QLabel(str(translate("Arch","Alignment"))) lay3.addWidget(label3) value3 = QtGui.QComboBox() items = ["Center","Left","Right"] value3.addItems(items) value3.setCurrentIndex(items.index(self.Align)) lay3.addWidget(value3) value4 = QtGui.QCheckBox(str(translate("Arch","Continue"))) lay0.addWidget(value4) QtCore.QObject.connect(value1,QtCore.SIGNAL("valueChanged(double)"),self.setWidth) QtCore.QObject.connect(value2,QtCore.SIGNAL("valueChanged(double)"),self.setHeight) QtCore.QObject.connect(value3,QtCore.SIGNAL("currentIndexChanged(int)"),self.setAlign) QtCore.QObject.connect(value4,QtCore.SIGNAL("stateChanged(int)"),self.setContinue) return w def setWidth(self,d): self.Width = d self.tracker.width(d) def setHeight(self,d): self.Height = d self.tracker.height(d) def setAlign(self,i): self.Align = ["Center","Left","Right"][i] def setContinue(self,i): self.continueCmd = bool(i) class _Wall(ArchComponent.Component): "The Wall object" def __init__(self,obj): ArchComponent.Component.__init__(self,obj) obj.addProperty("App::PropertyLength","Width","Base", str(translate("Arch","The width of this wall. Not used if this wall is based on a face"))) obj.addProperty("App::PropertyLength","Height","Base", str(translate("Arch","The height of this wall. Keep 0 for automatic. Not used if this wall is based on a solid"))) obj.addProperty("App::PropertyEnumeration","Align","Base", str(translate("Arch","The alignment of this wall on its base object, if applicable"))) obj.addProperty("App::PropertyVector","Normal","Base", str(translate("Arch","The normal extrusion direction of this object (keep (0,0,0) for automatic normal)"))) obj.addProperty("App::PropertyBool","ForceWire","Base", str(translate("Arch","If True, if this wall is based on a face, it will use its border wire as trace, and disconsider the face."))) obj.Align = ['Left','Right','Center'] obj.ForceWire = False self.Type = "Wall" obj.Width = 0.1 obj.Height = 0 def execute(self,obj): self.createGeometry(obj) def onChanged(self,obj,prop): if prop in ["Base","Height","Width","Align","Additions","Subtractions"]: self.createGeometry(obj) def getSubVolume(self,base,width,delta=None): "returns a subvolume from a base object" import Part max_length = 0 f = None for w in base.Shape.Wires: if w.BoundBox.DiagonalLength > max_length: max_length = w.BoundBox.DiagonalLength f = w if f: f = Part.Face(f) n = f.normalAt(0,0) v1 = DraftVecUtils.scaleTo(n,width) f.translate(v1) v2 = DraftVecUtils.neg(v1) v2 = DraftVecUtils.scale(v1,-2) f = f.extrude(v2) if delta: f.translate(delta) return f return None def createGeometry(self,obj): "builds the wall shape" if not obj.Base: return import Part, DraftGeomUtils flat = False if hasattr(obj.ViewObject,"DisplayMode"): flat = (obj.ViewObject.DisplayMode == "Flat 2D") width = 1.0 if hasattr(obj,"Width"): if obj.Width: width = obj.Width def getbase(wire): "returns a full shape from a base wire" dvec = DraftGeomUtils.vec(wire.Edges[0]).cross(normal) dvec.normalize() if obj.Align == "Left": dvec = dvec.multiply(width) w2 = DraftGeomUtils.offsetWire(wire,dvec) w1 = Part.Wire(DraftGeomUtils.sortEdges(wire.Edges)) sh = DraftGeomUtils.bind(w1,w2) elif obj.Align == "Right": dvec = dvec.multiply(width) dvec = DraftVecUtils.neg(dvec) w2 = DraftGeomUtils.offsetWire(wire,dvec) w1 = Part.Wire(DraftGeomUtils.sortEdges(wire.Edges)) sh = DraftGeomUtils.bind(w1,w2) elif obj.Align == "Center": dvec = dvec.multiply(width/2) w1 = DraftGeomUtils.offsetWire(wire,dvec) dvec = DraftVecUtils.neg(dvec) w2 = DraftGeomUtils.offsetWire(wire,dvec) sh = DraftGeomUtils.bind(w1,w2) # fixing self-intersections sh.fix(0.1,0,1) if height and (not flat): norm = Vector(normal).multiply(height) sh = sh.extrude(norm) return sh pl = obj.Placement # getting default values height = normal = None if obj.Height: height = obj.Height else: for p in obj.InList: if Draft.getType(p) == "Floor": height = p.Height if not height: height = 1 if obj.Normal == Vector(0,0,0): normal = Vector(0,0,1) else: normal = Vector(obj.Normal) # computing shape base = None if obj.Base.isDerivedFrom("Part::Feature"): if not obj.Base.Shape.isNull(): base = obj.Base.Shape.copy() if base.Solids: pass elif base.Faces and (not obj.ForceWire): if height: norm = normal.multiply(height) base = base.extrude(norm) elif base.Wires: temp = None for wire in obj.Base.Shape.Wires: sh = getbase(wire) if temp: temp = temp.fuse(sh) else: temp = sh base = temp elif base.Edges: wire = Part.Wire(base.Edges) sh = getbase(wire) if sh: base = sh else: FreeCAD.Console.PrintError(str(translate("Arch","Error: Invalid base object"))) if base: for app in obj.Additions: if hasattr(app,"Shape"): if app.Shape: if not app.Shape.isNull(): base = base.fuse(app.Shape) app.ViewObject.hide() #to be removed for hole in obj.Subtractions: if Draft.getType(hole) == "Window": # window if hole.Base and obj.Width: f = self.getSubVolume(hole.Base,width) if f: base = base.cut(f) elif Draft.isClone(hole,"Window"): if hole.Objects[0].Base and width: f = self.getSubVolume(hole.Objects[0].Base,width,hole.Placement.Base) if f: base = base.cut(f) elif hasattr(hole,"Shape"): if hole.Shape: if not hole.Shape.isNull(): base = base.cut(hole.Shape) hole.ViewObject.hide() # to be removed if not base.isNull(): try: base = base.removeSplitter() except: FreeCAD.Console.PrintError(str(translate("Arch","Error removing splitter from wall shape"))) obj.Shape = base if not DraftGeomUtils.isNull(pl): obj.Placement = pl class _ViewProviderWall(ArchComponent.ViewProviderComponent): "A View Provider for the Wall object" def __init__(self,vobj): ArchComponent.ViewProviderComponent.__init__(self,vobj) def getIcon(self): return ":/icons/Arch_Wall_Tree.svg" def getDisplayModes(self,vobj): return ["Flat 2D"] def setDisplayMode(self,mode): self.Object.Proxy.createGeometry(self.Object) if mode == "Flat 2D": return "Flat Lines" else: return mode def attach(self,vobj): self.Object = vobj.Object return FreeCADGui.addCommand('Arch_Wall',_CommandWall())