[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEdebugTreeTaskAOD.h

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/
//
// Debug tree to look at the distribution of the variable we are cutting on
//
//
#ifndef ALIHFEDEBUGTREETASKAOD_H
#define ALIHFEDEBUGTREETASKAOD_H

#include "AliAnalysisTaskSE.h"

class AliHFEcuts;
class TString;
class TTreeSRedirector;
class AliHFEpidTPC;
class AliAODMCHeader;
class TClonesArray;
class AliHFEsignalCuts;
class AliHFEextraCuts;

class AliHFEdebugTreeTaskAOD : public AliAnalysisTaskSE{
  public:
    AliHFEdebugTreeTaskAOD();
    AliHFEdebugTreeTaskAOD(const char *name);
    virtual ~AliHFEdebugTreeTaskAOD();

    virtual void UserCreateOutputObjects();
    virtual void UserExec(Option_t *);
    virtual void Terminate(Option_t *);

    // Setters for cuts
    void SetFileName(const char *filename);
    void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
    void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
    void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
    AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
    
  private:
    AliHFEdebugTreeTaskAOD(const AliHFEdebugTreeTaskAOD &);
    AliHFEdebugTreeTaskAOD &operator=(const AliHFEdebugTreeTaskAOD &);

    AliAODMCHeader *fAODMCHeader;     // ! MC info AOD
    TClonesArray *fAODArrayMCInfo;    // ! MC info particle AOD
    AliHFEcuts *fTrackCuts;           // Track
    AliHFEextraCuts *fExtraCuts;      // HFE IP info
    AliHFEsignalCuts *fSignalCuts;    // Signal Cuts
    AliHFEpidTPC *fTPCpid;            // TPC PID
    Int_t fEventNumber;               // Event Number
    Int_t fNclustersTPC;              // Min Number of clusters in TPC
    Int_t fNclustersTPCPID;           // Min Number of clusters for TPC PID
    Int_t fNclustersITS;              // Min Number of clusters in ITS
    TString fFilename;                // file name for the debug tree
    TTreeSRedirector *fDebugTree;     // Debug Tree
  
    ClassDef(AliHFEdebugTreeTaskAOD, 1)
};
#endif
Commit	Line	Data
76d0b522	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	//
	16	// Debug tree to look at the distribution of the variable we are cutting on
	17	//
	18	//
	19	#ifndef ALIHFEDEBUGTREETASKAOD_H
	20	#define ALIHFEDEBUGTREETASKAOD_H
	21
	22	#include "AliAnalysisTaskSE.h"
	23
	24	class AliHFEcuts;
	25	class TString;
	26	class TTreeSRedirector;
	27	class AliHFEpidTPC;
	28	class AliAODMCHeader;
	29	class TClonesArray;
	30	class AliHFEsignalCuts;
	31	class AliHFEextraCuts;
	32
	33	class AliHFEdebugTreeTaskAOD : public AliAnalysisTaskSE{
	34	public:
	35	AliHFEdebugTreeTaskAOD();
	36	AliHFEdebugTreeTaskAOD(const char *name);
	37	virtual ~AliHFEdebugTreeTaskAOD();
	38
	39	virtual void UserCreateOutputObjects();
	40	virtual void UserExec(Option_t *);
	41	virtual void Terminate(Option_t *);
	42
	43	// Setters for cuts
	44	void SetFileName(const char *filename);
	45	void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
	46	void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
	47	void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
	48	AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
	49
	50	private:
	51	AliHFEdebugTreeTaskAOD(const AliHFEdebugTreeTaskAOD &);
	52	AliHFEdebugTreeTaskAOD &operator=(const AliHFEdebugTreeTaskAOD &);
	53
	54	AliAODMCHeader *fAODMCHeader; // ! MC info AOD
	55	TClonesArray *fAODArrayMCInfo; // ! MC info particle AOD
	56	AliHFEcuts *fTrackCuts; // Track
	57	AliHFEextraCuts *fExtraCuts; // HFE IP info
	58	AliHFEsignalCuts *fSignalCuts; // Signal Cuts
	59	AliHFEpidTPC *fTPCpid; // TPC PID
	60	Int_t fEventNumber; // Event Number
	61	Int_t fNclustersTPC; // Min Number of clusters in TPC
	62	Int_t fNclustersTPCPID; // Min Number of clusters for TPC PID
	63	Int_t fNclustersITS; // Min Number of clusters in ITS
	64	TString fFilename; // file name for the debug tree
65	TTreeSRedirector *fDebugTree; // Debug Tree
66
959ea9d8	67	ClassDef(AliHFEdebugTreeTaskAOD, 1)
76d0b522	68	};
	69	#endif
	70