[u/mrichter/AliRoot.git] / JETAN / AliFastJetInput.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.                  *
 **************************************************************************/

#include <Riostream.h> 
#include <TChain.h>
#include <TFile.h>
#include <TF1.h>
#include <TRandom.h>
#include <TList.h>
#include <TLorentzVector.h>
#include <TArrayF.h>
#include <TClonesArray.h>

#include "AliJetHeader.h"
#include "AliJetReader.h"
#include "AliJetReaderHeader.h"
#include "AliJetHistos.h"

#include "AliFastJetFinder.h"
#include "AliFastJetHeaderV1.h"
#include "AliJetReaderHeader.h"
#include "AliJetReader.h"
#include "AliJetESDReader.h"
#include "AliJetUnitArray.h"
#include "AliFastJetInput.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(AliFastJetInput)

////////////////////////////////////////////////////////////////////////

AliFastJetInput::AliFastJetInput():
    fReader(0),
    fHeader(0),
    fInputParticles(0),
    fInputParticlesCh(0)
{
  // Default constructor
}
AliFastJetInput::AliFastJetInput(const AliFastJetInput &input):
    TObject(),
    fReader(input.fReader),
    fHeader(input.fHeader),
    fInputParticles(input.fInputParticles),
    fInputParticlesCh(input.fInputParticlesCh)
{
  // copy constructor
}
//______________________________________________________________________
AliFastJetInput& AliFastJetInput::operator=(const AliFastJetInput& source){
    // Assignment operator. 
    this->~AliFastJetInput();
    new(this) AliFastJetInput(source);
    return *this;

}
//___________________________________________________________
void AliFastJetInput::FillInput(){
  //cout<<"-------- AliFastJetInput::FillInput()  ----------------"<<endl;

  AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
  fInputParticles.clear();
  fInputParticlesCh.clear();

  Int_t debug  = header->GetDebug();     // debug option
  Int_t fOpt   = fReader->GetReaderHeader()->GetDetector();

  // RUN ALGORITHM  
  // read input particles -----------------------------
  vector<fastjet::PseudoJet> inputParticles;
  //  cout<<"=============== AliFastJetInput::FillInput()  =========== fOpt="<<fOpt<<endl;
  //cout<<"pointers --> fReader="<<fReader<<" header="<<header<<" fHeader="<<fHeader<<endl;
  //cout<<"Rparam="<<Rparam<<"  ghost_etamax="<<ghost_etamax<<"  ghost_area="<<ghost_area<<endl;
  //cout<<fReader->ClassName()<<endl;

  if(fOpt==0)
    {
      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; }
      Float_t px,py,pz,en;
      // load input vectors
      for(Int_t i = 0; i < nIn; i++){ // loop for all input particles
	if(fReader->GetCutFlag(i)!=1)continue; // pt cut
	TLorentzVector *lv = (TLorentzVector*) lvArray->At(i);
	px = lv->Px();
	py = lv->Py();
	pz = lv->Pz();
	en = lv->Energy();
	
	//	cout<<"in FillInput...... "<<i<<" "<<px<<" "<<py<<"  "<<pz<<" "<<en<<endl;
	fastjet::PseudoJet inputPart(px,py,pz,en); // create PseudoJet object
	inputPart.set_user_index(i); //label the particle into Fastjet algortihm
	fInputParticles.push_back(inputPart);  // back of the inputParticles vector  
      } // end loop 
    }
  else {
    TClonesArray* fUnit = fReader->GetUnitArray();
    if(fUnit == 0) { cout << "Could not get the momentum array" << endl; return; }
    Int_t         nIn = fUnit->GetEntries();
    if(nIn == 0) { if (debug) cout << "entries = 0 ; Event empty !!!" << endl ; return; }
   
    // Information extracted from fUnitArray
    // load input vectors and calculate total energy in array
    Float_t pt,eta,phi,theta,px,py,pz,en;
    Int_t ipart = 0;
    Int_t countUnit=0,countUnitNonZero=0;
    //cout<<" nIn = "<<nIn<<endl;

    for(Int_t i=0; i<nIn; i++) 
      {
	AliJetUnitArray *uArray = (AliJetUnitArray*)fUnit->At(i);
	if(uArray->GetUnitEnergy()>0.){
	  countUnit++;
	  // It is not necessary anymore to cut on particle pt
	  pt    = uArray->GetUnitEnergy();
	  eta   = uArray->GetUnitEta();
	  phi   = uArray->GetUnitPhi();
	  theta = EtaToTheta(eta);
	  en    = (TMath::Abs(TMath::Sin(theta)) == 0) ? pt : pt/TMath::Abs(TMath::Sin(theta));
	  px    = TMath::Cos(phi)*pt;
	  py    = TMath::Sin(phi)*pt;
	  pz    = en*TMath::TanH(eta);
	  if(debug) cout << "pt: " << pt << ", eta: " << eta << ", phi: " << phi << ", en: " << en << ", px: " << px << ", py: " << py << ", pz: " << pz << endl;
	  //cout << i<<" "<<"pt: " << pt << ", eta: " << eta << ", phi: " << phi << ", en: " << en << ", px: " << px << ", py: " << py << ", pz: " << pz << endl;
	  //cout<<"in FillInput...... "<<i<<" "<<px<<" "<<py<<"  "<<pz<<" "<<en<<endl;

	  fastjet::PseudoJet inputPart(px,py,pz,en); // create PseudoJet object
	  inputPart.set_user_index(ipart); //label the particle into Fastjet algortihm
	  fInputParticles.push_back(inputPart);  // back of the inputParticles vector 
	  ipart++;

	
	  //only for charged particles (TPC+ITS)
	  TRefArray* ref = uArray->GetUnitTrackRef();
	  Int_t nRef = ref->GetEntries();
	  for(Int_t j=0; j<nRef;j++){
	    Float_t pxj=0.;  Float_t pyj=0.;  Float_t pzj=0.;Float_t enj=0.;
	    pxj = ((AliVTrack*)ref->At(j))->Px();
	    pyj = ((AliVTrack*)ref->At(j))->Py();
	    pzj = ((AliVTrack*)ref->At(j))->Pz();
	    enj=TMath::Sqrt(pxj*pxj+pyj*pyj+pzj*pzj);
	    fastjet::PseudoJet inputPartCh(pxj,pyj,pzj,enj); // create PseudoJet object
	    inputPartCh.set_user_index(((AliVTrack*)ref->At(j))->GetID()); //label the particle into Fastjet algortihm
	    fInputParticlesCh.push_back(inputPartCh);  // back of the inputParticles vector 

	  }
	}
      } // End loop on UnitArray 
    if (debug) cout<<"countUnit(En>0) = "<<countUnit<<"  countUnit with Non ZeroSize = "<<countUnitNonZero<<endl;
  }

}

