proto::SaveSpacePoints Class Reference

Inheritance diagram for proto::SaveSpacePoints:

Public Member Functions
	SaveSpacePoints (fhicl::ParameterSet const &p)
	SaveSpacePoints (SaveSpacePoints const &)=delete
	SaveSpacePoints (SaveSpacePoints &&)=delete
SaveSpacePoints &	operator= (SaveSpacePoints const &)=delete
SaveSpacePoints &	operator= (SaveSpacePoints &&)=delete
void	analyze (art::Event const &e) override
void	beginJob () override
Public Member Functions inherited from art::EDAnalyzer
	EDAnalyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDAnalyzer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Analyzer
virtual	~Analyzer () noexcept
	Analyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	Analyzer (Table< Config > const &config)
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Observer
	~Observer () noexcept
	Observer (Observer const &)=delete
	Observer (Observer &&)=delete
Observer &	operator= (Observer const &)=delete
Observer &	operator= (Observer &&)=delete
void	registerProducts (ProductDescriptions &, ModuleDescription const &)
void	fillDescriptions (ModuleDescription const &)
fhicl::ParameterSetID	selectorConfig () const
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Attributes
const art::InputTag	fSpacePointModuleLabel
const art::InputTag	fBeamModuleLabel
const art::InputTag	fTrackModuleLabel
const art::InputTag	fTimeDecoderModuleLabel
protoana::ProtoDUNEBeamlineUtils	fBeamlineUtils
TTree *	fTree
int	run
int	subrun
int	event
int	trigger
double	evttime
std::vector< double >	vx
std::vector< double >	vy
std::vector< double >	vz
std::vector< double >	vcharge
std::vector< int >	vtpc
std::vector< int >	vtrackid
std::vector< int >	vpdg
std::vector< int >	vg4id
std::vector< int >	vorigin
std::vector< double >	beamPosx
std::vector< double >	beamPosy
std::vector< double >	beamPosz
std::vector< double >	beamDirx
std::vector< double >	beamDiry
std::vector< double >	beamDirz
std::vector< double >	beamMomentum
double	tof
short	ckov0status
short	ckov1status

Additional Inherited Members
Public Types inherited from art::EDAnalyzer
using	WorkerType = WorkerT< EDAnalyzer >
using	ModuleType = EDAnalyzer
Protected Member Functions inherited from art::Observer
std::string const &	processName () const
bool	wantAllEvents () const noexcept
bool	wantEvent (ScheduleID id, Event const &e) const
Handle< TriggerResults >	getTriggerResults (Event const &e) const
	Observer (fhicl::ParameterSet const &config)
	Observer (std::vector< std::string > const &select_paths, std::vector< std::string > const &reject_paths, fhicl::ParameterSet const &config)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 43 of file SaveSpacePoints_module.cc.

Constructor & Destructor Documentation

proto::SaveSpacePoints::SaveSpacePoints ( fhicl::ParameterSet const &  p )

explicit

Definition at line 107 of file SaveSpacePoints_module.cc.

  :
  EDAnalyzer(p),
  fSpacePointModuleLabel(p.get< art::InputTag >("SpacePointModuleLabel")),
  fBeamModuleLabel(p.get< art::InputTag >("BeamModuleLabel")),
  fTrackModuleLabel(p.get< art::InputTag >("TrackModuleLabel")),
  fTimeDecoderModuleLabel(p.get< art::InputTag >("TimeDecoderModuleLabel")),
  fBeamlineUtils(p.get<fhicl::ParameterSet>("BeamlineUtils"))
{}

proto::SaveSpacePoints::SaveSpacePoints ( SaveSpacePoints const &  )

delete

proto::SaveSpacePoints::SaveSpacePoints ( SaveSpacePoints &&  )

delete

Member Function Documentation

void proto::SaveSpacePoints::analyze ( art::Event const &  e )

overridevirtual

