[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTRDLoader.h

// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/
#ifndef AliEveTRDLoader_H
#define AliEveTRDLoader_H

////////////////////////////////////////////////////////////////////////
// - ALIEVE implementation -
// Loader for the TRD detector - base class
//    - AliEveTRDLoader - loader of TRD data (simulation + measured)
//    - AliEveTRDLoaderEditor - UI
//
// by A.Bercuci (A.Bercuci@gsi.de)   Fri Oct 27 2006
////////////////////////////////////////////////////////////////////////

#include <TEveElement.h>
#include <TGedFrame.h>
#include <TString.h>

class AliTRDv1;
class AliTRDgeometry;

class TGNumberEntry;
class TGColorSelect;
class TGTextEntry;
class TTree;

class TEveGValuator;

class AliEveTRDChamber;
class AliEveTRDLoaderManager;

class AliEveTRDLoader : public TEveElementList
{
  friend class AliEveTRDLoaderEditor;

public:
  enum TRDDataTypes {
    kTRDHits     = BIT(0),
    kTRDDigits   = BIT(1),
    kTRDClusters = BIT(2),
    kTRDTracklets= BIT(3),
    kTRDRawRoot  = BIT(4),
    kTRDRawDate  = BIT(5)
  };

  AliEveTRDLoader(const Text_t* n="AliEveTRDLoader", const Text_t* t=0);
  virtual ~AliEveTRDLoader() {}

  virtual void		AddChambers(int sm=-1, int stk=-1, int ly=-1);
  virtual AliEveTRDChamber*	GetChamber(int d);
  virtual Int_t   GetDataType() const {return fDataType;}
  virtual Int_t   GetEvent() const {return fEvent;}
  virtual Bool_t	GoToEvent(int ev);
          Bool_t  IsDataLinked() const {return TestBit(1);}

  virtual Bool_t	Open(const char *file, const char *dir = ".");
  inline virtual  Bool_t  NextEvent(Bool_t rewind = kTRUE);
  virtual void 		Paint(Option_t *option="");

  virtual void		SetDataType(UChar_t type = 0){fDataType = type;}


protected:
  virtual Bool_t	LoadHits(TTree *tH);
  virtual Bool_t	LoadClusters(TTree *tC);
  virtual Bool_t	LoadDigits(TTree *tD);
  virtual Bool_t	LoadTracklets(TTree *trklTree);
          void    SetDataLinked(Bool_t linked = kTRUE) {SetBit(1, linked);}

  virtual void		Unload();

  UChar_t fDataType;        // data type
  //Bool_t	fLoadHits, fLoadDigits, fLoadClusters, fLoadTracks; // flags for data-loading
  Char_t	fSM, fStack, fLy; // supermodule, stack, layer
  Short_t		fEvent;         // current event to be displayed
  AliTRDgeometry		*fGeo;  // the TRD geometry
  TString	fFilename;        // name of data file
  TString	fDir;             // data directory

private:
  AliEveTRDLoader(const AliEveTRDLoader&);            // Not implemented
  AliEveTRDLoader& operator=(const AliEveTRDLoader&); // Not implemented

  ClassDef(AliEveTRDLoader, 0); // Alieve Loader class for the TRD detector.
};

Bool_t AliEveTRDLoader::NextEvent(Bool_t)
{
  fEvent++;
  return GoToEvent(fEvent);
}

class AliEveTRDLoaderEditor : public TGedFrame
{
public:
  AliEveTRDLoaderEditor(const TGWindow* p=0, Int_t width=170, Int_t height=30,
			UInt_t options=kChildFrame, Pixel_t back=GetDefaultFrameBackground());
  virtual ~AliEveTRDLoaderEditor() {}

  virtual void	AddChambers();
  virtual void	FileOpen();
  virtual void	GoTo();
  virtual void	Next();
  virtual void	SetEvent(Double_t ev){fM->fEvent = (Int_t)ev;}
  virtual void	SetModel(TObject* obj);

protected:
  AliEveTRDLoader	*fM;     // Model object.
  TGTextEntry		*fFile;    // File name weed.
  TGTextButton  *fBrowse;  // browse button
  TEveGValuator		*fEvent; // Event no weed.
  TEveGValuator		*fSMNumber, *fStackNumber, *fPlaneNumber; // Detector id weeds.

private:
  AliEveTRDLoaderEditor(const AliEveTRDLoaderEditor&);            // Not implemented
  AliEveTRDLoaderEditor& operator=(const AliEveTRDLoaderEditor&); // Not implemented

  ClassDef(AliEveTRDLoaderEditor, 0); // Editor for AliEveTRDLoader.
};

#endif
Commit	Line	Data
	1	// $Id$
	2	// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
	3
	4	/**************************************************************************
	5	* Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
	6	* See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for *
	7	* full copyright notice. *
	8	**************************************************************************/
	9	#ifndef AliEveTRDLoader_H
	10	#define AliEveTRDLoader_H
	11
	12	////////////////////////////////////////////////////////////////////////
	13	// - ALIEVE implementation -
	14	// Loader for the TRD detector - base class
	15	// - AliEveTRDLoader - loader of TRD data (simulation + measured)
	16	// - AliEveTRDLoaderEditor - UI
	17	//
	18	// by A.Bercuci (A.Bercuci@gsi.de) Fri Oct 27 2006
	19	////////////////////////////////////////////////////////////////////////
	20
	21	#include <TEveElement.h>
	22	#include <TGedFrame.h>
	23	#include <TString.h>
	24
	25	class AliTRDv1;
	26	class AliTRDgeometry;
	27
	28	class TGNumberEntry;
	29	class TGColorSelect;
	30	class TGTextEntry;
	31	class TTree;
	32
	33	class TEveGValuator;
	34
	35	class AliEveTRDChamber;
	36	class AliEveTRDLoaderManager;
	37
	38	class AliEveTRDLoader : public TEveElementList
	39	{
	40	friend class AliEveTRDLoaderEditor;
	41
	42	public:
	43	enum TRDDataTypes {
	44	kTRDHits = BIT(0),
	45	kTRDDigits = BIT(1),
	46	kTRDClusters = BIT(2),
	47	kTRDTracklets= BIT(3),
	48	kTRDRawRoot = BIT(4),
	49	kTRDRawDate = BIT(5)
	50	};
	51
	52	AliEveTRDLoader(const Text_t* n="AliEveTRDLoader", const Text_t* t=0);
	53	virtual ~AliEveTRDLoader() {}
	54
	55	virtual void AddChambers(int sm=-1, int stk=-1, int ly=-1);
	56	virtual AliEveTRDChamber* GetChamber(int d);
	57	virtual Int_t GetDataType() const {return fDataType;}
	58	virtual Int_t GetEvent() const {return fEvent;}
	59	virtual Bool_t GoToEvent(int ev);
	60	Bool_t IsDataLinked() const {return TestBit(1);}
	61
	62	virtual Bool_t Open(const char file, const char dir = ".");
	63	inline virtual Bool_t NextEvent(Bool_t rewind = kTRUE);
	64	virtual void Paint(Option_t *option="");
	65
	66	virtual void SetDataType(UChar_t type = 0){fDataType = type;}
	67
	68
	69	protected:
	70	virtual Bool_t LoadHits(TTree *tH);
	71	virtual Bool_t LoadClusters(TTree *tC);
	72	virtual Bool_t LoadDigits(TTree *tD);
	73	virtual Bool_t LoadTracklets(TTree *trklTree);
	74	void SetDataLinked(Bool_t linked = kTRUE) {SetBit(1, linked);}
	75
	76	virtual void Unload();
	77
	78	UChar_t fDataType; // data type
	79	//Bool_t fLoadHits, fLoadDigits, fLoadClusters, fLoadTracks; // flags for data-loading
	80	Char_t fSM, fStack, fLy; // supermodule, stack, layer
	81	Short_t fEvent; // current event to be displayed
	82	AliTRDgeometry *fGeo; // the TRD geometry
	83	TString fFilename; // name of data file
	84	TString fDir; // data directory
	85
	86	private:
	87	AliEveTRDLoader(const AliEveTRDLoader&); // Not implemented
	88	AliEveTRDLoader& operator=(const AliEveTRDLoader&); // Not implemented
	89
	90	ClassDef(AliEveTRDLoader, 0); // Alieve Loader class for the TRD detector.
	91	};
	92
	93	Bool_t AliEveTRDLoader::NextEvent(Bool_t)
	94	{
	95	fEvent++;
	96	return GoToEvent(fEvent);
	97	}
	98
	99	class AliEveTRDLoaderEditor : public TGedFrame
	100	{
	101	public:
	102	AliEveTRDLoaderEditor(const TGWindow* p=0, Int_t width=170, Int_t height=30,
	103	UInt_t options=kChildFrame, Pixel_t back=GetDefaultFrameBackground());
	104	virtual ~AliEveTRDLoaderEditor() {}
	105
	106	virtual void AddChambers();
	107	virtual void FileOpen();
	108	virtual void GoTo();
	109	virtual void Next();
	110	virtual void SetEvent(Double_t ev){fM->fEvent = (Int_t)ev;}
	111	virtual void SetModel(TObject* obj);
	112
	113	protected:
	114	AliEveTRDLoader *fM; // Model object.
	115	TGTextEntry *fFile; // File name weed.
	116	TGTextButton *fBrowse; // browse button
	117	TEveGValuator *fEvent; // Event no weed.
	118	TEveGValuator fSMNumber, fStackNumber, *fPlaneNumber; // Detector id weeds.
	119
	120	private:
	121	AliEveTRDLoaderEditor(const AliEveTRDLoaderEditor&); // Not implemented
	122	AliEveTRDLoaderEditor& operator=(const AliEveTRDLoaderEditor&); // Not implemented
	123
	124	ClassDef(AliEveTRDLoaderEditor, 0); // Editor for AliEveTRDLoader.
	125	};
	126
	127	#endif