PlotTrackDetails_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
//
//  PlotTrackDetails_module.cc
//
//  Author: Leigh Whitehead (leigh.howard.whitehead@cern.ch)
//
//  Module to provide basic tracking information for protoDUNE
//  Nearline Monitoring
//
////////////////////////////////////////////////////////////////////////

#include "art/Framework/Core/EDAnalyzer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Principal/Run.h"
#include "art/Framework/Principal/SubRun.h"
#include "canvas/Utilities/InputTag.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include "larcore/Geometry/Geometry.h"
#include "lardata/DetectorInfoServices/LArPropertiesService.h"
#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"
#include "larcoreobj/SimpleTypesAndConstants/geo_types.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/AnalysisBase/T0.h"
#include "lardataobj/AnalysisBase/CosmicTag.h"
#include "art_root_io/TFileService.h"
#include "art_root_io/TFileDirectory.h"
#include "canvas/Persistency/Common/FindManyP.h"

#include <fstream>

#include "TH1F.h"

namespace nlana {
  class PlotTrackDetails;
}

class nlana::PlotTrackDetails : public art::EDAnalyzer {
public:
  explicit PlotTrackDetails(fhicl::ParameterSet const & p);
  // The compiler-generated destructor is fine for non-base
  // classes without bare pointers or other resource use.

  // Plugins should not be copied or assigned.
  PlotTrackDetails(PlotTrackDetails const &) = delete;
  PlotTrackDetails(PlotTrackDetails &&) = delete;
  PlotTrackDetails & operator = (PlotTrackDetails const &) = delete;
  PlotTrackDetails & operator = (PlotTrackDetails &&) = delete;

  // Required functions.
  void analyze(art::Event const & e) override;

  // Selected optional functions.
  void beginJob() override;
  void endJob() override;
  void reconfigure(fhicl::ParameterSet const & p) ;

private:

  std::string fTrackProducerLabel;

  // Number of tracks
  TH1F *fNTracksHist;

  // Cosmic tagged tracks
  TH1F *fNCosmicTagHist;

  // Number of T0 tag plots
  TH1F *fNT0Hist;

  // T0 plots  
  TH1F *fT0Hist;
  TH1F *fT0HistHiRes;
  TH1F *fCosmicT0Hist;
  TH1F *fCosmicT0HistHiRes;

};


nlana::PlotTrackDetails::PlotTrackDetails(fhicl::ParameterSet const & pset)
  :
  EDAnalyzer(pset)  // ,
 // More initializers here.
{
  reconfigure(pset);
  
}

//--------------------------------------------------------------------
void nlana::PlotTrackDetails::beginJob()
{
  // Implementation of optional member function here.
  
  art::ServiceHandle<art::TFileService> tfs;

  fNTracksHist = tfs->make<TH1F>("NTracks",";Number of tracks",50,0,200);
  fNCosmicTagHist = tfs->make<TH1F>("NCosmicTag",";Number of tracks tagged as cosmic",50,0,200);
  fNT0Hist = tfs->make<TH1F>("NT0s",";Number of tracks with reconstructed T0",50,0,40);

  fT0Hist = tfs->make<TH1F>("TrackT0",";T0 (us)",100,-4000,4000);
  fT0HistHiRes = tfs->make<TH1F>("TrackT0HiRes",";T0 (us)",1000,-4000,4000);
  fCosmicT0Hist = tfs->make<TH1F>("CosmicTrackT0",";T0 (us)",100,-4000,4000);
  fCosmicT0HistHiRes = tfs->make<TH1F>("CosmicTrackT0HiRes",";T0 (us)",1000,-4000,4000);

} // beginJob

//--------------------------------------------------------------------
void nlana::PlotTrackDetails::reconfigure(fhicl::ParameterSet const & pset)
{
  // The name of the module that produced the tracks
  fTrackProducerLabel  = pset.get<std::string>("TrackProducerLabel");
} // reconfigure


//--------------------------------------------------------------------
void nlana::PlotTrackDetails::endJob()
{

} // endJob

//--------------------------------------------------------------------
void nlana::PlotTrackDetails::analyze(art::Event const & evt)
{
  
  // Try finding some tracks
  art::ValidHandle< std::vector<recob::Track> > trackHandle
          = evt.getValidHandle<std::vector<recob::Track> >(fTrackProducerLabel);

  // Find the associations between tracks and T0
  const art::FindManyP<anab::T0> findTrackT0(trackHandle,evt,fTrackProducerLabel);

  // Also look for cosmic tags so we can make a T0 plot for cosmic tagged events only
  const art::FindManyP<anab::CosmicTag> findCosmicTag(trackHandle,evt,fTrackProducerLabel);

  fNTracksHist->Fill(trackHandle->size());

  unsigned int nT0s = 0;
  unsigned int nTags = 0;

  for ( size_t track_index = 0; track_index != trackHandle->size(); ++track_index )
  {
    const auto thisTrack = (*trackHandle)[track_index];

    // Did this track have an associated T0?
    auto const& t0s = findTrackT0.at(track_index);
    if(t0s.size() != 0){
      ++nT0s;
      fT0Hist->Fill(t0s[0]->Time());
      fT0HistHiRes->Fill(t0s[0]->Time());
    }
    // Did this track have a cosmic tag?
    auto const& tag = findCosmicTag.at(track_index);
    if(tag.size() != 0){
      ++nTags;
    }

    if(t0s.size() && tag.size()){
      fCosmicT0Hist->Fill(t0s[0]->Time());
      fCosmicT0HistHiRes->Fill(t0s[0]->Time());
    }
    
  }

  fNCosmicTagHist->Fill(nTags);
  fNT0Hist->Fill(nT0s);

} // analyze


DEFINE_ART_MODULE(nlana::PlotTrackDetails)