vispy 0.14.0__cp311-cp311-macosx_11_0_arm64.whl

vispy/scene/cameras/tests/test_link.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright (c) Vispy Development Team. All Rights Reserved.
+# Distributed under the (new) BSD License. See LICENSE.txt for more info.
+# -----------------------------------------------------------------------------
+from vispy.scene.widgets import ViewBox
+from vispy.testing import run_tests_if_main
+
+
+def test_turntable_camera_link():
+    vbs = [ViewBox(camera='turntable') for _ in range(3)]
+    cams = [vb.camera for vb in vbs]
+
+    for cam in cams:
+        cam.elevation = 45.0
+        cam.azimuth = 120.0
+        cam.scale_factor = 4.0
+
+    cams[0].link(cams[1])
+    cams[0].link(cams[2], props=['azimuth', 'elevation'])
+
+    cams[1].elevation = 30.0
+    cams[1].azimuth = 90.0
+    cams[1].scale_factor = 2.0
+
+    assert cams[0].elevation == 30.0
+    assert cams[0].azimuth == 90.0
+    assert cams[0].scale_factor == 2.0
+
+    assert cams[2].elevation == 30.0
+    assert cams[2].azimuth == 90.0
+    assert cams[2].scale_factor == 4.0
+
+
+def test_panzoom_link():
+    vbs = [ViewBox(camera='panzoom') for _ in range(4)]
+    cams = [vb.camera for vb in vbs]
+
+    for cam in cams:
+        cam.rect = (0, 0, 100, 100)
+
+    cams[0].link(cams[1])
+    cams[0].link(cams[2], axis='x')
+    cams[0].link(cams[3], axis='y')
+
+    cams[1].rect = (-20, -20, 130, 130)
+
+    assert cams[0].rect.pos == (-20, -20) and cams[0].rect.size == (130, 130)
+    assert cams[2].rect.pos == (-20, 0) and cams[2].rect.size == (130, 100)
+    assert cams[3].rect.pos == (0, -20) and cams[3].rect.size == (100, 130)
+
+
+run_tests_if_main()

vispy/scene/cameras/tests/test_perspective.py

