PixelUnpackingRegions.h

Go to the documentation of this file.

#ifndef PixelUnpackingRegions_H
#define PixelUnpackingRegions_H

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"

#include "CondFormats/DataRecord/interface/SiPixelFedCablingMapRcd.h"
#include "CondFormats/SiPixelObjects/interface/SiPixelFedCablingMap.h"
#include "CondFormats/SiPixelObjects/interface/SiPixelFedCablingTree.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include <cmath>
#include <vector>
#include <set>

class PixelUnpackingRegions {
public:
  struct Region {
    Region(const math::XYZVector& dir, float dphi = 0.5f, float maxz = 24.f) : v(dir), dPhi(dphi), maxZ(maxz) {
      cosphi = v.x() / v.rho();
      sinphi = v.y() / v.rho();
      atantheta = v.z() / v.rho();
    }
    math::XYZVector v;
    float dPhi, maxZ;
    float cosphi, sinphi, atantheta;
  };

  PixelUnpackingRegions(const edm::ParameterSet&, edm::ConsumesCollector&& iC);

  ~PixelUnpackingRegions() {}

  void run(const edm::Event& e, const edm::EventSetup& es);

  bool mayUnpackFED(unsigned int fed_n) const;

  bool mayUnpackModule(unsigned int id) const;

  const std::set<unsigned int>* modulesToUnpack() const { return &modules_; }

  unsigned int nFEDs() const { return feds_.size(); }
  unsigned int nModules() const { return modules_.size(); }
  unsigned int nBarrelModules() const;
  unsigned int nForwardModules() const;
  unsigned int nRegions() const { return nreg_; }

  struct Module {
    float phi;
    float x, y, z;
    unsigned int id;
    unsigned int fed;

    Module() {}
    Module(float ph) : phi(ph), x(0.f), y(0.f), z(0.f), id(0), fed(0) {}

    bool operator<(const Module& m) const {
      if (phi < m.phi)
        return true;
      if (phi == m.phi && id < m.id)
        return true;
      return false;
    }
  };

private:
  // input parameters
  std::vector<edm::InputTag> inputs_;
  std::vector<double> dPhi_;
  std::vector<double> maxZ_;
  edm::InputTag beamSpotTag_;

  edm::EDGetTokenT<reco::BeamSpot> tBeamSpot;
  std::vector<edm::EDGetTokenT<reco::CandidateView>> tCandidateView;

  std::set<unsigned int> feds_;
  std::set<unsigned int> modules_;
  unsigned int nreg_;

  void initialize(const edm::EventSetup& es);

  // add a new direction of a region of interest
  void addRegion(Region& r);

  // gather info into feds_ and modules_ from a range of a Module vector
  void gatherFromRange(Region& r, std::vector<Module>::const_iterator, std::vector<Module>::const_iterator);

  // addRegion for a local (BPIX or +-FPIX) container
  void addRegionLocal(Region& r, std::vector<Module>& container, const Module& lo, const Module& hi);

  // local containers of barrel and endcaps Modules sorted by phi
  std::vector<Module> phiBPIX_;
  std::vector<Module> phiFPIXp_;
  std::vector<Module> phiFPIXm_;

  std::unique_ptr<SiPixelFedCablingTree> cabling_;
  math::XYZPoint beamSpot_;

  edm::ESWatcher<SiPixelFedCablingMapRcd> watcherSiPixelFedCablingMap_;
};

#endif