AOD Tag creator task.

[u/mrichter/AliRoot.git] / JETAN / AliFastJetFinder.cxx

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 

//---------------------------------------------------------------------
// FastJet v2.3.4 finder algorithm interface
//
// Author: Rafael.Diaz.Valdes@cern.ch
//  
//---------------------------------------------------------------------

#include <Riostream.h>
#include <TLorentzVector.h>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TArrayF.h>
#include <TRandom.h>
#include <TClonesArray.h>

#include "AliFastJetFinder.h"
#include "AliFastJetHeader.h"
#include "AliJetReaderHeader.h"
#include "AliJetReader.h"
#include "AliJet.h"


#include "fastjet/PseudoJet.hh"
#include "fastjet/ClusterSequenceArea.hh"
#include "fastjet/AreaDefinition.hh"
#include "fastjet/JetDefinition.hh"
// get info on how fastjet was configured
#include "fastjet/config.h"

#ifdef ENABLE_PLUGIN_SISCONE
#include "fastjet/SISConePlugin.hh"
#endif

#include<sstream>  // needed for internal io
#include<vector> 
#include <cmath> 

using namespace std;


ClassImp(AliFastJetFinder);


//____________________________________________________________________________

AliFastJetFinder::AliFastJetFinder():
    AliJetFinder()
{
  // Constructor
}

//____________________________________________________________________________

AliFastJetFinder::~AliFastJetFinder()
{
  // destructor
}

//______________________________________________________________________________
void AliFastJetFinder::FindJets()
{
  
  Bool_t debug = kFALSE;
  
  //pick up fastjet header
  AliFastJetHeader *header = (AliFastJetHeader*)fHeader;

  // check if we are reading AOD jets
  TRefArray *refs = 0;
  Bool_t fromAod = !strcmp(fReader->ClassName(),"AliJetAODReader");
  if (fromAod) { refs = fReader->GetReferences(); }
  
  // RUN ALGORITHM  
  // read input particles -----------------------------
  vector<fastjet::PseudoJet> input_particles;
  TClonesArray *lvArray = fReader->GetMomentumArray();
  if(lvArray == 0) { cout << "Could not get the momentum array" << endl; return; }
  Int_t nIn =  lvArray->GetEntries();
  if(nIn == 0) { if (debug) cout << "entries = 0 ; Event empty !!!" << endl ; return; }
  Int_t nJets = 0; // n jets in this event
  fJets->SetNinput(nIn) ; // number of input objects
  Float_t px,py,pz,en;
  // load input vectors
  for(Int_t i = 0; i < nIn; i++){ // loop for all input particles
      TLorentzVector *lv = (TLorentzVector*) lvArray->At(i);
      px = lv->Px();
      py = lv->Py();
      pz = lv->Pz();
      en = lv->Energy();
      
      fastjet::PseudoJet input_part(px,py,pz,en); // create PseudoJet object
      input_part.set_user_index(i); //label the particle into Fastjet algortihm
      input_particles.push_back(input_part);  // back of the input_particles vector  
  } // end loop 
  
  
  // create an object that represents your choice of jet algorithm, and 
  // the associated parameters
  double Rparam = header->GetRparam();
  fastjet::Strategy strategy = header->GetStrategy();
  fastjet::RecombinationScheme recomb_scheme = header->GetRecombScheme();
  fastjet::JetAlgorithm algorithm = header->GetAlgorithm(); 
  fastjet::JetDefinition jet_def(algorithm, Rparam, recomb_scheme, strategy);
  
 
  // create an object that specifies how we to define the area
  fastjet::AreaDefinition area_def;
  double ghost_etamax = header->GetGhostEtaMax(); 
  double ghost_area   = header->GetGhostArea(); 
  int    active_area_repeats = header->GetActiveAreaRepeats(); 
  
  // now create the object that holds info about ghosts
  fastjet::GhostedAreaSpec ghost_spec(ghost_etamax, active_area_repeats, ghost_area);
  // and from that get an area definition
  fastjet::AreaType area_type = header->GetAreaType();
  area_def = fastjet::AreaDefinition(area_type,ghost_spec);
  
  // run the jet clustering with the above jet definition
  fastjet::ClusterSequenceArea clust_seq(input_particles, jet_def, area_def);
  
  
  // save a comment in the header
 
  TString comment = "Running FastJet algorithm with the following setup. ";
  comment+= "Jet definition: ";
  comment+= TString(jet_def.description());
  comment+= ". Area definition: ";
  comment+= TString(area_def.description());
  comment+= ". Strategy adopted by FastJet: ";
  comment+= TString(clust_seq.strategy_string());
  header->SetComment(comment);
  if(debug){
    cout << "--------------------------------------------------------" << endl;
    cout << comment << endl;
    cout << "--------------------------------------------------------" << endl;
  }
  //header->PrintParameters();
  
  
  // extract the inclusive jets with pt > ptmin, sorted by pt
  double ptmin = header->GetPtMin(); 
  vector<fastjet::PseudoJet> inclusive_jets = clust_seq.inclusive_jets(ptmin);
  
  //cout << "Number of unclustered particles: " << clust_seq.unclustered_particles().size() << endl;
 
 
  //subtract background // ===========================================
  // set the rapididty , phi range within which to study the background 
  double rap_max = header->GetRapMax(); 
  double rap_min = header->GetRapMin();
  double phi_max = header->GetPhiMax();
  double phi_min = header->GetPhiMin();
  fastjet::RangeDefinition range(rap_min, rap_max, phi_min, phi_max);
  // subtract background
  vector<fastjet::PseudoJet> sub_jets =  clust_seq.subtracted_jets(range,ptmin);  
  
  // print out
  //cout << "Printing inclusive sub jets with pt > "<< ptmin<<" GeV\n";
  //cout << "---------------------------------------\n";
  //cout << endl;
  //printf(" ijet   rap      phi        Pt         area  +-   err\n");
   
  // sort jets into increasing pt
  vector<fastjet::PseudoJet> jets = sorted_by_pt(sub_jets);  
  for (size_t j = 0; j < jets.size(); j++) { // loop for jets

    double area     = clust_seq.area(jets[j]);
    double area_error = clust_seq.area_error(jets[j]);

    //printf("%5u %9.5f %8.5f %10.3f %8.3f +- %6.3f\n",j,jets[j].rap(),jets[j].phi(),jets[j].perp(), area, area_error);
	
	// go to write AOD  info
    AliAODJet aodjet (jets[j].px(), jets[j].py(), jets[j].pz(), jets[j].E());
   //cout << "Printing jet " << endl;
    if(debug) aodjet.Print("");
   //cout << "Adding jet ... " ;
    AddJet(aodjet);
   //cout << "added \n" << endl;
   
  } // end loop for jets

    
}

