ShowerRecoTools::ShowerTrackFinderCheater Class Reference

Inheritance diagram for ShowerRecoTools::ShowerTrackFinderCheater:

Public Member Functions
	ShowerTrackFinderCheater (const fhicl::ParameterSet &pset)
int	CalculateElement (const art::Ptr< recob::PFParticle > &pfparticle, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder) override

Private Member Functions
int	RunShowerTool (const art::Ptr< recob::PFParticle > &pfparticle, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder, std::string evd_display_name_append="")
virtual void	InitialiseProducers ()
void	SetPtr (art::ProducesCollector *collector)
void	InitaliseProducerPtr (reco::shower::ShowerProducedPtrsHolder &uniqueproducerPtrs)
virtual int	AddAssociations (const art::Ptr< recob::PFParticle > &pfpPtr, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder)
const shower::LArPandoraShowerAlg &	GetLArPandoraShowerAlg () const
template<class T >
art::Ptr< T >	GetProducedElementPtr (std::string Name, reco::shower::ShowerElementHolder &ShowerEleHolder, int iter=-1)
template<class T >
void	InitialiseProduct (std::string Name, std::string InstanceName="")
template<class T , class A , class B >
void	AddSingle (A &a, B &b, std::string Name)
int	GetVectorPtrSize (std::string Name)
void	PrintPtrs ()
void	PrintPtr (std::string Name)

Private Attributes
shower::LArPandoraShowerCheatingAlg	fLArPandoraShowerCheatingAlg
const bool	fDebugEVD
const art::InputTag	fPFParticleLabel
const art::InputTag	fHitModuleLabel
const std::string	fTrueParticleIntputLabel
const std::string	fShowerStartPositionInputTag
const std::string	fShowerDirectionInputTag
const std::string	fInitialTrackHitsOutputLabel
const std::string	fInitialTrackSpacePointsOutputLabel

Detailed Description

Definition at line 21 of file ShowerTrackFinderCheater_tool.cc.

Constructor & Destructor Documentation

ShowerRecoTools::ShowerTrackFinderCheater::ShowerTrackFinderCheater

(

const fhicl::ParameterSet &

pset

)

Definition at line 47 of file ShowerTrackFinderCheater_tool.cc.

    : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
    , fLArPandoraShowerCheatingAlg(pset.get<fhicl::ParameterSet>("LArPandoraShowerCheatingAlg"))
    , fDebugEVD(pset.get<bool>("DebugEVD"))
    , fPFParticleLabel(pset.get<art::InputTag>("PFParticleLabel"))
    , fHitModuleLabel(pset.get<art::InputTag>("HitModuleLabel"))
    , fTrueParticleIntputLabel(pset.get<std::string>("TrueParticleIntputLabel"))
    , fShowerStartPositionInputTag(pset.get<std::string>("ShowerStartPositionInputTag"))
    , fShowerDirectionInputTag(pset.get<std::string>("ShowerDirectionInputTag"))
    , fInitialTrackHitsOutputLabel(pset.get<std::string>("InitialTrackHitsOutputLabel"))
    , fInitialTrackSpacePointsOutputLabel(
        pset.get<std::string>("InitialTrackSpacePointsOutputLabel"))
  {}

Member Function Documentation

virtual int ShowerRecoTools::IShowerTool::AddAssociations ( const art::Ptr< recob::PFParticle > &  pfpPtr, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder  )

inlinevirtualinherited

Reimplemented in ShowerRecoTools::ShowerPCADirection, ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder, ShowerRecoTools::ShowerExampleTool, and ShowerRecoTools::ShowerSkeletonTool.

Definition at line 72 of file IShowerTool.h.

    {
      return 0;
    }

template<class T , class A , class B >

void ShowerRecoTools::IShowerTool::AddSingle ( A &  a, B &  b, std::string  Name  )

inlineprotectedinherited

Definition at line 150 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::AddSingle().

    {
      UniquePtrs->AddSingle<T>(a, b, Name);
    }

int ShowerRecoTools::ShowerTrackFinderCheater::CalculateElement ( const art::Ptr< recob::PFParticle > &  pfparticle, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder  )

overridevirtual

Implements ShowerRecoTools::IShowerTool.

Definition at line 61 of file ShowerTrackFinderCheater_tool.cc.

