hit::HitAnaModule Class Reference

Inheritance diagram for hit::HitAnaModule:

Public Member Functions
	HitAnaModule (fhicl::ParameterSet const &p)

Private Types
using	HitWireAssns_t = art::Assns< recob::Hit, recob::Wire >

Private Member Functions
void	analyze (art::Event const &e) override
void	beginJob () override
void	createAssocVector (HitWireAssns_t const &, std::vector< std::vector< int >> &)
void	createAssocVector (std::vector< recob::Wire > const &, std::vector< recob::Hit > const &, std::vector< std::vector< int >> &)
void	createMCAssocVector (std::vector< recob::Wire > const &, std::vector< sim::MCHitCollection > const &, std::vector< std::vector< int >> &)

Private Attributes
std::vector< std::string >	fHitModuleLabels
std::string	fMCHitModuleLabel
std::string	fWireModuleLabel
TTree *	wireDataTree
std::vector< TTree * >	hitDataTree
HitAnaAlg	analysisAlg

Detailed Description

Definition at line 46 of file HitAnaModule_module.cc.

Member Typedef Documentation

using hit::HitAnaModule::HitWireAssns_t = art::Assns<recob::Hit, recob::Wire>

private

Definition at line 55 of file HitAnaModule_module.cc.

Constructor & Destructor Documentation

hit::HitAnaModule::HitAnaModule ( fhicl::ParameterSet const &  p )

explicit

Definition at line 78 of file HitAnaModule_module.cc.

                                                        : EDAnalyzer(p) // ,
// More initializers here.
{
  fHitModuleLabels = p.get<std::vector<std::string>>("HitModuleLabels");
  fWireModuleLabel = p.get<std::string>("WireModuleLabel");
  fMCHitModuleLabel = p.get<std::string>("MCHitModuleLabel");
}

Member Function Documentation

void hit::HitAnaModule::analyze ( art::Event const &  e )

overrideprivate

Definition at line 135 of file HitAnaModule_module.cc.

{
  //get event and run numbers
  unsigned int eventNumber = e.id().event();
  unsigned int runNumber = e.run();

  analysisAlg.ClearHitModules();

  // get the data
  auto const& wireVector = *e.getValidHandle<std::vector<recob::Wire>>(fWireModuleLabel);
  auto const& mcHitVector = *e.getValidHandle<std::vector<sim::MCHitCollection>>(fMCHitModuleLabel);

  // make the association vector. First index is wire index, second is
  // mcHitCollection index
  std::vector<std::vector<int>> WireMCHitAssocVector;
  createMCAssocVector(wireVector, mcHitVector, WireMCHitAssocVector);

  //get the hit data
  size_t nHitModules = fHitModuleLabels.size();
  std::vector<art::Handle<std::vector<recob::Hit>>> hitHandles(nHitModules);
  // for each hit module output (first index), for each wire (second index)
  // the list of hits associated with that wire is stored
  std::vector<std::vector<std::vector<int>>> WireHitAssocVectors(nHitModules);
  for (size_t iter = 0; iter < nHitModules; iter++) {

    e.getByLabel(fHitModuleLabels[iter], hitHandles[iter]);

    //create association vectors by hand for now
    //art::ValidHandle<HitWireAssns_t> HitToWireAssns
    //= e.getValidHandle<HitWireAssns_t>(fHitModuleLabels[iter]);
    //WireHitAssocVectors[iter].resize(wireVector.size());
    //createAssocVector(*HitToWireAssns,WireHitAssocVectors[iter]);

    WireHitAssocVectors[iter].resize(wireVector.size());
    art::Handle<HitWireAssns_t> HitToWireAssns;
    if (e.getByLabel(fHitModuleLabels[iter], HitToWireAssns))
      createAssocVector(*HitToWireAssns, WireHitAssocVectors[iter]);
    else
      createAssocVector(wireVector, *(hitHandles[iter]), WireHitAssocVectors[iter]);

    //load in this hit/assoc pair
    analysisAlg.LoadHitAssocPair(
      *(hitHandles[iter]), WireHitAssocVectors[iter], fHitModuleLabels[iter]);
  }

  // get time service
  auto const clock_data = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(e);

  // run the analzyer alg
  analysisAlg.AnalyzeWires(
    wireVector, mcHitVector, WireMCHitAssocVector, clock_data, eventNumber, runNumber);
}

