#include "AliJetUnitArray.h"
#include "AliFastJetInput.h"
#include "AliJetBkg.h"
-#include "AliAODPWG4JetEventBackground.h"
+#include "AliAODJetEventBackground.h"
+#include "AliAODTrack.h"
#include "fastjet/PseudoJet.hh"
#include "fastjet/ClusterSequenceArea.hh"
AliFastJetFinder::AliFastJetFinder():
AliJetFinder(),
- fInputFJ(0),
- fJetBkg(0)
+ fInputFJ(new AliFastJetInput()),
+ fJetBkg(new AliJetBkg())
{
// Constructor
}
AliFastJetFinder::~AliFastJetFinder()
{
// destructor
+ delete fInputFJ; fInputFJ = 0;
+ delete fJetBkg; fJetBkg = 0;
}
//______________________________________________________________________________
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;
// read input particles -----------------------------
vector<fastjet::PseudoJet> inputParticles=fInputFJ->GetInputParticles();
-
+ if(inputParticles.size()==0){
+ if(debug)Printf("%s:%d No input particles found, skipping event",(char*)__FILE__,__LINE__);
+ return;
+ }
// create an object that represents your choice of jet algorithm, and
// the associated parameters
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
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());
}
}
- 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);
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
}
}
+
+Bool_t AliFastJetFinder::ProcessEvent()
+{
+ //
+ // Process one event
+ // from meomntum array
+
+ Bool_t ok = fReader->FillMomentumArray();
+
+ if (!ok) return kFALSE;
+ fInputFJ->SetHeader(fHeader);
+ fInputFJ->SetReader(fReader);
+ fInputFJ->FillInput();
+ // Jets
+ FindJets();
+
+ 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);
+
+ fAODEvBkg->SetBackground(0,bkg1);
+ fAODEvBkg->SetBackground(1,bkg2);
+ fAODEvBkg->SetBackground(2,bkg3);
+ fAODEvBkg->SetBackground(3,bkg4);
+ */
+ Reset();
+ return kTRUE;
+
+}
+
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();
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff