delay TIGRE imports

Author: casperdcl

Wrappers/Python/cil/plugins/tigre/FBP.py

@@ -24,12 +24,6 @@
 from cil.framework.labels import AcquisitionDimension, ImageDimension
 from cil.plugins.tigre import CIL2TIGREGeometry
 
-try:
-    from tigre.algorithms import fdk, fbp
-except ModuleNotFoundError:
-    raise ModuleNotFoundError("This plugin requires the additional package TIGRE\n" +
-            "Please install it via conda as tigre from the ccpi channel")
-
 class FBP(DataProcessor):
 
     '''FBP Filtered Back Projection is a reconstructor for 2D and 3D parallel and cone-beam geometries.
@@ -82,7 +76,7 @@ def check_input(self, dataset):
 
         AcquisitionDimension.check_order_for_engine('tigre', dataset.geometry)
         return True
-    
+
     def _set_up(self):
         """
         Configure processor attributes that require the data to setup
@@ -91,7 +85,12 @@ def _set_up(self):
         self._shape_out = self.image_geometry.shape
 
     def process(self, out=None):
-
+        try:
+            from tigre.algorithms import fdk, fbp
+        except ModuleNotFoundError:
+            raise ModuleNotFoundError(
+                "This plugin requires the additional package TIGRE\n"
+                "Please install it via conda as tigre from the ccpi channel")
         if self.tigre_geom.is2D:
             data_temp = np.expand_dims(self.get_input().as_array(), axis=1)
 

Wrappers/Python/cil/plugins/tigre/Geometry.py

@@ -22,8 +22,7 @@
 try:
     from tigre.utilities.geometry import Geometry
 except ModuleNotFoundError:
-    raise ModuleNotFoundError("This plugin requires the additional package TIGRE\n" +
-            "Please install it via conda as tigre from the ccpi channel")
+    Geometry = object
 
 class CIL2TIGREGeometry(object):
     @staticmethod
@@ -50,10 +49,12 @@ def getTIGREGeometry(ig, ag):
         return tg, angles
 
 class TIGREGeometry(Geometry):
-
     def __init__(self, ig, ag):
-
-        Geometry.__init__(self)
+        if Geometry is object:
+            raise ModuleNotFoundError(
+                "This plugin requires the additional package TIGRE\n"
+                "Please install it via conda as tigre from the ccpi channel")
+        super().__init__()
 
         if ag.geom_type not in ['cone', 'parallel']:
             raise ValueError(f"CIL cannot use TIGRE to process geometries of type {ag.geom_type}.")

Wrappers/Python/cil/plugins/tigre/ProjectionOperator.py

@@ -26,15 +26,6 @@
 
 log = logging.getLogger(__name__)
 
-try:
-    from _Atb import _Atb_ext as Atb
-    from _Ax import _Ax_ext as Ax
-
-except ModuleNotFoundError:
-    raise ModuleNotFoundError(
-        "This plugin requires the additional package TIGRE\n" +
-        "Please install it via conda as tigre from the ccpi channel")
-
 try:
     from tigre.utilities.gpu import GpuIds
     has_gpu_sel = True
@@ -176,6 +167,13 @@ def __init__(self,
             self.gpuids = GpuIds()
 
     def __call_Ax(self, data):
+        try:
+            from _Ax import _Ax_ext as Ax
+        except ModuleNotFoundError:
+            raise ModuleNotFoundError(
+                "This plugin requires the additional package TIGRE\n"
+                "Please install it via conda as tigre from the ccpi channel")
+
         if has_gpu_sel:
             return Ax(data, self.tigre_geom, self.tigre_geom.angles,
                       self.method['direct'], self.tigre_geom.mode, self.gpuids)
@@ -207,6 +205,13 @@ def direct(self, x, out=None):
         return out
 
     def __call_Atb(self, data):
+        try:
+            from _Atb import _Atb_ext as Atb
+        except ModuleNotFoundError:
+            raise ModuleNotFoundError(
+                "This plugin requires the additional package TIGRE\n"
+                "Please install it via conda as tigre from the ccpi channel")
+
         if has_gpu_sel:
             return Atb(data, self.tigre_geom, self.tigre_geom.angles,
                        self.method['adjoint'], self.tigre_geom.mode,