Added option to shift time even more

1 files changed, 8 insertions, 2 deletions

diff --git a/log_get_energy.cpp b/log_get_energy.cpp
index 6a7be56..549bd21 100644
--- a/log_get_energy.cpp
+++ b/log_get_energy.cpp
@@ -18,6 +18,7 @@ int main(int argc, char* argv[]) {
   Real k = 0.1;
   unsigned pad = 2;
   Real logShift = 0;
+  bool shiftSquare = false;
 
   /* Iteration parameters */
   Real β₀ = 0;
@@ -26,7 +27,7 @@ int main(int argc, char* argv[]) {
 
   int opt;
 
-  while ((opt = getopt(argc, argv, "p:s:2:T:t:b:d:k:h:D:0:")) != -1) {
+  while ((opt = getopt(argc, argv, "p:s:2:T:t:b:d:k:h:D:0:S")) != -1) {
     switch (opt) {
     case 'p':
       p = atoi(optarg);
@@ -58,6 +59,9 @@ int main(int argc, char* argv[]) {
     case '0':
       β₀ = atof(optarg);
       break;
+    case 'S':
+      shiftSquare = true;
+      break;
     default:
       exit(1);
     }
@@ -67,7 +71,9 @@ int main(int argc, char* argv[]) {
   Real Γ₀ = 1;
   Real μ₀ = τ₀ > 0 ? (sqrt(1+4*Γ₀*τ₀)-1)/(2*τ₀) : Γ₀;
 
-  LogarithmicFourierTransform fft(N, k, Δτ, pad, μ₀ * pow(10, logShift));
+  Real shift = μ₀ * pow(10, logShift);
+  if (shiftSquare) shift *= μ₀;
+  LogarithmicFourierTransform fft(N, k, Δτ, pad, shift);
 
   std::vector<Real> C(N);
   std::vector<Real> R(N);