From a0438d434a8dde45e6fdab38d44181f0cd0cb2c6 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Tue, 29 Oct 2019 19:30:50 +0100
Subject: Rename target module to memory

---
 boltzgen/kernel/memory/__init__.py        |  3 +++
 boltzgen/kernel/memory/index.py           | 18 ++++++++++++++++++
 boltzgen/kernel/memory/layout/AOS.py      | 14 ++++++++++++++
 boltzgen/kernel/memory/layout/SOA.py      | 14 ++++++++++++++
 boltzgen/kernel/memory/layout/__init__.py |  2 ++
 boltzgen/kernel/memory/precision.py       | 16 ++++++++++++++++
 6 files changed, 67 insertions(+)
 create mode 100644 boltzgen/kernel/memory/__init__.py
 create mode 100644 boltzgen/kernel/memory/index.py
 create mode 100644 boltzgen/kernel/memory/layout/AOS.py
 create mode 100644 boltzgen/kernel/memory/layout/SOA.py
 create mode 100644 boltzgen/kernel/memory/layout/__init__.py
 create mode 100644 boltzgen/kernel/memory/precision.py

(limited to 'boltzgen/kernel/memory')

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
-- 
cgit v1.2.3