mfkrogh/docker launcher enshell

requirements/build.txt

@@ -4,3 +4,4 @@ requests>=2.20.1
 twine>=4.0.1
 pyansys-docker>=5.0.4
 junitparser>=2.8.0
+urllib3>=1.26.10

requirements/test.txt

@@ -6,3 +6,5 @@ pyansys-docker>=5.0.4
 pytest-mock>=3.10.0
 numpy>=1.13.0
 junitparser>=2.8.0
+requests<2.29.0
+urllib3>=1.26.10

setup.py

@@ -6,6 +6,7 @@
     "protobuf>=3.9.1",
     "grpcio>=1.23.0",
     "requests>=2.20.1",
+    "urllib3>=1.26.10",
     "dill>=0.3.5.1",
 ]
 

src/ansys/pyensight/__init__.py

@@ -15,3 +15,10 @@
     from ansys.pyensight.dockerlauncher import DockerLauncher
 except Exception:
     pass
+
+try:
+    from ansys.pyensight.dockerlauncherenshell import DockerLauncherEnShell
+except Exception:
+    pass
+
+from ansys.pyensight.launch_ensight import launch_ensight

src/ansys/pyensight/dockerlauncher.py

@@ -10,7 +10,7 @@
         launcher = DockerLauncher(data_directory="D:\\data")
         launcher.pull()
         session = launcher.start()
-        launcher.stop()
+        session.close()
 
 """
 import os.path
@@ -39,6 +39,11 @@ class DockerLauncher(pyensight.Launcher):
             In some cases where the EnSight session can take a significant amount of
             timme to start up, this is the number of seconds to wait before failing
             the connection.  The default is 120.0.
+        use_egl:
+            If True, EGL hardware accelerated graphics will be used. The platform
+            must be able to support it.
+        use_sos:
+            If None, don't use SOS. Otherwise, it's the number of EnSight Servers to use (int).
 
     Examples:
         ::
@@ -47,20 +52,21 @@ class DockerLauncher(pyensight.Launcher):
             launcher = DockerLauncher(data_directory="D:\\data")
             launcher.pull()
             session = launcher.start()
-            launcher.stop()
+            session.close()
 
     """
 
     def __init__(
         self,
         data_directory: str,
         docker_image_name: Optional[str] = None,
-        use_dev: bool = False,
-        timeout: float = 120.0,
+        use_dev: Optional[bool] = False,
+        timeout: Optional[float] = 120.0,
+        use_egl: Optional[bool] = False,
+        use_sos: Optional[int] = None,
     ) -> None:
-        super().__init__()
+        super().__init__(timeout=timeout, use_egl=use_egl, use_sos=use_sos)
 
-        self._timeout = timeout
         self._data_directory: str = data_directory
         self._container = None
 
@@ -118,16 +124,14 @@ def pull(self) -> None:
         except Exception:
             raise RuntimeError(f"Can't pull Docker image: {self._image_name}")
 
-    def start(self, host: str = "127.0.0.1", use_egl: bool = False) -> "pyensight.Session":
+    def start(self, host: str = "127.0.0.1") -> "pyensight.Session":
         """Start an EnSight session using the local Docker ensight image
         Launch a copy of EnSight in the container that supports the gRPC interface.  Create and
         bind a Session instance to the created gRPC session.  Return that session.
 
         Args:
             host:
                 Optional hostname on which the EnSight gRPC service is running
-            use_egl:
-                Specify True if EnSight should try to use EGL.  Beta flag.
 
         Returns:
             pyensight Session object instance
@@ -180,7 +184,7 @@ def start(self, host: str = "127.0.0.1", use_egl: bool = False) -> "pyensight.Se
         # FIXME_MFK: probably need a unique name for our container
         # in case the user launches multiple sessions
         egl_env = os.environ.get("PYENSIGHT_FORCE_ENSIGHT_EGL")
-        use_egl = use_egl or egl_env or self._has_egl()
+        use_egl = self._use_egl or egl_env or self._has_egl()
         if use_egl:
             self._container = self._docker_client.containers.run(
                 self._image_name,
@@ -250,6 +254,9 @@ def start(self, host: str = "127.0.0.1", use_egl: bool = False) -> "pyensight.Se
         if use_egl:
             cmd2 += " -egl"
 
+        if self._use_sos:
+            cmd2 += " -sos -nservers " + str(int(self._use_sos))
+
         cmd2 += " -grpc_server " + str(ports[0])
 
         vnc_url = f"vnc://%%3Frfb_port={ports[1]}%%26use_auth=0"