Removed some more duplicate code.

1 files changed, 7 insertions, 1 deletions

diff --git a/fourier.cpp b/fourier.cpp
index 7808989..8943521 100644
--- a/fourier.cpp
+++ b/fourier.cpp
@@ -71,7 +71,7 @@ Real energy(const std::vector<Real>& C, const std::vector<Real>& R, unsigned p,
 
 std::tuple<std::vector<Complex>, std::vector<Complex>> RddfCtdfCt(FourierTransform& fft, const std::vector<Real>& C, const std::vector<Real>& R, unsigned p, unsigned s, Real λ) {
   std::vector<Real> RddfC(C.size());
-  for (unsigned i = 0; i < C.size(); i++) {
+  for (unsigned i = 0; i < C.size() / 2; i++) {
     RddfC[i] = R[i] * ddf(λ, p, s, C[i]);
   }
   std::vector<Complex> RddfCt = fft.fourier(RddfC);
@@ -84,3 +84,9 @@ std::tuple<std::vector<Complex>, std::vector<Complex>> RddfCtdfCt(FourierTransfo
 
   return {RddfCt, dfCt};
 }
+
+Real estimateZ(FourierTransform& fft, const std::vector<Real>& C, const std::vector<Complex>& Ct, const std::vector<Real>& R, const std::vector<Complex>& Rt, unsigned p, unsigned s, Real λ, Real y) {
+  auto [RddfCt, dfCt] = RddfCtdfCt(fft, C, R, p, s, λ);
+
+  return ((std::conj(Rt[0]) + pow(y, 2) * (RddfCt[0] * Ct[0] + dfCt[0] * std::conj(Rt[0]))) / Ct[0]).real();
+}