TraceSamples< T > Struct Template Reference

Inheritance diagram for TraceSamples< T >:

Collaboration diagram for TraceSamples< T >:

Public Member Functions
	TraceSamples ()
	~TraceSamples ()
void	set_verbose (bool v)
int	size ()
void	assign_sample_index (const std::string &domain, const std::string &name, int index, std::string label="")
template<int D>
Array< T, D > *	checkout_array (const std::string &domain, const std::string &name, TinyVector< int, DMAX > shape)
template<int D>
Array< T, D > *	checkout_array (const ParticleSet &P, const std::string &name, TinyVector< int, DMAX > shape)
TraceSample< T > *	get_trace (const std::string &domain, const std::string &name)
CombinedTraceSample< T > *	get_combined_trace (const std::string &domain, const std::string &name)
bool	make_combined_trace (const std::string &name, std::vector< std::string > &names, std::vector< TraceReal > &weights)
void	set_unit_size (int usize)
void	screen_writes (TraceRequest &request)
void	order_by_size ()
void	set_buffer_ranges (int &starting_index)
int	total_size ()
int	min_buffer_index ()
int	max_buffer_index ()
void	combine_samples ()
void	reset_combined_samples ()
void	finalize ()
void	register_hdf_data (hdf_archive &f)
void	write_summary (std::string type, std::string pad=" ")
void	user_report (const std::string &type, const std::string &pad=" ")

Public Attributes
std::vector< TraceSample< T > * >	samples
std::map< std::string, std::map< std::string, int > >	sample_indices
std::vector< TraceSample< T > * >	ordered_samples
std::vector< CombinedTraceSample< T > * >	combined_samples
std::vector< Vector< T > * >	combined_sample_vectors
bool	verbose

Detailed Description

template<typename T>
struct qmcplusplus::TraceSamples< T >

Definition at line 751 of file TraceManager.h.

Constructor & Destructor Documentation

TraceSamples()

TraceSamples ( )

inline

Definition at line 760 of file TraceManager.h.

: verbose(false) {}

~TraceSamples()

~TraceSamples ( )

inline

Definition at line 762 of file TraceManager.h.

{ finalize(); }

Member Function Documentation

assign_sample_index()

void assign_sample_index ( const std::string &  domain, const std::string &  name, int  index, std::string  label = ""  )

inline

Definition at line 771 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::checkout_array(), and TraceSamples< std::complex< TraceReal > >::make_combined_trace().

  {
    if (sample_indices.count(domain) > 0 && sample_indices[domain].count(name) > 0)
    {
      APP_ABORT("TraceSamples::checkout " + label + " variable " + name + " already exists in domain " + domain);
    }
    else
    {
      sample_indices[domain][name] = index;
    }
  }

checkout_array() [1/2]

Array<T, D>* checkout_array ( const std::string &  domain, const std::string &  name, TinyVector< int, DMAX >  shape  )

inline

Definition at line 784 of file TraceManager.h.

Referenced by TraceManager::checkout_complex(), TraceManager::checkout_int(), TraceManager::checkout_real(), and qmcplusplus::TEST_CASE().

  {
    int index = samples.size();
    assign_sample_index(domain, name, index, "array");
    std::array<size_t, D> subshape;
    for (int idim = 0; idim < D; idim++)
      subshape[idim] = shape[idim];
    Array<T, D>* a    = new Array<T, D>(subshape);
    TraceSample<T>* s = new TraceSample<T>(domain, name, index, D, shape, a->storage());
    samples.push_back(s);
    if (verbose)
      app_log() << "TraceSamples::checkout_array  " << domain << " " << name << " " << index << std::endl;
    return a;
  }

checkout_array() [2/2]

Array<T, D>* checkout_array ( const ParticleSet &  P, const std::string &  name, TinyVector< int, DMAX >  shape  )

inline

Definition at line 801 of file TraceManager.h.

  {
    const std::string& domain = P.parentName();
    int index                 = samples.size();
    assign_sample_index(domain, name, index, "array");
    std::array<size_t, D> subshape;
    for (int idim = 0; idim < D; idim++)
      subshape[idim] = shape[idim];
    Array<T, D>* a    = new Array<T, D>(subshape);
    TraceSample<T>* s = new TraceSample<T>(domain, name, index, D, shape, a->storage());
    samples.push_back(s);
    s->array_trace = true;
    if (verbose)
      app_log() << "TraceSamples::checkout_array  " << domain << " " << name << " " << index << std::endl;
    return a;
  }

combine_samples()

void combine_samples ( )

inline

