refactor: excel parse
This commit is contained in:
Blizzard
@@ -0,0 +1,297 @@
|
||||
# Copyright (c) 2022, Manfred Moitzi
|
||||
# License: MIT License
|
||||
from __future__ import annotations
|
||||
from typing import Iterator, Iterable, Optional
|
||||
import ezdxf
|
||||
from ezdxf.addons.xqt import (
|
||||
QtWidgets,
|
||||
QtGui,
|
||||
QAction,
|
||||
QMessageBox,
|
||||
QFileDialog,
|
||||
Qt,
|
||||
QModelIndex,
|
||||
)
|
||||
from ezdxf.document import Drawing
|
||||
from ezdxf.entities import Body
|
||||
from ezdxf.lldxf.const import DXFStructureError
|
||||
from .data import AcisData, BinaryAcisData, TextAcisData
|
||||
|
||||
|
||||
APP_NAME = "ACIS Structure Browser"
|
||||
BROWSER_WIDTH = 1024
|
||||
BROWSER_HEIGHT = 768
|
||||
SELECTOR_WIDTH_FACTOR = 0.20
|
||||
FONT_FAMILY = "monospaced"
|
||||
|
||||
|
||||
def make_font():
|
||||
font = QtGui.QFont(FONT_FAMILY)
|
||||
font.setStyleHint(QtGui.QFont.Monospace)
|
||||
return font
|
||||
|
||||
|
||||
class AcisStructureBrowser(QtWidgets.QMainWindow):
|
||||
def __init__(
|
||||
self,
|
||||
filename: str = "",
|
||||
handle: str = "",
|
||||
):
|
||||
super().__init__()
|
||||
self.doc: Optional[Drawing] = None
|
||||
self.acis_entities: list[AcisData] = []
|
||||
self.current_acis_entity = AcisData()
|
||||
self.entity_selector = self.make_entity_selector()
|
||||
self.acis_content_viewer = self.make_content_viewer()
|
||||
self.statusbar = QtWidgets.QStatusBar(self)
|
||||
self.setup_actions()
|
||||
self.setup_menu()
|
||||
|
||||
if filename:
|
||||
self.load_dxf(filename)
|
||||
else:
|
||||
self.setWindowTitle(APP_NAME)
|
||||
|
||||
self.setStatusBar(self.statusbar)
|
||||
self.setCentralWidget(self.make_central_widget())
|
||||
self.resize(BROWSER_WIDTH, BROWSER_HEIGHT)
|
||||
self.connect_slots()
|
||||
if handle:
|
||||
try:
|
||||
int(handle, 16)
|
||||
except ValueError:
|
||||
msg = f"Given handle is not a hex value: '{handle}'"
|
||||
self.statusbar.showMessage(msg)
|
||||
print(msg)
|
||||
else:
|
||||
if not self.goto_handle(handle):
|
||||
msg = f"Handle '{handle}' not found."
|
||||
self.statusbar.showMessage(msg)
|
||||
print(msg)
|
||||
|
||||
def make_entity_selector(self):
|
||||
return QtWidgets.QListWidget(self)
|
||||
|
||||
def make_content_viewer(self):
|
||||
viewer = QtWidgets.QPlainTextEdit(self)
|
||||
viewer.setReadOnly(True)
|
||||
viewer.setLineWrapMode(QtWidgets.QPlainTextEdit.NoWrap)
|
||||
return viewer
|
||||
|
||||
def make_central_widget(self):
|
||||
container = QtWidgets.QSplitter(Qt.Horizontal)
|
||||
container.addWidget(self.entity_selector)
|
||||
container.addWidget(self.acis_content_viewer)
|
||||
selector_width = int(BROWSER_WIDTH * SELECTOR_WIDTH_FACTOR)
|
||||
entity_view_width = BROWSER_WIDTH - selector_width
|
||||
container.setSizes([selector_width, entity_view_width])
|
||||
container.setCollapsible(0, False)
|
||||
container.setCollapsible(1, False)
|
||||
return container
|
||||
|
||||
def connect_slots(self):
|
||||
self.entity_selector.clicked.connect(self.acis_entity_activated)
|
||||
self.entity_selector.activated.connect(self.acis_entity_activated)
|
||||
|
||||
# noinspection PyAttributeOutsideInit
|
||||
def setup_actions(self):
|
||||
self._open_action = self.make_action(
|
||||
"&Open DXF File...", self.open_dxf, shortcut="Ctrl+O"
|
||||
)
|
||||
self._reload_action = self.make_action(
|
||||
"Reload DXF File",
|
||||
self.reload_dxf,
|
||||
shortcut="Ctrl+R",
|
||||
)
|
||||
self._export_entity_action = self.make_action(
|
||||
"&Export Current Entity View...",
|
||||
self.export_entity,
|
||||
shortcut="Ctrl+E",
|
||||
)
|
||||
self._export_raw_data_action = self.make_action(
|
||||
"&Export Raw SAT/SAB Data...",
|
||||
self.export_raw_entity,
|
||||
shortcut="Ctrl+W",
|
||||
)
|
||||
self._quit_action = self.make_action(
|
||||
"&Quit", self.close, shortcut="Ctrl+Q"
|
||||
)
|
||||
|
||||
def make_action(
|
||||
self,
|
||||
name,
|
||||
slot,
|
||||
*,
|
||||
shortcut: str = "",
|
||||
tip: str = "",
|
||||
) -> QAction:
|
||||
action = QAction(name, self)
|
||||
if shortcut:
|
||||
action.setShortcut(shortcut)
|
||||
if tip:
|
||||
action.setToolTip(tip)
|
||||
action.triggered.connect(slot)
|
||||
return action
|
||||
|
||||
def setup_menu(self):
|
||||
menu = self.menuBar()
|
||||
file_menu = menu.addMenu("&File")
|
||||
file_menu.addAction(self._open_action)
|
||||
file_menu.addAction(self._reload_action)
|
||||
file_menu.addSeparator()
|
||||
file_menu.addAction(self._export_entity_action)
|
||||
file_menu.addAction(self._export_raw_data_action)
|
||||
file_menu.addSeparator()
|
||||
file_menu.addAction(self._quit_action)
|
||||
|
||||
def open_dxf(self):
|
||||
path, _ = QtWidgets.QFileDialog.getOpenFileName(
|
||||
self,
|
||||
caption="Select DXF file",
|
||||
filter="DXF Documents (*.dxf *.DXF)",
|
||||
)
|
||||
if path:
|
||||
self.load_dxf(path)
|
||||
|
||||
def load_dxf(self, path: str):
|
||||
try:
|
||||
doc = ezdxf.readfile(path)
|
||||
except IOError as e:
|
||||
QMessageBox.critical(self, "Loading Error", str(e))
|
||||
return
|
||||
except DXFStructureError as e:
|
||||
QMessageBox.critical(
|
||||
self,
|
||||
"DXF Structure Error",
|
||||
f'Invalid DXF file "{path}": {str(e)}',
|
||||
)
|
||||
return
|
||||
entities = list(get_acis_entities(doc))
|
||||
if len(entities):
|
||||
self.doc = doc
|
||||
self.set_acis_entities(entities)
|
||||
self.update_title(path)
|
||||
self.statusbar.showMessage(self.make_loading_message())
|
||||
else:
|
||||
msg = f"DXF file '{path}' contains no ACIS data"
|
||||
QMessageBox.information(self, "Loading Error", msg) # type: ignore
|
||||
|
||||
def make_loading_message(self) -> str:
|
||||
assert self.doc is not None
|
||||
dxfversion = self.doc.dxfversion
|
||||
acis_type = "SAB" if dxfversion >= "AC1027" else "SAT"
|
||||
return f"Loaded DXF file has version {self.doc.acad_release}/{dxfversion}" \
|
||||
f" and contains {acis_type} data"
|
||||
|
||||
def set_acis_entities(self, entities: list[AcisData]):
|
||||
self.acis_entities = entities
|
||||
self.update_entity_selector(entities)
|
||||
self.set_current_acis_entity(entities[0])
|
||||
|
||||
def reload_dxf(self):
|
||||
try:
|
||||
index = self.acis_entities.index(self.current_acis_entity)
|
||||
except IndexError:
|
||||
index = -1
|
||||
self.load_dxf(self.doc.filename)
|
||||
if index > 0:
|
||||
self.set_current_acis_entity(self.acis_entities[index])
|
||||
|
||||
def export_entity(self):
|
||||
dxf_entity = self.get_current_dxf_entity()
|
||||
if dxf_entity is None:
|
||||
return
|
||||
path, _ = QFileDialog.getSaveFileName(
|
||||
self,
|
||||
caption="Export Current Entity View",
|
||||
dir=f"{dxf_entity.dxftype()}-{dxf_entity.dxf.handle}.txt",
|
||||
filter="Text Files (*.txt *.TXT)",
|
||||
)
|
||||
if path:
|
||||
write_data(self.current_acis_entity, path)
|
||||
|
||||
def export_raw_entity(self):
|
||||
dxf_entity = self.get_current_dxf_entity()
|
||||
if dxf_entity is None:
|
||||
return
|
||||
filename = f"{dxf_entity.dxftype()}-{dxf_entity.dxf.handle}"
|
||||
sab = dxf_entity.has_binary_data
|
||||
if sab:
|
||||
filter_ = "Standard ACIS Binary Files (*.sab *.SAB)"
|
||||
filename += ".sab"
|
||||
else:
|
||||
filter_ = "Standard ACIS Text Files (*.sat *.SAT)"
|
||||
filename += ".sat"
|
||||
|
||||
path, _ = QFileDialog.getSaveFileName(
|
||||
self,
|
||||
caption="Export ACIS Raw Data",
|
||||
dir=filename,
|
||||
filter=filter_,
|
||||
)
|
||||
if path:
|
||||
if sab:
|
||||
with open(path, "wb") as fp:
|
||||
fp.write(dxf_entity.sab)
|
||||
else:
|
||||
with open(path, "wt") as fp:
|
||||
fp.write("\n".join(dxf_entity.sat))
|
||||
|
||||
def get_current_dxf_entity(self) -> Optional[Body]:
|
||||
current = self.current_acis_entity
|
||||
if not current.handle or self.doc is None:
|
||||
return None
|
||||
return self.doc.entitydb.get(current.handle) # type: ignore
|
||||
|
||||
def update_title(self, path: str):
|
||||
self.setWindowTitle(f"{APP_NAME} - {path}")
|
||||
|
||||
def acis_entity_activated(self, index: QModelIndex):
|
||||
if len(self.acis_entities) == 0:
|
||||
return
|
||||
try:
|
||||
self.set_current_acis_entity(self.acis_entities[index.row()])
|
||||
except IndexError:
|
||||
self.set_current_acis_entity(self.acis_entities[0])
|
||||
|
||||
def set_current_acis_entity(self, entity: AcisData):
|
||||
if entity:
|
||||
self.current_acis_entity = entity
|
||||
self.update_acis_content_viewer(entity)
|
||||
|
||||
def update_acis_content_viewer(self, entity: AcisData):
|
||||
viewer = self.acis_content_viewer
|
||||
viewer.clear()
|
||||
viewer.setPlainText("\n".join(entity.lines))
|
||||
|
||||
def update_entity_selector(self, entities: Iterable[AcisData]):
|
||||
viewer = self.entity_selector
|
||||
viewer.clear()
|
||||
viewer.addItems([e.name for e in entities])
|
||||
|
||||
def goto_handle(self, handle: str) -> bool:
|
||||
for entity in self.acis_entities:
|
||||
if entity.handle == handle:
|
||||
self.set_current_acis_entity(entity)
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def get_acis_entities(doc: Drawing) -> Iterator[AcisData]:
|
||||
for e in doc.entitydb.values():
|
||||
if isinstance(e, Body):
|
||||
handle = e.dxf.handle
|
||||
name = f"<{handle}> {e.dxftype()}"
|
||||
if e.has_binary_data:
|
||||
yield BinaryAcisData(e.sab, name, handle)
|
||||
else:
|
||||
yield TextAcisData(e.sat, name, handle)
|
||||
|
||||
|
||||
def write_data(entity: AcisData, path: str):
|
||||
try:
|
||||
with open(path, "wt") as fp:
|
||||
fp.write("\n".join(entity.lines))
|
||||
except IOError:
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user