//_____________________________________________________________________
Float_t  AliFastJetInput::EtaToTheta(Float_t arg)
{
  //  return (180./TMath::Pi())*2.*atan(exp(-arg));
  return 2.*atan(exp(-arg));


}
Double_t AliFastJetInput::Thermalspectrum(const Double_t *x, const Double_t *par){

  return x[0]*TMath::Exp(-x[0]/par[0]);

}
Commit	Line	Data
50c7d9f7	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	#include <Riostream.h>
	17	#include <TChain.h>
	18	#include <TFile.h>
	19	#include <TF1.h>
	20	#include <TRandom.h>
	21	#include <TList.h>
	22	#include <TLorentzVector.h>
	23	#include <TArrayF.h>
	24	#include <TClonesArray.h>
	25
	26	#include "AliJetHeader.h"
	27	#include "AliJetReader.h"
	28	#include "AliJetReaderHeader.h"
	29	#include "AliJetHistos.h"
	30
	31	#include "AliFastJetFinder.h"
	32	#include "AliFastJetHeaderV1.h"
	33	#include "AliJetReaderHeader.h"
	34	#include "AliJetReader.h"
	35	#include "AliJetESDReader.h"
	36	#include "AliJetUnitArray.h"
	37	#include "AliFastJetInput.h"
	38
	39	#include "fastjet/PseudoJet.hh"
	40	#include "fastjet/ClusterSequenceArea.hh"
	41	#include "fastjet/AreaDefinition.hh"
	42	#include "fastjet/JetDefinition.hh"
	43	// get info on how fastjet was configured
	44	#include "fastjet/config.h"
	45
	46	#ifdef ENABLE_PLUGIN_SISCONE
	47	#include "fastjet/SISConePlugin.hh"
	48	#endif
	49
	50	#include<sstream> // needed for internal io
	51	#include<vector>
	52	#include <cmath>
	53
	54	using namespace std;
	55
	56
	57	ClassImp(AliFastJetInput)
	58
	59	////////////////////////////////////////////////////////////////////////
	60
	61	AliFastJetInput::AliFastJetInput():
	62	fReader(0),
	63	fHeader(0),
	64	fInputParticles(0),
