TrackContainmentTagger_module.cc

Go to the documentation of this file.

// Class:       TrackContainmentTagger
// Module Type: producer
// File:        TrackContainmentTagger_module.cc
//

#include "art/Framework/Core/EDProducer.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"

#include "TrackContainment/TrackContainmentAlg.hh"
#include "larcore/Geometry/Geometry.h"
#include "lardata/Utilities/AssociationUtil.h"

namespace trk {
  class TrackContainmentTagger;
}

class trk::TrackContainmentTagger : public art::EDProducer {
public:
  explicit TrackContainmentTagger(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.
  TrackContainmentTagger(TrackContainmentTagger const&) = delete;
  TrackContainmentTagger(TrackContainmentTagger&&) = delete;
  TrackContainmentTagger& operator=(TrackContainmentTagger const&) = delete;
  TrackContainmentTagger& operator=(TrackContainmentTagger&&) = delete;

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

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

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

trk::TrackContainmentTagger::TrackContainmentTagger(fhicl::ParameterSet const& p) : EDProducer{p}
{
  art::ServiceHandle<art::TFileService const> tfs;
  fAlg.SetupOutputTree(tfs->make<TTree>("myanatree", "MyAnalysis Tree"));

  fAlg.Configure(p.get<fhicl::ParameterSet>("TrackContainmentAlg"));
  fTrackModuleLabels = p.get<std::vector<std::string>>("TrackModuleLabels");
  fApplyTags =
    p.get<std::vector<bool>>("ApplyTags", std::vector<bool>(fTrackModuleLabels.size(), true));

  if (fApplyTags.size() != fTrackModuleLabels.size())
    throw cet::exception("TrackContainmentTagger::TrackContainmentTagger")
      << "ApplyTags not same size as TrackModuleLabels. ABORT!!!";

  fAlg.setMakeCosmicTags();

  produces<std::vector<anab::CosmicTag>>();
  produces<art::Assns<recob::Track, anab::CosmicTag>>();
}

void trk::TrackContainmentTagger::produce(art::Event& e)
{

  std::unique_ptr<std::vector<anab::CosmicTag>> cosmicTagTrackVector(
    new std::vector<anab::CosmicTag>);
  std::unique_ptr<art::Assns<recob::Track, anab::CosmicTag>> assnOutCosmicTagTrack(
    new art::Assns<recob::Track, anab::CosmicTag>);

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

  std::vector<std::vector<recob::Track>> trackVectors;
  std::vector<art::Handle<std::vector<recob::Track>>> trackHandles;
  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);
    trackHandles.push_back(trackHandle);
  }

  art::ServiceHandle<geo::Geometry const> geoHandle;
  fAlg.ProcessTracks(trackVectors, *geoHandle);

  auto const& cosmicTags = fAlg.GetTrackCosmicTags();

  for (size_t i_tc = 0; i_tc < cosmicTags.size(); ++i_tc) {
    if (!fApplyTags[i_tc]) continue;
    for (size_t i_t = 0; i_t < fAlg.GetTrackCosmicTags()[i_tc].size(); ++i_t) {
      cosmicTagTrackVector->emplace_back(fAlg.GetTrackCosmicTags()[i_tc][i_t]);
      util::CreateAssn(e,
                       *cosmicTagTrackVector,
                       art::Ptr<recob::Track>(trackHandles[i_tc], i_t),
                       *assnOutCosmicTagTrack);
    }
  }

  e.put(std::move(cosmicTagTrackVector));
  e.put(std::move(assnOutCosmicTagTrack));
}

DEFINE_ART_MODULE(trk::TrackContainmentTagger)