#include <iostream>
#include <fstream>
#include <TFile.h>
#include <TTree.h>
#include <stdlib.h>
#include <TGraph.h>
#include <TCanvas.h>
#include <TH1.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH2.h>
#include <TF1.h>
#include <TROOT.h>
#include <TStyle.h>
#include "TMath.h"
#include </work/halla/g2p/disk1/ryan/mysql_root/g2p_mysql_lib.h>
#include </work/halla/g2p/disk1/pzhu/work/sql_root/runinfo_sqlite.h>
#include <cstring>
#include <iostream>
void beam_current_dependence(TString name ="2253_2252", int baserun=0){
	gStyle->SetTitleSize(0.06,"xyz");
	gStyle->SetTitleSize(0.08,"t");
    gStyle->SetTitleOffset(0.8,"x");	//X-axis title offset from axis
	gStyle->SetTitleOffset(0.8,"y");	//y-axis title offset from axis
	gStyle->SetTitleOffset(0.0,"z");	//y-axis title offset from axis
	gStyle->SetLabelSize(0.04,"xyz");

	runinfo info;
	TString runlist = "/work/halla/g2p/disk1/jie/g2psim2016/calc_model_xs/elastic_pb_yields/rastercut/lists_yields_tmp/";
	runlist +=name;
	Int_t runnum[3000],i,count;
	Double_t current[3000], posx[3000], posy[3000], posth[3000], posph[3000];
	ifstream file_list(runlist.Data(),ios::in);
	if(!file_list){
		cout << "Whoops, can't find the file list!" << endl;
		cout << "Exiting the program." << endl;
		exit(1);
	}

	else {
		i=0;
		file_list >> runnum[i];

		while(!file_list.eof()){
			cout <<" i= "<<i<< "   runnum = " << runnum[i]<< endl;
			i++;
			file_list >> runnum[i];
		}
		file_list.close();
		count=i;
		for (int j=baserun;j<count;j++)
		{
			//	cout<<" count "<<count<<"  now run :"<<runnum[j]<<endl;
			int krun = runnum[j];
			int temprun = krun;
			if(krun==4947||krun==4948) temprun = 4949;
			if(krun ==5067) temprun = 5066;
			if(krun==5198||krun==5197) temprun = 5199;
			if(krun ==5698) temprun = 5699;
			posx[j-baserun]= get_my_tgtx(temprun);

			posy[j-baserun]= get_my_tgty(temprun);
			posth[j-baserun]= 1000*get_my_tgtth(temprun);
			posph[j-baserun]= 1000*get_my_tgtph(temprun);
			double momentum= get_my_D1p(krun);
			double energy= get_my_beam_energy(krun);
			double slrasterx = info.fgetsqlinfo(temprun, "slrasterx");
			double slrastery = info.fgetsqlinfo(temprun, "slrastery");
			double fsrasterx = info.fgetsqlinfo(temprun, "fstrasterx");
			double fsrastery = info.fgetsqlinfo(temprun, "fstrastery");
			current[j-baserun]=info.fgetsqlinfo(temprun, "Current");
			int mID = get_my_materialID(krun);
			slrasterx = slrasterx/2.0 ;
			slrastery = slrastery/2.0 ;
			fsrasterx = fsrasterx/2.0 ;
			fsrastery = fsrastery/2.0 ;
			cout<<krun<<"\t"<<energy<<"\t"<<mID<<"\t"<<momentum<<"\t"<<current[j-baserun]<<"\t"<<posx[j-baserun]<<"\t"<<posth[j-baserun]<<"\t"<<posy[j-baserun]<<"\t"<<posph[j-baserun]<<"\t"<<slrasterx<<"\t"<<slrastery<<"\t"<<fsrasterx<<"\t"<<fsrastery<<endl;
		//cout<<current[j-baserun]<<endl;	
		//	      cout<<krun<<"\t"<<energy<<"\t"<<mID<<"\t"<<momentum<<"\t"<<yields[j]<<"\t"<<acc[j]<<"\t"<<endl;
		}

		TGraph *grx = new TGraph(count-baserun, current, posx);
		TGraph *gry = new TGraph(count-baserun, current, posy);
		TGraph *grth = new TGraph(count-baserun, current, posth);
		TGraph *grph = new TGraph(count-baserun, current, posph);
		grx->SetTitle("Beam x (Horizontal) vs Current;Current (nA); Beam x (mm)");
		gry->SetTitle("Beam y (Vertical) vs Current;Current (nA); Beam y (mm)");
		grth->SetTitle("Beam Tilt Angle dy/dz vs Current;Current (nA); Beam Tilt Angle (mrad)");
		grph->SetTitle("Beam Tilt Angle dx/dz vs Current;Current (nA); Beam Tilt Angle (mrad)");
		TCanvas *c1 = new TCanvas("c1", "c1",800,600);
		c1->Divide(2,2);
		c1->cd(1);
		grx->Draw("A*");
		c1->cd(2);
		grph->Draw("A*");
		c1->cd(3);
		gry->Draw("A*");
		c1->cd(4);
		grth->Draw("A*");
		c1->SaveAs(Form("beam_current_%s.png",name.Data()));
	}
}