plot_yfp_cuts.C

#include <TString.h>
#include <TLegend.h>
#include <TFile.h>
#include <TNtuple.h>
#include <TH1.h>
#include <TH2.h>
#include <TCanvas.h>
#include <TStyle.h>
#include <TROOT.h>
#include <TCutG.h>
#include <TMath.h>
#include <TProfile.h>
#include <TObjArray.h>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <math.h>
#include <iomanip>
#include <fstream>
using namespace std;


void plot_yfp_cuts(Int_t nrun=1814,Int_t FileID=-2) {
  
  gROOT->Reset();
  gStyle->SetOptStat(0);
  gStyle->SetOptFit(11);
  gStyle->SetTitleOffset(1.,"Y");
  gStyle->SetTitleOffset(.7,"X");
  gStyle->SetLabelSize(0.04,"XY");
  gStyle->SetTitleSize(0.06,"XY");
  gStyle->SetPadLeftMargin(0.14);
  const Int_t Number=3;
  Double_t Red[Number] = { 1.0,0.0,0.0};
  Double_t Blue[Number] = { 1.0,0.0,1.0};
  Double_t Green[Number] = { 0.0,1.0,0.0};
 Double_t Len[Number] = { 0.0,.5,1.0};
 Int_t nb=50;
 TColor::CreateGradientColorTable(Number,Len,Red,Green,Blue,nb);
  //  Get info for that optics run
 TString OpticsFile = "list_of_optics_run.dat";
   ifstream file_optics(OpticsFile.Data());
 TString opticsline;
  TString OpticsID="";
  Int_t RunNum=0.;
  Double_t CentAngle=0.;
  Int_t SieveFlag=1;
  Int_t NumFoil=0;
  TString temp;
 //
  vector <Double_t> ztar_foil;
  Int_t ndelcut;
  vector<Double_t > delcut;
  vector<Double_t > delwidth;
  if (file_optics.is_open()) {
    //
    cout << " Open file = " << OpticsFile << endl;
    while (RunNum!=nrun  ) {
      temp.ReadToDelim(file_optics,',');
      cout << temp << endl;
      if (temp.Atoi() == nrun) {
	RunNum = temp.Atoi();
      } else {
	temp.ReadLine(file_optics);
      }
    }
    if (RunNum==nrun) {
      temp.ReadToDelim(file_optics,',');
      OpticsID = temp;
      temp.ReadToDelim(file_optics,',');
      CentAngle = temp.Atof();
      temp.ReadToDelim(file_optics,',');
      NumFoil = temp.Atoi();
      temp.ReadToDelim(file_optics,',');
      SieveFlag = temp.Atoi();
      temp.ReadToDelim(file_optics);
      ndelcut = temp.Atoi();
      for (Int_t nf=0;nf<NumFoil-1;nf++) {
        temp.ReadToDelim(file_optics,',');
	ztar_foil.push_back(temp.Atof());
      }
        temp.ReadToDelim(file_optics);
	ztar_foil.push_back(temp.Atof());
      for (Int_t nd=0;nd<ndelcut-1;nd++) {
        temp.ReadToDelim(file_optics,',');
	delcut.push_back(temp.Atof());
      }
        temp.ReadToDelim(file_optics);
	delcut.push_back(temp.Atof());
	for (Int_t nw=0;nw<ndelcut-1;nw++) {
	temp.ReadToDelim(file_optics,',');
	delwidth.push_back(temp.Atof());
      }
      temp.ReadToDelim(file_optics);
      delwidth.push_back(temp.Atof());
    }
  } else {
    cout << " No file = " << OpticsFile << endl;    
  }
  cout << RunNum << " " << OpticsID << " " << CentAngle << " " << NumFoil << " " << SieveFlag << endl;
  if (NumFoil==0) return;
  //
  TString inputroot;
   TFile *fhistroot;
   TString outputpdf;
   inputroot=Form("hist/Optics_%s_%d_hist_v2.root",OpticsID.Data(),FileID);
   outputpdf = Form("plots/Optics__%s_%d_yfp_cuts",OpticsID.Data(),FileID);
     cout << " infile root = " << inputroot << endl;
   fhistroot =  new TFile(inputroot);
 //
   vector<vector<Double_t> > AxisRange;
   AxisRange.resize(ndelcut);
   TString AxisRangeFileName = "AxisRange_ypfp_yfp.dat";
   ifstream AxisRangeFile(AxisRangeFileName.Data());
   for  (Int_t nd=0;nd<ndelcut;nd++) { AxisRange[nd].resize(4) ;}
  //
  TString temp1;
  if (file_optics.is_open()) {
   for  (Int_t nd=0;nd<ndelcut;nd++) { 
     temp1.ReadToDelim(AxisRangeFile,',');
     AxisRange[nd][0] = temp1.Atof();
     temp1.ReadToDelim(AxisRangeFile,',');
     AxisRange[nd][1] = temp1.Atof();
     temp1.ReadToDelim(AxisRangeFile,',');
     AxisRange[nd][2] = temp1.Atof();
     temp1.ReadToDelim(AxisRangeFile);
     AxisRange[nd][3] = temp1.Atof();
      }
  }
  //

  //
	//
	vector<vector<TH2F*> > hYsXs_DelCut;
	vector<vector<TH2F*> > hYpFpYFp_DelCut;
	  vector<TH2F*> temp2d;
	TH2F* th;
	for  (Int_t nc=0;nc<NumFoil;nc++) {
	for  (Int_t nd=0;nd<ndelcut;nd++) {
	  th =  (TH2F*)fhistroot->Get(Form("hYsXs_Foil_%d_DelCut_%d",nc,nd));
	  if (!th) {
	    cout << " no hist : "<< Form("hYsXs_Foil_%d_DelCut_%d",nc,nd) << endl;
	    return;
	  }
	  temp2d.push_back(th);
	}	  
	hYsXs_DelCut.push_back(temp2d);
	  temp2d.clear();	
	}
	for  (Int_t nc=0;nc<NumFoil;nc++) {
	for  (Int_t nd=0;nd<ndelcut;nd++) {
	  th =  (TH2F*)fhistroot->Get(Form("hYpFpYFp_%d_DelCut_%d",nc,nd));
	  if (!th) {
	    cout << " no hist : "<< Form("hYpFpYFp_%d_DelCut_%d",nc,nd) << endl;
	    return;
	  }
	  temp2d.push_back(th);
	}	  
	hYpFpYFp_DelCut.push_back(temp2d);
	  temp2d.clear();	
	}
	//
	vector<vector<vector<TH2F*> > > hYsXs_DelCut_YpYfpCut;
	hYsXs_DelCut_YpYfpCut.resize(NumFoil);
	for  (Int_t nf=0;nf<NumFoil;nf++) {
	  hYsXs_DelCut_YpYfpCut[nf].resize(ndelcut);
	for  (Int_t nd=0;nd<ndelcut;nd++) {
	  hYsXs_DelCut_YpYfpCut[nf][nd].resize(11);
	}
	}
	for  (Int_t nc=0;nc<NumFoil;nc++) {
	for  (Int_t nd=0;nd<ndelcut;nd++) {
	for  (Int_t ny=0;ny<11;ny++) {
	  hYsXs_DelCut_YpYfpCut[nc][nd][ny] =  (TH2F*)fhistroot->Get(Form("hYsXs_Foil_%d_DelCut_%d_FpCut_%d",nc,nd,ny));
	}}}
	//
	  TLine* ys_line[11];
	  TText* ys_text[11];
  for (Int_t nys=0;nys<11;nys++) {
    Double_t pos=nys*1.64-1.64*5;
    ys_line[nys]= new TLine(pos,-15.,pos,15.);
    ys_text[nys]= new TText(pos,-17,Form("%d",nys));
     ys_text[nys]->SetTextColor(2);
     ys_line[nys]->SetLineColor(2);
     ys_line[nys]->SetLineWidth(1);
  }
	//
	/*
	TCanvas* can2d[NumFoil][ndelcut];
	for  (Int_t nc=0;nc<NumFoil;nc++) {
	    for  (Int_t nd=0;nd<ndelcut;nd++) {
	      can2d[nc][nd] = new TCanvas(Form("Can2d_%d_%d",nc,nd),Form("Foil %d Del %d",nc,nd), 700,700);
	      can2d[nc][nd]->Divide(3,4);
	  can2d[nc][nd]->cd(1);
	gPad->SetGridx();
	gPad->SetGridy();
	gPad->SetLogz();    
	  hYpFpYFp_DelCut[nc][nd]->Draw("colz");
	  hYpFpYFp_DelCut[nc][nd]->SetMinimum(1);
	  hYpFpYFp_DelCut[nc][nd]->GetYaxis()->SetRangeUser(AxisRange[nd][0],AxisRange[nd][1]);
	  hYpFpYFp_DelCut[nc][nd]->GetXaxis()->SetRangeUser(AxisRange[nd][2],AxisRange[nd][3]);
	  can2d[nc][nd]->cd(2);
	  hYsXs_DelCut[nc][nd]->Draw("colz");
	  hYsXs_DelCut[nc][nd]->SetMinimum(1);
          for (Int_t nys=0;nys<11;nys++) { ys_line[nys]->Draw();}
          for (Int_t nys=0;nys<11;nys++) { ys_text[nys]->Draw();}
	  Int_t incr=0;
	  	for  (Int_t ny=0;ny<11;ny++) {
		  if (hYsXs_DelCut_YpYfpCut[nc][nd][ny]->Integral()>0) {
		    can2d[nc][nd]->cd(3+incr);
		    hYsXs_DelCut_YpYfpCut[nc][nd][ny]->Draw("colz");
		    for (Int_t nys=0;nys<11;nys++) { ys_line[nys]->Draw();}
		    for (Int_t nys=0;nys<11;nys++) { ys_text[nys]->Draw();}
		    incr++;
		  }
		}
	  TString end = ".pdf";
	  if (nc==0 && nd==0) end=".pdf(";
	  if (nc==NumFoil-1 && nd==ndelcut-1) end=".pdf)";
	  can2d[nc][nd]->Print(outputpdf+end);
	    }}
	//
	*/
 //
  TString outCutFile;
  TFile *fcut;
  vector<vector<vector<TCutG*> > > ypfp_yfp_cut;
  vector<vector<vector<Int_t> > > ypfp_yfp_cut_flag;
  ypfp_yfp_cut.resize(NumFoil);
  ypfp_yfp_cut_flag.resize(NumFoil);
	for  (Int_t nf=0;nf<NumFoil;nf++) {
          ypfp_yfp_cut[nf].resize(ndelcut);
          ypfp_yfp_cut_flag[nf].resize(ndelcut);
   }
   outCutFile=Form("cuts/YpFpYFp_%s_%d_cut.root",OpticsID.Data(),FileID);
    fcut = new TFile(outCutFile);
    cout << " Cut file = " << outCutFile << endl;
    fcut->cd();
	for  (Int_t nf=0;nf<NumFoil;nf++) {
	for  (Int_t nd=0;nd<ndelcut;nd++) {
        for (Int_t nc=0;nc<11;nc++) {
	  TCutG* tempg  = (TCutG*)gROOT->FindObject(Form("hYpFpYFp_cut_yscol_%d_nfoil_%d_ndel_%d",nc,nf,nd));
	  if (tempg)  {
	    //cout << "hYpFpYFp_cut = " << nc << " " << nf << " " << nd << endl;
	  ypfp_yfp_cut[nf][nd].push_back(tempg);
      } else {
	    //cout << " No hYpFpYFp_cut = " << nc << " " << nf << " " << nd << endl;
     	  ypfp_yfp_cut[nf][nd].push_back(tempg);
      }
	}}}
	//
	TCanvas* canCut[NumFoil][ndelcut];
	for  (Int_t nf=0;nf<NumFoil;nf++) {
	    for  (Int_t nd=0;nd<ndelcut;nd++) {
	      canCut[nf][nd] = new TCanvas(Form("CanCut_%d_%d",nf,nd),Form("Foil %d Del %d",nf,nd), 700,700);
	      canCut[nf][nd]->Divide(2,2);
	  canCut[nf][nd]->cd(1);
	gPad->SetGridx();
	gPad->SetGridy();
	gPad->SetLogz();    
	  hYpFpYFp_DelCut[nf][nd]->Draw("colz");
	  hYpFpYFp_DelCut[nf][nd]->SetMinimum(1);
	  //hYpFpYFp_DelCut[nf][nd]->GetYaxis()->SetRangeUser(AxisRange[nd][0],AxisRange[nd][1]);
	  //hYpFpYFp_DelCut[nf][nd]->GetXaxis()->SetRangeUser(AxisRange[nd][2],AxisRange[nd][3]);
        for (Int_t nc=0;nc<11;nc++) {
          if (ypfp_yfp_cut[nf][nd][nc]) {
	    ypfp_yfp_cut[nf][nd][nc]->Draw("same");
             ypfp_yfp_cut[nf][nd][nc]->SetLineColor(2);
             ypfp_yfp_cut[nf][nd][nc]->SetLineWidth(2);
		 Double_t xcut,ycut;
                 ypfp_yfp_cut[nf][nd][nc]->GetPoint(0,xcut,ycut);
		 TText* ystext = new TText(xcut,ycut,Form("%d",nc));
		 ystext->Draw();
	  }
	}
	  canCut[nf][nd]->cd(2);
	gPad->SetLogz();    
	  hYsXs_DelCut[nf][nd]->Draw("colz");
	  hYsXs_DelCut[nf][nd]->SetMinimum(1);
          for (Int_t nys=0;nys<11;nys++) { ys_line[nys]->Draw();}
          for (Int_t nys=0;nys<11;nys++) { ys_text[nys]->Draw();}
	  canCut[nf][nd]->cd(3);
	gPad->SetLogz();    
	  	for  (Int_t ny=0;ny<11;ny++) {
		  if (hYsXs_DelCut_YpYfpCut[nf][nd][ny]->Integral()>0) {
		    if (ny==0) {
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->Draw("colz");
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->SetMinimum(1);
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->SetMaximum(300);
		    } else {
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->Draw("colz same");
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->SetMinimum(1);
		    hYsXs_DelCut_YpYfpCut[nf][nd][ny]->SetMaximum(300);
		    }
		  }
		}
		    for (Int_t nys=0;nys<11;nys++) { ys_line[nys]->Draw("same");}
		    for (Int_t nys=0;nys<11;nys++) { ys_text[nys]->Draw("same");}
	  TString end = ".pdf";
	  if (nf==0 && nd==0) end=".pdf(";
	  if (nf==NumFoil-1 && nd==ndelcut-1) end=".pdf)";
	  canCut[nf][nd]->Print(outputpdf+end);
	    }}
	//
	//
}