RecoJets/FFTJetProducers/interface/FFTJetInterface.h - master - githubmirror/cmssw

RecoJets/FFTJetProducers/interface/FFTJetInterface.h
119 строк3 KB
Igor
Changing FFTJet modules to stream modules
18 май 2022, 15:54
18 май 2022, 15:547be6095
// -*- C++ -*-
//
// Package:    FFTJetProducers
// Class:      FFTJetInterface
//
/**\class FFTJetInterface FFTJetInterface.h RecoJets/FFTJetProducers/interface/FFTJetInterface.hh

 Description: common facilities for the FFTJet interface code

*/
//
// Original Author:  Igor Volobouev
//         Created:  June 29 2010
//
//

#ifndef RecoJets_FFTJetProducers_FFTJetInterface_h
#define RecoJets_FFTJetProducers_FFTJetInterface_h

#include <memory>
#include <vector>
#include <cassert>

// FFTJet headers
#include "fftjet/Grid2d.hh"

// framework include files
#include "FWCore/ParameterSet/interface/ParameterSetfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"

#include "DataFormats/Candidate/interface/Particle.h"

#include "RecoJets/JetProducers/interface/AnomalousTower.h"

#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Common/interface/Handle.h"

#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

// local FFTJet-related definitions
#include "RecoJets/FFTJetAlgorithms/interface/fftjetTypedefs.h"
#include "RecoJets/FFTJetProducers/interface/JetType.h"

//
// class declaration
//
namespace fftjetcms {
  class FFTJetInterface : public edm::stream::EDProducer<> {
  public:
    // Explicitly disable other ways to construct this object
    FFTJetInterface() = delete;
    FFTJetInterface(const FFTJetInterface&) = delete;
    FFTJetInterface& operator=(const FFTJetInterface&) = delete;
    ~FFTJetInterface() override {}

  protected:
    explicit FFTJetInterface(const edm::ParameterSet&);

    template <class Ptr>
    void checkConfig(const Ptr& ptr, const char* message) {
      if (ptr.get() == nullptr)
        throw cms::Exception("FFTJetBadConfig") << message << std::endl;
    }

    void loadInputCollection(const edm::Event&);
    void discretizeEnergyFlow();
    double getEventScale() const;
    bool storeInSinglePrecision() const;

    const reco::Particle::Point& vertexUsed() const { return vertex_; }

    // Label for the input collection
    const edm::InputTag inputLabel;

    // Label for the output objects
    const std::string outputLabel;

    // Jet type to produce
    const JetType jetType;

    // Vertex correction-related stuff
    const bool doPVCorrection;

    // Label for the vertices
    const edm::InputTag srcPVs;

    // Try to equalize magnitudes in the energy flow grid?
    const std::vector<double> etaDependentMagnutideFactors;

    // Functor for finding anomalous towers
    const AnomalousTower anomalous;

    // Event data 4-vectors
    std::vector<fftjetcms::VectorLike> eventData;

    // Candidate number which corresponds to the item in the "eventData"
    std::vector<unsigned> candidateIndex;

    // The energy discretization grid
    std::unique_ptr<fftjet::Grid2d<fftjetcms::Real> > energyFlow;

    // The input handle for the collection of candidates
    edm::Handle<reco::CandidateView> inputCollection;

  private:
    const bool insertCompleteEvent;
    const double completeEventScale;
    reco::Particle::Point vertex_;

    edm::EDGetTokenT<reco::CandidateView> inputToken;
    edm::EDGetTokenT<reco::VertexCollection> srcPVsToken;
  };
}  // namespace fftjetcms

#endif  // RecoJets_FFTJetProducers_FFTJetInterface_h