[u/mrichter/AliRoot.git] / MUON / AliMUONData.h

#ifndef ALIMUONDATA_H
#define ALIMUONDATA_H
//
/*  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */
// Revision of includes 07/05/2004
//
/// \ingroup base
/// \class AliMUONData
/// \brief MUON data
///
/// Class containing MUON data: hits, digits, rawclusters, globaltrigger, 
/// localtrigger, etc ...
///
/// Author: Gines Martinez, Subatech,  September 2003

#include <TNamed.h>

class TArrayI;

#include "AliLoader.h"

class TClonesArray;
class TNamed;
class TObjArray;
class TTree;
class TIterator;

class AliMUONConstants;
class AliMUONRawCluster;
class AliMUONTrack;
class AliMUONTriggerTrack;
class AliMUONDigit;
class AliMUONHit;
class AliMUONLocalTrigger;
class AliMUONGlobalTrigger;

//__________________________________________________________________
/////////////////////////////////////////////////////////////////////
//                                                                 //
//  class AliMUONData                                              //
//                                                                 //
/////////////////////////////////////////////////////////////////////

class AliMUONData : public TNamed 
{
  public:
  
  enum EChamberIteration { kAllChambers, kTrackingChambers, kTriggerChambers };
  
    AliMUONData();
    AliMUONData(AliLoader * loader, const char* name, const char* title);
    virtual ~AliMUONData();  
    virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
			     Int_t* digits); 
    virtual void   AddSDigit(Int_t id, Int_t* tracks, Int_t* charges,
			     Int_t* digits); 
    virtual void   AddDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
    virtual void   AddSDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
    virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber, 
			  Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
			  Float_t tof, Float_t momentum, Float_t theta, 
			  Float_t phi, Float_t length, Float_t destep, 
			  Float_t Xref,Float_t Yref,Float_t Zref);
			  // TBR
    virtual void   AddHit2(Int_t fIshunt, Int_t track, Int_t detElemId, 
			  Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
			  Float_t tof, Float_t momentum, Float_t theta, 
			  Float_t phi, Float_t length, Float_t destep, 
			  Float_t Xref,Float_t Yref,Float_t Zref);
    
    virtual void   AddGlobalTrigger(Int_t *singlePlus, Int_t *singleMinus,
				    Int_t *singleUndef, Int_t *pairUnlike, 
				    Int_t *pairLike);
    virtual void   AddGlobalTrigger(const AliMUONGlobalTrigger& trigger); // use copy constructor

    virtual void   AddLocalTrigger(const Int_t* ltrigger, const TArrayI& digits);
    virtual void   AddLocalTrigger(const AliMUONLocalTrigger& trigger); // use copy constructor

    virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
    virtual void   AddRecTrack(const AliMUONTrack& track);
    virtual void   AddRecTriggerTrack(const AliMUONTriggerTrack& triggertrack);

    TClonesArray*  Hits() {return fHits;}
    TClonesArray*  Digits(Int_t DetectionPlane) const;
    TClonesArray*  SDigits(Int_t DetectionPlane) const;
    TClonesArray*  LocalTrigger() const;
    TClonesArray*  GlobalTrigger() const;    
    TClonesArray*  RawClusters(Int_t DetectionPlane);
    TClonesArray*  RecTracks() {return fRecTracks;}
    TClonesArray*  RecTriggerTracks() {return fRecTriggerTracks;}

    void           GetTrack(Int_t it) const  {fLoader->TreeH()->GetEvent(it);}
    Int_t          GetNtracks() const      {return (Int_t) fLoader->TreeH()->GetEntries();}
    void           GetDigits() const;
    void           GetSDigits() const {fLoader->TreeS()->GetEvent(0);}
    void           GetRawClusters() const {fLoader->TreeR()->GetEvent(0);}
    void           GetTrigger() const {fLoader->TreeR()->GetEvent(0);}
    void           GetTriggerD() const {fLoader->TreeD()->GetEvent(0);}
    Int_t          GetSplitLevel() const {return fSplitLevel;}
    void           GetRecTracks() const {fLoader->TreeT()->GetEvent(0);}
    void           GetRecTriggerTracks() const {fLoader->TreeT()->GetEvent(0);}

    Bool_t        IsRawClusterBranchesInTree();
    Bool_t        IsDigitsBranchesInTree();
    Bool_t        IsTriggerBranchesInTree();
    Bool_t        IsTriggerBranchesInTreeD();
    Bool_t        IsTrackBranchesInTree();
    Bool_t        IsTriggerTrackBranchesInTree();

    virtual AliLoader* GetLoader() const { return fLoader; }
    virtual void       SetLoader(AliLoader * loader) {fLoader=loader;}    
    
    virtual void   Fill(Option_t* opt=" ");
    virtual void   MakeBranch(Option_t *opt=" ");
    virtual void   SetTreeAddress(Option_t *opt=" ");
    
    void           SetSplitLevel(Int_t SplitLevel) {fSplitLevel=SplitLevel;}
    
    virtual void Print(Option_t* opt="") const;
    
    virtual void   ResetHits();
    virtual void   ResetDigits();
    virtual void   ResetSDigits();
    virtual void   ResetTrigger();
    virtual void   ResetRawClusters();
    virtual void   ResetRecTracks();
    virtual void   ResetRecTriggerTracks();
  
    TTree*         TreeH() {return fLoader->TreeH(); }
    TTree*         TreeD() {return fLoader->TreeD(); }
    TTree*         TreeS() {return fLoader->TreeS(); }
    TTree*         TreeR() {return fLoader->TreeR(); }
    TTree*         TreeT() {return fLoader->TreeT(); }
    TTree*         TreeP() {return fLoader->TreeP(); }

    TIterator* CreateDigitIterator(EChamberIteration type);
    
  protected: 
    AliMUONData(const AliMUONData& rhs);
    AliMUONData& operator=(const AliMUONData& rhs);

    AliLoader*  fLoader; //! Detector Loader pointer
    TClonesArray*   fHits;  // One event in treeH per primary track
    TObjArray*      fDigits; // One event in treeD and one branch per detection plane
    TObjArray*      fSDigits; // One event in treeS and one branch per detection plane
    TObjArray*      fRawClusters; //One event in TreeR/Rawcluster and one branch per tracking detection plane
    TClonesArray*   fGlobalTrigger;  // List of Global Trigger One event in TreeR/GlobalTriggerBranch
    TClonesArray*   fLocalTrigger;  // List of Local Trigger, One event in TreeR/LocalTriggerBranch
    TClonesArray*   fRecTracks; // pointer to array of reconstructed tracks
    TClonesArray*   fRecTriggerTracks; // pointer to array of reconstructed trigger tracks

    Int_t           fNhits; //!  Number of Hits
    Int_t*          fNdigits;//! Number of Digits
    Int_t*          fNSdigits;//! Number of Digits
    Int_t*          fNrawclusters;//! Number of Raw Clusters
    Int_t           fNglobaltrigger;//! Number of Global trigger
    Int_t           fNlocaltrigger;//! Number of Local trigger
    Int_t           fNrectracks; //! Number of reconstructed tracks
    Int_t           fNrectriggertracks; //! Number of reconstructed tracks
    Int_t           fSplitLevel; // Splitting of branches 0 no spitting (root files are smaller) 1 splitting (larger output files)

    mutable Int_t fCurrentEvent; // Current event we're dealing with\13
    
