265 lines
8.7 KiB
Python
265 lines
8.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# *****************************************************************************
|
|
# Copyright (c) 2015-2016 by the authors, see LICENSE
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it under
|
|
# the terms of the GNU General Public License as published by the Free Software
|
|
# Foundation; either version 2 of the License, or (at your option) any later
|
|
# version.
|
|
#
|
|
# This program 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 General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along with
|
|
# this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
# Module authors:
|
|
# Enrico Faulhaber <enrico.faulhaber@frm2.tum.de>
|
|
#
|
|
# *****************************************************************************
|
|
|
|
from secop.datatypes import *
|
|
|
|
from PyQt4.QtGui import QDialog, QPushButton, QLabel, QApplication, QLineEdit,\
|
|
QGroupBox, QSpinBox, QDoubleSpinBox, QComboBox, QCheckBox, QRadioButton, \
|
|
QVBoxLayout, QGridLayout, QScrollArea, QFrame
|
|
|
|
from secop.gui.util import loadUi
|
|
|
|
|
|
# XXX: implement live validators !!!!
|
|
|
|
class StringWidget(QLineEdit):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(StringWidget, self).__init__(parent)
|
|
self.datatype = datatype
|
|
if readonly:
|
|
self.setEnabled(False)
|
|
|
|
def get_value(self):
|
|
res = self.text()
|
|
return self.datatype.validate(res)
|
|
|
|
def set_value(self, value):
|
|
self.setText(value)
|
|
|
|
|
|
class BlobWidget(StringWidget):
|
|
# XXX: make an editable hex-table ?
|
|
pass
|
|
|
|
# or derive from widget and switch between combobox and radiobuttons?
|
|
class EnumWidget(QComboBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(EnumWidget, self).__init__(parent)
|
|
self.datatype = datatype
|
|
|
|
self._map = {}
|
|
self._revmap = {}
|
|
for idx, (val, name) in enumerate(datatype.entries.items()):
|
|
self._map[idx] = (name, val)
|
|
self._revmap[name] = idx
|
|
self._revmap[val] = idx
|
|
self.addItem(name, val)
|
|
# XXX: fill Combobox from datatype
|
|
|
|
def get_value(self):
|
|
# XXX: return integer corresponding to the selected item
|
|
return self._map[self.currentIndex()][1]
|
|
|
|
def set_value(self, value):
|
|
self.setCurrentIndex(self._revmap[value])
|
|
|
|
|
|
class BoolWidget(QCheckBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(BoolWidget, self).__init__(parent)
|
|
self.datatype = datatype
|
|
if readonly:
|
|
self.setEnabled(False)
|
|
|
|
def get_value(self):
|
|
return self.isChecked()
|
|
|
|
def set_value(self, value):
|
|
self.setChecked(bool(value))
|
|
|
|
|
|
class IntWidget(QSpinBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(IntWidget, self).__init__(parent)
|
|
self.datatype = datatype
|
|
if readonly:
|
|
self.setEnabled(False)
|
|
self.setMaximum(datatype.max)
|
|
self.setMinimum(datatype.min)
|
|
|
|
def get_value(self):
|
|
return int(self.value())
|
|
|
|
def set_value(self, value):
|
|
self.setValue(int(value))
|
|
|
|
|
|
class FloatWidget(QDoubleSpinBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(FloatWidget, self).__init__(parent)
|
|
self.datatype = datatype
|
|
if readonly:
|
|
self.setEnabled(False)
|
|
self.setMaximum(datatype.max)
|
|
self.setMinimum(datatype.min)
|
|
self.setDecimals(12)
|
|
|
|
def get_value(self):
|
|
return float(self.value())
|
|
|
|
def set_value(self, value):
|
|
self.setValue(float(value))
|
|
|
|
|
|
class TupleWidget(QFrame):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(TupleWidget, self).__init__(parent)
|
|
|
|
self.datatypes = datatype.subtypes
|
|
|
|
self.layout = QVBoxLayout()
|
|
self.subwidgets = []
|
|
for t in self.datatypes:
|
|
w = get_widget(t, readonly=readonly, parent=self)
|
|
w.show()
|
|
self.layout.addWidget(w)
|
|
self.subwidgets.append(w)
|
|
self.setLayout(self.layout)
|
|
self.show()
|
|
self.update()
|
|
|
|
def get_value(self):
|
|
return [v.validate(w.get_value()) for w,v in zip(self.subwidgets, self.datatypes)]
|
|
|
|
def set_value(self, value):
|
|
for w, v in zip(self.subwidgets, value):
|
|
w.set_value(value)
|
|
|
|
|
|
class StructWidget(QGroupBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(StructWidget, self).__init__(parent)
|
|
|
|
self.layout = QGridLayout()
|
|
self.subwidgets = {}
|
|
self.datatypes = []
|
|
self._labels = []
|
|
for idx, name in enumerate(sorted(datatype.named_subtypes)):
|
|
dt = datatype.named_subtypes[name]
|
|
w = get_widget(dt, readonly=readonly, parent=self)
|
|
l = QLabel(name)
|
|
self.layout.addWidget(l, idx, 0)
|
|
self.layout.addWidget(w, idx, 1)
|
|
self._labels.append(l)
|
|
self.subwidgets[name] = (w, dt)
|
|
self.datatypes.append(dt)
|
|
self.setLayout(self.layout)
|
|
|
|
def get_value(self):
|
|
res = {}
|
|
for name, entry in self.subwidgets.items():
|
|
w, dt = entry
|
|
res[name] = dt.validate(w.get_value())
|
|
return res
|
|
|
|
def set_value(self, value):
|
|
for k, v in value.items():
|
|
entry = self.subwidgets[k]
|
|
w, dt = entry
|
|
w.set_value(dt.validate(v))
|
|
|
|
|
|
class ArrayWidget(QGroupBox):
|
|
def __init__(self, datatype, readonly=False, parent=None):
|
|
super(ArrayWidget, self).__init__(parent)
|
|
self.datatype = datatype.subtype
|
|
|
|
self.layout = QVBoxLayout()
|
|
self.subwidgets = []
|
|
for i in range(datatype.maxsize):
|
|
w = get_widget(self.datatype, readonly=readonly, parent=self)
|
|
self.layout.addWidget(w)
|
|
self.subwidgets.append(w)
|
|
self.setLayout(self.layout)
|
|
|
|
def get_value(self):
|
|
return [self.datatype.validate(w.get_value()) for w in self.subwidgets]
|
|
|
|
def set_value(self, values):
|
|
for w, v in zip(self.subwidgets, values):
|
|
w.set_value(v)
|
|
|
|
|
|
|
|
def get_widget(datatype, readonly=False, parent=None):
|
|
return {FloatRange: FloatWidget,
|
|
IntRange: IntWidget,
|
|
StringType: StringWidget,
|
|
BLOBType: BlobWidget,
|
|
EnumType: EnumWidget,
|
|
BoolType: BoolWidget,
|
|
TupleOf: TupleWidget,
|
|
StructOf: StructWidget,
|
|
ArrayOf: ArrayWidget,
|
|
}.get(datatype.__class__)(datatype, readonly, parent)
|
|
|
|
|
|
class msg(QDialog):
|
|
def __init__(self, stuff, parent=None):
|
|
super(msg, self).__init__(parent)
|
|
loadUi(self, 'cmddialog.ui')
|
|
print(dir(self))
|
|
self.setWindowTitle('Please enter the arguments for calling command "blubb()"')
|
|
row = 0
|
|
# self.gridLayout.addWidget(QLabel('String'), row, 0); self.gridLayout.addWidget(StringWidget(StringType()), row, 1); row += 1
|
|
# self.gridLayout.addWidget(QLabel('Blob'), row, 0); self.gridLayout.addWidget(BlobWidget(BLOBType()), row, 1); row += 1
|
|
# self.gridLayout.addWidget(QLabel('Enum'), row, 0); self.gridLayout.addWidget(EnumWidget(EnumType(a=1,b=9)), row, 1); row += 1
|
|
# self.gridLayout.addWidget(QLabel('Bool'), row, 0); self.gridLayout.addWidget(BoolWidget(BoolType()), row, 1); row += 1
|
|
# self.gridLayout.addWidget(QLabel('int'), row, 0); self.gridLayout.addWidget(IntWidget(IntRange(0,9)), row, 1); row += 1
|
|
# self.gridLayout.addWidget(QLabel('float'), row, 0); self.gridLayout.addWidget(FloatWidget(FloatRange(-9,9)), row, 1); row += 1
|
|
|
|
#self.gridLayout.addWidget(QLabel('tuple'), row, 0);
|
|
#dt = TupleOf(BoolType(), EnumType(a=2,b=3))
|
|
#w = TupleWidget(dt)
|
|
#self.gridLayout.addWidget(w, row, 1)
|
|
#row+=1
|
|
|
|
#self.gridLayout.addWidget(QLabel('array'), row, 0);
|
|
#dt = ArrayOf(IntRange(0,10), 10)
|
|
#w = ArrayWidget(dt)
|
|
#self.gridLayout.addWidget(w, row, 1)
|
|
#row+=1
|
|
|
|
self.gridLayout.addWidget(QLabel('struct'), row, 0);
|
|
dt = StructOf(i=IntRange(0,10), f=FloatRange(), b=BoolType())
|
|
w = StructWidget(dt)
|
|
self.gridLayout.addWidget(w, row, 1)
|
|
row+=1
|
|
|
|
self.gridLayout.addWidget(QLabel('stuff'), row, 0, 1, 0); row += 1 # at pos (0,0) span 2 cols, 1 row
|
|
self.gridLayout.setRowStretch(row, 1)
|
|
self.setModal(True)
|
|
|
|
def accept(self):
|
|
print('accepted')
|
|
super(msg, self).accept()
|
|
|
|
def reject(self):
|
|
print('rejected')
|
|
super(msg, self).reject()
|
|
|
|
def done(self, how):
|
|
print('done(%r)' % how)
|
|
return super(msg, self).done(how)
|
|
|