sbn::CRUMBS Class Reference

Inheritance diagram for sbn::CRUMBS:

Public Member Functions
	CRUMBS (fhicl::ParameterSet const &p)
	CRUMBS (CRUMBS const &)=delete
	CRUMBS (CRUMBS &&)=delete
CRUMBS &	operator= (CRUMBS const &)=delete
CRUMBS &	operator= (CRUMBS &&)=delete
void	produce (art::Event &e) override
void	InitialiseMVAReader (TMVA::Reader &mvaReader, std::string &mvaName, std::string &mvaFileName)
void	ResetVars ()
void	GetMaps (art::Event const &e, std::map< int, int > &trackIDToGenMap, std::map< int, std::string > &genTypeMap, std::map< int, int > &genCCNCMap, std::map< int, int > &genNuTypeMap)
art::Ptr< recob::PFParticle >	GetSlicePrimary (art::Event const &e, const art::Ptr< recob::Slice > &slice, const art::ValidHandle< std::vector< recob::Slice > > &handleSlices)
std::vector< art::Ptr< anab::T0 > >	GetCRTTrackT0s (art::Event const &e, const art::Ptr< recob::Slice > &slice, const art::ValidHandle< std::vector< recob::PFParticle > > &handlePFPs, const art::ValidHandle< std::vector< recob::Slice > > &handleSlices)
std::vector< art::Ptr< anab::T0 > >	GetCRTHitT0s (art::Event const &e, const art::Ptr< recob::Slice > &slice, const art::ValidHandle< std::vector< recob::PFParticle > > &handlePFPs, const art::ValidHandle< std::vector< recob::Slice > > &handleSlices)
float	GetLongestTrackStoppingChi2Ratio (art::Event const &e, const art::Ptr< recob::Slice > &slice, const art::ValidHandle< std::vector< recob::PFParticle > > &handlePFPs, const art::ValidHandle< std::vector< recob::Slice > > &handleSlices)
void	FillCRTVars (const std::vector< art::Ptr< anab::T0 > > &trackT0s, const std::vector< art::Ptr< anab::T0 > > &hitT0s)
void	FillPandoraNuScoreVars (std::map< std::string, float > &propertiesMap)
std::vector< art::Ptr < recob::Hit > >	GetAllSliceHits (art::Event const &e, const art::Ptr< recob::Slice > &slice, const art::ValidHandle< std::vector< recob::Slice > > &handleSlices)
void	GetTruthMatching (art::Event const &e, const std::vector< art::Ptr< recob::Hit > > &sliceHits, const std::vector< art::Ptr< recob::Hit > > &allHits, std::map< int, int > &trackIDToGenMap, int &matchedID, double &purity, double &completeness)
int	SliceTruthId (std::map< int, float > &purities)

Private Attributes
bool	fTrainingMode
bool	fEvaluateResultInTrainingMode
bool	fProcessNeutrinos
bool	fProcessCosmics
std::string	fMCParticleModuleLabel
std::string	fGeneratorModuleLabel
std::string	fCosmicModuleLabel
std::string	fPFParticleModuleLabel
std::string	fHitModuleLabel
std::string	fTrackModuleLabel
std::string	fSliceModuleLabel
std::string	fFlashMatchModuleLabel
std::string	fCRTTrackMatchModuleLabel
std::string	fCRTHitMatchModuleLabel
std::string	fCalorimetryModuleLabel
std::string	fMVAName
std::string	fMVAFileName
std::string	fCCNuMuMVAName
std::string	fCCNuMuMVAFileName
std::string	fCCNuEMVAName
std::string	fCCNuEMVAFileName
std::string	fNCMVAName
std::string	fNCMVAFileName
fhicl::ParameterSet	fChi2FitParams
TTree *	fSliceTree
TMVA::Reader	fMVAReader
TMVA::Reader	fCCNuMuMVAReader
TMVA::Reader	fCCNuEMVAReader
TMVA::Reader	fNCMVAReader
float	tpc_NuScore
unsigned	eventID
unsigned	subRunID
unsigned	runID
unsigned	slicePDG
int	ccnc
int	nutype
std::string	matchedType
double	matchedPurity
double	matchedCompleteness
sbn::TrackStoppingChi2Alg	fTrackStoppingChi2Alg
sbn::TPCGeoAlg	fTpcGeo
float	tpc_CRFracHitsInLongestTrack
float	tpc_CRLongestTrackDeflection
float	tpc_CRLongestTrackDirY
float	tpc_CRNHitsMax
float	tpc_NuEigenRatioInSphere
float	tpc_NuNFinalStatePfos
float	tpc_NuNHitsTotal
float	tpc_NuNSpacePointsInSphere
float	tpc_NuVertexY
float	tpc_NuWeightedDirZ
float	tpc_StoppingChi2CosmicRatio
float	pds_FMTotalScore
float	pds_FMPE
float	pds_FMTime
float	crt_TrackScore
float	crt_HitScore
float	crt_TrackTime
float	crt_HitTime

Detailed Description

