From 1ab5bf2e6cd6f78916fa95586ab9b16d35eb84bb Mon Sep 17 00:00:00 2001
From: "Christopher J. White" <c.white.astro@gmail.com>
Date: Sun, 22 May 2022 16:25:14 -0400
Subject: [PATCH] made domain resizing occur less frequently

---
 src/simulation_reader/simulation_reader.cpp | 56 +++++++++++++--------
 src/simulation_reader/simulation_reader.hpp |  1 +
 2 files changed, 37 insertions(+), 20 deletions(-)

diff --git a/src/simulation_reader/simulation_reader.cpp b/src/simulation_reader/simulation_reader.cpp
index f096c83..a31eb87 100644
--- a/src/simulation_reader/simulation_reader.cpp
+++ b/src/simulation_reader/simulation_reader.cpp
@@ -1,6 +1,7 @@
 // Blacklight simulation reader
 
 // C++ headers
+#include <cmath>     // abs
 #include <cstdio>    // snprintf
 #include <fstream>   // ifstream
 #include <ios>       // ios_base
@@ -426,29 +427,44 @@ double SimulationReader::Read(int snapshot)
         data_stream.seekg(1, std::ios_base::cur);
         ConvertCoordinates();
       }
-      if (simulation_coord == Coordinates::sks and x2f.n2 == 1 and
-          (x2f(0,0) != 0.0 or x2f(0,x2f.n1-1) != Math::pi))
+    }
+
+    // Check coordinates
+    if (first_time)
+    {
+      if (simulation_coord == Coordinates::sks and x2f.n2 == 1)
       {
-        std::ostringstream message;
-        message.setf(std::ios_base::scientific);
-        message.precision(16);
-        message << "Changing theta range from [" << x2f(0,0) << ", " << x2f(0,x2f.n1-1);
-        message << "] to [0, pi].";
-        BlacklightWarning(message.str().c_str());
-        x2f(0,0) = 0.0;
-        x2f(0,x2f.n1-1) = Math::pi;
+        bool error_low = std::abs(x2f(0,0)) > (x2f(0,1) - x2f(0,0)) * angular_domain_tolerance;
+        bool error_high = std::abs(x2f(0,x2f.n1-1) - Math::pi)
+            > (x2f(0,x2f.n1-1) - x2f(0,x2f.n1-2)) * angular_domain_tolerance;
+        if (error_low or error_high)
+        {
+          std::ostringstream message;
+          message.setf(std::ios_base::scientific);
+          message.precision(16);
+          message << "Changing theta range from [" << x2f(0,0) << ", " << x2f(0,x2f.n1-1);
+          message << "] to [0, pi].";
+          BlacklightWarning(message.str().c_str());
+          x2f(0,0) = 0.0;
+          x2f(0,x2f.n1-1) = Math::pi;
+        }
       }
-      if (simulation_coord == Coordinates::sks and x3f.n2 == 1 and
-          (x3f(0,0) != 0.0 or x3f(0,x3f.n1-1) != 2.0 * Math::pi))
+      if (simulation_coord == Coordinates::sks and x3f.n2 == 1)
       {
-        std::ostringstream message;
-        message.setf(std::ios_base::scientific);
-        message.precision(16);
-        message << "Changing phi range from [" << x3f(0,0) << ", " << x3f(0,x3f.n1-1);
-        message << "] to [0, 2*pi].";
-        BlacklightWarning(message.str().c_str());
-        x3f(0,0) = 0.0;
-        x3f(0,x3f.n1-1) = 2.0 * Math::pi;
+        bool error_low = std::abs(x3f(0,0)) > (x3f(0,1) - x3f(0,0)) * angular_domain_tolerance;
+        bool error_high = std::abs(x3f(0,x3f.n1-1) - 2.0 * Math::pi)
+            > (x3f(0,x3f.n1-1) - x3f(0,x3f.n1-2)) * angular_domain_tolerance;
+        if (error_low or error_high)
+        {
+          std::ostringstream message;
+          message.setf(std::ios_base::scientific);
+          message.precision(16);
+          message << "Changing phi range from [" << x3f(0,0) << ", " << x3f(0,x3f.n1-1);
+          message << "] to [0, 2*pi].";
+          BlacklightWarning(message.str().c_str());
+          x3f(0,0) = 0.0;
+          x3f(0,x3f.n1-1) = 2.0 * Math::pi;
+        }
       }
     }
 
diff --git a/src/simulation_reader/simulation_reader.hpp b/src/simulation_reader/simulation_reader.hpp
index c5c8381..a2cfbe2 100644
--- a/src/simulation_reader/simulation_reader.hpp
+++ b/src/simulation_reader/simulation_reader.hpp
@@ -74,6 +74,7 @@ struct SimulationReader
   int num_arrays;
   int latest_file_number;
   const double extrapolation_tolerance = 1.0;
+  const double angular_domain_tolerance = 0.1;
 
   // Data
   int n_3_root;