TrackerHitCollection.cc

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#include "dd4pod/TrackerHitCollection.h"
 
 
// standard includes
#include <stdexcept>
#include <iomanip>
 
namespace dd4pod {
 
 
TrackerHitCollection::TrackerHitCollection() :
 m_isValid(false), m_isReadFromFile(false), m_collectionID(0), m_entries(),
 m_data(new TrackerHitDataContainer()) {
}
 
TrackerHitCollection::~TrackerHitCollection() {
 clear();
 if (m_data) delete m_data;
}
 
const TrackerHit TrackerHitCollection::operator[](unsigned int index) const {
 return TrackerHit(m_entries[index]);
}
 
const TrackerHit TrackerHitCollection::at(unsigned int index) const {
 return TrackerHit(m_entries.at(index));
}
 
TrackerHit TrackerHitCollection::operator[](unsigned int index) {
 return TrackerHit(m_entries[index]);
}
 
TrackerHit TrackerHitCollection::at(unsigned int index) {
 return TrackerHit(m_entries.at(index));
}
 
size_t TrackerHitCollection::size() const {
 return m_entries.size();
}
 
TrackerHit TrackerHitCollection::create() {
 auto obj = m_entries.emplace_back(new TrackerHitObj());
 
 obj->id = {int(m_entries.size() - 1), m_collectionID};
 return TrackerHit(obj);
}
 
void TrackerHitCollection::clear() {
 m_data->clear();
 for (auto& obj : m_entries) { delete obj; }
 m_entries.clear();
}
 
void TrackerHitCollection::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(); }
 
 
}
 
void TrackerHitCollection::prepareAfterRead() {
 int index = 0;
 for (auto& data : *m_data) {
 auto obj = new TrackerHitObj({index, m_collectionID}, data);
 
 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 TrackerHitCollection::setReferences(const podio::ICollectionProvider* collectionProvider) {
 
 return true; //TODO: check success
}
 
void TrackerHitCollection::push_back(ConstTrackerHit 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);
 
 } else {
 throw std::invalid_argument("Object already in a collection. Cannot add it to a second collection");
 }
}
 
void TrackerHitCollection::setBuffer(void* address) {
 if (m_data) delete m_data;
 m_data = static_cast<TrackerHitDataContainer*>(address);
}
 
const TrackerHit TrackerHitCollectionIterator::operator*() const {
 m_object.m_obj = (*m_collection)[m_index];
 return m_object;
}
 
const TrackerHit* TrackerHitCollectionIterator::operator->() const {
 m_object.m_obj = (*m_collection)[m_index];
 return &m_object;
}
 
const TrackerHitCollectionIterator& TrackerHitCollectionIterator::operator++() const {
 ++m_index;
 return *this;
}
 
std::ostream& operator<<(std::ostream& o, const TrackerHitCollection& v) {
 const auto old_flags = o.flags();
 o << " id: cellID: flag: g4ID: position [ x, y, z, t]: momentum [ x, y, z, t]: length: truth [ trackID, pdgID, deposit, time, length, x, y, z]:energyDeposit:" << '\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].cellID() << " "
 << std::setw(12) << v[i].flag() << " "
 << std::setw(12) << v[i].g4ID() << " "
 << std::setw(12) << v[i].position() << " "
 << std::setw(12) << v[i].momentum() << " "
 << std::setw(12) << v[i].length() << " "
 << std::setw(12) << v[i].truth() << " "
 << std::setw(12) << v[i].energyDeposit() << " "
 << std::endl;
 
 
 
 
 }
 
 o.flags(old_flags);
 return o;
}
 
} // namespace dd4pod