Changeset 7bca620 in git for external/TrackCovariance

Timestamp:

Feb 10, 2022, 11:42:45 AM (3 years ago)

Author:

Franco BEDESCHI <bed@…>

Branches:

master

Children:

5c03893

Parents:

b750b0a

Message:

Add feature to force starting radius in vertex fit

Location:

external/TrackCovariance

Files:

• ObsTrk.cc (modified) (2 diffs)
• VertexFit.cc (modified) (8 diffs)
• VertexFit.h (modified) (2 diffs)

external/TrackCovariance/ObsTrk.cc

@@ -74 +74 @@
         fCovILC.ResizeTo(5, 5);
         fGenPar = XPtoPar(fGenX, fGenP, Q);
+        fGenParMm = ParToMm(fGenPar);
         fGenParACTS = ParToACTS(fGenPar);
         fGenParILC = ParToILC(fGenPar);
         //
         fObsPar = GenToObsPar(fGenPar);
+        fObsParMm = ParToMm(fObsPar);
         fObsParACTS = ParToACTS(fObsPar);
         fObsParILC = ParToILC(fObsPar);
@@ -83 +85 @@
         fObsP = ParToP(fObsPar);
         fObsQ = ParToQ(fObsPar);
+        fCovMm = CovToMm(fCov);
         fCovACTS = CovToACTS(fObsPar, fCov);
         fCovILC = CovToILC(fCov);

external/TrackCovariance/VertexFit.cc

@@ -17 +17 @@
 {
         fNtr = 0;
-        fRold = -1.0;
+        fRstart = -1.0;
         fVtxDone = kFALSE;
         fVtxCst = kFALSE;
@@ -31 +31 @@
 {
         fNtr = Ntr;
-        fRold = -1.0;
+        fRstart = -1.0;
         fVtxDone = kFALSE;
         fVtxCst = kFALSE;
@@ -55 +55 @@
 {
         fNtr = Ntr;
-        fRold = -1.0;
+        fRstart = -1.0;
         fVtxDone = kFALSE;
         fVtxCst = kFALSE;
@@ -220 +220 @@
         std::vector<TVectorD*> x0i;                             // Tracks at ma
         std::vector<TVectorD*> ni;                              // Track derivative wrt phase
-        std::vector<TMatrixDSym*> Ci;                   // Position error matrix at fixed phase
+        std::vector<TMatrixDSym*> Ci;                           // Position error matrix at fixed phase
         std::vector<TVectorD*> wi;                              // Ci*ni
+        std::vector<Double_t> s_in;                             // Starting phase
+        //
+        // 
         //
         // Track loop
         for (Int_t i = 0; i < fNtr; i++)
         {
-                Double_t s = 0.;
                 TVectorD par = *fPar[i];
                 TMatrixDSym Cov = *fCov[i];
-                x0i.push_back(new TVectorD(Fill_x0(par)));
+                Double_t s = 0.;
+                // Case when starting radius is provided
+                if(fRstart > TMath::Abs(par(0))){
+                        s = 2.*TMath::ASin(par(2)*TMath::Sqrt((fRstart*fRstart-par(0)*par(0))/(1.+2.*par(2)*par(0))));
+                }
+                //
+                x0i.push_back(new TVectorD(Fill_x(par, s)));
                 ni.push_back(new TVectorD(derXds(par, s)));
                 TMatrixD A = derXdPar(par, s);
@@ -235 +243 @@
                 TMatrixDSym Cinv = RegInv(*Ci[i]);
                 wi.push_back(new TVectorD(Cinv * (*ni[i])));
+                s_in.push_back(s);
         }
         //std::cout << "Vtx init completed. fNtr = "<<fNtr << std::endl;
@@ -262 +271 @@
         for (Int_t i = 0; i < fNtr; i++){
                 Double_t si = Dot(*wi[i], fXv - (*x0i[i])) / Ci[i]->Similarity(*wi[i]);
-                ffi.push_back(si);
+                ffi.push_back(si+s_in[i]);
                 //TVectorD xvi = Fill_x(*fPar[i],si);
                 //std::cout << "Fast vertex "<<i<<": xvi = "<<xvi(0)<<", "<<xvi(1)<<", "<<xvi(2) 
@@ -280 +289 @@
         Ci.clear(); 
         wi.clear();
+        s_in.clear();
 }
 //
@@ -402 +412 @@
         //
         fVtxDone = kTRUE;               // Set fit completion flag
-        fRold = TMath::Sqrt(fXv(0)*fXv(0) + fXv(1)*fXv(1));     // Store fit 
+        //fRstart = TMath::Sqrt(fXv(0)*fXv(0) + fXv(1)*fXv(1)); // Store fit 
         //std::cout << "Found vertex " << fXv(0) << ", " << fXv(1) << ", " << fXv(2) 
         //      << ", after "<<Ntry<<" iterations"<<std::endl;

external/TrackCovariance/VertexFit.h

@@ -36 +36 @@
         // Results
         Bool_t fVtxDone;                        // Flag vertex fit completed
-        Double_t fRold;                         // Current value of vertex radius
+        Double_t fRstart;                       // Starting value of vertex radius (0 = none)
         TVectorD fXv;                           // Found vertex
         TMatrixDSym fcovXv;                     // Vertex covariance
@@ -83 +83 @@
         // Handle tracks/constraints
         void AddVtxConstraint(TVectorD xv, TMatrixDSym cov);    // Add gaussian vertex constraint
-        void AddTrk(TVectorD *par, TMatrixDSym *Cov);                           // Add track to input list
-        void RemoveTrk(Int_t iTrk);                                                             // Remove iTrk track
+        void AddTrk(TVectorD *par, TMatrixDSym *Cov);           // Add track to input list
+        void RemoveTrk(Int_t iTrk);                             // Remove iTrk track
+        void SetStartR(Double_t R) { fRstart = R; };            // Set starting radius
         //
 };