FEDBadModuleFilter

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172
// -*- C++ -*-
//
// Package:    SiStripTools
// Class:      FEDBadModuleFilter
//
/**\class FEDBadModuleFilter FEDBadModuleFilter.cc DPGAnalysis/SiStripTools/plugins/FEDBadModuleFilter.cc

 Description: template EDFilter to select events with selected modules with SiStripDigis or SiStripClusters

 Implementation:

*/
//
// Original Author:  Andrea Venturi
//         Created:  Wed Oct 22 17:54:30 CEST 2008
//
//

// system include files
#include <memory>
#include "TProfile.h"

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDFilter.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include <vector>
#include <set>

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/DetId/interface/DetIdVector.h"

#include "DPGAnalysis/SiStripTools/interface/RunHistogramManager.h"
#include "CommonTools/UtilAlgos/interface/DetIdSelector.h"

//
// class declaration
//

class FEDBadModuleFilter : public edm::one::EDFilter<edm::one::WatchRuns> {
public:
  explicit FEDBadModuleFilter(const edm::ParameterSet&);
  ~FEDBadModuleFilter() override;

private:
  void beginJob() override;
  bool filter(edm::Event&, const edm::EventSetup&) override;
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void endRun(const edm::Run&, const edm::EventSetup&) override {}
  void endJob() override;

  // ----------member data ---------------------------

  edm::EDGetTokenT<DetIdVector> m_digibadmodulecollectionToken;
  unsigned int m_modulethr;
  std::set<unsigned int> m_modules;
  DetIdSelector m_modsel;
  bool m_wantedhist;
  bool m_printlist;
  const unsigned int m_maxLS;
  const unsigned int m_LSfrac;
  RunHistogramManager m_rhm;
  TH1F** m_nbadrun;
  TProfile** m_nbadvsorbrun;
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
FEDBadModuleFilter::FEDBadModuleFilter(const edm::ParameterSet& iConfig)
    : m_digibadmodulecollectionToken(consumes(iConfig.getParameter<edm::InputTag>("collectionName"))),
      m_modulethr(iConfig.getParameter<unsigned int>("badModThr")),
      m_modsel(),
      m_wantedhist(iConfig.getUntrackedParameter<bool>("wantedHisto", false)),
      m_printlist(iConfig.getUntrackedParameter<bool>("printList", false)),
      m_maxLS(iConfig.getUntrackedParameter<unsigned int>("maxLSBeforeRebin", 100)),
      m_LSfrac(iConfig.getUntrackedParameter<unsigned int>("startingLSFraction", 4)),
      m_rhm(consumesCollector())

{
  //now do what ever initialization is needed

  if (m_wantedhist) {
    m_nbadrun = m_rhm.makeTH1F("nbadrun", "Number of bad channels", 500, -0.5, 499.5);
    m_nbadvsorbrun =
        m_rhm.makeTProfile("nbadvsorbrun", "Number of bad channels vs time", m_LSfrac * m_maxLS, 0, m_maxLS * 262144);
  }

  std::vector<unsigned int> modules =
      iConfig.getUntrackedParameter<std::vector<unsigned int> >("moduleList", std::vector<unsigned int>());
  m_modules = std::set<unsigned int>(modules.begin(), modules.end());

  if (iConfig.exists("moduleSelection")) {
    m_modsel = DetIdSelector(iConfig.getUntrackedParameter<edm::ParameterSet>("moduleSelection"));
  }
}

FEDBadModuleFilter::~FEDBadModuleFilter() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

//
// member functions
//

// ------------ method called on each new Event  ------------
bool FEDBadModuleFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;

  auto badmodules = iEvent.getHandle(m_digibadmodulecollectionToken);

  unsigned int nbad = 0;
  if (m_printlist || !m_modules.empty() || m_modsel.isValid()) {
    for (auto const& mod : *badmodules) {
      if ((m_modules.empty() || m_modules.find(mod) != m_modules.end()) &&
          (!m_modsel.isValid() || m_modsel.isSelected(mod))) {
        ++nbad;
        if (m_printlist)
          edm::LogInfo("FEDBadModule") << mod;
      }
    }
  } else {
    nbad = badmodules->size();
  }

  if (m_wantedhist) {
    if (m_nbadvsorbrun && *m_nbadvsorbrun)
      (*m_nbadvsorbrun)->Fill(iEvent.orbitNumber(), nbad);
    if (m_nbadrun && *m_nbadrun)
      (*m_nbadrun)->Fill(nbad);
  }

  return (nbad >= m_modulethr);
}

// ------------ method called once each job just before starting event loop  ------------
void FEDBadModuleFilter::beginJob() {}

// ------------ method called once each job just after ending the event loop  ------------
void FEDBadModuleFilter::endJob() {}

void FEDBadModuleFilter::beginRun(const edm::Run& iRun, const edm::EventSetup& iSetup) {
  if (m_wantedhist) {
    m_rhm.beginRun(iRun);
    if (*m_nbadvsorbrun) {
      (*m_nbadvsorbrun)->SetCanExtend(TH1::kXaxis);
      (*m_nbadvsorbrun)->GetXaxis()->SetTitle("time [Orb#]");
      (*m_nbadvsorbrun)->GetYaxis()->SetTitle("Bad Channels");
    }
  }
}

//define this as a plug-in
DEFINE_FWK_MODULE(FEDBadModuleFilter);