SiStripGainsPCLWorker.h

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    CalibTracker/SiStripChannelGain
// Class:      SiStripGainsPCLWorker
// 
//
//  Original Author: L. Quertermont (calibration algorithm)
//  Contributors:    M. Verzetti    (data access)
//                   A. Di Mattia   (PCL multi stream processing and monitoring)
//                   M. Delcourt    (monitoring)
//                   M. Musich      (migration to thread-safe DQMStore access)
//
//  Created:  Wed, 12 Apr 2017 14:46:48 GMT
//

// CMSSW includes
#include "CalibFormats/SiStripObjects/interface/SiStripDetCabling.h"
#include "CalibFormats/SiStripObjects/interface/SiStripGain.h"
#include "CalibFormats/SiStripObjects/interface/SiStripQuality.h"
#include "CalibTracker/Records/interface/SiStripDetCablingRcd.h"
#include "CalibTracker/Records/interface/SiStripGainRcd.h"
#include "CalibTracker/Records/interface/SiStripQualityRcd.h"
#include "CondFormats/SiStripObjects/interface/SiStripApvGain.h"
#include "DQMServices/Core/interface/DQMEDAnalyzer.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DQMServices/Core/interface/MonitorElement.h"
#include "DQMServices/Core/interface/MonitorElement.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/FEDRawData/interface/FEDNumbering.h"
#include "DataFormats/GeometrySurface/interface/RectangularPlaneBounds.h"
#include "DataFormats/GeometrySurface/interface/TrapezoidalPlaneBounds.h"
#include "DataFormats/SiStripCluster/interface/SiStripClusterCollection.h"
#include "DataFormats/SiStripDetId/interface/SiStripSubStructure.h"
#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2D.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit1D.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"
#include "Geometry/CommonDetUnit/interface/TrackingGeometry.h"
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"

#include "CalibTracker/SiStripChannelGain/interface/APVGainStruct.h"

// System includes
#include <unordered_map>

//
// class declaration
//

class SiStripGainsPCLWorker : public  DQMEDAnalyzer {
public:
    explicit SiStripGainsPCLWorker(const edm::ParameterSet&);
     
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
    virtual void beginJob() ;
    virtual void bookHistograms(DQMStore::IBooker &, edm::Run const &, edm::EventSetup const &);   
    virtual void dqmBeginRun(edm::Run const&, edm::EventSetup const&);
    virtual void analyze(const edm::Event&, const edm::EventSetup&);
    virtual void endJob() ;
    
    void processEvent(); //what really does the job
    virtual void checkBookAPVColls(const edm::EventSetup& setup);

    std::vector<std::string> dqm_tag_;

    int statCollectionFromMode(const char* tag) const;

    std::vector<MonitorElement*>  Charge_Vs_Index;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTIB;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTOB;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTIDP;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTIDM;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTECP1;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTECP2;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTECM1;
    std::vector<MonitorElement*>  Charge_Vs_PathlengthTECM2;

    unsigned int NEvent;    
    unsigned int NTrack;
    unsigned int NClusterStrip;
    unsigned int NClusterPixel;
    int NStripAPVs;
    int NPixelDets;
    unsigned int SRun;
    unsigned int ERun;
  
    double       MinTrackMomentum;
    double       MaxTrackMomentum;
    double       MinTrackEta;
    double       MaxTrackEta;
    unsigned int MaxNrStrips;
    unsigned int MinTrackHits;
    double       MaxTrackChiOverNdf;
    int          MaxTrackingIteration;
    bool         AllowSaturation;
    bool         FirstSetOfConstants;
    bool         Validation;
    bool         OldGainRemoving;
    bool         useCalibration;

    std::string  m_DQMdir; 
    std::string  m_calibrationMode;

    edm::ESHandle<TrackerGeometry> tkGeom_;
    const TrackerGeometry *bareTkGeomPtr_;   // ugly hack to fill APV colls only once, but checks

    //Data members for processing

    //Event data
    unsigned int                       eventnumber    =0;
    unsigned int                       runnumber      =0;
    const std::vector<bool>*           TrigTech       =0;  edm::EDGetTokenT<std::vector<bool>           > TrigTech_token_;

