WIP: Graph datahandling

This is the draft for a data handling that (optionally) forwards all calls to SerialDatahandling. All calls and data transfers get recorded for the creation of an execution graph.

Needs to be changed after the breaking changes in datahandling.

Needs a tiny change in lbmpy:

Instead of using TimeLoop(...) for time loop creation a custom function is used.

pystencils/datahandling/graph_datahandling.py

@@ -42,6 +42,9 @@ class DataTransfer:
         self.field = field
         self.kind = kind
 
+    def __str__(self):
+        return f'DataTransferKind: {self.kind} with {self.field}'
+
 
 class Swap(DataTransfer):
     def __init__(self, source, destination, gpu):
@@ -49,6 +52,9 @@ class Swap(DataTransfer):
         self.field = source
         self.destination = destination
 
+    def __str__(self):
+        return f'Swap: {self.field} with {self.destination}'
+
 
 class Communication(DataTransfer):
     def __init__(self, field, stencil, gpu):
@@ -80,6 +86,10 @@ class TimeloopRun:
                 + '\nPost:\n'
                 + '\n   '.join(str(f) for f in self.timeloop._post_run_functions))
 
+    @property
+    def asts(self):
+        return self.timeloop._single_step_asts
+
 
 class GraphDataHandling(pystencils.datahandling.SerialDataHandling):