Definition at line 963 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::collect_sample().

  {
    for (int i = 0; i < combined_samples.size(); ++i)
      combined_samples[i]->combine();
  }

finalize()

void finalize ( )

inline

Definition at line 977 of file TraceManager.h.

Referenced by TraceManager::finalize_traces(), and TraceSamples< std::complex< TraceReal > >::~TraceSamples().

  {
    //note combined_samples pointers are a subset of those in samples
    //  and so only samples needs to be deleted
    delete_iter(samples.begin(), samples.end());
    delete_iter(combined_sample_vectors.begin(), combined_sample_vectors.end());
    samples.resize(0);
    ordered_samples.resize(0);
    combined_samples.resize(0);
    combined_sample_vectors.resize(0);
    sample_indices.clear();
  }

get_combined_trace()

CombinedTraceSample<T>* get_combined_trace ( const std::string &  domain, const std::string &  name  )

inline

Definition at line 837 of file TraceManager.h.

Referenced by TraceManager::get_complex_combined_trace(), TraceManager::get_int_combined_trace(), and TraceManager::get_real_combined_trace().

  {
    CombinedTraceSample<T>* ts = NULL;
    for (int i = 0; i < combined_samples.size(); ++i)
    {
      CombinedTraceSample<T>& tsc = *combined_samples[i];
      if (tsc.domain == domain && tsc.name == name)
      {
        ts = &tsc;
        break;
      }
    }
    return ts;
  }

get_trace()

TraceSample<T>* get_trace ( const std::string &  domain, const std::string &  name  )

inline

Definition at line 819 of file TraceManager.h.

Referenced by TraceManager::get_complex_trace(), TraceManager::get_int_trace(), and TraceManager::get_real_trace().

  {
    TraceSample<T>* ts = NULL;
    for (int i = 0; i < samples.size(); ++i)
    {
      TraceSample<T>& tsc = *samples[i];
      if (tsc.domain == domain && tsc.name == name)
      {
        ts = &tsc;
        break;
      }
    }
    if (ts == NULL)
      APP_ABORT("TraceSamples::get_trace  failed to get trace for quantity " + name + " in domain " + domain);
    return ts;
  }

make_combined_trace()

bool make_combined_trace ( const std::string &  name, std::vector< std::string > &  names, std::vector< TraceReal > &  weights  )

inline

Definition at line 853 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::make_combined_trace().

  {
    bool created = false;
    if (names.size() != weights.size())
      APP_ABORT("TraceSamples::make_combined_trace  names and weights must be the same size");
    std::map<std::string, std::map<std::string, int>>::iterator it;
    for (it = sample_indices.begin(); it != sample_indices.end(); it++)
    {
      std::string domain                  = it->first;
      std::map<std::string, int>& indices = it->second;
      bool any_present                    = false;
      for (int i = 0; i < names.size(); ++i)
        any_present = any_present || indices.count(names[i]) > 0;
      if (any_present)
      {
        int index                        = samples.size();
        auto* sample                     = new Vector<T>;
        CombinedTraceSample<T>* combined = new CombinedTraceSample<T>(domain, name, index, 0, *sample);
        for (int i = 0; i < names.size(); ++i)
        {
          if (indices.count(names[i]) > 0)
          {
            TraceSample<T>* component = samples[indices[names[i]]];
            combined->add_component(component, weights[i]);
          }
        }
        assign_sample_index(domain, name, index);
        samples.push_back(combined);
        combined_samples.push_back(combined);
        combined_sample_vectors.push_back(sample);
        created = true;
      }
    }
    return created;
  }

max_buffer_index()

int max_buffer_index ( )

inline

Definition at line 954 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::test_buffer_write().

  {
    int max_index = -1;
    for (int i = 0; i < samples.size(); i++)
      max_index = std::max(max_index, samples[i]->buffer_end);
    return max_index;
  }

min_buffer_index()

int min_buffer_index ( )

inline

Definition at line 945 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::test_buffer_write().

  {
    int min_index = 2000000000;
    for (int i = 0; i < samples.size(); i++)
      min_index = std::min(min_index, samples[i]->buffer_start);
    return min_index;
  }

order_by_size()

void order_by_size ( )

inline

Definition at line 916 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::order_and_resize().

  {
    for (int i = 0; i < samples.size(); i++)
      samples[i]->set_data_size();
    ordered_samples.resize(samples.size());
    copy(samples.begin(), samples.end(), ordered_samples.begin());
    sort(ordered_samples.begin(), ordered_samples.end(), TraceSample_comp<T>);
  }

register_hdf_data()

void register_hdf_data ( hdf_archive &  f )

inline

