[ViewVC] Diff of: cvsroot/UserCode/grimes/VHbbAnalysisCode/src/MuonInfoPlotSet.cpp

Comparing UserCode/grimes/VHbbAnalysisCode/src/MuonInfoPlotSet.cpp (file contents):
Revision 1.1 by grimes, Tue Feb 14 01:43:16 2012 UTC vs.
Revision 1.2 by grimes, Fri Apr 27 13:52:14 2012 UTC

+#include <TDirectory.h>
+#include <TH1F.h>
-+
+#include <TTree.h>
-+
+#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbEventAuxInfo.h"
-<
+trkupgradeanalysis::MuonInfoPlotSet::MuonInfoPlotSet()
-<
+        : histogramHaveBeenBooked_(false)
->
+trkupgradeanalysis::MuonInfoPlotSet::MuonInfoPlotSet( bool createIsolationTree )
->
+        : histogramHaveBeenBooked_(false), createIsolationTree_(createIsolationTree), pIsolationTree_(NULL)
+        // No operation besides the initialiser list.
+        pPt_=new TH1F( "pT", "pT", 60, 0, 250 );
+        pPt_->SetDirectory(pDirectory);
-+
+        pChargedIsolation_=new TH1F( "chargedIsolation", "chargedIsolation", 60, 0, 30 );
-+
+        pChargedIsolation_->SetDirectory(pDirectory);
-+
-+
+        pPhotonIsolation_=new TH1F( "photonIsolation", "photonIsolation", 60, 0, 30 );
-+
+        pPhotonIsolation_->SetDirectory(pDirectory);
-+
-+
+        pNeutralIsolation_=new TH1F( "neutralIsolation", "neutralIsolation", 60, 0, 30 );
-+
+        pNeutralIsolation_->SetDirectory(pDirectory);
-+
-+
+        pPileupIsolation_=new TH1F( "pileupIsolation", "pileupIsolation", 60, 0, 30 );
-+
+        pPileupIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRelativeChargedIsolation_=new TH1F( "relativeChargedIsolation", "relativeChargedIsolation", 60, 0, 1 );
-+
+        pRelativeChargedIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRelativePhotonIsolation_=new TH1F( "relativePhotonIsolation", "relativePhotonIsolation", 60, 0, 1 );
-+
+        pRelativePhotonIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRelativeNeutralIsolation_=new TH1F( "relativeNeutralIsolation", "relativeNeutralIsolation", 60, 0, 1 );
-+
+        pRelativeNeutralIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRelativePileupIsolation_=new TH1F( "relativePileupIsolation", "relativePileupIsolation", 60, 0, 1 );
-+
+        pRelativePileupIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRelativeIsolation_=new TH1F( "relativeIsolation", "relativeIsolation", 120, 0, 2 );
-+
+        pRelativeIsolation_->SetDirectory(pDirectory);
-+
-+
+        pDeltaBetaCorrectedIsolation_=new TH1F( "deltaBetaCorrectedIsolation", "deltaBetaCorrectedIsolation", 120, 0, 2 );
-+
+        pDeltaBetaCorrectedIsolation_->SetDirectory(pDirectory);
-+
-+
+        pRho25CorrectedIsolation_=new TH1F( "rho25CorrectedIsolation", "rho25CorrectedIsolation", 120, 0, 2 );
-+
+        pRho25CorrectedIsolation_->SetDirectory(pDirectory);
-+
-+
+        if( createIsolationTree_ )
-+
+        {
-+
+                pIsolationTree_=new TTree("isolationTree","Data about the muon isolation");
-+
+                pIsolationTree_->Branch("numberOfPrimaryVertices",&numberOfPrimaryVertices_branch_,"numberOfPrimaryVertices/I");
-+
+                pIsolationTree_->Branch("chargedIsolation",&chargedIsolation_branch_,"chargedIsolation/F");
-+
+                pIsolationTree_->Branch("photonIsolation",&photonIsolation_branch_,"photonIsolation/F");
-+
+                pIsolationTree_->Branch("neutralIsolation",&neutralIsolation_branch_,"neutralIsolation/F");
-+
+                pIsolationTree_->Branch("pileupIsolation",&pileupIsolation_branch_,"pileupIsolation/F");
-+
+                pIsolationTree_->Branch("rho25",&rho25_branch_,"rho25/F");
-+
+                pIsolationTree_->Branch("pT",&pT_branch_,"pT/F");
-+
+                pIsolationTree_->SetDirectory( pDirectory );
-+
+        }
+        histogramHaveBeenBooked_=true;
-<
+void trkupgradeanalysis::MuonInfoPlotSet::fill( const VHbbEvent::MuonInfo& muon )
->
+void trkupgradeanalysis::MuonInfoPlotSet::fill( const VHbbEvent::MuonInfo& muon, const VHbbEventAuxInfo* pAuxInfo )
+        if( !histogramHaveBeenBooked_ ) throw std::runtime_error( "trkupgradeanalysis::MuonInfoPlotSet::book() - histograms have not been booked" );
-+
+        // Get the additional info about the event if it's available
-+
+        int numberOfPrimaryVertices=-1;
-+
+        float rho25=-1;
-+
+        if( pAuxInfo )
-+
+        {
-+
+                numberOfPrimaryVertices=pAuxInfo->pvInfo.nVertices;
-+
+                rho25=pAuxInfo->puInfo.rho25;
-+
+        }
-+
+        pGlobalChi2_->Fill( muon.globChi2 );
+        pNumberOfPixelHits_->Fill( muon.nPixelHits );
+        pNumberOfGlobalHits_->Fill( muon.globNHits );
+        pIPDB_->Fill( muon.ipDb );
+        pEta_->Fill( muon.p4.Eta() );
+        pPt_->Fill( muon.p4.Pt() );
-+
+        pChargedIsolation_->Fill( muon.pfChaIso );
-+
+        pPhotonIsolation_->Fill( muon.pfPhoIso );
-+
+        pNeutralIsolation_->Fill( muon.pfNeuIso );
-+
+        pPileupIsolation_->Fill( muon.pfChaPUIso );
-+
+        pRelativeChargedIsolation_->Fill( muon.pfChaIso/muon.p4.Pt() );
-+
+        pRelativePhotonIsolation_->Fill( muon.pfPhoIso/muon.p4.Pt() );
-+
+        pRelativeNeutralIsolation_->Fill( muon.pfNeuIso/muon.p4.Pt() );
-+
+        pRelativePileupIsolation_->Fill( muon.pfChaPUIso/muon.p4.Pt() );
-+
+        pRelativeIsolation_->Fill( combinedRelativeIsolation(muon) );
-+
+        pDeltaBetaCorrectedIsolation_->Fill( deltaBetaCorrectedIsolation(muon) );
-+
+        if( pAuxInfo ) pRho25CorrectedIsolation_->Fill( rho25CorrectedIsolation(muon,rho25) ); // Don't bother filling if I couldn't get the rho25 info
-+
-+
+        // If the TTree is not null then isolation data has been requested
-+
+        if( pIsolationTree_ )
-+
+        {
-+
+                numberOfPrimaryVertices_branch_=numberOfPrimaryVertices;
-+
+                chargedIsolation_branch_=muon.pfChaIso;
-+
+                photonIsolation_branch_=muon.pfPhoIso;
-+
+                neutralIsolation_branch_=muon.pfNeuIso;
-+
+                pileupIsolation_branch_=muon.pfChaPUIso;
-+
+                rho25_branch_=rho25;
-+
+                pT_branch_=muon.p4.Pt();
-+
-+
+                pIsolationTree_->Fill();
-+
+        }
-+
-+
+float trkupgradeanalysis::MuonInfoPlotSet::combinedRelativeIsolation( const VHbbEvent::MuonInfo& muon )
-+
+{
-+
+        return (muon.pfChaIso+muon.pfPhoIso+muon.pfNeuIso)/muon.p4.Pt();
-+
+}
-+
-+
+float trkupgradeanalysis::MuonInfoPlotSet::deltaBetaCorrectedIsolation( const VHbbEvent::MuonInfo& muon, float deltaBetaFactor )
-+
+{
-+
+        // deltaBetaFactor defaults to -0.5
-+
+        float correctedNeutralIsolation=muon.pfPhoIso+muon.pfNeuIso+deltaBetaFactor*muon.pfChaPUIso;
-+
+        if( correctedNeutralIsolation<0 ) correctedNeutralIsolation=muon.pfPhoIso+muon.pfNeuIso;
-+
+        return (muon.pfChaIso+correctedNeutralIsolation)/muon.p4.Pt();
-+
+}
-+
-+
+float trkupgradeanalysis::MuonInfoPlotSet::rho25CorrectedIsolation( const VHbbEvent::MuonInfo& muon, float rho25 )
-+
+{
-+
+        float uncorrectedIsolation=combinedRelativeIsolation(muon);
-+
+        float correctedIsolation=uncorrectedIsolation-(M_PI*0.4*0.4*rho25)/muon.p4.Pt();
-+
+        if( correctedIsolation<0 ) correctedIsolation=0;
-+
-+
+        return correctedIsolation;
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing UserCode/grimes/VHbbAnalysisCode/src/MuonInfoPlotSet.cpp (file contents): Revision 1.1 by grimes, Tue Feb 14 01:43:16 2012 UTC vs. Revision 1.2 by grimes, Fri Apr 27 13:52:14 2012 UTC

Diff Legend

Comparing UserCode/grimes/VHbbAnalysisCode/src/MuonInfoPlotSet.cpp (file contents):
Revision 1.1 by grimes, Tue Feb 14 01:43:16 2012 UTC vs.
Revision 1.2 by grimes, Fri Apr 27 13:52:14 2012 UTC