def create_states(self):
"""Generate states of the SM according to
the behavior of the GUI"""
# States
self.init = StateInit(self.machine, self.ui)
self.image_load = StateImageLoading(self.machine, self)
self.landmarking_selection = StateLanmarkingSelection(
self.machine, self)
self.save = StateSaveLandmarking(self.machine, self)
self.clear = StateClear(self.machine, self)
self.quit = StateExit(self.machine, self)
self.about = StateAbout(self.machine, self)
self.run = StateRunLandmarking(self.machine, self)
# Group of states for run
self.run_init = StateInitRun(self.run, self)
self.run_fwd = StateFoward(self.run, self)
self.run_bck = StateBack(self.run, self)
self.run_edit = StateEdit(self.run, self)
self.run.setInitialState(self.run_init)
def create_states(self):
"""Generate states of the SM according to
the behavior of the GUI"""
# States
self.init = StateInit(self.machine, self.ui)
self.image_load = StateImageLoading(self.machine, self)
self.landmarking_selection = StateLanmarkingSelection(self.machine, self)
self.save = StateSaveLandmarking(self.machine, self)
self.clear = StateClear(self.machine, self)
self.quit = StateExit(self.machine, self)
self.about = StateAbout(self.machine, self)
self.run = StateRunLandmarking(self.machine, self)
# Group of states for run
self.run_init = StateInitRun(self.run, self)
self.run_fwd = StateFoward(self.run, self)
self.run_bck = StateBack(self.run, self)
self.run_edit = StateEdit(self.run, self)
self.run.setInitialState(self.run_init)
class MainWindow(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.settings = QtCore.QSettings()
self.settings.setValue("last_dir", os.getcwd())
filename = os.path.dirname(__file__)
palette = QtGui.QPalette()
brush = QtGui.QBrush(
QtGui.QPixmap(os.path.join(filename, "Images/back.png")))
palette.setBrush(QtGui.QPalette.Background, brush)
self.setPalette(palette)
self.ui.scene = Scene()
self.ui.scene.setSceneRect(0, 0, 700, 300)
self.ui.graphicsView.setScene(self.ui.scene)
self.ui.graphicsView.setInteractive(True)
self.create_special_buttons(filename)
self.ui.graphicsView.show()
self.landn = 77
self.count = None
#self.number_of_landmarks = None
self.create_machine()
def create_special_buttons(self, filename):
self.ui.myButtonNext = MyButton(
os.path.join(filename, "Icons/next.png"), "Next ..", False)
self.ui.myButtonPrev = MyButton(
os.path.join(filename, "Icons/prev.png"), "Prev ..", False)
self.ui.myButtonEdit = MyButton(
os.path.join(filename, "Icons/learn.png"), "Edit ..", False)
self.ui.scene.buttonsForChecker(
self.ui.myButtonNext, self.ui.myButtonEdit,
self.ui.myButtonPrev)
def create_machine(self):
self.machine = QtCore.QStateMachine()
self.create_states()
self.create_transitions()
self.machine.setInitialState(self.init)
self.machine.start()
def create_states(self):
"""Generate states of the SM according to
the behavior of the GUI"""
# States
self.init = StateInit(self.machine, self.ui)
self.image_load = StateImageLoading(self.machine, self)
self.landmarking_selection = StateLanmarkingSelection(
self.machine, self)
self.save = StateSaveLandmarking(self.machine, self)
self.clear = StateClear(self.machine, self)
self.quit = StateExit(self.machine, self)
self.about = StateAbout(self.machine, self)
self.run = StateRunLandmarking(self.machine, self)
# Group of states for run
self.run_init = StateInitRun(self.run, self)
self.run_fwd = StateFoward(self.run, self)
self.run_bck = StateBack(self.run, self)
self.run_edit = StateEdit(self.run, self)
self.run.setInitialState(self.run_init)
def create_transitions(self):
"""Assign the trasintion to the SM according to
the behavior of the GUI"""
#home made QEvent transition (from future fix it)
"""clean_to_run = StringTransition("Hello")
clean_to_run.setTargetState(self.run)
self.clear.addTransition(clean_to_run)
"""
# Transitions
self.init.addTransition(self.ui.pushButton.clicked, self.image_load)
self.init.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.init.addTransition(self.ui.pushButton_6.clicked, self.about)
self.image_load.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.image_load.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.image_load.addTransition(self.ui.pushButton.clicked, self.clear)
self.image_load.addTransition(self.ui.pushButton_6.clicked, self.about)
self.landmarking_selection.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.landmarking_selection.addTransition(
self.ui.pushButton_3.clicked, self.run)
self.landmarking_selection.addTransition(
self.ui.pushButton_5.clicked, self.quit)
self.landmarking_selection.addTransition(
self.ui.pushButton_6.clicked, self.about)
self.landmarking_selection.addTransition(
self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.landmarking_selection.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.landmarking_selection.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.landmarking_selection.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.save.addTransition(self.ui.pushButton_4.clicked, self.save)
self.save.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.save.addTransition(self.ui.pushButton.clicked, self.clear)
self.save.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.save.addTransition(self.ui.pushButton_6.clicked, self.about)
self.save.addTransition(self.ui.pushButton_3.clicked, self.run)
# for the inner machine loop of <- ->
self.save.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.save.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.save.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.clear.addTransition(self.clear.finished, self.image_load)
self.about.addTransition(self.ui.pushButton.clicked, self.clear)
self.about.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.about.addTransition(self.ui.pushButton_3.clicked, self.run)
self.about.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.about.addTransition(self.ui.pushButton_6.clicked, self.about)
self.about.addTransition(self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.about.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.about.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.about.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.quit.addTransition(self.ui.pushButton.clicked, self.clear)
self.quit.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.quit.addTransition(self.ui.pushButton_3.clicked, self.run)
self.quit.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.quit.addTransition(self.ui.pushButton_6.clicked, self.about)
self.quit.addTransition(self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.quit.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.quit.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.quit.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run.addTransition(self.ui.pushButton.clicked, self.clear)
self.run.addTransition(
self.ui.pushButton_2.clicked, self.landmarking_selection)
self.run.addTransition(self.ui.pushButton_3.clicked, self.run)
self.run.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.run.addTransition(self.ui.pushButton_6.clicked, self.about)
# for the inner machine loop of <- ->
self.run.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
# Transitions under run
self.run_init.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_init.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_init.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_fwd.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_fwd.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_fwd.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_bck.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_bck.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_bck.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_fwd.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run_bck.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run_init.addTransition(self.ui.pushButton_4.clicked, self.save)
def draw_landmarks_in_scene(self):
"""Put the landmarked images in the scene and load a table
with the (nl, x, y)"""
# TODO remove this fuction .. translate to new state
my_table = Table(self.number_of_landmarks, ['x', 'y'])
my_landmarks = self.myfinder.landmarks[self.count]
my_filter_landmarks = [my_landmarks[i]
for i in self.number_of_landmarks]
load_model(my_filter_landmarks, my_table)
proxytabla = QtGui.QGraphicsProxyWidget()
proxytabla.setWidget(my_table)
proxytabla.setPos(320, 0)
qitmp = toqimage(self.images[self.count])
pix = QtGui.QPixmap(QtGui.QPixmap.fromImage(qitmp))
QtGui.QGraphicsPixmapItem(pix.scaledToWidth(310), scene=self.ui.scene)
self.ui.scene.addItem(proxytabla)
class MainWindow(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.settings = QtCore.QSettings()
self.settings.setValue("last_dir", os.getcwd())
filename = os.path.dirname(__file__)
palette = QtGui.QPalette()
brush = QtGui.QBrush(QtGui.QPixmap(os.path.join(filename, "Images/back.png")))
palette.setBrush(QtGui.QPalette.Background, brush)
self.setPalette(palette)
self.ui.scene = Scene()
self.ui.scene.setSceneRect(0, 0, 700, 300)
self.ui.graphicsView.setScene(self.ui.scene)
self.ui.graphicsView.setInteractive(True)
self.create_special_buttons(filename)
self.ui.graphicsView.show()
self.landn = 77
self.count = None
# self.number_of_landmarks = None
self.create_machine()
def create_special_buttons(self, filename):
self.ui.myButtonNext = MyButton(os.path.join(filename, "Icons/next.png"), "Next ..", False)
self.ui.myButtonPrev = MyButton(os.path.join(filename, "Icons/prev.png"), "Prev ..", False)
self.ui.myButtonEdit = MyButton(os.path.join(filename, "Icons/learn.png"), "Edit ..", False)
self.ui.scene.buttonsForChecker(self.ui.myButtonNext, self.ui.myButtonEdit, self.ui.myButtonPrev)
def create_machine(self):
self.machine = QtCore.QStateMachine()
self.create_states()
self.create_transitions()
self.machine.setInitialState(self.init)
self.machine.start()
def create_states(self):
"""Generate states of the SM according to
the behavior of the GUI"""
# States
self.init = StateInit(self.machine, self.ui)
self.image_load = StateImageLoading(self.machine, self)
self.landmarking_selection = StateLanmarkingSelection(self.machine, self)
self.save = StateSaveLandmarking(self.machine, self)
self.clear = StateClear(self.machine, self)
self.quit = StateExit(self.machine, self)
self.about = StateAbout(self.machine, self)
self.run = StateRunLandmarking(self.machine, self)
# Group of states for run
self.run_init = StateInitRun(self.run, self)
self.run_fwd = StateFoward(self.run, self)
self.run_bck = StateBack(self.run, self)
self.run_edit = StateEdit(self.run, self)
self.run.setInitialState(self.run_init)
def create_transitions(self):
"""Assign the trasintion to the SM according to
the behavior of the GUI"""
# home made QEvent transition (from future fix it)
"""clean_to_run = StringTransition("Hello")
clean_to_run.setTargetState(self.run)
self.clear.addTransition(clean_to_run)
"""
# Transitions
self.init.addTransition(self.ui.pushButton.clicked, self.image_load)
self.init.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.init.addTransition(self.ui.pushButton_6.clicked, self.about)
self.image_load.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.image_load.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.image_load.addTransition(self.ui.pushButton.clicked, self.clear)
self.image_load.addTransition(self.ui.pushButton_6.clicked, self.about)
self.landmarking_selection.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.landmarking_selection.addTransition(self.ui.pushButton_3.clicked, self.run)
self.landmarking_selection.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.landmarking_selection.addTransition(self.ui.pushButton_6.clicked, self.about)
self.landmarking_selection.addTransition(self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.landmarking_selection.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.landmarking_selection.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.landmarking_selection.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.save.addTransition(self.ui.pushButton_4.clicked, self.save)
self.save.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.save.addTransition(self.ui.pushButton.clicked, self.clear)
self.save.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.save.addTransition(self.ui.pushButton_6.clicked, self.about)
self.save.addTransition(self.ui.pushButton_3.clicked, self.run)
# for the inner machine loop of <- ->
self.save.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.save.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.save.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.clear.addTransition(self.clear.finished, self.image_load)
self.about.addTransition(self.ui.pushButton.clicked, self.clear)
self.about.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.about.addTransition(self.ui.pushButton_3.clicked, self.run)
self.about.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.about.addTransition(self.ui.pushButton_6.clicked, self.about)
self.about.addTransition(self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.about.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.about.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.about.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.quit.addTransition(self.ui.pushButton.clicked, self.clear)
self.quit.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.quit.addTransition(self.ui.pushButton_3.clicked, self.run)
self.quit.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.quit.addTransition(self.ui.pushButton_6.clicked, self.about)
self.quit.addTransition(self.ui.pushButton_4.clicked, self.save)
# for the inner machine loop of <- ->
self.quit.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.quit.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.quit.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run.addTransition(self.ui.pushButton.clicked, self.clear)
self.run.addTransition(self.ui.pushButton_2.clicked, self.landmarking_selection)
self.run.addTransition(self.ui.pushButton_3.clicked, self.run)
self.run.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run.addTransition(self.ui.pushButton_5.clicked, self.quit)
self.run.addTransition(self.ui.pushButton_6.clicked, self.about)
# for the inner machine loop of <- ->
self.run.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
# Transitions under run
self.run_init.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_init.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_init.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_fwd.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_fwd.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_fwd.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_bck.addTransition(self.ui.myButtonPrev.clicked, self.run_bck)
self.run_bck.addTransition(self.ui.myButtonNext.clicked, self.run_fwd)
self.run_bck.addTransition(self.ui.myButtonEdit.clicked, self.run_edit)
self.run_fwd.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run_bck.addTransition(self.ui.pushButton_4.clicked, self.save)
self.run_init.addTransition(self.ui.pushButton_4.clicked, self.save)
def draw_landmarks_in_scene(self):
"""Put the landmarked images in the scene and load a table
with the (nl, x, y)"""
# TODO remove this fuction .. translate to new state
my_table = Table(self.number_of_landmarks, ["x", "y"])
my_landmarks = self.myfinder.landmarks[self.count]
my_filter_landmarks = [my_landmarks[i] for i in self.number_of_landmarks]
load_model(my_filter_landmarks, my_table)
proxytabla = QtGui.QGraphicsProxyWidget()
proxytabla.setWidget(my_table)
proxytabla.setPos(320, 0)
qitmp = toqimage(self.images[self.count])
pix = QtGui.QPixmap(QtGui.QPixmap.fromImage(qitmp))
QtGui.QGraphicsPixmapItem(pix.scaledToWidth(310), scene=self.ui.scene)
self.ui.scene.addItem(proxytabla)