mirror of https://gitlab.com/pamhyr/pamhyr2
154 lines
4.4 KiB
Python
154 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from View.ASubWindow import ASubMainWindow
|
|
from View.ListedSubWindow import ListedSubWindow
|
|
|
|
from PyQt5.QtGui import (
|
|
QKeySequence,
|
|
)
|
|
|
|
from PyQt5.QtCore import (
|
|
Qt, QVariant, QAbstractTableModel, QCoreApplication,
|
|
)
|
|
|
|
from PyQt5.QtWidgets import (
|
|
QDialogButtonBox, QPushButton, QLineEdit,
|
|
QFileDialog, QTableView, QAbstractItemView,
|
|
QUndoStack, QShortcut, QAction, QItemDelegate,
|
|
QHeaderView,
|
|
)
|
|
|
|
from View.Plot.MplCanvas import MplCanvas
|
|
|
|
from View.BoundaryCondition.translate import long_types
|
|
from View.BoundaryCondition.Edit.Table import TableModel
|
|
from View.BoundaryCondition.Edit.Plot import Plot
|
|
|
|
_translate = QCoreApplication.translate
|
|
|
|
class EditBoundaryConditionWindow(ASubMainWindow, ListedSubWindow):
|
|
def __init__(self, title="Edit BoundaryConditions", data=None, parent=None):
|
|
super(EditBoundaryConditionWindow, self).__init__(
|
|
name=title, ui="EditBoundaryConditions", parent=parent
|
|
)
|
|
|
|
self._data = data
|
|
self._title = title
|
|
|
|
self.setup_window()
|
|
self.setup_sc()
|
|
self.setup_table()
|
|
self.setup_plot()
|
|
self.setup_connections()
|
|
|
|
def setup_window(self):
|
|
if self._data is not None:
|
|
node_name = (self._data.node.name if self._data.node is not None
|
|
else _translate("BoundaryCondition", "Not associate"))
|
|
title = (
|
|
_translate("BoundaryCondition", self._title) +
|
|
f" - {self._data.name} " +
|
|
f"({long_types[self._data.bctype]} - {node_name})"
|
|
)
|
|
self.ui.setWindowTitle(title)
|
|
else:
|
|
self.ui.setWindowTitle(_translate("BoundaryCondition", self._title))
|
|
|
|
def setup_sc(self):
|
|
self._undo_stack = QUndoStack()
|
|
|
|
self.undo_sc = QShortcut(QKeySequence.Undo, self)
|
|
self.redo_sc = QShortcut(QKeySequence.Redo, self)
|
|
self.copy_sc = QShortcut(QKeySequence.Copy, self)
|
|
self.paste_sc = QShortcut(QKeySequence.Paste, self)
|
|
|
|
def setup_table(self):
|
|
table = self.find(QTableView, "tableView")
|
|
self._table = TableModel(
|
|
data = self._data,
|
|
undo = self._undo_stack
|
|
)
|
|
table.setModel(self._table)
|
|
table.setSelectionBehavior(QAbstractItemView.SelectRows)
|
|
table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
|
|
table.setAlternatingRowColors(True)
|
|
|
|
def setup_plot(self):
|
|
self.canvas = MplCanvas(width=5, height=4, dpi=100)
|
|
self.canvas.setObjectName("canvas")
|
|
self.verticalLayout.addWidget(self.canvas)
|
|
|
|
self.plot = Plot(
|
|
canvas = self.canvas,
|
|
data = self._data,
|
|
)
|
|
self.plot.draw()
|
|
|
|
|
|
def setup_connections(self):
|
|
self.find(QAction, "action_add").triggered.connect(self.add)
|
|
self.find(QAction, "action_del").triggered.connect(self.delete)
|
|
self.find(QAction, "action_sort").triggered.connect(self.sort)
|
|
|
|
self.undo_sc.activated.connect(self.undo)
|
|
self.redo_sc.activated.connect(self.redo)
|
|
self.copy_sc.activated.connect(self.copy)
|
|
self.paste_sc.activated.connect(self.paste)
|
|
|
|
def index_selected_row(self):
|
|
table = self.find(QTableView, "tableView")
|
|
return table.selectionModel()\
|
|
.selectedRows()[0]\
|
|
.row()
|
|
|
|
def index_selected_rows(self):
|
|
table = self.find(QTableView, "tableView")
|
|
return list(
|
|
# Delete duplicate
|
|
set(
|
|
map(
|
|
lambda i: i.row(),
|
|
table.selectedIndexes()
|
|
)
|
|
)
|
|
)
|
|
|
|
|
|
def add(self):
|
|
rows = self.index_selected_rows()
|
|
if len(self._data) == 0 or len(rows) == 0:
|
|
self._table.add(0)
|
|
else:
|
|
self._table.add(rows[0])
|
|
|
|
def delete(self):
|
|
rows = self.index_selected_rows()
|
|
if len(rows) == 0:
|
|
return
|
|
|
|
self._table.delete(rows)
|
|
|
|
def sort(self):
|
|
self._table.sort(False)
|
|
|
|
def move_up(self):
|
|
row = self.index_selected_row()
|
|
self._table.move_up(row)
|
|
|
|
def move_down(self):
|
|
row = self.index_selected_row()
|
|
self._table.move_down(row)
|
|
|
|
|
|
def copy(self):
|
|
print("TODO")
|
|
|
|
def paste(self):
|
|
print("TODO")
|
|
|
|
def undo(self):
|
|
self._table.undo()
|
|
|
|
def redo(self):
|
|
self._table.redo()
|