From f8dd6b3f3179175c847b53c81d8d8702d4abc590 Mon Sep 17 00:00:00 2001
From: Alejandro Gallo <aamsgallo@gmail.com>
Date: Thu, 27 Jan 2022 20:46:23 +0100
Subject: [PATCH] Templatize Input

---
 atrip.org | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/atrip.org b/atrip.org
index 2c14694..a3aa110 100644
--- a/atrip.org
+++ b/atrip.org
@@ -1964,8 +1964,9 @@ namespace atrip {
     static int np;
     static void init();
 
+    template <typename F=double>
     struct Input {
-      CTF::Tensor<double> *ei = nullptr
+      CTF::Tensor<F> *ei = nullptr
                         , *ea = nullptr
                         , *Tph = nullptr
                         , *Tpphh = nullptr
@@ -1976,13 +1977,13 @@ namespace atrip {
       int maxIterations = 0, iterationMod = -1, percentageMod = -1;
       bool barrier = false;
       bool chrono = false;
-      Input& with_epsilon_i(CTF::Tensor<double> * t) { ei = t; return *this; }
-      Input& with_epsilon_a(CTF::Tensor<double> * t) { ea = t; return *this; }
-      Input& with_Tai(CTF::Tensor<double> * t) { Tph = t; return *this; }
-      Input& with_Tabij(CTF::Tensor<double> * t) { Tpphh = t; return *this; }
-      Input& with_Vabij(CTF::Tensor<double> * t) { Vpphh = t; return *this; }
-      Input& with_Vijka(CTF::Tensor<double> * t) { Vhhhp = t; return *this; }
-      Input& with_Vabci(CTF::Tensor<double> * t) { Vppph = t; return *this; }
+      Input& with_epsilon_i(CTF::Tensor<F> * t) { ei = t; return *this; }
+      Input& with_epsilon_a(CTF::Tensor<F> * t) { ea = t; return *this; }
+      Input& with_Tai(CTF::Tensor<F> * t) { Tph = t; return *this; }
+      Input& with_Tabij(CTF::Tensor<F> * t) { Tpphh = t; return *this; }
+      Input& with_Vabij(CTF::Tensor<F> * t) { Vpphh = t; return *this; }
+      Input& with_Vijka(CTF::Tensor<F> * t) { Vhhhp = t; return *this; }
+      Input& with_Vabci(CTF::Tensor<F> * t) { Vppph = t; return *this; }
       Input& with_maxIterations(int i) { maxIterations = i; return *this; }
       Input& with_iterationMod(int i) { iterationMod = i; return *this; }
       Input& with_percentageMod(int i) { percentageMod = i; return *this; }
@@ -1993,7 +1994,8 @@ namespace atrip {
     struct Output {
       double energy;
     };
-    static Output run(Input const& in);
+    template <typename F=double>
+    static Output run(Input<F> const& in);
   };
 
 }