MuonTrackAnalyzer.h

Go to the documentation of this file.

#ifndef Validation_RecoMuon_MuonTrackAnalyzer_H
#define Validation_RecoMuon_MuonTrackAnalyzer_H

// Base Class Headers
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Utilities/interface/InputTag.h"

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

#include "FWCore/Framework/interface/ESHandle.h"

#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"

#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"

namespace edm {class ParameterSet; class Event; class EventSetup;}
namespace reco {class TransientTrack;}

class TFile;
class TH1F; 
class TH2F;
class HTrackVariables;
class HTrack;

class TrajectoryStateOnSurface;
class FreeTrajectoryState;
class MuonServiceProxy;
class MuonPatternRecoDumper;
class TrajectorySeed;
class MuonUpdatorAtVertex;

class MuonTrackAnalyzer: public edm::EDAnalyzer {

 public:
  enum EtaRange{all,barrel,endcap};

 public:
  MuonTrackAnalyzer(const edm::ParameterSet& pset);

  virtual ~MuonTrackAnalyzer();

  // Operations

  void analyze(const edm::Event & event, const edm::EventSetup& eventSetup);
  void tracksAnalysis(const edm::Event & event, const edm::EventSetup& eventSetup,
              edm::Handle<edm::SimTrackContainer> simTracks);
  void seedsAnalysis(const edm::Event & event, const edm::EventSetup& eventSetup,
             edm::Handle<edm::SimTrackContainer> simTracks);
    

  virtual void beginJob() ;
  virtual void endJob() ;
  virtual void beginRun() ;
  virtual void endRun() ;
 protected:

 private:
  bool isInTheAcceptance(double eta);
  bool checkMuonSimHitPresence(const edm::Event & event,
                   edm::Handle<edm::SimTrackContainer> simTracks);

  std::pair<SimTrack,double> getSimTrack(TrajectoryStateOnSurface &tsos,
                       edm::Handle<edm::SimTrackContainer> simTracks);  

  void fillPlots(const edm::Event &event, edm::Handle<edm::SimTrackContainer> &simTracks);
  void fillPlots(reco::TransientTrack &track, SimTrack &simTrack);
  void fillPlots(TrajectoryStateOnSurface &recoTSOS,SimTrack &simState,
         HTrack*, MuonPatternRecoDumper&);
  void fillPlots(FreeTrajectoryState &recoFTS,SimTrack &simTrack,
         HTrack *histo, MuonPatternRecoDumper &debug);


  TrajectoryStateOnSurface getSeedTSOS(const TrajectorySeed& seed);

  DQMStore* dbe_;
  std::string dirName_;

  std::string out;
  //TFile* theFile;

  EtaRange theEtaRange;
  
  edm::InputTag theSimTracksLabel;
  edm::InputTag theSeedsLabel;
  edm::InputTag theTracksLabel;
  edm::InputTag theCSCSimHitLabel;
  edm::InputTag theDTSimHitLabel; 
  edm::InputTag theRPCSimHitLabel;

  edm::EDGetTokenT<edm::SimTrackContainer> theSimTracksToken;
  edm::EDGetTokenT<TrajectorySeedCollection> theSeedsToken;
  edm::EDGetTokenT<reco::TrackCollection> theTracksToken;
  edm::EDGetTokenT<std::vector<PSimHit> > theCSCSimHitToken;
  edm::EDGetTokenT<std::vector<PSimHit> > theDTSimHitToken;
  edm::EDGetTokenT<std::vector<PSimHit> > theRPCSimHitToken;


  bool doTracksAnalysis;
  bool doSeedsAnalysis;
  std::string theSeedPropagatorName;

  MuonServiceProxy *theService;
  MuonUpdatorAtVertex *theUpdator;

  // Histograms
  MonitorElement *hChi2;
  MonitorElement *hChi2Norm;
  MonitorElement *hHitsPerTrack;
  MonitorElement *hDof;
  MonitorElement *hChi2Prob;

  MonitorElement *hNumberOfTracks;
  MonitorElement *hNumberOfTracksVsEta;
  MonitorElement *hChargeVsEta;
  MonitorElement *hChargeVsPt;
  MonitorElement *hPtRecVsPtGen;

  MonitorElement *hChi2VsEta;
  MonitorElement *hChi2NormVsEta;
  MonitorElement *hHitsPerTrackVsEta;
  MonitorElement *hDofVsEta; 
  MonitorElement *hChi2ProbVsEta;
  MonitorElement *hDeltaPtVsEta;
  MonitorElement *hDeltaPt_In_Out_VsEta;

  HTrackVariables *hSimTracks;

  HTrack *hRecoSeedInner;
  HTrack *hRecoSeedPCA;
  HTrack *hRecoTracksPCA; 
  HTrack *hRecoTracksInner;
  HTrack *hRecoTracksOuter;

  // Counters
  int numberOfSimTracks;
  int numberOfRecTracks;
};
#endif