ParticleInventoryService.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////////
//
// ParticleInventoryService.cc
// Author: JStock
// EMail:  jason.stock@mines.sdsmt.edu
// 2017-09-12
//
// Maintinence Notes: When the ParticleInventory is initialized, none of the prep work (previously
// the BackTracker rebuild stage) will be done. Each function needs to check and make sure the
// needed data products have been loaded. To see what objects a function uses, you will have to
// check the appropriate part of ParticleInventory. After this, you will need to manually write the check
// into whatever function you are writing. You will also want to include a call to prepare the needed items
// if your check fails.
//
// Example:
// std::set<int> ParticleInventoryService::GetSetOfTrackIds(){
//   if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();} //The GetTrackIds in ParticleInventory needs the ParticleList.
//                                                                         So, we check if it's ready, and if it isn't we ready it.
//   return ParticleInventory::GetSetOfTrackIds();
// }
//
// If you have any questions about how to incorperate something in here, let me know. I know this is a rather odd
// use model. The rationale is to allow the BackTracker service to be lazy, while at the same time allowing gallery
// to use backtracker functions (the gallery implimentation is not lazy).
////////////////////////////////////////////////////////////////////////////

#include "larsim/MCCheater/ParticleInventoryService.h"

#include "nug4/ParticleNavigation/ParticleList.h"

#include "art/Framework/Principal/Event.h"
#include "art/Framework/Services/Registry/ActivityRegistry.h"
#include "messagefacility/MessageLogger/MessageLogger.h"
#include "fhiclcpp/ParameterSet.h"

namespace cheat{

  //----------------------------------------------------------------------
  ParticleInventoryService::ParticleInventoryService(const ParticleInventoryServiceConfig& config, art::ActivityRegistry& reg)
  :ParticleInventory(config.ParticleInventoryTable())
  {
//    std::cout<<"Config Dump from ParticleInventoryService using fhicl Table\n";
//    config.ParticleInventoryTable.print_allowed_configuration(std::cout);
    reg.sPreProcessEvent.watch(this, &ParticleInventoryService::priv_PrepEvent);
  }

  //----------------------------------------------------------------------
  ParticleInventoryService::ParticleInventoryService(const fhicl::ParameterSet& pSet, art::ActivityRegistry& reg)
  :ParticleInventory(pSet.get<fhicl::ParameterSet>("ParticleInventory"))
  {
//    std::cout<<"\n\n\n\nConfigDump from ParticleInventoryService using ParameterSet.\n"<<pSet.to_string()<<"\n\n\n\n";
    reg.sPreProcessEvent.watch(this, &ParticleInventoryService::priv_PrepEvent);
  }

  //_--Temporary Rebuild function (until the service can be lazy again).
  void ParticleInventoryService::Rebuild( const art::Event& evt){
    this->priv_PrepEvent(evt, art::ScheduleContext::invalid());
  }

  //----------------------------------------------------------------------
  void ParticleInventoryService::priv_PrepEvent(const art::Event& evt, art::ScheduleContext){
    //fEvt=&evt;
    ParticleInventory::ClearEvent();
    if( ! this->priv_CanRun(evt) ) { return; }
    this->priv_PrepParticleList(evt);
    this->priv_PrepMCTruthList(evt);
    this->priv_PrepTrackIdToMCTruthIndex(evt);
    //fEvt=nullptr; //dont keep the cached pointer since it will expire right after this, and I want to make sure bad calls to prep functions fail.
  }

  //----------------------------------------------------------------------
  bool ParticleInventoryService::priv_CanRun(const art::Event& evt) const{
    return ParticleInventory::CanRun(evt);
  }

  //----------------------------------------------------------------------
  void ParticleInventoryService::priv_PrepParticleList(const art::Event& evt){
    if(!this->priv_CanRun(evt)) {throw;}
    //if(!this->priv_CanRun(*fEvt)) {throw;}
    if(this->priv_ParticleListReady()){ return; }
    //try{ParticleInventory::PrepParticleList(*fEvt);}
    try{ParticleInventory::PrepParticleList(evt);}
    catch(...){ mf::LogWarning("ParticleInventory") << "Rebuild failed to get the MCParticles. This is expected when running on a generation or simulation step.";}
  }


