159 lines
5.9 KiB
Python
159 lines
5.9 KiB
Python
# IfcOpenShell - IFC toolkit and geometry engine
|
|
# Copyright (C) 2021 Thomas Krijnen <thomas@aecgeeks.com>
|
|
#
|
|
# This file is part of IfcOpenShell.
|
|
#
|
|
# IfcOpenShell 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 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# IfcOpenShell 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 IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import logging
|
|
import sys
|
|
from code import InteractiveConsole
|
|
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
|
try:
|
|
from PyQt5 import QtWidgets
|
|
except BaseException:
|
|
QtWidgets = QtGui
|
|
|
|
try:
|
|
from pyqode.core import api, modes, panels
|
|
from pyqode.core.api import CodeEdit
|
|
from pyqode.python import modes as pymodes
|
|
from pyqode.python import panels as pypanels
|
|
from pyqode.python.backend import server
|
|
from pyqode.python.modes import PythonSH
|
|
|
|
has_pyqode = True
|
|
except BaseException:
|
|
has_pyqode = False
|
|
CodeEdit = QtWidgets.QPlainTextEdit
|
|
|
|
|
|
class StdoutRedirector:
|
|
"""A class for redirecting stdout to this Text widget."""
|
|
|
|
def __init__(self, widget):
|
|
self.widget = widget
|
|
self.isError = False
|
|
|
|
def write(self, myStr):
|
|
self.widget.moveCursor(QtGui.QTextCursor.End)
|
|
if self.isError:
|
|
self.widget.setTextColor(QtCore.Qt.red)
|
|
else:
|
|
self.widget.setTextColor(QtCore.Qt.white)
|
|
self.widget.insertPlainText(myStr)
|
|
self.widget.moveCursor(QtGui.QTextCursor.End)
|
|
|
|
|
|
class code_edit(QtWidgets.QWidget):
|
|
class Console(InteractiveConsole):
|
|
def __init__(*args):
|
|
InteractiveConsole.__init__(*args)
|
|
|
|
def enter(self, source):
|
|
self.runcode(source)
|
|
|
|
def runCode(self):
|
|
sys.stdout = StdoutRedirector(self.output)
|
|
sys.stderr = StdoutRedirector(self.output)
|
|
sys.stderr.isError = True
|
|
|
|
if not self.model:
|
|
print("please load a model first", file=sys.stderr)
|
|
else:
|
|
self.c.enter(str(self.editor.toPlainText()))
|
|
|
|
sys.stdout = sys.__stdout__
|
|
sys.stderr = sys.__stderr__
|
|
|
|
def select(self, product):
|
|
self.c = self.Console({"model": self.model, "viewer": self.viewer, "selection": product})
|
|
|
|
def __init__(self, viewer, snippets=None):
|
|
self.model = None
|
|
self.viewer = viewer
|
|
QtWidgets.QWidget.__init__(self)
|
|
self.layout = QtWidgets.QVBoxLayout(self)
|
|
self.setLayout(self.layout)
|
|
self.c = None
|
|
self.tools = QtWidgets.QHBoxLayout(self)
|
|
self.layout.addLayout(self.tools)
|
|
self.runbutton = QtWidgets.QPushButton("Run")
|
|
width = self.runbutton.fontMetrics().boundingRect("Run").width() + 20
|
|
self.runbutton.setMaximumWidth(width)
|
|
self.tools.addWidget(self.runbutton)
|
|
self.runbutton.clicked.connect(self.runCode)
|
|
|
|
editor = CodeEdit()
|
|
if has_pyqode:
|
|
editor.backend.start(server.__file__)
|
|
editor.panels.append(panels.FoldingPanel())
|
|
editor.panels.append(panels.LineNumberPanel())
|
|
editor.panels.append(panels.SearchAndReplacePanel(), panels.SearchAndReplacePanel.Position.BOTTOM)
|
|
editor.panels.append(panels.EncodingPanel(), api.Panel.Position.TOP)
|
|
editor.add_separator()
|
|
editor.panels.append(pypanels.QuickDocPanel(), api.Panel.Position.BOTTOM)
|
|
sh = editor.modes.append(PythonSH(editor.document()))
|
|
editor.modes.append(modes.CaretLineHighlighterMode())
|
|
editor.modes.append(modes.CodeCompletionMode())
|
|
editor.modes.append(modes.ExtendedSelectionMode())
|
|
editor.modes.append(modes.FileWatcherMode())
|
|
editor.modes.append(modes.OccurrencesHighlighterMode())
|
|
editor.modes.append(modes.RightMarginMode())
|
|
editor.modes.append(modes.SmartBackSpaceMode())
|
|
editor.modes.append(modes.SymbolMatcherMode())
|
|
editor.modes.append(modes.ZoomMode())
|
|
editor.modes.append(pymodes.CommentsMode())
|
|
editor.modes.append(pymodes.CalltipsMode())
|
|
auto = pymodes.PyAutoCompleteMode()
|
|
auto.logger.setLevel(logging.CRITICAL)
|
|
editor.modes.append(auto)
|
|
editor.modes.append(pymodes.PyAutoIndentMode())
|
|
editor.modes.append(pymodes.PyIndenterMode())
|
|
editor.show()
|
|
else:
|
|
editor.setStyleSheet("font-size: 10pt; font-family: Consolas, Courier;")
|
|
|
|
self.editor = editor
|
|
self.snippets = snippets
|
|
if self.snippets:
|
|
self.list = QtWidgets.QComboBox(self)
|
|
self.replace_snippet(0)
|
|
for snip_name in self.snippets.keys():
|
|
self.list.addItem(snip_name)
|
|
self.tools.addWidget(self.list)
|
|
self.list.currentIndexChanged[int].connect(self.replace_snippet)
|
|
|
|
self.layout.addWidget(self.editor)
|
|
self.output = QtWidgets.QTextEdit()
|
|
self.output.setReadOnly(True)
|
|
self.output.setStyleSheet("font-size: 10pt; font-family: Consolas, Courier; background-color: #444;")
|
|
self.layout.addWidget(self.output)
|
|
|
|
def replace_snippet(self, number=None):
|
|
snip = list(self.snippets.values())[number]
|
|
if has_pyqode:
|
|
self.editor.setPlainText(snip, "", "")
|
|
else:
|
|
self.editor.setPlainText(snip)
|
|
|
|
def load_file(self, f, **kwargs):
|
|
output = []
|
|
sys.stdout = StdoutRedirector(self.output)
|
|
self.model = f
|
|
self.c = self.Console({"model": self.model, "selection": None, "viewer": self.viewer})
|
|
sys.stdout = sys.__stdout__
|