MuonTrackValidator.h

Go to the documentation of this file.

#ifndef MuonTrackValidator_h
#define MuonTrackValidator_h

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "Validation/RecoMuon/plugins/MuonTrackValidatorBase.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"

class MuonTrackValidator : public edm::EDAnalyzer, protected MuonTrackValidatorBase {
 public:
  MuonTrackValidator(const edm::ParameterSet& pset):MuonTrackValidatorBase(pset){
    dirName_ = pset.getParameter<std::string>("dirName");
    associatormap = pset.getParameter< edm::InputTag >("associatormap");
    UseAssociators = pset.getParameter< bool >("UseAssociators");
    tpSelector = TrackingParticleSelector(pset.getParameter<double>("ptMinTP"),
                      pset.getParameter<double>("minRapidityTP"),
                      pset.getParameter<double>("maxRapidityTP"),
                      pset.getParameter<double>("tipTP"),
                      pset.getParameter<double>("lipTP"),
                      pset.getParameter<int>("minHitTP"),
                      pset.getParameter<bool>("signalOnlyTP"),
                      pset.getParameter<bool>("chargedOnlyTP"),
                      pset.getParameter<bool>("stableOnlyTP"),
                      pset.getParameter<std::vector<int> >("pdgIdTP"));
    cosmictpSelector = CosmicTrackingParticleSelector(pset.getParameter<double>("ptMinTP"),
                              pset.getParameter<double>("minRapidityTP"),
                              pset.getParameter<double>("maxRapidityTP"),
                              pset.getParameter<double>("tipTP"),
                              pset.getParameter<double>("lipTP"),
                              pset.getParameter<int>("minHitTP"),
                              pset.getParameter<bool>("chargedOnlyTP"),
                              pset.getParameter<std::vector<int> >("pdgIdTP"));

    minPhi = pset.getParameter<double>("minPhi"); 
    maxPhi = pset.getParameter<double>("maxPhi");
    nintPhi = pset.getParameter<int>("nintPhi");
    useGsf = pset.getParameter<bool>("useGsf");
    BiDirectional_RecoToSim_association = pset.getParameter<bool>("BiDirectional_RecoToSim_association");

    // dump cfg parameters
    edm::LogVerbatim("MuonTrackValidator") << "constructing  MuonTrackValidator: " << pset.dump();
    
    // Declare consumes (also for the base class)
    bsSrc_Token = consumes<reco::BeamSpot>(bsSrc);
    tp_effic_Token = consumes<TrackingParticleCollection>(label_tp_effic);
    tp_fake_Token = consumes<TrackingParticleCollection>(label_tp_fake);
    for (unsigned int www=0;www<label.size();www++){
      track_Collection_Token.push_back(consumes<edm::View<reco::Track> >(label[www]));
    }
    simToRecoCollection_Token = consumes<reco::SimToRecoCollection>(associatormap);
    recoToSimCollection_Token = consumes<reco::RecoToSimCollection>(associatormap);

    MABH = false;
    if (!UseAssociators) {
      // flag MuonAssociatorByHits
      if (associators[0] == "MuonAssociationByHits") MABH = true;
      // reset string associators to the map label
      associators.clear();
      associators.push_back(associatormap.label());
      edm::LogVerbatim("MuonTrackValidator") << "--> associators reset to: " <<associators[0];
    }
    
    // inform on which SimHits will be counted 
    if (usetracker) edm::LogVerbatim("MuonTrackValidator")
      <<"\n usetracker = TRUE  : Tracker SimHits WILL be counted";
    else edm::LogVerbatim("MuonTrackValidator") 
      <<"\n usetracker = FALSE : Tracker SimHits WILL NOT be counted";    
    if (usemuon) edm::LogVerbatim("MuonTrackValidator")
      <<" usemuon = TRUE  : Muon SimHits WILL be counted";
    else edm::LogVerbatim("MuonTrackValidator") 
      <<" usemuon = FALSE : Muon SimHits WILL NOT be counted"<<std::endl;
    
    // loop over the reco::Track collections to validate: check for inconsistent input settings
    for (unsigned int www=0;www<label.size();www++) {
      std::string recoTracksLabel = label[www].label();  
      std::string recoTracksInstance = label[www].instance();
      
      // tracks with hits only on tracker
      if (recoTracksLabel=="generalTracks"                                  ||
      (recoTracksLabel.find("cutsRecoTracks") != std::string::npos)     ||
      recoTracksLabel=="ctfWithMaterialTracksP5LHCNavigation"           ||
      recoTracksLabel=="hltL3TkTracksFromL2"                            ||
      (recoTracksLabel=="hltL3Muons" && recoTracksInstance=="L2Seeded")) 
    {
      if (usemuon) {
        edm::LogWarning("MuonTrackValidator") 
          <<"\n*** WARNING : inconsistent input tracksTag = "<<label[www]
          <<"\n with usemuon == true"<<"\n ---> please change to usemuon == false ";
      }
      if (!usetracker) {
        edm::LogWarning("MuonTrackValidator") 
          <<"\n*** WARNING : inconsistent input tracksTag = "<<label[www]
          <<"\n with usetracker == false"<<"\n ---> please change to usetracker == true ";
      }   
    }
      
      // tracks with hits only on muon detectors
      else if (recoTracksLabel=="standAloneMuons"    || 
           recoTracksLabel=="standAloneSETMuons" ||
           recoTracksLabel=="cosmicMuons"        ||
           recoTracksLabel=="hltL2Muons")         
    {
      if (usetracker) {
        edm::LogWarning("MuonTrackValidator") 
          <<"\n*** WARNING : inconsistent input tracksTag = "<<label[www]
          <<"\n with usetracker == true"<<"\n ---> please change to usetracker == false ";
      }
      if (!usemuon) {
        edm::LogWarning("MuonTrackValidator") 
          <<"\n*** WARNING : inconsistent input tracksTag = "<<label[www]
          <<"\n with usemuon == false"<<"\n ---> please change to usemuon == true ";
      }
    }
      
    } //  for (unsigned int www=0;www<label.size();www++)
  }
    
