]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - JETAN/AliFastJetFinder.cxx
by default only barrel tracks for resolution task
[u/mrichter/AliRoot.git] / JETAN / AliFastJetFinder.cxx
index fe63f9a716c305de717a0808f380163cc162a66b..896c457e8e2026b9c85acf83a0ed502d44a9cae0 100644 (file)
@@ -40,6 +40,7 @@
 #include "AliFastJetInput.h"
 #include "AliJetBkg.h"
 #include "AliAODJetEventBackground.h"
+#include "AliAODTrack.h"
 
 #include "fastjet/PseudoJet.hh"
 #include "fastjet/ClusterSequenceArea.hh"
@@ -84,11 +85,12 @@ AliFastJetFinder::~AliFastJetFinder()
 //______________________________________________________________________________
 void AliFastJetFinder::FindJets()
 {
-  cout<<"----------in AliFastJetFinder::FindJets() ------------------"<<endl;
+
   //pick up fastjet header
   AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
-  Bool_t debug  = header->GetDebug();     // debug option
+  Int_t  debug  = header->GetDebug();     // debug option
   Bool_t bgMode = header->GetBGMode();    // choose to subtract BG or not
+  if(debug)cout<<"----------in AliFastJetFinder::FindJets() ------------------"<<endl;
 
   // check if we are reading AOD jets
   TRefArray *refs = 0;
@@ -100,7 +102,7 @@ void AliFastJetFinder::FindJets()
 
   vector<fastjet::PseudoJet> inputParticles=fInputFJ->GetInputParticles();
   if(inputParticles.size()==0){
-    Printf("%s:%d No input particles found, skipping event",(char*)__FILE__,__LINE__);
+    if(debug)Printf("%s:%d No input particles found, skipping event",(char*)__FILE__,__LINE__);
     return;
   }
 
@@ -124,7 +126,11 @@ void AliFastJetFinder::FindJets()
   fastjet::AreaType areaType = header->GetAreaType();
   areaDef = fastjet::AreaDefinition(areaType,ghostSpec);
   
-  if(bgMode) // BG subtraction
+
+
+
+
+  if(bgMode) // BG subtraction!!!!!!!!!! Not entering here
     {
       //***************************** JETS FINDING AND EXTRACTION
       // run the jet clustering with the above jet definition
@@ -179,7 +185,7 @@ void AliFastJetFinder::FindJets()
        double area     = clust_seq.area(jets[j]);
        double areaError = clust_seq.area_error(jets[j]);
        
-       printf("Jet found %5d %9.5f %8.5f %10.3f %8.3f +- %6.3f\n", (Int_t)j,jets[j].rap(),jets[j].phi(),jets[j].perp(), area, areaError);
+       if(debug) printf("Jet found %5d %9.5f %8.5f %10.3f %8.3f +- %6.3f\n", (Int_t)j,jets[j].rap(),jets[j].phi(),jets[j].perp(), area, areaError);
        
        // go to write AOD  info
        AliAODJet aodjet (jets[j].px(), jets[j].py(), jets[j].pz(), jets[j].E());
@@ -191,24 +197,19 @@ void AliFastJetFinder::FindJets()
        
       }
     }
