Adding extra check for GPU_FORCE_64BIT_PTR env var
[u/mrichter/AliRoot.git] / JETAN / AliFastJetInput.cxx
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 /* $Id$ */
17
18 //---------------------------------------------------------------------
19 // Class for input particles
20 // manages the search for jets
21 // Authors: Elena Bruna elena.bruna@yale.edu
22 //
23 // ** 2011 magali.estienne@subatech.in2p3.fr &  alexandre.shabetai@cern.ch
24 // Modified accordingly to reader/finder splitting and new handling of neutral information
25 //---------------------------------------------------------------------
26
27 #include <Riostream.h> 
28
29 #include "AliJetHeader.h"
30 #include "AliFastJetHeaderV1.h"
31 #include "AliFastJetInput.h"
32 #include "AliJetCalTrk.h"
33
34 #include "fastjet/PseudoJet.hh"
35
36 #include<vector> 
37
38 using namespace std;
39
40 ClassImp(AliFastJetInput)
41
42 ////////////////////////////////////////////////////////////////////////
43
44 AliFastJetInput::AliFastJetInput():
45   fHeader(0x0),
46   fCalTrkEvent(0x0),
47   fInputParticles(0),
48   fInputParticlesCh(0)
49 {
50   // Default constructor
51 }
52
53 //______________________________________________________________________
54 AliFastJetInput::AliFastJetInput(const AliFastJetInput &input):
55   TObject(input),
56   fHeader(input.fHeader),
57   fCalTrkEvent(input.fCalTrkEvent),
58   fInputParticles(input.fInputParticles),
59   fInputParticlesCh(input.fInputParticlesCh)
60 {
61   // copy constructor
62 }
63
64 //______________________________________________________________________
65 AliFastJetInput& AliFastJetInput::operator=(const AliFastJetInput& source)
66 {
67   // Assignment operator. 
68   if(this!=&source){
69    TObject::operator=(source);
70    fHeader = source.fHeader;
71    fCalTrkEvent = source.fCalTrkEvent;
72    fInputParticles = source.fInputParticles;
73    fInputParticlesCh = source.fInputParticlesCh;
74   }
75
76   return *this;
77
78 }
79
80 //___________________________________________________________
81 void AliFastJetInput::FillInput()
82 {
83   // fills input particles for FASTJET based analysis
84   
85   AliFastJetHeaderV1 *header = (AliFastJetHeaderV1*)fHeader;
86   Int_t debug  = header->GetDebug();     // debug option
87
88   if(debug>0) cout<<"-------- AliFastJetInput::FillInput()  ----------------"<<endl;
89
90   fInputParticles.clear();
91   fInputParticlesCh.clear();
92
93   // RUN ALGORITHM  
94   // read input particles -----------------------------
95   vector<fastjet::PseudoJet> inputParticles;
96
97   if(fCalTrkEvent == 0) { cout << "Could not get the CalTrk Event" << endl; return; }
98   Int_t nIn =  fCalTrkEvent->GetNCalTrkTracks() ;
99   if(nIn == 0) { if (debug>0) cout << "entries = 0 ; Event empty !!!" << endl ; return; }
100
101   // Information extracted from fCalTrkEvent
102   // load input vectors and calculate total energy in array
103   Float_t px = -999., py = -999., pz = -999., en = -999.; 
104  
105   // Fill charged tracks
106   for(Int_t i = 0; i < fCalTrkEvent->GetNCalTrkTracks(); i++)
107     { // loop for all input particles
108       if (fCalTrkEvent->GetCalTrkTrack(i)->GetCutFlag() != 1) continue;
109       px =  fCalTrkEvent->GetCalTrkTrack(i)->GetPx();
110       py =  fCalTrkEvent->GetCalTrkTrack(i)->GetPy();
111       pz =  fCalTrkEvent->GetCalTrkTrack(i)->GetPz();
112       en =  fCalTrkEvent->GetCalTrkTrack(i)->GetP();
113
114       fastjet::PseudoJet inputPart(px,py,pz,en);  // create PseudoJet object
115       inputPart.set_user_index(i);      //label the particle into Fastjet algortihm
116       fInputParticles.push_back(inputPart);       // back of the inputParticles vector
117  
118       // only for charged particles (TPC+ITS)
119       fastjet::PseudoJet inputPartCh(px,py,pz,en); // create PseudoJet object
120       inputPartCh.set_user_index(i);               //label the particle into Fastjet algortihm
121       fInputParticlesCh.push_back(inputPartCh);    // back of the inputParticles vector
122     } // End loop on CalTrk
123
124 }
125
126 //_____________________________________________________________________
127 Double_t AliFastJetInput::Thermalspectrum(const Double_t *x, const Double_t *par)
128 {
129   // compute an exponential function
130   return x[0]*TMath::Exp(-x[0]/par[0]);
131
132 }