Definition at line 51 of file CRUMBS_module.cc.

Constructor & Destructor Documentation

sbn::CRUMBS::CRUMBS ( fhicl::ParameterSet const &  p )

explicit

Definition at line 168 of file CRUMBS_module.cc.

    : EDProducer{p},
    fTrainingMode                 (p.get<bool>("TrainingMode",false)),
    fEvaluateResultInTrainingMode (p.get<bool>("EvaluateResultInTrainingMode",false)),
    fProcessNeutrinos             (p.get<bool>("ProcessNeutrinos",true)),
    fProcessCosmics               (p.get<bool>("ProcessCosmics",true)),
    fMCParticleModuleLabel        (p.get<std::string>("MCParticleModuleLabel","")),
    fGeneratorModuleLabel         (p.get<std::string>("GeneratorModuleLabel","")),
    fCosmicModuleLabel            (p.get<std::string>("CosmicModuleLabel","")),
    fPFParticleModuleLabel        (p.get<std::string>("PFParticleModuleLabel")),
    fHitModuleLabel               (p.get<std::string>("HitModuleLabel")),
    fTrackModuleLabel             (p.get<std::string>("TrackModuleLabel")),
    fSliceModuleLabel             (p.get<std::string>("SliceModuleLabel")),
    fFlashMatchModuleLabel        (p.get<std::string>("FlashMatchModuleLabel")),
    fCRTTrackMatchModuleLabel     (p.get<std::string>("CRTTrackMatchModuleLabel")),
    fCRTHitMatchModuleLabel       (p.get<std::string>("CRTHitMatchModuleLabel")),
    fCalorimetryModuleLabel       (p.get<std::string>("CalorimetryModuleLabel")),
    fMVAName                      (p.get<std::string>("MVAName")),
    fMVAFileName                  (p.get<std::string>("MVAFileName")),
    fCCNuMuMVAName                (p.get<std::string>("CCNuMuMVAName")),
    fCCNuMuMVAFileName            (p.get<std::string>("CCNuMuMVAFileName")),
    fCCNuEMVAName                 (p.get<std::string>("CCNuEMVAName")),
    fCCNuEMVAFileName             (p.get<std::string>("CCNuEMVAFileName")),
    fNCMVAName                    (p.get<std::string>("NCMVAName")),
    fNCMVAFileName                (p.get<std::string>("NCMVAFileName")),
    fChi2FitParams                (p.get<fhicl::ParameterSet>("Chi2FitParams")),
    fTrackStoppingChi2Alg(fChi2FitParams)
    {
      if(!fTrainingMode || fEvaluateResultInTrainingMode)
        {
          produces<std::vector<CRUMBSResult>>();
          produces<art::Assns<recob::Slice, CRUMBSResult>>();

          InitialiseMVAReader(fMVAReader, fMVAName, fMVAFileName);
          InitialiseMVAReader(fCCNuMuMVAReader, fCCNuMuMVAName, fCCNuMuMVAFileName);
          InitialiseMVAReader(fCCNuEMVAReader, fCCNuEMVAName, fCCNuEMVAFileName);
          InitialiseMVAReader(fNCMVAReader, fNCMVAName, fNCMVAFileName);
        }

      art::ServiceHandle<art::TFileService> tfs;
      if(fTrainingMode)
        {
          fSliceTree = tfs->make<TTree>("SliceTree","Slice data TTree");

          fSliceTree->Branch("tpc_NuScore",&tpc_NuScore);
          fSliceTree->Branch("tpc_CRFracHitsInLongestTrack",&tpc_CRFracHitsInLongestTrack);
          fSliceTree->Branch("tpc_CRLongestTrackDeflection",&tpc_CRLongestTrackDeflection);
          fSliceTree->Branch("tpc_CRLongestTrackDirY",&tpc_CRLongestTrackDirY);
          fSliceTree->Branch("tpc_CRNHitsMax",&tpc_CRNHitsMax);
          fSliceTree->Branch("tpc_NuEigenRatioInSphere",&tpc_NuEigenRatioInSphere);
          fSliceTree->Branch("tpc_NuNFinalStatePfos",&tpc_NuNFinalStatePfos);
          fSliceTree->Branch("tpc_NuNHitsTotal",&tpc_NuNHitsTotal);
          fSliceTree->Branch("tpc_NuNSpacePointsInSphere",&tpc_NuNSpacePointsInSphere);
          fSliceTree->Branch("tpc_NuVertexY",&tpc_NuVertexY);
          fSliceTree->Branch("tpc_NuWeightedDirZ",&tpc_NuWeightedDirZ);
          fSliceTree->Branch("tpc_StoppingChi2CosmicRatio",&tpc_StoppingChi2CosmicRatio);

          fSliceTree->Branch("pds_FMTotalScore",&pds_FMTotalScore);
          fSliceTree->Branch("pds_FMPE",&pds_FMPE);
          fSliceTree->Branch("pds_FMTime",&pds_FMTime);

          fSliceTree->Branch("crt_TrackScore",&crt_TrackScore);
          fSliceTree->Branch("crt_HitScore",&crt_HitScore);
          fSliceTree->Branch("crt_TrackTime",&crt_TrackTime);
          fSliceTree->Branch("crt_HitTime",&crt_HitTime);

          fSliceTree->Branch("eventID",&eventID);
          fSliceTree->Branch("subRunID",&subRunID);
          fSliceTree->Branch("runID",&runID);
          fSliceTree->Branch("slicePDG",&slicePDG);
          fSliceTree->Branch("matchedType",&matchedType);
          fSliceTree->Branch("matchedPurity",&matchedPurity);
          fSliceTree->Branch("matchedCompleteness",&matchedCompleteness);
          fSliceTree->Branch("ccnc",&ccnc);
          fSliceTree->Branch("nutype",&nutype);
        }
    }

