JETAN/AliJetFinder.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 // Jet finder base class
  20 // manages the search for jets
  21 // Authors: jgcn@mda.cinvestav.mx
  22 //          andreas.morsch@cern.ch
  23 //          magali.estienne@subatech.in2p3.fr
  24 //---------------------------------------------------------------------
  25
  26 #include <Riostream.h>
  27 #include <TFile.h>
  28
  29 #include "AliJetFinder.h"
  30 #include "AliJet.h"
  31 #include "AliAODJet.h"
  32 #include "AliJetControlPlots.h"
  33 #include "AliLeading.h"
  34 #include "AliAODEvent.h"
  35 #include "AliJetUnitArray.h"
  36
  37 class TProcessID;
  38 class TClonesArray;
  39
  40 ClassImp(AliJetFinder)
  41
  42 AliJetFinder::AliJetFinder():
  43     fTreeJ(0),
  44     fPlotMode(kFALSE),
  45     fJets(0),
  46     fGenJets(0),
  47     fLeading(0),
  48     fReader(0x0),
  49     fHeader(0x0),
  50     fAODjets(0x0),
  51     fNAODjets(0),
  52     fPlots(0x0),
  53     fOut(0x0)
  54
  55 {
  56   //
  57   // Constructor
  58   //
  59
  60   fJets    = new AliJet();
  61   fGenJets = new AliJet();
  62   fLeading = new AliLeading();
  63   fAODjets = 0;
  64 }
  65
  66 ////////////////////////////////////////////////////////////////////////
  67 AliJetFinder::~AliJetFinder()
  68 {
  69   //
  70   // Destructor
  71   //
  72
  73   // Reset and delete jets
  74   fJets->ClearJets();
  75   delete fJets;
  76   fGenJets->ClearJets();
  77   delete fGenJets;
  78   // close file
  79   if (fOut) {
  80     fOut->Close();
  81     fOut->Delete();
  82   }
  83   delete fOut;
  84   // reset and delete control plots
  85   if (fPlots) delete fPlots;
  86 }
  87
  88 ////////////////////////////////////////////////////////////////////////
  89 void AliJetFinder::SetOutputFile(const char */*name*/)
  90 {
  91   //  opens output file
  92   //  fOut = new TFile(name,"recreate");
  93 }
  94
  95 ////////////////////////////////////////////////////////////////////////
  96 void AliJetFinder::PrintJets()
  97 {
  98   // Print jet information
  99   cout << " Jets found with jet algorithm:" << endl;
 100   fJets->PrintJets();
 101   cout << " Jets found by pythia:" << endl;
 102   fGenJets->PrintJets();
 103 }
 104
 105 ////////////////////////////////////////////////////////////////////////
 106 void AliJetFinder::SetPlotMode(Bool_t b)
 107 {
 108   // Sets the plotting mode
 109   fPlotMode=b;
 110   if (b && !fPlots) fPlots = new AliJetControlPlots();
 111 }
 112
 113 ////////////////////////////////////////////////////////////////////////
 114 TTree* AliJetFinder::MakeTreeJ(char* name)
 115 {
 116     // Create the tree for reconstructed jets
 117     fTreeJ = new TTree(name, "AliJet");
 118     fTreeJ->Branch("FoundJet",   &fJets,   1000);
 119     fTreeJ->Branch("GenJet",     &fGenJets,1000);
 120     fTreeJ->Branch("LeadingPart",&fLeading,1000);
 121     return fTreeJ;
 122 }
 123
 124 ////////////////////////////////////////////////////////////////////////
 125 void AliJetFinder::WriteRHeaderToFile()
 126 {
 127   // write reader header
 128     AliJetReaderHeader *rh = fReader->GetReaderHeader();
 129     rh->Write();
 130 }
 131
 132 ////////////////////////////////////////////////////////////////////////
 133 void AliJetFinder::Run()
 134 {
 135   // Do some initialization
 136   Init();
 137   // connect files
 138   fReader->OpenInputFiles();
 139
 140   // write headers
 141   WriteHeaders();
 142   // loop over events
 143   Int_t nFirst, nLast, option, debug, arrayInitialised;
 144   nFirst = fReader->GetReaderHeader()->GetFirstEvent();
 145   nLast  = fReader->GetReaderHeader()->GetLastEvent();
 146
 147   option = fReader->GetReaderHeader()->GetDetector();
 148   debug  = fReader->GetReaderHeader()->GetDebug();
 149   arrayInitialised = fReader->GetArrayInitialised();
 150
 151   // loop over events
 152   for (Int_t i=nFirst;i<nLast;i++) {
 153       fReader->FillMomentumArray();
 154       fLeading->FindLeading(fReader);
 155       fReader->GetGenJets(fGenJets);
 156
 157       if (option == 0) { // TPC with fMomentumArray
 158           if(debug > 1)
 159               printf("In FindJetsC() routine: find jets with fMomentumArray !!!\n");
 160           FindJetsC();
 161       } else {
 162         if(debug > 1) printf("In FindJets() routine: find jets with fUnitArray !!!\n");
 163         FindJets();
 164       }
 165       if (fOut) {
 166           fOut->cd();
 167       }
 168
 169       if (fPlots) fPlots->FillHistos(fJets);
 170       fLeading->Reset();
 171       fGenJets->ClearJets();
 172       Reset();
 173   }
 174
 175   // write out
 176   if (fPlots) {
 177       fPlots->Normalize();
 178       fPlots->PlotHistos();
 179   }
 180   if (fOut) {
 181       fOut->cd();
 182       fPlots->Write();
 183       fOut->Close();
 184   }
 185 }
 186
 187
 188 //
 189 // The following methods have been added to allow for event steering from the outside
 190 //
 191
 192 ////////////////////////////////////////////////////////////////////////
 193 void AliJetFinder::ConnectTree(TTree* tree, TObject* data)
 194 {
 195     // Connect the input file
 196     fReader->ConnectTree(tree, data);
 197 }
 198
 199 ////////////////////////////////////////////////////////////////////////
 200 void AliJetFinder::WriteHeaders()
 201 {
 202     // Write the Headers
 203     TFile* f = new TFile("jets_local.root", "recreate");
 204     WriteRHeaderToFile();
 205     WriteJHeaderToFile();
 206     f->Close();
 207 }
 208
 209 ////////////////////////////////////////////////////////////////////////
 210 Bool_t AliJetFinder::ProcessEvent()
 211 {
 212   //
 213   // Process one event
 214   // Charged only jets
 215   //
 216
 217   Bool_t ok = fReader->FillMomentumArray();
 218   if (!ok) return kFALSE;
 219
 220   // Leading particles
 221   fLeading->FindLeading(fReader);
 222   // Jets
 223   FindJets(); // V1
 224   //  FindJetsC(); // V2
 225
 226   if (fPlots) fPlots->FillHistos(fJets);
 227   fLeading->Reset();
 228   fGenJets->ClearJets();
 229   Reset();
 230   return kTRUE;
 231 }
 232
 233 ////////////////////////////////////////////////////////////////////////
 234 Bool_t AliJetFinder::ProcessEvent2()
 235 {
 236   //
 237   // Process one event
 238   // Charged only or charged+neutral jets
 239   //
 240
 241   TRefArray* ref = new TRefArray();
 242   Bool_t procid = kFALSE;
 243   Bool_t ok = fReader->ExecTasks(procid,ref);
 244
 245   // Delete reference pointer
 246   if (!ok) {delete ref; return kFALSE;}
 247
 248   // Leading particles
 249   fLeading->FindLeading(fReader);
 250   // Jets
 251   FindJets();
 252
 253   Int_t         nEntRef    = ref->GetEntries();
 254
 255   for(Int_t i=0; i<nEntRef; i++)
 256     {
 257       // Reset the UnitArray content which were referenced
 258       ((AliJetUnitArray*)ref->At(i))->SetUnitTrackID(0);
 259       ((AliJetUnitArray*)ref->At(i))->SetUnitEnergy(0.);
 260       ((AliJetUnitArray*)ref->At(i))->SetUnitCutFlag(kPtSmaller);
 261       ((AliJetUnitArray*)ref->At(i))->SetUnitCutFlag2(kPtSmaller);
 262       ((AliJetUnitArray*)ref->At(i))->SetUnitSignalFlag(kBad);
 263       ((AliJetUnitArray*)ref->At(i))->SetUnitSignalFlagC(kTRUE,kBad);
 264       ((AliJetUnitArray*)ref->At(i))->SetUnitDetectorFlag(kTpc);
 265       ((AliJetUnitArray*)ref->At(i))->SetUnitFlag(kOutJet);
 266       ((AliJetUnitArray*)ref->At(i))->ClearUnitTrackRef();
 267
 268       // Reset process ID
 269       AliJetUnitArray* uA = (AliJetUnitArray*)ref->At(i);
 270       uA->ResetBit(kIsReferenced);
 271       uA->SetUniqueID(0);
 272     }
 273
 274   // Delete the reference pointer
 275   ref->Delete();
 276   delete ref;
 277
 278   if (fPlots) fPlots->FillHistos(fJets);
 279   fLeading->Reset();
 280   fGenJets->ClearJets();
 281   Reset();
 282
 283   return kTRUE;
 284 }
 285
 286 ////////////////////////////////////////////////////////////////////////
 287 void AliJetFinder::FinishRun()
 288 {
 289     // Finish a run
 290     if (fPlots) {
 291         fPlots->Normalize();
 292         fPlots->PlotHistos();
 293     }
 294
 295     if (fOut) {
 296          fOut->cd();
 297          if (fPlots) {
 298              fPlots->Write();
 299          }
 300          fOut->Close();
 301     }
 302 }
 303
 304 ////////////////////////////////////////////////////////////////////////
 305 void AliJetFinder::AddJet(AliAODJet p)
 306 {
 307 // Add new jet to the list
 308   new ((*fAODjets)[fNAODjets++]) AliAODJet(p);
 309 }
 310
 311 ////////////////////////////////////////////////////////////////////////
 312 void AliJetFinder::ConnectAOD(AliAODEvent* aod)
 313 {
 314 // Connect to the AOD
 315     fAODjets = aod->GetJets();
 316 }
 317
 318 ////////////////////////////////////////////////////////////////////////
 319 void AliJetFinder::ConnectAODNonStd(AliAODEvent* aod,const char *bname)
 320 {
 321
 322   fAODjets = dynamic_cast<TClonesArray*>(aod->FindListObject(bname));
 323   // how is this is reset? Cleared?
 324 }
 325