ParticleInventory.cc

Go to the documentation of this file.

//
// ParticleInventory.cc
// Author: JStock
// EMail:  jason.stock@mines.sdsmt.edu
// 2017-09-12
//

//STL includes
//ROOT includes
//Framework includes
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"
//LArSoft includes
#include "larsim/MCCheater/ParticleInventory.h"
#include "nusimdata/SimulationBase/MCParticle.h"

namespace cheat {

  ParticleInventory::ParticleInventory(const ParticleInventoryConfig& config)
    : fG4ModuleLabel(config.G4ModuleLabel())
    , fEveIdCalculator(config.EveIdCalculator())
    , fOverrideRealData(config.OverrideRealData())
  {}

  //----------------------------------------------------------------------
  ParticleInventory::ParticleInventory(const fhicl::ParameterSet& pSet)
    : fG4ModuleLabel(pSet.get<art::InputTag>("G4ModuleLabel", "largeant"))
    , fEveIdCalculator(pSet.get<std::string>("EveIdCalculator", "EmEveIdCalculator"))
    , fOverrideRealData(pSet.get<bool>("OverrideRealData", false))
  {}

  //-----------------------------------------------------------------------
  void ParticleInventory::ClearEvent()
  {
    fParticleList.clear();
    fMCTObj.fMCTruthList.clear();
    fMCTObj.fTrackIdToMCTruthIndex.clear();
  }

  //deliverables

  //-----------------------------------------------------------------------
  //TrackIdToParticlePtr
  const simb::MCParticle* ParticleInventory::TrackIdToParticle_P(int const& id) const
  {
    sim::ParticleList::const_iterator part_it = fParticleList.find(id);
    if (part_it == fParticleList.end()) {
      mf::LogWarning("ParticleInventory")
        << "Particle with TrackId: " << id << " not found in inventory. "
        << "Returning null pointer.";
      return 0;
    }
    return part_it->second;
  } //End TrackIdToParticle

  //-----------------------------------------------------------------------
  const simb::MCParticle* ParticleInventory::TrackIdToMotherParticle_P(int const& id) const
  {
    return this->TrackIdToParticle_P(fParticleList.EveId(abs(id)));
  }

  //-----------------------------------------------------------------------
  const art::Ptr<simb::MCTruth>& ParticleInventory::TrackIdToMCTruth_P(int const& id) const
  {
    // find the entry in the MCTruth collection for this track id
    auto mctItr = fMCTObj.fTrackIdToMCTruthIndex.find(abs(id));
    if (mctItr != fMCTObj.fTrackIdToMCTruthIndex.end()) {
      int partIndex = mctItr->second;
      return fMCTObj.fMCTruthList.at(partIndex);
    }
    else {
      throw cet::exception("ParticleInventory")
        << "Attempt to find MCTruth for TrackId: " << id << " has failed.";
    }
  }

  //-----------------------------------------------------------------------
  const art::Ptr<simb::MCTruth>& ParticleInventory::ParticleToMCTruth_P(
    const simb::MCParticle* p) const
  {
    return this->TrackIdToMCTruth_P(p->TrackId());
  }

  //-----------------------------------------------------------------------
  const std::vector<art::Ptr<simb::MCTruth>>& ParticleInventory::MCTruthVector_Ps() const
  {
    return fMCTObj.fMCTruthList;
  }

  //-----------------------------------------------------------------------
  std::vector<const simb::MCParticle*> ParticleInventory::MCTruthToParticles_Ps(
    art::Ptr<simb::MCTruth> const& mct) const
  {
    std::vector<const simb::MCParticle*> ret;
    // sim::ParticleList::value_type is a pair (track Id, particle pointer)
    for (const sim::ParticleList::value_type& TrackIdpair : fParticleList) {
      if (this->TrackIdToMCTruth_P(TrackIdpair.first) == mct) ret.push_back(TrackIdpair.second);
    }
    return ret;
  }

  //-----------------------------------------------------------------------
  std::set<int> ParticleInventory::GetSetOfTrackIds() const
  {
    std::set<int> ret;
    for (auto partItr = fParticleList.begin(); partItr != fParticleList.end(); ++partItr) {
      ret.emplace((partItr->second)->TrackId());
    }
    return ret;
  }

  //-----------------------------------------------------------------------
  std::set<int> ParticleInventory::GetSetOfEveIds() const
  {
    std::set<int> ret;
    std::set<int> tIds = this->GetSetOfTrackIds();
    for (auto tId : tIds) {
      ret.emplace(fParticleList.EveId(tId));
    }
    return ret;
  }

} //namespace