You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

173 lines
6.6 KiB
Python

# SPDX-License-Identifier: LGPL-2.1-or-later
# ***************************************************************************
# * *
# * Copyright (c) 2024 Yorik van Havre <yorik@uncreated.net> *
# * *
# * This file is part of FreeCAD. *
# * *
# * FreeCAD is free software: you can redistribute it and/or modify it *
# * under the terms of the GNU Lesser General Public License as *
# * published by the Free Software Foundation, either version 2.1 of the *
# * License, or (at your option) any later version. *
# * *
# * FreeCAD 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 *
# * Lesser General Public License for more details. *
# * *
# * You should have received a copy of the GNU Lesser General Public *
# * License along with FreeCAD. If not, see *
# * <https://www.gnu.org/licenses/>. *
# * *
# ***************************************************************************
"""BIM Panel-related Arch_"""
import FreeCAD
import FreeCADGui
QT_TRANSLATE_NOOP = FreeCAD.Qt.QT_TRANSLATE_NOOP
translate = FreeCAD.Qt.translate
PARAMS = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/BIM")
class Arch_Profile:
"""The FreeCAD Arch_Profile command definition"""
def GetResources(self):
return {
"Pixmap": "Arch_Profile",
"MenuText": QT_TRANSLATE_NOOP("Arch_Profile", "Profile"),
"Accel": "P, F",
"ToolTip": QT_TRANSLATE_NOOP("Arch_Profile", "Creates a profile"),
}
def IsActive(self):
v = hasattr(FreeCADGui.getMainWindow().getActiveWindow(), "getSceneGraph")
return v
def Activated(self):
import ArchProfile
FreeCAD.activeDraftCommand = self # register as a Draft command for auto grid on/off
self.doc = FreeCAD.ActiveDocument
self.Profile = None
self.Categories = []
self.Presets = ArchProfile.readPresets()
for pre in self.Presets:
if pre[1] not in self.Categories:
self.Categories.append(pre[1])
FreeCADGui.Snapper.getPoint(
callback=self.getPoint,
extradlg=[self.taskbox()],
title=translate("Arch", "Create profile"),
)
def taskbox(self):
"sets up a taskbox widget"
from PySide import QtGui
from draftutils import params
w = QtGui.QWidget()
ui = FreeCADGui.UiLoader()
w.setWindowTitle(translate("Arch", "Profile settings"))
grid = QtGui.QGridLayout(w)
# categories box
labelc = QtGui.QLabel(translate("Arch", "Category"))
self.vCategory = QtGui.QComboBox()
self.vCategory.addItems(self.Categories)
grid.addWidget(labelc, 1, 0, 1, 1)
grid.addWidget(self.vCategory, 1, 1, 1, 1)
# presets box
labelp = QtGui.QLabel(translate("Arch", "Preset"))
self.vPresets = QtGui.QComboBox()
self.pSelect = [None]
fpresets = [" "]
self.vPresets.addItems(fpresets)
grid.addWidget(labelp, 2, 0, 1, 1)
grid.addWidget(self.vPresets, 2, 1, 1, 1)
# restore preset
categoryIdx = -1
presetIdx = -1
stored = params.get_param_arch("ProfilePreset")
if stored and ";" in stored and len(stored.split(";")) >= 3:
stored = stored.split(";")
if stored[1] in self.Categories:
categoryIdx = self.Categories.index(stored[1])
self.setCategory(categoryIdx)
self.vCategory.setCurrentIndex(categoryIdx)
ps = [p[2] for p in self.pSelect]
if stored[2] in ps:
presetIdx = ps.index(stored[2])
self.setPreset(presetIdx)
self.vPresets.setCurrentIndex(presetIdx)
if categoryIdx == -1:
self.setCategory(0)
self.vCategory.setCurrentIndex(0)
if presetIdx == -1:
self.setPreset(0)
self.vPresets.setCurrentIndex(0)
# connect slots
self.vCategory.currentIndexChanged.connect(self.setCategory)
self.vPresets.currentIndexChanged.connect(self.setPreset)
return w
def getPoint(self, point=None, obj=None):
"this function is called by the snapper when it has a 3D point"
FreeCAD.activeDraftCommand = None
FreeCADGui.Snapper.off()
if not point:
return
if not self.Profile:
return
pt = "FreeCAD.Vector(" + str(point.x) + "," + str(point.y) + "," + str(point.z) + ")"
self.doc.openTransaction(translate("Arch", "Create Profile"))
FreeCADGui.addModule("Arch")
FreeCADGui.doCommand("p = Arch.makeProfile(" + str(self.Profile) + ")")
FreeCADGui.addModule("WorkingPlane")
FreeCADGui.doCommand("p.Placement = WorkingPlane.get_working_plane().get_placement()")
FreeCADGui.doCommand("p.Placement.Base = " + pt)
FreeCADGui.addModule("Draft")
FreeCADGui.doCommand("Draft.autogroup(p)")
self.doc.commitTransaction()
self.doc.recompute()
def setCategory(self, i):
from draftutils import params
self.vPresets.clear()
self.pSelect = [p for p in self.Presets if p[1] == self.Categories[i]]
fpresets = []
for p in self.pSelect:
f = FreeCAD.Units.Quantity(p[4], FreeCAD.Units.Length).getUserPreferred()
d = params.get_param("Decimals", path="Units")
s1 = str(round(p[4] / f[1], d))
s2 = str(round(p[5] / f[1], d))
s3 = str(f[2])
fpresets.append(p[2] + " (" + s1 + "x" + s2 + s3 + ")")
self.vPresets.addItems(fpresets)
self.setPreset(0)
def setPreset(self, i):
from draftutils import params
self.Profile = self.pSelect[i]
params.set_param_arch("ProfilePreset", ";".join([str(i) for i in self.Profile]))
FreeCADGui.addCommand("Arch_Profile", Arch_Profile())