sbn::CRUMBS::CRUMBS ( CRUMBS const &  )

delete

sbn::CRUMBS::CRUMBS ( CRUMBS &&  )

delete

Member Function Documentation

void sbn::CRUMBS::FillCRTVars	(	const std::vector< art::Ptr< anab::T0 > > &	trackT0s,
		const std::vector< art::Ptr< anab::T0 > > &	hitT0s
	)

Definition at line 459 of file CRUMBS_module.cc.

  {
    if (!trackT0s.empty()){
      crt_TrackScore = std::numeric_limits<float>::max();
      for(auto const crttrackmatcht0 : trackT0s)
        {
          if(crttrackmatcht0->TriggerConfidence() < crt_TrackScore)
            {
              crt_TrackScore = crttrackmatcht0->TriggerConfidence();
              crt_TrackTime = crttrackmatcht0->Time() * 1e-3;
            }
        }
    }
  
    if (!hitT0s.empty()){
      crt_HitScore = std::numeric_limits<float>::max();
      for(auto const crthitmatcht0 : hitT0s)
        {
          if(crthitmatcht0->TriggerConfidence() < crt_HitScore)
            {
              crt_HitScore = crthitmatcht0->TriggerConfidence();
              crt_HitTime = crthitmatcht0->Time() * 1e-3;
            }
        }
    }
  }

void sbn::CRUMBS::FillPandoraNuScoreVars

(

std::map< std::string, float > &

propertiesMap

)

Definition at line 486 of file CRUMBS_module.cc.

  {
    auto propertiesMapIter = propertiesMap.find("NuScore");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuScore" << std::endl;
      abort();
    }
    tpc_NuScore = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("CRFracHitsInLongestTrack");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- CRFracHitsInLongestTrack" << std::endl;
      abort();
    }
    tpc_CRFracHitsInLongestTrack = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("CRLongestTrackDeflection");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- CRLongestTrackDeflection" << std::endl;
      abort();
    }
    tpc_CRLongestTrackDeflection = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("CRLongestTrackDirY");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- CRLongestTrackDirY" << std::endl;
      abort();
    }
    tpc_CRLongestTrackDirY = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("CRNHitsMax");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- CRNHitsMax" << std::endl;
      abort();
    }
    tpc_CRNHitsMax = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuEigenRatioInSphere");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuEigenRatioInSphere" << std::endl;
      abort();
    }
    tpc_NuEigenRatioInSphere = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuNFinalStatePfos");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuNFinalStatePfos" << std::endl;
      abort();
    }
    tpc_NuNFinalStatePfos = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuNHitsTotal");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuNHitsTotal" << std::endl;
      abort();
    }
    tpc_NuNHitsTotal = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuNSpacePointsInSphere");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuNSpacePointsInSphere" << std::endl;
      abort();
    }
    tpc_NuNSpacePointsInSphere = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuVertexY");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuVertexY" << std::endl;
      abort();
    }
    tpc_NuVertexY = propertiesMapIter->second;

    propertiesMapIter = propertiesMap.find("NuWeightedDirZ");
    if (propertiesMapIter == propertiesMap.end()){
      std::cout << "CRUMBS_module: Error finding variable -- NuWeightedDirZ" << std::endl;
      abort();
    }
    tpc_NuWeightedDirZ = propertiesMapIter->second;
  }

std::vector< art::Ptr< recob::Hit > > sbn::CRUMBS::GetAllSliceHits	(	art::Event const &	e,
		const art::Ptr< recob::Slice > &	slice,
		const art::ValidHandle< std::vector< recob::Slice > > &	handleSlices
	)

Definition at line 566 of file CRUMBS_module.cc.

  {
    art::FindManyP<recob::Hit> sliceHitAssn(handleSlices,e,fSliceModuleLabel);
    return sliceHitAssn.at(slice.key());
  }

std::vector< art::Ptr< anab::T0 > > sbn::CRUMBS::GetCRTHitT0s	(	art::Event const &	e,
		const art::Ptr< recob::Slice > &	slice,
		const art::ValidHandle< std::vector< recob::PFParticle > > &	handlePFPs,
		const art::ValidHandle< std::vector< recob::Slice > > &	handleSlices
	)

