Changeset 34c28a9 in git

Timestamp:

Dec 9, 2019, 11:07:13 PM (5 years ago)

Author:

Kaan Yüksel Oyulmaz <kaanyukseloyulmaz@…>

Branches:

Timing

Children:

9117aaa4

Parents:

1363bf5

Message:

FCC-hh card with 4DVertex Algorithm

Files:

• cards/FCC/FCChh_PileUpVtx.tcl (modified) (8 diffs)
• cards/FCC/trackResolutionCMS.tcl (added)
• modules/TrackPileUpSubtractor.cc (modified) (4 diffs)

cards/FCC/FCChh_PileUpVtx.tcl

@@ -12 +12 @@
 #######################################
 
+set MaxEvents 100
+set RandomSeed 123
+
 set ExecutionPath {
 
+  BeamSpotFilter
+  PileUpMerger
   ParticlePropagator
 
@@ -25 +30 @@
 
   TrackMerger
+
+  TrackSmearing
+  TimeSmearing  
+
+  VertexFinderDA4D  
+
+  TrackPileUpSubtractor  
 
   ECal
@@ -68 +80 @@
 }
 
+#######################
+# GenBeamSpotFilter
+# Saves a particle intended to represent the beamspot
+#######################
+
+module BeamSpotFilter BeamSpotFilter {
+    set InputArray Delphes/stableParticles
+    set OutputArray beamSpotParticle
+
+}
+
+###############
+# PileUp Merger
+###############
+
+module PileUpMerger PileUpMerger {
+  set InputArray Delphes/stableParticles
+
+  set ParticleOutputArray stableParticles
+  set VertexOutputArray vertices
+
+  # pre-generated minbias input file
+  set PileUpFile MinBias.pileup
+
+  # average expected pile up
+  set MeanPileUp 10
+
+  # 0-poisson, 1-uniform, 2-delta
+  set PileUpDistribution 2
+
+  # maximum spread in the beam direction in m
+  set ZVertexSpread 0.25
+
+  # maximum spread in time in s
+  set TVertexSpread 800E-12
+
+  # vertex smearing formula f(z,t) (z,t need to be respectively given in m,s)
+
+  #set VertexDistributionFormula {exp(-(t^2/(2*(0.063/2.99792458E8*exp(-(z^2/(2*(0.063)^2))))^2)))}
+  set VertexDistributionFormula {exp(-(t^2/160e-12^2/2))*exp(-(z^2/0.053^2/2))}
+
+  # taking 5.3 cm x 160 ps
+
+  #set VertexDistributionFormula { (abs(t) <= 160e-12) * (abs(z) <= 0.053) * (1.00) +
+  #                                (abs(t) >  160e-12) * (abs(z) <= 0.053) * (0.00) +
+  #                               (abs(t) <= 160e-12) * (abs(z) > 0.053)  * (0.00) +
+  #                           (abs(t) >  160e-12) * (abs(z) > 0.053)  * (0.00)}
+
+}
+
 #####################################
 # Track propagation to calorimeters
@@ -73 +135 @@
 
 module ParticlePropagator ParticlePropagator {
-  set InputArray Delphes/stableParticles
+  set InputArray PileUpMerger/stableParticles
   set OutputArray stableParticles
   set ChargedHadronOutputArray chargedHadrons
@@ -203 +265 @@
 
 
-#### ADD TRACKSMEARING
-#### ADD TIMESMEARING
-
+########################################
+#   Smear tracks
+########################################
+
+module TrackSmearing TrackSmearing {
+  set InputArray TrackMerger/tracks
+  set OutputArray tracks
+  set ApplyToPileUp true
+
+  # from http://mersi.web.cern.ch/mersi/layouts/.private/Baseline_tilted_200_Pixel_1_1_1/index.html
+  source trackResolutionCMS.tcl 
+  # FIXME !!!! we need to add track resolution of FCC-hh baseline detector !!!!!
+}
+
+########################################
+#   Time Smearing
+########################################
+
+module TimeSmearing TimeSmearing {
+  set InputArray TrackSmearing/tracks
+  set OutputArray tracks
+
+  # assume 20 ps resolution for now
+  set TimeResolution 20E-12
+}
+
+##################################
+# Primary vertex reconstruction
+##################################
+
+
+module VertexFinderDA4D VertexFinderDA4D {
+  set InputArray TimeSmearing/tracks
+
+  set OutputArray tracks
+  set VertexOutputArray vertices
+
+  set Verbose 0
+  set MinPT 1.0
+
+  # in mm
+  set VertexSpaceSize 0.5
+
+  # in s
+  set VertexTimeSize 10E-12
+
+  set UseTc 1
+  set BetaMax 0.1
+  set BetaStop 1.0
+  set CoolingFactor 0.8
+  set MaxIterations 100
+
+  # in mm
+  set DzCutOff 40
+  set D0CutOff 30
+
+}
+
+##########################
+# Track pile-up subtractor
+##########################
+
+module TrackPileUpSubtractor TrackPileUpSubtractor {
+# add InputArray InputArray OutputArray
+
+  add InputArray ChargedHadronMomentumSmearing/chargedHadrons
+  add InputArray ElectronMomentumSmearing/electrons
+  add InputArray MuonMomentumSmearing/muons
+  
+  set VertexInputArray VertexFinderDA4D/vertices
+  # assume perfect pile-up subtraction for tracks with |z| > fZVertexResolution
+  # Z vertex resolution in m
+  set ZVertexResolution {0.0001}
+}
 
 
@@ -216 +349 @@
 module SimpleCalorimeter ECal {
   set ParticleInputArray ParticlePropagator/stableParticles
-  set TrackInputArray TrackMerger/tracks
+  set TrackInputArray TrackSmearing/tracks
 
   set TowerOutputArray ecalTowers
@@ -373 +506 @@
 }
 
+
 #################
 # Electron filter
@@ -891 +1025 @@
   add Branch MissingET/momentum MissingET MissingET
   add Branch ScalarHT/energy ScalarHT ScalarHT
-}
-
+  add Branch VertexFinderDA4D/vertices Vertex4D Vertex
+}
+

