evolver/qtapp.py at master · chrisbainbridge/evolver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
"""The main QT application for visualising simulations.

The QApplication is responsible for creating and managing the main
widget (currently qtgui.py) created by qt_designer. The main widget in
turn is responsible for creating and managing the glwidget (currently
TestDidget.py).

Events related to display (rotating, movement, shading etc.) are
handled in glwidget.py. Everything else is passed to the
QApplication here.

The QApplication also steps the simulator and schedules an opengl
redraw whenever it receives a specified timer event.

"""

import tempfile

from PyQt4 import QtCore, QtGui, uic
from PyQt4.QtGui import QApplication
from PyQt4 import QtOpenGL
from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *

from sim import DT

import logging
log = logging.getLogger('qtapp')

class MyApp(QApplication):
    def __init__(self, args, sim=None):
        """Initialise QApplication.

        args -- QT flags string
        sim -- simulation to run

        """
        log.debug('myapp init with args %s', args)
        QApplication.__init__(self, args)
        if not QtOpenGL.QGLFormat.hasOpenGL():
            raise Exception('No Qt OpenGL support.')
        log.debug('We have OpenGL support')
        # create the gui
        self.window = QtGui.QMainWindow()
        uic.loadUi('qt4gui.ui', self.window)
        self.window.statusBar().hide()
        self.glwidget = self.window.glwidget
        # tell GL widget what to draw
        self.glwidget.qtapp = self
        # set timerEvent for each timestep
        self.startTimer(DT*1000)
        # display main window
        self.window.show()
        # what to do when we quit
        QtCore.QObject.connect(self, QtCore.SIGNAL("lastWindowClosed()"), self.quit)
        # default end time
        self.end_time = 0
        self.window.progressBar1.hide()
        self.window.score_label.setText('0')
        self._old_sim_total_time = 0
        self._old_sim_score = 0
        self.window.progressBar1.show()
        self.steps = 0
        log.debug('end of qtapp init')
        self.frameno = 0
        if sim:
            self.setSim(sim)

    def setSim(self, sim):
        self.sim = sim
        self.glwidget.sim = sim
        self.window.progressBar1.maximum = round(self.sim.max_simsecs)
        self.window.show()

    def quit(self):
        "Quit the application"
        log.debug('quit : ends exec_loop()')
        # if recording is on, shut it down
        if self.glwidget.record:
            self.window.hide()
            self.glwidget.finaliseRecording()
        del self.sim
        del self.glwidget.sim
        QApplication.quit()

    def setRecord(self, record, avifile=''):
        """Turn recording on/off and set the output file name.

        record -- 0 for off, 1 for on
        avifile -- file name

        """
        # tell self.glwidget to record movie
        self.glwidget.record = record
        self.glwidget.avifile = avifile
        if self.glwidget.record:
            self.glwidget.screenshot_dir = tempfile.mkdtemp()

    def timerEvent(self, event):
        """Callback from QT event timer."""
        log.debug('qtapp.timerEvent on %s', self)
        # step the ODE simulation
        if not self.glwidget.pause and hasattr(self, 'sim'):
            self.sim.step()
            self.steps += 1
        # we tell it explicitly when to record a frame, otherwise spurious
        # repaints will cause the movie to go out of sync with the physics
        # Note: we only record 1/5th of the actual frames because grabbing
        # the images and saving them takes ages, and we want to do it in
        # realtime (this is why the movies are only 10fps).
        if self.glwidget.record and (self.frameno==0):
            self.glwidget.record_this_frame = 1
        self.frameno = (self.frameno + 1)%5
        # schedule a repaint
        self.glwidget.updateGL()
        # show simulation time counter
        if self.steps % 25 == 0:
            if self.sim.max_simsecs:
                self.window.progressBar1.setValue(round(self.sim.total_time))
            self.window.score_label.setText('%.1f : %.3f'%(self.sim.total_time, self.sim.score))
        if self.sim.finished:
            self.quit()