-
Notifications
You must be signed in to change notification settings - Fork 8
/
firesimgui.py
176 lines (138 loc) · 5.92 KB
/
firesimgui.py
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
import logging as log
import os.path
# This is a workaround for a Linux + NVIDIA + PyQt5 bug causing no graphics to be rendered
# because PyQt5 is linking to the wrong libGL.so / NVIDIA isn't overriding the MESA one.
# See: https://bugs.launchpad.net/ubuntu/+source/python-qt4/+bug/941826
from OpenGL import GL
from PyQt5.QtCore import (pyqtProperty, pyqtSignal, pyqtSlot, QObject, QUrl,
QTimer, QSize, QRect)
from PyQt5.QtQml import qmlRegisterType, QQmlComponent
from PyQt5.QtQuick import QQuickView
from PyQt5.QtWidgets import QApplication, QFileDialog
from PyQt5.QtGui import QIcon
from ui.canvasview import CanvasView
from lib.config import Config
from models.scene import Scene
from controllers.netcontroller import NetController
class FireSimGUI(QObject):
def __init__(self, args=None):
QObject.__init__(self)
self.app = QApplication(["FireSim"])
self.app.aboutToQuit.connect(self.about_to_quit)
self.args = args
self.config = Config("data/config.json")
if self.args.profile:
try:
import yappi
yappi.start()
except ImportError:
log.error("Could not enable YaPPI profiling")
self.selected_fixture = None
self.is_blurred = False
scene_file_path = (self.args.scene if self.args.scene is not None
else self.config.get("last-opened-scene"))
self.scene = Scene(scene_file_path)
qmlRegisterType(CanvasView, "FireSim", 1, 0, "Canvas")
self.view = QQuickView()
self.view.setResizeMode(QQuickView.SizeRootObjectToView)
self.view.closeEvent = self.on_close
self.context = self.view.rootContext()
self.context.setContextProperty('main', self)
self.context.setContextProperty('view', self.view)
self.context.setContextProperty('scene', self.scene)
self.view.setSource(QUrl('ui/qml/FireSimGUI.qml'))
self.root = self.view.rootObject()
self.canvas = self.root.findChild(CanvasView)
self.canvas.gui = self
self.canvas.model.scene = self.scene
self.canvas.update_target_fps.connect(self.set_target_fps)
self.set_properties_from_scene()
#self.net_thread = QThread()
#self.net_thread.start()
self.netcontroller = NetController(self)
#self.netcontroller.moveToThread(self.net_thread)
#self.netcontroller.start.emit()
self.redraw_timer = QTimer()
self.set_target_fps(60)
self.redraw_timer.timeout.connect(self.canvas.update)
self.redraw_timer.start()
self.netcontroller.new_frame.connect(self.canvas.controller.on_new_frame)
geom_str = self.config.get("window-geometry", None)
if geom_str is not None:
self.view.setGeometry(QRect(*geom_str))
else:
self.view.setMinimumSize(QSize(550, 550))
self.view.resize(QSize(700, 550))
self.view.show()
#self.view.showFullScreen()
#self.view.setGeometry(self.app.desktop().availableGeometry())
def set_properties_from_scene(self):
self.view.setTitle("FireSim - %s" % self.scene.name)
cw, ch = self.scene.extents
self.canvas.setWidth(cw)
self.canvas.setHeight(ch)
@pyqtSlot(float)
def set_target_fps(self, fps):
log.info("Target FPS: %d" % fps)
self.target_fps = fps
self.redraw_timer.stop()
self.redraw_timer.setInterval(int(1 / fps))
self.redraw_timer.start()
@pyqtSlot()
def about_to_quit(self):
rect = self.view.geometry()
self.config["window-geometry"] = [rect.x(), rect.y(),
rect.width(), rect.height()]
self.config.save()
@pyqtSlot()
def quit(self):
self.about_to_quit()
self.app.quit()
def run(self):
return self.app.exec_()
def on_close(self, e):
self.netcontroller.running = False
#self.net_thread.quit()
if self.args.profile:
try:
import yappi
yappi.get_func_stats().print_all()
except ImportError:
pass
@pyqtSlot()
def on_network_event(self):
self.canvas.update()
@pyqtSlot()
def on_btn_open(self):
file_name = QFileDialog.getOpenFileName(self.app.focusWidget(),
"Open Scene", "",
"Scene Files (*.json)")
if len(file_name[0]) > 0:
self.scene.save()
self.redraw_timer.stop()
self.scene.set_filepath_and_load(file_name[0])
self.config['last-opened-scene'] = file_name[0]
self.config.save()
self.set_properties_from_scene()
self.redraw_timer.start()
@pyqtSlot()
def on_btn_new(self):
self.scene.save()
self.scene.new()
self.set_properties_from_scene()
@pyqtSlot()
def on_btn_save(self):
if self.scene.filepath is None or self.scene.filepath == "":
filepath = QFileDialog.getSaveFileName(self.app.focusWidget(),
"Save Scene As", "",
"Scene Files (*.json)")
self.scene.filepath = filepath[0]
self.scene.save()
@pyqtSlot()
def on_btn_open_backdrop(self):
file_types = "Image Files (*.png *.jpg *.bmp *.jpeg *.xpm *.xbm *.ppm)"
file_name = QFileDialog.getOpenFileName(self.app.focusWidget(),
"Open Scene", "",
file_types)
if len(file_name[0]) > 0:
self.scene.backdrop_filepath = file_name[0]