#include<fstream>
#include<TH1.h>
#include<TH2.h>
#include<TFile.h>
#include<TTree.h>
#include<TCanvas.h>
#include<parameter.h>
#include<math.h>
//#include<iostream>
#include<iostream> using namespace std;
//ここまでで使うライブラリをincludeしてます.parameter.hというのは自分でつくってヘッダファイルで,キャリブレーション用のパラメターが書いてあります.
void macro_calb_oka(){ //Last modified 2011/11/22
char filename[99];//root file name
// sprintf(filname,"22na20111104.001.root");
//sprintf(filename,"22na-10.03.001-002.root");
//spintf(filename,"22na-11.10.20-21.root");
sprintf(filename
,"22na-20111028.002.root"); // sprintf(filename,"22na20111104.001.root");
// sprintf(filename,"22na-20111111.001.root");
// sprintf(filename,"22na-20111118.001.root");
// sprintf(filename,"22na-20111201.001.root");
//filenameという変数を宣言して,そこにファイル名を代入してます.この変数はファイルを開くのと,あとでPDFファイルをつくるときに使います.sprintfはprintfと似た関数で,変数に文字列をいれることができます.ROOTのマクロを書くときにはけっこう便利です.あと、文字列をいれるときは,charというクラスでその文字数以上の配列として宣言するみたいです。
TFile *f1=new TFile(filename);//opening root file
//ファイルを開いています。
Double_t p[9][9][9];
//pというDouble_t クラスの配列は、フィッティングのときに使うパラメターです。p[i][j][k]はチャネルiのj番目のピークのk番目のパラメターを意味しています。j=1は511keVのピーク、j=2は1275のピーク、j=3はサムピークです。k=0はガウシアン全体にかかる定数(使わない)k=1はMean,k=2はSigmaです。
Double_t ey[3];
TH1F *h[99];
h[1]=h101;
h[2]=h102;
h[3]=h103;
h[4]=h104;
h[5]=h105;
h[6]=h106;
//TH1Fは1次のFloatのHistgramのクラスです。ROOTファイルに入っているh101~h106までのヒストグラムをプログラム中で扱いやすいようにh[i]という配列を宣言して、そこに代入しています。
// end of front matter
// gStyle->SetOptFit(0001);
gStyle->SetOptFit(0000);
//ここはフィッティングのパラメータをどのようにキャンバスに出力するか決めています
TCanvas *c1 = new TCanvas("c1","Raw Data and Fitting");
c1->Divide(2,3);
TCanvas *c2 = new TCanvas("c2","Enegy Calibration");
c2->Divide(2,3);
TCanvas *c3 = new TCanvas("c3","Calibrated Hist");
c3->Divide(2,3);
//TCanvasというクラスはヒストグラムとかを出力するためのクラスです.キャンバスをDivideするとキャンバスを複数に分けることができます
for(Int_t ch=1;ch<=6;ch++){
/* char hclone[256];
sprintf(hclone,"hclone%d",ch);
hoge = h[ch]->Clone(hclone);
h[ch+10]=hoge;
//h101〜h106を違う目的で使いたいので複製しています*/
c1->cd(ch)->SetLogy();
h[ch]->Draw();
//まず,キャンバスc1の6つに分割したうちの1番目に移動し,ch=1のときはh[1]=h101をDrawします.
//Preparation of Fitting function
//ここからはピークをフィッティングするための関数を準備しています.
TF1 *fit1;
char fname[256];
char mean1[256],mean2[256],mean3[256];
char resol1[256],resol2[256],resol3[256];
//fnameは関数の名前で,ひとつめのピークにはチャネルごとにf[1]?f[6]をフィッティングします
fit1 = new TF1(fname,"gaus",p1start[ch],p1end[ch]);
//この行ではフィッティングのための関数fit1をつくっています.名前はfnameに入っている値,種類はガウシアン,定義域はp1startとp1endに入っている値です
h[ch]->Fit(fit1,"","",p1start[ch],p1end[ch]);
//h[ch]にfit1をフィッティングします
fit1->GetParameters(p[ch][1]);
//フィットしたfit1について,そのフィッティングパラメータを得ます.これによってp[ch][1][1]にはMeanが,p[ch][1][2]にはSigmaが入ります.
ey[0] = fit1->GetParError(1);
sprintf(mean1
," Mean =%lf",p
[ch
][1][1]); sprintf(resol1
,"Resol.=%lf",2*sqrt(2*log(2.0))*p
[ch
][1][2]/p
[ch
][1][1]); //凡例に表示するための文字列を準備しています
//同じことを2つ目,3つ目のピークについても行います.
TF1 *fit2;
fit2 = new TF1(fname,"gaus",p2start[ch],p2end[ch]);
h[ch]->Fit(fit2,"","",p2start[ch],p2end[ch]);
fit2->GetParameters(p[ch][2]);
ey[1] = fit2->GetParError(1);
sprintf(mean2
," Mean =%lf",p
[ch
][2][1]); sprintf(resol2
,"Resol.=%lf",2*sqrt(2*log(2.0))*p
[ch
][2][2]/p
[ch
][2][1]);
TF1 *fit3;
fit3 = new TF1(fname,"gaus",p3start[ch],p3end[ch]);
h[ch]->Fit(fname,"","",p3start[ch],p3end[ch]);
fit3->GetParameters(p[ch][3]);
ey[2] = fit3->GetParError(1);
sprintf(mean3
," Mean =%lf",p
[ch
][3][1]); sprintf(resol3
,"Resol.=%lf",2*sqrt(2*log(2.0))*p
[ch
][3][2]/p
[ch
][3][1]);
h[ch]->Draw();//もう1回Drawします.なぜかこうしないとうまくいかなかったきがする
fit1->Draw("sames");//fit1をDrawします."sames"と書くと上書きされます.
fit2->SetLineColor(3);
fit2->Draw("sames");//fit2の線の色を設定したのち,上書きします
fit3->SetLineColor(7);
fit3->Draw("sames");
//ここからは凡例(Legend)の設定です
TLegend *l = new TLegend(0.6255,0.75,0.875,0.45);
//TLegendクラスの変数 l を宣言しました
l->AddEntry(fit1,"511 keV","lp");
//lにfit1を追加し,その表示名を511 keVとしました
l->AddEntry((TObject*)0, mean1,"");
//文字列mean1をlに追加しました
l->AddEntry((TObject*)0, resol1,"");
l->AddEntry(fit2,"1274 keV","lp");
l->AddEntry((TObject*)0, mean2,"");
l->AddEntry((TObject*)0, resol2,"");
l->AddEntry(fit3,"511+1274 keV","lp");
l->AddEntry((TObject*)0, mean3,"");
l->AddEntry((TObject*)0, resol3,"");
l->SetFillColor(0);// l の背景を白にします
l->Draw();//lをキャンバスに描きます
// Graph of calibration
//エネルギーキャリブレーションの式を求めるグラフをつくります
c2->cd(ch);
//xとyの散布図をつくるために,値を準備しておきます
Double_t x[3],y[3];
y[0]=511;
y[1]=1274;
y[2]=511+1274;
x[0]=p[ch][1][1];
x[1]=p[ch][2][1];
x[2]=p[ch][3][1];
// TGraph *g=new TGraphErrors(3,x,y,ey,0);//gをxとyの散布図として宣言してつくります
TGraph *g=new TGraph(3,x,y);//gをxとyの散布図として宣言してつくります
char gtitle[256];
sprintf(gtitle
, "Calibration for ADC%d",ch
); //グラフのタイトルを準備します
g->SetTitle(gtitle);//グラフのタイトルをgtitleに入っている値にしました
g->SetMarkerStyle(5);//グラフのデータ点の設定
g->SetMarkerSize(1);//データ点の大きさ
g->SetMarkerColor(1);//データ点の色
g->GetXaxis()->SetTitle("ADC Channel");//x軸の名前
g->GetYaxis()->SetTitle("Energy (keV)");
g->Draw("AP");//グラフをDrawします.APってオプションはなんだか忘れました
char fgname[256];
TF1 *fg = new TF1(fgname,"pol1",0,4000);
g->Fit(fg,"","",0,4000);
//つくったグラフに対して1次元の1次関数fgをfitします
char func[256];
Double_t pcal[9][2];
fg->GetParameters(pcal[ch]);
sprintf(func
,"y=%lfx+%lf",pcal
[ch
][1],pcal
[ch
][0]); //fitしたfgに対してパラメータを取得し,funcという文字列に式を書いておきます
TLegend *lg = new TLegend(0.5255,0.40,0.875,0.15);
lg->AddEntry(g,"Data Point","p");
lg->AddEntry(fg,"Fitted line","l");
lg->AddEntry((TObject*)0, func, "");
lg->SetFillColor(0);
lg->Draw();
//canvas2にもlgというlegendをつくりました.
//次にヒストグラムの軸をkeVで書き直したものをつくります
c3->cd(ch)->SetLogy();
//hstartとhendは最後に横軸をkeVで書き換えたあと、各チャンネルのプロットする範囲を決める変数です。
double hstart, hend;
hstart=pcal[ch][1]*0+pcal[ch][0];
hend=pcal[ch][1]*4000.0+pcal[ch][0];
/* h[ch+10]->SetBins(4000,hstart,hend);
h[ch+10]->Draw();*/
//x軸を書き直す作業はこれだけで終わりです
//あとはさっきと同じことをしてる
//Gaussian: Resolution=FWHM/Mean
//FWHM=2*Sigma*sqrt(2*log2)
// 分解能出すときにcalibrationの後もう一度フィッティングしてからのほうがいいのかも
sprintf(mean1
," Mean =%lf keV",pcal
[ch
][1]*p
[ch
][1][1]+pcal
[ch
][0]); sprintf(resol1
,"Resol.=%lf",2*sqrt(2*log(2.0))*(pcal
[ch
][1]*p
[ch
][1][2])/(pcal
[ch
][1]*p
[ch
][1][1]+pcal
[ch
][0])); sprintf(mean2
," Mean =%lf keV",pcal
[ch
][1]*p
[ch
][2][1]+pcal
[ch
][0]); sprintf(resol2
,"Resol.=%lf",2*sqrt(2*log(2.0))*(pcal
[ch
][1]*p
[ch
][2][2])/(pcal
[ch
][1]*p
[ch
][2][1]+pcal
[ch
][0])); sprintf(mean3
," Mean =%lf keV",pcal
[ch
][1]*p
[ch
][3][1]+pcal
[ch
][0]); sprintf(resol3
,"Resol.=%lf",2*sqrt(2*log(2.0))*(pcal
[ch
][1]*p
[ch
][3][2])/(pcal
[ch
][1]*p
[ch
][3][1]+pcal
[ch
][0]));
TLegend *lc = new TLegend(0.6255,0.75,0.875,0.45);
// lc->AddEntry(fit1,"511 keV","lp");
lc->AddEntry((TObject*)0, mean1,"");
lc->AddEntry((TObject*)0, resol1,"");
// lc->AddEntry(fit2,"1274 keV","lp");
lc->AddEntry((TObject*)0, mean2,"");
lc->AddEntry((TObject*)0, resol2,"");
// lc->AddEntry(fit3,"511+1274 keV","lp");
lc->AddEntry((TObject*)0, mean3,"");
lc->AddEntry((TObject*)0, resol3,"");
lc->SetFillColor(0);
lc->Draw();
}
//こっからは新機能
//1次式+ガウシアンでフィットし直す
char fglname[99];
TF1 fgl[7][4];//function gauss + linear [ch No.][peak No.]
TCanvas *c4 = new TCanvas("c4","Fitting with Gaussian and Linear function");
c4->Divide(2,3);
Double_t paratest[7][5];
for (Int_t ch=1; ch<7; ch++){
c4->cd(ch)->SetLogy();
h[ch]->Draw();
fgl[ch][1] = new TF1(fglname,"[0]/sqrt(2.0*3.14)/[1]*exp(-(x-[2])*(x-[2])/2.0/[1]/[1])+[3]*x+[4]");//[0]:constant ,[1]:sigma, [2]:mean
fgl[ch][1]->SetParameters(p[ch][1][0],p[ch][1][2],p[ch][1][1],0,0);//パラメータの初期値の設定
h[ch]->Fit(fglname,"","",p1start[ch],p1end[ch]);
fgl[ch][1]->GetParameters(paratest[ch]);// parameterをgetしてこれを使えば直線の式が出せるはずだがなんかできない
//ftest= new TF1("ftest","[0]*x+[1]",p1start[ch],p1end[ch]);
TF1 *ftest = new TF1("ftest","[0]*x+[1]");
ftest->SetParameter(0,paratest[ch][3]);
ftest->SetParameter(1,paratest[ch][4]);
ftest->SetLineColor(7);
fgl[ch][1]->Draw("sames");
ftest->Draw("sames");
/* //2番目3番目のピークを重ね書きするのはまだできてない
sprintf(fglname,"fgl2%d",ch);
fgl[ch][2] = new TF1(fglname,"[0]/sqrt(2.0*3.14)/[1]*exp(-(x-[2])*(x-[2])/2.0/[1]/[1])+[3]*x+[4]");//[0]:constant ,[1]:sigma, [2]:mean
fgl[ch][2]->SetParameters(p[ch][2][0],p[ch][2][2],p[ch][2][1],0,0);
h[ch]->Fit(fglname,"","",p2start[ch],p2end[ch]);
fgl[ch][2]->Draw("sames");
sprintf(fglname,"fgl3%d",ch);
fgl[ch][3] = new TF1(fglname,"[0]/sqrt(2.0*3.14)/[1]*exp(-(x-[2])*(x-[2])/2.0/[1]/[1])+[3]*x+[4]");//[0]:constant ,[1]:sigma, [2]:mean
fgl[ch][3]->SetParameters(p[ch][3][0],p[ch][3][2],p[ch][3][1],0,0);
h[ch]->Fit(fglname,"","",p3start[ch],p3end[ch]);
fgl[ch][3]->Draw("sames");
*/
}
//キャンバスをPDFに出力します
//Print
char pdfstart[99];
sprintf(pdfstart
,"%s.pdf(",filename
); char pdfmiddle[99];
sprintf(pdfmiddle
,"%s.pdf",filename
); char pdfend[99];
sprintf(pdfend
,"%s.pdf)",filename
);
c1->Print(pdfstart);
c2->Print(pdfmiddle);
c3->Print(pdfmiddle);
c4->Print(pdfend);
//この下では別のプログラムで使うためのヘッダファイルをつくっています.あんまりきにしなくてもいいはず
//フィッティングパラメーターをファイルに書きこむ
char pfile[99];
sprintf(pfile
,"parameter%s.h",filename
); FILE *fp;
fprintf(fp
,"Double_t para1[7], para2[7],para3[7],para4[7];\nDouble_t para12,para13,para14,para15,para16, para23,para24,para25,para26,para34,para35,para36,para45,para46,para56;\n\n"); fprintf(fp
,"//Energy = Naiadc[ch]*para1[ch]+para2[ch]\n");
//Writing ADC Parameters
char para1[99];
char para2[99];
for(Int_t ch=1; ch<7; ch++){
sprintf(para1
,"para1[%d]=%lf;\n",ch
,pcal
[ch
-1][3]); sprintf(para2
,"para2[%d]=%lf;\n",ch
,pcal
[ch
-1][2]); }
//Writing TDC Parameters
fprintf(fp
,"para3[1]=0.5707;\npara4[1]=42.063;\npara3[2]=0.5679;\npara4[2]=47.576;\npara3[3]=0.5361;\npara4[3]=67.592;\npara3[4]=0.6024;\npara4[4]=12.809;\npara3[5]=0.5663;\npara4[5]=50.119;\npara3[6]=0.56559;\npara4[6]=45.407;\n\n"); fprintf(fp
,"para12=0;\npara13=0;\npara14=0;\npara15=0;\npara16=0;\npara23=0;\npara24=0;\npara25=0;\npara26=0;\npara34=0;\npara35=0;\npara36=0;\npara45=0;\npara46=0;\npara56=0;\n");
}