Definition at line 661 of file CRUMBS_module.cc.

  {
    std::vector<art::Ptr<anab::T0> > t0Vec;

    art::Handle<std::vector<recob::Track> > handleTracks;
    e.getByLabel(fTrackModuleLabel, handleTracks);

    art::FindManyP<recob::PFParticle> slicePFPAssn(handleSlices,e,fSliceModuleLabel);
    art::FindManyP<recob::Track> pfpTrackAssn(handlePFPs,e,fTrackModuleLabel);
    art::FindManyP<anab::T0> trackT0Assn(handleTracks,e,fCRTHitMatchModuleLabel);

    const std::vector<art::Ptr<recob::PFParticle> > pfps = slicePFPAssn.at(slice.key());
  
    for(auto const& pfp : pfps)
      {
        if(pfp->PdgCode() != 13)
          continue;

        const std::vector<art::Ptr<recob::Track> > tracks = pfpTrackAssn.at(pfp.key());

        if(tracks.size() != 1)
          continue;

        const art::Ptr<recob::Track> track = tracks.front();

        const std::vector<art::Ptr<anab::T0> > t0s = trackT0Assn.at(track.key());
        t0Vec.insert(t0Vec.end(), t0s.begin(), t0s.end());
      }
  
    return t0Vec;
  }

std::vector< art::Ptr< anab::T0 > > sbn::CRUMBS::GetCRTTrackT0s	(	art::Event const &	e,
		const art::Ptr< recob::Slice > &	slice,
		const art::ValidHandle< std::vector< recob::PFParticle > > &	handlePFPs,
		const art::ValidHandle< std::vector< recob::Slice > > &	handleSlices
	)

Definition at line 628 of file CRUMBS_module.cc.

  {
    std::vector<art::Ptr<anab::T0> > t0Vec;

    art::Handle<std::vector<recob::Track> > handleTracks;
    e.getByLabel(fTrackModuleLabel, handleTracks);

    art::FindManyP<recob::PFParticle> slicePFPAssn(handleSlices,e,fSliceModuleLabel);
    art::FindManyP<recob::Track> pfpTrackAssn(handlePFPs,e,fTrackModuleLabel);
    art::FindManyP<anab::T0> trackT0Assn(handleTracks,e,fCRTTrackMatchModuleLabel);

    const std::vector<art::Ptr<recob::PFParticle> > pfps = slicePFPAssn.at(slice.key());
  
    for(auto const& pfp : pfps)
      {
        if(pfp->PdgCode() != 13)
          continue;

        const std::vector<art::Ptr<recob::Track> > tracks = pfpTrackAssn.at(pfp.key());

        if(tracks.size() != 1)
          continue;

        const art::Ptr<recob::Track> track = tracks.front();

        const std::vector<art::Ptr<anab::T0> > t0s = trackT0Assn.at(track.key());
        t0Vec.insert(t0Vec.end(), t0s.begin(), t0s.end());
      }
  
    return t0Vec;
  }

float sbn::CRUMBS::GetLongestTrackStoppingChi2Ratio	(	art::Event const &	e,
		const art::Ptr< recob::Slice > &	slice,
		const art::ValidHandle< std::vector< recob::PFParticle > > &	handlePFPs,
		const art::ValidHandle< std::vector< recob::Slice > > &	handleSlices
	)

Definition at line 694 of file CRUMBS_module.cc.

  {
    art::Ptr<anab::Calorimetry> longestTrackCalo;
    float maxLength = -std::numeric_limits<float>::max();

    art::Handle<std::vector<recob::Track> > handleTracks;
    e.getByLabel(fTrackModuleLabel, handleTracks);

    art::FindManyP<recob::PFParticle> slicePFPAssn(handleSlices,e,fSliceModuleLabel);
    art::FindOneP<recob::Track> pfpTrackAssn(handlePFPs,e,fTrackModuleLabel);
    art::FindManyP<anab::Calorimetry> trackCaloAssn(handleTracks,e,fCalorimetryModuleLabel);

    const std::vector<art::Ptr<recob::PFParticle> > pfps = slicePFPAssn.at(slice.key());
  
    for(auto const& pfp : pfps)
      {
        if(pfp->PdgCode() != 13)
          continue;

        const art::Ptr<recob::Track> track = pfpTrackAssn.at(pfp.key());

        if(track.isNull())
          continue;

        const std::vector<art::Ptr<anab::Calorimetry> > calos = trackCaloAssn.at(track.key());

        const unsigned int maxHits(std::max({ calos[0]->dEdx().size(), calos[1]->dEdx().size(), calos[2]->dEdx().size() }));
        const int bestPlane((calos[2]->dEdx().size() == maxHits) ? 2 : (calos[0]->dEdx().size() == maxHits) ? 0 : (calos[1]->dEdx().size() == maxHits) ? 1 : -1);

        if (bestPlane == -1)
          continue;

        const art::Ptr<anab::Calorimetry> calo = calos.at(bestPlane);

        if(track->Length() > maxLength)
          {
            maxLength = track->Length();
            longestTrackCalo = calo;
          }
      }

    sbn::StoppingChi2Fit longestTrackFit = longestTrackCalo.isNull() ? StoppingChi2Fit() : fTrackStoppingChi2Alg.RunFitForCosmicID(*longestTrackCalo);

    if(longestTrackFit.pol0Chi2 < 0 || longestTrackFit.expChi2 <= 0)
      return -4.f;
  
    return longestTrackFit.pol0Chi2 / longestTrackFit.expChi2;
  }

