PWGDQ/dielectron/AliDielectronEvent.h

   1 #ifndef ALIDIELECTRONEVENT_H
   2 #define ALIDIELECTRONEVENT_H
   3
   4 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //#############################################################
   8 //#                                                           #
   9 //#         Class AliDielectronEvent                     #
  10 //#                                                           #
  11 //#  Authors:                                                 #
  12 //#   Jens      Wiechula, Uni Tübingen / Jens.Wiechula@cern.ch      #
  13 //#                                                           #
  14 //#############################################################
  15
  16 #include <TNamed.h>
  17 #include <TClonesArray.h>
  18
  19 #include "AliDielectronVarManager.h"
  20
  21
  22 class TObjArray;
  23 class TMap;
  24
  25 class AliDielectronEvent : public TNamed {
  26 public:
  27   AliDielectronEvent();
  28   AliDielectronEvent(const char*name, const char* title);
  29
  30   virtual ~AliDielectronEvent();
  31
  32   void SetESD();
  33   void SetAOD();
  34   Bool_t IsAOD() const { return fIsAOD; }
  35
  36   void SetTracks(const TObjArray &arrP, const TObjArray &arrN, const TObjArray &arrPairs);
  37   void SetEventData(const Double_t data[AliDielectronVarManager::kNMaxValues]);
  38   const Double_t* GetEventData() const {return fEventData;}
  39
  40   const TClonesArray* GetTrackArrayP() const { return &fArrTrackP; }
  41   const TClonesArray* GetTrackArrayN() const { return &fArrTrackN; }
  42   const TClonesArray* GetVertexArray() const { return &fArrVertex; }
  43
  44   Int_t GetNTracksP() const { return fNTracksP; }
  45   Int_t GetNTracksN() const { return fNTracksN; }
  46
  47 virtual void Clear(Option_t *opt="C");
  48
  49
  50 private:
  51   TClonesArray fArrTrackP;      //positive tracks
  52   TClonesArray fArrTrackN;      //negative tracks
  53   TClonesArray fArrVertex;      //track vertices
  54
  55   TClonesArray fArrPairs;       //Pair array
  56
  57   TMap *fMapStoredVertices;//! already buffered vertices in AOD
  58
  59   Int_t fNTracksP;              //number of positive tracks
  60   Int_t fNTracksN;              //number of negative tracks
  61
  62   Bool_t fIsAOD;                // if we deal with AODs
  63
  64   Double_t fEventData[AliDielectronVarManager::kNMaxValues]; // event informaion from the var manager
  65
  66   AliDielectronEvent(const AliDielectronEvent &c);
  67   AliDielectronEvent &operator=(const AliDielectronEvent &c);
  68
  69
  70   ClassDef(AliDielectronEvent,1)         // Dielectron Event
  71 };
  72
  73
  74
  75 #endif