574a66c65b
unfortunately IntEnum can't be bent like we would need it (extensible). So we had to write our own.... The members of the Enum still behave like ints, but also have .name and .value attributes, should they be needed. needed adoptions to correctly use (and test) the EnumType are included. Change-Id: Ie019d2f449a244c4fab00554b6c6daaac8948b59 Reviewed-on: https://forge.frm2.tum.de/review/17843 Tested-by: JenkinsCodeReview <bjoern_pedersen@frm2.tum.de> Reviewed-by: Enrico Faulhaber <enrico.faulhaber@frm2.tum.de>
248 lines
7.6 KiB
Python
248 lines
7.6 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 __future__ import print_function
|
|
|
|
from secop.datatypes import FloatRange, IntRange, StringType, BLOBType, \
|
|
EnumType, BoolType, TupleOf, StructOf, ArrayOf
|
|
|
|
from secop.gui.qt import QDialog, QLabel, QLineEdit,\
|
|
QGroupBox, QSpinBox, QDoubleSpinBox, QComboBox, QCheckBox, \
|
|
QVBoxLayout, QGridLayout, 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, member in enumerate(datatype._enum.members):
|
|
self._map[idx] = member
|
|
self._revmap[member.name] = idx
|
|
self._revmap[member.value] = idx
|
|
self.addItem(member.name, member.value)
|
|
|
|
def get_value(self):
|
|
return self._map[self.currentIndex()].value
|
|
|
|
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 or 1e6) # XXX!
|
|
self.setMinimum(datatype.min or 0) # XXX!
|
|
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, _ 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 _ 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('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)
|