DigitizerSaturationFilter_module.cc

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// Class:       DigitizerSaturationFilter
// Plugin Type: filter (art v2_12_01)
// File:        DigitizerSaturationFilter_module.cc
//
// Aidan Medcalf (University of Dallas) <amedcalf@fnal.gov>
//
// Discard events which saturate (clip) the digitizer
////////////////////////////////////////////////////////////////////////

#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/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 "art/Framework/Services/Optional/TFileService.h"
#include "art/Framework/Services/Optional/TFileDirectory.h"
#include "canvas/Persistency/Common/FindManyP.h"

#include "BeamlineRecoBase/BeamlineDigit.h"
#include "BeamlineRecoBase/ToF.h"
#include "RawData/RawBeamline.h"
#include "Utilities/AssociationUtil.h"

#include <iostream>
#include <map>

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

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

  private:
    std::vector<unsigned int> fChannels;
    art::InputTag fRawDataLabel;
    double        fThresholdADC;
    bool          fPulseIsNegative;
  };
}

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

void beamlinereco::DigitizerSaturationFilter::reconfigure(const fhicl::ParameterSet & pset) {
  fChannels        = pset.get<std::vector<unsigned int>>("Channels");
  fRawDataLabel    = pset.get<art::InputTag>("RawDataLabel");
  fThresholdADC    = pset.get<double>("ThresholdADC", 10);
  fPulseIsNegative = pset.get<bool>("PulseIsNegative", true);
}

bool beamlinereco::DigitizerSaturationFilter::filter(art::Event & e)
{
  // Get raw digits
  art::Handle<std::vector<rawdata::RawBeamlineDigit>> digitHandle;
  std::vector<art::Ptr<rawdata::RawBeamlineDigit>> digits;
  if (e.getByLabel(fRawDataLabel, digitHandle)) art::fill_ptr_vector(digits, digitHandle);

  for (auto d : digits) {
    // Skip if not in fChannels
    if (std::find_if(fChannels.begin(), fChannels.end(), [d](unsigned int ch){ return d->ChannelID().Channel == ch; } ) == fChannels.end()) {
      continue;
    }

    std::vector<uint16_t> adcs = d->ADCs();

    for (size_t i=0; i<adcs.size(); i++) {
      if (fPulseIsNegative) {
        if (adcs[i] < fThresholdADC) {
          //std::cout << e.run() << " " << e.subRun() << " " << e.event() << " ch" << d->Channel() << " below threshold" << std::endl;
          return false;
        }
      } else {
        if (adcs[i] > fThresholdADC) {
          return false;
        }
      }
    }
  }

  return true;
}


DEFINE_ART_MODULE(beamlinereco::DigitizerSaturationFilter)