void sbn::CRUMBS::GetMaps	(	art::Event const &	e,
		std::map< int, int > &	trackIDToGenMap,
		std::map< int, std::string > &	genTypeMap,
		std::map< int, int > &	genCCNCMap,
		std::map< int, int > &	genNuTypeMap
	)

Definition at line 293 of file CRUMBS_module.cc.

  {

    unsigned nNu(0), nCos(0);

    if(fProcessNeutrinos)
      {
        art::Handle<std::vector<simb::MCTruth> > handleMCTruthNu;
        e.getByLabel(fGeneratorModuleLabel, handleMCTruthNu);
        art::FindManyP<simb::MCParticle> truthNuMCPAssn(handleMCTruthNu,e,fMCParticleModuleLabel);

        for (unsigned int i = 0; i < handleMCTruthNu->size(); ++i){
          const art::Ptr<simb::MCTruth> mcTruth(handleMCTruthNu, i);
          const simb::MCParticle nu = mcTruth->GetNeutrino().Nu();

          if(!fTpcGeo.InVolume(nu))
            genTypeMap[i] = "DirtNu";
          else
            genTypeMap[i] = "Nu";
    
          const std::vector<art::Ptr<simb::MCParticle> > particles = truthNuMCPAssn.at(mcTruth.key());
    
          for (auto const& particle : particles)
            {
              trackIDToGenMap[particle->TrackId()] = i;
            }
          ++nNu;

          genCCNCMap[i]   = mcTruth->GetNeutrino().CCNC();
          genNuTypeMap[i] = mcTruth->GetNeutrino().Nu().PdgCode();
        }
      }

    if(fProcessCosmics)
      {
        art::Handle<std::vector<simb::MCTruth> > handleMCTruthCosmic;
        e.getByLabel(fCosmicModuleLabel, handleMCTruthCosmic);

        art::FindManyP<simb::MCParticle> truthCosmicMCPAssn(handleMCTruthCosmic,e,fMCParticleModuleLabel);

        for (unsigned int i = 0; i < handleMCTruthCosmic->size(); ++i){
          const art::Ptr<simb::MCTruth> mcTruth(handleMCTruthCosmic, i);

          genTypeMap[i + nNu] = "Cosmic";
    
          const std::vector<art::Ptr<simb::MCParticle> > particles = truthCosmicMCPAssn.at(mcTruth.key());
    
          for (auto const& particle : particles)
            {
              trackIDToGenMap[particle->TrackId()] = i + nNu;
            }
          ++nCos;

          genCCNCMap[i + nNu]   = -1;
          genNuTypeMap[i + nNu] = -1;
        }
      }

    eventID = e.event();
    subRunID = e.subRun();
    runID = e.run();
  }

art::Ptr< recob::PFParticle > sbn::CRUMBS::GetSlicePrimary	(	art::Event const &	e,
		const art::Ptr< recob::Slice > &	slice,
		const art::ValidHandle< std::vector< recob::Slice > > &	handleSlices
	)

Definition at line 572 of file CRUMBS_module.cc.

  {
    art::FindManyP<recob::PFParticle> slicePfpAssn(handleSlices,e,fSliceModuleLabel);
    std::vector<art::Ptr<recob::PFParticle> > pfps = slicePfpAssn.at(slice.key());

    for(auto const &pfp : pfps)
      {
        if(pfp->IsPrimary())
          return pfp;
      }
  
    art::Ptr<recob::PFParticle> nullReturn;
    return nullReturn;
  }

void sbn::CRUMBS::GetTruthMatching	(	art::Event const &	e,
		const std::vector< art::Ptr< recob::Hit > > &	sliceHits,
		const std::vector< art::Ptr< recob::Hit > > &	allHits,
		std::map< int, int > &	trackIDToGenMap,
		int &	matchedID,
		double &	purity,
		double &	completeness
	)

Definition at line 587 of file CRUMBS_module.cc.

  {
    std::map<int, int> sliceHitMap;
    std::map<int, float> slicePurityMap;

    auto clockData = art::ServiceHandle<detinfo::DetectorClocksService>()->DataFor(e);

    for (auto const& hit : sliceHits)
      {
        ++sliceHitMap[trackIDToGenMap[TruthMatchUtils::TrueParticleID(clockData,hit,true)]];
      }

    for (auto const& [id, nHits] : sliceHitMap)
      {
        slicePurityMap[id] = (float) nHits / (float) sliceHits.size();
      }

    for (auto const& [id, pur] : slicePurityMap)
      {
        if(pur > purity) 
          {
            matchedID = id;
            purity = pur;
          }
      }

    int totalTrueHits(0);

    for (auto const& hit : allHits)
      {
        if(trackIDToGenMap[TruthMatchUtils::TrueParticleID(clockData,hit,true)] == matchedID)
          ++totalTrueHits;
      }
  
    if(totalTrueHits == 0) 
      completeness = 0;
    else
      completeness = sliceHitMap[matchedID] / (float) totalTrueHits;
  }

