279 lines
9.3 KiB
Python
279 lines
9.3 KiB
Python
# /**********************************************************************
|
|
# * *
|
|
# * Copyright (c) 2021 Javier Braña <javier.branagutierrez@gmail.com> *
|
|
# * *
|
|
# * 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 ArchComponent
|
|
import FreeCAD
|
|
import Part
|
|
|
|
if FreeCAD.GuiUp:
|
|
import FreeCADGui
|
|
from PySide import QtCore
|
|
import draftguitools.gui_trackers as DraftTrackers
|
|
else:
|
|
# \cond
|
|
def translate(ctxt, txt):
|
|
return txt
|
|
def QT_TRANSLATE_NOOP(ctxt, txt):
|
|
return txt
|
|
# \endcond
|
|
|
|
try:
|
|
_fromUtf8 = QtCore.QString.fromUtf8
|
|
except AttributeError:
|
|
def _fromUtf8(s):
|
|
return s
|
|
|
|
import os
|
|
import PVPlantResources
|
|
from PVPlantResources import DirIcons as DirIcons
|
|
|
|
|
|
|
|
def makeManhole(name="Manhole"):
|
|
obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython", "Manhole")
|
|
obj.Label = name
|
|
_Manhole(obj)
|
|
_ViewProviderManhole(obj.ViewObject)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
return obj
|
|
|
|
class _Manhole(ArchComponent.Component):
|
|
"A Manhole Obcject"
|
|
|
|
def __init__(self, obj):
|
|
# Definición de Variables:
|
|
ArchComponent.Component.__init__(self, obj)
|
|
self.setProperties(obj)
|
|
|
|
def setProperties(self, obj):
|
|
pl = obj.PropertiesList
|
|
# Dimensions: --------------------------------------------------------------------------------------------------
|
|
if not "Height" in pl:
|
|
obj.addProperty("App::PropertyLength",
|
|
"Height",
|
|
"Manhole",
|
|
"The height of this object"
|
|
).Height = 1000
|
|
|
|
if not "Width" in pl:
|
|
obj.addProperty("App::PropertyLength",
|
|
"Width",
|
|
"Manhole",
|
|
"The width of this object"
|
|
).Width = 1000
|
|
|
|
if not "Length" in pl:
|
|
obj.addProperty("App::PropertyLength",
|
|
"Length",
|
|
"Manhole",
|
|
"The height of this object"
|
|
).Length = 2000
|
|
|
|
if not "Thickness" in pl:
|
|
obj.addProperty("App::PropertyLength",
|
|
"Thickness",
|
|
"Manhole",
|
|
"The height of this object"
|
|
).Thickness = 100
|
|
|
|
# outputs:
|
|
if not "InternalVolume" in pl:
|
|
obj.addProperty("App::PropertyVolume",
|
|
"InternalVolume",
|
|
"Outputs",
|
|
"The height of this object"
|
|
)
|
|
|
|
if not "ExternalVolume" in pl:
|
|
obj.addProperty("App::PropertyVolume",
|
|
"ExternalVolume",
|
|
"Outputs",
|
|
"The height of this object"
|
|
)
|
|
|
|
self.Type = "Manhole"
|
|
obj.Proxy = self
|
|
|
|
def onDocumentRestored(self, obj):
|
|
"""Method run when the document is restored.
|
|
Re-adds the Arch component, and Arch wall properties."""
|
|
ArchComponent.Component.onDocumentRestored(self, obj)
|
|
self.setProperties(obj)
|
|
|
|
def onChanged(self, obj, prop):
|
|
'''Do something when a property has changed'''
|
|
|
|
def execute(self, obj):
|
|
w_med = obj.Width.Value / 2
|
|
l_med = obj.Length.Value / 2
|
|
|
|
p1 = FreeCAD.Vector(-l_med, -w_med, 0)
|
|
p2 = FreeCAD.Vector( l_med, -w_med, 0)
|
|
p3 = FreeCAD.Vector( l_med, w_med, 0)
|
|
p4 = FreeCAD.Vector(-l_med, w_med, 0)
|
|
ext = Part.Face(Part.makePolygon([p1, p2, p3, p4, p1, ]))
|
|
ins = ext.makeOffset2D(-obj.Thickness.Value, join=2, openResult=True)
|
|
ext_sol = ext.extrude(FreeCAD.Vector(0, 0, -obj.Height.Value))
|
|
ins_sol = ins.extrude(FreeCAD.Vector(0, 0, -(obj.Height.Value - obj.Thickness.Value)))
|
|
|
|
obj.Shape = ext_sol.cut([ins_sol, ], 0.0)
|
|
|
|
|
|
|
|
|
|
class _ViewProviderManhole(ArchComponent.ViewProviderComponent):
|
|
"A View Provider for the Pipe object"
|
|
|
|
def __init__(self, vobj):
|
|
ArchComponent.ViewProviderComponent.__init__(self, vobj)
|
|
|
|
def getIcon(self):
|
|
return str(os.path.join(DirIcons, "manhole.svg"))
|
|
|
|
def setEdit(self, vobj, mode):
|
|
"""Method called when the document requests the object to enter edit mode.
|
|
|
|
Edit mode is entered when a user double clicks on an object in the tree
|
|
view, or when they use the menu option [Edit -> Toggle Edit Mode].
|
|
|
|
Just display the standard Arch component task panel.
|
|
|
|
Parameters
|
|
----------
|
|
mode: int or str
|
|
The edit mode the document has requested. Set to 0 when requested via
|
|
a double click or [Edit -> Toggle Edit Mode].
|
|
|
|
Returns
|
|
-------
|
|
bool
|
|
If edit mode was entered.
|
|
"""
|
|
|
|
if (mode == 0) and hasattr(self, "Object"):
|
|
taskd = _ManholeTaskPanel(self.Object)
|
|
taskd.obj = self.Object
|
|
# taskd.update()
|
|
FreeCADGui.Control.showDialog(taskd)
|
|
return True
|
|
|
|
return False
|
|
|
|
import draftguitools.gui_tool_utils as gui_tool_utils
|
|
|
|
class _ManholeTaskPanel:
|
|
def __init__(self, obj=None):
|
|
self.new = False
|
|
if obj is None:
|
|
self.new = True
|
|
obj = makeManhole()
|
|
|
|
self.obj = obj
|
|
|
|
self.form = FreeCADGui.PySideUic.loadUi(PVPlantResources.__dir__ + "/PVPlantManhole.ui")
|
|
|
|
self.node = None
|
|
self.view = FreeCADGui.ActiveDocument.ActiveView
|
|
self.tracker = DraftTrackers.ghostTracker(obj)
|
|
self.tracker.on()
|
|
self.call = self.view.addEventCallback("SoEvent", self.action)
|
|
|
|
def action(self, arg):
|
|
"""Handle the 3D scene events.
|
|
|
|
This is installed as an EventCallback in the Inventor view.
|
|
|
|
Parameters
|
|
----------
|
|
arg: dict
|
|
Dictionary with strings that indicates the type of event received
|
|
from the 3D view.
|
|
"""
|
|
|
|
if arg["Type"] == "SoKeyboardEvent" and arg["Key"] == "ESCAPE":
|
|
self.finish()
|
|
|
|
elif arg["Type"] == "SoLocation2Event":
|
|
point, ctrlPoint, info = gui_tool_utils.getPoint(self, arg)
|
|
if info:
|
|
self.tracker.move(FreeCAD.Vector(info["x"], info["y"], info["z"]))
|
|
else:
|
|
self.tracker.move(point)
|
|
|
|
elif (arg["Type"] == "SoMouseButtonEvent" and
|
|
arg["State"] == "DOWN" and
|
|
arg["Button"] == "BUTTON1"):
|
|
|
|
point, ctrlPoint, info = gui_tool_utils.getPoint(self, arg)
|
|
if info:
|
|
self.obj.Placement.Base = FreeCAD.Vector(info["x"], info["y"], info["z"])
|
|
else:
|
|
self.obj.Placement.Base = point
|
|
self.finish()
|
|
|
|
def finish(self):
|
|
self.accept()
|
|
|
|
def accept(self):
|
|
self.closeForm()
|
|
return True
|
|
|
|
def reject(self):
|
|
if self.new:
|
|
FreeCAD.ActiveDocument.removeObject(self.obj.Name)
|
|
self.closeForm()
|
|
return True
|
|
|
|
def closeForm(self):
|
|
self.tracker.finalize()
|
|
FreeCADGui.Control.closeDialog()
|
|
self.view.removeEventCallback("SoEvent", self.call)
|
|
|
|
|
|
class _CommandManhole:
|
|
"the Arch Building command definition"
|
|
|
|
def GetResources(self):
|
|
return {'Pixmap': str(os.path.join(DirIcons, "manhole.svg")),
|
|
'MenuText': "Manhole",
|
|
'Accel': "C, M",
|
|
'ToolTip': "Creates a Manhole object from setup dialog."}
|
|
|
|
def IsActive(self):
|
|
return not FreeCAD.ActiveDocument is None
|
|
|
|
if FreeCAD.ActiveDocument is not None:
|
|
if FreeCADGui.Selection.getCompleteSelection():
|
|
for ob in FreeCAD.ActiveDocument.Objects:
|
|
if ob.Name[:4] == "Site":
|
|
return True
|
|
|
|
def Activated(self):
|
|
TaskPanel = _ManholeTaskPanel()
|
|
FreeCADGui.Control.showDialog(TaskPanel)
|
|
return
|
|
|
|
|
|
if FreeCAD.GuiUp:
|
|
FreeCADGui.addCommand('PVPlantManhole', _CommandManhole())
|
|
|