145 lines
5.4 KiB
Python
145 lines
5.4 KiB
Python
"""Camera matrices recorded from the pre-refactor GL compatibility path.
|
|
|
|
References were captured with Mesa llvmpipe's OpenGL 4.5 compatibility context
|
|
using an 800x600 viewport, center (1.25, -2.5, 3.75), and extents (4, 6, 8).
|
|
OpenGL returns column-major matrices, hence the transpose before comparison.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
import numpy as np
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)),
|
|
"..", "..", "lib", "python"))
|
|
|
|
import glnav
|
|
|
|
|
|
class Navigation(glnav.GlNavBase):
|
|
def _redraw(self):
|
|
pass
|
|
|
|
def winfo_width(self):
|
|
return 800
|
|
|
|
def winfo_height(self):
|
|
return 600
|
|
|
|
def extents_info(self):
|
|
return (1.25, -2.5, 3.75), (4.0, 6.0, 8.0)
|
|
|
|
def is_lathe(self):
|
|
return self.lathe
|
|
|
|
|
|
def legacy_matrix(columns):
|
|
return np.asarray(columns, dtype=np.float64).T
|
|
|
|
|
|
REFERENCES = {
|
|
"rotate_snap": [[1, 0, 0, 0], [0, -4.3711388e-08, 1, 0],
|
|
[0, -1, -4.3711388e-08, 0], [0, 1.25, 6.25, 1]],
|
|
"translate_zoom_boost": [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0],
|
|
[0.1273418665, 0.0955063999, 0, 1]],
|
|
"view_x": [[-4.3711388e-08, 0, 1, 0], [1, -4.3711388e-08, 4.3711388e-08, 0],
|
|
[4.3711388e-08, 1, 1.9106855e-15, 0], [2.49999976, -3.75, -1.24999988, 1]],
|
|
"view_y": [[-4.3711388e-08, -1, 4.3711388e-08, 0], [0, -4.3711388e-08, -1, 0],
|
|
[1, -4.3711388e-08, 1.9106855e-15, 0], [-3.75, 1.25, -2.5, 1]],
|
|
"view_y2": [[-4.3711388e-08, 1, 4.3711388e-08, 0], [0, -4.3711388e-08, 1, 0],
|
|
[1, 4.3711388e-08, 1.9106855e-15, 0], [-3.75, -1.25000024, 2.5, 1]],
|
|
"view_z": [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [-1.25, 2.5, -3.75, 1]],
|
|
"view_z2": [[-4.3711388e-08, -1, 0, 0], [1, -4.3711388e-08, 0, 0],
|
|
[0, 0, 1, 0], [2.5, 1.24999988, -3.75, 1]],
|
|
"view_p": [[.9063076973, -.2113092095, .3659983277, 0],
|
|
[.4226184487, .4531538188, -.7848855257, 0],
|
|
[0, .8660254478, .4999999702, 0],
|
|
[-.0763385296, -1.850574255, -4.29471159, 1]],
|
|
}
|
|
|
|
|
|
class ExplicitCameraMatrixTest(unittest.TestCase):
|
|
def nav(self, lathe=False):
|
|
result = Navigation()
|
|
result.lathe = lathe
|
|
return result
|
|
|
|
def assert_matrix(self, name, nav):
|
|
np.testing.assert_allclose(nav.get_modelview_matrix(), legacy_matrix(REFERENCES[name]),
|
|
rtol=0, atol=3e-6)
|
|
|
|
def test_rotation_snaps_to_ninety_degrees(self):
|
|
nav = self.nav()
|
|
nav.set_centerpoint(1.25, -2.5, 3.75)
|
|
nav.lat = 89
|
|
nav.recordMouse(0, 0)
|
|
nav.rotate(4, 2) # 90° and 2° both snap to legacy 90°/0° angles.
|
|
self.assert_matrix("rotate_snap", nav)
|
|
|
|
def test_translate_uses_projection_and_zoom_boost(self):
|
|
nav = self.nav()
|
|
nav.distance = 2.0
|
|
nav.recordMouse(10, 20)
|
|
nav.translate(30, 5)
|
|
self.assert_matrix("translate_zoom_boost", nav)
|
|
|
|
def test_zoom_changes_distance_but_not_modelview(self):
|
|
nav = self.nav()
|
|
original = nav.get_modelview_matrix()
|
|
nav.zoomin()
|
|
nav.zoomout()
|
|
np.testing.assert_allclose(nav.get_modelview_matrix(), original)
|
|
self.assertAlmostEqual(nav.distance, 9.98)
|
|
|
|
def test_view_presets_match_legacy(self):
|
|
for name, lathe in (("x", False), ("y", True), ("y2", False),
|
|
("z", False), ("z2", False), ("p", False)):
|
|
with self.subTest(view=name):
|
|
nav = self.nav(lathe)
|
|
getattr(nav, "set_view_" + name)()
|
|
self.assert_matrix("view_" + name, nav)
|
|
|
|
def test_translate_modelview_matches_manual_composition(self):
|
|
nav = self.nav()
|
|
nav.set_view_p()
|
|
before = nav.get_modelview_matrix()
|
|
nav.translate_modelview(3.5, -1.75, 0.25)
|
|
expected = glnav.multiply(before, glnav.translation_matrix(3.5, -1.75, 0.25))
|
|
np.testing.assert_allclose(nav.get_modelview_matrix(), expected)
|
|
|
|
def test_translate_modelview_offset_survives_plasmac_sequence(self):
|
|
# The plasmac table view: top view, offset to the machine table centre,
|
|
# then refit the eyepoint. The offset has to survive the refit.
|
|
xcenter, ycenter = 12.0, -7.5
|
|
nav = self.nav()
|
|
nav.set_view_z()
|
|
after_preset = nav.get_modelview_matrix()
|
|
nav.translate_modelview(-xcenter, -ycenter, 0)
|
|
nav.set_eyepoint_from_extents(4.0, 6.0)
|
|
expected = glnav.multiply(after_preset,
|
|
glnav.translation_matrix(-xcenter, -ycenter, 0))
|
|
np.testing.assert_allclose(nav.get_modelview_matrix(), expected)
|
|
|
|
def test_translate_modelview_leaves_navigation_state_alone(self):
|
|
nav = self.nav()
|
|
nav.set_view_p()
|
|
nav.recordMouse(10, 20)
|
|
nav.translate(30, 5)
|
|
state = (nav.lat, nav.lon, nav.distance, nav._totalx, nav._totaly)
|
|
nav.translate_modelview(-4.0, 2.5, 0)
|
|
self.assertEqual((nav.lat, nav.lon, nav.distance, nav._totalx, nav._totaly), state)
|
|
|
|
def test_ortho_projection_matches_legacy_scale(self):
|
|
nav = self.nav()
|
|
nav.distance = 2.0
|
|
k = 2.0 ** .55555
|
|
expected = glnav.multiply(
|
|
glnav.ortho_matrix(-k, k, -k * 600 / 800, k * 600 / 800, -1000, 1000),
|
|
glnav.translation_matrix(0, 0, -1))
|
|
np.testing.assert_allclose(nav.get_projection_matrix(800, 600), expected)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|