BranchIO< SFNB > Class Template Reference

Collaboration diagram for BranchIO< SFNB >:

Public Types
using	RealType = typename SFNB::RealType
using	BranchModeType = typename SFNB::BranchModeType
using	IParamType = typename SFNB::IParamType
using	VParamType = typename SFNB::VParamType

Public Member Functions
	BranchIO (SFNB &source, Communicate *c)
bool	write (const std::string &fname)
bool	read (const std::string &fname)
void	bcast_state ()

Static Public Member Functions
static void	initAttributes ()

Public Attributes
SFNB &	ref
Communicate *	myComm

Static Public Attributes
static std::vector< std::string >	vParamName
static std::vector< std::string >	iParamName

Detailed Description

template<class SFNB>
class qmcplusplus::BranchIO< SFNB >

Definition at line 22 of file BranchIO.h.

Member Typedef Documentation

BranchModeType

using BranchModeType = typename SFNB::BranchModeType

Definition at line 26 of file BranchIO.h.

IParamType

using IParamType = typename SFNB::IParamType

Definition at line 27 of file BranchIO.h.

RealType

using RealType = typename SFNB::RealType

Definition at line 25 of file BranchIO.h.

VParamType

using VParamType = typename SFNB::VParamType

Definition at line 28 of file BranchIO.h.

Constructor & Destructor Documentation

BranchIO()

BranchIO ( SFNB &  source, Communicate *  c  )

inline

Definition at line 32 of file BranchIO.h.

: ref(source), myComm(c) {}

Member Function Documentation

bcast_state()

void bcast_state

(

)

Definition at line 210 of file BranchIO.cpp.

References copy(), and qmcplusplus::n.

{
  int n = ref.vParam.size() + ref.iParam.size();
  std::vector<RealType> pdata(n + 1 + 16, -1);

  if (myComm->rank() == 0)
  {
    copy(ref.vParam.begin(), ref.vParam.end(), pdata.begin());
    copy(ref.iParam.begin(), ref.iParam.end(), pdata.begin() + ref.vParam.size());
    int offset      = n;
    pdata[offset++] = ref.BranchMode.to_ulong();
    copy(ref.EnergyHist.properties, ref.EnergyHist.properties + 4, pdata.begin() + offset);
    offset += 4;
    copy(ref.VarianceHist.properties, ref.VarianceHist.properties + 4, pdata.begin() + offset);
    offset += 4;
    copy(ref.R2Accepted.properties, ref.R2Accepted.properties + 4, pdata.begin() + offset);
    offset += 4;
    copy(ref.R2Proposed.properties, ref.R2Proposed.properties + 4, pdata.begin() + offset);
  }

  //broadcast to the nodes : need to add a namespace mpi::
  myComm->bcast(pdata);

  if (myComm->rank())
  {
    int ii = 0;
    for (auto& vpar : ref.vParam)
      vpar = pdata[ii++];
    for (int i = 0; i < ref.iParam.size(); ++i, ++ii)
      ref.iParam[i] = static_cast<int>(pdata[ii]);
    ref.BranchMode = static_cast<unsigned long>(pdata[ii]);
  }
  {
    //update historgram
    int ii = n + 1;
    ref.EnergyHist.reset(pdata[ii], pdata[ii + 1], pdata[ii + 2]);
    ii += 4;
    ref.VarianceHist.reset(pdata[ii], pdata[ii + 1], pdata[ii + 2]);
    ii += 4;
    ref.R2Accepted.reset(pdata[ii], pdata[ii + 1], pdata[ii + 2]);
    ii += 4;
    ref.R2Proposed.reset(pdata[ii], pdata[ii + 1], pdata[ii + 2]);
  }
}

initAttributes()

void initAttributes ( )

static

Definition at line 82 of file BranchIO.cpp.

{
  if (vParamName.size())
    return;
  vParamName.resize(10);
  vParamName[0] = "tau";
  vParamName[1] = "taueff";
  vParamName[2] = "etrial";
  vParamName[3] = "eref";
  vParamName[4] = "branchmax";
  vParamName[5] = "branchcutoff";
  vParamName[6] = "branchfilter";
  vParamName[7] = "sigma";
  vParamName[8] = "acc_energy";
  vParamName[9] = "acc_samples";

  iParamName.resize(7);
  iParamName[0] = "warmupsteps";
  iParamName[1] = "energyupdateinterval";
  iParamName[2] = "counter";
  iParamName[3] = "targetwalkers";
  iParamName[4] = "maxwalkers";
  iParamName[5] = "minwalkers";
  iParamName[6] = "brnachinterval";
}

