MCHitCollection.h

Go to the documentation of this file.

#ifndef MCHITCOLLECTION_H
#define MCHITCOLLECTION_H

// C++ includes
#include "lardataobj/MCBase/MCHit.h"
#include <vector>

namespace sim {

  class MCHitCollection : public std::vector<sim::MCHit> {

  public:
    MCHitCollection(const unsigned int ch = ::sim::kINVALID_UINT)
    {
      Reset();
      fChannel = ch;
    }

    void Reset()
    {
      fChannel = ::sim::kINVALID_UINT;
      std::vector<sim::MCHit>::clear();
    }

  private:
    unsigned int fChannel; 

  public:
    unsigned int Channel() const { return fChannel; }

    inline bool operator<(const MCHitCollection& rhs) const { return fChannel < rhs.fChannel; }

    inline void push_back(const MCHit& hit)
    {

      bool sort = (!empty() && hit < (*rbegin()));

      std::vector<sim::MCHit>::push_back(hit);

      if (sort) std::sort(begin(), end());
    }
  };
}

// Define a pointer comparison
namespace std {
  template <>
  class less<sim::MCHitCollection*> {
  public:
    bool operator()(const sim::MCHitCollection* lhs, const sim::MCHitCollection* rhs)
    {
      return (*lhs) < (*rhs);
    }
  };
}

#endif