[u/mrichter/AliRoot.git] / PWGCF / Correlations / JCORRAN / AliJEfficiency.h

#ifndef ALIJEFFICIENCY_H
#define ALIJEFFICIENCY_H
// Class to Load and Get efficiency inforamtion
// ..
// TODO

#include <TString.h>
#include <TFile.h>
#include "AliJTrackCut.h"
#include "AliJRunTable.h"
#include "AliJConst.h"
#include <TGraphErrors.h>
#include <TAxis.h>
#include <iostream>
using namespace std;

class AliJEfficiency{
    public:
        enum Mode { kNotUse, kPeriod, kRunNumber, kAuto };
        enum Type { kRE, kMC, kMerge };
        AliJEfficiency();
        AliJEfficiency(const AliJEfficiency& obj);
        AliJEfficiency& operator=(const AliJEfficiency& obj);

        void SetMode( int i ){ fMode = i; }

        void SetDataPath(TString s ){ fDataPath=s; }
        void SetEffFile(TString s ){ fInputRootName=s; }
        void SetName(TString s ){ fName=s; }
        void SetPeriod(int period){ fPeriod = period; }
        void SetPeriod(TString s){ fPeriodStr = s; }
        void SetMCPeriod(TString s){ fMCPeriodStr = s; }
        void SetRunNumber( Long64_t runnum ){ fRunNumber=runnum; }
        void SetTag(TString s){ fTag=s; }

        TString GetName() const { return fName; }
        double GetCorrection( double pt, int icut, double cent ) const ;
        TString GetEffName() ;
        TString GetEffFullName() ;
        bool   Load();
        void   PrintOut() const {
            cout<<fInputRootName<<endl;
        }
        void Write();

    private:
        int      fMode;             // Mode. see enum Mode
        int      fPeriod;           // Data Period index
        AliJTrackCut fTrackCut;     // Track Cut Object. TODO:why not pointer?
        AliJRunTable fRunTable;     // run Table. TODO:why not pointer?

        TString fDataPath;          // locaction of eff files
        TString fName;              // name of efficiency. usually empty
        TString fPeriodStr;         // DATA period
        TString fMCPeriodStr;       // MC period
        Long64_t fRunNumber;        // Runnumber
        TString fTag;               // Tags to distinguish special eff file
        TString fInputRootName;     // name of input

        TFile * fInputRoot;         // input file  
        TDirectory * fEffDir[3];    // root directory of efficiency. only second item of fEffDir with "Efficiency" is being used.
        TGraphErrors * fCorrection[20][20][20]; // Storage of Correction factor 
        TAxis * fCentBin;     // Bin of Centrality. replace with AliJBin?
};
#endif
Commit	Line	Data
9dc4f671	1	#ifndef ALIJEFFICIENCY_H
	2	#define ALIJEFFICIENCY_H
	3	// Class to Load and Get efficiency inforamtion
	4	// ..
	5	// TODO
	6
	7	#include <TString.h>
	8	#include <TFile.h>
	9	#include "AliJTrackCut.h"
	10	#include "AliJRunTable.h"
	11	#include "AliJConst.h"
	12	#include <TGraphErrors.h>
	13	#include <TAxis.h>
	14	#include <iostream>
	15	using namespace std;
	16
	17	class AliJEfficiency{
	18	public:
	19	enum Mode { kNotUse, kPeriod, kRunNumber, kAuto };
	20	enum Type { kRE, kMC, kMerge };
	21	AliJEfficiency();
	22	AliJEfficiency(const AliJEfficiency& obj);
	23	AliJEfficiency& operator=(const AliJEfficiency& obj);
	24
	25	void SetMode( int i ){ fMode = i; }
	26
	27	void SetDataPath(TString s ){ fDataPath=s; }
	28	void SetEffFile(TString s ){ fInputRootName=s; }
	29	void SetName(TString s ){ fName=s; }
	30	void SetPeriod(int period){ fPeriod = period; }
	31	void SetPeriod(TString s){ fPeriodStr = s; }
	32	void SetMCPeriod(TString s){ fMCPeriodStr = s; }
	33	void SetRunNumber( Long64_t runnum ){ fRunNumber=runnum; }
	34	void SetTag(TString s){ fTag=s; }
	35
	36	TString GetName() const { return fName; }
	37	double GetCorrection( double pt, int icut, double cent ) const ;
	38	TString GetEffName() ;
	39	TString GetEffFullName() ;
	40	bool Load();
	41	void PrintOut() const {
	42	cout<<fInputRootName<<endl;
	43	}
	44	void Write();
	45
	46	private:
	47	int fMode; // Mode. see enum Mode
	48	int fPeriod; // Data Period index
	49	AliJTrackCut fTrackCut; // Track Cut Object. TODO:why not pointer?
	50	AliJRunTable fRunTable; // run Table. TODO:why not pointer?
	51
	52	TString fDataPath; // locaction of eff files
	53	TString fName; // name of efficiency. usually empty
	54	TString fPeriodStr; // DATA period
	55	TString fMCPeriodStr; // MC period
	56	Long64_t fRunNumber; // Runnumber
	57	TString fTag; // Tags to distinguish special eff file
	58	TString fInputRootName; // name of input
	59
	60	TFile * fInputRoot; // input file
bc514abb	61	TDirectory * fEffDir[3]; // root directory of efficiency. only second item of fEffDir with "Efficiency" is being used.
9dc4f671	62	TGraphErrors * fCorrection[20][20][20]; // Storage of Correction factor
	63	TAxis * fCentBin; // Bin of Centrality. replace with AliJBin?
	64	};
	65	#endif