spectrum_s.cc

Go to the documentation of this file.

/*! \file
 * \brief Main code for spectrum measurements
 */
 
#include "chroma.h"
#include "meas/gfix/coulgauge.h"
 
using namespace Chroma;
 
void setup_levelthree(QDP::multi1d<int> nrow, int *argc, char ***argv ) ;
 
 
 
 bool linkageHack(void)
{
  bool foo = true;
 
  // Inline Measurements
  foo &= StaggeredTypeFermActsEnv::registerAll();
  foo &= InlineAggregateEnv::registerAll();
  foo &= GaugeInitEnv::registerAll();
 
  return foo;
}
 
 
//! Spectrum measurements
/*! \defgroup spectrumain Spectrum measurements
 *  \ingroup main
 *
 * Main program for spectrum measurements
 */
 
int main(int argc, char **argv)
{
  // Put the machine into a known state
  Chroma::initialize(&argc, &argv);
 
  START_CODE();
 
  linkageHack();
  QDPIO::cerr << "Staggered measurement code." <<  std::endl;
 
  XMLReader xml_in ;
  std::string in_name = Chroma::getXMLInputFileName() ;
  try
  {
    xml_in.open(in_name);
  }
  catch (...)
  {
    QDPIO::cerr << "Error reading input file " << in_name << std::endl;
    QDPIO::cerr << "The input file name can be passed via the -i flag " << std::endl;
    QDPIO::cerr << "The default name is ./DATA" << std::endl;
    throw;
  }
 
  // Read data
  InlineStaggeredSpectrumParams input(xml_in, "/spectrum_s");
  InlineStaggeredSpectrum  meas(input);
 
  // Specify lattice size, shape, etc.
  Layout::setLattSize(input.param.nrow);
  Layout::create();
 
#if defined(BUILD_CPS_ASQTAD_INVERTER)
  setup_levelthree(input.param.nrow, &argc, &argv ) ;
#endif
 
 
  // Read gauge field info
  Cfg_t  cfg;
  try
  {
    read(xml_in, "/spectrum_s/Cfg", cfg);
  }
  catch(const std::string& e)
  {
    QDP_error_exit("Error reading in spectrum_s: %s", e.c_str());
  }
 
  // Start up the gauge field
  multi1d<LatticeColorMatrix> u(Nd);
  XMLBufferWriter config_xml;
  {
    XMLReader gauge_file_xml, gauge_xml;
 
    QDPIO::cout << "Initialize Gauge field" << std::endl;
    gaugeStartup(gauge_file_xml, gauge_xml, u, cfg);
    QDPIO::cout << "Gauge field initialized!" << std::endl;
 
    config_xml << gauge_xml;
  }
 
  // Check if the gauge field configuration is unitarized
  unitarityCheck(u);
 
  bool do_gauge_transform = false;
  read(xml_in, "/spectrum_s/Param/do_gauge_transform",do_gauge_transform );
  if( do_gauge_transform )
  {
    rgauge(u); 
    QDPIO::cout << "Random gauge transform on gauge fields done." << std::endl;
  }
 
 
  // Output
  XMLFileWriter& xml_out = Chroma::getXMLOutputInstance();
 
 
  bool gauge_invar_oper = false;
  read(xml_in, "/spectrum_s/Param/gauge_invar_oper",gauge_invar_oper );
 
  // GAUGE FIX 
  if( !gauge_invar_oper){
    //not using covariant shifts, should gauge-fix
 
    //read some input the sloppy way
    Real GFAccu;  // Gauge fixing tolerance 
    Real OrPara;  // Gauge fixing over-relaxation param
    int  GFMax;   // Maximum gauge fixing iterations
    int  j_decay; // the t-direction
 
    read(xml_in, "/spectrum_s/Param/GFAccu", GFAccu );
    read(xml_in, "/spectrum_s/Param/OrPara", OrPara );
    read(xml_in, "/spectrum_s/Param/GFMax", GFMax );
    read(xml_in, "/spectrum_s/Param/j_decay",j_decay  );
    int  n_gf;
 
 
    QDPIO::cout << "Not using covariant shift operators.";
    QDPIO::cout << "Must gauge-fix" <<  std::endl;
 
    QDPIO::cout << "Starting to fix to Coulomb gauge" <<  std::endl;
    coulGauge(u, n_gf, j_decay, GFAccu, GFMax, true,    OrPara);
    QDPIO::cout << "No. of gauge fixing iterations =" << n_gf << std::endl;
  }
 
 
  // Reset and set the default gauge field
  InlineDefaultGaugeField::reset();
  InlineDefaultGaugeField::set(u, config_xml);
 
  // Do the measurement
  try
  {
    unsigned long cur_update = 0;
    meas(cur_update, xml_out);
  }
  catch(std::bad_cast) 
  {
    QDPIO::cerr << "SPECTRUM_S: caught cast error" << std::endl;
    QDP_abort(1);
  }
  catch(const std::string& e) 
  {
    QDPIO::cerr << "SPECTRUM_S: Caught Exception: " << e << std::endl;
    QDP_abort(1);
  }
  catch(std::exception& e) 
  {
    QDPIO::cerr << "SPECTRUM_S: Caught standard library exception: " << e.what() << std::endl;
    QDP_abort(1);
  }
  catch(...)
  {
    QDPIO::cerr << "SPECTRUM_S: caught generic exception during measurement" << std::endl;
    QDP_abort(1);
  }
 
  // Reset the default gauge field
  InlineDefaultGaugeField::reset();
 
  xml_out.flush();
  xml_out.close();
 
  END_CODE();
 
  // Time to bolt
  Chroma::finalize();
 
  exit(0);
}