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.
305 lines
11 KiB
Python
305 lines
11 KiB
Python
2 years ago
|
import queue
|
||
|
|
import typing
|
||
|
|
|
||
|
|
from PyQt5 import QtCore
|
||
|
|
from PyQt5.QtCore import QAbstractTableModel, QModelIndex, Qt, QVariant, QAbstractListModel
|
||
|
|
from PyQt5.QtGui import QColor, QBrush
|
||
|
|
|
||
|
|
# stack = Stack()
|
||
|
|
from Agreement.CS.skio.worker.iomapping import get_bit_val
|
||
|
|
|
||
|
|
# from xps.myStack import Stack
|
||
|
|
|
||
|
|
que = queue.LifoQueue()
|
||
|
|
que1 = queue.LifoQueue()
|
||
|
|
|
||
|
|
|
||
|
|
class myTableModel(QAbstractTableModel):
|
||
|
|
def __init__(self, header, data: list):
|
||
|
|
QAbstractTableModel.__init__(self, parent=None)
|
||
|
|
self.datas = data
|
||
|
|
self.header = header
|
||
|
|
|
||
|
|
def append_data(self, x):
|
||
|
|
self.datas.append(x)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def remove_row(self, row):
|
||
|
|
self.datas.pop(row)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def rowCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
if len(self.datas) > 0:
|
||
|
|
return len(self.datas)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def columnCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
return len(self.header)
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def data(self, QModelIndex, role=None):
|
||
|
|
if not QModelIndex.isValid():
|
||
|
|
print("行或者列有问题")
|
||
|
|
return QVariant()
|
||
|
|
elif role != Qt.DisplayRole:
|
||
|
|
return QVariant()
|
||
|
|
return QVariant(self.datas[QModelIndex.row()][QModelIndex.column()])
|
||
|
|
|
||
|
|
def headerData(self, section: int, orientation: Qt.Orientation, role: int = ...) -> typing.Any:
|
||
|
|
if role != Qt.DisplayRole:
|
||
|
|
return None
|
||
|
|
|
||
|
|
if orientation == Qt.Horizontal:
|
||
|
|
return self.header[section]
|
||
|
|
|
||
|
|
|
||
|
|
class checkModel(myTableModel):
|
||
|
|
def __init__(self, header, data):
|
||
|
|
myTableModel.__init__(self, header, data)
|
||
|
|
|
||
|
|
def data(self, QModelIndex, role=None):
|
||
|
|
if not QModelIndex.isValid():
|
||
|
|
print("行或者列有问题")
|
||
|
|
return QVariant()
|
||
|
|
if role == Qt.BackgroundColorRole:
|
||
|
|
if QModelIndex.row() == self.datas[QModelIndex.row()][0] - 1 and self.datas[QModelIndex.row()][-1] == True:
|
||
|
|
return QBrush(QColor(Qt.green))
|
||
|
|
if QModelIndex.row() == self.datas[QModelIndex.row()][0] - 1 and self.datas[QModelIndex.row()][-1] == False:
|
||
|
|
return QBrush(QColor(Qt.red))
|
||
|
|
elif role != Qt.DisplayRole:
|
||
|
|
return QVariant()
|
||
|
|
return QVariant(self.datas[QModelIndex.row()][QModelIndex.column()])
|
||
|
|
|
||
|
|
|
||
|
|
class MyModel(myTableModel):
|
||
|
|
def __init__(self, header, data: list):
|
||
|
|
myTableModel.__init__(self, header, data)
|
||
|
|
self.checkList = ['Unchecked'] * len(self.datas)
|
||
|
|
self.supportedDragActions()
|
||
|
|
|
||
|
|
def data(self, index, role):
|
||
|
|
row = index.row()
|
||
|
|
col = index.column()
|
||
|
|
if role == Qt.DisplayRole:
|
||
|
|
return QVariant(self.datas[row][col])
|
||
|
|
elif role == Qt.CheckStateRole:
|
||
|
|
if col == 0:
|
||
|
|
return Qt.Checked if self.checkList[row] == 'Checked' else Qt.Unchecked
|
||
|
|
return QVariant()
|
||
|
|
|
||
|
|
def setData(self, index, value, role):
|
||
|
|
row = index.row()
|
||
|
|
col = index.column()
|
||
|
|
if role == Qt.CheckStateRole and col == 0:
|
||
|
|
self.checkList[row] = 'Checked' if value == Qt.Checked else 'Unchecked'
|
||
|
|
return True
|
||
|
|
|
||
|
|
def flags(self, index):
|
||
|
|
if index.isValid():
|
||
|
|
return Qt.ItemIsDragEnabled | Qt.ItemIsDropEnabled | Qt.ItemIsEditable | Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
|
||
|
|
return Qt.ItemIsDropEnabled | Qt.ItemIsEditable | Qt.ItemIsEnabled | Qt.ItemIsSelectable
|
||
|
|
|
||
|
|
def headerClick(self, isOn):
|
||
|
|
self.beginResetModel()
|
||
|
|
if isOn:
|
||
|
|
self.checkList = ['Checked'] * len(self.datas)
|
||
|
|
else:
|
||
|
|
self.checkList = ['Unchecked'] * len(self.datas)
|
||
|
|
self.endResetModel()
|
||
|
|
|
||
|
|
def dragMoveEvent(self, event):
|
||
|
|
event.setDropAction(QtCore.Qt.MoveAction)
|
||
|
|
event.accept()
|
||
|
|
|
||
|
|
def moveRow(self, sourceParent: QModelIndex, sourceRow: int, destinationParent: QModelIndex,
|
||
|
|
destinationChild: int) -> bool:
|
||
|
|
if self.datas[destinationChild] == self.datas[sourceRow]:
|
||
|
|
return
|
||
|
|
self.datas[sourceRow], self.datas[destinationChild] = self.datas[destinationChild], self.datas[sourceRow]
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
|
||
|
|
class smallTableModel(QAbstractTableModel):
|
||
|
|
err = QtCore.pyqtSignal(str)
|
||
|
|
|
||
|
|
def __init__(self, header, data):
|
||
|
|
QAbstractTableModel.__init__(self, parent=None)
|
||
|
|
self.datas = data
|
||
|
|
self.header = header
|
||
|
|
self.checkList = [[i, 'Unchecked'] for i in range(len(self.datas))]
|
||
|
|
|
||
|
|
def remove_row(self, row):
|
||
|
|
self.datas.pop(row)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def rowCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
if len(self.datas) > 0:
|
||
|
|
return len(self.datas)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def columnCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
return len(self.header)
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def data(self, QModelIndex, role=None):
|
||
|
|
from utils.core import MainWindowConfig
|
||
|
|
if not QModelIndex.isValid():
|
||
|
|
print("行或者列有问题")
|
||
|
|
return QVariant()
|
||
|
|
if role == Qt.BackgroundColorRole:
|
||
|
|
if MainWindowConfig.IOMapping.force_value_stact[self.datas[QModelIndex.row()]['sig_name']] == True:
|
||
|
|
return QBrush(QColor(Qt.green))
|
||
|
|
if role == Qt.CheckStateRole:
|
||
|
|
if QModelIndex.column() == 0:
|
||
|
|
if self.datas[QModelIndex.row()]['sig_type'] in ('AO', 'DO-24V', 'DO-48V', 'TC/T', 'PT100', 'DO', 'DI', 'TC/K', 'TC/E'):
|
||
|
|
return Qt.Checked if self.checkList[QModelIndex.row()][1] == 'Checked' else Qt.Unchecked
|
||
|
|
if role != Qt.DisplayRole:
|
||
|
|
return QVariant()
|
||
|
|
if role == Qt.DisplayRole:
|
||
|
|
if QModelIndex.column() == 0:
|
||
|
|
return QVariant(MainWindowConfig.IOMapping.force_value[self.datas[QModelIndex.row()]['sig_name']])
|
||
|
|
elif QModelIndex.column() == 1:
|
||
|
|
if self.datas[QModelIndex.row()]['offset'] != None:
|
||
|
|
return QVariant(
|
||
|
|
get_bit_val(
|
||
|
|
MainWindowConfig.IOMapping.current_value[int(self.datas[QModelIndex.row()]['reg']) - 1][
|
||
|
|
int(self.datas[QModelIndex.row()]['channel']) - 1],
|
||
|
|
int(self.datas[QModelIndex.row()]['offset'])))
|
||
|
|
else:
|
||
|
|
return QVariant(
|
||
|
|
MainWindowConfig.IOMapping.current_value[int(self.datas[QModelIndex.row()]['reg']) - 1][
|
||
|
|
int(self.datas[QModelIndex.row()]['channel']) - 1])
|
||
|
|
else:
|
||
|
|
return QVariant(self.datas[QModelIndex.row()][self.header[QModelIndex.column()][0]])
|
||
|
|
|
||
|
|
def setData(self, index: QModelIndex, value: typing.Any, role: int = ...) -> bool:
|
||
|
|
from utils.core import MainWindowConfig
|
||
|
|
row = index.row()
|
||
|
|
col = index.column()
|
||
|
|
if role == Qt.CheckStateRole and col == 0:
|
||
|
|
self.checkList[row][1] = 'Checked' if value == Qt.Checked else 'Unchecked'
|
||
|
|
if role != Qt.CheckStateRole and value != '':
|
||
|
|
self.datas[row]['force_value'] = float(value)
|
||
|
|
MainWindowConfig.IOMapping.force_value[self.datas[row]['sig_name']] = float(value)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def flags(self, index: QModelIndex) -> Qt.ItemFlags:
|
||
|
|
if index.column() == 0:
|
||
|
|
if self.datas[index.row()]['sig_type'] not in ('AI', 'DI-24V'):
|
||
|
|
return Qt.ItemIsEnabled | Qt.ItemIsUserCheckable | Qt.ItemIsEditable
|
||
|
|
else:
|
||
|
|
return Qt.ItemIsEnabled
|
||
|
|
if index.column() != 0:
|
||
|
|
return Qt.ItemIsEnabled
|
||
|
|
|
||
|
|
def headerClick(self, isOn):
|
||
|
|
self.beginResetModel()
|
||
|
|
if isOn:
|
||
|
|
self.checkList = [[i, 'Checked'] for i in range(len(self.datas))]
|
||
|
|
else:
|
||
|
|
self.checkList = [[i, 'Unchecked'] for i in range(len(self.datas))]
|
||
|
|
self.endResetModel()
|
||
|
|
|
||
|
|
def headerData(self, section: int, orientation: Qt.Orientation, role: int = ...) -> typing.Any:
|
||
|
|
if role != Qt.DisplayRole:
|
||
|
|
return None
|
||
|
|
|
||
|
|
if orientation == Qt.Horizontal:
|
||
|
|
return self.header[section][1]
|
||
|
|
|
||
|
|
|
||
|
|
class listViewModel(QAbstractListModel):
|
||
|
|
|
||
|
|
def __init__(self, query):
|
||
|
|
super().__init__()
|
||
|
|
self.ListItemDate = []
|
||
|
|
self.data_init(query)
|
||
|
|
|
||
|
|
def data_init(self, query):
|
||
|
|
for point in query:
|
||
|
|
ItemData = {'name': ''}
|
||
|
|
ItemData['name'] = point.sig_name
|
||
|
|
ItemData['data'] = point
|
||
|
|
self.ListItemDate.append(ItemData)
|
||
|
|
|
||
|
|
def data(self, index: QModelIndex, role: int = ...) -> typing.Any:
|
||
|
|
if index.isValid() or (0 <= index.row() < len(self.ListItemData)):
|
||
|
|
if role == Qt.DisplayRole:
|
||
|
|
return QVariant(self.ListItemDate[index.row()]['name'])
|
||
|
|
elif role == Qt.TextAlignmentRole:
|
||
|
|
return QVariant(int(Qt.AlignLeft | Qt.AlignVCenter))
|
||
|
|
else:
|
||
|
|
return QVariant()
|
||
|
|
|
||
|
|
def rowCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
return len(self.ListItemDate)
|
||
|
|
|
||
|
|
def addItem(self, itemData):
|
||
|
|
if itemData:
|
||
|
|
self.ListItemDate.append(itemData)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def deleteItem(self, index):
|
||
|
|
self.ListItemDate.pop(index)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def updataItem(self, lis):
|
||
|
|
self.ListItemDate = lis
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def getItem(self, index):
|
||
|
|
if index > -1 and index < len(self.ListItemData):
|
||
|
|
return self.ListItemData[index]
|
||
|
|
|
||
|
|
|
||
|
|
class variableModel(QAbstractTableModel):
|
||
|
|
|
||
|
|
def __init__(self, header, data: list):
|
||
|
|
QAbstractTableModel.__init__(self, parent=None)
|
||
|
|
self.datas = data
|
||
|
|
self.header = header
|
||
|
|
|
||
|
|
def append_data(self, x):
|
||
|
|
self.datas.append(x)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def remove_row(self, row):
|
||
|
|
self.datas.pop(row)
|
||
|
|
self.layoutChanged.emit()
|
||
|
|
|
||
|
|
def rowCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
if len(self.datas) > 0:
|
||
|
|
return len(self.datas)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def columnCount(self, parent: QModelIndex = ...) -> int:
|
||
|
|
return len(self.header)
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def data(self, QModelIndex, role=None):
|
||
|
|
from communication import iomapping
|
||
|
|
if not QModelIndex.isValid():
|
||
|
|
print("行或者列有问题")
|
||
|
|
return QVariant()
|
||
|
|
if role != Qt.DisplayRole:
|
||
|
|
return QVariant()
|
||
|
|
if role == Qt.DisplayRole:
|
||
|
|
if QModelIndex.column() == 5:
|
||
|
|
que1.put(self.datas[QModelIndex.row()][1])
|
||
|
|
return QVariant(iomapping.current_value[self.datas[QModelIndex.row()][1]])
|
||
|
|
else:
|
||
|
|
return QVariant(self.datas[QModelIndex.row()][QModelIndex.column()])
|
||
|
|
|
||
|
|
def headerData(self, section: int, orientation: Qt.Orientation, role: int = ...) -> typing.Any:
|
||
|
|
if role != Qt.DisplayRole:
|
||
|
|
return None
|
||
|
|
|
||
|
|
if orientation == Qt.Horizontal:
|
||
|
|
return self.header[section]
|