[u/mrichter/AliRoot.git] / PWG / muondep / AliAnalysisMuMuGraphUtil.h

#ifndef ALIANALYSISMUMUGRAPHUTIL_H
#define ALIANALYSISMUMUGRAPHUTIL_H

#ifndef ROOT_TObject
#  include "TObject.h"
#endif

#include <vector>
#include <string>
#include <set>

class TGraph;
class TGraphErrors;
class TCanvas;
class TObjArray;

#include "TAttFill.h"
#include "TAttMarker.h"
#include "TAttLine.h"
#include "TAttAxis.h"
#include "TString.h"

class AliAnalysisMuMuGraphUtil : public TObject
{
public:
  
  AliAnalysisMuMuGraphUtil(const char* ocdbpath="raw://");
  virtual ~AliAnalysisMuMuGraphUtil() {}

  static TGraphErrors* Combine(TObjArray& graph, Bool_t compact);
  
  static void Compact(TGraph& g);
  
  void DefaultStyle();

  TCanvas* DrawWith2Scales(TGraph& g1, TGraph& g2, const char* canvasName="c1");

  static Int_t GetRunNumber(const TGraph& g, Int_t i);

  void GetRuns(std::set<int>& runs, TGraph& graph) const;

  static Bool_t IsCompact(TGraph& g);
  
  void PlotSameWithLegend(TObjArray& a, Double_t ymin, Double_t ymax) const;

  void ShouldDrawPeriods(Bool_t value) { fShouldDrawPeriods = value; }

  void StyleGraph(TGraph& graph, UInt_t index) const;
  
  static void UnCompact(TGraph& g);
  
  static void GetYMinAndMax(TGraph& g, Double_t& ymin, Double_t& ymax);
  
  static TGraph* RelDif(TGraph& ga, TGraph& gb);

  
private:
  AliAnalysisMuMuGraphUtil(const AliAnalysisMuMuGraphUtil& rhs); // not implemented
  AliAnalysisMuMuGraphUtil& operator=(const AliAnalysisMuMuGraphUtil& rhs); // not implemented

  TString fOCDBPath; // OCDB path

  std::vector<TAttLine> fAttLine; // line attributes
  std::vector<TAttMarker> fAttMarker; // marker attributes
  std::vector<TAttFill> fAttFill; // fill attributes
  std::vector<TAttAxis> fAttXaxis; // x-axis attributes
  std::vector<TAttAxis> fAttYaxis; // y-axis attributes
  std::vector<std::string> fDrawOptions; // draw options
  
  Bool_t fShouldDrawPeriods; // draw period names on top of graphs
  
  ClassDef(AliAnalysisMuMuGraphUtil,0) // utility class to modify/plot graphs
};

#endif
Commit	Line	Data
81190958	1	#ifndef ALIANALYSISMUMUGRAPHUTIL_H
	2	#define ALIANALYSISMUMUGRAPHUTIL_H
	3
	4	#ifndef ROOT_TObject
	5	# include "TObject.h"
	6	#endif
	7
	8	#include <vector>
	9	#include <string>
	10	#include <set>
	11
	12	class TGraph;
	13	class TGraphErrors;
	14	class TCanvas;
	15	class TObjArray;
	16
	17	#include "TAttFill.h"
	18	#include "TAttMarker.h"
	19	#include "TAttLine.h"
	20	#include "TAttAxis.h"
	21	#include "TString.h"
	22
	23	class AliAnalysisMuMuGraphUtil : public TObject
	24	{
	25	public:
	26
	27	AliAnalysisMuMuGraphUtil(const char* ocdbpath="raw://");
	28	virtual ~AliAnalysisMuMuGraphUtil() {}
	29
	30	static TGraphErrors* Combine(TObjArray& graph, Bool_t compact);
	31
	32	static void Compact(TGraph& g);
	33
	34	void DefaultStyle();
	35
	36	TCanvas* DrawWith2Scales(TGraph& g1, TGraph& g2, const char* canvasName="c1");
	37
	38	static Int_t GetRunNumber(const TGraph& g, Int_t i);
	39
	40	void GetRuns(std::set<int>& runs, TGraph& graph) const;
	41
	42	static Bool_t IsCompact(TGraph& g);
	43
	44	void PlotSameWithLegend(TObjArray& a, Double_t ymin, Double_t ymax) const;
	45
	46	void ShouldDrawPeriods(Bool_t value) { fShouldDrawPeriods = value; }
	47
	48	void StyleGraph(TGraph& graph, UInt_t index) const;
	49
	50	static void UnCompact(TGraph& g);
	51
	52	static void GetYMinAndMax(TGraph& g, Double_t& ymin, Double_t& ymax);
	53
	54	static TGraph* RelDif(TGraph& ga, TGraph& gb);
	55
	56
	57	private:
	58	AliAnalysisMuMuGraphUtil(const AliAnalysisMuMuGraphUtil& rhs); // not implemented
	59	AliAnalysisMuMuGraphUtil& operator=(const AliAnalysisMuMuGraphUtil& rhs); // not implemented
	60
	61	TString fOCDBPath; // OCDB path
	62
	63	std::vector<TAttLine> fAttLine; // line attributes
	64	std::vector<TAttMarker> fAttMarker; // marker attributes
65	std::vector<TAttFill> fAttFill; // fill attributes
66	std::vector<TAttAxis> fAttXaxis; // x-axis attributes
67	std::vector<TAttAxis> fAttYaxis; // y-axis attributes
68	std::vector<std::string> fDrawOptions; // draw options
69
70	Bool_t fShouldDrawPeriods; // draw period names on top of graphs
71
72	ClassDef(AliAnalysisMuMuGraphUtil,0) // utility class to modify/plot graphs
73	};
74
75	#endif