//____________________________________________________________________________
void AliFastJetFinder::RunTest(const char* datafile)

{

   // This simple test run the kt algorithm for an ascii file testdata.dat
   // read input particles -----------------------------
  vector<fastjet::PseudoJet> input_particles;
  Float_t px,py,pz,en;
  ifstream in;
  Int_t nlines = 0;
  // we assume a file basic.dat in the current directory
  // this file has 3 columns of float data
  in.open(datafile);
  while (1) {
      in >> px >> py >> pz >> en;
      if (!in.good()) break;
      //printf("px=%8f, py=%8f, pz=%8fn",px,py,pz);
      nlines++;
      input_particles.push_back(fastjet::PseudoJet(px,py,pz,en)); 
   }
   //printf(" found %d pointsn",nlines);
   in.close();
   //////////////////////////////////////////////////
 
  // create an object that represents your choice of jet algorithm, and 
  // the associated parameters
  double Rparam = 1.0;
  fastjet::Strategy strategy = fastjet::Best;
  fastjet::RecombinationScheme recomb_scheme = fastjet::BIpt_scheme;
  fastjet::JetDefinition jet_def(fastjet::kt_algorithm, Rparam, recomb_scheme, strategy);
  
  
 
  // create an object that specifies how we to define the area
  fastjet::AreaDefinition area_def;
  double ghost_etamax = 7.0;
  double ghost_area    = 0.05;
  int    active_area_repeats = 1;
  

  // now create the object that holds info about ghosts
  fastjet::GhostedAreaSpec ghost_spec(ghost_etamax, active_area_repeats, ghost_area);
  // and from that get an area definition
  area_def = fastjet::AreaDefinition(fastjet::active_area,ghost_spec);
  

  // run the jet clustering with the above jet definition
  fastjet::ClusterSequenceArea clust_seq(input_particles, jet_def, area_def);
  
  
  // tell the user what was done
  cout << "--------------------------------------------------------" << endl;
  cout << "Jet definition was: " << jet_def.description() << endl;
  cout << "Area definition was: " << area_def.description() << endl;
  cout << "Strategy adopted by FastJet was "<< clust_seq.strategy_string()<<endl<<endl;
  cout << "--------------------------------------------------------" << endl;
 
  
  // extract the inclusive jets with pt > 5 GeV, sorted by pt
  double ptmin = 5.0;
  vector<fastjet::PseudoJet> inclusive_jets = clust_seq.inclusive_jets(ptmin);
  
  cout << "Number of unclustered particles: " << clust_seq.unclustered_particles().size() << endl;
 
 
  //subtract background // ===========================================
  // set the rapididty range within which to study the background 
  double rap_max = ghost_etamax - Rparam;
  fastjet::RangeDefinition range(rap_max);
  // subtract background
  vector<fastjet::PseudoJet> sub_jets =  clust_seq.subtracted_jets(range,ptmin);  
  
  // print them out //================================================
  cout << "Printing inclusive jets  after background subtraction \n";
  cout << "------------------------------------------------------\n";
  // sort jets into increasing pt
  vector<fastjet::PseudoJet> jets = sorted_by_pt(sub_jets);  

  printf(" ijet   rap      phi        Pt         area  +-   err\n");
  for (size_t j = 0; j < jets.size(); j++) {

    double area     = clust_seq.area(jets[j]);
    double area_error = clust_seq.area_error(jets[j]);

    printf("%5u %9.5f %8.5f %10.3f %8.3f +- %6.3f\n",j,jets[j].rap(),
	   jets[j].phi(),jets[j].perp(), area, area_error);
  }
  cout << endl;
  // ================================================================

  
 
}

//____________________________________________________________________________

void AliFastJetFinder::WriteJHeaderToFile()
{
  fHeader->Write();
}

//____________________________________________________________________________