hadron_corr_s.h

Go to the documentation of this file.

 
#ifndef HADRON_CORR_S_H
#define HADRON_CORR_S_H
 
#define NUM_STAG_PROPS   8
#define NUM_STAG_PIONS   16
 
 
#include "chromabase.h"
#include "stag_propShift_s.h"
 
namespace Chroma 
{
 
  /*
 
  This is generic code to compute staggered correlators.
 
  */
 
 
 
  class staggered_hadron_corr
  {
 
  public :
    virtual void compute(multi1d<LatticeStaggeredPropagator>& quark_props,
                         int j_decay) = 0 ;
 
    /*
      Write the correlators out
    */
    void dump(int t_source, XMLWriter &xml_out)
    {
      multi2d<Real> pi_corr(no_channel, t_length);
      multi2d<Real> re_corr_fn(no_channel, t_length);
      pi_corr = zero;
 
      // Take the real part of the correlator and average over the sources
      for(int i=0; i < no_channel ; i++){
        for(int t=0; t < t_length; t++){
          int t_eff = (t - t_source + t_length)% t_length;
          re_corr_fn[i][t_eff] = real(corr_fn[i][t]);
        }
      }
                  
      multi1d<Real> Pi(t_length) ; 
      push(xml_out, outer_tag);
      for(int i=0; i < no_channel; i++) {
        Pi = re_corr_fn[i];
        //      std::ostringstream tag;
        //tag << "re_pion" << i;
        // push(xml_out, tag.str());
        push(xml_out, tag_names[i]);
        write(xml_out, inner_tag, Pi);
        pop(xml_out);
      }
      pop(xml_out);
 
 
 
    }
 
    void binary_dump(int t_source, std::string start_name)
    {
      const int magic_number = 66618 ;
      std::string filename ;
      std::string resized_tag;
 
 
      filename = start_name + outer_tag + "."+inner_tag ;
      QDPIO::cout << "out tag, in tag: " <<outer_tag << inner_tag << std::endl;
      BinaryFileWriter speedy ;
      speedy.open(filename);
      write(speedy,magic_number) ;
      write(speedy,t_length) ;
      write(speedy,no_channel) ;
 
    QDPIO::cout << "FILENAME: "<< filename << std::endl;
 
     multi2d<Real> pi_corr(no_channel, t_length);
      multi2d<Real> re_corr_fn(no_channel, t_length);
      pi_corr = zero;
 
      // Take the real part of the correlator and average over the sources
      for(int i=0; i < no_channel ; i++){
        for(int t=0; t < t_length; t++){
          int t_eff = (t - t_source + t_length)% t_length;
          re_corr_fn[i][t_eff] = real(corr_fn[i][t]);
        }
      }
                  
      multi1d<Real> Pi(t_length) ; 
 
      for(int i=0; i < no_channel; i++) {
        Pi = re_corr_fn[i];
        resized_tag.assign(tag_names[i]);
        resized_tag.resize(45,' ');
 
 
        write(speedy, tag_names[i]);
        write(speedy, Pi);
 
      }
 
 
      speedy.close();
 
 
 
    }
 
 
 
 
 
 
    staggered_hadron_corr(int t_len, int t_chan, 
                          const multi1d<LatticeColorMatrix> & uin, 
                          Stag_shift_option type_of_shift_in  ) 
      : t_length(t_len) ,
        no_channel(t_chan)
    {
      corr_fn.resize(no_channel, t_length);
      u.resize(4) ; 
 
      if( uin.size() != 4 ) { 
        QDPIO::cerr << "staggered_hadron_corr: input guage config has wrong number of dimensions " << uin.size() << std::endl;
        QDP_abort(1);
      };
 
      u = uin ; 
      type_of_shift = type_of_shift_in ; 
    }
 
 
    virtual ~staggered_hadron_corr()
    {
      corr_fn.resize(1, 1);
    }
 
 
  protected:
 
    multi2d<DComplex> corr_fn ; 
    std::string outer_tag ; 
    std::string inner_tag ; 
    multi1d<std::string> tag_names ; 
    multi1d<LatticeColorMatrix> u ; // this should handle or state
 
    LatticeStaggeredPropagator shift_deltaProp(multi1d<int>& delta, 
                                               const LatticeStaggeredPropagator& src)
 
    {
 
      switch (type_of_shift)
        {
        case NON_GAUGE_INVAR :
          return shiftDeltaProp(delta,src) ;
          break ;
        case GAUGE_INVAR :
          return shiftDeltaPropCov(delta,src,u,false) ;
          break ;
        case SYM_GAUGE_INVAR :
          return shiftDeltaPropCov(delta,src,u,true) ; // symm shifting
          break ;
        case SYM_NON_GAUGE_INVAR:
          return shiftDeltaProp(delta,src,true) ; // symm shifting
          break ;
        default :
          /**************************************************************************/
 
          QDPIO::cerr << "Shift type " << type_of_shift << " unsupported." << std::endl;
          QDP_abort(1);
        }
 
      return zero;   // make compiler happy
    }
 
 
  private :
    int t_length ; 
    int no_channel ; 
 
    Stag_shift_option type_of_shift; 
 
 
 
 
  } ;
 
 
}  // end namespace Chroma
 
 
#endif