def load_scenario():
scene0 = Scene(
0,
"You are Homer Simpson, you want to drink a beer in bar... \n Are you ready?",
(GoodAction("yes", 1), ExitAction("no")))
scene1 = Scene(1, "You are in the living room, where do you go?",
(GoodAction("entrance", 2), GoodAction("kitchen", 3)))
scene2 = Scene(
2,
"You are opening the entrance door to leave but you meet Marge, she is angry and order you to stay at home to keep an eye on children... \n It's over for the beer today. \n Want to try again?",
(GoodAction("yes", 0), BadAction("no")))
scene3 = Scene(
3,
"You are now in the kitchen, you want go out through the window but you see the oven that catches fire. You have two choices; \n a: The most important is my beer, go out through the window \n b: Stop the fire \n What do you do?",
(GoodAction("a", 5), GoodAction("b", 4)))
scene4 = Scene(
4,
"You are trying to put out fire, but you burn yourself. \n You are now in the hospital for several days... \n Want to try again?",
(GoodAction("yes", 0), BadAction("no")))
scene5 = Scene(
5,
"You're finally out, you walk to the bar. Then you go into the bar, you are so happy! You ask the barman Moe for a beer. But your children and Marge are showing up! Your are learning the house is burned and Marge is very angry. \n You have two choices: Apologize to Marge and go to home or ignore the family and drink your beer.",
(GoodAction("apologize", 6), WinAction("drink")))
scene6 = Scene(
6,
"You're going to home and it's over for the beer... \n Want to try again?",
(GoodAction("yes", 0), BadAction("no")))
scenario = scene0, scene1, scene2, scene3, scene4, scene5, scene6
return scenario
def scene1(action3, scene2):
action1 = GoodAction('left', scene2)
action2 = GoodAction('right', scene2)
scene = Scene(1, 'my description', (action1, action2, action3))
return scene
def scene2():
return Scene(2, 'bla bla...')
def __init__(self, args=None):
QtCore.QObject.__init__(self)
self.app = QtGui.QApplication(["FireSim"])
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_strand = 0
self._selected_fixture_address = 0
self._selected_fixture_pixels = 0
self.selected_fixture = None
self.is_blurred = False
self.scene = Scene(os.path.join(self.config.get("scene_root"), self.args.scene) + ".json")
self.scenecontroller = SceneController(app=self, scene=self.scene)
QtDeclarative.qmlRegisterType(CanvasWidget, "FireSim", 1, 0, "SimCanvas")
QtDeclarative.qmlRegisterType(FixtureWidget, "FireSim", 1, 0, "Fixture")
self.view = QtDeclarative.QDeclarativeView()
self.view.setWindowTitle("FireSim")
self.view.setResizeMode(QtDeclarative.QDeclarativeView.SizeRootObjectToView)
self.view.closeEvent = self.on_close
self.context = self.view.rootContext()
self.context.setContextProperty('main', self)
self.fixture_info_list = []
self.context.setContextProperty('fixtureInfoModel', self.fixture_info_list)
self.view.setSource(QtCore.QUrl('ui/qml/FireSimGUI.qml'))
self.root = self.view.rootObject()
self.item_frame = self.root.findChild(QtDeclarative.QDeclarativeItem)
self.canvas = self.root.findChild(CanvasWidget)
self.canvas.gui = self
cw, ch = self.scenecontroller.scene.extents()
self.canvas.setWidth(cw)
self.canvas.setHeight(ch)
self.scenecontroller.set_canvas(self.canvas)
self.root.backdrop_showhide_callback.connect(self.on_btn_backdrop_showhide)
self.root.labels_showhide_callback.connect(self.on_btn_labels_showhide)
self.root.lock_callback.connect(self.on_btn_lock)
self.root.show_center_callback.connect(self.on_btn_show_center)
self.root.toggle_blurred_callback.connect(self.on_btn_toggle_blurred)
#self.net_thread = QtCore.QThread()
#self.net_thread.start()
self.netcontroller = NetController(self)
#self.netcontroller.moveToThread(self.net_thread)
#self.netcontroller.start.emit()
self.net_stats_timer = QtCore.QTimer()
self.net_stats_timer.setInterval(1000)
self.net_stats_timer.timeout.connect(self.update_net_stats)
self.net_stats_timer.start()
self.netcontroller.data_received.connect(self.on_network_event)
self.scenecontroller.new_frame.connect(self.netcontroller.frame_complete)
self.netcontroller.data_received.connect(self.scenecontroller.process_command)
self.view.setFixedSize(max(640, cw + 130), max(480, ch))
log.info("FireSimGUI Ready.")
self.view.show()
class FireSimGUI(QtCore.QObject):
def __init__(self, args=None):
QtCore.QObject.__init__(self)
self.app = QtGui.QApplication(["FireSim"])
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_strand = 0
self._selected_fixture_address = 0
self._selected_fixture_pixels = 0
self.selected_fixture = None
self.is_blurred = False
self.scene = Scene(os.path.join(self.config.get("scene_root"), self.args.scene) + ".json")
self.scenecontroller = SceneController(app=self, scene=self.scene)
QtDeclarative.qmlRegisterType(CanvasWidget, "FireSim", 1, 0, "SimCanvas")
QtDeclarative.qmlRegisterType(FixtureWidget, "FireSim", 1, 0, "Fixture")
self.view = QtDeclarative.QDeclarativeView()
self.view.setWindowTitle("FireSim")
self.view.setResizeMode(QtDeclarative.QDeclarativeView.SizeRootObjectToView)
self.view.closeEvent = self.on_close
self.context = self.view.rootContext()
self.context.setContextProperty('main', self)
self.fixture_info_list = []
self.context.setContextProperty('fixtureInfoModel', self.fixture_info_list)
self.view.setSource(QtCore.QUrl('ui/qml/FireSimGUI.qml'))
self.root = self.view.rootObject()
self.item_frame = self.root.findChild(QtDeclarative.QDeclarativeItem)
self.canvas = self.root.findChild(CanvasWidget)
self.canvas.gui = self
cw, ch = self.scenecontroller.scene.extents()
self.canvas.setWidth(cw)
self.canvas.setHeight(ch)
self.scenecontroller.set_canvas(self.canvas)
self.root.backdrop_showhide_callback.connect(self.on_btn_backdrop_showhide)
self.root.labels_showhide_callback.connect(self.on_btn_labels_showhide)
self.root.lock_callback.connect(self.on_btn_lock)
self.root.show_center_callback.connect(self.on_btn_show_center)
self.root.toggle_blurred_callback.connect(self.on_btn_toggle_blurred)
#self.net_thread = QtCore.QThread()
#self.net_thread.start()
self.netcontroller = NetController(self)
#self.netcontroller.moveToThread(self.net_thread)
#self.netcontroller.start.emit()
self.net_stats_timer = QtCore.QTimer()
self.net_stats_timer.setInterval(1000)
self.net_stats_timer.timeout.connect(self.update_net_stats)
self.net_stats_timer.start()
self.netcontroller.data_received.connect(self.on_network_event)
self.scenecontroller.new_frame.connect(self.netcontroller.frame_complete)
self.netcontroller.data_received.connect(self.scenecontroller.process_command)
self.view.setFixedSize(max(640, cw + 130), max(480, ch))
log.info("FireSimGUI Ready.")
self.view.show()
#self.view.showFullScreen()
#self.view.setGeometry(self.app.desktop().availableGeometry())
@QtCore.Slot()
def quit(self):
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
@QtCore.Slot()
def update_net_stats(self):
self.canvas.net_stats = self.netcontroller.get_stats()
@QtCore.Slot()
def on_network_event(self):
self.canvas.update()
@QtCore.Slot(result=bool)
def is_backdrop_enabled(self):
return self.scenecontroller.scene.get("backdrop_enable", False)
@QtCore.Slot(result=bool)
def are_labels_enabled(self):
return self.scenecontroller.scene.get("labels_enable", False)
@QtCore.Slot(result=bool)
def is_locked(self):
return self.scenecontroller.scene.get("locked", False)
@QtCore.Slot(result=bool)
def is_center_shown(self):
return self.scenecontroller.show_center
@QtCore.Slot(result=bool)
def is_blurred(self):
return self.is_blurred
@QtCore.Slot()
def on_btn_add_fixture(self):
self.scenecontroller.add_fixture()
@QtCore.Slot()
def on_btn_clear(self):
self.scenecontroller.clear_fixtures()
@QtCore.Slot()
def on_btn_save(self):
self.scenecontroller.save_scene()
@QtCore.Slot(result=bool)
def on_btn_backdrop_showhide(self, obj):
enabled = self.scenecontroller.toggle_background_enable()
if enabled:
obj.setProperty("text", "Hide Backdrop")
else:
obj.setProperty("text", "Show Backdrop")
return enabled
@QtCore.Slot(result=bool)
def on_btn_labels_showhide(self, obj):
enabled = self.scenecontroller.toggle_labels_enable()
if enabled:
obj.setProperty("text", "Hide Labels")
else:
obj.setProperty("text", "Show Labels")
return enabled
@QtCore.Slot(result=bool)
def on_btn_lock(self, obj):
locked = self.scenecontroller.toggle_locked()
if locked:
obj.setProperty("text", "Unlock Scene")
else:
obj.setProperty("text", "Lock Scene")
return locked
@QtCore.Slot(result=bool)
def on_btn_show_center(self, obj):
show_center = self.scenecontroller.toggle_show_center()
if show_center:
obj.setProperty("text", "Hide Center")
else:
obj.setProperty("text", "Show Center")
return show_center
@QtCore.Slot(result=bool)
def on_btn_toggle_blurred(self, obj):
self.is_blurred = not self.is_blurred
if self.is_blurred:
obj.setProperty("text", "In Focus")
else:
obj.setProperty("text", "Blurred")
return self.is_blurred
def widget_selected(self, selected, fixture, multi):
self.selected_fixture = None
if multi:
pass
else:
if selected:
self.selected_fixture_strand = fixture.strand()
self.selected_fixture_address = fixture.address()
self.selected_fixture_pixels = fixture.pixels()
else:
self.selected_fixture_strand = 0
self.selected_fixture_address = 0
self.selected_fixture_pixels = 0
if selected:
self.selected_fixture = fixture
self.update_selected_fixture_properties()
def update_selected_fixture_properties(self):
if self.selected_fixture is not None:
strand = int(self.selected_fixture_strand)
address = int(self.selected_fixture_address)
self.selected_fixture.set_strand(strand)
self.selected_fixture.set_address(address)
self.selected_fixture.set_pixels(int(self.selected_fixture_pixels))
self.selected_fixture.get_widget().update()
self.scene.update_address_cache(self.selected_fixture, strand, address)
self.scenecontroller.update_scene()
self.scenecontroller.create_pixel_array()
def _get_selected_fixture_strand(self):
return self._selected_fixture_strand
def _set_selected_fixture_strand(self, strand):
if self._selected_fixture_strand == strand:
return
self._selected_fixture_strand = strand
self.update_selected_fixture_properties()
self.on_selected_fixture_strand.emit()
def _get_selected_fixture_address(self):
return self._selected_fixture_address
def _set_selected_fixture_address(self, address):
if self._selected_fixture_address == address:
return
self._selected_fixture_address = address
self.update_selected_fixture_properties()
self.on_selected_fixture_address.emit()
def _get_selected_fixture_pixels(self):
return self._selected_fixture_pixels
def _set_selected_fixture_pixels(self, pixels):
if self._selected_fixture_pixels == pixels:
return
self._selected_fixture_pixels = pixels
self.update_selected_fixture_properties()
self.on_selected_fixture_pixels.emit()
on_selected_fixture_strand = QtCore.Signal()
on_selected_fixture_address = QtCore.Signal()
on_selected_fixture_pixels = QtCore.Signal()
selected_fixture_strand = QtCore.Property(int, _get_selected_fixture_strand, _set_selected_fixture_strand, notify=on_selected_fixture_strand)
selected_fixture_address = QtCore.Property(int, _get_selected_fixture_address, _set_selected_fixture_address, notify=on_selected_fixture_address)
selected_fixture_pixels = QtCore.Property(int, _get_selected_fixture_pixels, _set_selected_fixture_pixels, notify=on_selected_fixture_pixels)
def __init__(self, args=None):
QtCore.QObject.__init__(self)
self.app = QtGui.QApplication(["FireSim"])
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_strand = 0
self._selected_fixture_address = 0
self._selected_fixture_pixels = 0
self.selected_fixture = None
self.is_blurred = False
self.scene = Scene(
os.path.join(self.config.get("scene_root"), self.args.scene) +
".json")
self.scenecontroller = SceneController(app=self, scene=self.scene)
QtDeclarative.qmlRegisterType(CanvasWidget, "FireSim", 1, 0,
"SimCanvas")
QtDeclarative.qmlRegisterType(FixtureWidget, "FireSim", 1, 0,
"Fixture")
self.view = QtDeclarative.QDeclarativeView()
self.view.setWindowTitle("FireSim")
self.view.setResizeMode(
QtDeclarative.QDeclarativeView.SizeRootObjectToView)
self.view.closeEvent = self.on_close
self.context = self.view.rootContext()
self.context.setContextProperty('main', self)
self.fixture_info_list = []
self.context.setContextProperty('fixtureInfoModel',
self.fixture_info_list)
self.view.setSource(QtCore.QUrl('ui/qml/FireSimGUI.qml'))
self.root = self.view.rootObject()
self.item_frame = self.root.findChild(QtDeclarative.QDeclarativeItem)
self.canvas = self.root.findChild(CanvasWidget)
self.canvas.gui = self
cw, ch = self.scenecontroller.scene.extents()
self.canvas.setWidth(cw)
self.canvas.setHeight(ch)
self.scenecontroller.set_canvas(self.canvas)
self.root.backdrop_showhide_callback.connect(
self.on_btn_backdrop_showhide)
self.root.labels_showhide_callback.connect(self.on_btn_labels_showhide)
self.root.lock_callback.connect(self.on_btn_lock)
self.root.show_center_callback.connect(self.on_btn_show_center)
self.root.toggle_blurred_callback.connect(self.on_btn_toggle_blurred)
#self.net_thread = QtCore.QThread()
#self.net_thread.start()
self.netcontroller = NetController(self)
#self.netcontroller.moveToThread(self.net_thread)
#self.netcontroller.start.emit()
self.net_stats_timer = QtCore.QTimer()
self.net_stats_timer.setInterval(1000)
self.net_stats_timer.timeout.connect(self.update_net_stats)
self.net_stats_timer.start()
self.netcontroller.data_received.connect(self.on_network_event)
self.scenecontroller.new_frame.connect(
self.netcontroller.frame_complete)
self.netcontroller.data_received.connect(
self.scenecontroller.process_command)
self.view.setFixedSize(max(640, cw + 130), max(480, ch))
log.info("FireSimGUI Ready.")
self.view.show()
def get_scenes(self):
scenes = list(
Scene.filter(channel_id=str(self.id), archived=False).all())
return scenes
def main():
white_diffuse = BSDF(BSDFType.Diffuse, np.array([1, 1, 1]))
red_diffuse = BSDF(BSDFType.Diffuse, np.array([1, 0, 0]))
blue_diffuse = BSDF(BSDFType.Diffuse, np.array([0, 0, 1]))
light = BSDF(BSDFType.Light, np.array([2, 2, 2]))
size = 1.0
left = size
right = -size
top = size
bottom = -size
front = size - 2
back = -size - 2
# left
nl = np.array([-1, 0, 0])
tl0 = Triangle([
Vertex(np.array([left, bottom, back]), nl, np.array([0, 0])),
Vertex(np.array([left, bottom, front]), nl, np.array([0, 0])),
Vertex(np.array([left, top, front]), nl, np.array([0, 0])),
], red_diffuse)
tl1 = Triangle([
Vertex(np.array([left, top, front]), nl, np.array([0, 0])),
Vertex(np.array([left, top, back]), nl, np.array([0, 0])),
Vertex(np.array([left, bottom, back]), nl, np.array([0, 0])),
], red_diffuse)
# right
nr = np.array([1, 0, 0])
tr0 = Triangle([
Vertex(np.array([right, bottom, front]), nr, np.array([0, 0])),
Vertex(np.array([right, bottom, back]), nr, np.array([0, 0])),
Vertex(np.array([right, top, back]), nr, np.array([0, 0])),
], blue_diffuse)
tr1 = Triangle([
Vertex(np.array([right, top, back]), nr, np.array([0, 0])),
Vertex(np.array([right, top, front]), nr, np.array([0, 0])),
Vertex(np.array([right, bottom, front]), nr, np.array([0, 0])),
], blue_diffuse)
# top
nt = np.array([0, -1, 0])
tt0 = Triangle([
Vertex(np.array([left, top, front]), nt, np.array([0, 0])),
Vertex(np.array([right, top, front]), nt, np.array([0, 0])),
Vertex(np.array([right, top, back]), nt, np.array([0, 0])),
], white_diffuse)
tt1 = Triangle([
Vertex(np.array([right, top, back]), nt, np.array([0, 0])),
Vertex(np.array([left, top, back]), nt, np.array([0, 0])),
Vertex(np.array([left, top, front]), nt, np.array([0, 0])),
], white_diffuse)
# bottom
nb_ = np.array([0, 1, 0])
tb_0 = Triangle([
Vertex(np.array([left, bottom, front]), nb_, np.array([0, 0])),
Vertex(np.array([right, bottom, front]), nb_, np.array([0, 0])),
Vertex(np.array([right, bottom, back]), nb_, np.array([0, 0])),
], white_diffuse)
tb_1 = Triangle([
Vertex(np.array([right, bottom, back]), nb_, np.array([0, 0])),
Vertex(np.array([left, bottom, back]), nb_, np.array([0, 0])),
Vertex(np.array([left, bottom, front]), nb_, np.array([0, 0])),
], white_diffuse)
# back
nb = np.array([0, 0, 1])
tb0 = Triangle([
Vertex(np.array([left, bottom, back]), nb, np.array([0, 0])),
Vertex(np.array([right, bottom, back]), nb, np.array([0, 0])),
Vertex(np.array([right, top, back]), nb, np.array([0, 0])),
], white_diffuse)
tb1 = Triangle([
Vertex(np.array([right, top, back]), nb, np.array([0, 0])),
Vertex(np.array([left, top, back]), nb, np.array([0, 0])),
Vertex(np.array([left, bottom, back]), nb, np.array([0, 0])),
], white_diffuse)
# light
l0 = Triangle([
Vertex(np.array([left, top - 0.01, back]), nt, np.array([0, 0])),
Vertex(np.array([right, top - 0.01, back]), nt, np.array([0, 0])),
Vertex(np.array([0, top - 0.01, front]), nt, np.array([0, 0])),
], light)
scene = Scene([tl0, tl1, tr0, tr1, tt0, tt1, tb_0, tb_1, tb0, tb1, l0],
[l0])
world_matrix = np.array([[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0],
[0, 0, 0, 1]])
camera = Camera(pi / 2, Film(400, 400), world_matrix)
camera.look_at(np.array([0, 0, 10]), np.array([0, 0, 0]),
np.array([-1, 0, 0]))
render_options = RenderOptions(400, 400, 2, 2, 1, 1, 2, 2.2, 0)
renderer = Renderer(render_options, camera, scene)
renderer.render()
print('Rendering completed.')
print(datetime.now())
