HSCParticleProducer.h

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    HSCParticleProducer
// Class:      HSCParticleProducer
//
//
// Original Author:  Loic Quertenmont
//         Created:  Wed Oct 10 12:01:28 CEST 2007

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "CommonTools/UtilAlgos/interface/DeltaR.h"

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "SUSYBSMAnalysis/HSCP/interface/BetaCalculatorTK.h"
#include "SUSYBSMAnalysis/HSCP/interface/BetaCalculatorMUON.h"
#include "SUSYBSMAnalysis/HSCP/interface/BetaCalculatorRPC.h"
#include "SUSYBSMAnalysis/HSCP/interface/BetaCalculatorECAL.h"
#include "SUSYBSMAnalysis/HSCP/interface/CandidateSelector.h"

#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h"
#include "Math/GenVector/VectorUtil.h"

#include <TNtuple.h>
#include <TF1.h>

#include <vector>
#include <iostream>

//
// class decleration
//
class HSCParticleProducer : public edm::EDFilter {
public:
  explicit HSCParticleProducer(const edm::ParameterSet&);
  ~HSCParticleProducer() override;

private:
  void beginJob() override;
  bool filter(edm::Event&, const edm::EventSetup&) override;
  void endJob() override;

  std::vector<susybsm::HSCParticle> getHSCPSeedCollection(edm::Handle<reco::TrackCollection>& trackCollectionHandle,
                                                          edm::Handle<reco::MuonCollection>& muonCollectionHandle,
                                                          edm::Handle<reco::MuonCollection>& MTmuonCollectionHandle);

  // ----------member data ---------------------------
  bool Filter_;

  edm::EDGetTokenT<reco::TrackCollection> m_trackToken;
  edm::EDGetTokenT<reco::TrackCollection> m_trackIsoToken;
  edm::EDGetTokenT<reco::MuonCollection> m_muonsToken;
  edm::EDGetTokenT<reco::MuonCollection> m_MTmuonsToken;

  bool useBetaFromTk;
  bool useBetaFromMuon;
  bool useBetaFromRpc;
  bool useBetaFromEcal;

  float minTkP;
  float maxTkChi2;
  unsigned int minTkHits;
  float minMuP;
  float minSAMuPt;
  float minMTMuPt;
  float minDR;
  float minMTDR;
  float maxInvPtDiff;

  BetaCalculatorTK* beta_calculator_TK;
  BetaCalculatorMUON* beta_calculator_MUON;
  BetaCalculatorRPC* beta_calculator_RPC;
  BetaCalculatorECAL* beta_calculator_ECAL;

  std::vector<CandidateSelector*> Selectors;
};