Merge branch 'master' into TPCdev

[u/mrichter/AliRoot.git] / TPC / Upgrade / AliToyMCEvent.h

#ifndef AliToyMCEvent_H
#define AliToyMCEvent_H


#include <TClonesArray.h>
#include "AliToyMCTrack.h"

class AliToyMCEvent : public TObject {
 public:
  AliToyMCEvent();
  AliToyMCEvent(const AliToyMCEvent &event);
  virtual ~AliToyMCEvent() {}
  AliToyMCEvent& operator = (const AliToyMCEvent &event);

  enum EEventType {
    kPhysics=0,
    kLaser
  };
  
  static void SetInitialEventCounter(Int_t iev) { fgEvCounter=iev; }

  AliToyMCTrack* AddTrack(const AliToyMCTrack &track);
  AliToyMCTrack* AddTrack(Double_t xyz[3],Double_t pxpypz[3],
                          Double_t cv[21],Short_t sign);
  
  Int_t GetNumberOfTracks() const { return fTracks.GetEntriesFast(); }
  const AliToyMCTrack* GetTrack(Int_t track) const { return static_cast<const AliToyMCTrack*>(fTracks.At(track)); }
    
  void SetT0 (Float_t time)              { fT0   = time;          }
  void SetX(Float_t var)                 { fX = var;              }
  void SetY(Float_t var)                 { fY = var;              }
  void SetZ(Float_t var)                 { fZ = var;              }

  UInt_t GetEventNumber()    const {return fEventNumber;  }
  Float_t GetT0()            const {return fT0;           }
  Float_t GetX()             const {return fX;            }
  Float_t GetY()             const {return fY;            }
  Float_t GetZ()             const {return fZ;            }

  void SetEventType(EEventType type) { fEventType=type; }
  EEventType GetEventType() const    { return fEventType; }

  void     SetSCscale(Float_t  val)      { fSCscale = val;      }
  Float_t  GetSCscale() const            { return fSCscale;     }

  void     SetSCscaleChi2(Float_t  val)  { fSCscaleChi2 = val;  }
  Float_t  GetSCscaleChi2() const        { return fSCscaleChi2; }
  
 private:
  static Int_t fgEvCounter;      // static counter
    
  UInt_t fEventNumber;           // event number

  EEventType fEventType;         // type of the event
    
  Float_t fT0;                   // Interaction time of the event
  Float_t fX;                    // x-vertex position
  Float_t fY;                    // y-vertex position
  Float_t fZ;                    // z-vertex position

  Float_t fSCscale;              // scaling parameter for space charge correction
  Float_t fSCscaleChi2;          // chi2 of scaling parameter for space charge correction
  
  TClonesArray fTracks;          // array of tracks

  ClassDef(AliToyMCEvent, 2);    // MC event class

};

#endif