[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEInclusiveSpectrumQA.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.                  *
**************************************************************************/
//
// Class for spectrum correction
// Subtraction of hadronic background, Unfolding of the data and
// Renormalization done here
// For more information see the implementation file
//
#ifndef ALIHFEINCLUSIVESPECTRUMQA_H
#define ALIHFEINCLUSIVESPECTRUMQA_H

#ifndef ROOT_TNamed
#include <TNamed.h>
#endif


class TObjArray;
class TGraphErrors;
class TObject;


class AliHFEInclusiveSpectrumQA : public TNamed{
 public:

  enum Results_t{
    kDataProjection = 0,
    kCMProjection = 1,
    kBeforeSC = 2,
    kAfterSC = 3,
    kBeforeV0 = 4,
    kAfterV0 = 5,
    kV0Efficiency = 6,
    kBeforePE = 7,
    kAfterPE = 8,
    kPEfficiency = 9,
    kBeforeMCE = 10,
    kAfterMCE = 11,
    kMCEfficiency = 12,
    kBeforeU = 13,
    kAfterU = 14,
    kResidualU = 15,
    kUEfficiency = 16,
    kFinalResultUnfolded = 17,
    kFinalResultDirectEfficiency = 18,
    kBeforeSPB = 19,
    kAfterSPB = 20,
    kBeforeSCND = 21,
    kAfterSCND = 22,
    kHadronContaminationND = 23,
    kBeforeV0ND = 24,
    kAfterV0ND = 25,
    kV0EfficiencyND = 26,
    kBeforePEND = 27,
    kAfterPEND = 28,
    kPEfficiencyND = 29,
    kBeforeMCEND = 30,
    kAfterMCEND = 31,
    kMCEfficiencyND = 32,
    kNResults = 33
  };


  enum EfficiencyCorrection_t{
    kV0 = 0,
    kMC = 1,
    kParametrized = 2,
    kNTypeEfficiency = 3
  };
   
  void AddResultAt(TObject *obj,Int_t index);
  TObject *GetResult(Int_t index);
  
  void DrawProjections() const;
  void DrawSubtractContamination() const;
  void DrawSubtractContaminationND() const;
  void DrawSubtractPhotonicBackground() const;
  void DrawCorrectWithEfficiency(Int_t typeeff) const;
  void DrawCorrectWithEfficiencyND(Int_t typeeff) const;
  void DrawUnfolding() const;
  void DrawResult();
  
  void SetStyle() const;
  void SetWriteToFile(Bool_t writetofile) {fWriteToFile=writetofile; };
  void SetPtMax(Double_t ptmax) {fPtMax = ptmax; };


  TH1D *DivideSpectra(TGraphErrors *ga, TGraphErrors *gb);
  

  AliHFEInclusiveSpectrumQA();
  AliHFEInclusiveSpectrumQA(const char* name);
  ~AliHFEInclusiveSpectrumQA();
  
  protected:
  
  private:

  static const Char_t* fgkNameCanvas[kNTypeEfficiency];     // Name of canvas
  static const Char_t* fgkNameCanvasND[kNTypeEfficiency];     // Name of canvas

    AliHFEInclusiveSpectrumQA(const AliHFEInclusiveSpectrumQA &);
    AliHFEInclusiveSpectrumQA &operator=(const AliHFEInclusiveSpectrumQA &);
 
    Double_t   fPtMax;             // Pt max to plot
    TObjArray *fListOfResult;     // ListOfResults
    Bool_t fWriteToFile;           // Write plots to eps files
   
    ClassDef(AliHFEInclusiveSpectrumQA, 1) 
};
#endif
Commit	Line	Data
959ea9d8	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	// Class for spectrum correction
	17	// Subtraction of hadronic background, Unfolding of the data and
	18	// Renormalization done here
	19	// For more information see the implementation file
	20	//
	21	#ifndef ALIHFEINCLUSIVESPECTRUMQA_H
	22	#define ALIHFEINCLUSIVESPECTRUMQA_H
	23
	24	#ifndef ROOT_TNamed
	25	#include <TNamed.h>
	26	#endif
	27
	28
	29	class TObjArray;
	30	class TGraphErrors;
	31	class TObject;
	32
	33
	34	class AliHFEInclusiveSpectrumQA : public TNamed{
	35	public:
	36
	37	enum Results_t{
	38	kDataProjection = 0,
	39	kCMProjection = 1,
	40	kBeforeSC = 2,
	41	kAfterSC = 3,
	42	kBeforeV0 = 4,
	43	kAfterV0 = 5,
	44	kV0Efficiency = 6,
	45	kBeforePE = 7,
	46	kAfterPE = 8,
	47	kPEfficiency = 9,
	48	kBeforeMCE = 10,
	49	kAfterMCE = 11,
	50	kMCEfficiency = 12,
	51	kBeforeU = 13,
	52	kAfterU = 14,
	53	kResidualU = 15,
	54	kUEfficiency = 16,
	55	kFinalResultUnfolded = 17,
	56	kFinalResultDirectEfficiency = 18,
63bdf450	57	kBeforeSPB = 19,
63bdf450	58	kAfterSPB = 20,
ff8249bd	59	kBeforeSCND = 21,
	60	kAfterSCND = 22,
	61	kHadronContaminationND = 23,
	62	kBeforeV0ND = 24,
	63	kAfterV0ND = 25,
	64	kV0EfficiencyND = 26,
	65	kBeforePEND = 27,
	66	kAfterPEND = 28,
	67	kPEfficiencyND = 29,
	68	kBeforeMCEND = 30,
	69	kAfterMCEND = 31,
	70	kMCEfficiencyND = 32,
	71	kNResults = 33
959ea9d8	72	};
	73
	74
	75	enum EfficiencyCorrection_t{
	76	kV0 = 0,
	77	kMC = 1,
	78	kParametrized = 2,
	79	kNTypeEfficiency = 3
	80	};
	81
	82	void AddResultAt(TObject *obj,Int_t index);
	83	TObject *GetResult(Int_t index);
	84
	85	void DrawProjections() const;
	86	void DrawSubtractContamination() const;
ff8249bd	87	void DrawSubtractContaminationND() const;
63bdf450	88	void DrawSubtractPhotonicBackground() const;
959ea9d8	89	void DrawCorrectWithEfficiency(Int_t typeeff) const;
ff8249bd	90	void DrawCorrectWithEfficiencyND(Int_t typeeff) const;
959ea9d8	91	void DrawUnfolding() const;
	92	void DrawResult();
	93
	94	void SetStyle() const;
	95	void SetWriteToFile(Bool_t writetofile) {fWriteToFile=writetofile; };
	96	void SetPtMax(Double_t ptmax) {fPtMax = ptmax; };
	97
	98
	99	TH1D DivideSpectra(TGraphErrors ga, TGraphErrors *gb);
	100
	101
	102	AliHFEInclusiveSpectrumQA();
	103	AliHFEInclusiveSpectrumQA(const char* name);
	104	~AliHFEInclusiveSpectrumQA();
	105
	106	protected:
	107
	108	private:
	109
	110	static const Char_t* fgkNameCanvas[kNTypeEfficiency]; // Name of canvas
ff8249bd	111	static const Char_t* fgkNameCanvasND[kNTypeEfficiency]; // Name of canvas
959ea9d8	112
	113	AliHFEInclusiveSpectrumQA(const AliHFEInclusiveSpectrumQA &);
	114	AliHFEInclusiveSpectrumQA &operator=(const AliHFEInclusiveSpectrumQA &);
	115
	116	Double_t fPtMax; // Pt max to plot
	117	TObjArray *fListOfResult; // ListOfResults
	118	Bool_t fWriteToFile; // Write plots to eps files
	119
	120	ClassDef(AliHFEInclusiveSpectrumQA, 1)
	121	};
	122	#endif
	123