From a992e9e1f6554dc1674bad058e493016bd57a077 Mon Sep 17 00:00:00 2001
From: Victor Leite <leitevmd@gmail.com>
Date: Mon, 22 Mar 2021 11:14:37 -0300
Subject: [PATCH] dsl: testing

---
 devito/ir/equations/equation.py |  3 +-
 devito/ir/support/utils.py      | 61 ++++++++++++++++++++++++++++++++-
 2 files changed, 62 insertions(+), 2 deletions(-)

diff --git a/devito/ir/equations/equation.py b/devito/ir/equations/equation.py
index 3c13bc85531..0fec469f70f 100644
--- a/devito/ir/equations/equation.py
+++ b/devito/ir/equations/equation.py
@@ -4,7 +4,7 @@
 from devito.ir.equations.algorithms import dimension_sort, lower_exprs
 from devito.finite_differences.differentiable import diff2sympy
 from devito.ir.support import (IterationSpace, DataSpace, Interval, IntervalGroup,
-                               Stencil, detect_accesses, detect_oobs, detect_io,
+                               Stencil, detect_accesses, detect_oobs, detect_oobs_test, detect_io,
                                build_intervals, build_iterators)
 from devito.symbolics import CondEq, IntDiv, uxreplace
 from devito.tools import Pickable, frozendict
@@ -138,6 +138,7 @@ def __new__(cls, *args, **kwargs):
         # Analyze the expression
         mapper = detect_accesses(expr)
         oobs = detect_oobs(mapper)
+        detect_oobs_test(mapper)
         conditional_dimensions = [i for i in ordering if i.is_Conditional]
 
         # Construct Intervals for IterationSpace and DataSpace
diff --git a/devito/ir/support/utils.py b/devito/ir/support/utils.py
index c6588268fbb..aaa2e5361fd 100644
--- a/devito/ir/support/utils.py
+++ b/devito/ir/support/utils.py
@@ -6,7 +6,7 @@
 from devito.tools import as_tuple, flatten, filter_sorted
 from devito.types import Dimension
 
-__all__ = ['detect_accesses', 'detect_oobs', 'build_iterators', 'build_intervals',
+__all__ = ['detect_accesses', 'detect_oobs', 'detect_oobs_test', 'build_iterators', 'build_intervals',
            'detect_io']
 
 
@@ -69,6 +69,65 @@ def detect_oobs(mapper):
     return found | set().union(*[i._defines for i in found if i.is_Derived])
 
 
+def detect_oobs_test(mapper):
+    found = set()
+    for f, stencil in mapper.items():
+        # Excluding f.is_Array
+        if f is None or not f.is_DiscreteFunction:
+            continue
+        
+        for d, v in stencil.items():
+
+            # Ignore time dimensions
+            if not d.is_Space:
+                continue
+
+            # This should ignore subdimentions with thickness != 0
+            if d.is_Sub:
+                if not all(tick[1] == 0 for tick in d.thickness):
+                    continue
+
+            # Check only parent dimension limits only
+            p = d.parent if d.is_Sub else d
+
+            try:
+                Min = 0
+                Max = f._size_nodomain[p].left + f._size_halo[p].right
+                test0 = min(v) < 0
+                test1 = max(v) > Max
+                if test0 or test1:
+                    if p.is_NonlinearDerived:
+                        pass
+                    else:
+                        raise ValueError(' OOB access\n'
+                                        ' func: %s, type: %s\n'
+                                        ' indices: %s\n'
+                                        ' indices_ref: %s\n'
+                                        ' dim: %s, type: %s\n'
+                                        ' is_Time: %s \n'
+                                        #  ' (time_order: %s)\n'
+                                        ' is_Space: %s \n'
+                                        #  ' (space_order: %s)\n'
+                                        ' d.is_Sub %s (%s)\n'
+                                        ' stencil min max = (%s, %s)\n'
+                                        ' limits = (%s, %s)'
+                                        % (f,type(f),
+                                            f.indices,
+                                            f.indices_ref,
+                                            p.name,type(p),
+                                            p.is_Time,
+                                            # f.time_order if p.is_Time else '',
+                                            p.is_Space,
+                                            # f.space_order if p.is_Space else '',
+                                            d.is_Sub, d,
+                                            min(v), max(v),
+                                            Min, Max))
+
+            except KeyError:
+                pass
+    return found
+
+
 def build_iterators(mapper):
     """
     Given M as produced by :func:`detect_accesses`, return a mapper ``M' : D -> V``,