void hit::HitAnaModule::beginJob ( )

overrideprivate

Definition at line 189 of file HitAnaModule_module.cc.

{
  art::ServiceHandle<art::TFileService const> tfs;
  wireDataTree = tfs->make<TTree>("wireDataTree", "WireDataTree");
  analysisAlg.SetWireDataTree(wireDataTree);

  // The below creates a Tree with one branch - a recob::Hit branch - for each
  // Hit module specified in the fcl file. So, don't run this module once per Hit
  // Finder as one normally would. Just run it once and specify all HitFinders.
  // This was the design for the WireDataTree; we follow it here for the
  // Hit trees.
  for (auto const& label : fHitModuleLabels) {
    std::string firstArg("hitData_");
    firstArg += label;
    std::string secArg("HitDataTree_");
    secArg += label;
    TTree* intermediateTree = tfs->make<TTree>(firstArg.c_str(), secArg.c_str());
    hitDataTree.push_back(intermediateTree);
  }
  analysisAlg.SetHitDataTree(hitDataTree);
}

void hit::HitAnaModule::createAssocVector ( HitWireAssns_t const &  HitToWire, std::vector< std::vector< int >> &  WireHitAssocVector  )

private

Definition at line 87 of file HitAnaModule_module.cc.

{
  // WireHitAssocVector: for each wire, indices of all the hits associated to it

  // the iteration to art::Assns<Hit, Wire> points to a art::Ptr pair (assn_t)
  // with a hit as first element ("left") and a wire as the second one ("right")
  for (HitWireAssns_t::assn_t const& assn : HitToWire)
    WireHitAssocVector.at(assn.second.key()).push_back(assn.first.key());
}

void hit::HitAnaModule::createAssocVector ( std::vector< recob::Wire > const &  wireVector, std::vector< recob::Hit > const &  hitVector, std::vector< std::vector< int >> &  WireHitAssocVector  )

private

Definition at line 99 of file HitAnaModule_module.cc.

{
  WireHitAssocVector.resize(wireVector.size());
  for (size_t iwire = 0; iwire < wireVector.size(); iwire++) {
    for (size_t ihit = 0; ihit < hitVector.size(); ihit++) {
      if (hitVector[ihit].Channel() == wireVector[iwire].Channel())
        WireHitAssocVector[iwire].push_back(ihit);
    }
  }
}

void hit::HitAnaModule::createMCAssocVector ( std::vector< recob::Wire > const &  wireVector, std::vector< sim::MCHitCollection > const &  mcHitVector, std::vector< std::vector< int >> &  WireMCHitAssocVector  )

private

Definition at line 113 of file HitAnaModule_module.cc.

{

  WireMCHitAssocVector.clear();
  WireMCHitAssocVector.resize(wireVector.size());

  //first, store all the MCHitCollection indices in a map keyed on channel
  //then, loop through wires, and lookup mchitcollections based on the wire's channel

  std::map<unsigned int, std::vector<int>> mcHitIndicesByChannel;
  for (unsigned int icol = 0; icol < mcHitVector.size(); icol++)
    mcHitIndicesByChannel[mcHitVector[icol].Channel()].push_back(icol);

  for (unsigned int iwire = 0; iwire < wireVector.size(); iwire++)
    WireMCHitAssocVector[iwire].insert(WireMCHitAssocVector[iwire].end(),
                                       mcHitIndicesByChannel[wireVector[iwire].Channel()].begin(),
                                       mcHitIndicesByChannel[wireVector[iwire].Channel()].end());
}

Member Data Documentation

HitAnaAlg hit::HitAnaModule::analysisAlg

private

Definition at line 75 of file HitAnaModule_module.cc.

std::vector<std::string> hit::HitAnaModule::fHitModuleLabels

private

Definition at line 68 of file HitAnaModule_module.cc.

std::string hit::HitAnaModule::fMCHitModuleLabel

private

Definition at line 69 of file HitAnaModule_module.cc.

std::string hit::HitAnaModule::fWireModuleLabel

private

Definition at line 70 of file HitAnaModule_module.cc.

std::vector<TTree*> hit::HitAnaModule::hitDataTree

private

Definition at line 73 of file HitAnaModule_module.cc.

TTree* hit::HitAnaModule::wireDataTree

private

Definition at line 72 of file HitAnaModule_module.cc.

---

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

• HitAnaModule_module.cc