Implements art::EDAnalyzer.

Definition at line 117 of file SaveSpacePoints_module.cc.

{
  // Implementation of required member function here.
  run = evt.run();
  subrun = evt.subRun();
  event = evt.id().event();
  art::Timestamp ts = evt.time();
  //std::cout<<ts.timeHigh()<<" "<<ts.timeLow()<<std::endl;
  if (ts.timeHigh() == 0){
    TTimeStamp tts(ts.timeLow());
    evttime = tts.AsDouble();
  }
  else{
    TTimeStamp tts(ts.timeHigh(), ts.timeLow());
    evttime = tts.AsDouble();
  }
  vx.clear();
  vy.clear();
  vz.clear();
  vcharge.clear();
  vtpc.clear();
  vtrackid.clear();
  vpdg.clear();
  vg4id.clear();
  vorigin.clear();
  beamPosx.clear();
  beamPosy.clear();
  beamPosz.clear();
  beamDirx.clear();
  beamDiry.clear();
  beamDirz.clear();
  beamMomentum.clear();

  if (evt.isRealData()){
    if (fBeamlineUtils.IsGoodBeamlineTrigger(evt)){

      //Access the Beam Event
      auto beamHandle = evt.getValidHandle<std::vector<beam::ProtoDUNEBeamEvent>>("beamevent");
      
      std::vector<art::Ptr<beam::ProtoDUNEBeamEvent>> beamVec;
      if( beamHandle.isValid()){
        art::fill_ptr_vector(beamVec, beamHandle);
      }
      
      const beam::ProtoDUNEBeamEvent & beamEvent = *(beamVec.at(0)); //Should just have one
      
      //Access momentum
      const std::vector< double > & the_momenta = beamEvent.GetRecoBeamMomenta();
      std::cout << "Number of reconstructed momenta: " << the_momenta.size() << std::endl;
      
      beamMomentum.insert( beamMomentum.end(), the_momenta.begin(), the_momenta.end() );
      
      tof = -1;
      ckov0status = -1;
      ckov1status = -1;
      
      //Access time of flight
      const std::vector< double > & the_tofs  = beamEvent.GetTOFs();
      
      if( the_tofs.size() > 0){
        tof = the_tofs[0];
      }
      ckov0status = beamEvent.GetCKov0Status();
      ckov1status = beamEvent.GetCKov1Status();
    }
  
    trigger = -1;
    art::ValidHandle<std::vector<raw::RDTimeStamp>> timeStamps = evt.getValidHandle<std::vector<raw::RDTimeStamp>>(fTimeDecoderModuleLabel);

    // Check that we have good information
    if(timeStamps.isValid() && timeStamps->size() == 1){
      // Access the trigger information. Beam trigger flag = 0xc
      const raw::RDTimeStamp& timeStamp = timeStamps->at(0);
      trigger = timeStamp.GetFlags();
    }
  }

  std::vector< art::Ptr<recob::SpacePoint> > sps;
  auto spsHandle = evt.getHandle< std::vector<recob::SpacePoint> >(fSpacePointModuleLabel);
  if (spsHandle) art::fill_ptr_vector(sps, spsHandle);

  std::vector< art::Ptr<recob::PointCharge> > pcs;
  auto pcsHandle = evt.getHandle< std::vector<recob::PointCharge> >(fSpacePointModuleLabel);
  if (pcsHandle) art::fill_ptr_vector(pcs, pcsHandle);

  art::FindManyP<recob::Hit> fmhsp(spsHandle, evt, fSpacePointModuleLabel);

  //Services
  art::ServiceHandle<cheat::BackTrackerService> bt_serv;
  art::ServiceHandle<cheat::ParticleInventoryService> pi_serv;

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

  for (size_t i = 0; i<sps.size(); ++i){
    vx.push_back(sps[i]->XYZ()[0]);
    vy.push_back(sps[i]->XYZ()[1]);
    vz.push_back(sps[i]->XYZ()[2]);
    vcharge.push_back(pcs[i]->charge());
    vtrackid.push_back(-1);
    auto const& hits = fmhsp.at(i);
    if (!evt.isRealData()){
      int TrackID = 0;
      std::map<int,double> trkide;
      for (auto const & hit : hits){
        std::vector<sim::TrackIDE> TrackIDs = bt_serv->HitToTrackIDEs(clockData, hit);
        for(size_t e = 0; e < TrackIDs.size(); ++e){
          trkide[TrackIDs[e].trackID] += TrackIDs[e].energy;
        }
      }
      // Work out which IDE despoited the most charge in the hit if there was more than one.
      double maxe = -1;
      double tote = 0;
      for (std::map<int,double>::iterator ii = trkide.begin(); ii!=trkide.end(); ++ii){
        tote += ii->second;
        if ((ii->second)>maxe){
          maxe = ii->second;
          TrackID = ii->first;
        }
      }
      // Now have trackID, so get PdG code and T0 etc.
      vg4id.push_back(TrackID);
      const simb::MCParticle *particle = pi_serv->TrackIdToParticle_P(TrackID);
      if (particle){
        vpdg.push_back(particle->PdgCode());
        vorigin.push_back(pi_serv->ParticleToMCTruth_P(particle)->Origin());
      }
      else{
        vpdg.push_back(0);
        vorigin.push_back(0);
      }
    }
    else{
      vg4id.push_back(0);
      vpdg.push_back(0);
      vorigin.push_back(0);
    }
    int spTPC = -1;
    for (auto const & hit : hits){
      spTPC = hit->WireID().TPC;
    }
    vtpc.push_back(spTPC);
  }

  std::vector< art::Ptr<recob::Track> > trks;
  auto trkHandle = evt.getHandle< std::vector<recob::Track> >(fTrackModuleLabel);
  if (trkHandle) art::fill_ptr_vector(trks, trkHandle);

  for (size_t i = 0; i<trks.size(); ++i){
    auto & trk = trks[i];
    for (size_t j = 0; j<trk->NPoints(); ++j){
      if (trk->HasValidPoint(j)){
        vx.push_back(trk->TrajectoryPoint(j).position.X());
        vy.push_back(trk->TrajectoryPoint(j).position.Y());
        vz.push_back(trk->TrajectoryPoint(j).position.Z());
        vcharge.push_back(0);
        vtpc.push_back(-1);
        vtrackid.push_back(trk->ID());
        vpdg.push_back(-1);
        vg4id.push_back(-1);
        vorigin.push_back(-1);
      }
    }
  }

  fTree->Fill();
}

