MCParticleTrajectory.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/RelationsObject.h>
#include <simulation/dataobjects/MCTrajectoryPoint.h>
#include <vector>
 
namespace Belle2 {
  class MCParticleTrajectory: public RelationsObject {
  public:
    MCParticleTrajectory(): RelationsObject(), m_points() {}
 
    typedef std::vector<MCTrajectoryPoint>::iterator iterator;
    typedef std::vector<MCTrajectoryPoint>::const_iterator const_iterator;
    iterator begin() { return m_points.begin(); }
    iterator end() { return m_points.end(); }
    const_iterator begin() const { return m_points.begin(); }
    const_iterator end() const { return m_points.end(); }
    size_t size() const { return m_points.size(); }
    bool empty() const { return m_points.empty(); }
    const MCTrajectoryPoint& operator[](size_t index) const { return m_points[index]; }
    const MCTrajectoryPoint& front() const { return m_points.front(); }
    const MCTrajectoryPoint& back() const { return m_points.back(); }
    void addPoint(float x, float y, float z, float px, float py, float pz)
    {
      //Reserve a decent amount of space to avoid lots of relocations
      m_points.emplace_back(x, y, z, px, py, pz);
    }
 
    void simplify(float distanceTolerance);
 
  private:
    std::vector<MCTrajectoryPoint> m_points;
 
    ClassDef(MCParticleTrajectory, 1);
  };
 
} //Belle2 namespace