SusyScan/PlotScript/scan.h

#ifndef SCAN_H
#define SCAN_H

#include <vector>

#include <TH2.h>
#include <TGraph.h>
#include <TGraph2D.h>

class MyTGraph : public TGraph{
 public:
   MyTGraph(int b):TGraph(b){};
   void Add(double x,double y){ this->SetPoint( this->GetN(),x,y); };
   void ls(){  
      double x,y;
      for (int i=0; i<this->GetN(); ++i){
        this->GetPoint(i,x,y);
        std::cout << i << "th: "<<x<<", " <<y<<std::endl;
      }};
};

class SUSY_POINT {
 public:
  double MZERO;
  double MHALF;
  double TANB;
  double SGNMU;
  double AZERO;
  double MTOP;
  double muQ;
  double Q;
  double M1;
  double M2;
  double M3;
  double MGL;
  double MUL;
  double MB1;
  double MSN;
  double MNTAU;
  double MZ1;
  double MW1;
  double MHL;
  double MUR;
  double MB2;
  double MEL;
  double MTAU1;
  double MZ2;
  double MW2;
  double MHH;
  double MDL;
  double MT1;
  double MER;
  double MTAU2;
  double MZ3;
  double MHA;
  double MDR;
  double MT2;
  double MZ4;
  double MHp;
};


class TScan {
 public:
   TScan();
   ~TScan();

   MyTGraph * IsoMassLine(bool(*func)(SUSY_POINT,double), double);
   MyTGraph * GetContour(TH2F*, int flag=0);
   TH2F * Area(bool(*func)(SUSY_POINT));
   TH2F * Area51();
   //static bool ul(const SUSY_POINT point, const double mass);
 
   int DoStuff();
 
 private:
   void ReadGeneratedMasses(std::string file);

   int plot_id, bins_x, bins_y;
   double min_x, max_x, min_y, max_y;
   std::vector<SUSY_POINT> points;

};


#endif
Revision:	1.1.1.1 (vendor branch)
Committed:	Fri Nov 30 13:10:50 2007 UTC (17 years, 5 months ago) by auterman
Content type:	text/plain
Branch:	tex, Demo, SusyScan
Changes since 1.1:	+0 -0 lines
Log Message:	Scripts for a SUSY parameter scan
#	Content
1	#ifndef SCAN_H
2	#define SCAN_H
3
4	#include <vector>
5
6	#include <TH2.h>
7	#include <TGraph.h>
8	#include <TGraph2D.h>
9
10	class MyTGraph : public TGraph{
11	public:
12	MyTGraph(int b):TGraph(b){};
13	void Add(double x,double y){ this->SetPoint( this->GetN(),x,y); };
14	void ls(){
15	double x,y;
16	for (int i=0; i<this->GetN(); ++i){
17	this->GetPoint(i,x,y);
18	std::cout << i << "th: "<<x<<", " <<y<<std::endl;
19	}};
20	};
21
22	class SUSY_POINT {
23	public:
24	double MZERO;
25	double MHALF;
26	double TANB;
27	double SGNMU;
28	double AZERO;
29	double MTOP;
30	double muQ;
31	double Q;
32	double M1;
33	double M2;
34	double M3;
35	double MGL;
36	double MUL;
37	double MB1;
38	double MSN;
39	double MNTAU;
40	double MZ1;
41	double MW1;
42	double MHL;
43	double MUR;
44	double MB2;
45	double MEL;
46	double MTAU1;
47	double MZ2;
48	double MW2;
49	double MHH;
50	double MDL;
51	double MT1;
52	double MER;
53	double MTAU2;
54	double MZ3;
55	double MHA;
56	double MDR;
57	double MT2;
58	double MZ4;
59	double MHp;
60	};
61
62
63	class TScan {
64	public:
65	TScan();
66	~TScan();
67
68	MyTGraph * IsoMassLine(bool(*func)(SUSY_POINT,double), double);
69	MyTGraph * GetContour(TH2F*, int flag=0);
70	TH2F * Area(bool(*func)(SUSY_POINT));
71	TH2F * Area51();
72	//static bool ul(const SUSY_POINT point, const double mass);
73
74	int DoStuff();
75
76	private:
77	void ReadGeneratedMasses(std::string file);
78
79	int plot_id, bins_x, bins_y;
80	double min_x, max_x, min_y, max_y;
81	std::vector<SUSY_POINT> points;
82
83	};
84
85
86	#endif