Skip to content

Commit

Permalink
Changed HeavyIon/MuonAnalysis to MiniAOD using pat::Muon format (cms-…
Browse files Browse the repository at this point in the history 
…sw#271)

Co-authored-by: Andre Govinda Stahl Leiton <andre.govinda.stahl.leiton@cern.ch>
  • Loading branch information
@stahlleiton
stahlleiton and Andre Govinda Stahl Leiton committed Apr 1, 2020
1 parent 33e2fd8 commit a849484
Show file tree
Hide file tree
Showing 4 changed files with 749 additions and 0 deletions.
20 changes: 20 additions & 0 deletions HeavyIonsAnalysis/MuonAnalysis/BuildFile.xml
View file
@@ -0,0 +1,20 @@
<use name="clhep"/>
<use name="root"/>
<use name="FWCore/Framework"/>
<use name="FWCore/PluginManager"/>
<use name="FWCore/ParameterSet"/>
<use name="FWCore/ServiceRegistry"/>
<use name="DataFormats/HepMCCandidate"/>
<use name="DataFormats/PatCandidates"/>
<use name="DataFormats/HeavyIonEvent"/>
<use name="SimDataFormats/TrackingAnalysis"/>
<use name="CommonTools/UtilAlgos"/>
<use name="CondFormats/HIObjects"/>
<use name="CondFormats/DataRecord"/>
<use name="TrackingTools/TransientTrack"/>
<use name="RecoVertex/KalmanVertexFit"/>
<use name="DataFormats/TrackReco"/>
<use name="DataFormats/TrackerRecHit2D"/>
<use name="DataFormats/TrackingRecHit"/>
<use name="RecoTracker/TkTrackingRegions"/>
<flags EDM_PLUGIN="1"/>
197 changes: 197 additions & 0 deletions HeavyIonsAnalysis/MuonAnalysis/interface/HLTMuTree.h
View file
@@ -0,0 +1,197 @@
// system include files
#include <memory>
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

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

#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"
#include "HepMC/GenVertex.h"
#include "HepMC/HeavyIon.h"

// data formats
#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
#include "DataFormats/L1Trigger/interface/L1MuonParticle.h"
#include "DataFormats/L1Trigger/interface/L1MuonParticleFwd.h"
#include "TrackingTools/PatternTools/interface/ClosestApproachInRPhi.h"

//services and tools
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexFitter.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"

// root include files
#include "TROOT.h"
#include "TTree.h"
#include "TLorentzVector.h"

//
// class declaration
//

using namespace std;
using namespace reco;
using namespace edm;


class HLTMuTree : public edm::EDAnalyzer {
public:
explicit HLTMuTree(const edm::ParameterSet&);
~HLTMuTree();


private:
virtual void beginJob() ;
virtual void analyze(const edm::Event&, const edm::EventSetup&);
virtual void endJob() ;

// ----------member data ---------------------------
edm::EDGetTokenT< edm::View<pat::Muon> > tokRecoMu;
edm::EDGetTokenT< reco::VertexCollection > tokVtx;
edm::EDGetTokenT< reco::GenParticleCollection > tokGenPtl;

Bool_t doReco;
Bool_t doGen;
Bool_t doHLT;

TTree *treeMu;
edm::Service<TFileService> foutput;

int run, event, lumi, cbin;
float vx, vy, vz;

static const int nmax = 100;
typedef struct {
int nptl;
int pid[nmax];
int status[nmax];
int mom[nmax];
float pt[nmax];
float p[nmax];
float eta[nmax];
float phi[nmax];
} GENMU;

typedef struct {
int nptl;
int charge[nmax];
float pt[nmax];
float p[nmax];
float eta[nmax];
float phi[nmax];
float dxy[nmax];
float dz[nmax];
int nValMuHits[nmax];
int nValTrkHits[nmax];
int nTrkFound[nmax];
float glbChi2_ndof[nmax];
float trkChi2_ndof[nmax];
int pixLayerWMeas[nmax];
float trkDxy[nmax];
float trkDz[nmax];
int isArbitrated[nmax];
int trkLayerWMeas[nmax];
int nValPixHits[nmax];
int nMatchedStations[nmax];
} GLBMU;

typedef struct {
int nptl;
int charge[nmax];
float pt[nmax];
float p[nmax];
float eta[nmax];
float phi[nmax];
float dxy[nmax];
float dz[nmax];
} STAMU;

typedef struct {
int npair;
float vProb[nmax];
float mass[nmax];
float e[nmax];

float pt[nmax];
float pt1[nmax];
float pt2[nmax];
float eta[nmax];
float eta1[nmax];
float eta2[nmax];
float rapidity[nmax];
float phi[nmax];
float phi1[nmax];
float phi2[nmax];
int charge[nmax];
int charge1[nmax];
int charge2[nmax];
int isArb1[nmax];
int isArb2[nmax];
int nMuHit1[nmax];
int nMuHit2[nmax];
int nTrkHit1[nmax];
int nTrkHit2[nmax];
int nTrkLayers1[nmax];
int nTrkLayers2[nmax];
int nPixHit1[nmax];
int nPixHit2[nmax];
int nMatchedStations1[nmax];
int nMatchedStations2[nmax];
float trkChi2_1[nmax];
float trkChi2_2[nmax];
float glbChi2_1[nmax];
float glbChi2_2[nmax];
float dxy1[nmax];
float dxy2[nmax];
float dz1[nmax];
float dz2[nmax];
}DIMU;

GENMU GenMu;
GLBMU GlbMu;
STAMU StaMu;
DIMU DiMu;

float muonl2pt[nmax], muonl2eta[nmax], muonl2phi[nmax], muonl2dr[nmax], muonl2dz[nmax], muonl2vtxz[nmax];
float muonl3pt[nmax], muonl3eta[nmax], muonl3phi[nmax], muonl3dr[nmax], muonl3dz[nmax], muonl3vtxz[nmax], muonl3normchi2[nmax];
float muonl2pterr[nmax], muonl3pterr[nmax];
int nmu2cand, nmu3cand;
int muonl2chg[nmax], muonl2nhits[nmax], muonl3chg[nmax], muonl3nhits[nmax];
int muonl3ntrackerhits[nmax], muonl3nmuonhits[nmax];


};
//
// constants, enums and typedefs
//
9 changes: 9 additions & 0 deletions HeavyIonsAnalysis/MuonAnalysis/python/hltMuTree_cfi.py
View file
@@ -0,0 +1,9 @@
import FWCore.ParameterSet.Config as cms

hltMuTree = cms.EDAnalyzer("HLTMuTree",
muons = cms.InputTag("slimmedMuons"),
vertices = cms.InputTag("offlineSlimmedPrimaryVertices"),
doReco = cms.untracked.bool(True),
doGen = cms.untracked.bool(False),
genparticle = cms.InputTag("genParticles")
)

0 comments on commit a849484

Please sign in to comment.