[u/mrichter/AliRoot.git] / EMCAL / AliEMCALJetFinderAlgoUA1Revised2.cxx

//SARAH'S REVISED PERSONAL COPY!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!


/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//*--Author: Sarah Blyth (LBL)
//*--Based on UA1 jet algorithm from LUND JETSET called from EMC-erj

#include "TTask.h"
#include "AliEMCALJetFinderInput.h"
#include "AliEMCALJetFinderOutput.h"
#include "AliEMCALJetFinderAlgo.h"
#include "AliEMCALJetFinderAlgoUA1Revised2.h"
#include "AliEMCALJetFinderAlgoUA1Unit.h"
#include "AliEMCALGeometry.h"
#include "AliEMCAL.h"
#include "AliEMCALDigit.h"
#include "TParticle.h"
#include "AliRun.h"
#include "AliEMCALJet.h"
#include "TMath.h"


ClassImp(AliEMCALJetFinderAlgoUA1Revised2)

  AliEMCALJetFinderAlgoUA1Revised2::AliEMCALJetFinderAlgoUA1Revised2()
{
  //Default constructor
if (fDebug>0) Info("AliEMCALJetFinderAlgoUA1Revised","Beginning Default Constructor");

  fNumIter           = 0;
  fNumUnits          = 13824;     //Number of towers in EMCAL
  fESeed             = 5.0;       //Default value
  fConeRad           = 0.5;       //Default value
  fJetEMin           = 10.0;      //Default value
  fEtMin             = 0.28;      //Default value
  fMinMove           = 0.05;      //From original UA1 JetFinder
  fMaxMove           = 0.15;      //From original UA1 JetFinder
  fBGMaxMove         = 0.035;     //From original UA1 JetFinder
  fPtCut             = 0;         
  fHadCorr           = 0;       
  fEBGTotal          = 1.0;       //Set to 1 so that no div by zero in first FindJets() loop
  fEBGTotalOld       = 0.0;
  fEBGAve            = 0.0;
  fEnergy            = 0.0;
  fJetEta            = 0.0;
  fJetPhi            = 0.0;
  fEtaInit           = 0.0;
  fPhiInit           = 0.0;
  fEtaB              = 0.0;
  fPhiB              = 0.0;
  fJetESum           = 0.0;
  fJetEtaSum         = 0.0;
  fJetPhiSum         = 0.0;
  fDEta              = 0.0;
  fDPhi              = 0.0;
  fDistP             = 0.0;
  fDistI             = 0.0;
  fTempE             = 0.0;
  fRad               = 2.0;      //Set to 2 to start 
  fNumInCone         = 0;
  fNumJets           = 0;
  fArrayInitialised  = 0;        //Set to FALSE to start
}

 AliEMCALJetFinderAlgoUA1Revised2::~AliEMCALJetFinderAlgoUA1Revised2()
   {
     //Destructor
     if (fDebug>0) Info("AliEMCALJetFinderAlgoUA1Revised","Beginning Destructor");
     delete[] fUnit;
   }

 void AliEMCALJetFinderAlgoUA1Revised2::SetJetFindingParameters
                               (Int_t numUnits, Float_t eSeed, Float_t coneRad, Float_t jetEMin, Float_t etMin, 
                               Float_t minMove, Float_t maxMove, Float_t bgMaxMove)
   {
     //Sets parameters for the JetFinding algorithm
     if (fDebug>1) Info("SetJetFindingParameters","Setting parameters for JetFinding");

     SetNumUnits(numUnits);
     SetJetESeed(eSeed);
     SetConeRad(coneRad);
     SetJetEMin(jetEMin);
     SetEtMin(etMin);
     SetMinMove(minMove);
     SetMaxMove(maxMove);
     SetBGMaxMove(bgMaxMove);
   }

 void AliEMCALJetFinderAlgoUA1Revised2::SetJetFindingParameters
                               (Int_t numUnits, Float_t eSeed, Float_t coneRad, Float_t jetEMin, Float_t etMin)
   {
     //Sets fewer parameters for the JetFinding algorithm
     if (fDebug>1) Info("SetJetFindingParameters","Setting parameters for JetFinding");

     SetNumUnits(numUnits);
     SetJetESeed(eSeed);
     SetConeRad(coneRad);
     SetJetEMin(jetEMin);
     SetEtMin(etMin);
     SetMinMove(fMinMove);
     SetMaxMove(fMaxMove);
     SetBGMaxMove(fBGMaxMove);
   }

 void AliEMCALJetFinderAlgoUA1Revised2::InitUnitArray()
   {
     //Initialises unit array
     if(fArrayInitialised) delete[] fUnit;
     fUnit = new AliEMCALJetFinderAlgoUA1Unit[fNumUnits];
     fArrayInitialised = 1;
   }

 void AliEMCALJetFinderAlgoUA1Revised2::FillUnitArray(AliEMCALJetFinderAlgoUA1FillUnitFlagType_t flag)
   {
     if (fDebug>1) Info("FillUnitArray","Beginning FillUnitArray");
     AliEMCAL* pEMCAL = (AliEMCAL*) gAlice->GetModule("EMCAL");

         //   if (pEMCAL){ 
         //	     AliEMCALGeometry* geom =  AliEMCALGeometry::GetInstance(pEMCAL->GetTitle(), "");
         //     }else
         //    {
     AliEMCALGeometry* geom =  AliEMCALGeometry::GetInstance("EMCAL_5655_21", "");
        //    }
         
     AliEMCALJetFinderAlgoUA1FillUnitFlagType_t option = flag;
     Int_t         numTracks, numDigits;
    
     //Loops over all elements in the AliEMCALJetFinderAlgoUA1Unit array and 
     //fills the objects with relevant values from the Data Input object
     if (fDebug>1) Info("FillUnitArray","Filling array with Unit objects");
     if (fDebug>1) Info("FillUnitArray","NTracks %i NDigits %i",fInputPointer->GetNTracks(),fInputPointer->GetNDigits());
	 numTracks = fInputPointer->GetNTracks();
	 numDigits = fInputPointer->GetNDigits();
	 TParticle         *myPart;
	 AliEMCALDigit     *myDigit;

	 //Fill units with Track info if appropriate
	 if(option==kFillTracksOnly || option ==kFillAll) 
	   {          
	    for(Int_t j=0; j<numTracks; j++)
	    {
	     myPart = fInputPointer->GetTrack(j);
	     Float_t eta = myPart->Eta();
	     Float_t  phi = myPart->Phi();
	     Int_t towerID = geom->TowerIndexFromEtaPhi(eta,180.0/TMath::Pi()*phi);
	     Float_t  pT = myPart->Pt();
	     Float_t unitEnergy = fUnit[towerID-1].GetUnitEnergy(); 

   	     //Do Hadron Correction
              if(fHadCorr != 0)
	       {
		 Double_t   fullP = myPart->P();
		 Double_t   hCEnergy = fHadCorr->GetEnergy(fullP, (Double_t)eta);
		 unitEnergy -= hCEnergy*TMath::Sin(myPart->Theta());
		 fUnit[towerID-1].SetUnitEnergy(unitEnergy);
	       } //end Hadron Correction loop 
	     
	     //Do Pt cut on tracks
	     if(fPtCut != 0 && pT < fPtCut) continue;

	     fUnit[towerID-1].SetUnitEnergy(unitEnergy+pT);

             }//end tracks loop
	   }//end Tracks condition


	 //Fill units with Digit info if appropriate
	 if(option ==kFillDigitsOnly || option ==kFillAll)
	   {
            for(Int_t k=0; k<numDigits; k++)
	    {
	     myDigit = fInputPointer->GetDigit(k);
             if (fDebug>1) Info("FillUnitArray","getting digits %i %i numdigits",k,numDigits );
	     Int_t towerID = myDigit->GetId();
	     Int_t amplitude = myDigit->GetAmp();     //Gets the integer valued amplitude of the digit
	     Float_t amp = (Float_t)amplitude;        //Need to typecast to Float_t before doing real energy conversion
	     Float_t digitEnergy = amp/10000000.0;    //Factor of 10 million needed to convert!
	     Float_t unitEnergy = fUnit[towerID-1].GetUnitEnergy() + digitEnergy;
	     fUnit[towerID-1].SetUnitEnergy(unitEnergy);

	    }//end digits loop
	   }//end digits condition

	 //Set all unit flags, Eta, Phi
	 for(Int_t i=0; i<fNumUnits; i++)
	   {
             if (fDebug>1) Info("FillUnitArray","Setting all units outside jets");
	     fUnit[i].SetUnitFlag(kOutJet);           //Set all units to be outside a jet initially
	     fUnit[i].SetUnitID(i+1);
	     Float_t eta;
	     Float_t phi;
	     geom->EtaPhiFromIndex(fUnit[i].GetUnitID(), eta, phi);
	     fUnit[i].SetUnitEta(eta);
	     fUnit[i].SetUnitPhi(phi*TMath::Pi()/180.0);
	     //	     if(i>13000) cout<<"!!!!!!!!!!!!!!!!!For unit0, eta="<<eta<<" and phi="<<phi*TMath::Pi()/180.0<<" and ID="<<fUnit[i].GetUnitID()<<endl;
	     //  if(fUnit[i].GetUnitEnergy()>0) cout<<"Unit ID "<<fUnit[i].GetUnitID() <<"with eta="<<eta<<" and phi="<<phi*TMath::Pi()/180.0<<" has energy="<<fUnit[i].GetUnitEnergy()<<endl;
	   }//end loop over all units in array (same as all towers in EMCAL)
   }


 void AliEMCALJetFinderAlgoUA1Revised2::Sort(AliEMCALJetFinderAlgoUA1Unit *unit, Int_t length)
 {
   //Calls the recursive quicksort method to sort unit objects in decending order of Energy
   if (fDebug>1) Info("Sort","Sorting Unit objects");
   QS(unit, 0, length-1);
 }
  

 void AliEMCALJetFinderAlgoUA1Revised2::QS(AliEMCALJetFinderAlgoUA1Unit *unit, Int_t left, Int_t right)
 {
  //Sorts the AliEMCALJetFinderAlgoUA1Unit objects in decending order of Energy
   if (fDebug>111) Info("QS","QuickSorting Unit objects");   

   Int_t    i;
   Int_t    j;
   AliEMCALJetFinderAlgoUA1Unit  unitFirst;
   AliEMCALJetFinderAlgoUA1Unit  unitSecond;

   i = left;
   j = right;
   unitFirst = unit[(left+right)/2];

 do
  {
    while( (unit[i].GetUnitEnergy() > unitFirst.GetUnitEnergy()) && (i < right)) i++;
    while( (unitFirst.GetUnitEnergy() > unit[j].GetUnitEnergy()) && (j > left)) j--;

    if(i <= j)
      {
	unitSecond = unit[i];
	unit[i] = unit[j];
	unit[j] = unitSecond;
	i++;
	j--;
      }//end if
  }while(i <= j);

 if(left < j) QS(unit, left, j);
 if(i < right) QS(unit, i, right);
 }


 void AliEMCALJetFinderAlgoUA1Revised2::FindBG()
   {
     //Finds the background energy for the iteration
     if (fDebug>1) Info("FindBG","Finding Average Background"); 

     fEBGTotal          = 0.0;
     Int_t numCone      = 0;
     Int_t numOutWithE  = 0;

     //Loop over all unit objects in the array and sum the energy of those not in a jet
     //WITH NOMINAL ENERGY!!!
     for(Int_t i=0; i<fNumUnits; i++)
       {
	 if(fUnit[i].GetUnitFlag() != kInJet && fUnit[i].GetUnitEnergy()>0.00001)
	   {
	   fEBGTotal += fUnit[i].GetUnitEnergy();
	   numOutWithE++;
	   }
	 else numCone++;
       }//end for

     fEBGTotalOld = fEBGTotal;
     fEBGAve = fEBGTotal / numOutWithE;           // /(fNumUnits - numCone);
     if (fDebug>5) Info("FindBG","Average BG is %f: ",fEBGAve);      

     for(Int_t count=0; count<fNumUnits;count++)
       {
	 fUnit[count].SetUnitFlag(kOutJet);
       }//end for
   }


 void AliEMCALJetFinderAlgoUA1Revised2::FindJetEtaPhi(Int_t counter)
   {
     //Finds the eta and phi of the jet axis
     if (fDebug>1) Info("FindJetEtaPhi","Finding Jet Eta and Phi");

     fDEta = fUnit[counter].GetUnitEta() - fEtaInit;
     fDPhi = fUnit[counter].GetUnitPhi() - fPhiInit;

     fEnergy = fUnit[counter].GetUnitEnergy() - fEBGAve;
     fJetEtaSum += fEnergy * fDEta;
     fJetPhiSum += fEnergy * fDPhi;
     fJetESum += fEnergy;
     fJetEta = fEtaInit + (fJetEtaSum / fJetESum);
     fJetPhi = fPhiInit + (fJetPhiSum / fJetESum);
   }


 void AliEMCALJetFinderAlgoUA1Revised2::FindJetEnergy()
   {
     //Finds the energy of the jet after the final axis has been found
     if (fDebug>1) Info("FindJetEnergy","Finding Jet Energy");

     for(Int_t i=0; i<fNumUnits; i++)
       {
	 //Loop over all unit objects in the array and find if within cone radius
	 Float_t dEta = fUnit[i].GetUnitEta() - fJetEta;
	 Float_t dPhi = fUnit[i].GetUnitPhi() - fJetPhi;
	 Float_t rad = TMath::Sqrt( (dEta*dEta) + (dPhi*dPhi) );

	 if(fUnit[i].GetUnitFlag()==kOutJet && rad<= fConeRad)
	   {
	     fUnit[i].SetUnitFlag(kInCurrentJet);
	     Float_t energy = fUnit[i].GetUnitEnergy() - fEBGAve;
	     fJetESum += energy;                             
	     fJetEtaSum += energy * dEta;
	     fJetPhiSum += energy * dPhi;
	     fNumInCone++;                     //Increment the number of cells in the jet cone
	   }//end if
       }//end for
   }


 void AliEMCALJetFinderAlgoUA1Revised2::StoreJetInfo()
   {
     //Stores the resulting jet information in appropriate storage structure (TO BE DECIDED!!!!)
     if (fDebug>1) Info("StoreJetInfo","Storing Jet Information");
    
     //Store:
     //fJetESum is the final jet energy (background has been subtracted)
     //fJetEta is the final jet Eta
     //fJetPhi is the final jet Phi
     //fNumInCone is the final number of cells included in the jet cone
     //fEtaInit is the eta of the initiator cell
     //fPhiInit is the phi of the initiator cell
     fJet.SetEnergy(fJetESum);
     fJet.SetEta(fJetEta);
     fJet.SetPhi(fJetPhi);

      cout<<"For iteration "<<fNumIter <<" and Jet number " <<fNumJets <<endl;
      cout<<"The jet energy is: " <<fJetESum <<endl;
      cout<<"The jet eta is ---->" <<fJetEta <<endl;
      cout<<"The jet phi is ---->" <<fJetPhi <<endl;

     Int_t             numberTracks = fInputPointer->GetNTracks();
     TParticle         *myP;
     Int_t             numTracksInCone = 0;

     for(Int_t counter=0; counter<numberTracks; counter++)
       {
	myP = fInputPointer->GetTrack(counter);
	Float_t eta = myP->Eta();
	Float_t phi = myP->Phi(); 
	Float_t deta = fJetEta-eta;
	Float_t dphi = fJetPhi -phi;
	Float_t rad = TMath::Sqrt( (deta*deta) + (dphi*dphi));
	if(rad<=fConeRad) numTracksInCone++;
       }//end for

     Float_t    *pTArray = new Float_t[numTracksInCone];
     Float_t    *etaArray = new Float_t[numTracksInCone];
     Float_t    *phiArray = new Float_t[numTracksInCone];
     Int_t      *pdgArray = new Int_t[numTracksInCone];
     Int_t             index = 0;

     for(Int_t counter2=0; counter2<numberTracks; counter2++)
       {
	 myP = fInputPointer->GetTrack(counter2);
 	 Float_t eta = myP->Eta();
         Float_t phi = myP->Phi(); 
	 Float_t deta = fJetEta-eta;
	 Float_t dphi = fJetPhi -phi;
	 Float_t rad = TMath::Sqrt( (deta*deta) + (dphi*dphi));
	 if(rad<=fConeRad)
	   {
	     pTArray[index] = myP->Pt();
	     etaArray[index] = eta;
	     phiArray[index] = phi;
	     pdgArray[index] = myP->GetPdgCode();
	     index++;
	   }//end if
       }//end for

     fJet.SetTrackList(numTracksInCone,pTArray, etaArray, phiArray, pdgArray);
     fOutputObject.AddJet(&fJet);
     delete[] pTArray;
     delete[] etaArray;
     delete[] phiArray;
     delete[] pdgArray;
   }


 void AliEMCALJetFinderAlgoUA1Revised2::FindJets()
   {
     //Runs the complete UA1 JetFinding algorithm to find jets!
     if (fDebug>1) Info("FindJets","Starting Jet Finding!!!");

     //If the array of JetFinderUnit objects has not been initialised then initialise with default settings
     if(!fArrayInitialised) 
      {
       InitUnitArray();
       FillUnitArray(kFillAll);
      }//end if
     if (fDebug>1) Info("FindJets","Unit array filled");

     //Step 1. Sort the array in descending order of Energy
     Sort(fUnit,fNumUnits);

     //Step 2. Set the number of iterations and Number of jets found to zero to start
     fNumIter = 0;
     fNumJets = 0;

     //Step 3. Begin the iteration loop to find jets
     //Need to iterate the algorithm while number of iterations<2 OR number of iterations<10 AND 
     //the value of the average background has changed more than specified amount
     //Min iterations = 2, Max iterations = 10
     //while(fNumIter<2 || (fNumIter <10 && ( (fEBGTotal-fEBGTotalOld)/fEBGTotal) > fBGMaxMove) )

     while(fNumIter<2 || (fNumIter <10 && ( fEBGTotal-fEBGTotalOld) > fEBGTotal*fBGMaxMove) )
       {
        if (fDebug>1) Info("FindJets","Starting BIG iteration ---> %i",fNumIter);

         //Step 4. Find the value of the average background energy
	 FindBG();
	 fOutputObject.Reset(kResetJets); //Reset output object to store info for new iteration
	 fNumJets=0;

	 //Loop over the array of unit objects and flag those with energy below MinCellEt
         Int_t numbelow = 0;
	 for(Int_t j=0; j<fNumUnits; j++)
	   {
	     if( (fUnit[j].GetUnitEnergy()-fEBGAve) < fEtMin)
        	{       
                  fUnit[j].SetUnitFlag(kBelowMinEt);
                  numbelow++;
        	// if (fDebug>1) Info("FindJets","Below Min Et: %i",j);
                }//end if
	   }//end for

	 //Do quick check if there are no jets upfront
	 if(fUnit[0].GetUnitFlag() == kBelowMinEt)
	   {
            cout <<"There are no jets for this event!" <<endl;
	    break;
	   }//end if

         //Step 5. Begin with the first jet candidate cell (JET SEED LOOP)
         if (fDebug>5) Info("FindJets","Beginning JET SEED LOOP");
	 for(Int_t count=0; count<fNumUnits; count++)
	   {

//CHECK CONDITION HERE _ NOT SURE IF SHOULD MAYBE BE: GetUnitEnergy()-fEBGAve >fESeed?????????????????????????????
	     if(fUnit[count].GetUnitEnergy()>=fESeed && fUnit[count].GetUnitFlag()==kOutJet)
	       {
		 fEnergy = fUnit[count].GetUnitEnergy() - fEBGAve;
		 fJetEta = fUnit[count].GetUnitEta();
		 fJetPhi = fUnit[count].GetUnitPhi();
		 Int_t seedID = fUnit[count].GetUnitID();
                 if (fDebug>5) Info("FindJets","Inside first candidate jet seed loop for time : %i", count);
                 if (fDebug>5) Info("FindJets","Found candidate energy %f ",fEnergy);
                 if (fDebug>5) Info("FindJets","Found candidate eta %f ", fJetEta);
                 if (fDebug>5) Info("FindJets","Found candidate phi %f ", fJetPhi);
		 if (fDebug>5) Info("FindJets","Found candidate ID %i", seedID);

		 fEtaInit = fJetEta;
		 fPhiInit = fJetPhi;
		 fEtaB = fJetEta;
		 fPhiB = fJetPhi;
		 fJetESum = 0.0;
		 fJetEtaSum = 0.0;
		 fJetPhiSum = 0.0;
       
         //Step 6. Find Jet Eta and Phi
		 //Loop over all units in the array to find the ones in the jet cone and determine contrib to Jet eta, phi
		 do
		   {
		     for(Int_t count1=0; count1<fNumUnits; count1++)		   
		       {
			 if(fUnit[count1].GetUnitID() == seedID) continue;   //skip unit if the jetseed to avoid doublecounting
			 if(fUnit[count1].GetUnitFlag() == kOutJet)
			   {
			     fDEta = fUnit[count1].GetUnitEta() - fJetEta;
			     fDPhi = fUnit[count1].GetUnitPhi() - fJetPhi;
			     fRad = TMath::Sqrt( (fDEta*fDEta) + (fDPhi*fDPhi) );
			     if(fRad <= fConeRad)
			       {
				 FindJetEtaPhi(count1); 
			       }//end if
			   }//end if
		       }//end for (Jet Eta, Phi LOOP)
			     
		      //Find the distance cone centre moved from previous cone centre
		      if (fDebug>10) Info("FindJets","Checking if cone move small enough");
		      fDistP = TMath::Sqrt( ((fJetEta-fEtaB)*(fJetEta-fEtaB)) + ((fJetPhi-fPhiB)*(fJetPhi-fPhiB)) );
		      //     if(fDistP <= fMinMove) break;
			     

		      //Find the distance cone centre is from initiator cell
		      if (fDebug>10) Info("FindJets","Checking if cone move too large");
		      fDistI = TMath::Sqrt( ((fJetEtaSum/fJetESum)*(fJetEtaSum/fJetESum)) + ((fJetPhiSum/fJetESum)*
												    (fJetPhiSum/fJetESum)));

		      if(fDistP>fMinMove && fDistI<fMaxMove)
			{
			  fEtaB = fJetEta;
			  fPhiB = fJetPhi;
			}//end if
		 
		   }while(fDistP>fMinMove && fDistI<fMaxMove);
			  
		 fJetEta = fEtaB;
		 fJetPhi = fPhiB;


       //Step 7. Find the Jet Energy
                 if (fDebug>1) Info("FindJets","Looking for Jet energy");
		 fJetESum = 0.0;
		 fJetEtaSum = 0.0;
		 fJetPhiSum = 0.0;
		 fNumInCone = 0;
		 FindJetEnergy();

       //Step 8. Check if the jet is a valid jet
		 //Check if cluster energy is above Min allowed to be a jet
//DID NOT DO THE COSH COMPARISON HERE -> NEED TO CHECK WHICH COMPARISON IS BEST!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
                 if (fDebug>5) Info("FindJets","Checking cluster is valid jet");
		 if(fJetESum < fJetEMin)
		   {
		     for(Int_t count2=0; count2<fNumUnits; count2++)
		       {
			 if(fUnit[count2].GetUnitFlag()==kInCurrentJet || fUnit[count2].GetUnitFlag()==kOutJet)
			   fUnit[count2].SetUnitFlag(kOutJet);
		       }//end for
                   if (fDebug>10) Info("FindJets","NOT a valid jet cell");
		  }else
		    {
		     for(Int_t count2=0; count2<fNumUnits; count2++)
		       {
			 if(fUnit[count2].GetUnitFlag()==kInCurrentJet)
			   {
			     //	     cout<<"Setting unit #"<<count2 <<" to be officially in a jet!"<<endl;
                           fUnit[count2].SetUnitFlag(kInJet);
			   }
		       }//end for			

 //NEED TO CHECK FINAL WEIRD ITERATION OF ETA AND PHI CHANGES!!!!!!!!!
		     //	 fJetPhi += fJetPhiSum/fJetESum;        //CHECK!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
		     //  fJetEta += fJetEtaSum/fJetESum;        //CHECK!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

		     fNumJets++;              //Incrementing number of jets found
		     StoreJetInfo();          //Storing jet info

		 }//end if (check cluster above Min Jet Energy)
	       }//end if (Jet Seed condition)
	   }//end (JET SEED LOOP)

if (fDebug>5) Info("FindJets","End of BIG iteration number %i",fNumIter);
// this->Dump();
	 fNumIter++;
       }//end 10 iteration WHILE LOOP
 }
