GlobalRecHitsProducer.h

Go to the documentation of this file.

#ifndef GlobalRecHitsProducer_h
#define GlobalRecHitsProducer_h

// framework & common header files
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"

//DQM services
#include "DQMServices/Core/interface/DQMStore.h"
#include "FWCore/ServiceRegistry/interface/Service.h"



//#include "DataFormats/Common/interface/Provenance.h"
#include "DataFormats/Provenance/interface/Provenance.h"
#include "FWCore/Framework/interface/MakerMacros.h" 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// ecal calorimeter info
#include "DataFormats/EcalDigi/interface/EBDataFrame.h"
#include "DataFormats/EcalDigi/interface/EEDataFrame.h"
#include "DataFormats/EcalDigi/interface/ESDataFrame.h"
#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "DataFormats/EcalDetId/interface/ESDetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

// hcal calorimeter info
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h"
#include "DataFormats/HcalDetId/interface/HcalElectronicsId.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h"
#include "DataFormats/HcalDigi/interface/HcalQIESample.h"
#include "CalibFormats/HcalObjects/interface/HcalDbRecord.h"
#include "CalibFormats/HcalObjects/interface/HcalCoderDb.h"
#include "CalibFormats/HcalObjects/interface/HcalCalibrations.h"
#include "DataFormats/HcalDigi/interface/HBHEDataFrame.h"
#include "DataFormats/HcalDigi/interface/HFDataFrame.h"
#include "DataFormats/HcalDigi/interface/HODataFrame.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "CalibFormats/HcalObjects/interface/HcalDbService.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/HcalRecHit/interface/HcalSourcePositionData.h"

// silicon strip info
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/SiStripDigi/interface/SiStripDigi.h"
#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"
#include "SimTracker/TrackerHitAssociation/interface/TrackerHitAssociator.h" 
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/GluedGeomDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetType.h"
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/SiStripCluster/interface/SiStripCluster.h" 
#include "DataFormats/SiStripCluster/interface/SiStripClusterCollection.h" 
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h" 
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h" 
#include "DataFormats/Common/interface/OwnVector.h" 

// silicon pixel info
#include "DataFormats/SiPixelDigi/interface/PixelDigi.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetType.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"

// muon DT info
#include "DataFormats/DTDigi/interface/DTDigi.h"
#include "DataFormats/DTDigi/interface/DTDigiCollection.h"
#include "DataFormats/MuonDetId/interface/DTWireId.h"
#include "DataFormats/MuonDetId/interface/DTLayerId.h"
#include "DataFormats/DTRecHit/interface/DTRecHitCollection.h"
#include "Geometry/DTGeometry/interface/DTLayer.h"
#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
//#include "Validation/GlobalRecHits/interface/DTHitQualityUtils.h"
#include "Validation/DTRecHits/interface/DTHitQualityUtils.h"

// muon CSC info
#include "DataFormats/CSCDigi/interface/CSCStripDigi.h"
#include "DataFormats/CSCDigi/interface/CSCStripDigiCollection.h"
#include "DataFormats/CSCDigi/interface/CSCWireDigi.h"
#include "DataFormats/CSCDigi/interface/CSCWireDigiCollection.h"
#include "Geometry/CSCGeometry/interface/CSCGeometry.h"
#include "DataFormats/CSCRecHit/interface/CSCRecHit2DCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCRecHit2D.h"
#include "Geometry/CSCGeometry/interface/CSCLayer.h"

// muon RPC info
#include "Geometry/RPCGeometry/interface/RPCGeometry.h"
#include "DataFormats/RPCDigi/interface/RPCDigiCollection.h"
#include "DataFormats/MuonDetId/interface/RPCDetId.h"
#include "DataFormats/RPCRecHit/interface/RPCRecHitCollection.h"
#include "Geometry/RPCGeometry/interface/RPCRoll.h"

// event info
#include "SimDataFormats/ValidationFormats/interface/PValidationFormats.h"
#include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
#include "SimDataFormats/CrossingFrame/interface/MixCollection.h"
#include "SimDataFormats/CaloHit/interface/PCaloHit.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h" 

// general info 
#include "DataFormats/DetId/interface/DetId.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h" 
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h" 

// helper files
//#include <CLHEP/Vector/LorentzVector.h>
//#include <CLHEP/Units/SystemOfUnits.h>

#include <iostream>
#include <stdlib.h>
#include <string>
#include <memory>
#include <vector>
#include <map>
#include <math.h>

#include "TString.h"

class PGlobalRecHit;

class GlobalRecHitsProducer : public edm::EDProducer
{

 public:

  typedef std::vector<float> FloatVector;
  typedef std::vector<double> DoubleVector;
  typedef std::vector<int> IntVector;
  typedef std::map<uint32_t,float,std::less<uint32_t> > MapType;

