more stats, TEfficiency plots not saving

Marcel Hohmann
1 parent 7029bfa5
Showing 2 changed files with 98 additions and 27 deletions Show diff stats
DualTrackTagger/include/DualTrackTaggerModule.h
DualTrackTagger/src/DualTrackTaggerModule.cc
@@ -27,6 +27,7 @@
 #include "TMath.h"
 #include "TH1F.h"
 #include "TString.h"
+#include "TEfficiency.h"
  
 // Include MVA classes - TEST
 #include <analysis/modules/DualTrackTagger/TMVAClassification_BDT5.class.h>
@@ -46,7 +47,6 @@ namespace Belle2 {
  
     virtual ~DualTrackTaggerModule();
  
-    virtual void defineHisto(); //Define the histograms
  
     virtual void initialize() override;
  
@@ -79,9 +79,21 @@ namespace Belle2 {
     /** Calculates BN#1079 sec 4.1 chi2 */
     virtual double chi2(Particle* p);
  
+
+    // ---------------- Functions related to Histos ------------
+
+    /** define the histograms - only if MCFlag = true */
+    virtual void defineHisto();
+
     /** Used to fill histogram showing # of tracks associated with one MC particle */
     virtual void fill_h_genParticleIndexMulti();
  
+    /** Calculate TEfficiency Plots */
+    virtual void finalizeHistograms();
+
+    /** Fill Basic Histograms */
+    virtual void fillHistograms(Particle *p1, Particle *p2, bool taggedDuplicate);
+
 		std::vector<Particle*> m_candidateParticles; // particles with low pt
     std::vector<Particle*> m_allParticlesInList; // used for statistics
  
@@ -106,6 +118,15 @@ namespace Belle2 {
     TH1F *h_genParticleIndexMulti;
     TH1F *h_Duplicates;
     TH1F *h_Singles;
+    TH1F *h_taggedDuplicate;
+
+    TH1F *h_TruePosPt;
+    TH1F *h_FalsePosPt;
+    TH1F *h_TrueNegPt;
+    TH1F *h_FalseNegPt;
+
+    TEfficiency *h_effTruePosPt;
+    TEfficiency *h_effFalsePosPt;
   };
 }
 #endif
@@ -34,17 +34,6 @@
   {
   }
  
-  void DualTrackTaggerModule::defineHisto()
-  {
-    h_genParticleIndexMulti = new TH1F("h_genParticleIndexMulti","h_genParticleIndexMulti; # of particles with same genParticleIndex",20,0,20);
-  	//h_SameChargeMultAngle = new TH1F("h_SameChargeMultAngle", "h_SameChargeMultAngle; #Angle between two particles - both pos/neg",100,0,TMath::Pi());
-  	//h_OppoChargeMultAngle = new TH1F("h_OppoChargeMultAngle", "h_OppoChargeMultAngle; #Angle between two particles - one pos, one neg",100,0,TMath::Pi());
-    h_Duplicates      = new TH1F("h_Duplicates","h_Duplicates; 0 = Correctly Identified, 1 = Misidentified",2,0,2);
-    h_Singles         = new TH1F("h_Singles","h_Singles; 0 = Correctly Identified, 1 = Misidentified",2,0,2);
-    //h_pt_dual = new TH1F
-    //h_pt_sig = new TH1F
-  }
-
   void DualTrackTaggerModule::initialize()
   {
     m_particles.isRequired();
@@ -153,20 +142,7 @@
         } // duplicats
  
         // Fill histograms
-        if (Variable::genParticleIndex(p1) == Variable::genParticleIndex(p2)) {
-          if (duplicates) {
-            h_Duplicates -> Fill(0);
-          } else {
-            h_Duplicates -> Fill(1);
-          }
-        } else {
-          if (duplicates) {
-            h_Singles -> Fill(1);
-          } else {
-            h_Singles -> Fill(0);
-          }
-        } // different genParticleIndex
-
+        fillHistograms(p1, p2, duplicates);
       } //p2
     } //p1
   }
@@ -198,7 +174,7 @@
         return false;
       }
       return false;
-      
+
     } else { // BELLE 2
       // calculate Variables
       double magDiffP = (p1->getMomentum() - p2->getMomentum() ).Mag();
@@ -244,8 +220,29 @@
     return TMath::Power(gamma*Variable::particleDRho(p),2) + TMath::Power(Variable::particleDZ(p),2);
   }
  