  void ParticleInventoryService::priv_PrepTrackIdToMCTruthIndex(const art::Event& evt ){
    if(!this->priv_CanRun(evt)){throw;}
    //if(!this->priv_CanRun(*fEvt)){throw;}
    if( this->priv_TrackIdToMCTruthReady()){ return; }
    //try{ParticleInventory::PrepTrackIdToMCTruthIndex(*fEvt);}
    try{ParticleInventory::PrepTrackIdToMCTruthIndex(evt);}
    catch(...){ mf::LogWarning("ParticleInventory") << "Rebuild failed to get the MCParticles. This is expected when running on a generation or simulation step.";}
  }//End priv_PrepTrackIdToMCTruthIndexList

  void ParticleInventoryService::priv_PrepMCTruthList(const art::Event& evt ){
//    if(!this->priv_CanRun(*fEvt)){throw;}
    if(!this->priv_CanRun(evt)){throw;}
    if(this->priv_MCTruthListReady( ) ){ return;} //If the event is data or if the truth list is already built there is nothing for us to do.
    try{    ParticleInventory::PrepMCTruthList(evt); }
    //try{    ParticleInventory::PrepMCTruthList(*fEvt); }
    catch(...){ mf::LogWarning("ParticleInventory") << "Rebuild failed to get the MCParticles. This is expected when running on a generation or simulation step.";}
    //ToDo. Find out exactly which exception is thrown and catch only that.

  }//End PrepMCTruthList


  //Loop Event and grab MCTruths. Quick and clean as possible.

  //deliverables

  const sim::ParticleList& ParticleInventoryService::ParticleList() const {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    Not used for non lazy functions
    return ParticleInventory::ParticleList();
  } //This should be replaced with a public struct so we can get away from the nutools dependency.

  const std::vector< art::Ptr<simb::MCTruth> >& ParticleInventoryService::MCTruthVector_Ps() const {
    //if(!this->priv_MCTruthListReady()){priv_PrepMCTruthList();}
    // Not used for non-lazy mode
    return ParticleInventory::MCTruthVector_Ps();
  }

  //TrackIdToParticleP

  const simb::MCParticle* ParticleInventoryService::TrackIdToParticle_P(int const id) const {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    Not used for non-lazy mode
    return ParticleInventory::TrackIdToParticle_P(id);
  }//End TrackIdToParticle


  const simb::MCParticle* ParticleInventoryService::TrackIdToMotherParticle_P(int const id) const
  {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    Not used for non-lazy mode
    return ParticleInventory::TrackIdToMotherParticle_P(id);
  }

  const art::Ptr<simb::MCTruth>& ParticleInventoryService::TrackIdToMCTruth_P(int const id) const
  {
//    if(!this->priv_TrackIdToMCTruthReady()){this->priv_PrepTrackIdToMCTruthIndex();}
//    Not used for non-lazy mode
    return ParticleInventory::TrackIdToMCTruth_P(id);
  }

  int ParticleInventoryService::TrackIdToEveTrackId(const int tid) const
  {
    return ParticleInventory::TrackIdToEveTrackId(tid);
  }

  const art::Ptr<simb::MCTruth>& ParticleInventoryService::ParticleToMCTruth_P(const simb::MCParticle* p) const
  {
//    if(!this->priv_TrackIdToMCTruthReady()){this->priv_PrepTrackIdToMCTruthIndex();}
//    Not used for non-lazy mode
    return this->TrackIdToMCTruth_P(p->TrackId());
  }

  std::vector<const simb::MCParticle*> ParticleInventoryService::MCTruthToParticles_Ps(art::Ptr<simb::MCTruth> const& mct) const
  {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    if(!this->priv_MCTruthListReady()){this->priv_PrepMCTruthList();}
//    Not used for non-lazy mode
    return ParticleInventory::MCTruthToParticles_Ps(mct);
  }

  std::set<int> ParticleInventoryService::GetSetOfTrackIds() const {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    Not used for non-lazy mode
    return ParticleInventory::GetSetOfTrackIds();
  }

  std::set<int> ParticleInventoryService::GetSetOfEveIds() const {
//    if(!this->priv_ParticleListReady()){this->priv_PrepParticleList();}
//    Not used for non-lazy mode
    return ParticleInventory::GetSetOfEveIds();
  }

} //namespace