References shower::LArPandoraShowerCheatingAlg::CheatDebugEVD(), reco::shower::ShowerElementHolder::CheckElement(), simb::MCParticle::Daughter(), DEFINE_ART_CLASS_TOOL, fDebugEVD, fHitModuleLabel, fInitialTrackHitsOutputLabel, fInitialTrackSpacePointsOutputLabel, fLArPandoraShowerCheatingAlg, fPFParticleLabel, fShowerDirectionInputTag, fShowerStartPositionInputTag, fTrueParticleIntputLabel, reco::shower::ShowerElementHolder::GetElement(), shower::LArPandoraShowerCheatingAlg::GetTrueChain(), shower::LArPandoraShowerCheatingAlg::GetTrueParticleMap(), art::ProductRetriever::getValidHandle(), hits(), art::Ptr< T >::key(), simb::MCParticle::NumberDaughters(), simb::MCParticle::PdgCode(), reco::shower::ShowerElementHolder::SetElement(), simb::MCParticle::TrackId(), shower::LArPandoraShowerCheatingAlg::TrueParticleID(), and shower::LArPandoraShowerCheatingAlg::TrueParticleIDFromTrueChain().

  {
    const simb::MCParticle* trueParticle = nullptr;

    //Get the hits from the shower:
    auto const pfpHandle = Event.getValidHandle<std::vector<recob::PFParticle>>(fPFParticleLabel);
    auto const clockData =
      art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(Event);

    //Get the clusters
    auto const clusHandle = Event.getValidHandle<std::vector<recob::Cluster>>(fPFParticleLabel);
    art::FindManyP<recob::Cluster> fmc(pfpHandle, Event, fPFParticleLabel);
    std::vector<art::Ptr<recob::Cluster>> clusters = fmc.at(pfparticle.key());

    //Get the hit association
    art::FindManyP<recob::Hit> fmhc(clusHandle, Event, fPFParticleLabel);

    std::vector<art::Ptr<recob::Hit>> showerHits;
    for (auto const& cluster : clusters) {

      //Get the hits
      std::vector<art::Ptr<recob::Hit>> hits = fmhc.at(cluster.key());
      showerHits.insert(showerHits.end(), hits.begin(), hits.end());
    }

    if (ShowerEleHolder.CheckElement(fTrueParticleIntputLabel)) {
      ShowerEleHolder.GetElement(fTrueParticleIntputLabel, trueParticle);
    }
    else {

      //Could store these in the shower element holder and just calculate once?
      std::map<int, const simb::MCParticle*> trueParticles =
        fLArPandoraShowerCheatingAlg.GetTrueParticleMap();
      std::map<int, std::vector<int>> showersMothers =
        fLArPandoraShowerCheatingAlg.GetTrueChain(trueParticles);

      //Get the true particle from the shower
      std::pair<int, double> ShowerTrackInfo =
        fLArPandoraShowerCheatingAlg.TrueParticleIDFromTrueChain(
          clockData, showersMothers, showerHits, 2);

      if (ShowerTrackInfo.first == -99999) {
        mf::LogError("ShowerStartPosition") << "True Shower Not Found";
        return 1;
      }
      trueParticle = trueParticles[ShowerTrackInfo.first];
      ShowerEleHolder.SetElement(trueParticle, fTrueParticleIntputLabel);
    }

    if (!trueParticle) {
      mf::LogError("ShowerDirectionCheater") << "True shower not found, returning";
      return 1;
    }

    //This is all based on the shower vertex being known. If it is not lets not do the track
    if (!ShowerEleHolder.CheckElement(fShowerStartPositionInputTag)) {
      mf::LogError("ShowerTrackFinderCheater") << "Start position not set, returning " << std::endl;
      return 1;
    }
    if (!ShowerEleHolder.CheckElement(fShowerDirectionInputTag)) {
      mf::LogError("ShowerTrackFinderCheater") << "Direction not set, returning " << std::endl;
      return 1;
    }

    geo::Point_t ShowerStartPosition = {-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerStartPositionInputTag, ShowerStartPosition);

    geo::Vector_t ShowerDirection = {-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerDirectionInputTag, ShowerDirection);

    auto const hitHandle = Event.getValidHandle<std::vector<recob::Hit>>(fHitModuleLabel);

    // Get the hits associated with the space points
    art::FindManyP<recob::SpacePoint> fmsph(hitHandle, Event, fPFParticleLabel);
    if (!fmsph.isValid()) {
      throw cet::exception("ShowerTrackFinderCheater")
        << "Spacepoint and hit association not valid. Stopping.";
    }

    std::vector<int> trueParticleIdVec;

    // If we have an electron, take the hits from the primary only
    // This will also cover the cases Pandora misclassify a track as a shower
    if (trueParticle->PdgCode() != 22) { trueParticleIdVec.push_back(trueParticle->TrackId()); }
    else {
      // To check if we are rolling up showers, check the number of daughters the photon has
      const int nDaughters = trueParticle->NumberDaughters();
      if (nDaughters == 0) {
        // If we roll up showers, we have no choice but to take all of the hits from the photon
        trueParticleIdVec.push_back(-trueParticle->TrackId());
      }
      else {
        // If we do not roll up the showers, take all of the primary daughters
        for (int i = 0; i < nDaughters; i++) {
          trueParticleIdVec.push_back(trueParticle->Daughter(i));
        }
      }
    }

    std::vector<art::Ptr<recob::Hit>> trackHits;
    std::vector<art::Ptr<recob::SpacePoint>> trackSpacePoints;

    //Get the hits from the true particle
    for (auto hit : showerHits) {
      int trueHitId = fLArPandoraShowerCheatingAlg.TrueParticleID(clockData, hit);
      if (std::find(trueParticleIdVec.cbegin(), trueParticleIdVec.cend(), trueHitId) !=
          trueParticleIdVec.cend()) {
        trackHits.push_back(hit);
        std::vector<art::Ptr<recob::SpacePoint>> sps = fmsph.at(hit.key());
        if (sps.size() == 1) { trackSpacePoints.push_back(sps.front()); }
      }
    }

    if (trackHits.empty() || trackSpacePoints.empty())
      mf::LogWarning("ShowerTrackFinderCheater")
        << "Creating intial track with " << trackHits.size() << " hits and "
        << trackSpacePoints.size() << " spacepoints" << std::endl;

    ShowerEleHolder.SetElement(trackHits, fInitialTrackHitsOutputLabel);
    ShowerEleHolder.SetElement(trackSpacePoints, fInitialTrackSpacePointsOutputLabel);

    if (fDebugEVD) {
      fLArPandoraShowerCheatingAlg.CheatDebugEVD(
        clockData, trueParticle, Event, ShowerEleHolder, pfparticle);
    }

    return 0;
  }