void proto::SaveSpacePoints::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 284 of file SaveSpacePoints_module.cc.

{
  art::ServiceHandle<art::TFileService> tfs;
  fTree = tfs->make<TTree>("spt","space point tree");
  fTree->Branch("run",&run,"run/I");
  fTree->Branch("subrun",&subrun,"subrun/I");
  fTree->Branch("event",&event,"event/I");
  fTree->Branch("trigger",&trigger,"trigger/I");
  fTree->Branch("evttime",&evttime,"evttime/D");
  fTree->Branch("vx",&vx);
  fTree->Branch("vy",&vy);
  fTree->Branch("vz",&vz);
  fTree->Branch("vcharge",&vcharge);
  fTree->Branch("vtpc",&vtpc);
  fTree->Branch("vtrackid",&vtrackid);
  fTree->Branch("vpdg",&vpdg);
  fTree->Branch("vg4id",&vg4id);
  fTree->Branch("vorigin",&vorigin);
  fTree->Branch("beamPosx",&beamPosx);
  fTree->Branch("beamPosy",&beamPosy);
  fTree->Branch("beamPosz",&beamPosz);
  fTree->Branch("beamDirx",&beamDirx);
  fTree->Branch("beamDiry",&beamDiry);
  fTree->Branch("beamDirz",&beamDirz);
  fTree->Branch("beamMomentum",&beamMomentum);
  fTree->Branch("tof", &tof, "tof/D");
  fTree->Branch("ckov0status", &ckov0status, "ckov0status/S");
  fTree->Branch("ckov1status", &ckov1status, "ckov1status/S");
}

