GoodBeamlineEventFilter_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// Class:       GoodBeamlineEventFilter
// Plugin Type: filter (art v2_12_01)
// File:        GoodBeamlineEventFilter_module.cc
//
// John Rabaey::jrabaey@udallas.edu
//
// Description: ToF filter module
////////////////////////////////////////////////////////////////////////

#include "art/Framework/Core/EDFilter.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 "canvas/Persistency/Common/Ptr.h"
#include "canvas/Persistency/Common/PtrVector.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"

#include "BeamlineRecoBase/BeamlineDigit.h"
#include "BeamlineRecoBase/BeamlinePID.h"
#include "BeamlineRecoBase/ToF.h"
#include "BeamlineRecoBase/WCTrack.h"

#include <iostream>

namespace beamlinereco {
  class GoodBeamlineEventFilter : public art::EDFilter {
  public:
    explicit GoodBeamlineEventFilter(fhicl::ParameterSet const &p);

    bool filter(art::Event &e);
    void reconfigure(const fhicl::ParameterSet &p);

  private:
    art::InputTag fToFRecoLabel;
    art::InputTag fWCRecoLabel;
    art::InputTag fCherenkovRecoLabel;
    bool          fRequireCherenkov;
    bool          fRequireWC;
    bool          fRequireToF;

  };
}

beamlinereco::GoodBeamlineEventFilter::GoodBeamlineEventFilter(fhicl::ParameterSet const & p)
{
  reconfigure(p);
}

void beamlinereco::GoodBeamlineEventFilter::reconfigure(const fhicl::ParameterSet & pset) {
  fToFRecoLabel       = pset.get<art::InputTag>("ToFRecoLabel");
  fWCRecoLabel        = pset.get<art::InputTag>("WCRecoLabel");
  fCherenkovRecoLabel = pset.get<art::InputTag>("CherenkovRecoLabel");
  fRequireToF         = pset.get<bool>("RequireToF");
  fRequireWC          = pset.get<bool>("RequireWC");
  fRequireCherenkov   = pset.get<bool>("RequireCherenkov");
}

bool beamlinereco::GoodBeamlineEventFilter::filter(art::Event & e)
{
  // Get ToF objects
  art::Handle<std::vector<brb::ToF>> tofHandle;
  std::vector<art::Ptr<brb::ToF>> tofs;
  if (e.getByLabel(fToFRecoLabel, tofHandle)) art::fill_ptr_vector(tofs, tofHandle);

  // Cherenkov
  art::Handle<std::vector<brb::BeamlineDigit>> cherenkovHandle;
  std::vector<art::Ptr<brb::BeamlineDigit>>    cherenkovs;
  if (e.getByLabel(fCherenkovRecoLabel, cherenkovHandle))
    art::fill_ptr_vector(cherenkovs, cherenkovHandle);

  // Get WC reco
  art::Handle<std::vector<brb::WCTrack>> wcHandle;
  std::vector<art::Ptr<brb::WCTrack>>    wcs;
  if (e.getByLabel(fWCRecoLabel, wcHandle))
    art::fill_ptr_vector(wcs, wcHandle);

  bool hasToF       = tofs.size() > 0;
  bool hasWC        = wcs.size() > 0;
  bool hasCherenkov = cherenkovs.size() > 0;

  if ((!fRequireToF       || hasToF) &&
      (!fRequireWC        || hasWC)  &&
      (!fRequireCherenkov || hasCherenkov)) {
    return true;
  }

  return false;
}


DEFINE_ART_MODULE(beamlinereco::GoodBeamlineEventFilter)