Changeset 170a11d in git for external/TrackCovariance/SolGridCov.cc

Timestamp:

Dec 14, 2020, 10:53:31 AM (4 years ago)

Author:

michele <michele.selvaggi@…>

Branches:

master

Children:

a0db751

Parents:

527f67a

Message:

included acceptance and hits in TrackCovariance

File:

• external/TrackCovariance/SolGridCov.cc (modified) (3 diffs)

external/TrackCovariance/SolGridCov.cc

@@ -3 +3 @@
 #include <TMath.h>
 #include <TVectorD.h>
+#include <TVector3.h>
 #include <TMatrixDSym.h>
 #include <TDecompChol.h>
@@ -36 +37 @@
 {
   delete[] fCov;
+  delete fAcc;
 }
 
@@ -58 +60 @@
     }
   }
-}
+
+// Now make acceptance
+fAcc = new AcceptanceClx(G);
+}
+
+
+//
+Bool_t SolGridCov::IsAccepted(Double_t pt, Double_t Theta)
+{
+        //
+        // pt in GeV, Theta in degrees
+        //
+        Bool_t Accept = kFALSE;
+        if (fAcc->HitNumber(pt, Theta) >= fNminHits)Accept = kTRUE;
+        //
+        return Accept;
+}
+//
+Bool_t SolGridCov::IsAccepted(Double_t *p)
+{
+        //
+        // pt in GeV, Theta in degrees
+        //
+        Bool_t Accept = kFALSE;
+        Double_t pt = TMath::Sqrt(p[0] * p[0] + p[1] * p[1]);
+        Double_t th = 180. * TMath::ATan2(pt, p[2])/TMath::Pi();
+        if (fAcc->HitNumber(pt,th) >= fNminHits)Accept = kTRUE;
+        //
+        return Accept;
+}
+//
+Bool_t SolGridCov::IsAccepted(TVector3 p)
+{
+        //
+        // pt in GeV, Theta in degrees
+        //
+        Bool_t Accept = kFALSE;
+        Double_t pt = p.Pt();
+        Double_t th = 180.*TMath::ACos(p.CosTheta())/TMath::Pi();
+        if (fAcc->HitNumber(pt,th) >= fNminHits)Accept = kTRUE;
+        //
+        return Accept;
+}
+
+
 // Find bin in grid
 Int_t SolGridCov::GetMinIndex(Double_t xval, Int_t N, TVectorD x)