ABACUS/src/EXECS/LiebLin_Fourier_to_x_equal_...

/**********************************************************

This software is part of J.-S. Caux's ABACUS library.

Copyright (c) J.-S. Caux.

-----------------------------------------------------------

File:  LiebLin_Fourier_to_x_equal_t.cc

Purpose:  Fourier transform to static space correlator for LiebLin.

***********************************************************/

#include "ABACUS.h"

using namespace std;
using namespace ABACUS;


int main(int argc, char* argv[])
{
  if (argc != 9) { // provide some info

    cout << endl << "Welcome to ABACUS\t(copyright J.-S. Caux)." << endl;
    cout << endl << "Usage of LiebLin_Fourier_to_x_equal_t executable: " << endl;
    cout << endl << "Provide the following arguments:" << endl << endl;
    cout << "char whichDSF \t\t Which structure factor ?  Options are:  "
      "d for rho rho, g for psi psi{dagger}, o for psi{dagger} psi" << endl;
    cout << "DP c_int \t\t Value of the interaction parameter" << endl;
    cout << "DP L \t\t\t Length of the system" << endl;
    cout << "int N \t\t\t Number of particles" << endl;
    cout << "int iKmin" << endl << "int iKmax \t\t Min and max momentum integers scanned over" << endl;
    cout << "RAW file name" << endl;
    cout << "int Npts_x Number of points in space for the Fourier transform" << endl;
  }

  else { // (argc == 9), correct nr of arguments
    char whichDSF = *argv[1];
    DP c_int = atof(argv[2]);
    DP L = atof(argv[3]);
    int N = atoi(argv[4]);
    int iKmin = atoi(argv[5]);
    int iKmax = atoi(argv[6]);
    char* rawfilename = argv[7];
    int Npts_x = atoi(argv[8]);

    ifstream RAW_infile;
    RAW_infile.open(rawfilename);
    if (RAW_infile.fail()) {
      cout << rawfilename << endl;
      ABACUSerror("Could not open RAW_infile... ");
    }

    // Define the output file name: use the RAW file name but with different suffix

    stringstream SFT_stringstream;    string SFT_string;
    SFT_stringstream << rawfilename << "_sft";
    SFT_string = SFT_stringstream.str();    const char* SFT_Cstr = SFT_string.c_str();
    ofstream SFT_outfile;
    SFT_outfile.open(SFT_Cstr);
    if (SFT_outfile.fail()) ABACUSerror("Could not open SFT_outfile... ");

    // First compute the static structure factor from the RAW data:

    Vect_DP SSF(0.0, iKmax - iKmin + 1);

    DP omega;
    int iK;
    DP FF;
    DP dev;
    string label;

    while (RAW_infile.peek() != EOF) {
      RAW_infile >> omega >> iK >> FF >> dev >> label;
      if (iK >= iKmin && iK <= iKmax) {
	SSF[iK - iKmin] += FF * FF;
      }
    }
    RAW_infile.close();

    // Reset proper normalization:
    DP normalization = twoPI * L;
    for (int iK = 0; iK < iKmax - iKmin + 1; ++iK) SSF[iK] *= normalization/twoPI; // twoPI from integral over omega


    // Now define real-space coordinates: between 0 and L
    Vect_DP xlattice(Npts_x);
    for (int i = 0; i < Npts_x; ++i) xlattice[i] = (i + 0.5) * L/Npts_x;

    // Now the correlation at x:
    Vect_DP FTre(0.0, Npts_x);
    Vect_DP FTim(0.0, Npts_x);

    DP twopioverL = twoPI/L;

    // Fourier transform:
    for (int ix = 0; ix < Npts_x; ++ix) {
      for (int iK = iKmin; iK <= iKmax; ++iK) {
	FTre[ix] += SSF[iK - iKmin] * cos(twopioverL * iK * xlattice[ix]);
	FTim[ix] += SSF[iK - iKmin] * sin(twopioverL * iK * xlattice[ix]);
      }
      // Reset proper normalization: 1/L from space FT,
      FTre[ix] /= L;
      FTim[ix] /= L;

      // Outside of window iKmin, iKmax, we take the DSF to be a constant with delta function
      // at free energy k^2, so DSF = 2\pi N/L \delta(\omega - k^2) (to fit f-sumrule)
      // so SSF becomes N/L.
      // We thus need to correct above by adding
      // \frac{1}{L} \sum_{-\infty}^{iKmin - 1} SSF e^{ikx} + \frac{1}{L} \sum_{iKmax + 1}^\infty SSF e^{ikx}
      // Resumming carefully:
      //if (whichDSF == 'd') {
      //FTre[ix] += (sin(twopioverL * (iKmin - 0.5) * xlattice[ix]) - sin(twopioverL * (iKmax + 0.5) * xlattice[ix]))
      //  * N/(2.0 * L*L * sin(PI * xlattice[ix]/L));
      //FTim[ix] += (-cos(twopioverL * (iKmin - 0.5) * xlattice[ix]) + cos(twopioverL * (iKmax + 0.5) * xlattice[ix]))
      //  * N/(2.0 * L*L * sin(PI * xlattice[ix]/L));
      //}
    }


    // Output to file:
    for (int ix = 0; ix < Npts_x; ++ix) {
      if (ix > 0) SFT_outfile << endl;
      SFT_outfile << xlattice[ix] << "\t" << FTre[ix] << "\t" << FTim[ix];
    }

    SFT_outfile.close();
  }

  return(0);

}