Changes in examples/ExamplePVtxFit.C [0d65492:127644a] 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).
-To run compiled version:
-root
-root> .L examples/LoadPvtxFit.C+
-root> LoadPVtxFit()
-root> ExamplePVtxFit("infile.root", Nevents)
+assumed to be generated in (0,0,0)
+*/
 
-March 4, 2021
-F. Bedeschi, INFN-Pisa, Italy
-*/
+#ifdef __CLING__
+R__LOAD_LIBRARY(libDelphes)
 
 #include "classes/DelphesClasses.h"
@@ -18 +13 @@
 #include "external/TrackCovariance/VertexFit.h"
 
-#include <TClonesArray.h>
-#include <TChain.h>
-#include <TVectorD.h>
-#include <TMatrixDSym.h>
-#include <TH1.h>
-#include <TCanvas.h>
-#include <TStyle.h>
+#endif
+
 
 //------------------------------------------------------------------------------
@@ -56 +46 @@
                 treeReader->ReadEntry(entry);
                 Int_t Ntr = 0;  // # of tracks from primary vertex
-                Int_t NtrG = branchTrack->GetEntries(); // Total # of tracks
+                Int_t NtrG = branchTrack->GetEntries();
                 TVectorD** pr = new TVectorD * [NtrG];
                 TMatrixDSym** cv = new TMatrixDSym * [NtrG];
@@ -68 +58 @@
                                 Track* trk = (Track*)branchTrack->At(it);
                                 //
-                                // Get associated generated particle 
+                                // Get associated generated particle
                                 GenParticle* gp = (GenParticle*)trk->Particle.GetObject();
                                 //
-                                // Position of origin (mm)
-                                Double_t x = gp->X;
-                                Double_t y = gp->Y;
-                                Double_t z = gp->Z;
+                                // 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;
                                 //
-                                // Select group of tracks originating from the primary vertex
+                                // group tracks originating from the primary vertex
                                 if (x == 0.0 && y == 0.0)
                                 {
@@ -88 +78 @@
                                         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);
@@ -94 +85 @@
                                 }
                         }               // End loop on tracks
+                        //std::cout << "Total of " << Ntr << " primary tracks out of " << NtrG << " tracks" << std::endl;
                 }
                 //
@@ -115 +107 @@
                 }
 
+                //std::cout << "Vertex chi2/Ndof = " << Chi2 / Ndof << std::endl;
                 //
                 // Cleanup
@@ -125 +118 @@
         // Show resulting histograms
         //
-        TCanvas* Cnv = new TCanvas("Cnv", "Delphes primary vertex fit pulls", 50, 50, 900, 500);
+        TCanvas* Cnv = new TCanvas("Cnv", "Delphes generated track plots", 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);