From 3c08f1240527684c2d01d854d150a195bfb1349b Mon Sep 17 00:00:00 2001
From: nyanpasu64 <nyanpasu64@tuta.io>
Date: Sun, 29 Jul 2018 03:12:33 -0700
Subject: [PATCH] Edit CorrelationTrigger to window input, more when
 subsampling

---
 ovgenpy/triggers.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/ovgenpy/triggers.py b/ovgenpy/triggers.py
index 3ccd598..674fa86 100644
--- a/ovgenpy/triggers.py
+++ b/ovgenpy/triggers.py
@@ -97,16 +97,18 @@ class CorrelationTrigger(Trigger):
         use_edge_trigger = self.cfg.use_edge_trigger
 
         N = self._buffer_nsamp
+        halfN = N // 2
+
+        # data = windowed
         data = self._wave.get_around(index, N, self._subsampling)
+        data *= signal.gaussian(N, std = halfN / np.sqrt(self._subsampling))
 
         # prev_buffer = windowed step function + self._buffer
-        halfN = N // 2
         step = np.empty(N, dtype=FLOAT)     # type: np.ndarray[FLOAT]
         step[:halfN] = -trigger_strength / 2
         step[halfN:] = trigger_strength / 2
 
-        window = signal.gaussian(N, std = halfN // 3)
-        step *= window
+        step *= signal.gaussian(N, std = halfN / 3)
 
         prev_buffer = self._buffer + step