private:  

  ClassDef(AliMUONData,3) // Data accessor for MUON module
      
};


#endif
Commit	Line	Data
6309cf6e	1	#ifndef ALIMUONDATA_H
	2	#define ALIMUONDATA_H
	3	//
1391e633	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
50837721	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
30178c30	8	// Revision of includes 07/05/2004
6309cf6e	9	//
692de412	10	/// \ingroup base
	11	/// \class AliMUONData
	12	/// \brief MUON data
	13	///
	14	/// Class containing MUON data: hits, digits, rawclusters, globaltrigger,
	15	/// localtrigger, etc ...
	16	///
	17	/// Author: Gines Martinez, Subatech, September 2003
6309cf6e	18
30178c30	19	#include <TNamed.h>
f0945e46	20
f0945e46	21	class TArrayI;
30178c30	22
1a1cdff8	23	#include "AliLoader.h"
6309cf6e	24
6309cf6e	25	class TClonesArray;
1a1cdff8	26	class TNamed;
6309cf6e	27	class TObjArray;
6309cf6e	28	class TTree;
f0945e46	29	class TIterator;
6309cf6e	30
1a1cdff8	31	class AliMUONConstants;
6309cf6e	32	class AliMUONRawCluster;
dcd2690d	33	class AliMUONTrack;
276c44b7	34	class AliMUONTriggerTrack;
61adb9bd	35	class AliMUONDigit;
	36	class AliMUONHit;
	37	class AliMUONLocalTrigger;
	38	class AliMUONGlobalTrigger;