-  else { // No BG subtraction
+
+
+
+
+
+
+  else { // No BG subtraction!!!!!!!! Default header is bgmode=0.
    
-    TClonesArray* fUnit = fReader->GetUnitArray();
+    TClonesArray* fUnit = fReader->GetUnitArray(); //Big mmentum array
     if(fUnit == 0) { cout << "Could not get the momentum array" << endl; return; }
     Int_t         nIn = fUnit->GetEntries();
-    cout<<"===== check Unit Array in AliFastJetFinder ========="<<endl;
-    Int_t ppp=0;
-    for(Int_t ii=0; ii<nIn; ii++) 
-      {
-       AliJetUnitArray *uArray = (AliJetUnitArray*)fUnit->At(ii);
-       if(uArray->GetUnitEnergy()>0.){
-         Float_t eta   = uArray->GetUnitEta();
-         Float_t phi   = uArray->GetUnitPhi();
-         cout<<"ipart = "<<ppp<<" eta="<<eta<<"  phi="<<phi<<endl;
-         ppp++;
-       }
-      }
 
     //fastjet::ClusterSequence clust_seq(inputParticles, jetDef); 
     fastjet::ClusterSequenceArea clust_seq(inputParticles, jetDef, areaDef);
    
@@ -236,51 +237,51 @@ void AliFastJetFinder::FindJets()
     vector<fastjet::PseudoJet> jets = sorted_by_pt(inclusiveJets); // Added by me
     for (size_t j = 0; j < jets.size(); j++) { // loop for jets     // Added by me
       
-      printf("Jet found %5d %9.5f %8.5f %10.3f \n",(Int_t)j,jets[j].rap(),jets[j].phi(),jets[j].perp());
+      if(debug) printf("Jet found %5d %9.5f %8.5f %10.3f \n",(Int_t)j,jets[j].rap(),jets[j].phi(),jets[j].perp());
 
       vector<fastjet::PseudoJet> constituents = clust_seq.constituents(jets[j]);
       int nCon= constituents.size();
       TArrayI ind(nCon);
       Double_t area=clust_seq.area(jets[j]);
-      cout<<"area = "<<area<<endl;
       // go to write AOD  info
       AliAODJet aodjet (jets[j].px(), jets[j].py(), jets[j].pz(), jets[j].E());
       aodjet.SetEffArea(area,0);
       //cout << "Printing jet " << endl;
       if(debug) aodjet.Print("");
-      // cout << "Adding jet ... " <<j<<endl;
+      Int_t count=0;
       for (int i=0; i < nCon; i++)
        {
        fastjet::PseudoJet mPart=constituents[i];
        ind[i]=mPart.user_index();
-       //cout<<i<<"  index="<<ind[i]<<endl;
+       //      cout<<i<<"  index="<<ind[i]<<endl;
        
-       //internal oop over all the unit cells
+       //internal loop over all the unit cells
        Int_t ipart = 0;
+       //      count=0;
        for(Int_t ii=0; ii<nIn; ii++) 
          {
            AliJetUnitArray *uArray = (AliJetUnitArray*)fUnit->At(ii);
            if(uArray->GetUnitEnergy()>0.){
              uArray->SetUnitTrackID(ipart);//used to have the index available in AliJEtBkg
              if(ipart==ind[i]){
-               aodjet.AddTrack(uArray);
+               TRefArray* trackArray = (TRefArray*)uArray->GetUnitTrackRef();
+               Int_t tracksInCell = trackArray->GetEntries();
+               for(int ji = 0; ji < tracksInCell; ji++){
+                 AliAODTrack * track = (AliAODTrack*)trackArray->At(ji);
+                 aodjet.AddTrack(track);
+               }
+
+               count++;
              }
              ipart++;
            }
          }
       }
 
-      AddJet(aodjet);
-      //cout << "added \n" << endl;
-
-
-      ///----> set in the aod the reference to the unit cells
-      //  in FastJetFinder: 1) loop over the unit array. 2) select those unit cells belonging to the jet (via user_index). 3) use AliAODJet->AddTrack(unitRef)
-      //  in AliJetBkg: 1) loop over the unit arrays --> get ind of the unit cell 2) internal loop on jet unit cells; 3) check if i_cell = UID of the trackRefs of the AODJet
-      // should work hopefully
-
+    
 
-      
+      AddJet(aodjet);
+           
     } // end loop for jets
   } 
 
@@ -468,15 +469,15 @@ Bool_t AliFastJetFinder::ProcessEvent2()
   fJetBkg->SetHeader(fHeader);
   fJetBkg->SetReader(fReader);
   fJetBkg->SetFastJetInput(fInputFJ);
-  Double_t bkg1=fJetBkg->BkgFastJet();
-  Double_t bkg2=fJetBkg->BkgChargedFastJet();
-  Double_t bkg3=fJetBkg->BkgFastJetCone(fAODjets);
-  Double_t bkg4=fJetBkg->BkgRemoveJetLeading(fAODjets);
+//  Double_t bkg1=fJetBkg->BkgFastJet();
+//  Double_t bkg2=fJetBkg->BkgChargedFastJet();
+//  Double_t bkg3=fJetBkg->BkgFastJetCone(fAODjets);
+//  Double_t bkg4=fJetBkg->BkgRemoveJetLeading(fAODjets);
   
-  fAODEvBkg->SetBackground(0,bkg1);
-  fAODEvBkg->SetBackground(1,bkg2);
-  fAODEvBkg->SetBackground(2,bkg3);
-  fAODEvBkg->SetBackground(3,bkg4);
+//  fAODEvBkg->SetBackground(0,bkg1);
+//  fAODEvBkg->SetBackground(1,bkg2);
+//  fAODEvBkg->SetBackground(2,bkg3);
+//  fAODEvBkg->SetBackground(3,bkg4);
   
   Int_t nEntRef    = ref->GetEntries();