Definition at line 991 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::register_hdf_data().

  {
    std::map<std::string, std::map<std::string, int>>::iterator it;
    std::map<std::string, int>::iterator it2;
    for (it = sample_indices.begin(); it != sample_indices.end(); it++)
    {
      const std::string& domain           = it->first;
      std::map<std::string, int>& indices = it->second;
      f.push(domain);
      for (it2 = indices.begin(); it2 != indices.end(); ++it2)
      {
        const std::string& quantity = it2->first;
        TraceSample<T>& sample      = *samples[it2->second];
        if (sample.write)
        {
          f.push(quantity);
          f.write(sample.dimension, "dimension");
          f.write(sample.shape, "shape");
          f.write(sample.size, "size");
          f.write(sample.unit_size, "unit_size");
          f.write(sample.buffer_start, "row_start");
          f.write(sample.buffer_end, "row_end");
          f.pop();
        }
      }
      f.pop();
    }
  }

reset_combined_samples()

void reset_combined_samples ( )

inline

Definition at line 970 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::collect_sample().

  {
    for (int i = 0; i < combined_samples.size(); ++i)
      combined_samples[i]->reset();
  }

screen_writes()

void screen_writes ( TraceRequest &  request )

inline

Definition at line 899 of file TraceManager.h.

Referenced by TraceManager::screen_writes().

  {
    for (int i = 0; i < samples.size(); i++)
    {
      TraceSample<T>& s = *samples[i];
      bool stream       = request.screen_sample(s.domain, s.name, s.write);
      if (verbose)
        app_log() << "TraceRequest screening " << s.name << " in domain " << s.domain << ". stream: " << stream
                  << " write: " << s.write << std::endl;
      if (!stream && !s.is_combined())
        app_log() << "warning: quantity " + s.name + " in domain " + s.domain +
                " was not requested but is streaming anyway"
                  << std::endl;
    }
  }

set_buffer_ranges()

void set_buffer_ranges ( int &  starting_index )

inline

Definition at line 926 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::order_and_resize().

  {
    for (int i = 0; i < ordered_samples.size(); i++)
    {
      TraceSample<T>& sample = *ordered_samples[i];
      sample.set_buffer_range(starting_index);
    }
  }

set_unit_size()

void set_unit_size ( int  usize )

inline

Definition at line 892 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::order_and_resize(), and TraceSamples< std::complex< TraceReal > >::set_unit_size().

  {
    for (int i = 0; i < samples.size(); i++)
      samples[i]->set_unit_size(usize);
  }

set_verbose()

void set_verbose ( bool  v )

inline

Definition at line 765 of file TraceManager.h.

Referenced by TraceManager::distribute().

{ verbose = v; }

size()

int size ( void  )

inline

Definition at line 768 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::test_buffer_write().

{ return samples.size(); }

total_size()

int total_size ( )

inline

Definition at line 936 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::test_buffer_write().

  {
    int s = 0;
    for (int i = 0; i < samples.size(); i++)
      s += samples[i]->sample.size() * samples[i]->unit_size;
    return s;
  }

user_report()

void user_report ( const std::string &  type, const std::string &  pad = "  "  )

inline

Definition at line 1059 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::user_report().

  {
    std::string pad2 = pad + "  ";
    app_log() << pad << type << " traces provided by estimators" << std::endl;
    std::map<std::string, std::map<std::string, int>>::iterator it;
    std::map<std::string, int>::iterator it2;
    for (it = sample_indices.begin(); it != sample_indices.end(); it++)
    {
      const std::string& domain           = it->first;
      std::map<std::string, int>& indices = it->second;
      app_log() << pad2 << "domain " << domain << ":  ";
      int n = 0;
      for (it2 = indices.begin(); it2 != indices.end(); ++it2)
      {
        if (n % 5 == 0)
          app_log() << std::endl << pad2 << "  ";
        n++;
        const std::string& quantity = it2->first;
        app_log() << quantity << " ";
      }
      app_log() << std::endl;
    }
  }

write_summary()

void write_summary ( std::string  type, std::string  pad = "  "  )

inline

