3 #pragma GCC system_header
10 #include "TObjArray.h"
11 #include "TObjString.h"
12 #include "TPaletteAxis.h"
13 #include "TMultiLayerPerceptron.h"
14 #include "TMLPAnalyzer.h"
43 sprintf(NNi2,
"%s",NN_FILE.Data());
48 if((NNi2[p+1]==
'N')&&(NNi2[p+2]==
'\/')) {
51 while (NNi2[hh]!=
'\0'){NNi[hh-p-3]=NNi2[hh];hh=hh+1;}
62 sprintf(Filei2,
"%s",TEST_FILE.Data());
65 if(Filei2[q]==
'n'&&Filei2[q+1]==
'n'&&Filei2[q+2]==
'T'&&Filei2[q+3]==
'R'&&Filei2[q+4]==
'E'&&(Filei2[q+5]==
'E')&&(Filei2[q+6]==
'\/')) {
68 while (Filei2[hh]!=
'\0') {Filei[hh-p-7]=Filei2[hh];hh=hh+1;
71 for (
int h0=hh-p-7;h0<1024;h0++) Filei[h0]=
'\0';
77 cout<<NNi<<
" original String: "<<NNi2<<endl;
78 cout<<Filei<<
" original String: "<<Filei2<<endl;
83 NNi0.ReplaceAll(
".root",
"");
84 Filei0.ReplaceAll(
".root",
"");
87 TFile* fnet =TFile::Open(NN_FILE.Data());
88 TMultiLayerPerceptron *mlp = (TMultiLayerPerceptron*)fnet->Get(
"TMultiLayerPerceptron");
89 if(mlp==NULL) {cout <<
"Error getting mlp" << endl;
exit(1);}
90 TTree* infot=(TTree*)fnet->Get(
"info");
95 infot->SetBranchAddress(
"Rand_start_Sig",&NNs);
96 infot->SetBranchAddress(
"Rand_start_Bg",&NNb);
97 infot->SetBranchAddress(
"#trainSig",&NNnTS);
98 infot->SetBranchAddress(
"#trainBg",&NNnTB);
101 TFile* fTEST =TFile::Open(TEST_FILE.Data());
102 TTree* NNTree=(TTree*)fTEST->Get(
"nnTree");
103 int entries=NNTree->GetEntries();
104 cout<<
"entries: "<<entries<<endl;
114 NNTree->SetBranchAddress(
"#Entries_type",&entriesTot);
115 NNTree->SetBranchAddress(
"Matrix_dim",&ndim);
116 NNTree->SetBranchAddress(
"#inputs",&ninp);
117 NNTree->SetBranchAddress(
"amplitude_mode",&y);
120 sig_entries=entriesTot;
121 NNTree->GetEntry(entries-1);
122 bg_entries=entriesTot;
127 cout<<
"NDIM "<<NDIM<<endl;
128 cout<<
"nINP"<<nINP<<endl;
129 cout<<
"sig e "<<sig_entries<<endl;
130 cout<<
"bg e "<<bg_entries<<endl;
132 if (sig_entries>bg_entries) minevents=bg_entries;
133 else minevents=sig_entries;
135 if(b==0) b=sig_entries;
141 cout<<
"Error: Bg index<sig_entries"<<endl;
144 if((TS>sig_entries||
TB>bg_entries)&&(TS==
TB)) {TS=minevents-
s;
TB=minevents-b+sig_entries;}
145 if((TS>sig_entries||
TB>bg_entries)&&(TS!=
TB)) {TS=sig_entries-
s;
TB=bg_entries-b+sig_entries;}
149 sprintf(NOMEtot,
"NN_norm_%s_FILE_%s_TS%i_TB%i_st%i_bt%i_dANN%1.2f_uf%i",NNi0.Data(),Filei0.Data(),TS,
TB,
s,b,
deltaANN,uf);
150 cout<<
"nome: "<<NOMEtot<<endl;
155 NNTree->SetBranchAddress(
"Files_name",&FILE_NAME);
158 NNTree->GetEntry(entries-1);
160 cout<<
"fine ifdef RHO_CC"<<endl;
162 TChain sigTree(
"waveburst");
163 sigTree.Add(SIG_FILE.Data());
166 cout <<
"sig entries2 : " << sig_entries2 << endl;
168 TChain bgTree(
"waveburst");
169 bgTree.Add(BG_FILE.Data());
172 cout <<
"bg entries2 : " << bg_entries2 << endl;
174 cout<<
"b: "<<b<<endl;
175 cout<<
"s: "<<s<<endl;
180 for (
int jj=0; jj<nINP;jj++) x[jj]=0.;
181 char ilabel[nINP][16];
184 for(
int i=0;
i<nINP;
i++) {
186 NNTree->SetBranchAddress(ilabel[i], &xo[i]);
190 sprintf(ofile,
"outfile/%s.root",NOMEtot);
192 TFile*
f=
new TFile(ofile,
"RECREATE");
193 TTree* NNTree2=
new TTree(
"Parameters",
"Parameters");
194 NNTree2->SetDirectory(f);
198 NNTree2->Branch(
"ANNout",&out,
"ANNout/D");
199 NNTree2->Branch(
"cc",&NNcc,
"cc/D");
200 NNTree2->Branch(
"rho",&NNrho,
"rho/D");
201 char NNfilename[1024];
202 NNTree2->Branch(
"NNname",&NNfilename,
"NNname/C");
203 sprintf(NNfilename,
"%s",NN_FILE.Data());
205 NNTree2->Branch(
"TestFile",&Testf,
"TestFile/C");
206 sprintf(Testf,
"%s",TEST_FILE.Data());
208 NNTree2->Branch(
"#TestSig",&nTestS,
"#TestSig/I");
210 cout<<
"nTestS: "<<nTestS<<
" TS: "<<TS<<endl;
211 cout<<
"dopo def tree"<<endl;
227 for(
int n=s;
n<s+TS;
n++) {
228 if (uf!=0&&
n>=NNs&&
n<=(NNs+NNnTS))
continue;
235 for (
int i=0;
i<nINP;
i++) params[
i]=0.;
237 for(
int n=s;
n<s+TS;
n++) {
238 if (uf!=0&&
n>=NNs&&
n<=(NNs+NNnTS))
continue;
241 NNcc=(double)signal.
netcc[1];
242 NNrho=(
double)signal.
rho[0];
245 for (
int i=0;
i<nINP;
i++) {
if(xo[
i]>xo_max) xo_max=xo[
i];}
246 for (
int i=0;
i<nINP;
i++){
251 double output=mlp->Evaluate(0,params);
253 if (out<0.6) sig_05=sig_05+1;
254 cout<<
"rho: "<<signal.
rho[0]<<
" cc "<<signal.
netcc[1]<<
" out: "<<out<<endl;
258 cout<<
"riempito sig"<<endl;
261 for(
int n=b;
n<b+
TB;
n++) {
262 if (uf!=0&&
n>=NNb&&
n<=(NNb+NNnTB))
continue;
264 cout<<
"n: "<<
n<<
"Bg index"<<(
n-sig_entries)<<endl;
266 NNcc=(double)background.
netcc[1];
267 NNrho=(
double)background.
rho[0];
270 for (
int i=0;
i<nINP;
i++){
if(xo[
i]>xo_max) xo_max=xo[
i];}
271 for (
int i=0;
i<nINP;
i++){
277 double output=mlp->Evaluate(0,params);
279 if(out>0.6) bg_05=bg_05+1;
280 cout<<
"rho: "<<background.
rho[0]<<
" cc "<<background.
netcc[1]<<
" out: "<<out<<endl;
283 cout<<
"s "<<s<<
" TS "<<TS<<
" b "<<b<<
" TB "<< TB<<endl;
284 cout<<
"riempito bg"<<endl;
286 Realentries=NNTree2->GetEntries();
287 cout<<
" Realentries "<<Realentries<<endl;
290 cout<<
" Realentries "<<Realentries<<endl;
291 cout<<
"chiuso file"<<endl;
294 cout<<
"dopo richiamo funzione"<<endl;
298 cout<<
" Sig with out<06: "<<sig_05<<endl;
299 cout<<
" Bg with out>06: "<<bg_05<<endl;
300 cout<<
" Realentries "<<Realentries<<endl;
306 name.ReplaceAll(
"outfile/",
"");
307 TFile* fTEST =TFile::Open(ifile.Data());
308 TTree* NNTree2=(TTree*)fTEST->Get(
"Parameters");
313 NNTree2->SetBranchAddress(
"ANNout",&out);
314 NNTree2->SetBranchAddress(
"cc",&cc);
315 NNTree2->SetBranchAddress(
"rho",&rho);
316 NNTree2->SetBranchAddress(
"#TestSig",&nSi);
317 NNTree2->GetEntry(0);
319 cout<<
"nSig: "<<nSig<<
" nSi: "<<nSi<<endl;
321 cout<<
"dentro funzione dopodef"<<endl;
323 double* rhoSig[ncurve];
324 for (
int i=0;
i<ncurve;
i++) rhoSig[
i]=
new double[nSig];
325 cout<<
"dopo def rhoSig"<<endl;
329 int const nBg=NNTree2->GetEntries()-nSig;
330 for (
int i=0;
i<ncurve;
i++) {
332 for (
int j=0;
j<nSig;
j++) rhoSig[
i][
j]=0.;
334 cout<<
"dopo def rhoSig"<<endl;
335 double* rhoBg[ncurve];
336 for (
int i=0;
i<ncurve;
i++) rhoBg[
i]=
new double[nBg];
339 for (
int i=0;
i<ncurve;
i++) {
341 for (
int j=0;
j<nBg;
j++) rhoBg[
i][
j]=0.;
343 cout<<
"dopo def rhoBg"<<endl;
345 for (
int i=0;
i<
nCC;
i++) ccTh[
i]=0.;
347 for (
int i=0;
i<
nANN;
i++) NNTh[
i]=0.;
353 cout<<NNTree2->GetEntries()<<endl;
354 for(
int n=0;
n<NNTree2->GetEntries();
n++){
356 NNTree2->GetEntry(
n);
357 cout<<
"rho "<<rho<<
" cc "<<cc<<
" out "<<out<<endl;
360 if(cc<ccTh[
i])
continue;
362 if(
j==0) NNTh[
j]=-1000.;
368 if(out<NNTh[
j])
continue;
371 NBg[i*nANN+
j]= NBg[i*nANN+
j]+1;
372 while(rhoBg[i*nANN+j][ni]!=0)ni=ni+1;
373 rhoBg[i*nANN+
j][ni]=
rho;
378 NSig[i*nANN+
j]= NSig[i*nANN+
j]+1;
379 while(rhoSig[i*nANN+j][ni]!=0)ni=ni+1;
380 rhoSig[i*nANN+
j][ni]=
rho;
388 cout<<
"dopo riempimento variabili"<<endl;
390 for (
int i=0;
i<ncurve;
i++) indexS[
i]=
new int[nSig];
393 for (
int y=0;
y<ncurve;
y++) {
406 TMath::Sort(nSig,rhoSig[
y],indexS[y],
false);
408 for (
int k=0;
k<nSig;
k++) {
413 int ij=indexS[
y][
k-1];
415 if(rhoSig[y][ii]!=0) {
418 if(rhoSig[y][ii]!=rhoSig[y][ij]) gS[
y]->SetPoint(igS_p++,rhoSig[y][ii],yy);
419 cout<<
"igS"<<igS<<
" x "<<rhoSig[
y][ii]<<
" y: "<<yy<<endl;
424 if(rhoSig[y][ii]!=0){
426 gS[
y]->SetPoint(0,rhoSig[y][ii],yy);
437 for (
int i=0;
i<ncurve;
i++) indexB[
i]=
new int[nBg];
440 for (
int y=0;
y<ncurve;
y++) {
444 TMath::Sort(nBg,rhoBg[
y],indexB[y],
false);
446 for (
int k=0;
k<nBg;
k++) {
450 int ij=indexB[
y][
k-1];
452 if(rhoBg[y][ii]!=0) {
455 if(rhoBg[y][ii]!=rhoBg[y][ij]) gB[
y]->SetPoint(igB_p++,rhoBg[y][ii],yy);
461 if(rhoBg[y][ii]!=0) {
463 gB[
y]->SetPoint(0,rhoBg[y][ii],yy);
470 cout<<
"dopo inserimento puntiB"<<endl;
473 TCanvas* cS=
new TCanvas(
"Efficiency_vs_rho",
"Efficiency_vs_rho",0,0,1200,700);
475 cS->cd(1)->SetLogy();
476 TMultiGraph* mg1=
new TMultiGraph();
478 gS[0]->SetMarkerColor(2);
479 gS[0]->SetLineColor(2);
480 mg1->SetTitle(
"cc=0.5;rho;#Events");
481 if(gS[0]->GetN()!=0) mg1->Add(gS[0]);
484 gS[
h]->SetMarkerColor(3);
485 gS[
h]->SetLineColor(3);
487 if(gS[
h]->GetN()!=0) mg1->Add(gS[
h]);
491 cS->cd(2)->SetLogy();
492 TMultiGraph* mg2=
new TMultiGraph();
494 gS[
nANN]->SetMarkerColor(2);
495 gS[
nANN]->SetLineColor(2);
496 mg2->SetTitle(
"cc=0.55;rho;#Events");
497 if(gS[nANN]->GetN()!=0) mg2->Add(gS[nANN]);
499 gS[nANN+
h]->SetMarkerColor(3);
500 gS[nANN+
h]->SetLineColor(3);
502 if(gS[nANN+
h]->GetN()!=0) mg2->Add(gS[nANN+
h]);
506 cS->cd(3)->SetLogy();
507 TMultiGraph* mg3=
new TMultiGraph();
509 gS[nANN*2]->SetMarkerColor(2);
510 gS[nANN*2]->SetLineColor(2);
511 mg3->SetTitle(
"cc=0.6;rho;#Events");
512 if(gS[nANN*2]->GetN()!=0) mg3->Add(gS[nANN*2]);
514 gS[2*nANN+
h]->SetMarkerColor(3);
515 gS[2*nANN+
h]->SetLineColor(3);
517 if(gS[2*nANN+
h]->GetN()!=0) mg3->Add(gS[2*nANN+
h]);
521 cS->cd(4)->SetLogy();
522 TMultiGraph* mg4=
new TMultiGraph();
524 gS[nANN*3]->SetMarkerColor(2);
525 gS[nANN*3]->SetLineColor(2);
526 mg4->SetTitle(
"cc=0.65;rho;#Events");
527 if(gS[nANN*3]->GetN()!=0) mg4->Add(gS[nANN*3]);
530 gS[3*nANN+
h]->SetMarkerColor(3);
531 gS[3*nANN+
h]->SetLineColor(3);
533 if(gS[3*nANN+
h]->GetN()!=0) mg4->Add(gS[3*nANN+
h]);
537 cout<<
"nuovo canv"<<endl;
538 TCanvas* cB=
new TCanvas(
"Number_vs_rho",
"Number_vs_rho",0,0,1200,700);
540 cB->cd(1)->SetLogy();
541 TMultiGraph* mg1B=
new TMultiGraph();
543 gB[0]->SetMarkerColor(2);
544 gB[0]->SetLineColor(2);
545 mg1B->SetTitle(
"cc=0.5;rho;#Events");
546 if(gB[0]->GetN()!=0) mg1B->Add(gB[0]);
548 gB[
h]->SetMarkerColor(3);
549 gB[
h]->SetLineColor(3);
551 if(gB[
h]->GetN()!=0) mg1B->Add(gB[
h]);
555 cB->cd(2)->SetLogy();
556 TMultiGraph* mg2B=
new TMultiGraph();
558 gB[
nANN]->SetMarkerColor(2);
559 gB[
nANN]->SetLineColor(2);
560 mg2B->SetTitle(
"cc=0.55;rho;#Events");
561 if(gB[nANN]->GetN()!=0) mg2B->Add(gB[nANN]);
563 gB[nANN+
h]->SetMarkerColor(3);
564 gB[nANN+
h]->SetLineColor(3);
566 if(gB[nANN+
h]->GetN()!=0) mg2B->Add(gB[nANN+
h]);
570 cB->cd(3)->SetLogy();
571 TMultiGraph* mg3B=
new TMultiGraph();
573 gB[2*
nANN]->SetMarkerColor(2);
574 gB[2*
nANN]->SetLineColor(2);
575 mg3B->SetTitle(
"cc=0.6;rho;#Events");
576 if(gB[2*nANN]->GetN()!=0) mg3B->Add(gB[2*nANN]);
578 gB[2*nANN+
h]->SetMarkerColor(3);
579 gB[2*nANN+
h]->SetLineColor(3);
581 if(gB[2*nANN+
h]->GetN()!=0) mg3B->Add(gB[2*nANN+
h]);
585 cB->cd(4)->SetLogy();
586 TMultiGraph* mg4B=
new TMultiGraph();
588 gB[3*
nANN]->SetMarkerColor(2);
589 gB[3*
nANN]->SetLineColor(2);
590 mg4B->SetTitle(
"cc=0.65;rho;#Events");
591 if(gB[3*nANN]->GetN()!=0) mg4B->Add(gB[3*nANN]);
593 gB[3*nANN+
h]->SetMarkerColor(3);
594 gB[3*nANN+
h]->SetLineColor(3);
596 if(gB[3*nANN+
h]->GetN()!=0) mg4B->Add(gB[3*nANN+
h]);
608 char CnameS2root[1024];
609 char CnameB2root[1024];
610 CnameS.ReplaceAll(
".root",
".png");
611 CnameB.ReplaceAll(
".root",
".png");
612 sprintf(CnameS2,
"logN_rho/logN_rho_S_dANN%1.2f_%s",
deltaANN,CnameS.Data());
613 sprintf(CnameB2,
"logN_rho/logN_rho_B_dANN%1.2f_%s",
deltaANN,CnameB.Data());
614 sprintf(CnameS2root,
"logN_rho/logN_rho_S_dANN%1.2f_%s",
deltaANN,CnameSroot.Data());
615 sprintf(CnameB2root,
"logN_rho/logN_rho_B_dANN%1.2f_%s",
deltaANN,CnameBroot.Data());
618 cS->Print(CnameS2root);
619 cB->Print(CnameB2root);
627 name.ReplaceAll(
"outfile/",
"");
628 TFile* fTEST =TFile::Open(ifile.Data());
629 TTree* NNTree2=(TTree*)fTEST->Get(
"Parameters");
634 NNTree2->SetBranchAddress(
"ANNout",&out);
635 NNTree2->SetBranchAddress(
"cc",&cc);
636 NNTree2->SetBranchAddress(
"rho",&rho);
637 NNTree2->SetBranchAddress(
"#TestSig",&nSi);
638 NNTree2->GetEntry(0);
644 double* ANNSig[ncurve2];
645 for (
int i=0;
i<ncurve2;
i++) ANNSig[
i]=
new double[nSig];
649 int const nBg=NNTree2->GetEntries()-nSig;
650 for (
int i=0;
i<ncurve2;
i++) {
652 for (
int j=0;
j<nSig;
j++) ANNSig[
i][
j]=0.;
654 double* ANNBg[ncurve2];
655 for (
int i=0;
i<ncurve2;
i++) ANNBg[
i]=
new double[nBg];
659 for (
int i=0;
i<ncurve2;
i++) {
661 for (
int j=0;
j<nBg;
j++) ANNBg[
i][
j]=0.;
664 for (
int i=0;
i<
nCC;
i++) ccTh[
i]=0.;
666 for (
int i=0;
i<
nRHO;
i++) rhoTh[
i]=0.;
673 for(
int n=0;
n<NNTree2->GetEntries();
n++){
674 NNTree2->GetEntry(
n);
675 cout<<
"rho "<<rho<<
" cc "<<cc<<
" out "<<out<<endl;
678 if(cc<ccTh[
i])
continue;
681 if(rho<rhoTh[
j])
continue;
684 NBg[i*nRHO+
j]= NBg[i*nRHO+
j]+1;
685 while(ANNBg[i*nRHO+j][ni]!=0)ni=ni+1;
686 ANNBg[i*nRHO+
j][ni]=
out;
690 NSig[i*nRHO+
j]= NSig[i*nRHO+
j]+1;
691 while(ANNSig[i*nRHO+j][ni]!=0)ni=ni+1;
692 ANNSig[i*nRHO+
j][ni]=
out;
699 int* indexS[ncurve2];
700 for (
int i=0;
i<ncurve2;
i++) indexS[
i]=
new int[nSig];
702 TGraph * gS[ncurve2];
703 for (
int y=0;
y<ncurve2;
y++) {
707 TMath::Sort(nSig,ANNSig[
y],indexS[y],
false);
708 for (
int k=0;
k<nSig;
k++) {
713 int ij=indexS[
y][
k-1];
715 if(ANNSig[y][ii]!=0) {
718 if(ANNSig[y][ii]!=ANNSig[y][ij]) gS[
y]->SetPoint(igS_p++,ANNSig[y][ii],yy);
725 if(ANNSig[y][ii]!=0){
727 gS[
y]->SetPoint(0,ANNSig[y][ii],yy);
738 int* indexB[ncurve2];
739 for (
int i=0;
i<ncurve2;
i++) indexB[
i]=
new int[nBg];
741 TGraph * gB[ncurve2];
742 for (
int y=0;
y<ncurve2;
y++) {
746 TMath::Sort(nBg,ANNBg[
y],indexB[y],
false);
748 for (
int k=0;
k<nBg;
k++) {
752 int ij=indexB[
y][
k-1];
754 if(ANNBg[y][ii]!=0) {
757 if(ANNBg[y][ii]!=ANNBg[y][ij]) gB[
y]->SetPoint(igB_p++,ANNBg[y][ii],yy);
763 if(ANNBg[y][ii]!=0) {
765 gB[
y]->SetPoint(0,ANNBg[y][ii],yy);
774 TCanvas* cS=
new TCanvas(
"Efficiency_vs_ANN",
"Efficiency_vs_ANN",0,0,1200,700);
778 TMultiGraph* mg1=
new TMultiGraph();
779 mg1->SetTitle(
"cc=0.5;ANN;#Events");
781 gS[
h]->SetLineColor(4);
782 if(gS[
h]->GetN()!=0) mg1->Add(gS[
h]);
787 TMultiGraph* mg2=
new TMultiGraph();
788 mg2->SetTitle(
"cc=0.55;ANN;#Events");
790 gS[nRHO+
h]->SetLineColor(4);
791 if(gS[nRHO+
h]->GetN()!=0) mg2->Add(gS[nRHO+
h]);
797 TMultiGraph* mg3=
new TMultiGraph();
798 mg3->SetTitle(
"cc=0.6;ANN;#Events");
800 gS[2*nRHO+
h]->SetLineColor(4);
801 if(gS[2*nRHO+
h]->GetN()!=0) mg3->Add(gS[2*nRHO+
h]);
806 TMultiGraph* mg4=
new TMultiGraph();
807 mg4->SetTitle(
"cc=0.65;ANN;#Events");
809 gS[3*nRHO+
h]->SetLineColor(4);
810 if(gS[3*nRHO+
h]->GetN()!=0) mg4->Add(gS[3*nRHO+
h]);
814 TCanvas* cB=
new TCanvas(
"Number_vs_ANN",
"Number_vs_ANN",0,0,1200,700);
816 cB->cd(1)->SetLogy();
817 TMultiGraph* mg1B=
new TMultiGraph();
818 mg1B->SetTitle(
"cc=0.5;ANN;#Events");
820 gB[
h]->SetLineColor(4);
821 if(gB[
h]->GetN()!=0) mg1B->Add(gB[
h]);
824 cB->cd(2)->SetLogy();
825 TMultiGraph* mg2B=
new TMultiGraph();
826 mg2B->SetTitle(
"cc=0.55;ANN;#Events");
828 gB[nRHO+
h]->SetLineColor(4);
829 if(gB[nRHO+
h]->GetN()!=0) mg2B->Add(gB[nRHO+
h]);
832 cB->cd(3)->SetLogy();
833 TMultiGraph* mg3B=
new TMultiGraph();
834 mg3B->SetTitle(
"cc=0.6;ANN;#Events");
836 gB[2*nRHO+
h]->SetLineColor(4);
837 if(gB[2*nRHO+
h]->GetN()!=0) mg3B->Add(gB[2*nRHO+
h]);
840 cB->cd(4)->SetLogy();
841 TMultiGraph* mg4B=
new TMultiGraph();
842 mg4B->SetTitle(
"cc=0.6;ANN;#Events");
844 gB[3*nRHO+
h]->SetLineColor(4);
845 if(gB[3*nRHO+
h]->GetN()!=0) mg4B->Add(gB[3*nRHO+
h]);
857 char CnameS2root[1024];
858 char CnameB2root[1024];
859 CnameS.ReplaceAll(
".root",
".png");
860 CnameB.ReplaceAll(
".root",
".png");
861 sprintf(CnameS2,
"ANNthres/N_ANN_S_%s",CnameS.Data());
862 sprintf(CnameB2,
"ANNthres/N_ANN_B_%s",CnameB.Data());
863 sprintf(CnameS2root,
"ANNthres/N_ANN_S_%s",CnameSroot.Data());
864 sprintf(CnameB2root,
"ANNthres/N_ANN_B_%s",CnameBroot.Data());
867 cS->Print(CnameS2root);
868 cB->Print(CnameB2root);
880 name.ReplaceAll(
"outfile/",
"");
881 TFile* fTEST =TFile::Open(ifile.Data());
882 TTree* NNTree2=(TTree*)fTEST->Get(
"Parameters");
887 NNTree2->SetBranchAddress(
"ANNout",&out);
888 NNTree2->SetBranchAddress(
"cc",&cc);
889 NNTree2->SetBranchAddress(
"rho",&rho);
890 NNTree2->SetBranchAddress(
"#TestSig",&nSi);
891 NNTree2->GetEntry(0);
893 cout<<
"nSig: "<<nSig<<
" nSi: "<<nSi<<endl;
894 int const nBg=NNTree2->GetEntries()-nSig;
923 for (
int i=0;
i<nSig;
i++){
931 for (
int i=0;
i<nBg;
i++){
937 for (
int n = 0;
n <NNTree2->GetEntries();
n++){
938 NNTree2->GetEntry(
n);
966 for (
int k=0;
k<nBg;
k++)indexB[
k]=0;
967 TMath::Sort(nBg,aRHOB,indexB,
false);
971 double ANN1th[
nth+1];
973 for (
int k=0;
k<(
nth);
k++){
982 for (
int k=0;
k<(
nth);
k++){
983 ANN1th[
k+1]=0.+dANNth*(
k+1);
984 cc1th[
k]=0.5+dccth*
k;
999 for (
int a=0;
a<nBg;
a++){
1000 cout<<
"tutti: a "<<
a<<
" rho "<<aRHOB[
a]<<
" aNNB "<<aANNB[
a]<<
" cc "<<aCCB[
a]<<endl;
1001 if(aRHOB[
a]<
RHOth)
continue;
1002 cout<<
"dopo rho: a "<<
a<<
" rho "<<aRHOB[
a]<<
" aNNB "<<aANNB[
a]<<
" cc "<<aCCB[
a]<<endl;
1003 for (
int b=0;b<
nth;b++){
1004 if(aCCB[
a]<cc1th[b])
continue;
1005 cout<<
"dopo cc: "<<cc1th[b]<<
"a "<<
a<<
" rho "<<aRHOB[
a]<<
" aNNB "<<aANNB[
a]<<
" cc "<<aCCB[
a]<<endl;
1006 for(
int c=0;
c<nth+1;
c++){
1007 if(aANNB[
a]<ANN1th[
c])
continue;
1008 cout<<
"dopo ANN:"<<ANN1th[
c]<<
" a "<<
a<<
" rho "<<aRHOB[
a]<<
" aNNB "<<aANNB[
a]<<
" cc "<<aCCB[
a]<<endl;
1009 Z[b*(nth+1)+c]=Z[b*(nth+1)+
c]+1;
1010 cout<<
"Zindex: "<<b*(nth+1)+c<<
" Z: "<<Z[b*(nth+1)+
c]<<endl;
1021 for (
int d=0; d<
nth+1;d++) bins[d]=0.;
1023 for (
int d=0; d<nth+1;d++) bins[d+1]=0.+(d+1)*dANNth;
1024 TH2D* hglitch=
new TH2D(
"Survived_glitches(RHO_cut:5.5)",
"Survived_glitches(RHO_cut:5.5)",nth,0.5,1,nth+1,bins);
1026 hglitch->SetDirectory(0);
1028 for (
int b=0;b<
nth;b++)
for(
int c=0;
c<nth+1;
c++) {
1030 hglitch->Fill(cc1th[b]+dccth/2.,bins[
c]+dANNth/2.,Z[(nth+1)*b+
c]);
1031 cout<<
" Zindex: "<<(nth+1)*b+
c<<
" x: "<<cc1th[b]<<
" y: "<<ANN1th[
c]<<
" z: "<<Z[(nth+1)*b+
c]<<endl;
1034 gB[0]=
new TGraph(nBg,aCCB,aRHOB);
1035 gS[0]=
new TGraph(nSig,aCCS,aRHOS);
1036 gB[1]=
new TGraph(nBg,aCCB,aANNB);
1037 gS[1]=
new TGraph(nSig,aCCS,aANNS);
1038 gB[2]=
new TGraph(nBg,aRHOB,aANNB);
1039 gS[2]=
new TGraph(nSig,aRHOS,aANNS);
1041 for(
int i=0;
i<3;
i++){
1042 gS[
i]->SetMarkerColor(2);
1043 gB[
i]->SetMarkerColor(4);
1044 gS[
i]->SetMarkerStyle(6);
1045 gB[
i]->SetMarkerStyle(7);
1048 TCanvas*
c=
new TCanvas(
"Plots",
"Plots",0,0,1200,700);
1051 TMultiGraph* mg1=
new TMultiGraph();
1052 mg1->SetTitle(
"cc_rho;cc;rho");
1053 if(gB[0]->GetN()!=0) mg1->Add(gB[0]);
1054 if(gS[0]->GetN()!=0) mg1->Add(gS[0]);
1057 TMultiGraph* mg2=
new TMultiGraph();
1058 mg2->SetTitle(
"cc_ANNoutput;cc;ANNoutput");
1059 if(gB[1]->GetN()!=0) mg2->Add(gB[1]);
1060 if(gS[1]->GetN()!=0) mg2->Add(gS[1]);
1063 TMultiGraph* mg3=
new TMultiGraph();
1064 mg3->SetTitle(
"rho_ANNoutput;rho;ANNoutput");
1065 if(gB[2]->GetN()!=0) mg3->Add(gB[2]);
1066 if(gS[2]->GetN()!=0) mg3->Add(gS[2]);
1069 TText*
text=
new TText(0.37,0.0,
"no cuts on ANN");
1070 hglitch->SetStats(0);
1071 hglitch->GetXaxis()->SetTitle(
"cc");
1072 hglitch->GetYaxis()->SetTitle(
"ANNoutput");
1073 hglitch->GetZaxis()->SetTitle(
"count");
1074 hglitch->Draw(
"colz");
1081 char Cname2root[1024];
1082 Cname.ReplaceAll(
".root",
".png");
1083 sprintf(Cname2,
"CC_RHO_ANN_Plots/CC_RHO_ANN_Plots_%s",Cname.Data());
1084 sprintf(Cname2root,
"CC_RHO_ANN_Plots/CC_RHO_ANN_Plots_%s",Cnameroot.Data());
1086 c->Print(Cname2root);
1088 TCanvas*
c2=
new TCanvas(
"Plots_Bkg_on_Sig",
"Plots_Bkg_on_Sig",0,0,1200,700);
1092 gB[0]->Draw(
"p,same");
1096 gB[1]->Draw(
"p,same");
1099 gB[2]->Draw(
"p,same");
1101 TText* text2=
new TText(0.37,0.0,
"no cuts on ANN");
1102 hglitch->SetStats(0);
1103 hglitch->GetXaxis()->SetTitle(
"cc");
1104 hglitch->GetYaxis()->SetTitle(
"ANNoutput");
1105 hglitch->GetZaxis()->SetTitle(
"count");
1106 hglitch->Draw(
"colz");
1112 char Cname2_2[1024];
1113 char Cname2_2root[1024];
1114 Cname_2.ReplaceAll(
".root",
".png");
1115 sprintf(Cname2_2,
"CC_RHO_ANN_Plots/CC_RHO_ANN_Plots_BoS_%s",Cname_2.Data());
1116 sprintf(Cname2_2root,
"CC_RHO_ANN_Plots/CC_RHO_ANN_Plots_BoS_%s",Cname_2root.Data());
1117 c2->SaveAs(Cname2_2);
1118 c2->Print(Cname2_2root);
Float_t * rho
biased null statistics
wavearray< double > a(hp.size())
void Plots(TString ifile)
cout<< endl;cout<< "ts size = "<< ts.size()<< " ts rate = "<< ts.rate()<< endl;tf.Forward(ts, wdm);int levels=tf.getLevel();cout<< "tf size = "<< tf.size()<< endl;double dF=tf.resolution();double dT=1./(2 *dF);cout<< "rate(hz) : "<< RATE<< "\t layers : "<< nLAYERS<< "\t dF(hz) : "<< dF<< "\t dT(ms) : "<< dT *1000.<< endl;int itime=TIME_PIXEL_INDEX;int ifreq=FREQ_PIXEL_INDEX;int index=(levels+1)*itime+ifreq;double time=itime *dT;double freq=(ifreq >0)?ifreq *dF:dF/4;cout<< endl;cout<< "PIXEL TIME = "<< time<< " sec "<< endl;cout<< "PIXEL FREQ = "<< freq<< " Hz "<< endl;cout<< endl;wavearray< double > x
void Annth(TString ifile)
void graph(TString ifile)
cout<< "Injected signals: "<< mdc.GetEntries()<< endl;cout<< "Injected signals in histogram factor_events_inj: "<< NEVTS<< endl;float myifar, ecor, m1, m2, netcc[3], neted, penalty;float rho[2];float chirp[6];float range[2];float frequency[2];float iSNR[3], sSNR[3];sim.SetBranchAddress("mass", mass);sim.SetBranchAddress("factor",&factor);sim.SetBranchAddress("range", range);sim.SetBranchAddress("chirp", chirp);sim.SetBranchAddress("rho", rho);sim.SetBranchAddress("netcc", netcc);sim.SetBranchAddress("neted",&neted);sim.SetBranchAddress("ifar",&myifar);sim.SetBranchAddress("ecor",&ecor);sim.SetBranchAddress("penalty",&penalty);sim.SetBranchAddress("time", mytime);sim.SetBranchAddress("iSNR", iSNR);sim.SetBranchAddress("sSNR", sSNR);sim.SetBranchAddress("spin", spin);sim.SetBranchAddress("frequency", frequency);float **volume=new float *[NBINS_mass1];float **volume_first_shell=new float *[NBINS_mass1];float **radius=new float *[NBINS_mass1];float **error_volume=new float *[NBINS_mass1];float **error_volume_first_shell=new float *[NBINS_mass1];float **error_radius=new float *[NBINS_mass1];for(int i=0;i< NBINS_mass1;i++){volume[i]=new float[NBINS_mass2];volume_first_shell[i]=new float[NBINS_mass2];radius[i]=new float[NBINS_mass2];error_volume[i]=new float[NBINS_mass2];error_volume_first_shell[i]=new float[NBINS_mass2];error_radius[i]=new float[NBINS_mass2];for(int j=0;j< NBINS_mass2;j++){volume[i][j]=0.;volume_first_shell[i][j]=0.;radius[i][j]=0.;error_volume[i][j]=0.;error_volume_first_shell[i][j]=0.;error_radius[i][j]=0.;}}float **spin_mtot_volume=new float *[NBINS_MTOT+1];float **spin_mtot_radius=new float *[NBINS_MTOT+1];float **error_spin_mtot_volume=new float *[NBINS_MTOT+1];float **error_spin_mtot_radius=new float *[NBINS_MTOT+1];for(int i=0;i< NBINS_MTOT+1;i++){spin_mtot_volume[i]=new float[NBINS_SPIN+1];spin_mtot_radius[i]=new float[NBINS_SPIN+1];error_spin_mtot_volume[i]=new float[NBINS_SPIN+1];error_spin_mtot_radius[i]=new float[NBINS_SPIN+1];for(int j=0;j< NBINS_SPIN+1;j++){spin_mtot_volume[i][j]=0.;error_spin_mtot_volume[i][j]=0.;spin_mtot_radius[i][j]=0.;error_spin_mtot_radius[i][j]=0.;}}char fname[1024];sprintf(fname,"%s/recovered_signals.txt", netdir);ofstream fev;fev.open(fname, std::ofstream::out);sprintf(line,"#GPS@L1 FAR[Hz] eFAR[Hz] Pval ""ePval factor rho frequency iSNR sSNR \n");fev<< line<< endl;ofstream *fev_single=new ofstream[nfactor];for(int l=1;l< nfactor+1;l++){sprintf(fname,"%s/recovered_signals_%d.txt", netdir, l);fev_single[l-1].open(fname, std::ofstream::out);fev_single[l-1]<< line<< endl;}double Vrho[RHO_NBINS], eVrho[RHO_NBINS], Rrho[RHO_NBINS], eRrho[RHO_NBINS], Trho[RHO_NBINS];for(int i=0;i< RHO_NBINS;i++){Vrho[i]=0.;eVrho[i]=0.;Rrho[i]=0.;eRrho[i]=0.;Trho[i]=RHO_MIN+i *RHO_BIN;}double dV, dV1, dV_spin_mtot, nevts, internal_volume;int nT;int countv=0;int cnt=0;int cnt2=0;int cntfreq=0;bool bcut=false;double liveTot=sim.GetMaximum("ifar");double BKG_LIVETIME_yr=liveTot/CYS;double BKG_LIVETIME_Myr=BKG_LIVETIME_yr/(1.e6);cout.precision(14);cout<< "Total live time ---> background
void Test0_dANN_var_norm(TString NN_FILE, TString TEST_FILE, int TS=0, int TB=0, int s=0, int b=0, int uf=0)
Float_t * netcc
effective correlated SNR
sprintf(tfres,"(1/%g)x(%g) (sec)x(Hz)", 2 *df, df)