Set default yaw axis Vector3::UNIT_Y (tabletop mode) + Optional X, Z …

…and free rotation
OGRECave · Apr 20, 2024 · 0125d1c · 0125d1c
1 parent c01afeb
commit 0125d1c
Showing 1 changed file with 37 additions and 2 deletions.
diff --git a/ogre_mesh_viewer.py b/ogre_mesh_viewer.py
@@ -248,6 +248,22 @@ def preRenderTargetUpdate(self, evt):
                 if ImGui.MenuItem("Wireframe Mode", "W", app.cam.getPolygonMode() == Ogre.PM_WIREFRAME):
                     self.app._toggle_wireframe_mode()
 
+                if ImGui.BeginMenu("Fixed Camera Yaw"):
+                    if ImGui.MenuItem("Disabled", "", self.app.fixed_yaw_axis == -1):
+                        self.app.fixed_yaw_axis = -1
+                        self.app.set_orientation()
+                    ImGui.Separator()
+                    if ImGui.MenuItem("X Axis", "", self.app.fixed_yaw_axis == 0):
+                        self.app.fixed_yaw_axis = 0
+                        self.app.set_orientation()
+                    if ImGui.MenuItem("Y Axis", "", self.app.fixed_yaw_axis == 1):
+                        self.app.fixed_yaw_axis = 1
+                        self.app.set_orientation()
+                    if ImGui.MenuItem("Z Axis", "", self.app.fixed_yaw_axis == 2):
+                        self.app.fixed_yaw_axis = 2
+                        self.app.set_orientation()
+                    ImGui.EndMenu()
+
                 if entity.hasSkeleton() and ImGui.MenuItem("Show Skeleton", None, entity.getDisplaySkeleton()):
                     entity.setDisplaySkeleton(not entity.getDisplaySkeleton())
                 ImGui.EndMenu()
@@ -441,6 +457,9 @@ def __init__(self, infile, rescfg):
         self.highlight_mat = None
         self.restart = False
         self.axes_visible = False
+        self.fixed_yaw_axes = [(1, 0, 0), (0, 1, 0), (0, 0, 1)]
+        self.fixed_yaw_axis = 1
+        self.default_tilt = Ogre.Degree(20)
 
         self.active_controllers = {}
 
@@ -522,6 +541,23 @@ def _save_screenshot(self):
         self.getRenderWindow().writeContentsToTimestampedFile(outpath, ".png")
         self.cam.getViewport().setOverlaysEnabled(True)
 
+    def set_orientation(self):
+        camnode = self.camman.getCamera()
+        diam = camnode.getPosition().length()
+        camnode.setOrientation(Ogre.Quaternion.IDENTITY)
+        if self.fixed_yaw_axis >= 0:
+            camnode.setFixedYawAxis(True, self.fixed_yaw_axes[self.fixed_yaw_axis])
+            if self.fixed_yaw_axis == 0:
+                self.camman.setYawPitchDist(0, 0, diam)
+                camnode.roll(-Ogre.Degree(90))
+            elif self.fixed_yaw_axis == 1:
+                self.camman.setYawPitchDist(0, self.default_tilt, diam)
+            elif self.fixed_yaw_axis == 2:
+                self.camman.setYawPitchDist(0, self.default_tilt + Ogre.Degree(90), diam)
+        else:
+            self.camman.setYawPitchDist(0, self.default_tilt, diam)
+        self.camman.setFixedYaw(self.fixed_yaw_axis >= 0)
+
     def reload(self):
         if app.infile:
             app.restart = True
@@ -655,8 +691,7 @@ def setup(self):
 
         self.camman = OgreBites.CameraMan(camnode)
         self.camman.setStyle(OgreBites.CS_ORBIT)
-        self.camman.setYawPitchDist(0, 0.3, diam)
-        self.camman.setFixedYaw(False)
+        self.camman.setYawPitchDist(0, self.default_tilt, diam)
 
         self.input_dispatcher = OgreBites.InputListenerChain([self.getImGuiInputListener(), self.camman, self])
         self.addInputListener(self.input_dispatcher)