248 lines
11 KiB
Python
248 lines
11 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 *
|
|
# * *
|
|
# ***********************************************************************
|
|
|
|
__title__="FreeCAD Fotovoltaic Power Plant Toolkit"
|
|
__author__ = "Javier Braña"
|
|
__url__ = "sn"
|
|
|
|
|
|
|
|
class PVPlantWorkbench (Workbench):
|
|
import os
|
|
from PVPlantResources import DirIcons as DirIcons
|
|
|
|
MenuText = "PVPlant"
|
|
ToolTip = "Workbench for PV design"
|
|
Icon = str(os.path.join(DirIcons, "icon.svg"))
|
|
|
|
def Initialize(self):
|
|
import sys
|
|
sys.path.append(r"C:\Users\javie\AppData\Roaming\FreeCAD\Mod")
|
|
|
|
# Mias
|
|
import PVPlantGeoreferencing, PVPlantPlacement, \
|
|
PVPlantTerrainAnalisys, PVPlantSite, PVPlantImportGrid, PVPlantFence,\
|
|
PVPlantFoundation, PVPlantCreateTerrainMesh, \
|
|
PVPlantTreeGenerator, PVPlantBuilding, PVPlantTrench, PVPlantEarthWorks, \
|
|
PVPlantStringing, \
|
|
PVPlantPad, PVPlantRoad, PVPlantTerrain, PVPlantManhole, \
|
|
GraphProfile, Utils.PVPlantTrace,\
|
|
reload
|
|
import PVPlantRackChecking
|
|
|
|
from Project.Area import PVPlantArea, PVPlantAreaUtils
|
|
from Project import ProjectSetup
|
|
from Export import exportPVSyst, PVPlantBOQMechanical, PVPlantBOQElectrical, PVPlantBOQCivil,\
|
|
exportDXF
|
|
from Importer import importDXF
|
|
|
|
from Mechanical.Frame import PVPlantFrame
|
|
|
|
from Electrical.Cable import PVPlantCable, PVPlantElectricalLine
|
|
from Electrical.CombinerBox import PVPlantStringBox
|
|
from Electrical.Inverter import PVPlantInverter
|
|
|
|
# A list of command names created in the line above
|
|
self.projectlist = ["Reload",
|
|
"PVPlantSite",
|
|
"PVPlantGeoreferencing",
|
|
"ProjectSetup",
|
|
#"ImportGrid",
|
|
"Terrain",
|
|
"PointsGroup",
|
|
"PVPlantCreateTerrainMesh",
|
|
"PVPlantAreas",
|
|
"SplitArea",
|
|
"TerrainAnalisys",
|
|
"Trenches",
|
|
"PVPlantEarthworks",
|
|
"PVPlantPad",
|
|
"PVPlantRoad",
|
|
"PVPlantManhole",
|
|
#"PVPlantFoundation"
|
|
"GraphTerrainProfile",
|
|
"Trace",
|
|
]
|
|
self.framelist = [
|
|
"RackType",
|
|
"PVPlantRackCheck",
|
|
"Separator",
|
|
"PVPlantPlacement",
|
|
"PVPlantAdjustToTerrain",
|
|
"PVPlantConvertTo",
|
|
"PVArea"
|
|
]
|
|
|
|
self.objectlist = [
|
|
"PVPlantTree",
|
|
"PVPlantBuilding",
|
|
"PVPlantFenceGroup",
|
|
]
|
|
|
|
self.inportExportlist = ["BOQCivil",
|
|
"BOQMechanical",
|
|
"BOQElectrical",
|
|
"Separator",
|
|
"exportDXF",
|
|
#"importDXF",
|
|
"ExportToPVSyst",
|
|
]
|
|
|
|
self.electricalList = ["PVPlantStringBox",
|
|
"PVPlantCable",
|
|
"PVPlanElectricalLine",
|
|
"Conduit",
|
|
"Stringing",
|
|
"Separator",
|
|
"StringInverter",
|
|
]
|
|
|
|
self.roads = ["PVPlantRoad",
|
|
|
|
]
|
|
|
|
self.pads = ["PVPlantPad",
|
|
"Separator"
|
|
]
|
|
|
|
# Toolbar
|
|
self.appendToolbar("Civil", self.projectlist) # creates a new toolbar with your commands
|
|
self.appendToolbar("PVPlant", self.framelist) # creates a new toolbar with your commands
|
|
self.appendToolbar("Shadow", self.objectlist) # creates a new toolbar with your commands
|
|
self.appendToolbar("Outputs", self.inportExportlist) # creates a new toolbar with your commands
|
|
self.appendToolbar("Electrical", self.electricalList) # creates a new toolbar with your commands
|
|
|
|
# Menu
|
|
self.appendMenu("&Civil", self.projectlist) # creates a new menu
|
|
self.appendMenu("&PVPlant", self.framelist) # creates a new menu
|
|
self.appendMenu("&Shadow", self.objectlist) # creates a new menu
|
|
self.appendMenu("&Outputs", self.inportExportlist) # creates a new menu
|
|
self.appendMenu("&Electrical", self.electricalList) # creates a new menu
|
|
|
|
# Draft tools
|
|
from DraftTools import translate
|
|
self.drafttools = ["Draft_Line","Draft_Wire","Draft_Circle","Draft_Arc","Draft_Ellipse",
|
|
"Draft_Polygon","Draft_Rectangle", "Draft_Text",
|
|
"Draft_Dimension", "Draft_BSpline","Draft_Point",
|
|
"Draft_Facebinder","Draft_BezCurve","Draft_Label"]
|
|
self.draftmodtools = ["Draft_Move","Draft_Rotate","Draft_Offset",
|
|
"Draft_Trimex", "Draft_Upgrade", "Draft_Downgrade", "Draft_Scale",
|
|
"Draft_Shape2DView","Draft_Draft2Sketch","Draft_Array",
|
|
"Draft_Clone"]
|
|
self.draftextratools = ["Draft_WireToBSpline","Draft_ShapeString",
|
|
"Draft_PathArray","Draft_Mirror","Draft_Stretch"]
|
|
self.draftcontexttools = ["Draft_ApplyStyle","Draft_ToggleDisplayMode","Draft_AddToGroup","Draft_AutoGroup",
|
|
"Draft_SelectGroup","Draft_SelectPlane",
|
|
"Draft_ShowSnapBar","Draft_ToggleGrid",]
|
|
self.draftutils = ["Draft_Heal","Draft_FlipDimension",
|
|
"Draft_ToggleConstructionMode","Draft_ToggleContinueMode","Draft_Edit",
|
|
"Draft_Slope","Draft_AddConstruction"]
|
|
self.snapList = ['Draft_Snap_Lock','Draft_Snap_Midpoint','Draft_Snap_Perpendicular',
|
|
'Draft_Snap_Grid','Draft_Snap_Intersection','Draft_Snap_Parallel',
|
|
'Draft_Snap_Endpoint','Draft_Snap_Angle','Draft_Snap_Center',
|
|
'Draft_Snap_Extension','Draft_Snap_Near','Draft_Snap_Ortho','Draft_Snap_Special',
|
|
'Draft_Snap_Dimensions','Draft_Snap_WorkingPlane']
|
|
|
|
|
|
def QT_TRANSLATE_NOOP(scope, text): return text
|
|
self.appendToolbar(QT_TRANSLATE_NOOP("Workbench", "Draft tools"), self.drafttools)
|
|
self.appendToolbar(QT_TRANSLATE_NOOP("Workbench", "Draft mod tools"), self.draftmodtools)
|
|
self.appendMenu(QT_TRANSLATE_NOOP("arch", "&Draft"), self.drafttools + self.draftmodtools + self.draftextratools)
|
|
self.appendMenu([QT_TRANSLATE_NOOP("arch", "&Draft"), QT_TRANSLATE_NOOP("arch", "Utilities")], self.draftutils + self.draftcontexttools)
|
|
self.appendMenu([QT_TRANSLATE_NOOP("arch", "&Draft"), QT_TRANSLATE_NOOP("arch", "Snapping")], self.snapList)
|
|
|
|
import Part
|
|
self.measureTools = ["Part_Measure_Linear",
|
|
"Part_Measure_Angular",
|
|
"Separator",
|
|
"Part_Measure_Refresh",
|
|
"Part_Measure_Clear_All",
|
|
"Part_Measure_Toggle_All",
|
|
"Part_Measure_Toggle_3D",
|
|
"Part_Measure_Toggle_Delta"
|
|
]
|
|
self.appendToolbar("Medir", self.measureTools)
|
|
self.appendMenu("&Medir", self.measureTools)
|
|
|
|
self.observer = None
|
|
|
|
from widgets import CountSelection
|
|
|
|
def Activated(self):
|
|
"This function is executed when the workbench is activated"
|
|
import SelectionObserver
|
|
import FreeCADGui
|
|
|
|
self.observer = SelectionObserver.SelObserver()
|
|
FreeCADGui.Selection.addObserver(self.observer) # installe la fonction en mode resident
|
|
return
|
|
|
|
def Deactivated(self):
|
|
"This function is executed when the workbench is deactivated"
|
|
FreeCADGui.Selection.removeObserver(self.observer)
|
|
return
|
|
|
|
def ContextMenu(self, recipient):
|
|
"This is executed whenever the user right-clicks on screen"
|
|
# "recipient" will be either "view" or "tree"
|
|
|
|
#if FreeCAD.activeDraftCommand is None:
|
|
if recipient.lower() == "view":
|
|
print("Menus en la 'View'")
|
|
#if FreeCAD.activeDraftCommand is None:
|
|
presel = FreeCADGui.Selection.getPreselection()
|
|
print(presel.SubElementNames, " - ", presel.PickedPoints)
|
|
if not presel is None:
|
|
if presel.Object.Proxy.Type == "Road":
|
|
self.appendContextMenu("Road", self.roads)
|
|
elif presel.Object.Proxy.Type == "Pad":
|
|
self.appendContextMenu("Pad", self.pads)
|
|
|
|
'''
|
|
self.contextMenu = QtGui.QMenu()
|
|
menu_item_remove_selected = self.contextMenu.addAction("Remove selected geometry")
|
|
menu_item_remove_all = self.contextMenu.addAction("Clear list")
|
|
if not self.references:
|
|
menu_item_remove_selected.setDisabled(True)
|
|
menu_item_remove_all.setDisabled(True)
|
|
self.connect(
|
|
menu_item_remove_selected,
|
|
QtCore.SIGNAL("triggered()"),
|
|
self.remove_selected_reference
|
|
)
|
|
self.connect(
|
|
menu_item_remove_all,
|
|
QtCore.SIGNAL("triggered()"),
|
|
self.remove_all_references
|
|
)
|
|
parentPosition = self.list_References.mapToGlobal(QtCore.QPoint(0, 0))
|
|
self.contextMenu.move(parentPosition + QPos)
|
|
self.contextMenu.show()
|
|
'''
|
|
|
|
def GetClassName(self):
|
|
# this function is mandatory if this is a full python workbench
|
|
return "Gui::PythonWorkbench"
|
|
|
|
|
|
Gui.addWorkbench(PVPlantWorkbench())
|