[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpostAnalysis.h

#ifndef ALIHFEPOSTANALYSIS_H
#define ALIHFEPOSTANALYSIS_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.                  *
**************************************************************************/

/* $Id$ */ 

//
// Post analysis class
// Creating results and draw pictures
// Called in AliAnalysisTaskHFE::Terminate or in a macro
//
#ifndef ROOT_THnSparse
#include <THnSparse.h>
#endif

class AliHFEcontainer;
class TH1;
class TList;

class AliHFEpostAnalysis : public TObject{
  public:
    AliHFEpostAnalysis();
    AliHFEpostAnalysis(const AliHFEpostAnalysis &ref);
    AliHFEpostAnalysis &operator=(const AliHFEpostAnalysis &ref);
    ~AliHFEpostAnalysis();

    Int_t SetTaskResults(AliHFEcontainer *trackContainer) { fEfficiencyContainer = trackContainer; return 1; };
    Int_t SetTaskQA(const TList *qa);
    void StoreOutput(const char *filename = "HFEresults.root");

    void DrawMCSignal2Background();
    void DrawEfficiency();
    void DrawPIDperformance();
    void DrawCutEfficiency(Bool_t MC = kTRUE, Int_t source = -1);
  private:
    enum{
      kCFC,
      kPIDperf,
      kSigBackg
    };
    TH1 *CreateHistoSignalToBackgroundMC(Int_t mode, Int_t charge);
    TH1 *CreateHistoPIDperformance(Int_t mode, Int_t charge);

    TList *fResults;                          // Container for output objects
    UChar_t fAnalysisObjects;                       // S
    AliHFEcontainer *fEfficiencyContainer;     // Task Results
    THnSparseF *fPIDperformance;              // PID Performance Studies
    THnSparseF *fSignalToBackgroundMC;        // Signal To Background Studies

    ClassDef(AliHFEpostAnalysis, 1)           // Result Creator class
};

#endif
Commit	Line	Data
27de2dfb	1	#ifndef ALIHFEPOSTANALYSIS_H
	2	#define ALIHFEPOSTANALYSIS_H
	3
d2af20c5	4	/**************************************************************************
	5	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	6	* *
	7	* Author: The ALICE Off-line Project. *
	8	* Contributors are mentioned in the code where appropriate. *
	9	* *
	10	* Permission to use, copy, modify and distribute this software and its *
	11	* documentation strictly for non-commercial purposes is hereby granted *
	12	* without fee, provided that the above copyright notice appears in all *
	13	* copies and that both the copyright notice and this permission notice *
	14	* appear in the supporting documentation. The authors make no claims *
	15	* about the suitability of this software for any purpose. It is *
	16	* provided "as is" without express or implied warranty. *
	17	**************************************************************************/
27de2dfb	18
	19	/* $Id$ */
	20
d2af20c5	21	//
	22	// Post analysis class
	23	// Creating results and draw pictures
	24	// Called in AliAnalysisTaskHFE::Terminate or in a macro
	25	//
d2af20c5	26	#ifndef ROOT_THnSparse
	27	#include <THnSparse.h>
	28	#endif
	29
3a72645a	30	class AliHFEcontainer;
70da6c5a	31	class TH1;
d2af20c5	32	class TList;
3a72645a	33
d2af20c5	34	class AliHFEpostAnalysis : public TObject{
	35	public:
	36	AliHFEpostAnalysis();
	37	AliHFEpostAnalysis(const AliHFEpostAnalysis &ref);
	38	AliHFEpostAnalysis &operator=(const AliHFEpostAnalysis &ref);
	39	~AliHFEpostAnalysis();
	40
3a72645a	41	Int_t SetTaskResults(AliHFEcontainer *trackContainer) { fEfficiencyContainer = trackContainer; return 1; };
3a72645a	42	Int_t SetTaskQA(const TList *qa);
d2af20c5	43	void StoreOutput(const char *filename = "HFEresults.root");
	44
	45	void DrawMCSignal2Background();
	46	void DrawEfficiency();
	47	void DrawPIDperformance();
67fe7bd0	48	void DrawCutEfficiency(Bool_t MC = kTRUE, Int_t source = -1);
d2af20c5	49	private:
	50	enum{
	51	kCFC,
	52	kPIDperf,
	53	kSigBackg
	54	};
70da6c5a	55	TH1 *CreateHistoSignalToBackgroundMC(Int_t mode, Int_t charge);
	56	TH1 *CreateHistoPIDperformance(Int_t mode, Int_t charge);
	57
d2af20c5	58	TList *fResults; // Container for output objects
d2af20c5	59	UChar_t fAnalysisObjects; // S
3a72645a	60	AliHFEcontainer *fEfficiencyContainer; // Task Results
d2af20c5	61	THnSparseF *fPIDperformance; // PID Performance Studies
	62	THnSparseF *fSignalToBackgroundMC; // Signal To Background Studies
	63
	64	ClassDef(AliHFEpostAnalysis, 1) // Result Creator class
	65	};
	66
	67	#endif