  virtual ~MuonTrackValidator(){ }

  void beginRun(edm::Run const&, edm::EventSetup const&);
  void analyze(const edm::Event&, const edm::EventSetup& );
  void endRun(edm::Run const&, edm::EventSetup const&);

private:
  void getRecoMomentum (const reco::Track& track, double& pt, double& ptError,
            double& qoverp, double& qoverpError, double& lambda, double& lambdaError,  
            double& phi, double& phiError ) const;
  void getRecoMomentum (const reco::GsfTrack& gsfTrack, double& pt, double& ptError,
            double& qoverp, double& qoverpError, double& lambda, double& lambdaError,  
            double& phi, double& phiError) const;

 private:
  std::string dirName_;
  edm::InputTag associatormap;
  edm::EDGetTokenT<reco::SimToRecoCollection> simToRecoCollection_Token;
  edm::EDGetTokenT<reco::RecoToSimCollection> recoToSimCollection_Token;
  bool UseAssociators;
  double minPhi, maxPhi;
  int nintPhi;
  bool useGsf;
  // select tracking particles 
  //(i.e. "denominator" of the efficiency ratio)
  TrackingParticleSelector tpSelector;                    
  CosmicTrackingParticleSelector cosmictpSelector;

  // flag new validation logic (bidirectional RecoToSim association)
  bool BiDirectional_RecoToSim_association;
  // flag MuonAssociatorByHits
  bool MABH;
  
  //1D
  std::vector<MonitorElement*> h_nchi2, h_nchi2_prob, h_losthits;

  //2D  
  std::vector<MonitorElement*> chi2_vs_nhits, etares_vs_eta;
  std::vector<MonitorElement*> h_ptshifteta;
  std::vector<MonitorElement*> ptres_vs_phi, chi2_vs_phi, nhits_vs_phi, phires_vs_phi;

  //Profile2D
  std::vector<MonitorElement*> ptmean_vs_eta_phi, phimean_vs_eta_phi;

  //assoc chi2
  std::vector<MonitorElement*> h_assochi2, h_assochi2_prob;

  //chi2 and # lost hits vs eta: to be used with doProfileX
  std::vector<MonitorElement*> chi2_vs_eta, nlosthits_vs_eta;
  std::vector<MonitorElement*> h_chi2meanh, h_losthits_eta;
  std::vector<MonitorElement*> h_hits_phi;  
  std::vector<MonitorElement*> h_chi2meanhitsh, h_chi2mean_vs_phi;

  //resolution of track params: to be used with fitslicesytool
  std::vector<MonitorElement*> dxyres_vs_eta, ptres_vs_eta, dzres_vs_eta, phires_vs_eta, cotThetares_vs_eta;
  std::vector<MonitorElement*> dxyres_vs_pt, ptres_vs_pt, dzres_vs_pt, phires_vs_pt, cotThetares_vs_pt;

  //pulls of track params vs eta: to be used with fitslicesytool
  std::vector<MonitorElement*> dxypull_vs_eta, ptpull_vs_eta, dzpull_vs_eta, phipull_vs_eta, thetapull_vs_eta;
  std::vector<MonitorElement*> ptpull_vs_phi, phipull_vs_phi, thetapull_vs_phi;
  std::vector<MonitorElement*> h_dxypulleta, h_ptpulleta, h_dzpulleta, h_phipulleta, h_thetapulleta;
  std::vector<MonitorElement*> h_ptpullphi, h_phipullphi, h_thetapullphi;

};


#endif