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