  explicit GlobalRecHitsProducer(const edm::ParameterSet&);
  virtual ~GlobalRecHitsProducer();
  virtual void beginJob();
  virtual void endJob();  
  virtual void produce(edm::Event&, const edm::EventSetup&);
  
 private:

  // production related methods
  void fillECal(edm::Event&, const edm::EventSetup&);
  void storeECal(PGlobalRecHit&);
  void fillHCal(edm::Event&, const edm::EventSetup&);
  void storeHCal(PGlobalRecHit&);
  void fillTrk(edm::Event&, const edm::EventSetup&);
  void storeTrk(PGlobalRecHit&);
  void fillMuon(edm::Event&, const edm::EventSetup&);
  void storeMuon(PGlobalRecHit&);  

  void clear();

 private:

  //  parameter information
  std::string fName;
  int verbosity;
  int frequency;
  std::string label;
  bool getAllProvenances;
  bool printProvenanceInfo;

  // Electromagnetic info
  // ECal info
 
  FloatVector EBRE; 
  FloatVector EBSHE;

  FloatVector EERE; 
  FloatVector EESHE;

  FloatVector ESRE; 
  FloatVector ESSHE;

  edm::InputTag ECalEBSrc_;
  edm::InputTag ECalUncalEBSrc_;
  edm::InputTag ECalEESrc_;
  edm::InputTag ECalUncalEESrc_;
  edm::InputTag ECalESSrc_;
  edm::EDGetTokenT<EBRecHitCollection> ECalEBSrc_Token_;
  edm::EDGetTokenT<EERecHitCollection> ECalEESrc_Token_;
  edm::EDGetTokenT<ESRecHitCollection> ECalESSrc_Token_;
  edm::EDGetTokenT<EBUncalibratedRecHitCollection> ECalUncalEBSrc_Token_;
  edm::EDGetTokenT<EEUncalibratedRecHitCollection> ECalUncalEESrc_Token_;
  edm::EDGetTokenT<CrossingFrame<PCaloHit>> EBHits_Token_;
  edm::EDGetTokenT<CrossingFrame<PCaloHit>> EEHits_Token_;
  edm::EDGetTokenT<CrossingFrame<PCaloHit>> ESHits_Token_;

  // HCal info

  FloatVector HBCalREC;
  FloatVector HBCalR;
  FloatVector HBCalSHE;

  FloatVector HECalREC;
  FloatVector HECalR;
  FloatVector HECalSHE;

  FloatVector HOCalREC;
  FloatVector HOCalR;
  FloatVector HOCalSHE;

  FloatVector HFCalREC;
  FloatVector HFCalR;
  FloatVector HFCalSHE;

  edm::InputTag HCalSrc_;
  edm::EDGetTokenT<edm::PCaloHitContainer> HCalSrc_Token_;

  // Tracker info
  // SiStrip
  
  FloatVector TIBL1RX, TIBL2RX, TIBL3RX, TIBL4RX;
  FloatVector TIBL1RY, TIBL2RY, TIBL3RY, TIBL4RY;
  FloatVector TIBL1SX, TIBL2SX, TIBL3SX, TIBL4SX;
  FloatVector TIBL1SY, TIBL2SY, TIBL3SY, TIBL4SY;

  FloatVector TOBL1RX, TOBL2RX, TOBL3RX, TOBL4RX;
  FloatVector TOBL1RY, TOBL2RY, TOBL3RY, TOBL4RY;
  FloatVector TOBL1SX, TOBL2SX, TOBL3SX, TOBL4SX;
  FloatVector TOBL1SY, TOBL2SY, TOBL3SY, TOBL4SY;

  FloatVector TIDW1RX, TIDW2RX, TIDW3RX;
  FloatVector TIDW1RY, TIDW2RY, TIDW3RY;
  FloatVector TIDW1SX, TIDW2SX, TIDW3SX;
  FloatVector TIDW1SY, TIDW2SY, TIDW3SY;

  FloatVector TECW1RX, TECW2RX, TECW3RX, TECW4RX, TECW5RX, TECW6RX, TECW7RX,
    TECW8RX;
  FloatVector TECW1RY, TECW2RY, TECW3RY, TECW4RY, TECW5RY, TECW6RY, TECW7RY,
    TECW8RY;
  FloatVector TECW1SX, TECW2SX, TECW3SX, TECW4SX, TECW5SX, TECW6SX, TECW7SX,
    TECW8SX;
  FloatVector TECW1SY, TECW2SY, TECW3SY, TECW4SY, TECW5SY, TECW6SY, TECW7SY,
    TECW8SY;

  edm::InputTag SiStripSrc_;
  edm::EDGetTokenT<SiStripMatchedRecHit2DCollection> SiStripSrc_Token_;

  std::vector<PSimHit> matched;
  std::pair<LocalPoint,LocalVector> 
    projectHit( const PSimHit& hit,
        const StripGeomDetUnit* stripDet,
        const BoundPlane& plane);
  edm::ParameterSet conf_;