modules/TrackPileUpSubtractor.cc

@@ -117 +117 @@
 void TrackPileUpSubtractor::Process()
 {
-  Candidate *candidate, *particle;
+  Candidate *candidate, *particle, *particleTest;
   map<TIterator *, TObjArray *>::iterator itInputMap;
   TIterator *iterator;
@@ -123 +123 @@
   Double_t z, zvtx = 0;
   Double_t pt, eta, phi, e;
-
+  Double_t sumPT2 = 0;
+  Double_t sumSquare = 0;
+  int counter = 0;
+  Double_t tempPTSquare = 0;
+  Double_t tempZVertex = 0; 
   // find z position of primary vertex
 
+  cout << " ---------- NEW EVENT --------- " << endl; 
+
   fItVertexInputArray->Reset();
+  cout << " NUMBER OF VERTICES : " << fVertexInputArray->GetEntriesFast() << endl;
   while((candidate = static_cast<Candidate *>(fItVertexInputArray->Next())))
   {
-    if(!candidate->IsPU)
-    {
-      zvtx = candidate->Position.Z();
-      // break;
-    }
+      cout << " ---------- NEW VERTEX --------- " << candidate->IsPU << endl; 
+      tempZVertex = candidate->Position.Z();
+      tempPTSquare = candidate->SumPT2;
+      if(tempPTSquare > sumSquare)
+      {
+        sumSquare = tempPTSquare;
+        zvtx = tempZVertex;
+        cout << " Sum Square : " << sumSquare << " Z of Vertex : " << zvtx << "  Is PileUp : " << candidate->IsPU << endl;
+      } 
+      
+      /*
+      for(int i = 0; i < candidate->GetCandidates()->GetEntriesFast(); i++)
+      {
+        particleTest = static_cast<Candidate *>(candidate->GetCandidates()->At(i));
+        TLorentzVector &candidateMomentumNew = particleTest->Momentum;
+        if(candidateMomentumNew.Pt() > 1.0)
+          sumSquare += (candidateMomentumNew.Pt()*candidateMomentumNew.Pt());
+        cout << candidateMomentumNew.Pt() << "  " << candidate->SumPT2 << " counter : " << candidate->GetCandidates()->GetEntriesFast() << endl;
+      }  
+      */
   }
 
@@ -147 +169 @@
     {
       particle = static_cast<Candidate *>(candidate->GetCandidates()->At(0));
-      const TLorentzVector &candidateMomentum = particle->Momentum;
+      const TLorentzVector &candidateMomentum = candidate->Momentum;
 
       eta = candidateMomentum.Eta();
@@ -154 +176 @@
       e = candidateMomentum.E();
 
-      z = particle->Position.Z();
+      z = candidate->Position.Z();
+      counter ++; 
+      //sum = pt*pt;
+      if(pt > 1.0)
+        sumPT2 += pt*pt;
 
       // apply pile-up subtraction
       // assume perfect pile-up subtraction for tracks outside fZVertexResolution
-
-      if(candidate->Charge != 0 && candidate->IsPU && TMath::Abs(z - zvtx) > fFormula->Eval(pt, eta, phi, e) * 1.0e3)
+      // cout << particle->IsRecoPU << " ParticleSumSquare : " << sumPT2 << " VertexResult : " << sumSquare << " Added SumSquare : " << zvtx << endl;
+      //cout << pt << "  " << candidate->IsPU << "  " <<  TMath::Abs(z - zvtx) << "   " << sumPT2 << "   " << sumPT2 << endl;
+
+      
+      if(particle->Charge != 0 && TMath::Abs(z - zvtx) > fFormula->Eval(pt, eta, phi, e) * 1.0e3)
       {
         candidate->IsRecoPU = 1;
+        //cout <<  TMath::Abs(z - zvtx) << "   " << fFormula->Eval(pt, eta, phi, e) * 1.0e3 << endl;
       }
       else