Changes in examples/ExamplePVtxFit.C [127644a:0d65492] in git

File:

• examples/ExamplePVtxFit.C (modified) (8 diffs)

examples/ExamplePVtxFit.C

@@ -1 +1 @@
 /*
 Example of using vertex fitter class to fit primary vertex
-assumed to be generated in (0,0,0)
+assumed to be generated in (0,0,0).
+To run compiled version:
+root
+root> .L examples/LoadPvtxFit.C+
+root> LoadPVtxFit()
+root> ExamplePVtxFit("infile.root", Nevents)
+
+March 4, 2021
+F. Bedeschi, INFN-Pisa, Italy
 */
-
-#ifdef __CLING__
-R__LOAD_LIBRARY(libDelphes)
 
 #include "classes/DelphesClasses.h"
@@ -13 +18 @@
 #include "external/TrackCovariance/VertexFit.h"
 
-#endif
-
+#include <TClonesArray.h>
+#include <TChain.h>
+#include <TVectorD.h>
+#include <TMatrixDSym.h>
+#include <TH1.h>
+#include <TCanvas.h>
+#include <TStyle.h>
 
 //------------------------------------------------------------------------------
@@ -46 +56 @@
                 treeReader->ReadEntry(entry);
                 Int_t Ntr = 0;  // # of tracks from primary vertex
-                Int_t NtrG = branchTrack->GetEntries();
+                Int_t NtrG = branchTrack->GetEntries(); // Total # of tracks
                 TVectorD** pr = new TVectorD * [NtrG];
                 TMatrixDSym** cv = new TMatrixDSym * [NtrG];
@@ -58 +68 @@
                                 Track* trk = (Track*)branchTrack->At(it);
                                 //
-                                // Get associated generated particle
+                                // Get associated generated particle 
                                 GenParticle* gp = (GenParticle*)trk->Particle.GetObject();
                                 //
-                                // Position of origin in meters
-                                Double_t x = 1.0e-3 * gp->X;
-                                Double_t y = 1.0e-3 * gp->Y;
-                                Double_t z = 1.0e-3 * gp->Z;
+                                // Position of origin (mm)
+                                Double_t x = gp->X;
+                                Double_t y = gp->Y;
+                                Double_t z = gp->Z;
                                 //
-                                // group tracks originating from the primary vertex
+                                // Select group of tracks originating from the primary vertex
                                 if (x == 0.0 && y == 0.0)
                                 {
@@ -78 +88 @@
                                         Double_t oPar[5] = { obsD0, obsPhi, obsC, obsZ0, obsCtg };
                                         TVectorD obsPar(5, oPar);       // Fill observed parameters
-                                        TVector3 xv(x, y, z);
                                         //
                                         pr[Ntr] = new TVectorD(obsPar);
@@ -85 +94 @@
                                 }
                         }               // End loop on tracks
-                        //std::cout << "Total of " << Ntr << " primary tracks out of " << NtrG << " tracks" << std::endl;
                 }
                 //
@@ -107 +115 @@
                 }
 
-                //std::cout << "Vertex chi2/Ndof = " << Chi2 / Ndof << std::endl;
                 //
                 // Cleanup
@@ -118 +125 @@
         // Show resulting histograms
         //
-        TCanvas* Cnv = new TCanvas("Cnv", "Delphes generated track plots", 50, 50, 900, 500);
+        TCanvas* Cnv = new TCanvas("Cnv", "Delphes primary vertex fit pulls", 50, 50, 900, 500);
         Cnv->Divide(2, 2);
-        Cnv->cd(1); gPad->SetLogy(1); gStyle->SetOptFit(1111);
+        Cnv->cd(1); gPad->SetLogy(1); gStyle->SetOptFit(1111);  
         hXpull->Fit("gaus"); hXpull->Draw();
-        Cnv->cd(2); gPad->SetLogy(1); gStyle->SetOptFit(1111);
+        Cnv->cd(2); gPad->SetLogy(1); gStyle->SetOptFit(1111);  
         hYpull->Fit("gaus"); hYpull->Draw();
         Cnv->cd(3); gPad->SetLogy(1); gStyle->SetOptFit(1111);