From 8e8beae3156f38be507bf44837add1c78003688b Mon Sep 17 00:00:00 2001
From: Simon Humpohl <simon.humpohl@rwth-aachen.de>
Date: Tue, 22 Feb 2022 16:56:20 +0100
Subject: [PATCH] WIP of subroutine insertion

---
 qupulse/_program/seqc.py | 68 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/qupulse/_program/seqc.py b/qupulse/_program/seqc.py
index 52d6042a..23106852 100644
--- a/qupulse/_program/seqc.py
+++ b/qupulse/_program/seqc.py
@@ -1115,6 +1115,10 @@ def stepping_hash(self) -> int:
     def same_stepping(self, other: 'SEQCNode'):
         pass
 
+    @abc.abstractmethod
+    def same_playback(self, other: 'SEQCNode') -> bool:
+        pass
+
     @abc.abstractmethod
     def iter_waveform_playbacks(self) -> Iterator['WaveformPlayback']:
         pass
@@ -1186,6 +1190,11 @@ def same_stepping(self, other: 'Scope'):
                 len(self.nodes) == len(other.nodes) and
                 all(n1.same_stepping(n2) for n1, n2 in zip(self.nodes, other.nodes)))
 
+    def same_playback(self, other: 'SEQCNode') -> bool:
+        return (type(other) is Scope and
+                len(self.nodes) == len(other.nodes) and
+                all(n1.same_playback(n2) for n1, n2 in zip(self.nodes, other.nodes)))
+
     def _visit_nodes(self, waveform_manager: ProgramWaveformManager):
         for node in self.nodes:
             node._visit_nodes(waveform_manager)
@@ -1241,6 +1250,11 @@ def same_stepping(self, other: 'Repeat'):
                 self.repetition_count == other.repetition_count and
                 self.scope.same_stepping(other.scope))
 
+    def same_playback(self, other: 'Repeat'):
+        return (type(self) == type(other) and
+                self.repetition_count == other.repetition_count and
+                self.scope.same_playback(other.scope))
+
     def stepping_hash(self) -> int:
         return hash((type(self), self.repetition_count, self.scope.stepping_hash()))
 
@@ -1351,6 +1365,11 @@ def same_stepping(self, other: 'SteppingRepeat'):
                 len(self.node_cluster) == len(other.node_cluster) and
                 self.node_cluster[0].same_stepping(other.node_cluster[0]))
 
+    def same_playback(self, other: 'SEQCNode') -> bool:
+        return (type(other) is SteppingRepeat and
+                len(self.node_cluster) == len(other.node_cluster) and
+                all(n1.same_playback(n2) for n1, n2 in zip(self.node_cluster, other.node_cluster)))
+
     def _visit_nodes(self, waveform_manager: ProgramWaveformManager):
         for node in self.node_cluster:
             node._visit_nodes(waveform_manager)
@@ -1421,6 +1440,10 @@ def same_stepping(self, other: 'WaveformPlayback') -> bool:
         else:
             return same_type and self.samples() == other.samples()
 
+    def same_playback(self, other: 'SEQCNode') -> bool:
+        return (type(other) is WaveformPlayback
+                and self.waveform == other.waveform)
+
     def iter_waveform_playbacks(self) -> Iterator['WaveformPlayback']:
         yield self
 
@@ -1446,3 +1469,48 @@ def to_source_code(self, waveform_manager: ProgramWaveformManager,
             else:
                 advance_cmd = self.ADVANCE_DISABLED_COMMENT
             yield play_cmd + advance_cmd
+
+
+class HackWF:
+    def __init__(self):
+        self.memory = 0
+
+    def request_concatenated(self, wf: Tuple[BinaryWaveform, ...]):
+        self.memory += wf[0].data.size
+
+
+def detect_subroutines(program: SEQCNode, min_wf_memory: int):
+    candidates: List[List[SEQCNode]] = []
+    stack: List[SEQCNode] = [program]
+    while stack:
+        current = stack.pop()
+        m = HackWF()
+        current._visit_nodes(m)
+        if m.memory < min_wf_memory:
+            continue
+
+        for cs in candidates:
+            if cs[0].same_playback(current):
+                cs.append(current)
+            else:
+                cs.append([current])
+                if isinstance(current, Scope):
+                    stack.extend(current.nodes)
+                elif isinstance(current, Repeat):
+                    stack.append(current.scope)
+                elif isinstance(current, SteppingRepeat):
+                    stack.extend(current.node_cluster)
+                else:
+                    assert isinstance(stack, WaveformPlayback)
+
+    candidates = [cs
+                  for cs in candidates
+                  if len(cs) > 1]
+
+    
+
+
+
+
+
+