6309cf6e	39
	40	//__________________________________________________________________
	41	/////////////////////////////////////////////////////////////////////
	42	// //
	43	// class AliMUONData //
	44	// //
	45	/////////////////////////////////////////////////////////////////////
	46
53af523e	47	class AliMUONData : public TNamed
	48	{
	49	public:
f0945e46	50
	51	enum EChamberIteration { kAllChambers, kTrackingChambers, kTriggerChambers };
	52
6309cf6e	53	AliMUONData();
6309cf6e	54	AliMUONData(AliLoader * loader, const char* name, const char* title);
6309cf6e	55	virtual ~AliMUONData();
	56	virtual void AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
	57	Int_t* digits);
d1775029	58	virtual void AddSDigit(Int_t id, Int_t* tracks, Int_t* charges,
d1775029	59	Int_t* digits);
e3ea1889	60	virtual void AddDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
e3ea1889	61	virtual void AddSDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
6309cf6e	62	virtual void AddHit(Int_t fIshunt, Int_t track, Int_t iChamber,
	63	Int_t idpart, Float_t X, Float_t Y, Float_t Z,
	64	Float_t tof, Float_t momentum, Float_t theta,
d59b2ec3	65	Float_t phi, Float_t length, Float_t destep,
	66	Float_t Xref,Float_t Yref,Float_t Zref);
	67	// TBR
	68	virtual void AddHit2(Int_t fIshunt, Int_t track, Int_t detElemId,
1391e633	69	Int_t idpart, Float_t X, Float_t Y, Float_t Z,
	70	Float_t tof, Float_t momentum, Float_t theta,
	71	Float_t phi, Float_t length, Float_t destep,
	72	Float_t Xref,Float_t Yref,Float_t Zref);
61adb9bd	73
6309cf6e	74	virtual void AddGlobalTrigger(Int_t singlePlus, Int_t singleMinus,
	75	Int_t singleUndef, Int_t pairUnlike,
	76	Int_t *pairLike);
61adb9bd	77	virtual void AddGlobalTrigger(const AliMUONGlobalTrigger& trigger); // use copy constructor
61adb9bd	78
2be06f1e	79	virtual void AddLocalTrigger(const Int_t* ltrigger, const TArrayI& digits);
61adb9bd	80	virtual void AddLocalTrigger(const AliMUONLocalTrigger& trigger); // use copy constructor
61adb9bd	81
6309cf6e	82	virtual void AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
dcd2690d	83	virtual void AddRecTrack(const AliMUONTrack& track);
276c44b7	84	virtual void AddRecTriggerTrack(const AliMUONTriggerTrack& triggertrack);
dcd2690d	85
6309cf6e	86	TClonesArray* Hits() {return fHits;}
c82862d3	87	TClonesArray* Digits(Int_t DetectionPlane) const;
c82862d3	88	TClonesArray* SDigits(Int_t DetectionPlane) const;
f0945e46	89	TClonesArray* LocalTrigger() const;
f0945e46	90	TClonesArray* GlobalTrigger() const;
1eccde20	91	TClonesArray* RawClusters(Int_t DetectionPlane);
dcd2690d	92	TClonesArray* RecTracks() {return fRecTracks;}
276c44b7	93	TClonesArray* RecTriggerTracks() {return fRecTriggerTracks;}
1a1cdff8	94
30178c30	95	void GetTrack(Int_t it) const {fLoader->TreeH()->GetEvent(it);}
30178c30	96	Int_t GetNtracks() const {return (Int_t) fLoader->TreeH()->GetEntries();}
f0945e46	97	void GetDigits() const;
935b9895	98	void GetSDigits() const {fLoader->TreeS()->GetEvent(0);}
30178c30	99	void GetRawClusters() const {fLoader->TreeR()->GetEvent(0);}
30178c30	100	void GetTrigger() const {fLoader->TreeR()->GetEvent(0);}
ce3e25a8	101	void GetTriggerD() const {fLoader->TreeD()->GetEvent(0);}
30178c30	102	Int_t GetSplitLevel() const {return fSplitLevel;}
	103	void GetRecTracks() const {fLoader->TreeT()->GetEvent(0);}
	104	void GetRecTriggerTracks() const {fLoader->TreeT()->GetEvent(0);}
