SiPixelPhase1RecHits.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     SiPixelPhase1RecHits
// Class:       SiPixelPhase1RecHits
//

// Original Author: Marcel Schneider

#include "DQM/SiPixelPhase1Common/interface/SiPixelPhase1Base.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

namespace {

class SiPixelPhase1RecHits final : public SiPixelPhase1Base {
  enum {
    NRECHITS,
    CLUST_X,
    CLUST_Y,
    ERROR_X,
    ERROR_Y,
    POS,
    CLUSTER_PROB
  };

  public:
  explicit SiPixelPhase1RecHits(const edm::ParameterSet& conf);
  void analyze(const edm::Event&, const edm::EventSetup&) override;

  private:
    edm::EDGetTokenT<reco::TrackCollection> srcToken_;
    edm::EDGetTokenT<reco::VertexCollection> offlinePrimaryVerticesToken_;

    bool onlyValid_;
    bool applyVertexCut_;
};

SiPixelPhase1RecHits::SiPixelPhase1RecHits(const edm::ParameterSet& iConfig) :
  SiPixelPhase1Base(iConfig) 
{
  srcToken_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("src"));

  offlinePrimaryVerticesToken_ = consumes<reco::VertexCollection>(std::string("offlinePrimaryVertices"));

  onlyValid_=iConfig.getParameter<bool>("onlyValidHits");

  applyVertexCut_=iConfig.getUntrackedParameter<bool>("VertexCut",true);

}

void SiPixelPhase1RecHits::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  if( !checktrigger(iEvent,iSetup,DCS) ) return;

  edm::ESHandle<TrackerGeometry> tracker;
  iSetup.get<TrackerDigiGeometryRecord>().get(tracker);
  assert(tracker.isValid());

  edm::Handle<reco::TrackCollection> tracks;
  iEvent.getByToken( srcToken_, tracks);
  if (!tracks.isValid()) return;

  edm::Handle<reco::VertexCollection> vertices;
  iEvent.getByToken(offlinePrimaryVerticesToken_, vertices);

  if (applyVertexCut_ && (!vertices.isValid() || vertices->empty())) return;


  for (auto const & track : *tracks) {

    if (applyVertexCut_ && (track.pt() < 0.75 || std::abs( track.dxy(vertices->at(0).position()) ) > 5*track.dxyError())) continue;

    bool isBpixtrack = false, isFpixtrack = false;

    auto const & trajParams = track.extra()->trajParams();
    auto hb = track.recHitsBegin();
    for(unsigned int h=0;h<track.recHitsSize();h++){
     
      auto hit = *(hb+h);
      if(!hit->isValid()) continue;
      
      DetId id = hit->geographicalId();
      uint32_t subdetid = (id.subdetId());

      if (subdetid == PixelSubdetector::PixelBarrel) isBpixtrack = true;
      if (subdetid == PixelSubdetector::PixelEndcap) isFpixtrack = true;
    }

    if (!isBpixtrack && !isFpixtrack) continue;

    // then, look at each hit
    for(unsigned int h=0;h<track.recHitsSize();h++){
      auto rechit = *(hb+h);
    
      if(!rechit->isValid()) continue;

      //continue if not a Pixel recHit
      DetId id = rechit->geographicalId();
      uint32_t subdetid = (id.subdetId());

      if (   subdetid != PixelSubdetector::PixelBarrel 
         && subdetid != PixelSubdetector::PixelEndcap) continue;

      bool isHitValid   = rechit->getType()==TrackingRecHit::valid;
      if (onlyValid_ && !isHitValid) continue; //useful to run on cosmics where the TrackEfficiency plugin is not used

      const SiPixelRecHit* prechit = dynamic_cast<const SiPixelRecHit*>(rechit);//to be used to get the associated cluster and the cluster probability      

      int sizeX=0, sizeY=0;

      if (isHitValid){
    SiPixelRecHit::ClusterRef const& clust = prechit->cluster();
    sizeX = (*clust).sizeX();
    sizeY = (*clust).sizeY();
      }

      const PixelGeomDetUnit* geomdetunit = dynamic_cast<const PixelGeomDetUnit*> ( tracker->idToDet(id) );
      const PixelTopology& topol = geomdetunit->specificTopology();

      LocalPoint lp = trajParams[h].position();
      MeasurementPoint mp = topol.measurementPosition(lp);
      
      int row = (int) mp.x();
      int col = (int) mp.y();
      
      float rechit_x = lp.x();
      float rechit_y = lp.y();
      
      LocalError lerr = rechit->localPositionError();
      float lerr_x = sqrt(lerr.xx());
      float lerr_y = sqrt(lerr.yy());

      histo[NRECHITS].fill(id, &iEvent, col, row); //in general a inclusive counter of missing/valid/inactive hits

      if (isHitValid){
    histo[CLUST_X].fill(sizeX, id, &iEvent, col, row);
    histo[CLUST_Y].fill(sizeY, id, &iEvent, col, row);
      }

      histo[ERROR_X].fill(lerr_x, id, &iEvent);
      histo[ERROR_Y].fill(lerr_y, id, &iEvent);

      histo[POS].fill(rechit_x, rechit_y, id, &iEvent);
      
      if (isHitValid){
    double clusterProbability= prechit->clusterProbability(0);
    if (clusterProbability > 0)
      histo[CLUSTER_PROB].fill(log10(clusterProbability), id, &iEvent);
      }
    }
  }

  histo[NRECHITS].executePerEventHarvesting(&iEvent);
}

} //namespace

DEFINE_FWK_MODULE(SiPixelPhase1RecHits);