fastplotlib
diff --git a/‎fastplotlib/layouts/_base.py
Copy file name to clipboardExpand all lines: fastplotlib/layouts/_base.py
+35-1Lines changed: 35 additions & 1 deletion b/‎fastplotlib/layouts/_base.py
Copy file name to clipboardExpand all lines: fastplotlib/layouts/_base.py
+35-1Lines changed: 35 additions & 1 deletion
@@ -1,11 +1,11 @@
-from warnings import warn
 from typing import *
 import weakref
 
 import numpy as np
 
 from pygfx import Scene, OrthographicCamera, PerspectiveCamera, PanZoomController, OrbitController, \
     Viewport, WgpuRenderer
+from pygfx.linalg import Vector3
 from wgpu.gui.auto import WgpuCanvas
 
 from ..graphics._base import Graphic, GraphicCollection
@@ -146,6 +146,40 @@ def get_rect(self) -> Tuple[float, float, float, float]:
         """allows setting the region occupied by the viewport w.r.t. the parent"""
         raise NotImplementedError("Must be implemented in subclass")
 
+    def map_screen_to_world(self, pos: Tuple[float, float]) -> Vector3:
+        """
+        Map screen position to world position
+
+        Parameters
+        ----------
+        pos: (float, float)
+            (x, y) screen coordinates
+
+        """
+
+        if not self.viewport.is_inside(*pos):
+            return None
+
+        vs = self.viewport.logical_size
+
+        # get position relative to viewport
+        pos_rel = (
+            pos[0] - self.viewport.rect[0],
+            pos[1] - self.viewport.rect[1],
+        )
+
+        # convert screen position to NDC
+        pos_ndc = Vector3(
+            pos_rel[0] / vs[0] * 2 - 1,
+            -(pos_rel[1] / vs[1] * 2 - 1),
+            0
+        )
+
+        # get world position
+        pos_world = self.camera.position.clone().project(self.camera).add(pos_ndc).unproject(self.camera)
+
+        return pos_world
+
     def set_viewport_rect(self, *args):
         self.viewport.rect = self.get_rect()