Commit	Line	Data
f7d5860b	1	//SARAH'S REVISED PERSONAL COPY!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
	2
	3
	4
	5	/**************************************************************************
	6	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	7	* *
	8	* Author: The ALICE Off-line Project. *
	9	* Contributors are mentioned in the code where appropriate. *
	10	* *
	11	* Permission to use, copy, modify and distribute this software and its *
	12	* documentation strictly for non-commercial purposes is hereby granted *
	13	* without fee, provided that the above copyright notice appears in all *
	14	* copies and that both the copyright notice and this permission notice *
	15	* appear in the supporting documentation. The authors make no claims *
	16	* about the suitability of this software for any purpose. It is *
	17	* provided "as is" without express or implied warranty. *
	18	**************************************************************************/
	19
ee6b678f	20	/* $Id$ */
f7d5860b	21
	22	//*--Author: Sarah Blyth (LBL)
	23	//*--Based on UA1 jet algorithm from LUND JETSET called from EMC-erj
	24
	25	#include "TTask.h"
	26	#include "AliEMCALJetFinderInput.h"
	27	#include "AliEMCALJetFinderOutput.h"
	28	#include "AliEMCALJetFinderAlgo.h"
	29	#include "AliEMCALJetFinderAlgoUA1Revised2.h"
	30	#include "AliEMCALJetFinderAlgoUA1Unit.h"
	31	#include "AliEMCALGeometry.h"
	32	#include "AliEMCAL.h"
	33	#include "AliEMCALDigit.h"
	34	#include "TParticle.h"
	35	#include "AliRun.h"
	36	#include "AliEMCALJet.h"
	37	#include "TMath.h"
	38
	39
	40	ClassImp(AliEMCALJetFinderAlgoUA1Revised2)
	41
	42	AliEMCALJetFinderAlgoUA1Revised2::AliEMCALJetFinderAlgoUA1Revised2()
	43	{
	44	//Default constructor
	45	if (fDebug>0) Info("AliEMCALJetFinderAlgoUA1Revised","Beginning Default Constructor");
	46
	47	fNumIter = 0;
	48	fNumUnits = 13824; //Number of towers in EMCAL
	49	fESeed = 5.0; //Default value
	50	fConeRad = 0.5; //Default value
	51	fJetEMin = 10.0; //Default value
	52	fEtMin = 0.28; //Default value
	53	fMinMove = 0.05; //From original UA1 JetFinder
	54	fMaxMove = 0.15; //From original UA1 JetFinder
	55	fBGMaxMove = 0.035; //From original UA1 JetFinder
	56	fPtCut = 0;
	57	fHadCorr = 0;
	58	fEBGTotal = 1.0; //Set to 1 so that no div by zero in first FindJets() loop
	59	fEBGTotalOld = 0.0;
	60	fEBGAve = 0.0;
	61	fEnergy = 0.0;
	62	fJetEta = 0.0;
	63	fJetPhi = 0.0;
	64	fEtaInit = 0.0;
	65	fPhiInit = 0.0;
	66	fEtaB = 0.0;
	67	fPhiB = 0.0;
	68	fJetESum = 0.0;
	69	fJetEtaSum = 0.0;
	70	fJetPhiSum = 0.0;
	71	fDEta = 0.0;
	72	fDPhi = 0.0;
	73	fDistP = 0.0;
	74	fDistI = 0.0;
	75	fTempE = 0.0;
	76	fRad = 2.0; //Set to 2 to start
	77	fNumInCone = 0;
	78	fNumJets = 0;
	79	fArrayInitialised = 0; //Set to FALSE to start
	80	}
	81
	82	AliEMCALJetFinderAlgoUA1Revised2::~AliEMCALJetFinderAlgoUA1Revised2()
	83	{
	84	//Destructor
85	if (fDebug>0) Info("AliEMCALJetFinderAlgoUA1Revised","Beginning Destructor");
86	delete[] fUnit;
87	}
88
89	void AliEMCALJetFinderAlgoUA1Revised2::SetJetFindingParameters
90	(Int_t numUnits, Float_t eSeed, Float_t coneRad, Float_t jetEMin, Float_t etMin,
91	Float_t minMove, Float_t maxMove, Float_t bgMaxMove)
92	{
93	//Sets parameters for the JetFinding algorithm
94	if (fDebug>1) Info("SetJetFindingParameters","Setting parameters for JetFinding");
95
96	SetNumUnits(numUnits);
97	SetJetESeed(eSeed);
98	SetConeRad(coneRad);
99	SetJetEMin(jetEMin);
100	SetEtMin(etMin);
101	SetMinMove(minMove);
102	SetMaxMove(maxMove);
103	SetBGMaxMove(bgMaxMove);
104	}
105
106	void AliEMCALJetFinderAlgoUA1Revised2::SetJetFindingParameters
107	(Int_t numUnits, Float_t eSeed, Float_t coneRad, Float_t jetEMin, Float_t etMin)
108	{
109	//Sets fewer parameters for the JetFinding algorithm
110	if (fDebug>1) Info("SetJetFindingParameters","Setting parameters for JetFinding");
111
112	SetNumUnits(numUnits);
113	SetJetESeed(eSeed);
114	SetConeRad(coneRad);
115	SetJetEMin(jetEMin);
116	SetEtMin(etMin);
117	SetMinMove(fMinMove);
118	SetMaxMove(fMaxMove);
119	SetBGMaxMove(fBGMaxMove);
120	}
121
122	void AliEMCALJetFinderAlgoUA1Revised2::InitUnitArray()
123	{
124	//Initialises unit array
125	if(fArrayInitialised) delete[] fUnit;
126	fUnit = new AliEMCALJetFinderAlgoUA1Unit[fNumUnits];
127	fArrayInitialised = 1;
128	}
129
130	void AliEMCALJetFinderAlgoUA1Revised2::FillUnitArray(AliEMCALJetFinderAlgoUA1FillUnitFlagType_t flag)
131	{
132	if (fDebug>1) Info("FillUnitArray","Beginning FillUnitArray");
133	AliEMCAL* pEMCAL = (AliEMCAL*) gAlice->GetModule("EMCAL");
134
135	// if (pEMCAL){
136	// AliEMCALGeometry* geom = AliEMCALGeometry::GetInstance(pEMCAL->GetTitle(), "");
137	// }else
138	// {
139	AliEMCALGeometry* geom = AliEMCALGeometry::GetInstance("EMCAL_5655_21", "");
140	// }
141
142	AliEMCALJetFinderAlgoUA1FillUnitFlagType_t option = flag;
143	Int_t numTracks, numDigits;
144
145	//Loops over all elements in the AliEMCALJetFinderAlgoUA1Unit array and
146	//fills the objects with relevant values from the Data Input object
147	if (fDebug>1) Info("FillUnitArray","Filling array with Unit objects");
148	if (fDebug>1) Info("FillUnitArray","NTracks %i NDigits %i",fInputPointer->GetNTracks(),fInputPointer->GetNDigits());
149	numTracks = fInputPointer->GetNTracks();
150	numDigits = fInputPointer->GetNDigits();
151	TParticle *myPart;
152	AliEMCALDigit *myDigit;
153
154	//Fill units with Track info if appropriate
155	if(option==kFillTracksOnly \|\| option ==kFillAll)
156	{
157	for(Int_t j=0; j<numTracks; j++)
158	{
159	myPart = fInputPointer->GetTrack(j);
160	Float_t eta = myPart->Eta();
161	Float_t phi = myPart->Phi();
162	Int_t towerID = geom->TowerIndexFromEtaPhi(eta,180.0/TMath::Pi()*phi);
163	Float_t pT = myPart->Pt();
164	Float_t unitEnergy = fUnit[towerID-1].GetUnitEnergy();
165
166	//Do Hadron Correction
167	if(fHadCorr != 0)
168	{
169	Double_t fullP = myPart->P();
170	Double_t hCEnergy = fHadCorr->GetEnergy(fullP, (Double_t)eta);
171	unitEnergy -= hCEnergy*TMath::Sin(myPart->Theta());
172	fUnit[towerID-1].SetUnitEnergy(unitEnergy);
173	} //end Hadron Correction loop
174
175	//Do Pt cut on tracks
176	if(fPtCut != 0 && pT < fPtCut) continue;
177
178	fUnit[towerID-1].SetUnitEnergy(unitEnergy+pT);
179
180	}//end tracks loop
181	}//end Tracks condition
182
183
184	//Fill units with Digit info if appropriate
185	if(option ==kFillDigitsOnly \|\| option ==kFillAll)
186	{
187	for(Int_t k=0; k<numDigits; k++)
188	{
189	myDigit = fInputPointer->GetDigit(k);
190	if (fDebug>1) Info("FillUnitArray","getting digits %i %i numdigits",k,numDigits );
191	Int_t towerID = myDigit->GetId();
192	Int_t amplitude = myDigit->GetAmp(); //Gets the integer valued amplitude of the digit
193	Float_t amp = (Float_t)amplitude; //Need to typecast to Float_t before doing real energy conversion
194	Float_t digitEnergy = amp/10000000.0; //Factor of 10 million needed to convert!
195	Float_t unitEnergy = fUnit[towerID-1].GetUnitEnergy() + digitEnergy;
196	fUnit[towerID-1].SetUnitEnergy(unitEnergy);
197
198	}//end digits loop
199	}//end digits condition
200
201	//Set all unit flags, Eta, Phi
202	for(Int_t i=0; i<fNumUnits; i++)
203	{
204	if (fDebug>1) Info("FillUnitArray","Setting all units outside jets");
205	fUnit[i].SetUnitFlag(kOutJet); //Set all units to be outside a jet initially
206	fUnit[i].SetUnitID(i+1);
207	Float_t eta;
208	Float_t phi;
209	geom->EtaPhiFromIndex(fUnit[i].GetUnitID(), eta, phi);
210	fUnit[i].SetUnitEta(eta);
211	fUnit[i].SetUnitPhi(phi*TMath::Pi()/180.0);
212	// if(i>13000) cout<<"!!!!!!!!!!!!!!!!!For unit0, eta="<<eta<<" and phi="<<phi*TMath::Pi()/180.0<<" and ID="<<fUnit[i].GetUnitID()<<endl;
213	// if(fUnit[i].GetUnitEnergy()>0) cout<<"Unit ID "<<fUnit[i].GetUnitID() <<"with eta="<<eta<<" and phi="<<phi*TMath::Pi()/180.0<<" has energy="<<fUnit[i].GetUnitEnergy()<<endl;
214	}//end loop over all units in array (same as all towers in EMCAL)
215	}
216
217
218	void AliEMCALJetFinderAlgoUA1Revised2::Sort(AliEMCALJetFinderAlgoUA1Unit *unit, Int_t length)
219	{
220	//Calls the recursive quicksort method to sort unit objects in decending order of Energy
221	if (fDebug>1) Info("Sort","Sorting Unit objects");
222	QS(unit, 0, length-1);
223	}
224
225
226	void AliEMCALJetFinderAlgoUA1Revised2::QS(AliEMCALJetFinderAlgoUA1Unit *unit, Int_t left, Int_t right)
227	{
228	//Sorts the AliEMCALJetFinderAlgoUA1Unit objects in decending order of Energy
229	if (fDebug>111) Info("QS","QuickSorting Unit objects");
230
231	Int_t i;
232	Int_t j;
233	AliEMCALJetFinderAlgoUA1Unit unitFirst;
234	AliEMCALJetFinderAlgoUA1Unit unitSecond;
235
236	i = left;
237	j = right;
238	unitFirst = unit[(left+right)/2];
239
240	do
241	{
242	while( (unit[i].GetUnitEnergy() > unitFirst.GetUnitEnergy()) && (i < right)) i++;
243	while( (unitFirst.GetUnitEnergy() > unit[j].GetUnitEnergy()) && (j > left)) j--;
244
245	if(i <= j)
246	{
247	unitSecond = unit[i];
248	unit[i] = unit[j];
249	unit[j] = unitSecond;
250	i++;
251	j--;
252	}//end if
253	}while(i <= j);
254
255	if(left < j) QS(unit, left, j);
256	if(i < right) QS(unit, i, right);
257	}
258
259
260	void AliEMCALJetFinderAlgoUA1Revised2::FindBG()
261	{
262	//Finds the background energy for the iteration
263	if (fDebug>1) Info("FindBG","Finding Average Background");
264
265	fEBGTotal = 0.0;
266	Int_t numCone = 0;
267	Int_t numOutWithE = 0;
268
269	//Loop over all unit objects in the array and sum the energy of those not in a jet
270	//WITH NOMINAL ENERGY!!!
271	for(Int_t i=0; i<fNumUnits; i++)
272	{
273	if(fUnit[i].GetUnitFlag() != kInJet && fUnit[i].GetUnitEnergy()>0.00001)
274	{
275	fEBGTotal += fUnit[i].GetUnitEnergy();
276	numOutWithE++;
277	}
278	else numCone++;
279	}//end for
280
281	fEBGTotalOld = fEBGTotal;
282	fEBGAve = fEBGTotal / numOutWithE; // /(fNumUnits - numCone);
283	if (fDebug>5) Info("FindBG","Average BG is %f: ",fEBGAve);
284
285	for(Int_t count=0; count<fNumUnits;count++)
286	{
287	fUnit[count].SetUnitFlag(kOutJet);
288	}//end for
289	}
290
291
292	void AliEMCALJetFinderAlgoUA1Revised2::FindJetEtaPhi(Int_t counter)
293	{
294	//Finds the eta and phi of the jet axis
295	if (fDebug>1) Info("FindJetEtaPhi","Finding Jet Eta and Phi");
296
297	fDEta = fUnit[counter].GetUnitEta() - fEtaInit;
298	fDPhi = fUnit[counter].GetUnitPhi() - fPhiInit;
299
300	fEnergy = fUnit[counter].GetUnitEnergy() - fEBGAve;
301	fJetEtaSum += fEnergy * fDEta;
302	fJetPhiSum += fEnergy * fDPhi;
303	fJetESum += fEnergy;
304	fJetEta = fEtaInit + (fJetEtaSum / fJetESum);
305	fJetPhi = fPhiInit + (fJetPhiSum / fJetESum);
306	}
307
308
309	void AliEMCALJetFinderAlgoUA1Revised2::FindJetEnergy()
310	{
311	//Finds the energy of the jet after the final axis has been found
312	if (fDebug>1) Info("FindJetEnergy","Finding Jet Energy");
313
314	for(Int_t i=0; i<fNumUnits; i++)
315	{
316	//Loop over all unit objects in the array and find if within cone radius
317	Float_t dEta = fUnit[i].GetUnitEta() - fJetEta;
318	Float_t dPhi = fUnit[i].GetUnitPhi() - fJetPhi;
319	Float_t rad = TMath::Sqrt( (dEtadEta) + (dPhidPhi) );
320
321	if(fUnit[i].GetUnitFlag()==kOutJet && rad<= fConeRad)
322	{
323	fUnit[i].SetUnitFlag(kInCurrentJet);
324	Float_t energy = fUnit[i].GetUnitEnergy() - fEBGAve;
325	fJetESum += energy;
326	fJetEtaSum += energy * dEta;
327	fJetPhiSum += energy * dPhi;
328	fNumInCone++; //Increment the number of cells in the jet cone
329	}//end if
330	}//end for
331	}
332
333
334	void AliEMCALJetFinderAlgoUA1Revised2::StoreJetInfo()
335	{
336	//Stores the resulting jet information in appropriate storage structure (TO BE DECIDED!!!!)
337	if (fDebug>1) Info("StoreJetInfo","Storing Jet Information");
338
339	//Store:
340	//fJetESum is the final jet energy (background has been subtracted)
341	//fJetEta is the final jet Eta
342	//fJetPhi is the final jet Phi
343	//fNumInCone is the final number of cells included in the jet cone
344	//fEtaInit is the eta of the initiator cell
345	//fPhiInit is the phi of the initiator cell
346	fJet.SetEnergy(fJetESum);
347	fJet.SetEta(fJetEta);
348	fJet.SetPhi(fJetPhi);
349
350	cout<<"For iteration "<<fNumIter <<" and Jet number " <<fNumJets <<endl;
351	cout<<"The jet energy is: " <<fJetESum <<endl;
352	cout<<"The jet eta is ---->" <<fJetEta <<endl;
353	cout<<"The jet phi is ---->" <<fJetPhi <<endl;
354
355	Int_t numberTracks = fInputPointer->GetNTracks();
356	TParticle *myP;
357	Int_t numTracksInCone = 0;
358
359	for(Int_t counter=0; counter<numberTracks; counter++)
360	{
361	myP = fInputPointer->GetTrack(counter);
362	Float_t eta = myP->Eta();
363	Float_t phi = myP->Phi();
364	Float_t deta = fJetEta-eta;
365	Float_t dphi = fJetPhi -phi;
366	Float_t rad = TMath::Sqrt( (detadeta) + (dphidphi));
367	if(rad<=fConeRad) numTracksInCone++;
368	}//end for
369
370	Float_t *pTArray = new Float_t[numTracksInCone];
371	Float_t *etaArray = new Float_t[numTracksInCone];
372	Float_t *phiArray = new Float_t[numTracksInCone];
373	Int_t *pdgArray = new Int_t[numTracksInCone];
374	Int_t index = 0;
375
376	for(Int_t counter2=0; counter2<numberTracks; counter2++)
377	{
378	myP = fInputPointer->GetTrack(counter2);
379	Float_t eta = myP->Eta();
380	Float_t phi = myP->Phi();
381	Float_t deta = fJetEta-eta;
382	Float_t dphi = fJetPhi -phi;
383	Float_t rad = TMath::Sqrt( (detadeta) + (dphidphi));
384	if(rad<=fConeRad)
385	{
386	pTArray[index] = myP->Pt();
387	etaArray[index] = eta;
388	phiArray[index] = phi;
389	pdgArray[index] = myP->GetPdgCode();
390	index++;
391	}//end if
392	}//end for
393
394	fJet.SetTrackList(numTracksInCone,pTArray, etaArray, phiArray, pdgArray);
395	fOutputObject.AddJet(&fJet);
396	delete[] pTArray;
397	delete[] etaArray;
398	delete[] phiArray;
399	delete[] pdgArray;
400	}
401
402
403	void AliEMCALJetFinderAlgoUA1Revised2::FindJets()
404	{
405	//Runs the complete UA1 JetFinding algorithm to find jets!
406	if (fDebug>1) Info("FindJets","Starting Jet Finding!!!");
407
408	//If the array of JetFinderUnit objects has not been initialised then initialise with default settings
409	if(!fArrayInitialised)
410	{
411	InitUnitArray();
412	FillUnitArray(kFillAll);
413	}//end if
414	if (fDebug>1) Info("FindJets","Unit array filled");
415
416	//Step 1. Sort the array in descending order of Energy
417	Sort(fUnit,fNumUnits);
418
419	//Step 2. Set the number of iterations and Number of jets found to zero to start
420	fNumIter = 0;
421	fNumJets = 0;
422
423	//Step 3. Begin the iteration loop to find jets
424	//Need to iterate the algorithm while number of iterations<2 OR number of iterations<10 AND
425	//the value of the average background has changed more than specified amount
426	//Min iterations = 2, Max iterations = 10
427	//while(fNumIter<2 \|\| (fNumIter <10 && ( (fEBGTotal-fEBGTotalOld)/fEBGTotal) > fBGMaxMove) )
428
429	while(fNumIter<2 \|\| (fNumIter <10 && ( fEBGTotal-fEBGTotalOld) > fEBGTotal*fBGMaxMove) )
430	{
431	if (fDebug>1) Info("FindJets","Starting BIG iteration ---> %i",fNumIter);
432
433	//Step 4. Find the value of the average background energy
434	FindBG();
435	fOutputObject.Reset(kResetJets); //Reset output object to store info for new iteration
436	fNumJets=0;
437
438	//Loop over the array of unit objects and flag those with energy below MinCellEt
439	Int_t numbelow = 0;
440	for(Int_t j=0; j<fNumUnits; j++)
441	{
442	if( (fUnit[j].GetUnitEnergy()-fEBGAve) < fEtMin)
443	{
444	fUnit[j].SetUnitFlag(kBelowMinEt);
445	numbelow++;
446	// if (fDebug>1) Info("FindJets","Below Min Et: %i",j);
447	}//end if
448	}//end for
449
450	//Do quick check if there are no jets upfront
451	if(fUnit[0].GetUnitFlag() == kBelowMinEt)
452	{
453	cout <<"There are no jets for this event!" <<endl;
454	break;
455	}//end if
456
457	//Step 5. Begin with the first jet candidate cell (JET SEED LOOP)
458	if (fDebug>5) Info("FindJets","Beginning JET SEED LOOP");
459	for(Int_t count=0; count<fNumUnits; count++)
460	{
461
462	//CHECK CONDITION HERE _ NOT SURE IF SHOULD MAYBE BE: GetUnitEnergy()-fEBGAve >fESeed?????????????????????????????
463	if(fUnit[count].GetUnitEnergy()>=fESeed && fUnit[count].GetUnitFlag()==kOutJet)
464	{
465	fEnergy = fUnit[count].GetUnitEnergy() - fEBGAve;
466	fJetEta = fUnit[count].GetUnitEta();
467	fJetPhi = fUnit[count].GetUnitPhi();
468	Int_t seedID = fUnit[count].GetUnitID();
469	if (fDebug>5) Info("FindJets","Inside first candidate jet seed loop for time : %i", count);
470	if (fDebug>5) Info("FindJets","Found candidate energy %f ",fEnergy);
471	if (fDebug>5) Info("FindJets","Found candidate eta %f ", fJetEta);
472	if (fDebug>5) Info("FindJets","Found candidate phi %f ", fJetPhi);
473	if (fDebug>5) Info("FindJets","Found candidate ID %i", seedID);
474
475	fEtaInit = fJetEta;
476	fPhiInit = fJetPhi;
477	fEtaB = fJetEta;
478	fPhiB = fJetPhi;
479	fJetESum = 0.0;
480	fJetEtaSum = 0.0;
481	fJetPhiSum = 0.0;
482
483	//Step 6. Find Jet Eta and Phi
484	//Loop over all units in the array to find the ones in the jet cone and determine contrib to Jet eta, phi
485	do
486	{
487	for(Int_t count1=0; count1<fNumUnits; count1++)
488	{
489	if(fUnit[count1].GetUnitID() == seedID) continue; //skip unit if the jetseed to avoid doublecounting
490	if(fUnit[count1].GetUnitFlag() == kOutJet)
491	{
492	fDEta = fUnit[count1].GetUnitEta() - fJetEta;
493	fDPhi = fUnit[count1].GetUnitPhi() - fJetPhi;
494	fRad = TMath::Sqrt( (fDEtafDEta) + (fDPhifDPhi) );
495	if(fRad <= fConeRad)
496	{
497	FindJetEtaPhi(count1);
498	}//end if
499	}//end if
500	}//end for (Jet Eta, Phi LOOP)
501
502	//Find the distance cone centre moved from previous cone centre
503	if (fDebug>10) Info("FindJets","Checking if cone move small enough");
504	fDistP = TMath::Sqrt( ((fJetEta-fEtaB)(fJetEta-fEtaB)) + ((fJetPhi-fPhiB)(fJetPhi-fPhiB)) );
505	// if(fDistP <= fMinMove) break;
506
507
508	//Find the distance cone centre is from initiator cell
509	if (fDebug>10) Info("FindJets","Checking if cone move too large");
510	fDistI = TMath::Sqrt( ((fJetEtaSum/fJetESum)(fJetEtaSum/fJetESum)) + ((fJetPhiSum/fJetESum)
511	(fJetPhiSum/fJetESum)));
512
513	if(fDistP>fMinMove && fDistI<fMaxMove)
514	{
515	fEtaB = fJetEta;
516	fPhiB = fJetPhi;
517	}//end if
518
519	}while(fDistP>fMinMove && fDistI<fMaxMove);
520
521	fJetEta = fEtaB;
522	fJetPhi = fPhiB;
523
524
525	//Step 7. Find the Jet Energy
526	if (fDebug>1) Info("FindJets","Looking for Jet energy");
527	fJetESum = 0.0;
528	fJetEtaSum = 0.0;
529	fJetPhiSum = 0.0;
530	fNumInCone = 0;
531	FindJetEnergy();
532
533	//Step 8. Check if the jet is a valid jet
534	//Check if cluster energy is above Min allowed to be a jet
535	//DID NOT DO THE COSH COMPARISON HERE -> NEED TO CHECK WHICH COMPARISON IS BEST!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
536	if (fDebug>5) Info("FindJets","Checking cluster is valid jet");
537	if(fJetESum < fJetEMin)
538	{
539	for(Int_t count2=0; count2<fNumUnits; count2++)
540	{
541	if(fUnit[count2].GetUnitFlag()==kInCurrentJet \|\| fUnit[count2].GetUnitFlag()==kOutJet)
542	fUnit[count2].SetUnitFlag(kOutJet);
543	}//end for
544	if (fDebug>10) Info("FindJets","NOT a valid jet cell");
545	}else
546	{
547	for(Int_t count2=0; count2<fNumUnits; count2++)
548	{
549	if(fUnit[count2].GetUnitFlag()==kInCurrentJet)
550	{
551	// cout<<"Setting unit #"<<count2 <<" to be officially in a jet!"<<endl;
552	fUnit[count2].SetUnitFlag(kInJet);
553	}
554	}//end for
555
556	//NEED TO CHECK FINAL WEIRD ITERATION OF ETA AND PHI CHANGES!!!!!!!!!
557	// fJetPhi += fJetPhiSum/fJetESum; //CHECK!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
558	// fJetEta += fJetEtaSum/fJetESum; //CHECK!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
559
560	fNumJets++; //Incrementing number of jets found
561	StoreJetInfo(); //Storing jet info
562
563	}//end if (check cluster above Min Jet Energy)
564	}//end if (Jet Seed condition)
565	}//end (JET SEED LOOP)
566
567	if (fDebug>5) Info("FindJets","End of BIG iteration number %i",fNumIter);
568	// this->Dump();
569	fNumIter++;
570	}//end 10 iteration WHILE LOOP
571	}
572
573
574
575
576
577
578
579
580
581
582