MCWireCollection.h

Go to the documentation of this file.

#ifndef MCWIRECOLLECTION_H
#define MCWIRECOLLECTION_H

// C++ includes
#include "lardataobj/MCBase/MCWire.h"

namespace sim {

  class MCWireCollection : public std::vector<sim::MCWire> {

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

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

  private:
    unsigned int fChannel;

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

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

    inline void push_back(const MCWire& wire)
    {

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

      std::vector<sim::MCWire>::push_back(wire);

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

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

#endif