1 files changed, 11 insertions, 0 deletions
diff --git a/model.py b/model.py
index 5e8a2ce..5f20075 100644
--- a/model.py
+++ b/model.py
@@ -350,11 +350,22 @@ class PrimedGenerator(Runner):
                 x[:, i] = int((math.sin(i/44100 * primer_freq) + 1) / 2 * (q_width) + q_min)
             return x
 
+        def _saw(x):
+            primer_width = int(n_samples / 2)
+            for i in range(n_samples):
+                # x[:, i] = #int((math.sin(i/44100 * primer_freq) + 1) / 2 * (q_width) + q_min)
+                if (i % primer_width) == 0:
+                    x[:, i] = q_levels
+                else:
+                    x[:, i] = int(q_levels / 2)
+            return x
+
         sequence_lookup = {
           'zero': lambda x: x.fill_(utils.q_zero(self.model.q_levels)),
           'noise': _noise,
           'sin': _sin,
           'saw': _saw,
+          'spike': _spike,
         }
 
         sequences = sequence_lookup.get(primer, 'zero')(sequences)