From 10190ecef89ba80d0cf22b7aa59c7d558089a6cf Mon Sep 17 00:00:00 2001
From: march3 <gang.wei@qq.com>
Date: Sat, 29 Jul 2023 18:46:15 +0800
Subject: [PATCH] =?UTF-8?q?Python=E8=B6=85=E4=BA=BA-=E5=AE=87=E5=AE=99?=
 =?UTF-8?q?=E6=A8=A1=E6=8B=9F=E5=99=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sim_scenes/func.py                           | 14 +++++++++-----
 simulators/ursina/entities/video_recorder.py |  4 ++--
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/sim_scenes/func.py b/sim_scenes/func.py
index 6ca214c..ce62dd1 100644
--- a/sim_scenes/func.py
+++ b/sim_scenes/func.py
@@ -123,6 +123,7 @@ def ursina_run(bodies,
                show_timer=False,
                timer_enabled=False,
                save_as_json=None,
+               save_cube_map=False,
                save_as_video=False,
                view_closely=False):
     """
@@ -172,9 +173,12 @@ def ursina_run(bodies,
     view_azimuth = 0  # 暂时未用
     player = UrsinaPlayer(position, view_azimuth, simulator.ursina_views)
 
-    # if save_as_video:
-    #     from common.video_recorder import VideoRecorder
-    #     vr = VideoRecorder()
+    if save_cube_map:
+        from simulators.ursina.entities.video_recorder import VideoRecorder
+        from ursina import camera
+        import sys
+        sys.modules["__main__"].video_recorder = VideoRecorder()
+        camera.fov = 90
 
     def callback_update():
         UrsinaEvent.on_application_run()
@@ -183,8 +187,8 @@ def ursina_run(bodies,
             if ursina_view.appeared:
                 ursina_view.update()
 
-        # if save_as_video:
-        #     vr.screenshot()
+        if save_cube_map:
+            sys.modules["__main__"].video_recorder.screenshot()
 
     import sys
     from simulators.ursina.ursina_config import UrsinaConfig
diff --git a/simulators/ursina/entities/video_recorder.py b/simulators/ursina/entities/video_recorder.py
index 9a8d047..953c71e 100644
--- a/simulators/ursina/entities/video_recorder.py
+++ b/simulators/ursina/entities/video_recorder.py
@@ -38,8 +38,8 @@ class VideoRecorder(Entity):
 
         if self.t >= 1 / self.fps:
             base.saveCubeMap(
-                namePrefix=f'\\{self.temp_dir}\\cmap_' + str(self.i).zfill(self.sd) + '_#.png',
-                size=1024
+                namePrefix=f'\\{self.temp_dir}\\cmap_' + str(self.i).zfill(self.sd) + '_#.jpg',
+                size=2048
                 # namePrefix = 'cube_map_#.png'
             )
         self.t = 0
-- 
GitLab