Definition at line 1021 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::write_summary(), and TraceBuffer< TraceInt >::write_summary().

  {
    std::string pad2 = pad + "  ";
    std::string pad3 = pad2 + "  ";
    std::string pad4 = pad3 + "  ";
    app_log() << pad << "TraceSamples<" << type << ">" << std::endl;
    app_log() << pad2 << "nsamples          = " << samples.size() << std::endl;
    app_log() << pad2 << "ncombined_samples = " << combined_samples.size() << std::endl;
    app_log() << pad2 << "sample_indices" << std::endl;
    std::map<std::string, std::map<std::string, int>>::iterator it;
    std::map<std::string, int>::iterator it2;
    for (it = sample_indices.begin(); it != sample_indices.end(); it++)
    {
      const std::string& domain           = it->first;
      std::map<std::string, int>& indices = it->second;
      app_log() << pad3 << domain << std::endl;
      for (it2 = indices.begin(); it2 != indices.end(); ++it2)
        app_log() << pad4 << it2->first << " = " << it2->second << std::endl;
    }
    app_log() << pad2 << "end sample_indices" << std::endl;
    app_log() << pad2 << "combined_sample_vectors = ";
    for (int i = 0; i < combined_sample_vectors.size(); ++i)
      app_log() << (size_t)combined_sample_vectors[i] << " ";
    app_log() << std::endl;
    app_log() << pad2 << "combined_samples" << std::endl;
    for (int i = 0; i < combined_samples.size(); ++i)
      combined_samples[i]->write_summary_combined(i, pad3);
    app_log() << pad2 << "end combined_samples" << std::endl;
    app_log() << pad2 << "samples" << std::endl;
    for (int i = 0; i < ordered_samples.size(); ++i)
      ordered_samples[i]->write_summary(i, pad3);
    //for(int i=0; i<samples.size(); ++i)
    //  samples[i]->write_summary(i,pad3);
    app_log() << pad2 << "end samples" << std::endl;
    app_log() << pad << "end TraceSamples<" << type << ">" << std::endl;
  }

Member Data Documentation

combined_sample_vectors

std::vector<Vector<T>*> combined_sample_vectors

Definition at line 757 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::finalize(), TraceSamples< std::complex< TraceReal > >::make_combined_trace(), and TraceSamples< std::complex< TraceReal > >::write_summary().

combined_samples

std::vector<CombinedTraceSample<T>*> combined_samples

Definition at line 756 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::combine_samples(), TraceSamples< std::complex< TraceReal > >::finalize(), TraceSamples< std::complex< TraceReal > >::get_combined_trace(), TraceSamples< std::complex< TraceReal > >::make_combined_trace(), TraceSamples< std::complex< TraceReal > >::reset_combined_samples(), and TraceSamples< std::complex< TraceReal > >::write_summary().

ordered_samples

std::vector<TraceSample<T>*> ordered_samples

Definition at line 755 of file TraceManager.h.

Referenced by TraceBuffer< TraceInt >::collect_sample(), TraceSamples< std::complex< TraceReal > >::finalize(), TraceSamples< std::complex< TraceReal > >::order_by_size(), TraceSamples< std::complex< TraceReal > >::set_buffer_ranges(), TraceBuffer< TraceInt >::test_buffer_write(), and TraceSamples< std::complex< TraceReal > >::write_summary().

sample_indices

std::map<std::string, std::map<std::string, int> > sample_indices

Definition at line 754 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::assign_sample_index(), TraceSamples< std::complex< TraceReal > >::finalize(), TraceSamples< std::complex< TraceReal > >::make_combined_trace(), TraceSamples< std::complex< TraceReal > >::register_hdf_data(), TraceSamples< std::complex< TraceReal > >::user_report(), and TraceSamples< std::complex< TraceReal > >::write_summary().

samples

std::vector<TraceSample<T>*> samples

Definition at line 753 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::checkout_array(), TraceSamples< std::complex< TraceReal > >::finalize(), TraceSamples< std::complex< TraceReal > >::get_trace(), TraceSamples< std::complex< TraceReal > >::make_combined_trace(), TraceSamples< std::complex< TraceReal > >::max_buffer_index(), TraceSamples< std::complex< TraceReal > >::min_buffer_index(), TraceSamples< std::complex< TraceReal > >::order_by_size(), TraceSamples< std::complex< TraceReal > >::register_hdf_data(), TraceSamples< std::complex< TraceReal > >::screen_writes(), TraceSamples< std::complex< TraceReal > >::set_unit_size(), TraceSamples< std::complex< TraceReal > >::size(), TraceSamples< std::complex< TraceReal > >::total_size(), and TraceSamples< std::complex< TraceReal > >::write_summary().

verbose

bool verbose

Definition at line 758 of file TraceManager.h.

Referenced by TraceSamples< std::complex< TraceReal > >::checkout_array(), TraceSamples< std::complex< TraceReal > >::screen_writes(), and TraceSamples< std::complex< TraceReal > >::set_verbose().

---

The documentation for this struct was generated from the following file:

• /home/pk7/projects/qmc/for_cron_doxygen/qmcpack/src/Estimators/TraceManager.h