Alignment/MuonAlignmentAlgorithms/plugins/CSCOverlapsBeamSplashCut.cc - master - githubmirror/cmssw

Alignment/MuonAlignmentAlgorithms/plugins/CSCOverlapsBeamSplashCut.cc

95 строк3 KB

Sunanda

Use tokens rather than labels in accessing collections within Alignment?MuonAlignmentAlgorithms

30 окт 2022, 08:16

30 окт 2022, 08:160e194d1

// -*- C++ -*-
//
// Package:    CSCOverlapsBeamSplashCut
// Class:      CSCOverlapsBeamSplashCut
//
/**\class CSCOverlapsBeamSplashCut CSCOverlapsBeamSplashCut.cc Alignment/CSCOverlapsBeamSplashCut/src/CSCOverlapsBeamSplashCut.cc

 Description: <one line class summary>

 Implementation:
     <Notes on implementation>
*/
//
// Original Author:  Jim Pivarski
//         Created:  Sat Nov 21 21:18:04 CET 2009
//

// system include files
#include <memory>

// 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/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"

// references
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "TH1F.h"

//
// class decleration
//

class CSCOverlapsBeamSplashCut : public edm::one::EDFilter<edm::one::SharedResources> {
public:
  explicit CSCOverlapsBeamSplashCut(const edm::ParameterSet&);
  ~CSCOverlapsBeamSplashCut() override = default;
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  bool filter(edm::Event&, const edm::EventSetup&) override;
  // ----------member data ---------------------------
  const edm::InputTag m_src;
  const int m_maxSegments;
  edm::EDGetTokenT<CSCSegmentCollection> cscToken_;
  TH1F* m_numSegments;
};

//
// constructors and destructor
//
CSCOverlapsBeamSplashCut::CSCOverlapsBeamSplashCut(const edm::ParameterSet& iConfig)
    : m_src(iConfig.getParameter<edm::InputTag>("src")),
      m_maxSegments(iConfig.getParameter<int>("maxSegments")),
      cscToken_(consumes<CSCSegmentCollection>(m_src)) {
  usesResource(TFileService::kSharedResource);
  edm::Service<TFileService> tFileService;
  m_numSegments = tFileService->make<TH1F>("numSegments", "", 201, -0.5, 200.5);
}

//
// member functions
//
void CSCOverlapsBeamSplashCut::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("src", edm::InputTag("ALCARECOMuAlBeamHaloOverlaps"));
  desc.add<int>("maxSegments", 4);
  descriptions.add("cscOverlapsBeamSplashCut", desc);
}

// ------------ method called to produce the data  ------------
bool CSCOverlapsBeamSplashCut::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  const edm::Handle<CSCSegmentCollection>& cscSegments = iEvent.getHandle(cscToken_);

  m_numSegments->Fill(cscSegments->size());

  if (m_maxSegments < 0)
    return true;

  else if (static_cast<int>(cscSegments->size()) <= m_maxSegments)
    return true;

  else
    return false;
}

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