const shower::LArPandoraShowerAlg& ShowerRecoTools::IShowerTool::GetLArPandoraShowerAlg ( ) const

inlineprotectedinherited

Definition at line 80 of file IShowerTool.h.

Referenced by ShowerRecoTools::ShowerLengthPercentile::CalculateElement(), ShowerRecoTools::Shower3DCylinderTrackHitFinder::CalculateElement(), ShowerRecoTools::ShowerPCAPropergationStartPosition::CalculateElement(), ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::CalculateElement(), ShowerRecoTools::ShowerPFPVertexStartPosition::CalculateElement(), ShowerRecoTools::ShowerUnidirectiondEdx::CalculateElement(), ShowerRecoTools::ShowerDirectionCheater::CalculateElement(), ShowerRecoTools::ShowerPCADirection::CalculateElement(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::CalculateElement(), ShowerRecoTools::ShowerTrajPointdEdx::CalculateElement(), ShowerRecoTools::ShowerPCADirection::CalculateShowerPCA(), ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::FindInitialTrackHits(), ShowerRecoTools::Shower3DCylinderTrackHitFinder::FindTrackSpacePoints(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::FitSegmentAndCalculateResidual(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::PruneFrontOfSPSPool(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::PruneTrack(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::RunTestOfIncrementalSpacePointFinder(), ShowerRecoTools::ShowerTrackPCADirection::ShowerPCAVector(), and ShowerRecoTools::ShowerIncrementalTrackHitFinder::ShowerPCAVector().

    {
      return fLArPandoraShowerAlg;
    };

template<class T >

art::Ptr<T> ShowerRecoTools::IShowerTool::GetProducedElementPtr ( std::string  Name, reco::shower::ShowerElementHolder &  ShowerEleHolder, int  iter = -1  )

inlineprotectedinherited

Definition at line 100 of file IShowerTool.h.

References reco::shower::ShowerElementHolder::CheckElement(), reco::shower::ShowerProducedPtrsHolder::CheckUniqueProduerPtr(), reco::shower::ShowerProducedPtrsHolder::GetArtPtr(), and reco::shower::ShowerElementHolder::GetShowerNumber().

    {

      //Check the element has been set
      bool check_element = ShowerEleHolder.CheckElement(Name);
      if (!check_element) {
        throw cet::exception("IShowerTool") << "tried to get a element that does not exist. Failed "
                                               "at making the art ptr for Element: "
                                            << Name << std::endl;
      }

      //Check the unique ptr has been set.
      bool check_ptr = UniquePtrs->CheckUniqueProduerPtr(Name);
      if (!check_ptr) {
        throw cet::exception("IShowerTool")
          << "tried to get a ptr that does not exist. Failed at making the art ptr for Element"
          << Name;
      }

      //Check if the user has defined an index if not just use the current shower index/
      int index;
      if (iter != -1) { index = iter; }
      else {
        index = ShowerEleHolder.GetShowerNumber();
      }

      //Make the ptr
      return UniquePtrs->GetArtPtr<T>(Name, index);
    }

int ShowerRecoTools::IShowerTool::GetVectorPtrSize ( std::string  Name )

inlineprotectedinherited

Definition at line 156 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::GetVectorPtrSize().

Referenced by ShowerRecoTools::ShowerExampleTool::AddAssociations(), ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder::AddAssociations(), and ShowerRecoTools::ShowerPCADirection::AddAssociations().

{ return UniquePtrs->GetVectorPtrSize(Name); }

void ShowerRecoTools::IShowerTool::InitaliseProducerPtr ( reco::shower::ShowerProducedPtrsHolder &  uniqueproducerPtrs )

inlineinherited

Definition at line 66 of file IShowerTool.h.

    {
      UniquePtrs = &uniqueproducerPtrs;
    }

virtual void ShowerRecoTools::IShowerTool::InitialiseProducers ( )

inlinevirtualinherited

Reimplemented in ShowerRecoTools::ShowerPCADirection, ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder, and ShowerRecoTools::ShowerExampleTool.

Definition at line 60 of file IShowerTool.h.

{}

template<class T >

void ShowerRecoTools::IShowerTool::InitialiseProduct ( std::string  Name, std::string  InstanceName = ""  )

inlineprotectedinherited

Definition at line 135 of file IShowerTool.h.

References art::ProducesCollector::produces(), and reco::shower::ShowerProducedPtrsHolder::SetShowerUniqueProduerPtr().

    {

      if (collectorPtr == nullptr) {
        mf::LogWarning("IShowerTool") << "The art::ProducesCollector ptr has not been set";
        return;
      }

      collectorPtr->produces<T>(InstanceName);
      UniquePtrs->SetShowerUniqueProduerPtr(type<T>(), Name, InstanceName);
    }

void ShowerRecoTools::IShowerTool::PrintPtr ( std::string  Name )

inlineprotectedinherited

Definition at line 160 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::PrintPtr().

Referenced by ShowerRecoTools::ShowerExampleTool::CalculateElement().

{ UniquePtrs->PrintPtr(Name); }

void ShowerRecoTools::IShowerTool::PrintPtrs ( )

inlineprotectedinherited

Definition at line 158 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::PrintPtrs().

Referenced by ShowerRecoTools::ShowerExampleTool::CalculateElement().

{ UniquePtrs->PrintPtrs(); }

int ShowerRecoTools::IShowerTool::RunShowerTool ( const art::Ptr< recob::PFParticle > &  pfparticle, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder, std::string  evd_display_name_append = ""  )

inlineinherited

Definition at line 44 of file IShowerTool.h.

    {

      int calculation_status = CalculateElement(pfparticle, Event, ShowerEleHolder);
      if (calculation_status != 0) return calculation_status;
      if (fRunEventDisplay) {
        IShowerTool::GetLArPandoraShowerAlg().DebugEVD(
          pfparticle, Event, ShowerEleHolder, evd_display_name_append);
      }
      return calculation_status;
    }

void ShowerRecoTools::IShowerTool::SetPtr ( art::ProducesCollector *  collector )

inlineinherited

Definition at line 63 of file IShowerTool.h.

{ collectorPtr = collector; }

Member Data Documentation

const bool ShowerRecoTools::ShowerTrackFinderCheater::fDebugEVD

private

Definition at line 36 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const art::InputTag ShowerRecoTools::ShowerTrackFinderCheater::fHitModuleLabel

private

Definition at line 38 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const std::string ShowerRecoTools::ShowerTrackFinderCheater::fInitialTrackHitsOutputLabel

private

Definition at line 43 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const std::string ShowerRecoTools::ShowerTrackFinderCheater::fInitialTrackSpacePointsOutputLabel

private

Definition at line 44 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

shower::LArPandoraShowerCheatingAlg ShowerRecoTools::ShowerTrackFinderCheater::fLArPandoraShowerCheatingAlg

private

Definition at line 33 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const art::InputTag ShowerRecoTools::ShowerTrackFinderCheater::fPFParticleLabel

private

Definition at line 37 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const std::string ShowerRecoTools::ShowerTrackFinderCheater::fShowerDirectionInputTag

private

Definition at line 42 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const std::string ShowerRecoTools::ShowerTrackFinderCheater::fShowerStartPositionInputTag

private

Definition at line 41 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

const std::string ShowerRecoTools::ShowerTrackFinderCheater::fTrueParticleIntputLabel

private

Definition at line 40 of file ShowerTrackFinderCheater_tool.cc.

Referenced by CalculateElement().

---

The documentation for this class was generated from the following file:

• larpandora/v10_00_14/source/larpandora/LArPandoraEventBuilding/LArPandoraShower/Tools/Cheating/ShowerTrackFinderCheater_tool.cc