AreaSeededTrackingRegionsProducer.h

Go to the documentation of this file.

#ifndef RecoTracker_TkTrackingRegions_AreaSeededTrackingRegionsProducer_h
#define RecoTracker_TkTrackingRegions_AreaSeededTrackingRegionsProducer_h


#include "RecoTracker/TkTrackingRegions/interface/RectangularEtaPhiTrackingRegion.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Math/interface/PtEtaPhiMass.h"

#include "VertexBeamspotOrigins.h"
#include "AreaSeededTrackingRegionsBuilder.h"

#include <array>
#include <limits>

class AreaSeededTrackingRegionsProducer {
public:

  typedef enum {BEAM_SPOT_FIXED, BEAM_SPOT_SIGMA, VERTICES_FIXED, VERTICES_SIGMA } Mode;

  AreaSeededTrackingRegionsProducer(const edm::ParameterSet& conf, edm::ConsumesCollector&& iC):
    m_origins(conf.getParameter<edm::ParameterSet>("RegionPSet"), iC),
    m_builder(conf.getParameter<edm::ParameterSet>("RegionPSet"), iC)
  {
    edm::ParameterSet regPSet = conf.getParameter<edm::ParameterSet>("RegionPSet");
    for(const auto& area: regPSet.getParameter<std::vector<edm::ParameterSet> >("areas")) {
      m_areas.emplace_back(area.getParameter<double>("rmin"),
                           area.getParameter<double>("rmax"),
                           area.getParameter<double>("phimin"),
                           area.getParameter<double>("phimax"),
                           area.getParameter<double>("zmin"),
                           area.getParameter<double>("zmax"));
    }
    if(m_areas.empty())
      throw cms::Exception("Configuration") << "Empty 'areas' parameter.";
  }
  
  ~AreaSeededTrackingRegionsProducer() = default;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;

    edm::ParameterSetDescription descAreas;
    descAreas.add<double>("rmin", 0.0);
    descAreas.add<double>("rmax", 0.0);
    descAreas.add<double>("zmin", 0.0);
    descAreas.add<double>("zmax", 0.0);
    descAreas.add<double>("phimin", 0.0);
    descAreas.add<double>("phimax", 0.0);
    std::vector<edm::ParameterSet> vDefaults;
    desc.addVPSet("areas", descAreas, vDefaults);

    VertexBeamspotOrigins::fillDescriptions(desc);
    AreaSeededTrackingRegionsBuilder::fillDescriptions(desc);

    // Only for backwards-compatibility
    edm::ParameterSetDescription descRegion;
    descRegion.add<edm::ParameterSetDescription>("RegionPSet", desc);

    descriptions.add("areaSeededTrackingRegion", descRegion);
  }

  std::vector<std::unique_ptr<TrackingRegion> > regions(const edm::Event& e, const edm::EventSetup& es) const
  {
    auto origins = m_origins.origins(e);
    auto builder = m_builder.beginEvent(e);
    return builder.regions(origins, m_areas);
  }
  
private:
  VertexBeamspotOrigins m_origins;
  AreaSeededTrackingRegionsBuilder m_builder;
  std::vector<AreaSeededTrackingRegionsBuilder::Area> m_areas;
};

#endif