Commit | Line | Data |
---|---|---|
8ee66edd | 1 | import argparse |
4894bb48 | 2 | import simplepbr |
420ce99a | 3 | #import gltf |
63e7aeb2 | 4 | from glob import glob |
a2a89363 FC |
5 | from importlib import import_module |
6 | from inspect import isclass | |
63e7aeb2 | 7 | from sys import platform, argv, exit |
94a18c21 | 8 | from logging import info, debug |
8ee66edd FC |
9 | from os.path import exists |
10 | from os import makedirs | |
6fff1464 | 11 | from panda3d.core import Filename, load_prc_file_data, AntialiasAttrib, \ |
882c058d | 12 | Texture, WindowProperties, LVector2i, TextNode |
1be87278 | 13 | from panda3d.bullet import BulletWorld, BulletDebugNode |
8ee66edd | 14 | from direct.showbase.ShowBase import ShowBase |
882c058d | 15 | from direct.gui.OnscreenText import OnscreenText |
5964572b | 16 | from direct.fsm.FSM import FSM |
b35b1f62 FC |
17 | from audio.music import MusicMgr |
18 | from logics.items.background import Background | |
19 | from gui.menu import Menu | |
20 | from logics.scene import Scene | |
21 | from logics.scenes.scene_basketball import SceneBasketBall | |
22 | from logics.scenes.scene_box import SceneBox | |
23 | from logics.scenes.scene_domino_box_basketball import SceneDominoBoxBasketball | |
24 | from logics.scenes.scene_domino_box import SceneDominoBox | |
25 | from logics.scenes.scene_domino import SceneDomino | |
26 | from logics.scenes.scene_teeter_domino_box_basketball import SceneTeeterDominoBoxBasketball | |
27 | from logics.scenes.scene_teeter_tooter import SceneTeeterTooter | |
2d1773b1 | 28 | from logics.posmgr import PositionMgr |
b35b1f62 FC |
29 | from ya2.utils.dictfile import DctFile |
30 | from ya2.p3d.p3d import LibP3d | |
31 | from ya2.utils.lang import LangMgr | |
32 | from ya2.utils.log import LogMgr | |
33 | from ya2.utils.functional import FunctionalTest | |
8ee66edd FC |
34 | |
35 | ||
5964572b FC |
36 | class MainFsm(FSM): |
37 | ||
38 | def __init__(self, pmachines): | |
39 | super().__init__('Main FSM') | |
40 | self._pmachines = pmachines | |
41 | ||
42 | def enterMenu(self): | |
43 | self._pmachines.on_menu_enter() | |
44 | ||
45 | def exitMenu(self): | |
46 | self._pmachines.on_menu_exit() | |
47 | ||
8c9bf90e FC |
48 | def enterScene(self, cls): |
49 | self._pmachines.on_scene_enter(cls) | |
5964572b FC |
50 | |
51 | def exitScene(self): | |
52 | self._pmachines.on_scene_exit() | |
53 | ||
54 | ||
c8035584 | 55 | class PmachinesApp: |
8ee66edd | 56 | |
6168d0c2 FC |
57 | scenes = [ |
58 | SceneDomino, | |
59 | SceneBox, | |
60 | SceneDominoBox, | |
61 | SceneBasketBall, | |
62 | SceneDominoBoxBasketball, | |
63 | SceneTeeterTooter, | |
64 | SceneTeeterDominoBoxBasketball] | |
65 | ||
8ee66edd | 66 | def __init__(self): |
8ee66edd FC |
67 | info('platform: %s' % platform) |
68 | info('exists main.py: %s' % exists('main.py')) | |
e982cdde | 69 | self._args = args = self._parse_args() |
d18f757d FC |
70 | self._configure(args) |
71 | self.base = ShowBase() | |
8ce16d6c | 72 | self._pipeline = None |
8ee66edd FC |
73 | self.updating = args.update |
74 | self.version = args.version | |
8ce16d6c | 75 | self.log_mgr = LogMgr.init_cls()(self) |
2d1773b1 | 76 | self._pos_mgr = PositionMgr() |
8ce16d6c | 77 | self._prepare_window(args) |
8ee66edd FC |
78 | if args.update: |
79 | return | |
361d3942 | 80 | if args.functional_test: |
edeef6f9 | 81 | self._options['settings']['volume'] = 0 |
e1e44d5c | 82 | self._music = MusicMgr(self._options['settings']['volume']) |
a0b33e12 | 83 | self.lang_mgr = LangMgr(self._options['settings']['language'], |
2aaa10d3 FC |
84 | 'pmachines', |
85 | 'assets/locale/') | |
5964572b | 86 | self._fsm = MainFsm(self) |
a747111f | 87 | if args.screenshot: |
6168d0c2 FC |
88 | cls = [cls for cls in self.scenes if cls.__name__ == args.screenshot][0] |
89 | scene = cls(BulletWorld(), None, True, False, lambda: None, self.scenes) | |
a747111f FC |
90 | scene.screenshot() |
91 | scene.destroy() | |
63e7aeb2 FC |
92 | exit() |
93 | elif self._options['development']['auto_start']: | |
b35b1f62 | 94 | mod_name = 'logics.scenes.scene_' + self._options['development']['auto_start'] |
a2a89363 FC |
95 | for member in import_module(mod_name).__dict__.values(): |
96 | if isclass(member) and issubclass(member, Scene) and \ | |
97 | member != Scene: | |
98 | cls = member | |
99 | self._fsm.demand('Scene', cls) | |
100 | else: | |
101 | self._fsm.demand('Menu') | |
edeef6f9 | 102 | if args.functional_test or args.functional_ref: |
2d1773b1 | 103 | FunctionalTest(args.functional_ref, self._pos_mgr) |
5964572b FC |
104 | |
105 | def on_menu_enter(self): | |
106 | self._menu_bg = Background() | |
a9aba267 FC |
107 | self._menu = Menu( |
108 | self._fsm, self.lang_mgr, self._options, self._music, | |
2d1773b1 FC |
109 | self._pipeline, self.scenes, self._args.functional_test or self._args.functional_ref, |
110 | self._pos_mgr) | |
5964572b FC |
111 | |
112 | def on_home(self): | |
113 | self._fsm.demand('Menu') | |
114 | ||
115 | def on_menu_exit(self): | |
116 | self._menu_bg.destroy() | |
4071c6d8 | 117 | self._menu.destroy() |
5964572b | 118 | |
8c9bf90e | 119 | def on_scene_enter(self, cls): |
1be87278 | 120 | self._set_physics() |
8c9bf90e | 121 | self._scene = cls( |
e669403e | 122 | self.world, self.on_home, |
31237524 | 123 | self._options['development']['auto_close_instructions'], |
9914cfc9 | 124 | self._options['development']['debug_items'], |
6168d0c2 | 125 | self.reload, |
2d1773b1 FC |
126 | self.scenes, |
127 | self._pos_mgr) | |
5964572b FC |
128 | |
129 | def on_scene_exit(self): | |
130 | self._unset_physics() | |
131 | self._scene.destroy() | |
8ee66edd | 132 | |
9914cfc9 FC |
133 | def reload(self, cls): |
134 | self._fsm.demand('Scene', cls) | |
135 | ||
d18f757d | 136 | def _configure(self, args): |
8ee66edd | 137 | load_prc_file_data('', 'window-title pmachines') |
4894bb48 | 138 | load_prc_file_data('', 'framebuffer-srgb true') |
a9e8696e | 139 | load_prc_file_data('', 'sync-video true') |
d982c0a5 | 140 | if args.functional_test or args.functional_ref: |
addec9c9 | 141 | load_prc_file_data('', 'win-size 1360 768') |
d982c0a5 | 142 | # otherwise it is not centered in exwm |
9914cfc9 FC |
143 | # load_prc_file_data('', 'threading-model Cull/Draw') |
144 | # it freezes when you go to the next scene | |
a747111f | 145 | if args.screenshot: |
d18f757d FC |
146 | load_prc_file_data('', 'window-type offscreen') |
147 | load_prc_file_data('', 'audio-library-name null') | |
8ee66edd FC |
148 | |
149 | def _parse_args(self): | |
150 | parser = argparse.ArgumentParser() | |
151 | parser.add_argument('--update', action='store_true') | |
152 | parser.add_argument('--version', action='store_true') | |
9ba5488b | 153 | parser.add_argument('--optfile') |
a747111f | 154 | parser.add_argument('--screenshot') |
361d3942 | 155 | parser.add_argument('--functional-test', action='store_true') |
edeef6f9 | 156 | parser.add_argument('--functional-ref', action='store_true') |
8ee66edd FC |
157 | cmd_line = [arg for arg in iter(argv[1:]) if not arg.startswith('-psn_')] |
158 | args = parser.parse_args(cmd_line) | |
159 | return args | |
160 | ||
9ba5488b | 161 | def _prepare_window(self, args): |
8ee66edd FC |
162 | data_path = '' |
163 | if (platform.startswith('win') or platform.startswith('linux')) and ( | |
164 | not exists('main.py') or __file__.startswith('/app/bin/')): | |
165 | # it is the deployed version for windows | |
166 | data_path = str(Filename.get_user_appdata_directory()) + '/pmachines' | |
167 | home = '/home/flavio' # we must force this for wine | |
168 | if data_path.startswith('/c/users/') and exists(home + '/.wine/'): | |
169 | data_path = home + '/.wine/drive_' + data_path[1:] | |
170 | info('creating dirs: %s' % data_path) | |
171 | makedirs(data_path, exist_ok=True) | |
9ba5488b FC |
172 | optfile = args.optfile if args.optfile else 'options.ini' |
173 | info('data path: %s' % data_path) | |
174 | info('option file: %s' % optfile) | |
175 | info('fixed path: %s' % LibP3d.fixpath(data_path + '/' + optfile)) | |
176 | default_opt = { | |
177 | 'settings': { | |
a0b33e12 | 178 | 'volume': 1, |
6fff1464 FC |
179 | 'language': 'en', |
180 | 'fullscreen': 1, | |
a9aba267 | 181 | 'resolution': '', |
5fdf77d0 FC |
182 | 'antialiasing': 1, |
183 | 'shadows': 1}, | |
9ba5488b | 184 | 'development': { |
a5dc83f4 | 185 | 'simplepbr': 1, |
9ba5488b | 186 | 'verbose_log': 0, |
e669403e FC |
187 | 'physics_debug': 0, |
188 | 'auto_start': 0, | |
189 | 'auto_close_instructions': 0, | |
31237524 | 190 | 'show_buffers': 0, |
b41381b2 | 191 | 'debug_items': 0, |
882c058d | 192 | 'mouse_coords': 0, |
b41381b2 | 193 | 'fps': 0}} |
9ba5488b FC |
194 | opt_path = LibP3d.fixpath(data_path + '/' + optfile) if data_path else optfile |
195 | opt_exists = exists(opt_path) | |
196 | self._options = DctFile( | |
197 | LibP3d.fixpath(data_path + '/' + optfile) if data_path else optfile, | |
198 | default_opt) | |
199 | if not opt_exists: | |
200 | self._options.store() | |
6fff1464 FC |
201 | res = self._options['settings']['resolution'] |
202 | if res: | |
203 | res = LVector2i(*[int(_res) for _res in res.split('x')]) | |
204 | else: | |
dd32d640 FC |
205 | resolutions = [] |
206 | if not self.version: | |
207 | d_i = base.pipe.get_display_information() | |
208 | def _res(idx): | |
209 | return d_i.get_display_mode_width(idx), \ | |
210 | d_i.get_display_mode_height(idx) | |
211 | resolutions = [ | |
212 | _res(idx) for idx in range(d_i.get_total_display_modes())] | |
6fff1464 | 213 | res = sorted(resolutions)[-1] |
edeef6f9 | 214 | fullscreen = self._options['settings']['fullscreen'] |
d982c0a5 | 215 | props = WindowProperties() |
edeef6f9 | 216 | if args.functional_test or args.functional_ref: |
edeef6f9 | 217 | fullscreen = False |
d982c0a5 FC |
218 | else: |
219 | props.set_size(res) | |
edeef6f9 | 220 | props.set_fullscreen(fullscreen) |
7e487769 | 221 | props.set_icon_filename('assets/images/icon/pmachines.ico') |
8ce16d6c | 222 | if not args.screenshot and not self.version: |
d18f757d | 223 | base.win.request_properties(props) |
420ce99a | 224 | #gltf.patch_loader(base.loader) |
8ce16d6c | 225 | if self._options['development']['simplepbr'] and not self.version: |
a9aba267 | 226 | self._pipeline = simplepbr.init( |
a5dc83f4 FC |
227 | use_normal_maps=True, |
228 | use_emission_maps=False, | |
a9aba267 | 229 | use_occlusion_maps=True, |
5fdf77d0 FC |
230 | msaa_samples=4 if self._options['settings']['antialiasing'] else 1, |
231 | enable_shadows=int(self._options['settings']['shadows'])) | |
94a18c21 FC |
232 | debug(f'msaa: {self._pipeline.msaa_samples}') |
233 | debug(f'shadows: {self._pipeline.enable_shadows}') | |
4894bb48 | 234 | render.setAntialias(AntialiasAttrib.MAuto) |
1be87278 FC |
235 | self.base.set_background_color(0, 0, 0, 1) |
236 | self.base.disable_mouse() | |
e669403e FC |
237 | if self._options['development']['show_buffers']: |
238 | base.bufferViewer.toggleEnable() | |
b41381b2 FC |
239 | if self._options['development']['fps']: |
240 | base.set_frame_rate_meter(True) | |
651713a9 FC |
241 | #self.base.accept('window-event', self._on_win_evt) |
242 | self.base.accept('aspectRatioChanged', self._on_aspect_ratio_changed) | |
882c058d FC |
243 | if self._options['development']['mouse_coords']: |
244 | coords_txt = OnscreenText( | |
245 | '', parent=base.a2dTopRight, scale=0.04, | |
246 | pos=(-.03, -.06), fg=(.9, .9, .9, 1), align=TextNode.A_right) | |
247 | def update_coords(task): | |
248 | txt = '%s %s' % (int(base.win.get_pointer(0).x), | |
249 | int(base.win.get_pointer(0).y)) | |
250 | coords_txt['text'] = txt | |
251 | return task.cont | |
252 | taskMgr.add(update_coords, 'update_coords') | |
253 | ||
1be87278 | 254 | def _set_physics(self): |
9ba5488b FC |
255 | if self._options['development']['physics_debug']: |
256 | debug_node = BulletDebugNode('Debug') | |
257 | debug_node.show_wireframe(True) | |
258 | debug_node.show_constraints(True) | |
259 | debug_node.show_bounding_boxes(True) | |
260 | debug_node.show_normals(True) | |
5964572b FC |
261 | self._debug_np = render.attach_new_node(debug_node) |
262 | self._debug_np.show() | |
1be87278 FC |
263 | self.world = BulletWorld() |
264 | self.world.set_gravity((0, 0, -9.81)) | |
9ba5488b | 265 | if self._options['development']['physics_debug']: |
5964572b | 266 | self.world.set_debug_node(self._debug_np.node()) |
1be87278 FC |
267 | def update(task): |
268 | dt = globalClock.get_dt() | |
0625cf49 | 269 | self.world.do_physics(dt, 10, 1/180) |
1be87278 | 270 | return task.cont |
5964572b FC |
271 | self._phys_tsk = taskMgr.add(update, 'update') |
272 | ||
273 | def _unset_physics(self): | |
274 | if self._options['development']['physics_debug']: | |
275 | self._debug_np.remove_node() | |
276 | self.world = None | |
277 | taskMgr.remove(self._phys_tsk) | |
651713a9 FC |
278 | |
279 | def _on_aspect_ratio_changed(self): | |
5964572b FC |
280 | if self._fsm.state == 'Scene': |
281 | self._scene.on_aspect_ratio_changed() |