226 lines
8.2 KiB
Python
226 lines
8.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# ***************************************************************************
|
||
|
|
# * Copyright (c) 2016 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, ArchComponent
|
||
|
|
|
||
|
|
if FreeCAD.GuiUp:
|
||
|
|
import FreeCADGui, os
|
||
|
|
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
|
||
|
|
|
||
|
|
from PVPlantResources import DirIcons as DirIcons
|
||
|
|
|
||
|
|
__title__="FreeCAD Fotovoltaic Power Plant Toolkit"
|
||
|
|
__author__ = "Javier Braña"
|
||
|
|
__url__ = "sn"
|
||
|
|
|
||
|
|
|
||
|
|
def makeFoundation(baseobj=None, diameter=200, length=700, placement=None, name="Foundation"):
|
||
|
|
"makeFoundation([baseobj,diamerter,length,placement,name]):"
|
||
|
|
"creates an Foundation object from the given base object"
|
||
|
|
|
||
|
|
if not FreeCAD.ActiveDocument:
|
||
|
|
FreeCAD.Console.PrintError("No active document. Aborting\n")
|
||
|
|
return
|
||
|
|
|
||
|
|
obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython", name)
|
||
|
|
obj.Label = name
|
||
|
|
_PVPlantFoundation(obj)
|
||
|
|
|
||
|
|
if FreeCAD.GuiUp:
|
||
|
|
_ViewProviderFoundation(obj.ViewObject)
|
||
|
|
if baseobj:
|
||
|
|
baseobj.ViewObject.hide()
|
||
|
|
if baseobj:
|
||
|
|
obj.Base = baseobj
|
||
|
|
else:
|
||
|
|
if length:
|
||
|
|
obj.Length = length
|
||
|
|
else:
|
||
|
|
obj.Length = 1000
|
||
|
|
if diameter:
|
||
|
|
obj.Diameter = diameter
|
||
|
|
else:
|
||
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Arch")
|
||
|
|
obj.Diameter = p.GetFloat("PipeDiameter", 50)
|
||
|
|
if placement:
|
||
|
|
obj.Placement = placement
|
||
|
|
return obj
|
||
|
|
|
||
|
|
|
||
|
|
class _PVPlantFoundation(ArchComponent.Component):
|
||
|
|
"the PVPlant Foundation object"
|
||
|
|
|
||
|
|
def __init__(self, obj):
|
||
|
|
ArchComponent.Component.__init__(self, obj)
|
||
|
|
self.Type = ""
|
||
|
|
self.setProperties(obj)
|
||
|
|
|
||
|
|
def setProperties(self, obj):
|
||
|
|
|
||
|
|
pl = obj.PropertiesList
|
||
|
|
if not "Diameter" in pl:
|
||
|
|
obj.addProperty("App::PropertyLength", "Diameter", "Foundation", QT_TRANSLATE_NOOP("App::Property",
|
||
|
|
"The diameter of this Foundation, if not based on a profile"))
|
||
|
|
if not "Length" in pl:
|
||
|
|
obj.addProperty("App::PropertyLength", "Length", "Foundation", QT_TRANSLATE_NOOP("App::Property",
|
||
|
|
"The length of this Foundation, if not based on an edge"))
|
||
|
|
if not "Profile" in pl:
|
||
|
|
obj.addProperty("App::PropertyLink", "Profile", "Foundation",
|
||
|
|
QT_TRANSLATE_NOOP("App::Property", "An optional closed profile to base this Foundation on"))
|
||
|
|
|
||
|
|
self.Type = "Foundation"
|
||
|
|
obj.IfcType = "Element"
|
||
|
|
|
||
|
|
def onDocumentRestored(self, obj):
|
||
|
|
ArchComponent.Component.onDocumentRestored(self, obj)
|
||
|
|
self.setProperties(obj)
|
||
|
|
|
||
|
|
def execute(self, obj):
|
||
|
|
|
||
|
|
import DraftGeomUtils, math
|
||
|
|
pl = obj.Placement
|
||
|
|
w = self.getWire(obj)
|
||
|
|
if not w:
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "Unable to build the base path") + "\n")
|
||
|
|
return
|
||
|
|
|
||
|
|
p = self.getProfile(obj)
|
||
|
|
if not p:
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "Unable to build the profile") + "\n")
|
||
|
|
return
|
||
|
|
|
||
|
|
# move and rotate the profile to the first point
|
||
|
|
delta = w.Vertexes[0].Point - p.CenterOfMass
|
||
|
|
p.translate(delta)
|
||
|
|
import Draft
|
||
|
|
if Draft.getType(obj.Base) == "BezCurve":
|
||
|
|
v1 = obj.Base.Placement.multVec(obj.Base.Points[1]) - w.Vertexes[0].Point
|
||
|
|
else:
|
||
|
|
v1 = w.Vertexes[1].Point - w.Vertexes[0].Point
|
||
|
|
v2 = DraftGeomUtils.getNormal(p)
|
||
|
|
rot = FreeCAD.Rotation(v2, v1)
|
||
|
|
p.rotate(p.CenterOfMass, rot.Axis, math.degrees(rot.Angle))
|
||
|
|
|
||
|
|
try:
|
||
|
|
sh = w.makePipeShell([p], True, False, 2)
|
||
|
|
except:
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "Unable to build the Foundation -- ") + "\n")
|
||
|
|
else:
|
||
|
|
obj.Shape = sh
|
||
|
|
if obj.Base:
|
||
|
|
obj.Length = w.Length
|
||
|
|
else:
|
||
|
|
obj.Placement = pl
|
||
|
|
|
||
|
|
def getWire(self, obj):
|
||
|
|
|
||
|
|
import Part
|
||
|
|
if obj.Base:
|
||
|
|
if not hasattr(obj.Base, 'Shape'):
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "The base object is not a Part") + "\n")
|
||
|
|
return
|
||
|
|
if len(obj.Base.Shape.Wires) != 1:
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "Too many wires in the base shape") + "\n")
|
||
|
|
return
|
||
|
|
if obj.Base.Shape.Wires[0].isClosed():
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "The base wire is closed") + "\n")
|
||
|
|
return
|
||
|
|
w = obj.Base.Shape.Wires[0]
|
||
|
|
else:
|
||
|
|
if obj.Length.Value == 0:
|
||
|
|
return
|
||
|
|
w = Part.Wire([Part.LineSegment(FreeCAD.Vector(0, 0, 0), FreeCAD.Vector(0, 0, obj.Length.Value)).toShape()])
|
||
|
|
return w
|
||
|
|
|
||
|
|
def getProfile(self, obj):
|
||
|
|
|
||
|
|
import Part
|
||
|
|
if obj.Profile:
|
||
|
|
if not obj.Profile.getLinkedObject().isDerivedFrom("Part::Part2DObject"):
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "The profile is not a 2D Part") + "\n")
|
||
|
|
return
|
||
|
|
if len(obj.Profile.Shape.Wires) != 1:
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "Too many wires in the profile") + "\n")
|
||
|
|
return
|
||
|
|
if not obj.Profile.Shape.Wires[0].isClosed():
|
||
|
|
FreeCAD.Console.PrintError(translate("Arch", "The profile is not closed") + "\n")
|
||
|
|
return
|
||
|
|
p = obj.Profile.Shape.Wires[0]
|
||
|
|
else:
|
||
|
|
if obj.Diameter.Value == 0:
|
||
|
|
return
|
||
|
|
p = Part.Wire(
|
||
|
|
[Part.Circle(FreeCAD.Vector(0, 0, 0), FreeCAD.Vector(0, 0, 1), obj.Diameter.Value / 2).toShape()])
|
||
|
|
return p
|
||
|
|
|
||
|
|
|
||
|
|
class _ViewProviderFoundation(ArchComponent.ViewProviderComponent):
|
||
|
|
"A View Provider for the Foundation object"
|
||
|
|
|
||
|
|
def __init__(self, vobj):
|
||
|
|
ArchComponent.ViewProviderComponent.__init__(self, vobj)
|
||
|
|
|
||
|
|
def getIcon(self):
|
||
|
|
return ":/icons/Arch_Pipe_Tree.svg"
|
||
|
|
|
||
|
|
|
||
|
|
def _FoundationTaskPanel():
|
||
|
|
''''''
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class _CommandFoundation:
|
||
|
|
|
||
|
|
def GetResources(self):
|
||
|
|
"""Set icon, menu and tooltip."""
|
||
|
|
return {'Pixmap': str(os.path.join(DirIcons, "trench.svg")),
|
||
|
|
'MenuText': "Trench",
|
||
|
|
'Accel': "C, T",
|
||
|
|
'ToolTip': "Creates a Trench object from setup dialog."}
|
||
|
|
|
||
|
|
def IsActive(self):
|
||
|
|
if FreeCAD.ActiveDocument:
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def Activated(self):
|
||
|
|
"""Execute when the command is called."""
|
||
|
|
task = _FoundationTaskPanel();
|
||
|
|
sel = FreeCADGui.Selection.getSelection()
|
||
|
|
|
||
|
|
|
||
|
|
if FreeCAD.GuiUp:
|
||
|
|
FreeCADGui.addCommand('PVPlantFoundation', _CommandFoundation())
|