Arch: initial BuildingPart development
This commit is contained in:
Yorik van Havre
337
src/Mod/Arch/ArchBuildingPart.py
Normal file
337
src/Mod/Arch/ArchBuildingPart.py
Normal file
@@ -0,0 +1,337 @@
|
||||
# -*- coding: utf8 -*-
|
||||
|
||||
#***************************************************************************
|
||||
#* *
|
||||
#* Copyright (c) 2018 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,ArchCommands,DraftVecUtils,sys
|
||||
if FreeCAD.GuiUp:
|
||||
import FreeCADGui
|
||||
from PySide import QtCore, QtGui
|
||||
from DraftTools import translate
|
||||
from PySide.QtCore import QT_TRANSLATE_NOOP
|
||||
else:
|
||||
# \cond
|
||||
def translate(ctxt,txt):
|
||||
return txt
|
||||
def QT_TRANSLATE_NOOP(ctxt,txt):
|
||||
return txt
|
||||
# \endcond
|
||||
|
||||
## @package ArchBuildingPart
|
||||
# \ingroup ARCH
|
||||
# \brief The BuildingPart object and tools
|
||||
#
|
||||
# This module provides tools to build BuildingPart objects.
|
||||
# BuildingParts are used to group different Arch objects
|
||||
|
||||
__title__="FreeCAD Arch BuildingPart"
|
||||
__author__ = "Yorik van Havre"
|
||||
__url__ = "http://www.freecadweb.org"
|
||||
|
||||
|
||||
# possible roles for BuildingPart objects
|
||||
Roles = ["Undefined","Storey","Zone","SpatialZone","Component","Group"]
|
||||
|
||||
|
||||
def makeBuildingPart(objectslist=None,name="BuildingPart"):
|
||||
|
||||
'''makeBuildingPart(objectslist): creates a buildingPart including the
|
||||
objects from the given list.'''
|
||||
|
||||
obj = FreeCAD.ActiveDocument.addObject("App::GeometryPython",name)
|
||||
obj.Label = translate("Arch",name)
|
||||
BuildingPart(obj)
|
||||
if FreeCAD.GuiUp:
|
||||
ViewProviderBuildingPart(obj.ViewObject)
|
||||
if objectslist:
|
||||
obj.addObjects(objectslist)
|
||||
return obj
|
||||
|
||||
|
||||
def convertFloors(floor=None):
|
||||
|
||||
"""convert the given Floor (or all Arch Floors from the active document if none is given) into BuildingParts"""
|
||||
|
||||
todel = []
|
||||
if floor:
|
||||
objset = [floor]
|
||||
else:
|
||||
objset = FreeCAD.ActiveDocument.Objects
|
||||
for obj in objset:
|
||||
if Draft.getType(obj) == "Floor":
|
||||
nobj = makeBuildingPart(obj.Group)
|
||||
nobj.Role = "Storey"
|
||||
label = obj.Label
|
||||
for parent in obj.InList:
|
||||
if hasattr(parent,"Group"):
|
||||
if obj in parent.Group:
|
||||
parent.addObject(nobj)
|
||||
#g = parent.Group
|
||||
#g.append(nobj)
|
||||
#parent.Group = g
|
||||
else:
|
||||
print("Warning: couldn't add new object '"+label+"' to parent object '"+parent.Label+"'")
|
||||
todel.append(obj.Name)
|
||||
if obj.ViewObject:
|
||||
obj.ViewObject.Proxy.Object = None # some bug makes this trigger even efter the object has been deleted...
|
||||
obj.Label = obj.Label+" to delete" # in case FreeCAD doesn't allow 2 objs with same label
|
||||
nobj.Label = label
|
||||
for n in todel:
|
||||
from DraftGui import todo
|
||||
todo.delay(FreeCAD.ActiveDocument.removeObject,n)
|
||||
|
||||
|
||||
class CommandBuildingPart:
|
||||
|
||||
|
||||
"the Arch BuildingPart command definition"
|
||||
|
||||
def GetResources(self):
|
||||
return {'Pixmap' : 'Arch_BuildingPart',
|
||||
'MenuText': QT_TRANSLATE_NOOP("Arch_BuildingPart","BuildingPart"),
|
||||
'Accel': "B, P",
|
||||
'ToolTip': QT_TRANSLATE_NOOP("Arch_BuildingPart","Creates a BuildingPart object including selected objects")}
|
||||
|
||||
def IsActive(self):
|
||||
return not FreeCAD.ActiveDocument is None
|
||||
|
||||
def Activated(self):
|
||||
sel = FreeCADGui.Selection.getSelection()
|
||||
ss = "[ "
|
||||
for o in sel:
|
||||
ss += "FreeCAD.ActiveDocument." + o.Name + ", "
|
||||
ss += "]"
|
||||
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create BuildingPart"))
|
||||
FreeCADGui.addModule("Arch")
|
||||
FreeCADGui.doCommand("obj = Arch.makeBuildingPart("+ss+")")
|
||||
FreeCADGui.addModule("Draft")
|
||||
FreeCADGui.doCommand("Draft.autogroup(obj)")
|
||||
FreeCAD.ActiveDocument.commitTransaction()
|
||||
FreeCAD.ActiveDocument.recompute()
|
||||
|
||||
|
||||
|
||||
class BuildingPart:
|
||||
|
||||
|
||||
"The BuildingPart object"
|
||||
|
||||
def __init__(self,obj):
|
||||
|
||||
obj.addExtension('App::OriginGroupExtensionPython', self)
|
||||
obj.addProperty("App::PropertyLength","Height","Arch",QT_TRANSLATE_NOOP("App::Property","The height of this object"))
|
||||
obj.addProperty("App::PropertyLength","LevelOffset","Arch",QT_TRANSLATE_NOOP("App::Property","The level of the (0,0,0) point of this level"))
|
||||
obj.addProperty("App::PropertyArea","Area", "Arch",QT_TRANSLATE_NOOP("App::Property","The computed floor area of this floor"))
|
||||
obj.addProperty("App::PropertyEnumeration","Role","Component",QT_TRANSLATE_NOOP("App::Property","The role of this object"))
|
||||
obj.addProperty("App::PropertyLink","CloneOf","Component",QT_TRANSLATE_NOOP("App::Property","The object this component is cloning"))
|
||||
obj.addProperty("App::PropertyString","Description","Component",QT_TRANSLATE_NOOP("App::Property","An optional description for this component"))
|
||||
obj.addProperty("App::PropertyString","Tag","Component",QT_TRANSLATE_NOOP("App::Property","An optional tag for this component"))
|
||||
obj.addProperty("App::PropertyMap","IfcAttributes","Component",QT_TRANSLATE_NOOP("App::Property","Custom IFC properties and attributes"))
|
||||
obj.Role = Roles
|
||||
self.Type = "BuildingPart"
|
||||
obj.Proxy = self
|
||||
|
||||
def __getstate__(self):
|
||||
return self.Type
|
||||
|
||||
def __setstate__(self,state):
|
||||
if state:
|
||||
self.Type = state
|
||||
|
||||
def onChanged(self,obj,prop):
|
||||
if (prop == "Height"):
|
||||
for o in obj.Group:
|
||||
if Draft.getType(o) in ["Wall","Structure"]:
|
||||
if not o.Height.Value:
|
||||
o.Proxy.execute(o)
|
||||
|
||||
def execute(self,obj):
|
||||
pass
|
||||
|
||||
def getSpaces(self,obj):
|
||||
"gets the list of Spaces that have this object as their Zone property"
|
||||
g = []
|
||||
for o in obj.OutList:
|
||||
if hasattr(o,"Zone"):
|
||||
if o.Zone == obj:
|
||||
g.append(o)
|
||||
return g
|
||||
|
||||
|
||||
|
||||
class ViewProviderBuildingPart:
|
||||
|
||||
|
||||
"A View Provider for the BuildingPart object"
|
||||
|
||||
def __init__(self,vobj):
|
||||
vobj.addExtension("Gui::ViewProviderGeoFeatureGroupExtensionPython", self)
|
||||
vobj.addProperty("App::PropertyFloat","LineWidth","Base","")
|
||||
vobj.addProperty("App::PropertyString","OverrideUnit","Base",QT_TRANSLATE_NOOP("App::Property","An optional unit to express levels"))
|
||||
vobj.addProperty("App::PropertyFont","FontName","Base","")
|
||||
vobj.addProperty("App::PropertyLength","FontSize","Base","")
|
||||
vobj.addProperty("App::PropertyPlacement","DisplayOffset","Base",QT_TRANSLATE_NOOP("App::Property","A transformation to apply to the level mark"))
|
||||
vobj.addProperty("App::PropertyBool","ShowLevel","Base",QT_TRANSLATE_NOOP("App::Property","If true, show the level"))
|
||||
vobj.addProperty("App::PropertyBool","ShowUnit","Base",QT_TRANSLATE_NOOP("App::Property","If true, show the unit on the level tag"))
|
||||
vobj.addProperty("App::PropertyBool","SetWorkingPlane","Base",QT_TRANSLATE_NOOP("App::Property","If true, when activated, the working plane will automatically adapt to this level"))
|
||||
vobj.addProperty("App::PropertyBool","OriginOffset","Base",QT_TRANSLATE_NOOP("App::Property","If true, when activated, Display offset will affect the origin mark too"))
|
||||
vobj.addProperty("App::PropertyBool","ShowLabel","Base",QT_TRANSLATE_NOOP("App::Property","If true, when activated, the object's label is displayed"))
|
||||
vobj.FontName = Draft.getParam("textfont","Arial")
|
||||
vobj.FontSize = Draft.getParam("textheight",2.0)
|
||||
vobj.ShapeColor = (0.13,0.15,0.37)
|
||||
vobj.LineWidth = 1
|
||||
vobj.SetWorkingPlane = True
|
||||
vobj.ShowLevel = True
|
||||
vobj.Proxy = self
|
||||
|
||||
def getIcon(self):
|
||||
import Arch_rc
|
||||
if hasattr(self,"Object"):
|
||||
if self.Object.Role == "Storey":
|
||||
return ":/icons/Arch_Floor_Tree.svg"
|
||||
return ":/icons/Arch_BuildingPart_Tree.svg"
|
||||
|
||||
def attach(self,vobj):
|
||||
self.Object = vobj.Object
|
||||
from pivy import coin
|
||||
self.sep = coin.SoSeparator()
|
||||
self.mat = coin.SoMaterial()
|
||||
self.sep.addChild(self.mat)
|
||||
self.dst = coin.SoDrawStyle()
|
||||
self.sep.addChild(self.dst)
|
||||
self.lco = coin.SoCoordinate3()
|
||||
self.sep.addChild(self.lco)
|
||||
lin = coin.SoType.fromName("SoBrepEdgeSet").createInstance()
|
||||
lin.coordIndex.setValues([0,1,-1,2,3,-1,4,5,-1])
|
||||
self.sep.addChild(lin)
|
||||
self.tra = coin.SoTransform()
|
||||
self.tra.rotation.setValue(FreeCAD.Rotation(0,0,90).Q)
|
||||
self.sep.addChild(self.tra)
|
||||
self.fon = coin.SoFont()
|
||||
self.sep.addChild(self.fon)
|
||||
self.txt = coin.SoAsciiText()
|
||||
self.txt.justification = coin.SoText2.LEFT
|
||||
self.txt.string.setValue("level")
|
||||
self.sep.addChild(self.txt)
|
||||
self.onChanged(vobj,"ShapeColor")
|
||||
self.onChanged(vobj,"FontName")
|
||||
self.onChanged(vobj,"FontSize")
|
||||
self.onChanged(vobj,"ShowLevel")
|
||||
return
|
||||
|
||||
def updateData(self,obj,prop):
|
||||
if prop in ["Placement","LevelOffset"]:
|
||||
self.onChanged(obj.ViewObject,"OverrideUnit")
|
||||
|
||||
def onChanged(self,vobj,prop):
|
||||
if prop == "ShapeColor":
|
||||
l = vobj.ShapeColor
|
||||
self.mat.diffuseColor.setValue([l[0],l[1],l[2]])
|
||||
elif prop == "LineWidth":
|
||||
self.dst.lineWidth = vobj.LineWidth
|
||||
elif prop == "FontName":
|
||||
if vobj.FontName:
|
||||
if sys.version_info.major < 3:
|
||||
self.fon.name = vobj.FontName.encode("utf8")
|
||||
else:
|
||||
self.fon.name = vobj.FontName
|
||||
elif prop in ["FontSize","DisplayOffset","OriginOffset"]:
|
||||
fs = vobj.FontSize.Value
|
||||
if fs:
|
||||
self.fon.size = fs
|
||||
b = vobj.DisplayOffset.Base
|
||||
self.tra.translation.setValue([b.x+fs/8,b.y,b.z+fs/8])
|
||||
if vobj.OriginOffset:
|
||||
self.lco.point.setValues([[b.x-fs,b.y,b.z],[b.x+fs,b.y,b.z],[b.x,b.y-fs,b.z],[b.x,b.y+fs,b.z],[b.x,b.y,b.z-fs],[b.x,b.y,b.z+fs]])
|
||||
else:
|
||||
self.lco.point.setValues([[-fs,0,0],[fs,0,0],[0,-fs,0],[0,fs,0],[0,0,-fs],[0,0,fs]])
|
||||
elif prop in ["ShowLevel","ShowLabel"]:
|
||||
rn = vobj.RootNode
|
||||
if vobj.ShowLevel or vobj.ShowLabel:
|
||||
if rn.findChild(self.sep) == -1:
|
||||
rn.addChild(self.sep)
|
||||
self.onChanged(vobj,"ShowUnit")
|
||||
else:
|
||||
if rn.findChild(self.sep) != -1:
|
||||
rn.removeChild(self.sep)
|
||||
elif prop in ["OverrideUnit","ShowUnit"]:
|
||||
z = vobj.Object.Placement.Base.z + vobj.Object.LevelOffset.Value
|
||||
q = FreeCAD.Units.Quantity(z,FreeCAD.Units.Length)
|
||||
txt = ""
|
||||
if vobj.ShowLabel:
|
||||
txt += vobj.Object.Label
|
||||
if vobj.ShowLevel:
|
||||
if txt:
|
||||
txt += " "
|
||||
if z >= 0:
|
||||
txt = "+"
|
||||
if vobj.OverrideUnit:
|
||||
u = vobj.OverrideUnit
|
||||
else:
|
||||
u = q.getUserPreferred()[2]
|
||||
q = q.getValueAs(u)
|
||||
d = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units").GetInt("Decimals",0)
|
||||
fmt = "{0:."+ str(d) + "f}"
|
||||
if not vobj.ShowUnit:
|
||||
u = ""
|
||||
txt += fmt.format(float(q)) + str(u)
|
||||
if isinstance(txt,unicode):
|
||||
txt = txt.encode("utf8")
|
||||
self.txt.string.setValue(txt)
|
||||
|
||||
def doubleClicked(self,vobj):
|
||||
self.activate(vobj)
|
||||
FreeCADGui.Selection.clearSelection()
|
||||
return True
|
||||
|
||||
def activate(self,vobj):
|
||||
if FreeCADGui.ActiveDocument.ActiveView.getActiveObject("Arch") == vobj.Object:
|
||||
FreeCADGui.ActiveDocument.ActiveView.setActiveObject("Arch",None)
|
||||
else:
|
||||
FreeCADGui.ActiveDocument.ActiveView.setActiveObject("Arch",vobj.Object)
|
||||
if vobj.SetWorkingPlane:
|
||||
self.setWorkingPlane()
|
||||
|
||||
def setupContextMenu(self,vobj,menu):
|
||||
from PySide import QtCore,QtGui
|
||||
action1 = QtGui.QAction(QtGui.QIcon(":/icons/Draft_SelectPlane.svg"),"Set working plane",menu)
|
||||
QtCore.QObject.connect(action1,QtCore.SIGNAL("triggered()"),self.setWorkingPlane)
|
||||
menu.addAction(action1)
|
||||
|
||||
def setWorkingPlane(self):
|
||||
if hasattr(self,"Object") and hasattr(FreeCAD,"DraftWorkingPlane"):
|
||||
import FreeCADGui
|
||||
FreeCAD.DraftWorkingPlane.setFromPlacement(self.Object.Placement,rebase=True)
|
||||
FreeCAD.DraftWorkingPlane.weak = False
|
||||
if hasattr(FreeCADGui,"Snapper"):
|
||||
FreeCADGui.Snapper.setGrid()
|
||||
if hasattr(FreeCADGui,"draftToolBar"):
|
||||
FreeCADGui.draftToolBar.wplabel.setText(self.Object.Label)
|
||||
|
||||
def __getstate__(self):
|
||||
return None
|
||||
|
||||
def __setstate__(self,state):
|
||||
return None
|
||||
|
||||
if FreeCAD.GuiUp:
|
||||
FreeCADGui.addCommand('Arch_BuildingPart',CommandBuildingPart())
|
||||
Reference in New Issue
Block a user