65	fInputParticlesCh(0)
66	{
67	// Default constructor
68	}
69	AliFastJetInput::AliFastJetInput(const AliFastJetInput &input):
70	TObject(),
71	fReader(input.fReader),
72	fHeader(input.fHeader),
73	fInputParticles(input.fInputParticles),
74	fInputParticlesCh(input.fInputParticlesCh)
75	{
76	// copy constructor
77	}
78	//______________________________________________________________________
79	AliFastJetInput& AliFastJetInput::operator=(const AliFastJetInput& source){
80	// Assignment operator.
81	this->~AliFastJetInput();
82	new(this) AliFastJetInput(source);
83	return *this;
84
85	}
86	//___________________________________________________________
87	void AliFastJetInput::FillInput(){
88	//cout<<"-------- AliFastJetInput::FillInput() ----------------"<<endl;
89
90	AliFastJetHeaderV1 header = (AliFastJetHeaderV1)fHeader;
91	fInputParticles.clear();
92	fInputParticlesCh.clear();
93
cc6a2227	94	Int_t debug = header->GetDebug(); // debug option
cc6a2227	95	Int_t fOpt = fReader->GetReaderHeader()->GetDetector();
50c7d9f7	96
50c7d9f7	97	// RUN ALGORITHM
	98	// read input particles -----------------------------
	99	vector<fastjet::PseudoJet> inputParticles;
	100	// cout<<"=============== AliFastJetInput::FillInput() =========== fOpt="<<fOpt<<endl;
	101	//cout<<"pointers --> fReader="<<fReader<<" header="<<header<<" fHeader="<<fHeader<<endl;
	102	//cout<<"Rparam="<<Rparam<<" ghost_etamax="<<ghost_etamax<<" ghost_area="<<ghost_area<<endl;
	103	//cout<<fReader->ClassName()<<endl;
	104
	105	if(fOpt==0)
	106	{
	107	TClonesArray *lvArray = fReader->GetMomentumArray();
	108	if(lvArray == 0) { cout << "Could not get the momentum array" << endl; return; }
	109	Int_t nIn = lvArray->GetEntries();
	110	if(nIn == 0) { if (debug) cout << "entries = 0 ; Event empty !!!" << endl ; return; }
	111	Float_t px,py,pz,en;
	112	// load input vectors
	113	for(Int_t i = 0; i < nIn; i++){ // loop for all input particles
9e7ec179	114	if(fReader->GetCutFlag(i)!=1)continue; // pt cut
50c7d9f7	115	TLorentzVector lv = (TLorentzVector) lvArray->At(i);
	116	px = lv->Px();
	117	py = lv->Py();
	118	pz = lv->Pz();
	119	en = lv->Energy();
9e7ec179	120
50c7d9f7	121	// cout<<"in FillInput...... "<<i<<" "<<px<<" "<<py<<" "<<pz<<" "<<en<<endl;
	122	fastjet::PseudoJet inputPart(px,py,pz,en); // create PseudoJet object
	123	inputPart.set_user_index(i); //label the particle into Fastjet algortihm
	124	fInputParticles.push_back(inputPart); // back of the inputParticles vector
	125	} // end loop
	126	}
	127	else {
	128	TClonesArray* fUnit = fReader->GetUnitArray();
	129	if(fUnit == 0) { cout << "Could not get the momentum array" << endl; return; }
	130	Int_t nIn = fUnit->GetEntries();
	131	if(nIn == 0) { if (debug) cout << "entries = 0 ; Event empty !!!" << endl ; return; }
	132
	133	// Information extracted from fUnitArray
	134	// load input vectors and calculate total energy in array
	135	Float_t pt,eta,phi,theta,px,py,pz,en;
	136	Int_t ipart = 0;
	137	Int_t countUnit=0,countUnitNonZero=0;
	138	//cout<<" nIn = "<<nIn<<endl;
	139
	140	for(Int_t i=0; i<nIn; i++)
	141	{
	142	AliJetUnitArray uArray = (AliJetUnitArray)fUnit->At(i);
	143	if(uArray->GetUnitEnergy()>0.){
	144	countUnit++;
	145	// It is not necessary anymore to cut on particle pt
	146	pt = uArray->GetUnitEnergy();
	147	eta = uArray->GetUnitEta();
	148	phi = uArray->GetUnitPhi();
	149	theta = EtaToTheta(eta);
	150	en = (TMath::Abs(TMath::Sin(theta)) == 0) ? pt : pt/TMath::Abs(TMath::Sin(theta));
	151	px = TMath::Cos(phi)*pt;
	152	py = TMath::Sin(phi)*pt;
	153	pz = en*TMath::TanH(eta);
	154	if(debug) cout << "pt: " << pt << ", eta: " << eta << ", phi: " << phi << ", en: " << en << ", px: " << px << ", py: " << py << ", pz: " << pz << endl;
	155	//cout << i<<" "<<"pt: " << pt << ", eta: " << eta << ", phi: " << phi << ", en: " << en << ", px: " << px << ", py: " << py << ", pz: " << pz << endl;
	156	//cout<<"in FillInput...... "<<i<<" "<<px<<" "<<py<<" "<<pz<<" "<<en<<endl;
	157
	158	fastjet::PseudoJet inputPart(px,py,pz,en); // create PseudoJet object
	159	inputPart.set_user_index(ipart); //label the particle into Fastjet algortihm
	160	fInputParticles.push_back(inputPart); // back of the inputParticles vector
	161	ipart++;
	162
	163
	164
	165	//only for charged particles (TPC+ITS)
	166	TRefArray* ref = uArray->GetUnitTrackRef();
	167	Int_t nRef = ref->GetEntries();
	168	for(Int_t j=0; j<nRef;j++){
	169	Float_t pxj=0.; Float_t pyj=0.; Float_t pzj=0.;Float_t enj=0.;
	170	pxj = ((AliVTrack*)ref->At(j))->Px();
	171	pyj = ((AliVTrack*)ref->At(j))->Py();
	172	pzj = ((AliVTrack*)ref->At(j))->Pz();
	173	enj=TMath::Sqrt(pxjpxj+pyjpyj+pzj*pzj);
	174	fastjet::PseudoJet inputPartCh(pxj,pyj,pzj,enj); // create PseudoJet object
	175	inputPartCh.set_user_index(((AliVTrack*)ref->At(j))->GetID()); //label the particle into Fastjet algortihm
	176	fInputParticlesCh.push_back(inputPartCh); // back of the inputParticles vector
	177
	178	}
	179	}
	180	} // End loop on UnitArray
c01e7c1b	181	if (debug) cout<<"countUnit(En>0) = "<<countUnit<<" countUnit with Non ZeroSize = "<<countUnitNonZero<<endl;
50c7d9f7	182	}
	183
	184	}
	185
	186	//_____________________________________________________________________
	187	Float_t AliFastJetInput::EtaToTheta(Float_t arg)
	188	{
	189	// return (180./TMath::Pi())2.atan(exp(-arg));
	190	return 2.*atan(exp(-arg));
	191
	192
	193	}
1240edf5	194	Double_t AliFastJetInput::Thermalspectrum(const Double_t x, const Double_t par){
50c7d9f7	195
	196	return x[0]*TMath::Exp(-x[0]/par[0]);
	197
	198	}