  // SiPxl

  FloatVector BRL1RX, BRL2RX, BRL3RX;
  FloatVector BRL1RY, BRL2RY, BRL3RY;
  FloatVector BRL1SX, BRL2SX, BRL3SX;
  FloatVector BRL1SY, BRL2SY, BRL3SY;

  FloatVector FWD1pRX, FWD1nRX, FWD2pRX, FWD2nRX;
  FloatVector FWD1pRY, FWD1nRY, FWD2pRY, FWD2nRY;
  FloatVector FWD1pSX, FWD1nSX, FWD2pSX, FWD2nSX;
  FloatVector FWD1pSY, FWD1nSY, FWD2pSY, FWD2nSY;

  edm::InputTag SiPxlSrc_;
  edm::EDGetTokenT<SiPixelRecHitCollection> SiPxlSrc_Token_;

  // Muon info
  // DT

  FloatVector DTRHD;
  FloatVector DTSHD;

  edm::InputTag MuDTSrc_;
  edm::InputTag MuDTSimSrc_;
  edm::EDGetTokenT<DTRecHitCollection> MuDTSrc_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> MuDTSimSrc_Token_;

  // Return a map between DTRecHit1DPair and wireId
  std::map<DTWireId, std::vector<DTRecHit1DPair> >
    map1DRecHitsPerWire(const DTRecHitCollection* dt1DRecHitPairs);
  
  // Compute SimHit distance from wire (cm)
  float simHitDistFromWire(const DTLayer* layer,
               DTWireId wireId,
               const PSimHit& hit);
  
  // Find the RecHit closest to the muon SimHit
  template  <typename type>
    const type* 
    findBestRecHit(const DTLayer* layer,
           DTWireId wireId,
           const std::vector<type>& recHits,
           const float simHitDist);
  
  // Compute the distance from wire (cm) of a hits in a DTRecHit1DPair
  float recHitDistFromWire(const DTRecHit1DPair& hitPair, 
               const DTLayer* layer);
  // Compute the distance from wire (cm) of a hits in a DTRecHit1D
  float recHitDistFromWire(const DTRecHit1D& recHit, const DTLayer* layer);
    
  // Does the real job
  template  <typename type>
    int compute(const DTGeometry *dtGeom,
         const std::map<DTWireId, std::vector<PSimHit> >& simHitsPerWire,
         const std::map<DTWireId, std::vector<type> >& recHitsPerWire,
         int step);

  // CSC

  FloatVector CSCRHPHI;
  FloatVector CSCRHPERP;
  FloatVector CSCSHPHI;

  edm::InputTag MuCSCSrc_;
  edm::EDGetTokenT<CSCRecHit2DCollection> MuCSCSrc_Token_;
  edm::EDGetTokenT<CrossingFrame<PSimHit>> MuCSCHits_Token_;

  std::map<int, edm::PSimHitContainer> theMap;
  void plotResolution(const PSimHit &simHit, const CSCRecHit2D &recHit,
              const CSCLayer *layer, int chamberType);

  // RPC

  FloatVector RPCRHX;
  FloatVector RPCSHX;

  edm::InputTag MuRPCSrc_;
  edm::InputTag MuRPCSimSrc_;
  edm::EDGetTokenT<RPCRecHitCollection> MuRPCSrc_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> MuRPCSimSrc_Token_;

  // private statistics information
  unsigned int count;

}; // end class declaration

#endif

#ifndef GlobalHitMap
#define GlobalHitMap

// geometry mapping
static const int dTrk             = 1;
static const int sdPxlBrl         = 1;
static const int sdPxlFwd         = 2;
static const int sdSiTIB          = 3;
static const int sdSiTID          = 4;
static const int sdSiTOB          = 5;
static const int sdSiTEC          = 6;

static const int dMuon            = 2;
static const int sdMuonDT         = 1;
static const int sdMuonCSC        = 2;
static const int sdMuonRPC        = 3;
static const int sdMuonRPCRgnBrl  = 0;
static const int sdMuonRPCRgnFwdp = 1;
static const int sdMuonRPCRgnFwdn = -1;

static const int dEcal            = 3;
static const int sdEcalBrl        = 1;
static const int sdEcalFwd        = 2;
static const int sdEcalPS         = 3;
static const int sdEcalTT         = 4;
static const int sdEcalLPnD       = 5;

static const int dHcal            = 4;
static const int sdHcalEmpty      = 0;
static const int sdHcalBrl        = 1;
static const int sdHcalEC         = 2;
static const int sdHcalOut        = 3;
static const int sdHcalFwd        = 4;
static const int sdHcalTT         = 5;
static const int sdHcalCalib      = 6;
static const int sdHcalCompst     = 7;

#endif //PGlobalRecHitsProducer_h