1a1cdff8	105
c1d45bdf	106	Bool_t IsRawClusterBranchesInTree();
ce3e25a8	107	Bool_t IsDigitsBranchesInTree();
c1d45bdf	108	Bool_t IsTriggerBranchesInTree();
ce3e25a8	109	Bool_t IsTriggerBranchesInTreeD();
276c44b7	110	Bool_t IsTrackBranchesInTree();
276c44b7	111	Bool_t IsTriggerTrackBranchesInTree();
c1d45bdf	112
f0945e46	113	virtual AliLoader* GetLoader() const { return fLoader; }
ce3f5e87	114	virtual void SetLoader(AliLoader * loader) {fLoader=loader;}
dcd2690d	115
1a1cdff8	116	virtual void Fill(Option_t* opt=" ");
6309cf6e	117	virtual void MakeBranch(Option_t *opt=" ");
ce3f5e87	118	virtual void SetTreeAddress(Option_t *opt=" ");
dcd2690d	119
d652f85c	120	void SetSplitLevel(Int_t SplitLevel) {fSplitLevel=SplitLevel;}
d652f85c	121
c82862d3	122	virtual void Print(Option_t* opt="") const;
c82862d3	123
6309cf6e	124	virtual void ResetHits();
6309cf6e	125	virtual void ResetDigits();
d1775029	126	virtual void ResetSDigits();
6309cf6e	127	virtual void ResetTrigger();
6309cf6e	128	virtual void ResetRawClusters();
dcd2690d	129	virtual void ResetRecTracks();
276c44b7	130	virtual void ResetRecTriggerTracks();
6309cf6e	131
	132	TTree* TreeH() {return fLoader->TreeH(); }
	133	TTree* TreeD() {return fLoader->TreeD(); }
d1775029	134	TTree* TreeS() {return fLoader->TreeS(); }
6309cf6e	135	TTree* TreeR() {return fLoader->TreeR(); }
	136	TTree* TreeT() {return fLoader->TreeT(); }
	137	TTree* TreeP() {return fLoader->TreeP(); }
	138
f0945e46	139	TIterator* CreateDigitIterator(EChamberIteration type);
f0945e46	140
53af523e	141	protected:
e3ea1889	142	AliMUONData(const AliMUONData& rhs);
	143	AliMUONData& operator=(const AliMUONData& rhs);
	144
9140dcef	145	AliLoader* fLoader; //! Detector Loader pointer
	146	TClonesArray* fHits; // One event in treeH per primary track
	147	TObjArray* fDigits; // One event in treeD and one branch per detection plane
d1775029	148	TObjArray* fSDigits; // One event in treeS and one branch per detection plane
9140dcef	149	TObjArray* fRawClusters; //One event in TreeR/Rawcluster and one branch per tracking detection plane
	150	TClonesArray* fGlobalTrigger; // List of Global Trigger One event in TreeR/GlobalTriggerBranch
	151	TClonesArray* fLocalTrigger; // List of Local Trigger, One event in TreeR/LocalTriggerBranch
	152	TClonesArray* fRecTracks; // pointer to array of reconstructed tracks
276c44b7	153	TClonesArray* fRecTriggerTracks; // pointer to array of reconstructed trigger tracks
9140dcef	154
	155	Int_t fNhits; //! Number of Hits
	156	Int_t* fNdigits;//! Number of Digits
d1775029	157	Int_t* fNSdigits;//! Number of Digits
9140dcef	158	Int_t* fNrawclusters;//! Number of Raw Clusters
	159	Int_t fNglobaltrigger;//! Number of Global trigger
	160	Int_t fNlocaltrigger;//! Number of Local trigger
	161	Int_t fNrectracks; //! Number of reconstructed tracks
276c44b7	162	Int_t fNrectriggertracks; //! Number of reconstructed tracks
d652f85c	163	Int_t fSplitLevel; // Splitting of branches 0 no spitting (root files are smaller) 1 splitting (larger output files)
6309cf6e	164
f0945e46	165	mutable Int_t fCurrentEvent; // Current event we're dealing with\13
	166
	167	private:
53af523e	168
f0945e46	169	ClassDef(AliMUONData,3) // Data accessor for MUON module
f0945e46	170
53af523e	171	};
f0945e46	172
f0945e46	173
6309cf6e	174	#endif
8547965d	175