MUON/AliMUONData.h

   1 #ifndef ALIMUONDATA_H
   2 #define ALIMUONDATA_H
   3 //
   4 /*  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8
   9 // AliMUONData
  10 // Class containing MUON data: hits, digits, rawclusters, globaltrigger, localtrigger, etc ...
  11 // Gines Martinez, Subatech,  September 2003
  12 //
  13
  14 #include "AliLoader.h"
  15
  16 class TClonesArray;
  17 class TNamed;
  18 class TObjArray;
  19 class TTree;
  20
  21
  22 class AliMUONConstants;
  23 class AliMUONRawCluster;
  24 class AliMUONTrack;
  25 class AliMUONTriggerTrack;
  26 class AliMUONDigit;
  27 class AliMUONHit;
  28 class AliMUONLocalTrigger;
  29 class AliMUONGlobalTrigger;
  30
  31 //__________________________________________________________________
  32 /////////////////////////////////////////////////////////////////////
  33 //                                                                 //
  34 //  class AliMUONData                                              //
  35 //                                                                 //
  36 /////////////////////////////////////////////////////////////////////
  37
  38 class AliMUONData : public TNamed {
  39  public:
  40     AliMUONData();
  41     AliMUONData(AliLoader * loader, const char* name, const char* title);
  42     AliMUONData(const AliMUONData& rMUONData);
  43     virtual ~AliMUONData();
  44     virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
  45                              Int_t* digits);
  46     virtual void   AddDigit(Int_t, const AliMUONDigit& ); // use copy constructor
  47     virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber,
  48                           Int_t idpart, Float_t X, Float_t Y, Float_t Z,
  49                           Float_t tof, Float_t momentum, Float_t theta,
  50                           Float_t phi, Float_t length, Float_t destep);
  51     virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber,
  52                           Int_t idpart, Float_t X, Float_t Y, Float_t Z,
  53                           Float_t tof, Float_t momentum, Float_t theta,
  54                           Float_t phi, Float_t length, Float_t destep,
  55                           Float_t Xref,Float_t Yref,Float_t Zref);
  56     virtual void   AddHit(const AliMUONHit& ); // use copy constructor
  57
  58     virtual void   AddGlobalTrigger(Int_t *singlePlus, Int_t *singleMinus,
  59                                     Int_t *singleUndef, Int_t *pairUnlike,
  60                                     Int_t *pairLike);
  61     virtual void   AddGlobalTrigger(const AliMUONGlobalTrigger& trigger); // use copy constructor
  62
  63     virtual void   AddLocalTrigger(Int_t* ltrigger);
  64     virtual void   AddLocalTrigger(const AliMUONLocalTrigger& trigger); // use copy constructor
  65
  66     virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
  67     virtual void   AddRecTrack(const AliMUONTrack& track);
  68     virtual void   AddRecTriggerTrack(const AliMUONTriggerTrack& triggertrack);
  69
  70     TClonesArray*  Hits() {return fHits;}
  71     TClonesArray*  Digits(Int_t DetectionPlane);
  72 //    TClonesArray*  LocalTrigger() {return fLocalTrigger;}
  73 //    TClonesArray*  GlobalTrigger() {return fGlobalTrigger;}
  74     TClonesArray*  LocalTrigger();
  75     TClonesArray*  GlobalTrigger();
  76     TClonesArray*  RawClusters(Int_t DetectionPlane);
  77     TClonesArray*  RecTracks() {return fRecTracks;}
  78     TClonesArray*  RecTriggerTracks() {return fRecTriggerTracks;}
  79
  80     void           GetTrack(Int_t it) {fLoader->TreeH()->GetEvent(it);}
  81     Int_t          GetNtracks()       {return (Int_t) fLoader->TreeH()->GetEntries();}
  82     void           GetCathode(Int_t ic) {fLoader->TreeD()->GetEvent(ic);}
  83     void           GetRawClusters() {fLoader->TreeR()->GetEvent(0);}
  84     void           GetTrigger() {fLoader->TreeR()->GetEvent(0);}
  85     Int_t          GetSplitLevel() {return fSplitLevel;}
  86     void           GetRecTracks() {fLoader->TreeT()->GetEvent(0);}
  87     void           GetRecTriggerTracks() {fLoader->TreeT()->GetEvent(0);}
  88
  89     Bool_t        IsRawClusterBranchesInTree();
  90     Bool_t        IsTriggerBranchesInTree();
  91     Bool_t        IsTrackBranchesInTree();
  92     Bool_t        IsTriggerTrackBranchesInTree();
  93
  94     virtual AliLoader* GetLoader() {return fLoader;}
  95     virtual void       SetLoader(AliLoader * loader) {fLoader=loader;}
  96
  97     virtual void   Fill(Option_t* opt=" ");
  98     virtual void   MakeBranch(Option_t *opt=" ");
  99     virtual void   SetTreeAddress(Option_t *opt=" ");
 100
 101     void           SetSplitLevel(Int_t SplitLevel) {fSplitLevel=SplitLevel;}
 102
 103
 104     virtual void   ResetHits();
 105     virtual void   ResetDigits();
 106     virtual void   ResetTrigger();
 107     virtual void   ResetRawClusters();
 108     virtual void   ResetRecTracks();
 109     virtual void   ResetRecTriggerTracks();
 110
 111     TTree*         TreeH() {return fLoader->TreeH(); }
 112     TTree*         TreeD() {return fLoader->TreeD(); }
 113     TTree*         TreeR() {return fLoader->TreeR(); }
 114     TTree*         TreeT() {return fLoader->TreeT(); }
 115     TTree*         TreeP() {return fLoader->TreeP(); }
 116
 117  private:
 118     //descendant classes should
 119     //use protected interface methods to access these folders
 120
 121
 122  protected:
 123     AliLoader*  fLoader; //! Detector Loader pointer
 124     TClonesArray*   fHits;  // One event in treeH per primary track
 125     TObjArray*      fDigits; // One event in treeD and one branch per detection plane
 126     TObjArray*      fRawClusters; //One event in TreeR/Rawcluster and one branch per tracking detection plane
 127     TClonesArray*   fGlobalTrigger;  // List of Global Trigger One event in TreeR/GlobalTriggerBranch
 128     TClonesArray*   fLocalTrigger;  // List of Local Trigger, One event in TreeR/LocalTriggerBranch
 129     TClonesArray*   fRecTracks; // pointer to array of reconstructed tracks
 130     TClonesArray*   fRecTriggerTracks; // pointer to array of reconstructed trigger tracks
 131
 132     Int_t           fNhits; //!  Number of Hits
 133     Int_t*          fNdigits;//! Number of Digits
 134     Int_t*          fNrawclusters;//! Number of Raw Clusters
 135     Int_t           fNglobaltrigger;//! Number of Global trigger
 136     Int_t           fNlocaltrigger;//! Number of Local trigger
 137     Int_t           fNrectracks; //! Number of reconstructed tracks
 138     Int_t           fNrectriggertracks; //! Number of reconstructed tracks
 139     Int_t           fSplitLevel; // Splitting of branches 0 no spitting (root files are smaller) 1 splitting (larger output files)
 140
 141     ClassDef(AliMUONData,1)
 142  };
 143 #endif
 144