PWGJE/EMCALJetTasks/AliJetEmbeddingFromGenTask.cxx

   1 // $Id: AliJetEmbeddingFromGenTask.cxx 57324 2012-06-21 04:33:52Z loizides $
   2 //
   3 // Jet embedding task.
   4 //
   5 // Author: S.Aiola, C.Loizides
   6
   7 #include "AliJetEmbeddingFromGenTask.h"
   8
   9 #include <TClonesArray.h>
  10 #include <TLorentzVector.h>
  11 #include <TRandom3.h>
  12
  13 #include "AliAnalysisManager.h"
  14 #include "AliEMCALDigit.h"
  15 #include "AliEMCALGeometry.h"
  16 #include "AliEMCALRecPoint.h"
  17 #include "AliGenerator.h"
  18 #include "AliLog.h"
  19 #include "AliPicoTrack.h"
  20 #include "AliStack.h"
  21 #include "AliVCluster.h"
  22 #include "AliVEvent.h"
  23
  24 ClassImp(AliJetEmbeddingFromGenTask)
  25
  26 //________________________________________________________________________
  27 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask() :
  28   AliJetModelBaseTask("AliJetEmbeddingFromGenTask"),
  29   fGen(0)
  30 {
  31   // Default constructor.
  32   SetSuffix("EmbeddedFromGen");
  33 }
  34
  35 //________________________________________________________________________
  36 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask(const char *name) :
  37   AliJetModelBaseTask(name),
  38   fGen(0)
  39 {
  40   // Standard constructor.
  41   SetSuffix("EmbeddedFromGen");
  42 }
  43
  44 //________________________________________________________________________
  45 AliJetEmbeddingFromGenTask::~AliJetEmbeddingFromGenTask()
  46 {
  47   // Destructor
  48 }
  49
  50 //________________________________________________________________________
  51 void AliJetEmbeddingFromGenTask::UserExec(Option_t *)
  52 {
  53   // Execute per event.
  54
  55   if (!fIsInit) {
  56     ExecOnce();
  57     fIsInit = 1;
  58   }
  59   Run();
  60 }
  61
  62 //________________________________________________________________________
  63 void AliJetEmbeddingFromGenTask::Run()
  64 {
  65   // Embed particles.
  66
  67   fGen->Generate();
  68
  69   AliStack *stack = fGen->GetStack();
  70   const Int_t nprim = stack->GetNprimary();
  71   for (Int_t i=0;i<nprim;++i) {
  72     TParticle *part = stack->Particle(i);
  73   }
  74 }