[u/mrichter/AliRoot.git] / TPC / AliTPCCalibViewer.h

#ifndef ALITPCCALIBVIEWER_H
#define ALITPCCALIBVIEWER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliTPCCalibViewer.h,v */

///////////////////////////////////////////////////
//                                               //
//  TPC calibration viewer/visualization class   //
//  use Tree for visualization                   //
///////////////////////////////////////////////////

#include <TObject.h>
#include <TTree.h>
#include <TFile.h>
#include "AliTPCCalPad.h"
#include "AliTPCCalROC.h"
#include "TFriendElement.h"

#include "AliMathBase.h"
class TLegend;
class TGraph;


class AliTPCCalibViewer : public TObject {
public:
   AliTPCCalibViewer();
   AliTPCCalibViewer(const AliTPCCalibViewer &c);
   AliTPCCalibViewer(TTree* tree);
   AliTPCCalibViewer(char* fileName, char* treeName = "calPads");
   AliTPCCalibViewer &operator = (const AliTPCCalibViewer & param);
   virtual ~AliTPCCalibViewer();
   virtual void Delete(Option_t* option = "");
   
   TString GetAbbreviation() const { return fAbbreviation;  }
   TString GetAppendString() const { return fAppendString; }
   void SetAbbreviation(char* abr) { fAbbreviation = abr; }
   void SetAppendString(char* str) { fAppendString = str; }
   
   virtual void     Draw(Option_t* opt="") { fTree->Draw(opt); }
   virtual Long64_t Draw(const char* varexp, const TCut& selection, Option_t* option = "", Long64_t nentries = 1000000000, Long64_t firstentry = 0) { return fTree->Draw(varexp, selection, option, nentries, firstentry); };
   virtual Long64_t Draw(const char* varexp, const char* selection, Option_t* option = "", Long64_t nentries = 1000000000, Long64_t firstentry = 0) { return fTree->Draw(varexp, selection, option, nentries, firstentry); };