void sbn::CRUMBS::InitialiseMVAReader	(	TMVA::Reader &	mvaReader,
		std::string &	mvaName,
		std::string &	mvaFileName
	)

Definition at line 246 of file CRUMBS_module.cc.

  {
    mvaReader.AddVariable("tpc_CRFracHitsInLongestTrack",&tpc_CRFracHitsInLongestTrack);
    mvaReader.AddVariable("tpc_CRLongestTrackDeflection",&tpc_CRLongestTrackDeflection);
    mvaReader.AddVariable("tpc_CRLongestTrackDirY",&tpc_CRLongestTrackDirY);
    mvaReader.AddVariable("tpc_CRNHitsMax",&tpc_CRNHitsMax);
    mvaReader.AddVariable("tpc_NuEigenRatioInSphere",&tpc_NuEigenRatioInSphere);
    mvaReader.AddVariable("tpc_NuNFinalStatePfos",&tpc_NuNFinalStatePfos);
    mvaReader.AddVariable("tpc_NuNHitsTotal",&tpc_NuNHitsTotal);
    mvaReader.AddVariable("tpc_NuNSpacePointsInSphere",&tpc_NuNSpacePointsInSphere);
    mvaReader.AddVariable("tpc_NuVertexY",&tpc_NuVertexY);
    mvaReader.AddVariable("tpc_NuWeightedDirZ",&tpc_NuWeightedDirZ);
    mvaReader.AddVariable("tpc_StoppingChi2CosmicRatio",&tpc_StoppingChi2CosmicRatio);

    mvaReader.AddVariable("pds_FMTotalScore",&pds_FMTotalScore);
    mvaReader.AddVariable("pds_FMPE",&pds_FMPE);
    mvaReader.AddVariable("pds_FMTime",&pds_FMTime);

    mvaReader.AddVariable("crt_TrackScore",&crt_TrackScore);
    mvaReader.AddVariable("crt_HitScore",&crt_HitScore);
    mvaReader.AddVariable("crt_TrackTime",&crt_TrackTime);
    mvaReader.AddVariable("crt_HitTime",&crt_HitTime);

    cet::search_path searchPath("FW_SEARCH_PATH");
    std::string weightFileFullPath;
    if (!searchPath.find_file(mvaFileName, weightFileFullPath))
      throw cet::exception("CRUMBS") << "Unable to find weight file: " << mvaFileName << " in FW_SEARCH_PATH: " << searchPath.to_string();

    mvaReader.BookMVA(mvaName, weightFileFullPath);
  }

CRUMBS& sbn::CRUMBS::operator= ( CRUMBS const &  )

delete

CRUMBS& sbn::CRUMBS::operator= ( CRUMBS &&  )

delete

void sbn::CRUMBS::produce ( art::Event &  e )

override

