Merge pull request #558 from bnmajor/jupyterlite

Jupyterlite

Author: thewtex

docs/jupyterlite/files/Hello3DWorld.ipynb

@@ -0,0 +1,36 @@
+from enum import Enum
+
+
+class Env(Enum):
+    JUPYTER_NOTEBOOK = 'notebook'
+    JUPYTERLAB = 'lab'
+    JUPYTERLITE = 'lite'
+    SAGEMAKER = 'sagemaker'
+    HYPHA = 'hypha'
+    COLAB = 'colab'
+
+
+def find_env():
+    try:
+        from google.colab import files
+        return Env.COLAB
+    except:
+        try:
+            from IPython import get_ipython
+            parent_header = get_ipython().parent_header
+            username = parent_header['header']['username']
+            if username == '':
+                return Env.JUPYTERLAB
+            elif username == 'username':
+                return Env.JUPYTER_NOTEBOOK
+            else:
+                return Env.SAGEMAKER
+        except AttributeError:
+            try:
+                import js
+                return Env.JUPYTERLITE
+            except ImportError:
+                return Env.HYPHA
+
+
+ENVIRONMENT = find_env()

docs/jupyterlite/pypi/dask_image-2022.9.0-py2.py3-none-any.whl

@@ -11,6 +11,7 @@
 from ._initialization_params import init_params_dict, init_key_aliases
 from ._method_types import deferred_methods
 from .integrations import _detect_render_type, _get_viewer_image, _get_viewer_point_sets
+from .integrations.environment import ENVIRONMENT, Env
 from .render_types import RenderType
 from .viewer_config import ITK_VIEWER_SRC, PYDATA_SPHINX_HREF, MUI_HREF
 
@@ -35,8 +36,9 @@ def __init__(
         self.init_data = {}
         self.img = display(HTML(f'<div />'), display_id=str(uuid.uuid4()))
         self.wid = None
-        self.viewer_event = threading.Event()
-        self.data_event = threading.Event()
+        if ENVIRONMENT is not Env.JUPYTERLITE:
+            self.viewer_event = threading.Event()
+            self.data_event = threading.Event()
 
     def _get_input_data(self):
         input_options = ["data", "image", "label_image", "point_sets"]
@@ -98,11 +100,12 @@ async def run(self, ctx):
             config=config,
         )
         _viewer_count += 1
-        itk_viewer.registerEventListener(
-            'renderedImageAssigned', self.set_event
-        )
-        # Once the viewer has been created any queued requests can be run
-        asyncio.get_running_loop().call_soon_threadsafe(self.viewer_event.set)
+        if ENVIRONMENT is not Env.JUPYTERLITE:
+            itk_viewer.registerEventListener(
+                'renderedImageAssigned', self.set_event
+            )
+            # Once the viewer has been created any queued requests can be run
+            asyncio.get_running_loop().call_soon_threadsafe(self.viewer_event.set)
 
         self.set_default_ui_values(itk_viewer)
         self.itk_viewer = itk_viewer
@@ -155,19 +158,24 @@ def __init__(
         self, ui_collapsed=True, rotate=False, ui="pydata-sphinx", **add_data_kwargs
     ):
         """Create a viewer."""
-        self.bg_jobs = bg.BackgroundJobManager()
         self.viewer_rpc = ViewerRPC(
             ui_collapsed=ui_collapsed, rotate=rotate, ui=ui, **add_data_kwargs
         )
-        self.queue = queue.Queue()
-        self.deferred_queue = queue.Queue()
-        self.bg_thread = self.bg_jobs.new(self.queue_worker)
+        if ENVIRONMENT is not Env.JUPYTERLITE:
+            self.bg_jobs = bg.BackgroundJobManager()
+            self.queue = queue.Queue()
+            self.deferred_queue = queue.Queue()
+            self.bg_thread = self.bg_jobs.new(self.queue_worker)
         api.export(self.viewer_rpc)
 
     @property
     def loop(self):
         return asyncio.get_running_loop()
 
+    @property
+    def has_viewer(self):
+        return hasattr(self.viewer_rpc, 'itk_viewer')
+
     async def run_queued_requests(self):
         def _run_queued_requests(queue):
             method_name, args, kwargs = queue.get()
@@ -190,7 +198,7 @@ def queue_worker(self):
         loop.run_until_complete(task)
 
     def queue_request(self, method, *args, **kwargs):
-        if hasattr(self.viewer_rpc, 'itk_viewer'):
+        if ENVIRONMENT is Env.JUPYTERLITE or self.has_viewer:
             fn = getattr(self.viewer_rpc.itk_viewer, method)
             fn(*args, **kwargs)
         elif method in deferred_methods():