primera subida
This commit is contained in:
225
PVPlantFoundation.py
Normal file
225
PVPlantFoundation.py
Normal file
@@ -0,0 +1,225 @@
|
||||
# -*- 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())
|
||||
Reference in New Issue
Block a user