Definition at line 357 of file CRUMBS_module.cc.

  {
    std::map<int, int> trackIDToGenMap;
    std::map<int, std::string> genTypeMap;
    std::map<int, int> genCCNCMap;
    std::map<int, int> genNuTypeMap;

    if(fTrainingMode)
      this->GetMaps(e, trackIDToGenMap, genTypeMap, genCCNCMap, genNuTypeMap);

    auto resultsVec = std::make_unique<std::vector<CRUMBSResult>>();
    auto sliceAssns = std::make_unique<art::Assns<recob::Slice, CRUMBSResult>>();

    auto const handleSlices(e.getValidHandle<std::vector<recob::Slice>>(fSliceModuleLabel));
    std::vector<art::Ptr<recob::Slice>> slices;
    art::fill_ptr_vector(slices, handleSlices);

    auto const handlePFPs(e.getValidHandle<std::vector<recob::PFParticle>>(fPFParticleModuleLabel));
    std::vector<art::Ptr<recob::PFParticle>> pfps;
    art::fill_ptr_vector(pfps, handlePFPs);

    auto const handleHits(e.getValidHandle<std::vector<recob::Hit>>(fHitModuleLabel));
    std::vector<art::Ptr<recob::Hit>> allHits;
    art::fill_ptr_vector(allHits, handleHits);

    art::FindManyP<larpandoraobj::PFParticleMetadata> pfpMetadataAssoc(handlePFPs, e, fPFParticleModuleLabel);
    art::FindManyP<sbn::SimpleFlashMatch> pfpFMAssoc(handlePFPs, e, fFlashMatchModuleLabel);

    for(auto const &slice : slices)
      {
        this->ResetVars();

        auto const primary = this->GetSlicePrimary(e, slice, handleSlices);

        if(primary.isNull())
          continue;

        if(primary->PdgCode() == 13 || primary->PdgCode() == 11)
          continue;

        const std::vector<art::Ptr<larpandoraobj::PFParticleMetadata> > pfpMetaVec = pfpMetadataAssoc.at(primary.key());
        const std::vector<art::Ptr<sbn::SimpleFlashMatch> > pfpFMVec = pfpFMAssoc.at(primary.key());
        const std::vector<art::Ptr<anab::T0> > sliceCRTTrackT0s = this->GetCRTTrackT0s(e, slice, handlePFPs, handleSlices);
        const std::vector<art::Ptr<anab::T0> > sliceCRTHitT0s = this->GetCRTHitT0s(e, slice, handlePFPs, handleSlices);

        this->FillCRTVars(sliceCRTTrackT0s, sliceCRTHitT0s);

        const art::Ptr<larpandoraobj::PFParticleMetadata> pfpMeta = pfpMetaVec.front();
        std::map<std::string, float> propertiesMap = pfpMeta->GetPropertiesMap();
      
        this->FillPandoraNuScoreVars(propertiesMap);

        tpc_StoppingChi2CosmicRatio = this->GetLongestTrackStoppingChi2Ratio(e, slice, handlePFPs, handleSlices);
    
        const art::Ptr<sbn::SimpleFlashMatch> flashmatch = pfpFMVec.front();
        pds_FMTotalScore = flashmatch->score.total;
        pds_FMPE = flashmatch->light.pe;
        pds_FMTime = std::max(flashmatch->time, -100.);
      
        if(!fTrainingMode || fEvaluateResultInTrainingMode)
          {
            const float score       = fMVAReader.EvaluateMVA(fMVAName);
            const float ccnumuscore = fCCNuMuMVAReader.EvaluateMVA(fCCNuMuMVAName);
            const float ccnuescore  = fCCNuEMVAReader.EvaluateMVA(fCCNuEMVAName);
            const float ncscore     = fNCMVAReader.EvaluateMVA(fNCMVAName);

            const float bestscore   = (ccnumuscore > ccnuescore && ccnumuscore > ncscore) ? ccnumuscore : (ccnuescore > ncscore) ? ccnuescore : ncscore;
            const int   bestid      = (ccnumuscore > ccnuescore && ccnumuscore > ncscore) ? 14 : (ccnuescore > ncscore) ? 12 : 1;
            
            resultsVec->emplace_back(score, ccnumuscore, ccnuescore, ncscore, bestscore, bestid, tpc_CRFracHitsInLongestTrack, tpc_CRLongestTrackDeflection, 
                                     tpc_CRLongestTrackDirY, std::round(tpc_CRNHitsMax), tpc_NuEigenRatioInSphere, std::round(tpc_NuNFinalStatePfos), 
                                     std::round(tpc_NuNHitsTotal), std::round(tpc_NuNSpacePointsInSphere), tpc_NuVertexY, tpc_NuWeightedDirZ, 
                                     tpc_StoppingChi2CosmicRatio, pds_FMTotalScore, pds_FMPE, pds_FMTime, crt_TrackScore, crt_HitScore, 
                                     crt_TrackTime, crt_HitTime);
            
            util::CreateAssn(*this, e, *resultsVec, slice, *sliceAssns);
          }

        if(fTrainingMode)
          {
            std::vector<art::Ptr<recob::Hit> > sliceHits = this->GetAllSliceHits(e, slice, handleSlices);

            int matchedID(-1);
            this->GetTruthMatching(e, sliceHits, allHits, trackIDToGenMap, matchedID, matchedPurity, matchedCompleteness);

            slicePDG = primary->PdgCode();
            matchedType = genTypeMap[matchedID];
      
            ccnc   = genCCNCMap[matchedID];
            nutype = genNuTypeMap[matchedID];

            fSliceTree->Fill();
          }
      }

    if(!fTrainingMode || fEvaluateResultInTrainingMode)
      {
        e.put(std::move(resultsVec));
        e.put(std::move(sliceAssns));
      }
  }

void sbn::CRUMBS::ResetVars

(

)

Definition at line 277 of file CRUMBS_module.cc.

  {
    tpc_NuScore = -999999.; tpc_CRFracHitsInLongestTrack = -999999.; tpc_CRLongestTrackDeflection = -999999.; tpc_CRLongestTrackDirY = -999999.; tpc_CRNHitsMax = -999999.;
    tpc_NuEigenRatioInSphere = -999999.; tpc_NuNFinalStatePfos = -999999.; tpc_NuNHitsTotal = -999999.; tpc_NuNSpacePointsInSphere = -999999.; tpc_NuVertexY = -999999.;
    tpc_NuWeightedDirZ = -999999.; tpc_StoppingChi2CosmicRatio = -4.;

    pds_FMTotalScore = -999999.; pds_FMPE = -999999.; pds_FMTime = -500.;

    crt_TrackScore = -4.; crt_HitScore = -4.; crt_TrackTime = -3000; crt_HitTime = -3000;

    slicePDG = 999999;
    matchedType = "";
    matchedPurity = -999999.; matchedCompleteness = -999999.;
    ccnc = 999999; nutype = 999999;
  }