    // Track data
    const std::vector<double>*         trackchi2ndof  =0;  edm::EDGetTokenT<std::vector<double>         > trackchi2ndof_token_;
    const std::vector<float>*          trackp         =0;  edm::EDGetTokenT<std::vector<float>          > trackp_token_;
    const std::vector<float>*          trackpt        =0;  edm::EDGetTokenT<std::vector<float>          > trackpt_token_;
    const std::vector<double>*         tracketa       =0;  edm::EDGetTokenT<std::vector<double>         > tracketa_token_;
    const std::vector<double>*         trackphi       =0;  edm::EDGetTokenT<std::vector<double>         > trackphi_token_;
    const std::vector<unsigned int>*   trackhitsvalid =0;  edm::EDGetTokenT<std::vector<unsigned int>   > trackhitsvalid_token_;
    const std::vector<int>*            trackalgo      =0;  edm::EDGetTokenT<std::vector<int>   >          trackalgo_token_;

    // CalibTree data
    const std::vector<int>*            trackindex     =0;  edm::EDGetTokenT<std::vector<int>            > trackindex_token_;
    const std::vector<unsigned int>*   rawid          =0;  edm::EDGetTokenT<std::vector<unsigned int>   > rawid_token_;
    const std::vector<double>*         localdirx      =0;  edm::EDGetTokenT<std::vector<double>         > localdirx_token_;
    const std::vector<double>*         localdiry      =0;  edm::EDGetTokenT<std::vector<double>         > localdiry_token_;
    const std::vector<double>*         localdirz      =0;  edm::EDGetTokenT<std::vector<double>         > localdirz_token_;
    const std::vector<unsigned short>* firststrip     =0;  edm::EDGetTokenT<std::vector<unsigned short> > firststrip_token_;
    const std::vector<unsigned short>* nstrips        =0;  edm::EDGetTokenT<std::vector<unsigned short> > nstrips_token_;
    const std::vector<bool>*           saturation     =0;  edm::EDGetTokenT<std::vector<bool>           > saturation_token_;
    const std::vector<bool>*           overlapping    =0;  edm::EDGetTokenT<std::vector<bool>           > overlapping_token_;
    const std::vector<bool>*           farfromedge    =0;  edm::EDGetTokenT<std::vector<bool>           > farfromedge_token_;
    const std::vector<unsigned int>*   charge         =0;  edm::EDGetTokenT<std::vector<unsigned int>   > charge_token_;
    const std::vector<double>*         path           =0;  edm::EDGetTokenT<std::vector<double>         > path_token_;
    const std::vector<double>*         chargeoverpath =0;  edm::EDGetTokenT<std::vector<double>         > chargeoverpath_token_;
    const std::vector<unsigned char>*  amplitude      =0;  edm::EDGetTokenT<std::vector<unsigned char>  > amplitude_token_;
    const std::vector<double>*         gainused       =0;  edm::EDGetTokenT<std::vector<double>         > gainused_token_;

    std::string EventPrefix_; //("");
    std::string EventSuffix_; //("");
    std::string TrackPrefix_; //("track");
    std::string TrackSuffix_; //("");
    std::string CalibPrefix_; //("GainCalibration");
    std::string CalibSuffix_; //("");

    std::vector<std::shared_ptr<stAPVGain> > APVsCollOrdered;
    std::unordered_map<unsigned int, std::shared_ptr<stAPVGain> > APVsColl; 
    
};

inline int
SiStripGainsPCLWorker::statCollectionFromMode(const char* tag) const
{
  std::vector<std::string>::const_iterator it=dqm_tag_.begin();
  while(it!=dqm_tag_.end()) {
    if(*it==std::string(tag)) return it-dqm_tag_.begin();
    it++;
  }
  
  if (std::string(tag)=="") return 0;  // return StdBunch calibration mode for backward compatibility
  
  return None;
}

template<typename T>
inline edm::Handle<T> connect(const T* &ptr, edm::EDGetTokenT<T> token, const edm::Event &evt) {
  edm::Handle<T> handle;
  evt.getByToken(token, handle);
  ptr = handle.product();
  return handle; //return handle to keep alive pointer (safety first)
}