6 files changed, 67 insertions, 0 deletions
diff --git a/boltzgen/kernel/memory/__init__.py b/boltzgen/kernel/memory/__init__.py
new file mode 100644
index 0000000..5927d1b
--- /dev/null
+++ b/boltzgen/kernel/memory/__init__.py
@@ -0,0 +1,3 @@
+from . import layout
+from . import precision
+from . import index
diff --git a/boltzgen/kernel/memory/index.py b/boltzgen/kernel/memory/index.py
new file mode 100644
index 0000000..e89e9ee
--- /dev/null
+++ b/boltzgen/kernel/memory/index.py
@@ -0,0 +1,18 @@
+class CellIndex:
+    def __init__(self, geometry):
+        self.geometry   = geometry
+
+    def neighbor(self, c_i):
+        return self.gid(*[ c for _, c in enumerate(c_i) ])
+
+class XYZ(CellIndex):
+    pass
+
+    def gid(self, x, y, z=0):
+        return x*self.geometry.size_y*self.geometry.size_z + y*self.geometry.size_z + z
+
+class ZYX(CellIndex):
+    pass
+
+    def gid(self, x, y, z=0):
+        return z*self.geometry.size_x*self.geometry.size_y + y*self.geometry.size_x + x
diff --git a/boltzgen/kernel/memory/layout/AOS.py b/boltzgen/kernel/memory/layout/AOS.py
new file mode 100644
index 0000000..88e5377
--- /dev/null
+++ b/boltzgen/kernel/memory/layout/AOS.py
@@ -0,0 +1,14 @@
+class AOS:
+    def __init__(self, descriptor, index, geometry):
+        self.descriptor = descriptor
+        self.index    = index(geometry)
+        self.geometry = geometry
+
+    def cell_preshift(self, gid):
+        return "(%s)*%d" % (gid, self.descriptor.q)
+
+    def pop_offset(self, i):
+        return i
+
+    def neighbor_offset(self, c_i):
+        return self.descriptor.q * self.index.neighbor(c_i)
diff --git a/boltzgen/kernel/memory/layout/SOA.py b/boltzgen/kernel/memory/layout/SOA.py
new file mode 100644
index 0000000..05e9e84
--- /dev/null
+++ b/boltzgen/kernel/memory/layout/SOA.py
@@ -0,0 +1,14 @@
+class SOA:
+    def __init__(self, descriptor, index, geometry):
+        self.descriptor = descriptor
+        self.index    = index(geometry)
+        self.geometry = geometry
+
+    def cell_preshift(self, gid):
+        return gid
+
+    def pop_offset(self, i):
+        return i * self.geometry.volume
+
+    def neighbor_offset(self, c_i):
+        return self.index.neighbor(c_i)
diff --git a/boltzgen/kernel/memory/layout/__init__.py b/boltzgen/kernel/memory/layout/__init__.py
new file mode 100644
index 0000000..731fbe8
--- /dev/null
+++ b/boltzgen/kernel/memory/layout/__init__.py
@@ -0,0 +1,2 @@
+from .AOS import AOS
+from .SOA import SOA
diff --git a/boltzgen/kernel/memory/precision.py b/boltzgen/kernel/memory/precision.py
new file mode 100644
index 0000000..d22e922
--- /dev/null
+++ b/boltzgen/kernel/memory/precision.py
@@ -0,0 +1,16 @@
+class common:
+    @classmethod
+    def get_float_type(self, precision):
+        if precision not in ['single', 'double']:
+            raise Exception("Precision must be either 'single' or 'double'")
+
+        return {
+            'single': 'float',
+            'double': 'double'
+        }.get(precision)
+
+class cpp(common):
+    pass
+
+class cl(common):
+    pass