PWGGA/EMCALJetTasks/AliJetEmbeddingTask.cxx

   1 // $Id$
   2 //
   3 // Jet embedding task.
   4 //
   5 // Author: S.Aiola, C.Loizides
   6
   7 #include <TClonesArray.h>
   8 #include <TLorentzVector.h>
   9 #include <TRandom3.h>
  10
  11 #include "AliAnalysisManager.h"
  12 #include "AliVEvent.h"
  13 #include "AliVCluster.h"
  14 #include "AliEMCALDigit.h"
  15 #include "AliEMCALRecPoint.h"
  16 #include "AliPicoTrack.h"
  17 #include "AliEMCALGeometry.h"
  18 #include "AliLog.h"
  19
  20 #include "AliJetEmbeddingTask.h"
  21
  22 ClassImp(AliJetEmbeddingTask)
  23
  24 //________________________________________________________________________
  25 AliJetEmbeddingTask::AliJetEmbeddingTask() :
  26   AliJetModelBaseTask("AliJetEmbeddingTask")
  27 {
  28   // Default constructor.
  29   SetSuffix("Embedded");
  30 }
  31
  32 //________________________________________________________________________
  33 AliJetEmbeddingTask::AliJetEmbeddingTask(const char *name) :
  34   AliJetModelBaseTask(name)
  35 {
  36   // Standard constructor.
  37   SetSuffix("Embedded");
  38 }
  39
  40 //________________________________________________________________________
  41 AliJetEmbeddingTask::~AliJetEmbeddingTask()
  42 {
  43   // Destructor
  44 }
  45
  46 //________________________________________________________________________
  47 void AliJetEmbeddingTask::Run()
  48 {
  49   // Embed particles.
  50
  51   if (fNClusters > 0 && fOutClusters) {
  52     if (fCopyArray)
  53       CopyClusters();
  54     for (Int_t i = 0; i < fNClusters; ++i) {
  55       AddCluster();
  56     }
  57   }
  58
  59   if (fNTracks > 0 && fOutTracks) {
  60     if (fCopyArray)
  61       CopyTracks();
  62     for (Int_t i = 0; i < fNTracks; ++i) {
  63       AddTrack();
  64     }
  65   }
  66 }
  67
  68 //________________________________________________________________________
  69 void AliJetEmbeddingTask::UserExec(Option_t *)
  70 {
  71   // Execute per event.
  72
  73   Init();
  74
  75   Run();
  76 }