SaveSpacePoints& proto::SaveSpacePoints::operator= ( SaveSpacePoints const &  )

delete

SaveSpacePoints& proto::SaveSpacePoints::operator= ( SaveSpacePoints &&  )

delete

Member Data Documentation

std::vector<double> proto::SaveSpacePoints::beamDirx

private

Definition at line 94 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamDiry

private

Definition at line 95 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamDirz

private

Definition at line 96 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamMomentum

private

Definition at line 98 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamPosx

private

Definition at line 90 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamPosy

private

Definition at line 91 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::beamPosz

private

Definition at line 92 of file SaveSpacePoints_module.cc.

short proto::SaveSpacePoints::ckov0status

private

Definition at line 101 of file SaveSpacePoints_module.cc.

short proto::SaveSpacePoints::ckov1status

private

Definition at line 102 of file SaveSpacePoints_module.cc.

int proto::SaveSpacePoints::event

private

Definition at line 74 of file SaveSpacePoints_module.cc.

double proto::SaveSpacePoints::evttime

private

Definition at line 76 of file SaveSpacePoints_module.cc.

protoana::ProtoDUNEBeamlineUtils proto::SaveSpacePoints::fBeamlineUtils

private

Definition at line 68 of file SaveSpacePoints_module.cc.

const art::InputTag proto::SaveSpacePoints::fBeamModuleLabel

private

Definition at line 64 of file SaveSpacePoints_module.cc.

const art::InputTag proto::SaveSpacePoints::fSpacePointModuleLabel

private

Definition at line 63 of file SaveSpacePoints_module.cc.

const art::InputTag proto::SaveSpacePoints::fTimeDecoderModuleLabel

private

Definition at line 66 of file SaveSpacePoints_module.cc.

const art::InputTag proto::SaveSpacePoints::fTrackModuleLabel

private

Definition at line 65 of file SaveSpacePoints_module.cc.

TTree* proto::SaveSpacePoints::fTree

private

Definition at line 70 of file SaveSpacePoints_module.cc.

int proto::SaveSpacePoints::run

private

Definition at line 72 of file SaveSpacePoints_module.cc.

int proto::SaveSpacePoints::subrun

private

Definition at line 73 of file SaveSpacePoints_module.cc.

double proto::SaveSpacePoints::tof

private

Definition at line 100 of file SaveSpacePoints_module.cc.

int proto::SaveSpacePoints::trigger

private

Definition at line 75 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::vcharge

private

Definition at line 82 of file SaveSpacePoints_module.cc.

std::vector<int> proto::SaveSpacePoints::vg4id

private

Definition at line 86 of file SaveSpacePoints_module.cc.

std::vector<int> proto::SaveSpacePoints::vorigin

private

Definition at line 87 of file SaveSpacePoints_module.cc.

std::vector<int> proto::SaveSpacePoints::vpdg

private

Definition at line 85 of file SaveSpacePoints_module.cc.

std::vector<int> proto::SaveSpacePoints::vtpc

private

Definition at line 83 of file SaveSpacePoints_module.cc.

std::vector<int> proto::SaveSpacePoints::vtrackid

private

Definition at line 84 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::vx

private

Definition at line 79 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::vy

private

Definition at line 80 of file SaveSpacePoints_module.cc.

std::vector<double> proto::SaveSpacePoints::vz

private

Definition at line 81 of file SaveSpacePoints_module.cc.

---

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

• protoduneana/protoduneana/Utilities/SaveSpacePoints_module.cc