read()

bool read

(

const std::string &

fname

)

Definition at line 166 of file BranchIO.cpp.

{
  int found_config = 0;

  if (myComm->rank() == 0)
  {
    initAttributes();
    using boost::property_tree::ptree;
    ptree pt;
    std::string xname = fname + ".qmc.xml";
    read_xml(xname, pt);
    if (!pt.empty())
    {
      ref.BranchMode = pt.get<BranchModeType>("state.branchmode");

      get_histogram("state.energy", ref.EnergyHist, pt);
      get_histogram("state.variance", ref.VarianceHist, pt);
      get_histogram("state.r2accepted", ref.R2Accepted, pt);
      get_histogram("state.r2proposed", ref.R2Proposed, pt);

      int i = 0;
      BOOST_FOREACH (const ptree::value_type& v, pt.get_child("state.iparam"))
      {
        ref.iParam[i++] = v.second.get_value<int>();
      }
      auto it_vparam = ref.vParam.begin();
      BOOST_FOREACH (const ptree::value_type& v, pt.get_child("state.vparam"))
      {
        *(it_vparam++) = v.second.get_value<double>();
      }
      found_config = 1;
    }
  }
  myComm->bcast(found_config);

  if (!found_config)
    return false;

  bcast_state();

  return true;
}

write()

bool write

(

const std::string &

fname

)

Definition at line 109 of file BranchIO.cpp.

References hdf_archive::create(), qmcplusplus::hdf::main_state, hdf_archive::push(), qmcplusplus::hdf::qmc_status, and hdf_archive::write().

Referenced by SimpleFixedNodeBranch::write().

{
  if (myComm->rank())
    return true;

#if defined(HAVE_LIBBOOST)
  initAttributes();
  using boost::property_tree::ptree;
  ptree pt;

  // Put log filename in property tree
  pt.put("state.branchmode", ref.BranchMode);

  auto it_vparam = ref.vParam.begin();
  for (auto& name : vParamName)
    pt.put("state.vparam." + name, *(it_vparam++));
  for (int i = 0; i < iParamName.size(); ++i)
    pt.put("state.iparam." + iParamName[i], ref.iParam[i]);

  put_histogram("state.energy", ref.EnergyHist, pt);
  put_histogram("state.variance", ref.VarianceHist, pt);
  put_histogram("state.r2accepted", ref.R2Accepted, pt);
  put_histogram("state.r2proposed", ref.R2Proposed, pt);
  std::string xname = fname + ".qmc.xml";
  write_xml(xname, pt);
#else
  //append .qmc.h5 if missing
  std::string h5name(fname);
  if (fname.find("qmc.h5") >= fname.size())
    h5name.append(".qmc.h5");
  hdf_archive dump(myComm);
  hid_t fid = dump.create(h5name);
  dump.push(hdf::main_state);
  dump.push(hdf::qmc_status);
  std::string v_header("tau:taueff:etrial:eref:branchmax:branchcutoff:branchfilter:sigma:acc_energy:acc_samples");
  std::string i_header("warmupsteps:energyupdateinterval:counter:targetwalkers:maxwalkers:minwalkers:branchinterval");
  dump.write(v_header, "vparam_def");
  dump.write(i_header, "iparam_def");
  dump.write(ref.vParam, "vparam");
  dump.write(ref.iParam, "iparam");
  dump.write(ref.BranchMode, "branchmode");
  dump.push("histogram");
  dump.write(ref.EnergyHist, "energy");
  dump.write(ref.VarianceHist, "variance");
  dump.write(ref.R2Accepted, "r2accepted");
  dump.write(ref.R2Proposed, "r2proposed");
  //PopHist is not being used in 2010-10-19
  //if(ref.BranchMode[SimpleFixedNodeBranch::B_DMC])
  //{
  //  dump.push("population");
  //  dump.write(ref.PopHist.myData,"histogram");
  //}
#endif
  return true;
}

Member Data Documentation

iParamName

std::vector< std::string > iParamName

static

Definition at line 40 of file BranchIO.h.

myComm

Communicate* myComm

Definition at line 31 of file BranchIO.h.

ref

SFNB& ref

Definition at line 30 of file BranchIO.h.

vParamName

std::vector< std::string > vParamName

static

Definition at line 39 of file BranchIO.h.

---

The documentation for this class was generated from the following files:

• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/QMCDrivers/BranchIO.h
• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/QMCDrivers/BranchIO.cpp