TrackContainmentAnalyzer_module.cc

Go to the documentation of this file.

// Class:       TrackContainmentAnalyzer
// Module Type: analyzer
// File:        TrackContainmentAnalyzer_module.cc
//
// Generated at Tue Nov 10 13:06:09 2015 by Wesley Ketchum using artmod
// from cetpkgsupport v1_08_07.

#include "TrackContainment/TrackContainmentAlg.hh"
#include "larcore/Geometry/Geometry.h"
#include "larcore/Geometry/WireReadout.h"

#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_root_io/TFileService.h"
#include "fhiclcpp/ParameterSet.h"

#include "TTree.h"

namespace trk {
  class TrackContainmentAnalyzer;
}

class trk::TrackContainmentAnalyzer : public art::EDAnalyzer {
public:
  explicit TrackContainmentAnalyzer(fhicl::ParameterSet const& p);
  // The destructor generated by the compiler is fine for classes
  // without bare pointers or other resource use.

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

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

private:
  // Declare member data here.
  trk::TrackContainmentAlg fAlg;

  std::vector<std::string> fTrackModuleLabels;
};

trk::TrackContainmentAnalyzer::TrackContainmentAnalyzer(fhicl::ParameterSet const& p)
  : EDAnalyzer(p)
{
  fAlg.Configure(p.get<fhicl::ParameterSet>("TrackContainmentAlg"));
  fTrackModuleLabels = p.get<std::vector<std::string>>("TrackModuleLabels");
  fAlg.setFillOutputTree();
  art::ServiceHandle<art::TFileService const> tfs;
  fAlg.SetupOutputTree(tfs->make<TTree>("myanatree", "MyAnalysis Tree"));
}

void trk::TrackContainmentAnalyzer::analyze(art::Event const& e)
{

  fAlg.SetRunEvent(e.run(), e.event());

  std::vector<std::vector<recob::Track>> trackVectors;
  for (size_t i_l = 0; i_l < fTrackModuleLabels.size(); ++i_l) {
    art::Handle<std::vector<recob::Track>> trackHandle;
    e.getByLabel(fTrackModuleLabels[i_l], trackHandle);
    trackVectors.push_back(*trackHandle);
  }

  art::ServiceHandle<geo::Geometry const> geoHandle;

  fAlg.ProcessTracks(trackVectors, *geoHandle, art::ServiceHandle<geo::WireReadout>()->Get());
}

DEFINE_ART_MODULE(trk::TrackContainmentAnalyzer)