Subversion Repositories f9daq

#include "TROOT.h"
#include "TFile.h"
#include "TBenchmark.h"
#include "TH1.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TH3F.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TPad.h"
#include "TF1.h"
#include "TGraph.h"
#include "TSpectrum.h"
#include "TAttFill.h"
#include "stdio.h"

#include "include/RTUtil.h"

//#include "include/RTUtil.h"

int tdc(char filename[256] = "test", int chX=0, int chY=0, double rangeLeft=87, double rangeRight=107, bool debug = false)
{
  
  int map[8][8]={{32,34,53,55,40,42,61,63},
                 {48,50,37,39,56,58,45,47},
                 {33,35,52,54,41,43,60,62},
                 {49,51,36,38,57,59,44,46},
                 {17,19,4,6,25,27,12,14},
                 {1,3,20,22,9,11,28,30},
                 {16,18,5,7,24,26,13,15},
                 {0,2,21,23,8,10,29,31}
                };
  
  char fnameroot[256];
  TFile* rootfile;
        sprintf(fnameroot, "root/%s.root", filename); 
        rootfile = (TFile *) gROOT->FindObject(filename); 
        if(rootfile==NULL) rootfile = new TFile(fnameroot);
        if(rootfile==NULL) {
          printf("Cannot open root file %s!!!\n",fnameroot); 
          return(0);
        }

        // set draw style
        /*
        gStyle->SetPalette(1, 0);
        
        gStyle->SetPaperSize(TStyle::kA4);
        gStyle->SetStatBorderSize(1);
        gStyle->SetFrameBorderMode(0);
        gStyle->SetFrameFillColor(0);
        gStyle->SetCanvasBorderMode(0);
        gStyle->SetPadBorderMode(0);
        gStyle->SetPadColor(0);
        gStyle->SetCanvasColor(0);
        gStyle->SetStatColor(0);
        gStyle->SetOptFit(1111);
        gStyle->SetOptStat("ne");
        gStyle->SetPadTopMargin(0.15);
        gStyle->SetPadBottomMargin(0.15);
        gStyle->SetPadRightMargin(0.15);
        gStyle->SetPadLeftMargin(0.15);
        //gStyle->SetTitleYOffset(1.4);*/
        RTSetStyle(gStyle);
         
    TCanvas* canvas1 = new TCanvas("canvas1","canvas1",1000,1000);
    TH2F* h_tdc = (TH2F*) rootfile->Get("htdc");
    canvas1->cd();
    h_tdc->Draw("colz");
    
    //TH3F* h_3D = (TH3F*) rootfile->Get("h_correctedTDC");
    TH2F* h_correctedTDC = (TH2F*) rootfile->Get("h_correctedTDC");
    TCanvas* canvas2 = new TCanvas("canvas2","canvas2",1000,1000);
    canvas2->cd();
    h_correctedTDC->Draw("colz");
    
    
    //TH2D* h_correctedTDC = (TH2D*) h_3D->Project3D("xz");
    //h_correctedTDC->SetTitle("; t [ns]; Channel");
    //h_correctedTDC->Draw("colz");
    TCanvas* canvas3 = new TCanvas("canvas3","canvas3",1600,1600);
    canvas3->cd();
    //TH1D* h_allTDCbins = h_correctedTDC->ProjectionX("", 1, 64);
    //h_allTDCbins->Draw();
    
    
    int binY = map[chX][chY];
    if (debug) printf("%d\n", binY);
    //TH1D* channelY = h_correctedTDC->ProjectionX("",binY+1,binY+1);
    TH1D* channelY = h_tdc->ProjectionX("",binY+1,binY+1);
    channelY->SetStats(1);
    
    char title[256];
    sprintf(title,";Time [ns];Events");
    channelY->SetTitle(title);
    //channelY->GetYaxis()->SetTitleOffset(1.7);
    //TAxis* xAxis = h_correctedTDC->GetXaxis();
    //int range = xAxis->GetBinUpEdge(xAxis->GetLast()+1);
    //channelY->GetXaxis()->SetRangeUser(-range, range);
    //gStyle->SetOptStat("nemr");
    //gStyle->SetOptFit(111);
    channelY->GetXaxis()->SetRangeUser(rangeLeft,rangeRight);
    channelY->Draw();
    
    // Fit function: linear for background c + k*x
    // Signal: norm*gaussian(mu,sigma)
    TF1* f_gaus1 = new TF1("f_gaus1","[0] +[1]*x + gaus(2)", rangeLeft, rangeRight);
    //TF1* f_gaus1 = new TF1("f_gaus1","[0] +[1]*x + gaus(2)");
    f_gaus1->SetParNames("Const","Linear","Norm","#mu","#sigma");
    //f_gaus1->SetLineColor(kAzure+1);
    f_gaus1->SetLineColor(kBlack);
    f_gaus1->SetLineWidth(4);
    
    f_gaus1->SetParameter(0, channelY->GetMinimum());
    f_gaus1->SetParameter(1, 0.0);
    f_gaus1->SetParameter(2, channelY->GetMaximum() - channelY->GetMinimum());
    f_gaus1->SetParameter(3, channelY->GetBinCenter(channelY->GetMaximumBin()));
    f_gaus1->SetParameter(4, 1.0);
    //channelY->Fit(f_gaus1,"qr");
    //channelY->Fit(f_gaus1,"r");
    
    //Fit in range +- 3 sigma
    //channelY->Fit(f_gaus1,"r","", f_gaus1->GetParameter(3)-3*f_gaus1->GetParameter(4),
    channelY->Fit(f_gaus1,"","");
                                  //f_gaus1->GetParameter(3)+3*f_gaus1->GetParameter(4));
                                  
    double N = channelY->GetEntries();
    double Nbkg = (rangeRight - rangeLeft)*f_gaus1->GetParameter(0);
    double Nsig = N - Nbkg;
    printf("\nN = %f, Nsig = %f, Nbkg = %f, SN = %f***\n\n", N, Nsig, Nbkg, Nsig/Nbkg);
   /* 
    //to draw a shaded area above and below an histogram range, we create
   //a TGraph object (here we shade bins 60 to 80).
   Int_t i;
   Int_t n = 2*(80-60);
   TGraph *gr = new TGraph(2*n);
   for (i=0;i<20;i++) {
      Float_t xlow = h1->GetBinLowEdge(60+i);
      Float_t xup  = h1->GetBinLowEdge(60+i+1);
      Float_t y    = h1->GetBinContent(60+i);
      Float_t yup  = 1.1*y;
      Float_t ydown= 0.9*y;
      gr->SetPoint(2*i,  xlow,yup);
      gr->SetPoint(2*i+1,xup, yup);
      gr->SetPoint(2*n-2*i-1,xlow, ydown);
      gr->SetPoint(2*n-2*i-2,xup, ydown);
   }
   gr->SetFillColor(2);
   gr->Draw("lf");*/

   TH1D* timeWindow = (TH1D*) channelY->Clone();
    timeWindow->GetXaxis()->SetRangeUser(92,102);
    //timeWindow->SetFillColor(kAzure-1);
    timeWindow->SetFillColor(kBlack);
    timeWindow->SetFillStyle(3004);
    timeWindow->Draw("same");
    
  
  char drawing[128];
  sprintf(drawing, "ps/%sTDC.pdf", filename);
  canvas3->SaveAs(drawing);
  
  gPad->Update();                          
  return (0);
  }