+
+  void DualTrackTaggerModule::defineHisto()
+  {
+    h_genParticleIndexMulti = new TH1F("h_genParticleIndexMulti","h_genParticleIndexMulti; # of particles with same genParticleIndex",20,0,20);
+    //h_SameChargeMultAngle = new TH1F("h_SameChargeMultAngle", "h_SameChargeMultAngle; #Angle between two particles - both pos/neg",100,0,TMath::Pi());
+    //h_OppoChargeMultAngle = new TH1F("h_OppoChargeMultAngle", "h_OppoChargeMultAngle; #Angle between two particles - one pos, one neg",100,0,TMath::Pi());
+    h_Duplicates      = new TH1F("h_Duplicates","True Duplicates; 0 = Correctly Identified, 1 = False Negative",2,0,2);
+    h_Singles         = new TH1F("h_Singles","True Single Track Pairs; 0 = Correctly Identified, 1 = False Positive",2,0,2);
+    h_taggedDuplicate = new TH1F("h_taggedDuplicate", "Pairs tagged as Duplicate; 0 = correctly Identified, 1 = False Positive",2,0,2);
+
+    h_TruePosPt = new TH1F("h_TruePosPt","True Positive Duplicates ;pT",100,0,0.5);
+    h_FalsePosPt = new TH1F("h_FalsePosPt","False Positive Duplicates ;pT",100,0,0.5);
+    h_TrueNegPt = new TH1F("h_TrueNegPt","True Negative Singles ;pT",100,0,0.5);
+    h_FalseNegPt = new TH1F("h_FalseNegPt","False Negative Singles ;pT",100,0,0.5);
+
+    h_effTruePosPt = new TEfficiency;
+    h_effFalsePosPt = new TEfficiency;
+  }
+
   void DualTrackTaggerModule::fill_h_genParticleIndexMulti()
   {
+    if (!m_MCFlag) return;
+
     std::vector<std::pair<int,int>> v_MCIndex;
  
     const unsigned int nParticles = m_allParticlesInList.size();
@@ -277,6 +274,59 @@
     }
   }
  
+  void DualTrackTaggerModule::fillHistograms(Particle *p1, Particle *p2, bool taggedDuplicate)
+  {
+    if (!m_MCFlag) return;
+
+    bool MCduplicate = Variable::genParticleIndex(p1) == Variable::genParticleIndex(p2);
+
+    if (MCduplicate && taggedDuplicate) {
+      h_Duplicates -> Fill(0);
+      h_taggedDuplicate -> Fill(0);
+
+      h_TruePosPt -> Fill(Variable::particlePt(p1));
+      h_TruePosPt -> Fill(Variable::particlePt(p2));
+    }
+
+    if (MCduplicate && !taggedDuplicate) {
+      h_Duplicates -> Fill(1);
+
+      h_FalseNegPt -> Fill(Variable::particlePt(p1));
+      h_FalseNegPt -> Fill(Variable::particlePt(p2));
+    }
+
+    if (!MCduplicate && taggedDuplicate) {
+      h_Singles -> Fill(1);
+      h_taggedDuplicate -> Fill(1);
+
+      h_FalsePosPt -> Fill(Variable::particlePt(p1));
+      h_FalsePosPt -> Fill(Variable::particlePt(p2));
+
+    }
+
+    if (!MCduplicate && !taggedDuplicate) {
+      h_Singles -> Fill(0);
+
+      h_TrueNegPt -> Fill(Variable::particlePt(p1));
+      h_TrueNegPt -> Fill(Variable::particlePt(p2));
+    }
+
+  } // fill histograms
+
+  void DualTrackTaggerModule::finalizeHistograms()
+  {
+    if (!m_MCFlag) return;
+
+    TH1F *h_AllDuplicatesPt = (TH1F*)h_TruePosPt->Clone();
+    h_AllDuplicatesPt -> Add(h_FalseNegPt);
+    TH1F *h_AllSinglesPt = (TH1F*)h_FalsePosPt->Clone();
+    h_AllSinglesPt -> Add(h_TrueNegPt);
+
+
+    h_effTruePosPt  = new TEfficiency(*h_TruePosPt, *h_AllDuplicatesPt);
+    h_effFalsePosPt = new TEfficiency(*h_FalsePosPt, *h_AllSinglesPt);
+  }
+
   //void DualTrackTaggerModule::GroupParticles()
   //{
   //}