Changeset 307 in svn for trunk/src/JetsUtil.cc

Timestamp:

Mar 9, 2009, 9:47:21 PM (16 years ago)

Author:

severine ovyn

Message:

LHCO advanced

File:

• trunk/src/JetsUtil.cc (modified) (6 diffs)

trunk/src/JetsUtil.cc

@@ -120 +120 @@
 }
 
-vector<fastjet::PseudoJet> JetsUtil::RunJets(const vector<fastjet::PseudoJet>&  input_particles)
+vector<fastjet::PseudoJet> JetsUtil::RunJets(const vector<fastjet::PseudoJet>&  input_particles,  const vector<TRootTracks> & TrackCentral,  vector<int> &NTrackJet)
 {
   inclusive_jets.clear();
@@ -131 +131 @@
       double ptmin = 5.0;
       inclusive_jets = clust_seq.inclusive_jets(ptmin);
+      
       // sort jets into increasing pt
       sorted_jets = sorted_by_pt(inclusive_jets);
+
+      //check number of tracks in jets 
+      float  iEtaTrack[TrackCentral.size()];
+      float  iPhiTrack[TrackCentral.size()];
+      for(unsigned int t = 0; t < TrackCentral.size(); t++)
+         {
+           DET->BinEtaPhi(TrackCentral[t].PhiOuter,TrackCentral[t].EtaOuter,iPhiTrack[t],iEtaTrack[t]); 
+         }
+      int numTrackJet;
+      for (unsigned int i = 0; i < sorted_jets.size(); i++) 
+          {
+            numTrackJet=0;
+            vector<fastjet::PseudoJet> constituents = clust_seq.constituents(sorted_jets[i]);
+            for(unsigned int it = 0; it < TrackCentral.size(); it++)
+               {
+                 for (unsigned int j = 0; j < constituents.size(); j++)
+                     {
+                       if(DeltaR(iPhiTrack[it], iEtaTrack[it], constituents[j].phi(), constituents[j].eta())<0.001)numTrackJet++;
+                     }
+               }
+            NTrackJet.push_back(numTrackJet);
+       }
     }
   
@@ -138 +161 @@
 }
 
-void JetsUtil::RunJetBtagging(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchJet,const vector<fastjet::PseudoJet> & sorted_jets,const TSimpleArray<TRootGenParticle>& NFCentralQ)
+
+
+void JetsUtil::RunJetBtagging(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchJet,const vector<fastjet::PseudoJet> & sorted_jets,const TSimpleArray<TRootGenParticle>& NFCentralQ, const vector<int> &NTrackJet)
 {
   TRootJet *elementJet;
   TLorentzVector JET;
   for (unsigned int i = 0; i < sorted_jets.size(); i++) {
+
     JET.SetPxPyPzE(sorted_jets[i].px(),sorted_jets[i].py(),sorted_jets[i].pz(),sorted_jets[i].E());
     if(JET.Pt() > DET->PTCUT_jet)
@@ -148 +174 @@
         elementJet = (TRootJet*) branchJet->NewEntry();
         elementJet->Set(JET);
+        elementJet->NTracks = NTrackJet[i];
+
         // b-jets
         bool btag=false;
@@ -157 +185 @@
 }
 
-//void JetsUtil::RunTauJets(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchTauJet,const vector<fastjet::PseudoJet> & sorted_jets,const vector<PhysicsTower> & towers, const vector<TLorentzVector> & TrackCentral)
-void JetsUtil::RunTauJets(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchTauJet,const vector<fastjet::PseudoJet> & sorted_jets,const vector<PhysicsTower> & towers, const vector<TRootTracks> & TrackCentral)
+void JetsUtil::RunTauJets(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchTauJet,const vector<fastjet::PseudoJet> & sorted_jets,const vector<PhysicsTower> & towers, const vector<TRootTracks> & TrackCentral, const vector<int> &NTrackJet)
 {
   TRootTauJet *elementTauJet;
@@ -176 +203 @@
         elementTauJet = (TRootTauJet*) branchTauJet->NewEntry();
         elementTauJet->Set(JET);
+        elementTauJet->NTracks=NTrackJet[i];
         elementTauJet->Charge = charge;
       } // if tau jet