int sbn::CRUMBS::SliceTruthId

(

std::map< int, float > &

purities

)

Member Data Documentation

int sbn::CRUMBS::ccnc

private

Definition at line 126 of file CRUMBS_module.cc.

float sbn::CRUMBS::crt_HitScore

private

Definition at line 162 of file CRUMBS_module.cc.

float sbn::CRUMBS::crt_HitTime

private

Definition at line 164 of file CRUMBS_module.cc.

float sbn::CRUMBS::crt_TrackScore

private

Definition at line 161 of file CRUMBS_module.cc.

float sbn::CRUMBS::crt_TrackTime

private

Definition at line 163 of file CRUMBS_module.cc.

unsigned sbn::CRUMBS::eventID

private

Definition at line 125 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCalorimetryModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCCNuEMVAFileName

private

Definition at line 112 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCCNuEMVAName

private

Definition at line 112 of file CRUMBS_module.cc.

TMVA::Reader sbn::CRUMBS::fCCNuEMVAReader

private

Definition at line 121 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCCNuMuMVAFileName

private

Definition at line 112 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCCNuMuMVAName

private

Definition at line 112 of file CRUMBS_module.cc.

TMVA::Reader sbn::CRUMBS::fCCNuMuMVAReader

private

Definition at line 121 of file CRUMBS_module.cc.

fhicl::ParameterSet sbn::CRUMBS::fChi2FitParams

private

Definition at line 115 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCosmicModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCRTHitMatchModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fCRTTrackMatchModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

bool sbn::CRUMBS::fEvaluateResultInTrainingMode

private

Definition at line 105 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fFlashMatchModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fGeneratorModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fHitModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fMCParticleModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fMVAFileName

private

Definition at line 112 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fMVAName

private

Definition at line 112 of file CRUMBS_module.cc.

TMVA::Reader sbn::CRUMBS::fMVAReader

private

Definition at line 121 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fNCMVAFileName

private

Definition at line 112 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fNCMVAName

private

Definition at line 112 of file CRUMBS_module.cc.

TMVA::Reader sbn::CRUMBS::fNCMVAReader

private

Definition at line 121 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fPFParticleModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

bool sbn::CRUMBS::fProcessCosmics

private

Definition at line 105 of file CRUMBS_module.cc.

bool sbn::CRUMBS::fProcessNeutrinos

private

Definition at line 105 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fSliceModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

TTree* sbn::CRUMBS::fSliceTree

private

Definition at line 118 of file CRUMBS_module.cc.

sbn::TPCGeoAlg sbn::CRUMBS::fTpcGeo

private

Definition at line 132 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::fTrackModuleLabel

private

Definition at line 108 of file CRUMBS_module.cc.

sbn::TrackStoppingChi2Alg sbn::CRUMBS::fTrackStoppingChi2Alg

private

Definition at line 131 of file CRUMBS_module.cc.

bool sbn::CRUMBS::fTrainingMode

private

Definition at line 105 of file CRUMBS_module.cc.

double sbn::CRUMBS::matchedCompleteness

private

Definition at line 128 of file CRUMBS_module.cc.

double sbn::CRUMBS::matchedPurity

private

Definition at line 128 of file CRUMBS_module.cc.

std::string sbn::CRUMBS::matchedType

private

Definition at line 127 of file CRUMBS_module.cc.

int sbn::CRUMBS::nutype

private

Definition at line 126 of file CRUMBS_module.cc.

float sbn::CRUMBS::pds_FMPE

private

Definition at line 157 of file CRUMBS_module.cc.

float sbn::CRUMBS::pds_FMTime

private

Definition at line 158 of file CRUMBS_module.cc.

float sbn::CRUMBS::pds_FMTotalScore

private

Definition at line 156 of file CRUMBS_module.cc.

unsigned sbn::CRUMBS::runID

private

Definition at line 125 of file CRUMBS_module.cc.

unsigned sbn::CRUMBS::slicePDG

private

Definition at line 125 of file CRUMBS_module.cc.

unsigned sbn::CRUMBS::subRunID

private

Definition at line 125 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_CRFracHitsInLongestTrack

private

Definition at line 139 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_CRLongestTrackDeflection

private

Definition at line 140 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_CRLongestTrackDirY

private

Definition at line 141 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_CRNHitsMax

private

Definition at line 142 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuEigenRatioInSphere

private

Definition at line 145 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuNFinalStatePfos

private

Definition at line 146 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuNHitsTotal

private

Definition at line 147 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuNSpacePointsInSphere

private

Definition at line 148 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuScore

private

Definition at line 124 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuVertexY

private

Definition at line 149 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_NuWeightedDirZ

private

Definition at line 150 of file CRUMBS_module.cc.

float sbn::CRUMBS::tpc_StoppingChi2CosmicRatio

private

Definition at line 153 of file CRUMBS_module.cc.

---

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

• CRUMBS_module.cc