[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronDebugTree.h

#ifndef ALIDIELECTRONDEBUGTREE_H
#define ALIDIELECTRONDEBUGTREE_H

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#############################################################
//#                                                           # 
//#         Class AliDielectronDebugTree                     #
//#                                                           #
//#  Authors:                                                 #
//#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
//#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
//#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
//#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
//#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
//#                                                           #
//#############################################################

#include <TNamed.h>
#include <TString.h>

#include "AliDielectronVarManager.h"

class TTreeSRedirector;
class AliDielectronPair;
class AliDielectron;

class AliDielectronDebugTree : public TNamed {
public:
  AliDielectronDebugTree();
  AliDielectronDebugTree(const char*name, const char* title);

  virtual ~AliDielectronDebugTree();

  void SetOutputFileName(const char* file) { fFileName=file; }
  
  void AddPairVariable(AliDielectronVarManager::ValueTypes type) { fVariables[fNVars++]=(Int_t)type; }
  void AddLegVariable(AliDielectronVarManager::ValueTypes type)  { fVariablesLeg[fNVarsLeg++]=(Int_t)type; }
  
  void Fill(AliDielectronPair *pair);

  void SetDielectron(AliDielectron * const dielectron) { fDielectron=dielectron; }
  
  void DeleteStreamer();
  void WriteTree();
private:
  TString fFileName;                                          //output file name
  
  Int_t  fNVars;                                              //number of configured variables
  Int_t  fVariables[AliDielectronVarManager::kNMaxValues];    //configured variables
  Int_t  fNVarsLeg;                                           //number of configured variables
  Int_t  fVariablesLeg[AliDielectronVarManager::kNMaxValues]; //configured variables for the legs

  TTreeSRedirector *fStreamer;     //! Tree Redirector
  AliDielectron *fDielectron;      //! pointer to mother dielectron manager

  AliDielectronDebugTree(const AliDielectronDebugTree &c);
  AliDielectronDebugTree &operator=(const AliDielectronDebugTree &c);

  
  ClassDef(AliDielectronDebugTree,1)         // Dielectron DebugTree
};


#endif
Commit	Line	Data
572b0139	1	#ifndef ALIDIELECTRONDEBUGTREE_H
	2	#define ALIDIELECTRONDEBUGTREE_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 AliDielectronDebugTree #
	10	//# #
	11	//# Authors: #
	12	//# Anton Andronic, GSI / A.Andronic@gsi.de #
	13	//# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de #
	14	//# Julian Book, Uni Ffm / Julian.Book@cern.ch #
	15	//# Frederick Kramer, Uni Ffm, / Frederick.Kramer@cern.ch #
	16	//# Magnus Mager, CERN / Magnus.Mager@cern.ch #
	17	//# WooJin J. Park, GSI / W.J.Park@gsi.de #
	18	//# Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
	19	//# #
	20	//#############################################################
	21
	22	#include <TNamed.h>
	23	#include <TString.h>
	24
	25	#include "AliDielectronVarManager.h"
	26
	27	class TTreeSRedirector;
	28	class AliDielectronPair;
8df8e382	29	class AliDielectron;
572b0139	30
	31	class AliDielectronDebugTree : public TNamed {
	32	public:
	33	AliDielectronDebugTree();
	34	AliDielectronDebugTree(const charname, const char title);
	35
	36	virtual ~AliDielectronDebugTree();
	37
	38	void SetOutputFileName(const char* file) { fFileName=file; }
	39
	40	void AddPairVariable(AliDielectronVarManager::ValueTypes type) { fVariables[fNVars++]=(Int_t)type; }
	41	void AddLegVariable(AliDielectronVarManager::ValueTypes type) { fVariablesLeg[fNVarsLeg++]=(Int_t)type; }
	42
	43	void Fill(AliDielectronPair *pair);
	44
8df8e382	45	void SetDielectron(AliDielectron * const dielectron) { fDielectron=dielectron; }
8df8e382	46
572b0139	47	void DeleteStreamer();
	48	void WriteTree();
	49	private:
	50	TString fFileName; //output file name
	51
	52	Int_t fNVars; //number of configured variables
	53	Int_t fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
	54	Int_t fNVarsLeg; //number of configured variables
	55	Int_t fVariablesLeg[AliDielectronVarManager::kNMaxValues]; //configured variables for the legs
	56
	57	TTreeSRedirector *fStreamer; //! Tree Redirector
8df8e382	58	AliDielectron *fDielectron; //! pointer to mother dielectron manager
8df8e382	59
572b0139	60	AliDielectronDebugTree(const AliDielectronDebugTree &c);
	61	AliDielectronDebugTree &operator=(const AliDielectronDebugTree &c);
	62
	63
	64	ClassDef(AliDielectronDebugTree,1) // Dielectron DebugTree
	65	};
	66
	67
	68
	69	#endif