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