Geant4ParticleCollection.cc

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#include "dd4pod/Geant4ParticleCollection.h"
 
#include <numeric>
 
// standard includes
#include <stdexcept>
#include <iomanip>
 
namespace dd4pod {
 
 
Geant4ParticleCollection::Geant4ParticleCollection() :
  m_isValid(false), m_isReadFromFile(false), m_collectionID(0), m_entries(),
  m_data(new Geant4ParticleDataContainer()) {
  m_vecmem_info.emplace_back("int", &m_vec_parents);
  m_vec_parents = new std::vector<int>();
  m_vecmem_info.emplace_back("int", &m_vec_daughters);
  m_vec_daughters = new std::vector<int>();
}
 
Geant4ParticleCollection::~Geant4ParticleCollection() {
  clear();
  if (m_data) delete m_data;
  if (m_vec_parents) delete m_vec_parents;
  if (m_vec_daughters) delete m_vec_daughters;
}
 
const Geant4Particle Geant4ParticleCollection::operator[](unsigned int index) const {
  return Geant4Particle(m_entries[index]);
}
 
const Geant4Particle Geant4ParticleCollection::at(unsigned int index) const {
  return Geant4Particle(m_entries.at(index));
}
 
Geant4Particle Geant4ParticleCollection::operator[](unsigned int index) {
  return Geant4Particle(m_entries[index]);
}
 
Geant4Particle Geant4ParticleCollection::at(unsigned int index) {
  return Geant4Particle(m_entries.at(index));
}
 
size_t Geant4ParticleCollection::size() const {
  return m_entries.size();
}
 
Geant4Particle Geant4ParticleCollection::create() {
  auto obj = m_entries.emplace_back(new Geant4ParticleObj());
  m_vecs_parents.push_back(obj->m_parents);
  m_vecs_daughters.push_back(obj->m_daughters);
 
  obj->id = {int(m_entries.size() - 1), m_collectionID};
  return Geant4Particle(obj);
}
 
void Geant4ParticleCollection::clear() {
  m_data->clear();
  m_vec_parents->clear();
  for (auto& vec : m_vecs_parents) { delete vec; }
  m_vecs_parents.clear();
 
  m_vec_daughters->clear();
  for (auto& vec : m_vecs_daughters) { delete vec; }
  m_vecs_daughters.clear();
 
  for (auto& obj : m_entries) { delete obj; }
  m_entries.clear();
}
 
void Geant4ParticleCollection::prepareForWrite() {
  const auto size = m_entries.size();
  m_data->reserve(size);
  for (auto& obj : m_entries) { m_data->push_back(obj->data); }
 
  // if the collection has been read from a file the rest of the information is
  // already in the correct format and we have to skip it, since the temporary
  // buffers are invalid
  if (m_isReadFromFile) return;
  for (auto& pointer : m_refCollections) { pointer->clear(); }
 
  const int parents_size = std::accumulate(m_entries.begin(), m_entries.end(), 0, [](int sum, const Geant4ParticleObj* obj) { return sum + obj->m_parents->size(); });
  m_vec_parents->reserve(parents_size);
  int parents_index = 0;
  const int daughters_size = std::accumulate(m_entries.begin(), m_entries.end(), 0, [](int sum, const Geant4ParticleObj* obj) { return sum + obj->m_daughters->size(); });
  m_vec_daughters->reserve(daughters_size);
  int daughters_index = 0;
  for (int i = 0, size = m_data->size(); i != size; ++i) {
    (*m_data)[i].parents_begin = parents_index;
    (*m_data)[i].parents_end += parents_index;
    parents_index = (*m_data)[i].parents_end;
    for (const auto& it : (*m_vecs_parents[i])) { m_vec_parents->push_back(it); }
 
    (*m_data)[i].daughters_begin = daughters_index;
    (*m_data)[i].daughters_end += daughters_index;
    daughters_index = (*m_data)[i].daughters_end;
    for (const auto& it : (*m_vecs_daughters[i])) { m_vec_daughters->push_back(it); }
 
  }
}
 
void Geant4ParticleCollection::prepareAfterRead() {
  int index = 0;
  for (auto& data : *m_data) {
    auto obj = new Geant4ParticleObj({index, m_collectionID}, data);
 
    obj->m_parents = m_vec_parents;
    obj->m_daughters = m_vec_daughters;
    m_entries.emplace_back(obj);
    ++index;
  }
 
  // at this point we are done with the I/O buffer and can safely clear it to not
  // have a redundant (but now useless) copy of the data
  m_data->clear();
  m_isValid = true;
  m_isReadFromFile = true;
}
 
bool Geant4ParticleCollection::setReferences(const podio::ICollectionProvider* collectionProvider) {
 
  return true; //TODO: check success
}
 
void Geant4ParticleCollection::push_back(ConstGeant4Particle object) {
  const int size = m_entries.size();
  auto obj = object.m_obj;
  if (obj->id.index == podio::ObjectID::untracked) {
    obj->id = {size, m_collectionID};
    m_entries.push_back(obj);
 
    m_vecs_parents.push_back(obj->m_parents);
    m_vecs_daughters.push_back(obj->m_daughters);
  } else {
    throw std::invalid_argument("Object already in a collection. Cannot add it to a second collection");
  }
}
 
void Geant4ParticleCollection::setBuffer(void* address) {
  if (m_data) delete m_data;
  m_data = static_cast<Geant4ParticleDataContainer*>(address);
}
 
const Geant4Particle Geant4ParticleCollectionIterator::operator*() const {
  m_object.m_obj = (*m_collection)[m_index];
  return m_object;
}
 
const Geant4Particle* Geant4ParticleCollectionIterator::operator->() const {
  m_object.m_obj = (*m_collection)[m_index];
  return &m_object;
}
 
const Geant4ParticleCollectionIterator& Geant4ParticleCollectionIterator::operator++() const {
  ++m_index;
  return *this;
}
 
std::ostream& operator<<(std::ostream& o, const Geant4ParticleCollection& v) {
  const auto old_flags = o.flags();
  o << "          id:          ID:    g4Parent:      reason:        mask:       steps: secondaries:       pdgID:      status:   colorFlow:   genStatus:      charge:       spare:        spin:         vsx:         vsy:         vsz:         vex:         vey:         vez:         psx:         psy:         psz:         pex:         pey:         pez:        mass:        time:  properTime:" << '\n';
 
  for (size_t i = 0; i < v.size(); ++i) {
    o << std::scientific << std::showpos << std::setw(12) << v[i].id() << " "
      << std::setw(12) << v[i].ID() << " "
      << std::setw(12) << v[i].g4Parent() << " "
      << std::setw(12) << v[i].reason() << " "
      << std::setw(12) << v[i].mask() << " "
      << std::setw(12) << v[i].steps() << " "
      << std::setw(12) << v[i].secondaries() << " "
      << std::setw(12) << v[i].pdgID() << " "
      << std::setw(12) << v[i].status() << " "
      << std::setw(12) << v[i].genStatus() << " "
      << std::setw(12) << v[i].charge() << " "
      << std::setw(12) << v[i].vsx() << " "
      << std::setw(12) << v[i].vsy() << " "
      << std::setw(12) << v[i].vsz() << " "
      << std::setw(12) << v[i].vex() << " "
      << std::setw(12) << v[i].vey() << " "
      << std::setw(12) << v[i].vez() << " "
      << std::setw(12) << v[i].psx() << " "
      << std::setw(12) << v[i].psy() << " "
      << std::setw(12) << v[i].psz() << " "
      << std::setw(12) << v[i].pex() << " "
      << std::setw(12) << v[i].pey() << " "
      << std::setw(12) << v[i].pez() << " "
      << std::setw(12) << v[i].mass() << " "
      << std::setw(12) << v[i].time() << " "
      << std::setw(12) << v[i].properTime() << " "
      << std::endl;
 
 
 
    o << "      parents : ";
    for (unsigned j = 0, N = v[i].parents_size(); j < N; ++j) {
      o << v[i].parents(j) << " ";
    }
    o << std::endl;
    o << "      daughters : ";
    for (unsigned j = 0, N = v[i].daughters_size(); j < N; ++j) {
      o << v[i].daughters(j) << " ";
    }
    o << std::endl;
 
  }
 
  o.flags(old_flags);
  return o;
}
 
} // namespace dd4pod