def add_refresh_button(self, name, func):
layout = self.v_layout
btn = QPushButton(name, self)
btn.setMaximumWidth(60)
layout.insertWidget(0, btn)
btn.clicked.connect(self._add_items_to_filter)
btn.clicked.connect(super().accept)
btn.clicked.connect(func)
class NuggetListWidget(QWidget):
def __init__(self, interactive_matching_widget):
super(NuggetListWidget, self).__init__(interactive_matching_widget)
self.interactive_matching_widget = interactive_matching_widget
self.layout = QVBoxLayout(self)
self.layout.setContentsMargins(0, 0, 0, 0)
self.layout.setSpacing(10)
# top widget
self.top_widget = QWidget()
self.top_layout = QHBoxLayout(self.top_widget)
self.top_layout.setContentsMargins(0, 0, 0, 0)
self.top_layout.setSpacing(10)
self.layout.addWidget(self.top_widget)
self.description = QLabel(
"Below you see a list of guessed matches for you to confirm or correct."
)
self.description.setFont(LABEL_FONT)
self.top_layout.addWidget(self.description)
self.stop_button = QPushButton("Continue With Next Attribute")
self.stop_button.setFont(BUTTON_FONT)
self.stop_button.clicked.connect(self._stop_button_clicked)
self.stop_button.setMaximumWidth(240)
self.top_layout.addWidget(self.stop_button)
# nugget list
self.nugget_list = CustomScrollableList(self, NuggetListItemWidget)
self.layout.addWidget(self.nugget_list)
def update_nuggets(self, nuggets):
max_start_chars = max([
nugget[CachedContextSentenceSignal]["start_char"]
for nugget in nuggets
])
self.nugget_list.update_item_list(nuggets, max_start_chars)
def _stop_button_clicked(self):
self.interactive_matching_widget.main_window.give_feedback_task(
{"message": "stop-interactive-matching"})
def enable_input(self):
self.stop_button.setEnabled(True)
self.nugget_list.enable_input()
def disable_input(self):
self.stop_button.setDisabled(True)
self.nugget_list.disable_input()
def exerciseLayout(self, index):
hLayout = QHBoxLayout()
setSpinBox = QSpinBox()
setSpinBox.setRange(1, 100)
setSpinBox.setValue(5)
setSpinBox.setMaximumWidth(80)
startButton = QPushButton(self.classifyExercises.exercises[index].name)
startButton.setMaximumWidth(120)
assignedKey = QPushButton(
self.classifyExercises.exercises[index].assigned_key[0])
assignedKey.setStyleSheet('background-color: grey; color: white;')
assignedKey.setMaximumWidth(50)
assignedKey.setEnabled(False)
progress = QProgressBar()
progress.setMaximumWidth(150)
progress.setValue(randint(1, 100))
progress.setAlignment(QtCore.Qt.Alignment.AlignCenter)
latency = QLabel(str(randint(50, 250)) + 'ms')
avgLatency = QLabel(str(randint(50, 250)) + 'ms')
hLayout.addWidget(setSpinBox)
hLayout.setAlignment(setSpinBox, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(startButton)
hLayout.setAlignment(startButton, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(assignedKey)
hLayout.setAlignment(assignedKey, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(progress)
hLayout.setAlignment(progress, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(latency)
hLayout.setAlignment(latency, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(avgLatency)
hLayout.setAlignment(avgLatency, QtCore.Qt.Alignment.AlignHCenter)
hLayout.setContentsMargins(0, 10, 0, 10)
self.exerciseLayouts.append(hLayout)
return hLayout
class Ui_TrainPanel(object):
def setupUi(self, TrainPanel):
self.parent = TrainPanel
self.mainLayout = QHBoxLayout(TrainPanel)
self.bigFont = QLabel().font()
self.bigFont.setPointSize(13)
self.actionsLayout = QHBoxLayout(TrainPanel)
# self.checkRecording = QCheckBox(TrainPanel)
self.calibrateButton = QPushButton("Calibrate", parent=TrainPanel)
self.calibrateButton.setFont(self.bigFont)
self.calibrateButton.setStyleSheet(CustomQStyles.buttonStyle)
self.calibrateButton.setMinimumHeight(50)
self.calibrateButton.setMaximumWidth(170)
self.calibrateButton.setMinimumWidth(120)
self.sessionButton = QPushButton("Session", parent=TrainPanel)
self.sessionButton.setFont(self.bigFont)
self.sessionButton.setStyleSheet(CustomQStyles.outlineButtonStyle)
self.sessionButton.setMinimumHeight(50)
self.sessionButton.setMaximumWidth(170)
self.sessionButton.setMinimumWidth(120)
self.listFiles = QListWidget(TrainPanel)
self.listFiles.setFont(self.bigFont)
self.label = QLabel('or select', parent=TrainPanel)
self.subjectLayout = QVBoxLayout(TrainPanel)
print("init")
self.box1 = QGroupBox(parent=TrainPanel)
self.box2 = QGroupBox(parent=TrainPanel)
self.wizard = CalibrateWizard(parent=TrainPanel)
self.setPatientsBox(TrainPanel) # right panel
self.setActionsBox(TrainPanel) # left panel
self.mainLayout.addWidget(self.box1, stretch=2)
self.mainLayout.addWidget(self.box2, stretch=3)
# self.mainLayout.setAlignment(self.box2, Qt.Alignment.AlignCenter)
# Display progress bar, checkbox - to record new gestures or not, start train
def setActionsBox(self, TrainPanel):
hLayout = QVBoxLayout()
hLayout.addWidget(self.calibrateButton)
hLayout.addWidget(self.sessionButton)
hLayout.setAlignment(self.calibrateButton, Qt.Alignment.AlignCenter)
hLayout.setAlignment(self.sessionButton, Qt.Alignment.AlignCenter)
self.box2.setLayout(hLayout)
# Display list of subjects, or new subject
def setPatientsBox(self, TrainPanel):
# Button to create new patient
self.addPatient = QPushButton('Create Patient')
self.addPatient.setFixedSize(120, 35)
self.addPatient.setStyleSheet(CustomQStyles.outlineButtonStyle)
self.label.setSizePolicy(QSizePolicy.Policy.Expanding,
QSizePolicy.Policy.Expanding)
self.label.setAlignment(QtCore.Qt.Alignment.AlignCenter)
self.loadPatientList()
self.listFiles.setStyleSheet(CustomQStyles.listStyle)
self.listFiles.setFocusPolicy(Qt.FocusPolicy.NoFocus)
self.subjectLayout.addWidget(self.addPatient)
self.subjectLayout.setAlignment(self.addPatient,
Qt.Alignment.AlignCenter)
self.subjectLayout.addWidget(self.label)
self.subjectLayout.addWidget(self.listFiles)
self.box1.setLayout(self.subjectLayout)
def loadPatientList(self):
self.listFiles.clear()
self.parent.patients.clear()
files = [
f for f in listdir(PATIENTS_PATH) if isfile(join(PATIENTS_PATH, f))
]
for x, ind in zip(files, range(0, len(files))):
item = QListWidgetItem(x.split('.')[0])
item.setTextAlignment(Qt.Alignment.AlignHCenter)
self.listFiles.addItem(item)
with open(PATIENTS_PATH + x, 'r') as f:
person_dict = json.load(f)
patient = Patient(person_dict['Name'], person_dict['Age'],
person_dict['Exercises'])
self.parent.patients.append(patient)
print(self.parent.patients)
def deleteItemsOfLayout(self, layout):
if layout is not None:
while layout.count():
item = layout.takeAt(0)
widget = item.widget()
if widget is not None:
widget.setParent(None)
else:
self.deleteItemsOfLayout(item.layout())
