MCShowerRecoPart.h

Go to the documentation of this file.

#ifndef MCSHOWERRECOPART_H
#define MCSHOWERRECOPART_H

// ART includes
namespace fhicl {
  class ParameterSet;
}
#include "cetlib_except/exception.h"

// LArSoft
namespace sim {
  class MCRecoPart;
}

// STL
#include <map>
#include <vector>

namespace sim {

  class MCShowerRecoPart {

  public:
    static const unsigned int kINVALID_UINT;
    static const int kINVALID_INT;

  public:
    explicit MCShowerRecoPart(fhicl::ParameterSet const& pset);
    //ClusterMergeAlg(fhicl::ParameterSet const& pset, art::ActivityRegistry& reg);

    void ConstructShower(const MCRecoPart& part_v);

    const std::vector<unsigned int>& ShowerDaughters(const unsigned int shower_id) const
    {
      if (shower_id >= _shower_daughters.size())
        throw cet::exception(__FUNCTION__) << "Invalid shower index!";
      return _shower_daughters.at(shower_id);
    }

    const std::vector<unsigned int> ShowerMothers() const
    {
      std::vector<unsigned int> mothers(_shower_index.size(), 0);
      for (auto mother_iter = _shower_index.begin(); mother_iter != _shower_index.end();
           ++mother_iter)
        mothers.at((*mother_iter).second) = (*mother_iter).first;
      return mothers;
    }

    //--------------- Particle Information Getters -----------------//

    int ShowerIndex(const unsigned int part_index) const
    {
      if (_shower_id.size() <= part_index) return kINVALID_INT;
      return _shower_id.at(part_index);
    }

  protected:
    bool _debug_mode;

    std::vector<int> _shower_id;

    //
    // shower-indexed-variables
    //
    std::map<unsigned int, unsigned int> _shower_index;

    std::vector<std::vector<unsigned int>> _shower_daughters;

  }; // class MCShowerRecoPart

} //namespace cluster
#endif