@@ -0,0 +1,122 @@
+# -*- coding: utf-8 -*-
+# -----------------------------------------------------------------------------
+# Copyright (c) Vispy Development Team. All Rights Reserved.
+# Distributed under the (new) BSD License. See LICENSE.txt for more info.
+# -----------------------------------------------------------------------------
+import sys
+
+import numpy as np
+import pytest
+
+from vispy import scene, io
+from vispy.testing import (requires_application, TestingCanvas,
+                           run_tests_if_main)
+from vispy.testing.image_tester import assert_image_approved
+
+
+@pytest.mark.xfail('darwin' in sys.platform, reason="Differences in OSX rendering")
+@requires_application()
+def test_perspective_render():
+    with TestingCanvas(size=(120, 200)) as canvas:
+
+        grid = canvas.central_widget.add_grid()
+        imdata = io.load_crate().astype('float32') / 255
+
+        views = []
+        images = []
+        for i, imethod in enumerate(['impostor', 'subdivide']):
+            v = grid.add_view(row=i, col=0, border_color='white')
+            v.camera = 'turntable'
+            v.camera.fov = 50
+            v.camera.distance = 30
+
+            views.append(v)
+            image = scene.visuals.Image(imdata, method=imethod,
+                                        grid=(4, 4))
+            image.transform = scene.STTransform(translate=(-12.8, -12.8),
+                                                scale=(0.1, 0.1))
+            v.add(image)
+            images.append(image)
+
+        image = canvas.render()
+        canvas.close()
+
+        # Allow many pixels to differ by a small amount--texture sampling and
+        # exact triangle position will differ across platforms. However a
+        # change in perspective or in the widget borders should trigger a
+        # failure.
+        assert_image_approved(image, 'scene/cameras/perspective_test.png',
+                              'perspective test 1: 2 identical views with '
+                              'correct perspective',
+                              px_threshold=20,
+                              px_count=60,
+                              max_px_diff=200)
+
+
+@requires_application()
+def test_panzoom_center():
+    with TestingCanvas(size=(120, 200)) as canvas:
+        grid = canvas.central_widget.add_grid()
+        imdata = io.load_crate().astype('float32') / 255
+
+        v = grid.add_view(row=0, col=0)
+        v.camera = 'panzoom'
+
+        image = scene.visuals.Image(imdata)
+        image.transform = scene.STTransform(translate=(-12.8, -12.8),
+                                            scale=(0.1, 0.1))
+        v.add(image)
+
+        result1 = canvas.render()[..., :3]
+        assert v.camera.center == (0.5, 0.5, 0)
+
+        v.camera.center = (-12.8, -12.8, 0)
+        result2 = canvas.render()[..., :3]
+
+        assert not np.allclose(result1, result2)
+        # we moved to the lower-left corner of the image that means only the
+        # upper-right quadrant should have data, the rest is black background
+        np.testing.assert_allclose(result2[100:, :], 0)
+        np.testing.assert_allclose(result2[:, :60], 0)
+        assert not np.allclose(result2[:100, 60:], 0)
+        assert v.camera.center == (-12.8, -12.8, 0)
+
+
+@requires_application()
+def test_panzoom_gesture_zoom():
+    with TestingCanvas(size=(120, 200)) as canvas:
+        view = canvas.central_widget.add_view()
+        imdata = io.load_crate().astype('float32') / 255
+        scene.visuals.Image(imdata, parent=view.scene)
+        view.camera = scene.PanZoomCamera(aspect=1)
+
+        assert view.camera.rect.size == (1, 1)
+
+        canvas.events.touch(
+            type="gesture_zoom",
+            pos=(60, 100),
+            scale=-1.0,
+        )
+
+        assert view.camera.rect.size == (2, 2)
+
+
+@requires_application()
+def test_turntable_gesture_zoom():
+    with TestingCanvas(size=(120, 200)) as canvas:
+        view = canvas.central_widget.add_view()
+        imdata = io.load_crate().astype('float32') / 255
+        scene.visuals.Image(imdata, parent=view.scene)
+        view.camera = scene.TurntableCamera()
+
+        initial_scale_factor = view.camera.scale_factor
+        canvas.events.touch(
+            type="gesture_zoom",
+            pos=(60, 100),
+            scale=-1.0,
+        )
+
+        assert view.camera.scale_factor == 2 * initial_scale_factor
+
+
+run_tests_if_main()

vispy/scene/cameras/turntable.py

