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
  24 class AliDielectronEvent : public TNamed {
  25 public:
  26   AliDielectronEvent();
  27   AliDielectronEvent(const char*name, const char* title);
  28
  29   virtual ~AliDielectronEvent();
  30
  31   void SetAOD();
  32
  33   void SetTracks(const TObjArray &arrP, const TObjArray &arrN, const TObjArray &arrPairs);
  34   void SetEventData(const Double_t data[AliDielectronVarManager::kNMaxValues]);
  35   const Double_t* GetEventData() const {return fEventData;}
  36
  37
  38   const TClonesArray* GetTrackArrayP() const { return &fArrTrackP; }
  39   const TClonesArray* GetTrackArrayN() const { return &fArrTrackN; }
  40
  41   Int_t GetNTracksP() const { return fNTracksP; }
  42   Int_t GetNTracksN() const { return fNTracksN; }
  43
  44 virtual void Clear(Option_t *opt="C");
  45
  46
  47 private:
  48   TClonesArray fArrTrackP;      //positive tracks
  49   TClonesArray fArrTrackN;      //negative tracks
  50
  51   TClonesArray fArrPairs;       //Pair array
  52
  53   Int_t fNTracksP;              //number of positive tracks
  54   Int_t fNTracksN;              //number of negative tracks
  55
  56   Bool_t fIsAOD;                // if we deal with AODs
  57
  58   Double_t fEventData[AliDielectronVarManager::kNMaxValues]; // event informaion from the var manager
  59
  60   AliDielectronEvent(const AliDielectronEvent &c);
  61   AliDielectronEvent &operator=(const AliDielectronEvent &c);
  62
  63
  64   ClassDef(AliDielectronEvent,1)         // Dielectron Event
  65 };
  66
  67
  68
  69 #endif