def __init__(self, category, name):
super().__init__()
uic.loadUi('View\Description.ui', self)
self._exerciseManager = ExerciseManager(category, name)
self.textEdit.insertPlainText(self._exerciseManager.get_description())
self.setAction()
self.exec_()
class Ui_Remove(QtWidgets.QDialog):
def __init__(self, category, name):
super().__init__()
uic.loadUi('View\RemoveVideo.ui', self)
self._exerciseCategory = category
self._exerciseName = name
self._currentExercise = None
self.setAction()
self.initialize_gui()
self.exec_()
def initialize_gui(self):
self.comboBox.addItems(next(os.walk("./Exercises/"+self._exerciseCategory+"/"+self._exerciseName))[1])
self.changeCategory()
def setAction(self):
# Imposto le azioni per gli elementi della GUI che generano eventi
self.pushButton.clicked.connect(self.removeVideo)
self.pushButton_2.clicked.connect(self.close)
self.comboBox.currentIndexChanged.connect(self.changeCategory)
def removeVideo(self):
if self.comboBox_2.currentText():
self._currentExercise = ExerciseManager(self._exerciseCategory, self._exerciseName)
self._currentExercise.remove_video(self.comboBox.currentText(), self.comboBox_2.currentText()[:-4])
self.close()
def changeCategory(self):
self.comboBox_2.clear()
files = [fname for fname in sorted(os.listdir("./Exercises/"+self._exerciseCategory+"/"+self._exerciseName+"/"+self.comboBox.currentText()), key = len) if fname.endswith('.mp4')]
x = len(files)
self.comboBox_2.addItems(files)
self.comboBox_2.setCurrentIndex(0)
def trainLearner(self):
exerciseManager = ExerciseManager(self._exerciseCategory,
self._exerciseName)
metricLearner = MetricLearner()
metricLearner.train_metric_learner_supervised(
exerciseManager.get_landmarks()[2],
exerciseManager.get_landmarks()[3], "Exercises" + '\\' +
self._exerciseCategory + '\\' + self._exerciseName + '\\')
def execute_exercise(self):
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
res = self._currentExercise.execute_exercise()
if (res != 0):
self.result = Ui_Results(res, self.comboBox.currentText(),
self.comboBox_2.currentText())
else:
self.error_message = Ui_CameraError()
def changeCategory(self):
self.comboBox_2.clear()
self.comboBox_2.addItems(
sorted(
os.listdir("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText()),
key=len))
self.comboBox_2.setCurrentIndex(0)
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
def createNewVideo(self, typeVideo):
exerciseManager = ExerciseManager(self._exerciseCategory,
self._exerciseName)
res = exerciseManager.insert_new_video(typeVideo)
if res != 0:
self._windowVideoChoice = Ui_VideoTraining(self._exerciseCategory,
self._exerciseName,
typeVideo)
else:
self.error_message = Ui_CameraError()
self.initialize()
def changeExercise(self):
if (self.comboBox_2.currentText()):
self._currentExercise = ExerciseManager(
self.comboBox.currentText(), self.comboBox_2.currentText())
self.label_5.setText(self._currentExercise.get_description())
self.play_video(self.comboBox.currentText(),
self.comboBox_2.currentText())
if path.exists("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText() +
"\\TreanedModel.sav"):
self.pushButton.setEnabled(True)
else:
self.pushButton.setEnabled(False)
def initialize_gui(self):
diect = []
for file in os.listdir("Exercises"):
if not (file.endswith(".npy")) and not (
file.endswith(".png")) and not (file.endswith(".mp4")):
diect.append(file)
self.comboBox.addItems(diect)
self.changeCategory()
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
self.play_video(self.comboBox.currentText(),
self.comboBox_2.currentText())
self.label_5.setText(self._currentExercise.get_description())
if path.exists("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText() +
"\\TreanedModel.sav"):
self.pushButton.setEnabled(True)
else:
self.pushButton.setEnabled(False)
class Ui_Description(QtWidgets.QDialog):
def __init__(self, category, name):
super().__init__()
uic.loadUi('View\Description.ui', self)
self._exerciseManager = ExerciseManager(category, name)
self.textEdit.insertPlainText(self._exerciseManager.get_description())
self.setAction()
self.exec_()
def setAction(self):
# Imposto le azioni per gli elementi della GUI che generano eventi
self.pushButton.clicked.connect(self.saveDescription)
self.pushButton_2.clicked.connect(self.close)
def saveDescription(self):
self._exerciseManager.replace_description(self.textEdit.toPlainText())
self.close()
def removeVideo(self):
self.window_remove = Ui_Remove(self._exerciseCategory,
self._exerciseName)
def removeVideo(self):
if self.comboBox_2.currentText():
self._currentExercise = ExerciseManager(self._exerciseCategory, self._exerciseName)
self._currentExercise.remove_video(self.comboBox.currentText(), self.comboBox_2.currentText()[:-4])
self.close()
class Ui(QtWidgets.QMainWindow):
def __init__(self):
super(Ui, self).__init__()
self.ui = uic.loadUi('View\MainPage.ui', self)
self._currentExercise = None
self.player = QtMultimedia.QMediaPlayer(
None, QtMultimedia.QMediaPlayer.VideoSurface)
self.player.setVideoOutput(self.ui.widget_8)
self.initialize_gui()
self.setAction()
self.show()
def initialize_gui(self):
diect = []
for file in os.listdir("Exercises"):
if not (file.endswith(".npy")) and not (
file.endswith(".png")) and not (file.endswith(".mp4")):
diect.append(file)
self.comboBox.addItems(diect)
self.changeCategory()
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
self.play_video(self.comboBox.currentText(),
self.comboBox_2.currentText())
self.label_5.setText(self._currentExercise.get_description())
if path.exists("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText() +
"\\TreanedModel.sav"):
self.pushButton.setEnabled(True)
else:
self.pushButton.setEnabled(False)
def setAction(self):
# Imposto le azioni per gli elementi della GUI che generano eventi
self.comboBox.currentIndexChanged.connect(self.changeCategory)
self.comboBox_2.currentIndexChanged.connect(self.changeExercise)
self.pushButton_3.clicked.connect(self.show_settings)
self.pushButton.clicked.connect(self.execute_exercise)
self.player.stateChanged.connect(self.player.play)
def changeCategory(self):
self.comboBox_2.clear()
self.comboBox_2.addItems(
sorted(
os.listdir("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText()),
key=len))
self.comboBox_2.setCurrentIndex(0)
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
def changeExercise(self):
if (self.comboBox_2.currentText()):
self._currentExercise = ExerciseManager(
self.comboBox.currentText(), self.comboBox_2.currentText())
self.label_5.setText(self._currentExercise.get_description())
self.play_video(self.comboBox.currentText(),
self.comboBox_2.currentText())
if path.exists("Exercises" + '\\' + self.comboBox.currentText() +
'\\' + self.comboBox_2.currentText() +
"\\TreanedModel.sav"):
self.pushButton.setEnabled(True)
else:
self.pushButton.setEnabled(False)
def show_settings(self):
self.window_setting = Ui_settingss(self.comboBox.currentText(),
self.comboBox_2.currentText())
def play_video(self, category, name):
file = "Exercises/" + category + "/" + name + "/demo.wmv"
self.player.setMedia(
QtMultimedia.QMediaContent(QtCore.QUrl.fromLocalFile(file)))
self.player.play()
def execute_exercise(self):
self._currentExercise = ExerciseManager(self.comboBox.currentText(),
self.comboBox_2.currentText())
res = self._currentExercise.execute_exercise()
if (res != 0):
self.result = Ui_Results(res, self.comboBox.currentText(),
self.comboBox_2.currentText())
else:
self.error_message = Ui_CameraError()