@@ -0,0 +1,173 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) Vispy Development Team. All Rights Reserved.
+# Distributed under the (new) BSD License. See LICENSE.txt for more info.
+
+from __future__ import division
+
+import numpy as np
+
+from ...util import transforms
+from .perspective import Base3DRotationCamera
+
+
+class TurntableCamera(Base3DRotationCamera):
+    """3D camera class that orbits around a center point while
+    maintaining a view on a center point.
+
+    For this camera, the ``scale_factor`` indicates the zoom level, and
+    the ``center`` indicates the position to put at the center of the
+    view.
+
+    When ``elevation`` and ``azimuth`` are set to 0, the camera
+    points along the +y axis.
+
+    Parameters
+    ----------
+    fov : float
+        Field of view. 0.0 means orthographic projection,
+        default is 45.0 (some perspective)
+    elevation : float
+        Elevation angle in degrees. The elevation angle represents a
+        rotation of the camera around the current scene x-axis. The
+        camera points along the x-y plane when the angle is 0.
+    azimuth : float
+        Azimuth angle in degrees. The azimuth angle represents a
+        rotation of the camera around the scene z-axis according to the
+        right-hand screw rule. The camera points along the y-z plane when
+        the angle is 0.
+    roll : float
+        Roll angle in degrees. The roll angle represents a rotation of
+        the camera around the current scene y-axis.
+    distance : float | None
+        The distance of the camera from the rotation point (only makes sense
+        if fov > 0). If None (default) the distance is determined from the
+        scale_factor and fov.
+    translate_speed : float
+        Scale factor on translation speed when moving the camera center point.
+    **kwargs : dict
+        Keyword arguments to pass to `BaseCamera`.
+
+    Notes
+    -----
+    Interaction:
+
+        * LMB: orbits the view around its center point.
+        * RMB or scroll: change scale_factor (i.e. zoom level)
+        * SHIFT + LMB: translate the center point
+        * SHIFT + RMB: change FOV
+
+    """
+
+    _state_props = Base3DRotationCamera._state_props + ("elevation", "azimuth", "roll")
+
+    def __init__(
+        self,
+        fov=45.0,
+        elevation=30.0,
+        azimuth=30.0,
+        roll=0.0,
+        distance=None,
+        translate_speed=1.0,
+        **kwargs
+    ):
+        super(TurntableCamera, self).__init__(fov=fov, **kwargs)
+
+        # Set camera attributes
+        self.azimuth = azimuth
+        self.elevation = elevation
+        self.roll = roll
+        self.distance = distance  # None means auto-distance
+        self.translate_speed = translate_speed
+
+    @property
+    def elevation(self):
+        """Get the camera elevation angle in degrees.
+        
+        The camera points along the x-y plane when the angle is 0.
+        """
+        return self._elevation
+
+    @elevation.setter
+    def elevation(self, elev):
+        elev = float(elev)
+        self._elevation = min(90, max(-90, elev))
+        self.view_changed()
+
+    @property
+    def azimuth(self):
+        """Get the camera azimuth angle in degrees.
+        
+        The camera points along the y-z plane when the angle is 0.
+        """
+        return self._azimuth
+
+    @azimuth.setter
+    def azimuth(self, azim):
+        azim = float(azim)
+        while azim < -180:
+            azim += 360
+        while azim > 180:
+            azim -= 360
+        self._azimuth = azim
+        self.view_changed()
+
+    @property
+    def roll(self):
+        """Get the camera roll angle in degrees."""
+        return self._roll
+
+    @roll.setter
+    def roll(self, roll):
+        roll = float(roll)
+        while roll < -180:
+            roll += 360
+        while roll > 180:
+            roll -= 360
+        self._roll = roll
+        self.view_changed()
+
+    def orbit(self, azim, elev):
+        """Orbits the camera around the center position.
+
+        Parameters
+        ----------
+        azim : float
+            Angle in degrees to rotate horizontally around the center point.
+        elev : float
+            Angle in degrees to rotate vertically around the center point.
+        """
+        self.azimuth += azim
+        self.elevation = np.clip(self.elevation + elev, -90, 90)
+        self.view_changed()
+
+    def _update_rotation(self, event):
+        """Update rotation parmeters based on mouse movement"""
+        p1 = event.mouse_event.press_event.pos
+        p2 = event.mouse_event.pos
+        if self._event_value is None:
+            self._event_value = self.azimuth, self.elevation
+        self.azimuth = self._event_value[0] - (p2 - p1)[0] * 0.5
+        self.elevation = self._event_value[1] + (p2 - p1)[1] * 0.5
+
+    def _get_rotation_tr(self):
+        """Return a rotation matrix based on camera parameters"""
+        up, forward, right = self._get_dim_vectors()
+        matrix = (
+            transforms.rotate(self.elevation, -right)
+            .dot(transforms.rotate(self.azimuth, up))
+            .dot(transforms.rotate(self.roll, forward))
+        )
+        return matrix
+
+    def _dist_to_trans(self, dist):
+        """Convert mouse x, y movement into x, y, z translations"""
+        rae = np.array([self.roll, self.azimuth, self.elevation]) * np.pi / 180
+        sro, saz, sel = np.sin(rae)
+        cro, caz, cel = np.cos(rae)
+        d0, d1 = dist[0], dist[1]
+        dx = (+ d0 * (cro * caz + sro * sel * saz)
+              + d1 * (sro * caz - cro * sel * saz)) * self.translate_speed
+        dy = (+ d0 * (cro * saz - sro * sel * caz)
+              + d1 * (sro * saz + cro * sel * caz)) * self.translate_speed
+        dz = (- d0 * sro * cel + d1 * cro * cel) * self.translate_speed
+        return dx, dy, dz