inline_wilson_flow.cc

Go to the documentation of this file.

// 
/*! \file
 * \brief Wilson Flow driver
 *
 */
 
#include "inline_wilson_flow.h"
#include "meas/inline/abs_inline_measurement_factory.h"
#include "meas/glue/mesplq.h"
#include "util/info/proginfo.h"
#include "meas/inline/make_xml_file.h"
#include "meas/inline/io/named_objmap.h"
 
// The wilson flow code
#include "meas/glue/wilson_flow_w.h"
 
 
namespace Chroma 
{ 
 
  namespace InlineWilsonFlowEnv 
  {
    //! Read input -- gauge fields
    void read(XMLReader& xml, const std::string& path, Params::NamedObject_t& input)
    {
      XMLReader inputtop(xml, path);
 
      read(inputtop, "gauge_in", input.gauge_in);
      read(inputtop, "gauge_out", input.gauge_out);
    }
 
    //! write output -- gauge fields
    void write(XMLWriter& xml, const std::string& path, const Params::NamedObject_t& input)
    {
      push(xml, path);
 
      write(xml, "gauge_in", input.gauge_in);
      write(xml, "gauge_out", input.gauge_out);
 
      pop(xml);
    }
 
 
    //! read input
    void read(XMLReader& xml, const std::string& path, Params::Param_t& input)
    {
      XMLReader inputtop(xml, path);
 
      read(inputtop, "version", input.version);
      read(inputtop, "nstep", input.nstep);
      read(inputtop, "wtime", input.wtime);
      read(inputtop, "t_dir",input.t_dir);
 
      switch (input.version) 
      {
      case 1:
        input.smear_dirs.resize(Nd);
        input.smear_dirs = true;
        break;
 
      case 2:
        read(inputtop, "smear_dirs", input.smear_dirs);
        break;
 
      default:
        QDPIO::cerr << "WILSON_FLOW: Input version " << input.version 
                    << " unsupported." << std::endl;
        QDP_abort(1);
      }
 
    }
 
    //! write output
    void write(XMLWriter& xml, const std::string& path, const Params::Param_t& input)
    {
      push(xml, path);
    
      write(xml, "version", input.version);
      write(xml, "nstep", input.nstep);
      write(xml, "wtime", input.wtime);
      write(xml, "t_dir",input.t_dir);
      write(xml, "smear_dirs", input.smear_dirs);
        
      pop(xml);
    }
 
 
    //! read input
    void read(XMLReader& xml, const std::string& path, Params& input)
    {
      InlineWilsonFlowEnv::Params tmp(xml, path);
      input = tmp;
    }
 
    //! write output
    void write(XMLWriter& xml, const std::string& path, const Params& input)
    {
      push(xml, path);
    
      write(xml, "Param", input.param);
      write(xml, "NamedObject", input.named_obj);
 
      pop(xml);
    }
  } // namespace InlineWilsonFlowEnv 
 
 
  namespace InlineWilsonFlowEnv 
  {
    namespace
    {
      AbsInlineMeasurement* createMeasurement(XMLReader& xml_in, 
                                              const std::string& path) 
      {
        return new InlineMeas(Params(xml_in, path));
      }
 
      //! Local registration flag
      bool registered = false;
    }
      
    const std::string name = "WILSON_FLOW";
 
    //! Register all the factories
    bool registerAll() 
    {
      bool success = true; 
      if (! registered)
      {
        success &= TheInlineMeasurementFactory::Instance().registerObject(name, createMeasurement);
        registered = true;
      }
      return success;
    }
 
 
    //--------------------------------------------------------------------------
    // Param stuff
    Params::Params() { frequency = 0; }
 
    Params::Params(XMLReader& xml_in, const std::string& path) 
    {
      try 
      {
        XMLReader paramtop(xml_in, path);
 
        if (paramtop.count("Frequency") == 1)
          read(paramtop, "Frequency", frequency);
        else
          frequency = 1;
 
        // Parameters for source construction
        read(paramtop, "Param", param);
 
        // Read in the output propagator/source configuration info
        read(paramtop, "NamedObject", named_obj);
 
        // Possible alternate XML file pattern
        if (paramtop.count("xml_file") != 0) 
        {
          read(paramtop, "xml_file", xml_file);
        }
      }
      catch(const std::string& e) 
      {
        QDPIO::cerr << __func__ << ": Caught Exception reading XML: " << e << std::endl;
        QDP_abort(1);
      }
    }
 
 
 
