1023 lines
46 KiB
Python
1023 lines
46 KiB
Python
#***************************************************************************
|
|
#* *
|
|
#* Copyright (c) 2011 *
|
|
#* Yorik van Havre <yorik@uncreated.net> *
|
|
#* *
|
|
#* 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,Draft,ArchComponent,DraftVecUtils,ArchCommands,math
|
|
from FreeCAD import Vector
|
|
if FreeCAD.GuiUp:
|
|
import FreeCADGui
|
|
from PySide import QtCore, QtGui
|
|
from DraftTools import translate
|
|
from PySide.QtCore import QT_TRANSLATE_NOOP
|
|
else:
|
|
# \cond
|
|
def translate(ctxt,txt, utf8_decode=False):
|
|
return txt
|
|
def QT_TRANSLATE_NOOP(ctxt,txt):
|
|
return txt
|
|
# \endcond
|
|
|
|
## @package ArchWall
|
|
# \ingroup ARCH
|
|
# \brief The Wall object and tools
|
|
#
|
|
# This module provides tools to build Wall objects.
|
|
# Walls are simple objects, usually vertical, obtained
|
|
# by giving a thickness to a base line, then extruding it
|
|
# vertically.
|
|
|
|
__title__="FreeCAD Wall"
|
|
__author__ = "Yorik van Havre"
|
|
__url__ = "http://www.freecadweb.org"
|
|
|
|
|
|
|
|
def makeWall(baseobj=None,length=None,width=None,height=None,align="Center",face=None,name="Wall"):
|
|
|
|
'''makeWall([obj],[length],[width],[height],[align],[face],[name]): creates a wall based on the
|
|
given object, which can be a sketch, a draft object, a face or a solid, or no object at
|
|
all, then you must provide length, width and height. Align can be "Center","Left" or "Right",
|
|
face can be an index number of a face in the base object to base the wall on.'''
|
|
|
|
if not FreeCAD.ActiveDocument:
|
|
FreeCAD.Console.PrintError("No active document. Aborting\n")
|
|
return
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch")
|
|
obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython","Wall")
|
|
obj.Label = translate("Arch",name)
|
|
_Wall(obj)
|
|
if FreeCAD.GuiUp:
|
|
_ViewProviderWall(obj.ViewObject)
|
|
if baseobj:
|
|
if baseobj.isDerivedFrom("Part::Feature") or baseobj.isDerivedFrom("Mesh::Feature"):
|
|
obj.Base = baseobj
|
|
else:
|
|
FreeCAD.Console.PrintWarning(str(translate("Arch","Walls can only be based on Part or Mesh objects")))
|
|
if face:
|
|
obj.Face = face
|
|
if length:
|
|
obj.Length = length
|
|
if width:
|
|
obj.Width = width
|
|
else:
|
|
obj.Width = p.GetFloat("WallWidth",200)
|
|
if height:
|
|
obj.Height = height
|
|
else:
|
|
obj.Height = p.GetFloat("WallHeight",3000)
|
|
obj.Align = align
|
|
if obj.Base and FreeCAD.GuiUp:
|
|
if Draft.getType(obj.Base) != "Space":
|
|
obj.Base.ViewObject.hide()
|
|
return obj
|
|
|
|
def joinWalls(walls,delete=False):
|
|
|
|
"""joins the given list of walls into one sketch-based wall. If delete
|
|
is True, merged wall objects are deleted"""
|
|
|
|
if not walls:
|
|
return None
|
|
if not isinstance(walls,list):
|
|
walls = [walls]
|
|
if not areSameWallTypes(walls):
|
|
return None
|
|
deleteList = []
|
|
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 w.Base.Shape.Faces:
|
|
for e in w.Base.Shape.Edges:
|
|
sk.addGeometry(e.Curve)
|
|
deleteList.append(w.Name)
|
|
if delete:
|
|
for n in deleteList:
|
|
FreeCAD.ActiveDocument.removeObject(n)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
base.ViewObject.show()
|
|
return base
|
|
|
|
def mergeShapes(w1,w2):
|
|
|
|
"returns a Shape built on two walls that share same properties and have a coincident endpoint"
|
|
|
|
if not areSameWallTypes([w1,w2]):
|
|
return None
|
|
if (not hasattr(w1.Base,"Shape")) or (not hasattr(w2.Base,"Shape")):
|
|
return None
|
|
if w1.Base.Shape.Faces or w2.Base.Shape.Faces:
|
|
return None
|
|
|
|
# TODO fix this
|
|
return None
|
|
|
|
eds = w1.Base.Shape.Edges + w2.Base.Shape.Edges
|
|
import DraftGeomUtils
|
|
w = DraftGeomUtils.findWires(eds)
|
|
if len(w) == 1:
|
|
#print("found common wire")
|
|
normal,length,width,height = w1.Proxy.getDefaultValues(w1)
|
|
print(w[0].Edges)
|
|
sh = w1.Proxy.getBase(w1,w[0],normal,width,height)
|
|
print(sh)
|
|
return sh
|
|
return None
|
|
|
|
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': QT_TRANSLATE_NOOP("Arch_Wall","Wall"),
|
|
'Accel': "W, A",
|
|
'ToolTip': QT_TRANSLATE_NOOP("Arch_Wall","Creates a wall object from scratch or from a selected object (wire, face or solid)")}
|
|
|
|
def IsActive(self):
|
|
|
|
return not FreeCAD.ActiveDocument is None
|
|
|
|
def Activated(self):
|
|
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch")
|
|
self.Align = ["Center","Left","Right"][p.GetInt("WallAlignment",0)]
|
|
self.MultiMat = None
|
|
self.Length = None
|
|
self.lengthValue = 0
|
|
self.continueCmd = False
|
|
self.Width = p.GetFloat("WallWidth",200)
|
|
self.Height = p.GetFloat("WallHeight",3000)
|
|
self.JOIN_WALLS_SKETCHES = p.GetBool("joinWallSketches",False)
|
|
self.AUTOJOIN = p.GetBool("autoJoinWalls",True)
|
|
sel = FreeCADGui.Selection.getSelectionEx()
|
|
done = False
|
|
self.existing = []
|
|
|
|
if sel:
|
|
# automatic mode
|
|
import Draft
|
|
if Draft.getType(sel[0].Object) != "Wall":
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Wall"))
|
|
FreeCADGui.addModule("Arch")
|
|
for selobj in sel:
|
|
if Draft.getType(selobj.Object) == "Space":
|
|
spacedone = False
|
|
if selobj.HasSubObjects:
|
|
if "Face" in selobj.SubElementNames[0]:
|
|
idx = int(selobj.SubElementNames[0][4:])
|
|
FreeCADGui.doCommand("obj = Arch.makeWall(FreeCAD.ActiveDocument."+selobj.Object.Name+",face="+str(idx)+")")
|
|
spacedone = True
|
|
if not spacedone:
|
|
FreeCADGui.doCommand('obj = Arch.makeWall(FreeCAD.ActiveDocument.'+selobj.Object.Name+')')
|
|
else:
|
|
FreeCADGui.doCommand('obj = Arch.makeWall(FreeCAD.ActiveDocument.'+selobj.Object.Name+')')
|
|
FreeCADGui.addModule("Draft")
|
|
FreeCADGui.doCommand("Draft.autogroup(obj)")
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
done = True
|
|
|
|
if not done:
|
|
# interactive mode
|
|
import DraftTrackers
|
|
self.points = []
|
|
self.tracker = DraftTrackers.boxTracker()
|
|
if hasattr(FreeCAD,"DraftWorkingPlane"):
|
|
FreeCAD.DraftWorkingPlane.setup()
|
|
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.width(self.Width)
|
|
self.tracker.height(self.Height)
|
|
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
|
|
l = Part.LineSegment(FreeCAD.DraftWorkingPlane.getLocalCoords(self.points[0]),FreeCAD.DraftWorkingPlane.getLocalCoords(self.points[1]))
|
|
self.tracker.finalize()
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Wall"))
|
|
FreeCADGui.addModule("Arch")
|
|
FreeCADGui.doCommand('import Part')
|
|
FreeCADGui.doCommand('trace=Part.LineSegment(FreeCAD.'+str(l.StartPoint)+',FreeCAD.'+str(l.EndPoint)+')')
|
|
if not self.existing:
|
|
# no existing wall snapped, just add a default wall
|
|
self.addDefault(l)
|
|
else:
|
|
if self.JOIN_WALLS_SKETCHES:
|
|
# join existing subwalls first if possible, then add the new one
|
|
w = joinWalls(self.existing)
|
|
if w:
|
|
if areSameWallTypes([w,self]):
|
|
FreeCADGui.doCommand('FreeCAD.ActiveDocument.'+w.Name+'.Base.addGeometry(trace)')
|
|
else:
|
|
# if not possible, add new wall as addition to the existing one
|
|
self.addDefault(l)
|
|
if self.AUTOJOIN:
|
|
FreeCADGui.doCommand('Arch.addComponents(FreeCAD.ActiveDocument.'+FreeCAD.ActiveDocument.Objects[-1].Name+',FreeCAD.ActiveDocument.'+w.Name+')')
|
|
else:
|
|
self.addDefault(l)
|
|
else:
|
|
# add new wall as addition to the first existing one
|
|
self.addDefault(l)
|
|
if self.AUTOJOIN:
|
|
FreeCADGui.doCommand('Arch.addComponents(FreeCAD.ActiveDocument.'+FreeCAD.ActiveDocument.Objects[-1].Name+',FreeCAD.ActiveDocument.'+self.existing[0].Name+')')
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
if self.continueCmd:
|
|
self.Activated()
|
|
|
|
def addDefault(self,l):
|
|
|
|
FreeCADGui.addModule("Draft")
|
|
if FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").GetBool("WallSketches",True):
|
|
FreeCADGui.doCommand('base=FreeCAD.ActiveDocument.addObject("Sketcher::SketchObject","WallTrace")')
|
|
FreeCADGui.doCommand('base.Placement = FreeCAD.DraftWorkingPlane.getPlacement()')
|
|
FreeCADGui.doCommand('base.addGeometry(trace)')
|
|
else:
|
|
FreeCADGui.doCommand('base=Draft.makeLine(trace)')
|
|
FreeCADGui.doCommand('wall = Arch.makeWall(base,width='+str(self.Width)+',height='+str(self.Height)+',align="'+str(self.Align)+'")')
|
|
FreeCADGui.doCommand('wall.Normal = FreeCAD.DraftWorkingPlane.getNormal()')
|
|
if self.MultiMat:
|
|
FreeCADGui.doCommand("wall.Material = FreeCAD.ActiveDocument."+self.MultiMat.Name)
|
|
FreeCADGui.doCommand("Draft.autogroup(wall)")
|
|
|
|
def update(self,point,info):
|
|
|
|
"this function is called by the Snapper when the mouse is moved"
|
|
|
|
if FreeCADGui.Control.activeDialog():
|
|
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 = dv.negative()
|
|
self.tracker.update([b.add(dv),point.add(dv)])
|
|
if self.Length:
|
|
self.Length.setText(FreeCAD.Units.Quantity(bv.Length,FreeCAD.Units.Length).UserString)
|
|
|
|
def taskbox(self):
|
|
|
|
"sets up a taskbox widget"
|
|
|
|
w = QtGui.QWidget()
|
|
ui = FreeCADGui.UiLoader()
|
|
w.setWindowTitle(translate("Arch","Wall options", utf8_decode=True))
|
|
grid = QtGui.QGridLayout(w)
|
|
|
|
matCombo = QtGui.QComboBox()
|
|
matCombo.addItem(translate("Arch","Wall Presets..."))
|
|
self.multimats = []
|
|
self.MultiMat = None
|
|
for o in FreeCAD.ActiveDocument.Objects:
|
|
if Draft.getType(o) == "MultiMaterial":
|
|
self.multimats.append(o)
|
|
matCombo.addItem(o.Label)
|
|
if hasattr(FreeCAD,"LastArchMultiMaterial"):
|
|
for i,o in enumerate(self.multimats):
|
|
if o.Name == FreeCAD.LastArchMultiMaterial:
|
|
matCombo.setCurrentIndex(i+1)
|
|
self.MultiMat = o
|
|
grid.addWidget(matCombo,0,0,1,2)
|
|
|
|
label5 = QtGui.QLabel(translate("Arch","Length", utf8_decode=True))
|
|
self.Length = ui.createWidget("Gui::InputField")
|
|
self.Length.setText("0.00 mm")
|
|
grid.addWidget(label5,1,0,1,1)
|
|
grid.addWidget(self.Length,1,1,1,1)
|
|
|
|
label1 = QtGui.QLabel(translate("Arch","Width", utf8_decode=True))
|
|
value1 = ui.createWidget("Gui::InputField")
|
|
value1.setText(FreeCAD.Units.Quantity(self.Width,FreeCAD.Units.Length).UserString)
|
|
grid.addWidget(label1,2,0,1,1)
|
|
grid.addWidget(value1,2,1,1,1)
|
|
|
|
label2 = QtGui.QLabel(translate("Arch","Height", utf8_decode=True))
|
|
value2 = ui.createWidget("Gui::InputField")
|
|
value2.setText(FreeCAD.Units.Quantity(self.Height,FreeCAD.Units.Length).UserString)
|
|
grid.addWidget(label2,3,0,1,1)
|
|
grid.addWidget(value2,3,1,1,1)
|
|
|
|
label3 = QtGui.QLabel(translate("Arch","Alignment", utf8_decode=True))
|
|
value3 = QtGui.QComboBox()
|
|
items = ["Center","Left","Right"]
|
|
value3.addItems(items)
|
|
value3.setCurrentIndex(items.index(self.Align))
|
|
grid.addWidget(label3,4,0,1,1)
|
|
grid.addWidget(value3,4,1,1,1)
|
|
|
|
label4 = QtGui.QLabel(translate("Arch","Con&tinue", utf8_decode=True))
|
|
value4 = QtGui.QCheckBox()
|
|
value4.setObjectName("ContinueCmd")
|
|
value4.setLayoutDirection(QtCore.Qt.RightToLeft)
|
|
label4.setBuddy(value4)
|
|
if hasattr(FreeCADGui,"draftToolBar"):
|
|
value4.setChecked(FreeCADGui.draftToolBar.continueMode)
|
|
self.continueCmd = FreeCADGui.draftToolBar.continueMode
|
|
grid.addWidget(label4,5,0,1,1)
|
|
grid.addWidget(value4,5,1,1,1)
|
|
|
|
QtCore.QObject.connect(self.Length,QtCore.SIGNAL("valueChanged(double)"),self.setLength)
|
|
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)
|
|
QtCore.QObject.connect(self.Length,QtCore.SIGNAL("returnPressed()"),value1.setFocus)
|
|
QtCore.QObject.connect(self.Length,QtCore.SIGNAL("returnPressed()"),value1.selectAll)
|
|
QtCore.QObject.connect(value1,QtCore.SIGNAL("returnPressed()"),value2.setFocus)
|
|
QtCore.QObject.connect(value1,QtCore.SIGNAL("returnPressed()"),value2.selectAll)
|
|
QtCore.QObject.connect(value2,QtCore.SIGNAL("returnPressed()"),self.createFromGUI)
|
|
QtCore.QObject.connect(matCombo,QtCore.SIGNAL("currentIndexChanged(int)"),self.setMat)
|
|
return w
|
|
|
|
def setMat(self,d):
|
|
|
|
if d == 0:
|
|
self.MultiMat = None
|
|
del FreeCAD.LastArchMultiMaterial
|
|
elif d <= len(self.multimats):
|
|
self.MultiMat = self.multimats[d-1]
|
|
FreeCAD.LastArchMultiMaterial = self.MultiMat.Name
|
|
|
|
def setLength(self,d):
|
|
|
|
self.lengthValue = d
|
|
|
|
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]
|
|
FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").SetInt("WallAlignment",i)
|
|
|
|
def setContinue(self,i):
|
|
|
|
self.continueCmd = bool(i)
|
|
if hasattr(FreeCADGui,"draftToolBar"):
|
|
FreeCADGui.draftToolBar.continueMode = bool(i)
|
|
|
|
def createFromGUI(self):
|
|
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Wall"))
|
|
FreeCADGui.addModule("Arch")
|
|
FreeCADGui.doCommand('wall = Arch.makeWall(length='+str(self.lengthValue)+',width='+str(self.Width)+',height='+str(self.Height)+',align="'+str(self.Align)+'")')
|
|
if self.MultiMat:
|
|
FreeCADGui.doCommand("wall.Material = FreeCAD.ActiveDocument."+self.MultiMat.Name)
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
if hasattr(FreeCADGui,"draftToolBar"):
|
|
FreeCADGui.draftToolBar.escape()
|
|
|
|
|
|
|
|
class _CommandMergeWalls:
|
|
|
|
"the Arch Merge Walls command definition"
|
|
|
|
def GetResources(self):
|
|
|
|
return {'Pixmap' : 'Arch_MergeWalls',
|
|
'MenuText': QT_TRANSLATE_NOOP("Arch_MergeWalls","Merge Walls"),
|
|
'ToolTip': QT_TRANSLATE_NOOP("Arch_MergeWalls","Merges the selected walls, if possible")}
|
|
|
|
def IsActive(self):
|
|
|
|
return bool(FreeCADGui.Selection.getSelection())
|
|
|
|
def Activated(self):
|
|
|
|
walls = FreeCADGui.Selection.getSelection()
|
|
if len(walls) == 1:
|
|
if Draft.getType(walls[0]) == "Wall":
|
|
ostr = "FreeCAD.ActiveDocument."+ walls[0].Name
|
|
ok = False
|
|
for o in walls[0].Additions:
|
|
if Draft.getType(o) == "Wall":
|
|
ostr += ",FreeCAD.ActiveDocument." + o.Name
|
|
ok = True
|
|
if ok:
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Merge Wall"))
|
|
FreeCADGui.addModule("Arch")
|
|
FreeCADGui.doCommand("Arch.joinWalls(["+ostr+"],delete=True)")
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
return
|
|
else:
|
|
FreeCAD.Console.PrintWarning(translate("Arch","The selected wall contains no subwall to merge"))
|
|
return
|
|
else:
|
|
FreeCAD.Console.PrintWarning(translate("Arch","Please select only wall objects"))
|
|
return
|
|
for w in walls:
|
|
if Draft.getType(w) != "Wall":
|
|
FreeCAD.Console.PrintMessage(translate("Arch","Please select only wall objects"))
|
|
return
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Merge Walls"))
|
|
FreeCADGui.addModule("Arch")
|
|
FreeCADGui.doCommand("Arch.joinWalls(FreeCADGui.Selection.getSelection(),delete=True)")
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
|
|
|
|
|
|
class _Wall(ArchComponent.Component):
|
|
|
|
"The Wall object"
|
|
|
|
def __init__(self,obj):
|
|
|
|
ArchComponent.Component.__init__(self,obj)
|
|
self.setProperties(obj)
|
|
obj.IfcRole = "Wall"
|
|
|
|
def setProperties(self,obj):
|
|
|
|
lp = obj.PropertiesList
|
|
if not "Length" in lp:
|
|
obj.addProperty("App::PropertyLength","Length","Wall",QT_TRANSLATE_NOOP("App::Property","The length of this wall. Not used if this wall is based on an underlying object"))
|
|
if not "Width" in lp:
|
|
obj.addProperty("App::PropertyLength","Width","Wall",QT_TRANSLATE_NOOP("App::Property","The width of this wall. Not used if this wall is based on a face"))
|
|
if not "Height" in lp:
|
|
obj.addProperty("App::PropertyLength","Height","Wall",QT_TRANSLATE_NOOP("App::Property","The height of this wall. Keep 0 for automatic. Not used if this wall is based on a solid"))
|
|
if not "Align" in lp:
|
|
obj.addProperty("App::PropertyEnumeration","Align","Wall",QT_TRANSLATE_NOOP("App::Property","The alignment of this wall on its base object, if applicable"))
|
|
obj.Align = ['Left','Right','Center']
|
|
if not "Normal" in lp:
|
|
obj.addProperty("App::PropertyVector","Normal","Wall",QT_TRANSLATE_NOOP("App::Property","The normal extrusion direction of this object (keep (0,0,0) for automatic normal)"))
|
|
if not "Face" in lp:
|
|
obj.addProperty("App::PropertyInteger","Face","Wall",QT_TRANSLATE_NOOP("App::Property","The face number of the base object used to build this wall"))
|
|
if not "Offset" in lp:
|
|
obj.addProperty("App::PropertyDistance","Offset","Wall",QT_TRANSLATE_NOOP("App::Property","The offset between this wall and its baseline (only for left and right alignments)"))
|
|
|
|
if not "MakeBlocks" in lp:
|
|
obj.addProperty("App::PropertyBool","MakeBlocks","Blocks",QT_TRANSLATE_NOOP("App::Property","Enable this to make the wall generate blocks"))
|
|
if not "BlockLength" in lp:
|
|
obj.addProperty("App::PropertyLength","BlockLength","Blocks",QT_TRANSLATE_NOOP("App::Property","The length of each block"))
|
|
if not "BlockHeight" in lp:
|
|
obj.addProperty("App::PropertyLength","BlockHeight","Blocks",QT_TRANSLATE_NOOP("App::Property","The height of each block"))
|
|
if not "OffsetFirst" in lp:
|
|
obj.addProperty("App::PropertyLength","OffsetFirst","Blocks",QT_TRANSLATE_NOOP("App::Property","The horizontal offset of the first line of blocks"))
|
|
if not "OffsetSecond" in lp:
|
|
obj.addProperty("App::PropertyLength","OffsetSecond","Blocks",QT_TRANSLATE_NOOP("App::Property","The horizontal offset of the second line of blocks"))
|
|
if not "Joint" in lp:
|
|
obj.addProperty("App::PropertyLength","Joint","Blocks",QT_TRANSLATE_NOOP("App::Property","The size of the joints between each block"))
|
|
if not "CountEntire" in lp:
|
|
obj.addProperty("App::PropertyInteger","CountEntire","Blocks",QT_TRANSLATE_NOOP("App::Property","The number of entire blocks"))
|
|
obj.setEditorMode("CountEntire",1)
|
|
if not "CountBroken" in lp:
|
|
obj.addProperty("App::PropertyInteger","CountBroken","Blocks",QT_TRANSLATE_NOOP("App::Property","The number of broken blocks"))
|
|
obj.setEditorMode("CountBroken",1)
|
|
self.Type = "Wall"
|
|
|
|
def onDocumentRestored(self,obj):
|
|
|
|
ArchComponent.Component.onDocumentRestored(self,obj)
|
|
self.setProperties(obj)
|
|
|
|
def execute(self,obj):
|
|
|
|
"builds the wall shape"
|
|
|
|
if self.clone(obj):
|
|
return
|
|
|
|
import Part, DraftGeomUtils
|
|
base = None
|
|
pl = obj.Placement
|
|
extdata = self.getExtrusionData(obj)
|
|
if extdata:
|
|
bplates = extdata[0]
|
|
extv = extdata[2].Rotation.multVec(extdata[1])
|
|
if isinstance(bplates,list):
|
|
shps = []
|
|
for b in bplates:
|
|
b.Placement = extdata[2].multiply(b.Placement)
|
|
b = b.extrude(extv)
|
|
shps.append(b)
|
|
base = Part.makeCompound(shps)
|
|
else:
|
|
bplates.Placement = extdata[2].multiply(bplates.Placement)
|
|
base = bplates.extrude(extv)
|
|
if obj.Base:
|
|
if obj.Base.isDerivedFrom("Part::Feature"):
|
|
if obj.Base.Shape.isNull():
|
|
return
|
|
if not obj.Base.Shape.isValid():
|
|
if not obj.Base.Shape.Solids:
|
|
# let pass invalid objects if they have solids...
|
|
return
|
|
elif obj.Base.Shape.Solids:
|
|
base = obj.Base.Shape.copy()
|
|
|
|
# blocks calculation
|
|
elif hasattr(obj,"MakeBlocks") and hasattr(self,"basewires"):
|
|
if obj.MakeBlocks and self.basewires and extdata and obj.Width and obj.BlockLength.Value and obj.BlockHeight.Value:
|
|
#print "calculating blocks"
|
|
if len(self.basewires) == 1:
|
|
blocks = []
|
|
n = FreeCAD.Vector(extv)
|
|
n.normalize()
|
|
cuts1 = []
|
|
cuts2 = []
|
|
for i in range(2):
|
|
if i == 0:
|
|
offset = obj.OffsetFirst.Value
|
|
else:
|
|
offset = obj.OffsetSecond.Value
|
|
for edge in self.basewires[0].Edges:
|
|
while offset < (edge.Length-obj.Joint.Value):
|
|
#print i," Edge ",edge," : ",edge.Length," - ",offset
|
|
if offset:
|
|
t = edge.tangentAt(offset)
|
|
p = t.cross(n)
|
|
p.multiply(1.1*obj.Width.Value)
|
|
p1 = edge.valueAt(offset).add(p)
|
|
p2 = edge.valueAt(offset).add(p.negative())
|
|
sh = Part.LineSegment(p1,p2).toShape()
|
|
if obj.Joint.Value:
|
|
sh = sh.extrude(t.multiply(obj.Joint.Value))
|
|
sh = sh.extrude(n)
|
|
if i == 0:
|
|
cuts1.append(sh)
|
|
else:
|
|
cuts2.append(sh)
|
|
offset += (obj.BlockLength.Value+obj.Joint.Value)
|
|
else:
|
|
offset -= (edge.Length-obj.Joint.Value)
|
|
if cuts1 and cuts2:
|
|
if isinstance(bplates,list):
|
|
bplates = bplates[0]
|
|
fsize = obj.BlockHeight.Value+obj.Joint.Value
|
|
bvec = FreeCAD.Vector(n)
|
|
bvec.multiply(obj.BlockHeight.Value)
|
|
svec = FreeCAD.Vector(n)
|
|
svec.multiply(fsize)
|
|
plate1 = bplates.cut(cuts1).Faces
|
|
blocks1 = Part.makeCompound([f.extrude(bvec) for f in plate1])
|
|
plate2 = bplates.cut(cuts2).Faces
|
|
blocks2 = Part.makeCompound([f.extrude(bvec) for f in plate2])
|
|
interval = extv.Length/(fsize)
|
|
entires = int(interval)
|
|
rest = (interval - entires)
|
|
for i in range(entires):
|
|
if i % 2: # odd
|
|
b = blocks2.copy()
|
|
else:
|
|
b = blocks1.copy()
|
|
if i:
|
|
t = FreeCAD.Vector(svec)
|
|
t.multiply(i)
|
|
b.translate(t)
|
|
blocks.append(b)
|
|
if rest:
|
|
rest = extv.Length-(entires*fsize)
|
|
rvec = FreeCAD.Vector(n)
|
|
rvec.multiply(rest)
|
|
if entires % 2:
|
|
b = Part.makeCompound([f.extrude(rvec) for f in plate2])
|
|
else:
|
|
b = Part.makeCompound([f.extrude(rvec) for f in plate1])
|
|
t = FreeCAD.Vector(svec)
|
|
t.multiply(entires)
|
|
b.translate(t)
|
|
blocks.append(b)
|
|
if blocks:
|
|
base = Part.makeCompound(blocks)
|
|
else:
|
|
FreeCAD.Console.PrintWarning(translate("Arch","Error computing block cuts for wall")+obj.Label+"\n")
|
|
else:
|
|
FreeCAD.Console.PrintWarning(translate("Arch","Cannot compute blocks for wall")+obj.Label+"\n")
|
|
|
|
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:
|
|
FreeCAD.Console.PrintError(translate("Arch","Error: Invalid base object")+"\n")
|
|
return
|
|
|
|
base = self.processSubShapes(obj,base,pl)
|
|
self.applyShape(obj,base,pl)
|
|
|
|
# count blocks
|
|
if hasattr(obj,"MakeBlocks"):
|
|
if obj.MakeBlocks:
|
|
fvol = obj.BlockLength.Value * obj.BlockHeight.Value * obj.Width.Value
|
|
if fvol:
|
|
#print("base volume:",fvol)
|
|
#for s in base.Solids:
|
|
#print(abs(s.Volume - fvol))
|
|
ents = [s for s in base.Solids if abs(s.Volume - fvol) < 1]
|
|
obj.CountEntire = len(ents)
|
|
obj.CountBroken = len(base.Solids) - len(ents)
|
|
else:
|
|
obj.CountEntire = 0
|
|
obj.CountBroken = 0
|
|
|
|
# set the length property
|
|
if obj.Base:
|
|
if obj.Base.isDerivedFrom("Part::Feature"):
|
|
if obj.Base.Shape.Edges:
|
|
if not obj.Base.Shape.Faces:
|
|
if hasattr(obj.Base.Shape,"Length"):
|
|
l = obj.Base.Shape.Length
|
|
if obj.Length.Value != l:
|
|
obj.Length = l
|
|
|
|
def onChanged(self,obj,prop):
|
|
|
|
if prop == "Length":
|
|
if obj.Base and obj.Length.Value:
|
|
if obj.Base.isDerivedFrom("Part::Feature"):
|
|
if len(obj.Base.Shape.Edges) == 1:
|
|
import DraftGeomUtils
|
|
e = obj.Base.Shape.Edges[0]
|
|
if DraftGeomUtils.geomType(e) == "Line":
|
|
if e.Length != obj.Length.Value:
|
|
v = e.Vertexes[-1].Point.sub(e.Vertexes[0].Point)
|
|
v.normalize()
|
|
v.multiply(obj.Length.Value)
|
|
p2 = e.Vertexes[0].Point.add(v)
|
|
if Draft.getType(obj.Base) == "Wire":
|
|
#print "modifying p2"
|
|
obj.Base.End = p2
|
|
elif Draft.getType(obj.Base) == "Sketch":
|
|
obj.Base.movePoint(0,2,p2,0)
|
|
else:
|
|
FreeCAD.Console.PrintError(translate("Arch","Error: Unable to modify the base object of this wall")+"\n")
|
|
self.hideSubobjects(obj,prop)
|
|
ArchComponent.Component.onChanged(self,obj,prop)
|
|
|
|
def getFootprint(self,obj):
|
|
|
|
faces = []
|
|
if obj.Shape:
|
|
for f in obj.Shape.Faces:
|
|
if f.normalAt(0,0).getAngle(FreeCAD.Vector(0,0,-1)) < 0.01:
|
|
if abs(abs(f.CenterOfMass.z) - abs(obj.Shape.BoundBox.ZMin)) < 0.001:
|
|
faces.append(f)
|
|
return faces
|
|
|
|
def getExtrusionData(self,obj):
|
|
|
|
"""returns (shape,extrusion vector,placement) or None"""
|
|
import Part,DraftGeomUtils
|
|
data = ArchComponent.Component.getExtrusionData(self,obj)
|
|
if data:
|
|
if not isinstance(data[0],list):
|
|
# multifuses not considered here
|
|
return data
|
|
length = obj.Length.Value
|
|
width = obj.Width.Value
|
|
height = obj.Height.Value
|
|
if not height:
|
|
for p in obj.InList:
|
|
if Draft.getType(p) in ["Floor","BuildingPart"]:
|
|
if p.Height.Value:
|
|
height = p.Height.Value
|
|
if not height:
|
|
return None
|
|
if obj.Normal == Vector(0,0,0):
|
|
normal = Vector(0,0,1)
|
|
else:
|
|
normal = Vector(obj.Normal)
|
|
base = None
|
|
placement = None
|
|
self.basewires = None
|
|
# build wall layers
|
|
layers = []
|
|
if hasattr(obj,"Material"):
|
|
if obj.Material:
|
|
if hasattr(obj.Material,"Materials"):
|
|
varwidth = 0
|
|
restwidth = width - sum(obj.Material.Thicknesses)
|
|
if restwidth > 0:
|
|
varwidth = [t for t in obj.Material.Thicknesses if t == 0]
|
|
if varwidth:
|
|
varwidth = restwidth/len(varwidth)
|
|
for t in obj.Material.Thicknesses:
|
|
if t:
|
|
layers.append(t)
|
|
elif varwidth:
|
|
layers.append(varwidth)
|
|
if obj.Base:
|
|
if obj.Base.isDerivedFrom("Part::Feature"):
|
|
if obj.Base.Shape:
|
|
if obj.Base.Shape.Solids:
|
|
return None
|
|
elif obj.Face > 0:
|
|
if len(obj.Base.Shape.Faces) >= obj.Face:
|
|
face = obj.Base.Shape.Faces[obj.Face-1]
|
|
# this wall is based on a specific face of its base object
|
|
if obj.Normal != Vector(0,0,0):
|
|
normal = face.normalAt(0,0)
|
|
if normal.getAngle(Vector(0,0,1)) > math.pi/4:
|
|
normal.multiply(width)
|
|
base = face.extrude(normal)
|
|
if obj.Align == "Center":
|
|
base.translate(normal.negative().multiply(0.5))
|
|
elif obj.Align == "Right":
|
|
base.translate(normal.negative())
|
|
else:
|
|
normal.multiply(height)
|
|
base = face.extrude(normal)
|
|
base,placement = self.rebase(base)
|
|
return (base,normal,placement)
|
|
elif obj.Base.Shape.Faces:
|
|
if not DraftGeomUtils.isCoplanar(obj.Base.Shape.Faces):
|
|
return None
|
|
else:
|
|
base,placement = self.rebase(obj.Base.Shape)
|
|
elif len(obj.Base.Shape.Edges) == 1:
|
|
self.basewires = [Part.Wire(obj.Base.Shape.Edges)]
|
|
else:
|
|
# self.basewires = obj.Base.Shape.Wires
|
|
self.basewires = []
|
|
for cluster in Part.getSortedClusters(obj.Base.Shape.Edges):
|
|
for c in Part.sortEdges(cluster):
|
|
self.basewires.append(Part.Wire(c))
|
|
|
|
if self.basewires and width:
|
|
if (len(self.basewires) == 1) and layers:
|
|
self.basewires = [self.basewires[0] for l in layers]
|
|
layeroffset = 0
|
|
baseface = None
|
|
for i,wire in enumerate(self.basewires):
|
|
e = wire.Edges[0]
|
|
if isinstance(e.Curve,Part.Circle):
|
|
dvec = e.Vertexes[0].Point.sub(e.Curve.Center)
|
|
else:
|
|
dvec = DraftGeomUtils.vec(wire.Edges[0]).cross(normal)
|
|
if not DraftVecUtils.isNull(dvec):
|
|
dvec.normalize()
|
|
sh = None
|
|
if obj.Align == "Left":
|
|
off = obj.Offset.Value
|
|
if layers:
|
|
off = off+layeroffset
|
|
dvec.multiply(layers[i])
|
|
layeroffset += layers[i]
|
|
else:
|
|
dvec.multiply(width)
|
|
if off:
|
|
dvec2 = DraftVecUtils.scaleTo(dvec,off)
|
|
wire = DraftGeomUtils.offsetWire(wire,dvec2)
|
|
w2 = DraftGeomUtils.offsetWire(wire,dvec)
|
|
w1 = Part.Wire(Part.__sortEdges__(wire.Edges))
|
|
sh = DraftGeomUtils.bind(w1,w2)
|
|
elif obj.Align == "Right":
|
|
dvec = dvec.negative()
|
|
off = obj.Offset.Value
|
|
if layers:
|
|
off = off+layeroffset
|
|
dvec.multiply(layers[i])
|
|
layeroffset += layers[i]
|
|
else:
|
|
dvec.multiply(width)
|
|
if off:
|
|
dvec2 = DraftVecUtils.scaleTo(dvec,off)
|
|
wire = DraftGeomUtils.offsetWire(wire,dvec2)
|
|
w2 = DraftGeomUtils.offsetWire(wire,dvec)
|
|
w1 = Part.Wire(Part.__sortEdges__(wire.Edges))
|
|
sh = DraftGeomUtils.bind(w1,w2)
|
|
elif obj.Align == "Center":
|
|
if layers:
|
|
off = width/2-layeroffset
|
|
d1 = Vector(dvec).multiply(off)
|
|
w1 = DraftGeomUtils.offsetWire(wire,d1)
|
|
layeroffset += layers[i]
|
|
off = width/2-layeroffset
|
|
d1 = Vector(dvec).multiply(off)
|
|
w2 = DraftGeomUtils.offsetWire(wire,d1)
|
|
else:
|
|
dvec.multiply(width/2)
|
|
w1 = DraftGeomUtils.offsetWire(wire,dvec)
|
|
dvec = dvec.negative()
|
|
w2 = DraftGeomUtils.offsetWire(wire,dvec)
|
|
sh = DraftGeomUtils.bind(w1,w2)
|
|
if sh:
|
|
sh.fix(0.1,0,1) # fixes self-intersecting wires
|
|
f = Part.Face(sh)
|
|
if baseface:
|
|
if layers:
|
|
baseface.append(f)
|
|
else:
|
|
baseface = baseface.fuse(f)
|
|
# baseface = baseface.removeSplitter()
|
|
s = DraftGeomUtils.removeSplitter(baseface)
|
|
if s:
|
|
baseface = s
|
|
else:
|
|
if layers:
|
|
baseface = [f]
|
|
else:
|
|
baseface = f
|
|
if baseface:
|
|
base,placement = self.rebase(baseface)
|
|
else:
|
|
if layers:
|
|
totalwidth = sum(layers)
|
|
offset = 0
|
|
base = []
|
|
for l in layers:
|
|
l2 = length/2 or 0.5
|
|
w1 = -totalwidth/2 + offset
|
|
w2 = w1 + l
|
|
v1 = Vector(-l2,w1,0)
|
|
v2 = Vector(l2,w1,0)
|
|
v3 = Vector(l2,w2,0)
|
|
v4 = Vector(-l2,w2,0)
|
|
base.append(Part.Face(Part.makePolygon([v1,v2,v3,v4,v1])))
|
|
offset += l
|
|
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)
|
|
base = Part.Face(Part.makePolygon([v1,v2,v3,v4,v1]))
|
|
placement = FreeCAD.Placement()
|
|
if base and placement:
|
|
extrusion = normal.multiply(height)
|
|
if placement.Rotation.Angle > 0:
|
|
extrusion = placement.inverse().Rotation.multVec(extrusion)
|
|
return (base,extrusion,placement)
|
|
return None
|
|
|
|
class _ViewProviderWall(ArchComponent.ViewProviderComponent):
|
|
|
|
"A View Provider for the Wall object"
|
|
|
|
def __init__(self,vobj):
|
|
|
|
ArchComponent.ViewProviderComponent.__init__(self,vobj)
|
|
vobj.ShapeColor = ArchCommands.getDefaultColor("Wall")
|
|
|
|
def getIcon(self):
|
|
|
|
import Arch_rc
|
|
if hasattr(self,"Object"):
|
|
if self.Object.CloneOf:
|
|
return ":/icons/Arch_Wall_Clone.svg"
|
|
elif (not self.Object.Base) and self.Object.Additions:
|
|
return ":/icons/Arch_Wall_Tree_Assembly.svg"
|
|
return ":/icons/Arch_Wall_Tree.svg"
|
|
|
|
def attach(self,vobj):
|
|
|
|
self.Object = vobj.Object
|
|
from pivy import coin
|
|
tex = coin.SoTexture2()
|
|
tex.image = Draft.loadTexture(Draft.svgpatterns()['simple'][1], 128)
|
|
texcoords = coin.SoTextureCoordinatePlane()
|
|
s = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch").GetFloat("patternScale",0.01)
|
|
texcoords.directionS.setValue(s,0,0)
|
|
texcoords.directionT.setValue(0,s,0)
|
|
self.fcoords = coin.SoCoordinate3()
|
|
self.fset = coin.SoIndexedFaceSet()
|
|
sep = coin.SoSeparator()
|
|
sep.addChild(tex)
|
|
sep.addChild(texcoords)
|
|
sep.addChild(self.fcoords)
|
|
sep.addChild(self.fset)
|
|
vobj.RootNode.addChild(sep)
|
|
ArchComponent.ViewProviderComponent.attach(self,vobj)
|
|
|
|
def updateData(self,obj,prop):
|
|
|
|
if prop in ["Placement","Shape","Material"]:
|
|
if obj.ViewObject.DisplayMode == "Footprint":
|
|
obj.ViewObject.Proxy.setDisplayMode("Footprint")
|
|
if hasattr(obj,"Material"):
|
|
if obj.Material and obj.Shape:
|
|
if hasattr(obj.Material,"Materials"):
|
|
if len(obj.Material.Materials) == len(obj.Shape.Solids):
|
|
cols = []
|
|
for i,mat in enumerate(obj.Material.Materials):
|
|
c = obj.ViewObject.ShapeColor
|
|
c = (c[0],c[1],c[2],obj.ViewObject.Transparency/100.0)
|
|
if 'DiffuseColor' in mat.Material:
|
|
if "(" in mat.Material['DiffuseColor']:
|
|
c = tuple([float(f) for f in mat.Material['DiffuseColor'].strip("()").split(",")])
|
|
if 'Transparency' in mat.Material:
|
|
c = (c[0],c[1],c[2],float(mat.Material['Transparency']))
|
|
cols.extend([c for j in range(len(obj.Shape.Solids[i].Faces))])
|
|
obj.ViewObject.DiffuseColor = cols
|
|
ArchComponent.ViewProviderComponent.updateData(self,obj,prop)
|
|
if len(obj.ViewObject.DiffuseColor) > 1:
|
|
# force-reset colors if changed
|
|
obj.ViewObject.DiffuseColor = obj.ViewObject.DiffuseColor
|
|
|
|
def getDisplayModes(self,vobj):
|
|
|
|
modes = ArchComponent.ViewProviderComponent.getDisplayModes(self,vobj)+["Footprint"]
|
|
return modes
|
|
|
|
def setDisplayMode(self,mode):
|
|
|
|
self.fset.coordIndex.deleteValues(0)
|
|
self.fcoords.point.deleteValues(0)
|
|
if mode == "Footprint":
|
|
if hasattr(self,"Object"):
|
|
faces = self.Object.Proxy.getFootprint(self.Object)
|
|
if faces:
|
|
verts = []
|
|
fdata = []
|
|
idx = 0
|
|
for face in faces:
|
|
tri = face.tessellate(1)
|
|
for v in tri[0]:
|
|
verts.append([v.x,v.y,v.z])
|
|
for f in tri[1]:
|
|
fdata.extend([f[0]+idx,f[1]+idx,f[2]+idx,-1])
|
|
idx += len(tri[0])
|
|
self.fcoords.point.setValues(verts)
|
|
self.fset.coordIndex.setValues(0,len(fdata),fdata)
|
|
return "Wireframe"
|
|
return ArchComponent.ViewProviderComponent.setDisplayMode(self,mode)
|
|
|
|
|
|
if FreeCAD.GuiUp:
|
|
FreeCADGui.addCommand('Arch_Wall',_CommandWall())
|
|
FreeCADGui.addCommand('Arch_MergeWalls',_CommandMergeWalls())
|