compiler: Started ghosted subdomain implementation

Author: ZoeLeibowitz

devito/operator/operator.py

@@ -42,6 +42,7 @@
 from devito.types.dimension import Thickness
 from devito.petsc.iet.passes import lower_petsc
 from devito.petsc.clusters import petsc_preprocess
+from devito.petsc.equations import lower_exprs_petsc
 
 __all__ = ['Operator']
 
@@ -368,6 +369,9 @@ def _lower_exprs(cls, expressions, **kwargs):
         # in particular uniqueness across expressions is ensured
         expressions = concretize_subdims(expressions, **kwargs)
 
+        # rename etc
+        expressions = lower_exprs_petsc(expressions, **kwargs)
+
         processed = [LoweredEq(i) for i in expressions]
 
         return processed

devito/types/dimension.py

@@ -21,7 +21,7 @@
            'CustomDimension', 'SteppingDimension', 'SubDimension',
            'MultiSubDimension', 'ConditionalDimension', 'ModuloDimension',
            'IncrDimension', 'BlockDimension', 'StencilDimension',
-           'VirtualDimension', 'Spacing', 'dimensions']
+           'VirtualDimension', 'Spacing', 'dimensions', 'CustomBoundSubDimension']
 
 
 SubDimensionThickness = namedtuple('SubDimensionThickness', 'left right')
@@ -822,6 +822,39 @@ def __init_finalize__(self, name, parent, thickness, functions=None,
     @cached_property
     def bound_symbols(self):
         return self.parent.bound_symbols
+    
+
+class CustomBoundSubDimension(SubDimension):
+
+    # have is_CustomSub = True ... here?
+
+    __rargs__ = SubDimension.__rargs__ + ('custom_left', 'custom_right')
+
+    def __init_finalize__(self, name, parent, thickness, local,
+                          custom_left=0, custom_right=0, **kwargs):
+        self._custom_left = custom_left
+        self._custom_right = custom_right
+        super().__init_finalize__(name, parent, thickness, local)
+
+    @property
+    def custom_left(self):
+        return self._custom_left
+
+    @property
+    def custom_right(self):
+        return self._custom_right
+
+    # @cached_property
+    # def _interval(self):
+    #     left = self.parent.symbolic_min + self._offset_left
+    #     right = self.parent.symbolic_max - self._offset_right
+    #     return sympy.Interval(left, right)
+
+    @cached_property
+    def _interval(self):
+        left = self.custom_left
+        right = self.custom_right
+        return sympy.Interval(left, right)
 
 
 class SubsamplingFactor(Scalar):