Added threshold for zeroing or oneing

1 files changed, 4 insertions, 4 deletions

diff --git a/log-fourier_integrator.cpp b/log-fourier_integrator.cpp
index 6984e96..2359c80 100644
--- a/log-fourier_integrator.cpp
+++ b/log-fourier_integrator.cpp
@@ -150,11 +150,11 @@ int main(int argc, char* argv[]) {
       bool trigger0 = false;
       bool trigger1 = false;
       for (unsigned i = 0; i < N; i++) {
-        if (Cₜ₊₁[i] < 0 || trigger0) {
+        if (Cₜ₊₁[i] < ε || trigger0) {
           Cₜ₊₁[i] = 0;
           trigger0 = true;
         }
-        if (Cₜ₊₁[N-1-i] > 1 || trigger1) {
+        if (Cₜ₊₁[N-1-i] > 1 - ε || trigger1) {
           Cₜ₊₁[N-1-i] = 1;
           trigger1 = true;
         }
@@ -162,11 +162,11 @@ int main(int argc, char* argv[]) {
       trigger0 = false;
       trigger1 = false;
       for (unsigned i = 0; i < N; i++) {
-        if (Rₜ₊₁[i] < 0 || trigger0) {
+        if (Rₜ₊₁[i] < ε || trigger0) {
           Rₜ₊₁[i] = 0;
           trigger0 = true;
         }
-        if (Rₜ₊₁[N-1-i] > 1 || trigger1) {
+        if (Rₜ₊₁[N-1-i] > 1 - ε || trigger1) {
           Rₜ₊₁[N-1-i] = 1;
           trigger1 = true;
         }