   Int_t EasyDraw(const char* drawCommand, const char* sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const;   // easy drawing of data, use '~' for abbreviation of '.fElements'
   Int_t EasyDraw(const char* drawCommand, Int_t sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const;   // easy drawing of data, use '~' for abbreviation of '.fElements'
   Int_t EasyDraw1D(const char* drawCommand, const char* sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const;   // easy drawing of data, use '~' for abbreviation of '.fElements'
   Int_t EasyDraw1D(const char* drawCommand, Int_t sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const;   // easy drawing of data, use '~' for abbreviation of '.fElements'
   void FormatHistoLabels(TH1 *histo) const; // formats title and axis labels of histo, removes '.fElements'
   
   Int_t  DrawHisto1D(const char* drawCommand,       Int_t sector, const char* cuts = 0, const char *sigmas = "2;4;6", Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE) const; // draws 1d histograms and superimposes mean, median, ltm and several sigma cuts
   Int_t  DrawHisto1D(const char* drawCommand, const char* sector, const char* cuts = 0, const char *sigmas = "2;4;6", Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE) const; // draws 1d histograms and superimposes mean, median, ltm and several sigma cuts
   Int_t     SigmaCut(const char* drawCommand,       Int_t sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws fraction of used pads over different sigma cuts
   Int_t  SigmaCutNew(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws fraction of used pads over different sigma cuts
   Int_t     SigmaCut(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws fraction of used pads over different sigma cuts
   Int_t    Integrate(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws an integrated histogram
   Int_t    Integrate(const char* drawCommand,       Int_t sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws an integrated histogram
   Int_t IntegrateOld(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const;    // draws an integrated histogram
   
   AliTPCCalPad* GetCalPad(const char* desiredData, char* cuts = "", char* calPadName = "NoName") const;     // returns an AliTPCCalPad object containing the specified data with cuts applied
   AliTPCCalROC* GetCalROC(const char* desiredData, UInt_t sector, char* cuts = "") const;  // returns an AliTPCCalROC object containing the specified data for sector with cuts applied
   
   TObjArray* GetArrayOfCalPads();
   TObjArray* GetListOfVariables(Bool_t printList = kFALSE);
   TObjArray* GetListOfNormalizationVariables(Bool_t printList = kFALSE) const;
   
   static void MakeTreeWithObjects(const char * fileName, TObjArray * array, const char * mapFileName = 0);
   static void MakeTree(const char * fileName, TObjArray * array, const char * mapFileName = 0, AliTPCCalPad* outlierPad = 0, Float_t ltmFraction = 0.9);
   static void MakeTree(const char *outPutFileName, const Char_t *inputFileName, AliTPCCalPad *outlierPad = 0, Float_t ltmFraction = 0.9, const char *mapFileName = "$ALICE_ROOT/TPC/Calib/MapCalibrationObjects.root");
   static void CreateObjectList(const Char_t *filename, TObjArray *calibObjects);
   
   TFriendElement* AddReferenceTree(const char* filename, const char* treename = "calPads", const char* refname = "R");
   TFriendElement* AddFriend(const char* treename, const char* filename) {return fTree->AddFriend(treename, filename);};
   TFriendElement* AddFriend(TTree* tree, const char* alias, Bool_t warn=kFALSE) {return fTree->AddFriend(tree, alias, warn);};
   TFriendElement* AddFriend(const char* treename, TFile* file) {return fTree->AddFriend(treename, file);};
   TTree * GetTree() { return fTree;}

   TString* Fit(const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, TVectorD &fitParam, TMatrixD &covMatrix);

  // 
  // Array tools
  // 
  static Double_t GetLTM(Int_t n, Double_t *array, Double_t *sigma = 0, Double_t fraction = 0.9);
  static Int_t GetBin(Float_t value, Int_t nbins, Double_t binLow, Double_t binUp);
  static TH1F* SigmaCut(Int_t n, Float_t *array, Float_t mean, Float_t sigma, Int_t nbins, Float_t binLow, Float_t binUp, Float_t sigmaMax, Float_t sigmaStep = -1, Bool_t pm = kFALSE);
  static TH1F* SigmaCut(TH1F *histogram, Float_t mean, Float_t sigma, Float_t sigmaMax, Float_t sigmaStep = -1, Bool_t pm = kFALSE);
  static TH1F* Integrate(TH1F *histogram, Float_t mean = 0, Float_t sigma = 0, Float_t sigmaMax = 0, Float_t sigmaStep = -1);
  static TH1F* Integrate(Int_t n, Float_t *array, Int_t nbins, Float_t binLow, Float_t binUp, Float_t mean = 0, Float_t sigma = 0, Float_t sigmaMax = 0, Float_t sigmaStep = -1);
  
  static TH1F* SigmaCut(Int_t n, Double_t *array, Double_t mean, Double_t sigma, Int_t nbins, Double_t *xbins, Double_t sigmaMax);
   
   
protected:
   TTree* fTree;     // tree containing visualization data (e.g. written by AliTPCCalPad::MakeTree(...)
   TFile* fFile;     // file that contains a calPads tree (e.g. written by AliTPCCalPad::MakeTree(...)
   TObjArray* fListOfObjectsToBeDeleted;  //Objects, that will be deleted when the destructor ist called
   Bool_t fTreeMustBeDeleted;  // decides weather the tree must be deleted in destructor or not 
   TString fAbbreviation;       // the abreviation for '.fElements'
   TString fAppendString;      // '.fElements', stored in a TStrig
   
   void DrawLines(TH1F *cutHistoMean, TVectorF nsigma, TLegend *legend, Int_t color, Bool_t pm) const;
   void DrawLines(TGraph *graph, TVectorF nsigma, TLegend *legend, Int_t color, Bool_t pm) const;
  
   
   ClassDef(AliTPCCalibViewer,1)    //  TPC calibration viewer class
};

#endif
Commit	Line	Data
39bcd65d	1	#ifndef ALITPCCALIBVIEWER_H
	2	#define ALITPCCALIBVIEWER_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliTPCCalibViewer.h,v */
	7
	8	///////////////////////////////////////////////////
	9	// //
	10	// TPC calibration viewer/visualization class //
	11	// use Tree for visualization //
	12	///////////////////////////////////////////////////
	13
	14	#include <TObject.h>
	15	#include <TTree.h>
	16	#include <TFile.h>
	17	#include "AliTPCCalPad.h"
	18	#include "AliTPCCalROC.h"
	19	#include "TFriendElement.h"
	20
72d0ab7e	21	#include "AliMathBase.h"
72d0ab7e	22	class TLegend;
4f3934a1	23	class TGraph;
72d0ab7e	24
39bcd65d	25
	26	class AliTPCCalibViewer : public TObject {
	27	public:
	28	AliTPCCalibViewer();
	29	AliTPCCalibViewer(const AliTPCCalibViewer &c);
	30	AliTPCCalibViewer(TTree* tree);
	31	AliTPCCalibViewer(char* fileName, char* treeName = "calPads");
	32	AliTPCCalibViewer &operator = (const AliTPCCalibViewer & param);
	33	virtual ~AliTPCCalibViewer();
a6d2bd0c	34	virtual void Delete(Option_t* option = "");
39bcd65d	35
832bb12c	36	TString GetAbbreviation() const { return fAbbreviation; }
	37	TString GetAppendString() const { return fAppendString; }
	38	void SetAbbreviation(char* abr) { fAbbreviation = abr; }
	39	void SetAppendString(char* str) { fAppendString = str; }
	40
39bcd65d	41	virtual void Draw(Option_t* opt="") { fTree->Draw(opt); }
	42	virtual Long64_t Draw(const char* varexp, const TCut& selection, Option_t* option = "", Long64_t nentries = 1000000000, Long64_t firstentry = 0) { return fTree->Draw(varexp, selection, option, nentries, firstentry); };
	43	virtual Long64_t Draw(const char* varexp, const char* selection, Option_t* option = "", Long64_t nentries = 1000000000, Long64_t firstentry = 0) { return fTree->Draw(varexp, selection, option, nentries, firstentry); };
	44
	45	Int_t EasyDraw(const char* drawCommand, const char* sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const; // easy drawing of data, use '~' for abbreviation of '.fElements'
	46	Int_t EasyDraw(const char* drawCommand, Int_t sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const; // easy drawing of data, use '~' for abbreviation of '.fElements'
	47	Int_t EasyDraw1D(const char* drawCommand, const char* sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const; // easy drawing of data, use '~' for abbreviation of '.fElements'
	48	Int_t EasyDraw1D(const char* drawCommand, Int_t sector, const char* cuts = 0, const char* drawOptions = 0, Bool_t writeDrawCommand = kFALSE) const; // easy drawing of data, use '~' for abbreviation of '.fElements'
832bb12c	49	void FormatHistoLabels(TH1 *histo) const; // formats title and axis labels of histo, removes '.fElements'
72d0ab7e	50
	51	Int_t DrawHisto1D(const char* drawCommand, Int_t sector, const char* cuts = 0, const char *sigmas = "2;4;6", Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE) const; // draws 1d histograms and superimposes mean, median, ltm and several sigma cuts
	52	Int_t DrawHisto1D(const char* drawCommand, const char* sector, const char* cuts = 0, const char *sigmas = "2;4;6", Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE) const; // draws 1d histograms and superimposes mean, median, ltm and several sigma cuts
	53	Int_t SigmaCut(const char* drawCommand, Int_t sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws fraction of used pads over different sigma cuts
4f3934a1	54	Int_t SigmaCutNew(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws fraction of used pads over different sigma cuts
72d0ab7e	55	Int_t SigmaCut(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, Bool_t pm = kFALSE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws fraction of used pads over different sigma cuts
	56	Int_t Integrate(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws an integrated histogram
	57	Int_t Integrate(const char* drawCommand, Int_t sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws an integrated histogram
4f3934a1	58	Int_t IntegrateOld(const char* drawCommand, const char* sector, const char* cuts = 0, Float_t sigmaMax = 5, Bool_t plotMean = kTRUE, Bool_t plotMedian = kTRUE, Bool_t plotLTM = kTRUE, const char *sigmas = "", Float_t sigmaStep = -1) const; // draws an integrated histogram
39bcd65d	59
	60	AliTPCCalPad* GetCalPad(const char* desiredData, char* cuts = "", char* calPadName = "NoName") const; // returns an AliTPCCalPad object containing the specified data with cuts applied
	61	AliTPCCalROC* GetCalROC(const char* desiredData, UInt_t sector, char* cuts = "") const; // returns an AliTPCCalROC object containing the specified data for sector with cuts applied
	62
	63	TObjArray* GetArrayOfCalPads();
	64	TObjArray* GetListOfVariables(Bool_t printList = kFALSE);
72d0ab7e	65	TObjArray* GetListOfNormalizationVariables(Bool_t printList = kFALSE) const;
39bcd65d	66
	67	static void MakeTreeWithObjects(const char * fileName, TObjArray * array, const char * mapFileName = 0);
	68	static void MakeTree(const char * fileName, TObjArray * array, const char * mapFileName = 0, AliTPCCalPad* outlierPad = 0, Float_t ltmFraction = 0.9);
4f3934a1	69	static void MakeTree(const char outPutFileName, const Char_t inputFileName, AliTPCCalPad outlierPad = 0, Float_t ltmFraction = 0.9, const char mapFileName = "$ALICE_ROOT/TPC/Calib/MapCalibrationObjects.root");
4f3934a1	70	static void CreateObjectList(const Char_t filename, TObjArray calibObjects);
39bcd65d	71
	72	TFriendElement* AddReferenceTree(const char* filename, const char* treename = "calPads", const char* refname = "R");
	73	TFriendElement* AddFriend(const char* treename, const char* filename) {return fTree->AddFriend(treename, filename);};
	74	TFriendElement* AddFriend(TTree* tree, const char* alias, Bool_t warn=kFALSE) {return fTree->AddFriend(tree, alias, warn);};
	75	TFriendElement* AddFriend(const char* treename, TFile* file) {return fTree->AddFriend(treename, file);};
a6d2bd0c	76	TTree * GetTree() { return fTree;}
	77
	78	TString* Fit(const char* drawCommand, const char* formula, const char* cuts, Double_t & chi2, TVectorD &fitParam, TMatrixD &covMatrix);
72d0ab7e	79
	80	//
	81	// Array tools
	82	//
	83	static Double_t GetLTM(Int_t n, Double_t array, Double_t sigma = 0, Double_t fraction = 0.9);
	84	static Int_t GetBin(Float_t value, Int_t nbins, Double_t binLow, Double_t binUp);
	85	static TH1F* SigmaCut(Int_t n, Float_t *array, Float_t mean, Float_t sigma, Int_t nbins, Float_t binLow, Float_t binUp, Float_t sigmaMax, Float_t sigmaStep = -1, Bool_t pm = kFALSE);
	86	static TH1F* SigmaCut(TH1F *histogram, Float_t mean, Float_t sigma, Float_t sigmaMax, Float_t sigmaStep = -1, Bool_t pm = kFALSE);
	87	static TH1F* Integrate(TH1F *histogram, Float_t mean = 0, Float_t sigma = 0, Float_t sigmaMax = 0, Float_t sigmaStep = -1);
	88	static TH1F* Integrate(Int_t n, Float_t *array, Int_t nbins, Float_t binLow, Float_t binUp, Float_t mean = 0, Float_t sigma = 0, Float_t sigmaMax = 0, Float_t sigmaStep = -1);
	89
	90	static TH1F* SigmaCut(Int_t n, Double_t array, Double_t mean, Double_t sigma, Int_t nbins, Double_t xbins, Double_t sigmaMax);
	91
	92
	93
39bcd65d	94	protected:
	95	TTree* fTree; // tree containing visualization data (e.g. written by AliTPCCalPad::MakeTree(...)
	96	TFile* fFile; // file that contains a calPads tree (e.g. written by AliTPCCalPad::MakeTree(...)
	97	TObjArray* fListOfObjectsToBeDeleted; //Objects, that will be deleted when the destructor ist called
a6d2bd0c	98	Bool_t fTreeMustBeDeleted; // decides weather the tree must be deleted in destructor or not
832bb12c	99	TString fAbbreviation; // the abreviation for '.fElements'
832bb12c	100	TString fAppendString; // '.fElements', stored in a TStrig
39bcd65d	101
72d0ab7e	102	void DrawLines(TH1F cutHistoMean, TVectorF nsigma, TLegend legend, Int_t color, Bool_t pm) const;
4f3934a1	103	void DrawLines(TGraph graph, TVectorF nsigma, TLegend legend, Int_t color, Bool_t pm) const;
72d0ab7e	104
72d0ab7e	105
39bcd65d	106	ClassDef(AliTPCCalibViewer,1) // TPC calibration viewer class
	107	};
	108
	109	#endif
4f3934a1	110
	111
	112