    // Function call
    void 
    InlineMeas::operator()(unsigned long update_no,
                           XMLWriter& xml_out) 
    {
      // If xml file not empty, then use alternate
      if (params.xml_file != "")
      {
        std::string xml_file = makeXMLFileName(params.xml_file, update_no);
 
        push(xml_out, "WilsonFlow");
        write(xml_out, "update_no", update_no);
        write(xml_out, "xml_file", xml_file);
        pop(xml_out);
 
        XMLFileWriter xml(xml_file);
        func(update_no, xml);
      }
      else
      {
        func(update_no, xml_out);
      }
    }
 
 
    // Real work done here
    // Create the diluted source and apply Lanczos quarature 
    void 
    InlineMeas::func(unsigned long update_no,
                     XMLWriter& xml_out) 
    {
      START_CODE();
 
      StopWatch snoop;
      snoop.reset();
      snoop.start();
 
      push(xml_out, "WilsonFlow");
      write(xml_out, "update_no", update_no);
 
      QDPIO::cout << name << ": Wilson Flowing a configuration" << std::endl;
 
      proginfo(xml_out);    // Print out basic program info
 
      push(xml_out, "Output_version");
      write(xml_out, "out_version", 1);
      pop(xml_out);
 
      // Write out the input
      write(xml_out, "Input", params);
 
      // Test and grab a reference to the gauge field
      // -- we really need only two gauge fields --
      multi1d<LatticeColorMatrix> u ; 
     
      push(xml_out,"GaugeFieldInfo");
      XMLBufferWriter gauge_xml;
      try
        {
          u = TheNamedObjMap::Instance().getData< multi1d<LatticeColorMatrix> >(params.named_obj.gauge_in);
          TheNamedObjMap::Instance().get(params.named_obj.gauge_in).getRecordXML(gauge_xml);
        }
      catch( std::bad_cast ) 
        {
          QDPIO::cerr << name << ": caught dynamic cast error" << std::endl;
          QDP_abort(1);
        }
      catch (const std::string& e) 
        {
          QDPIO::cerr << name << ": std::map call failed: " << e << std::endl;
          QDP_abort(1);
        }
      // Write out the config header
      write(xml_out, "Config_info", gauge_xml);
 
      pop(xml_out);
      
 
      // Calculate some gauge invariant observables just for info.
      MesPlq(xml_out, "GaugeObservables", u);
      
      
      
      multi1d<LatticeColorMatrix> wf_u = u ; 
      Real eps  = params.param.wtime/params.param.nstep ;
 
      wilson_flow(xml_out, wf_u, params.param.nstep, eps, params.param.t_dir, params.param.smear_dirs);
 
 
      // Calculate some gauge invariant observables just for info.
      MesPlq(xml_out, "WilsonFlowGaugeObservables", wf_u);
 
 
       // Now store the configuration to a memory object
      {
        XMLBufferWriter file_xml, record_xml;
        push(file_xml, "gauge");
        write(file_xml, "id", int(0));
        write(file_xml, "WilsonFlowParams", params.param);
        pop(file_xml);
        record_xml << gauge_xml;
 
        // Store the gauge field
        TheNamedObjMap::Instance().create< multi1d<LatticeColorMatrix> >(params.named_obj.gauge_out);
        TheNamedObjMap::Instance().getData< multi1d<LatticeColorMatrix> >(params.named_obj.gauge_out) = wf_u;
        TheNamedObjMap::Instance().get(params.named_obj.gauge_out).setFileXML(file_xml);
        TheNamedObjMap::Instance().get(params.named_obj.gauge_out).setRecordXML(record_xml);
      }
 
      pop(xml_out);  // WilsonFlow
      
      snoop.stop();
      QDPIO::cout << name << ": total time = "
                  << snoop.getTimeInSeconds() 
                  << " secs" << std::endl;
      
      QDPIO::cout << name << ": ran successfully" << std::endl;
      
      END_CODE();
    }
 
  }
  
 
} // namespace Chroma