#include "TH1.h"
 
#include "TH2.h"
 
#include "TGraph.h"
 
#include "TCanvas.h"
 
#include "TStyle.h"
 
#include "TLegend.h"
 
//#include "TROOT.h"
 
 
 
#include <stdio.h>
 
 
 
#define DATA_PATH  "./data/"
 
#define CFRAC 0.4
 
#define DWIDTH 1024
 
void drs(int nev=1000, int updfrq=20, const char *FileName="./data/x1.dat", int trgch=0, int debug=0)
 
{
 
  int i,j,ich,iev;
 
  int stat;
 
  char histname[128];
 
//  char FileName[300]="./data/n2.dat";
 
  FILE *fin;
 
 
 
  char recid[5]="SAMO";
 
  unsigned short *srecid = (unsigned short *) &recid[0];
 
  unsigned int * irecid;
 
  struct evrec {
 
    unsigned long evn;
 
    unsigned short tYear,tMonth,tDay,tHour,tMin,tSec,tmSec,reserved;
 
  } evrec;
 
 
 
  float dtdata[4][DWIDTH];
 
  float tcdata[4][DWIDTH];
 
  float tccalib[4];
 
 
 
  short lwfm[4];
 
  int nwfm[4];
 
  int board;
 
 
 
//  int imin[4]={250,250,250,250};
 
//  int imax[4]={750,750,750,750};
 
  unsigned short ichdat[4][DWIDTH];
 
  float chdat[4][DWIDTH],achdat[4][DWIDTH],chmax[4][2],chmin[4][2];
 
  float letime[4][2],cftime[4][2],qdc[4];
 
//  float threshold[4][2]={{0.,-0.02},{0.,0.},{0.,0.},{0.,-0.25}};
 
//  float twin[4][2]={{120.,130.},{100.,150.},{100.,150.},{70.,90.}};
 
//  float adcgate[4][2]={{124.,136.},{120.,140.},{120.,140.},{70.,90.}};
 
//  float threshold[4][2]={{0.,-0.05},{0.,-0.05},{0.,-0.25},{0.,-0.25}};
 
//  float twin[4][2]={{80.,100.},{80.,100.},{90.,110.},{50.,70.}};
 
//  float adcgate[4][2]={{80.,110.},{80.,110.},{90.,110.},{50.,70.}};
 
  float threshold[4][2]={{0.,-0.1},{0.,-0.05},{0.,-0.25},{0.,-0.25}};
 
  float twin[4][2]={{100.,120.},{100.,120.},{40.,130.},{60.,80.}};
 
  float adcgate[4][2]={{105.,125.},{105.,125.},{110.,130.},{60.,80.}};
 
  float vcut[4][2]={{-1,1},{-1,1},{-1,1},{-1,1}};
 
  
 
//  gROOT->Reset();
 
 
 
  gStyle->SetOptStat(1111111);
 
//  gStyle->SetPalette(52);
 
 
 
  TH1F *h1_chmax[4];
 
  TH1F *h1_chmaxt[4];
 
  TH1F *h1_chmin[4];
 
  TH1F *h1_chmint[4];
 
  TH1F *h1_letime[4][2];
 
  TH1F *h1_cftime[4][2];
 
  TH1F *h1_cftdif[4];
 
  TH1F *h1_qdc[4];
 
  TH2F *h2_dpo[4];
 
  TH2F *h2_tdcadc[4];
 
  TH2F *h2_ctdcadc[4];
 
  TH2F *h2_ctalk;
 
  TGraph *gr_wfm[4];
 
  TCanvas *c_wfm= new TCanvas("c_wfm","WFM",50,50,700,700);
 
  TLegend *leg = new TLegend(0.8, 0.8, 1, 1);
 
  c_wfm->Divide(1,1);
 
  c_wfm->cd(1);
 
 
 
  for (i=0;i<4;i++) {
 
    h1_chmax[i]=new TH1F(histname,histname,1100,-0.55,0.55);
 
    h1_chmaxt[i]=new TH1F(histname,histname,DWIDTH,-0.1,204.7);
 
    h1_chmin[i]=new TH1F(histname,histname,1100,-0.55,0.55);
 
    h1_chmint[i]=new TH1F(histname,histname,DWIDTH,-0.1,204.7);
 
    sprintf(histname
,"letimep_%02d",i
);  
    h1_letime[i][0]=new TH1F(histname,histname,500,twin[i][0],twin[i][1]);
 
    sprintf(histname
,"letimen_%02d",i
);  
    h1_letime[i][1]=new TH1F(histname,histname,500,twin[i][0],twin[i][1]);
 
    sprintf(histname
,"cftimep_%02d",i
);  
    h1_cftime[i][0]=new TH1F(histname,histname,500,twin[i][0],twin[i][1]);
 
    sprintf(histname
,"cftimen_%02d",i
);  
    h1_cftime[i][1]=new TH1F(histname,histname,500,twin[i][0],twin[i][1]);
 
    h1_cftdif[i]=new TH1F(histname,histname,2000,twin[i][0],twin[i][1]);
 
    h1_qdc[i]=new TH1F(histname,histname,500,-0.1,4.9);
 
    h2_dpo[i]=new TH2F(histname,histname,DWIDTH,-0.1,204.7,1100,-0.55,0.55);
 
    h2_tdcadc[i]=new TH2F(histname,histname,250,0.,0.5,400,twin[i][0],twin[i][1]);
 
    sprintf(histname
,"ctdcadc_%02d",i
);  
    h2_ctdcadc[i]=new TH2F(histname,histname,250,0.,0.5,400,twin[i][0],twin[i][1]);
 
    gr_wfm[i]=new TGraph(DWIDTH);
 
    gr_wfm[i]->SetMinimum(-0.55);
 
    gr_wfm[i]->SetMaximum(0.55);
 
    gr_wfm[i]->SetLineColor(i+1);
 
    gr_wfm[i]->SetMarkerColor(i+1);
 
    leg->AddEntry(gr_wfm[i], histname, "l");
 
    if (i) gr_wfm[i]->Draw("LP"); else gr_wfm[i]->Draw("ALP");
 
    
 
  }
 
  leg->Draw();
 
  h2_ctalk=new TH2F("ctalk","ctalk",250,0.,0.5,250,0.,0.5);
 
 
 
  fin
=fopen(FileName
,"rb"); 
 
 
  printf("%s, %lu\n",recid
,sizeof(evrec
));  
  
 
 
 
  for (i=0;i<4;i++){
 
    nwfm[i]=0;
 
    for (j=0;j<DWIDTH;j++){
 
      achdat[i][j]=0;
 
    }
 
  }
 
  unsigned int trgcell = 0;
 
  int event = -1;
 
 
 
  iev=0;
 
    if (iev>=nev) break;
 
    stat
=fread(recid
,1,4,fin
); 
    if (debug){
 
      if (recid
[1]=='#') printf("%c%c%u\n",recid
[0],recid
[1],srecid
[1]);  
    }
 
    switch (recid[0]){
 
       case 'T':
 
           switch (recid[1]){
 
             case 'I': event=0; break;
 
             case '#': trgcell = srecid[1]; break;
 
           } 
 
           break;
 
       case 'B':
 
           board = srecid[1]; 
 
           break;
 
       case 'C':
 
           ich=ich-1;
 
           if (event) {
 
             stat
=fread(ichdat
[ich
],1,sizeof(ichdat
[ich
]),fin
); 
           } else {
 
             stat
=fread(dtdata
[ich
],1,sizeof(float)*DWIDTH
,fin
); 
             
 
           }
 
           break;
 
       case 'E':
 
           stat
=fread(&evrec.
evn,1,sizeof(evrec
),fin
); 
           iev++;
 
           event=1;
 
           break;
 
       default:
 
           break; 
 
    }
 
 
 
 
 
    if (recid[0] == 'E'){
 
         for ( i=0;i<4;i++){
 
           if (!lwfm[i]) continue;
 
           if ((chmin[i][0]>vcut[i][0])&&(chmin[i][0]<vcut[i][1])) {
 
             h1_cftdif[i]->Fill(cftime[i][1]-cftime[trgch][1]+110.);
 
           }
 
//      h2_tdcadc[i]->Fill(-chmin[i][0],letime[i][1]-cftime[trgch][1]+72);
 
//      h2_ctdcadc[i]->Fill(-chmin[i][0],cftime[i][1]-cftime[trgch][1]+72);
 
           h2_tdcadc[i]->Fill(-chmin[0][0],letime[i][1]-cftime[trgch][1]+72);
 
           h2_ctdcadc[i]->Fill(-chmin[0][0],cftime[i][1]-cftime[trgch][1]+72);
 
         }
 
         h2_ctalk->Fill(-chmin[0][0],-chmin[1][0]);
 
    }
 
    if (event<1) continue;
 
    if (recid[0]!='C') continue;
 
 
 
    
 
    
 
    lwfm[ich]=0;
 
    qdc[ich]=0;
 
    chmax[ich][0]=-1; chmax[ich][1]=-1;
 
    chmin[ich][0]=1; chmin[ich][1]=-1;
 
    letime[ich][0]=-1.; letime[ich][1]=-1.;
 
    cftime[ich][0]=-1.; cftime[ich][1]=-1.;
 
    
 
 
 
      lwfm[ich]=1;
 
      nwfm[ich]+=1;
 
 
 
      float sum=0; 
 
      for (i=0;i<DWIDTH;i++) {
 
         sum+= dtdata[ich][(i+trgcell)%DWIDTH];
 
         tcdata[ich][i]=sum;         
 
      }
 
      tccalib[ich]=tcdata[ich][(DWIDTH-trgcell)%DWIDTH];
 
 
 
      for (i=0;i<DWIDTH;i++) {
 
         tcdata[ich][i]-=(tccalib[ich]-tccalib[0]); // assume ich 0 is connected and always first in the data          
 
      }
 
 
 
      for (i=0;i<DWIDTH;i++) {
 
//      for (i=imin[ich];i<imax[ich];i++) {
 
        chdat[ich][i]=-0.5+(float)ichdat[ich][i]/0xFFFF;
 
        achdat[ich][i]+=chdat[ich][i];
 
        if (chdat[ich][i]>chmax[ich][0]) {
 
          chmax[ich][0]=chdat[ich][i];
 
          chmax[ich][1]=tcdata[ich][i];
 
        }
 
        if (chdat[ich][i]<chmin[ich][0]) {
 
          chmin[ich][0]=chdat[ich][i];
 
          chmin[ich][1]=tcdata[ich][i];
 
        }
 
        float tdata = tcdata[ich][i];
 
        float tdatap = tcdata[ich][i+1];
 
        float tdatan = tcdata[ich][i-1];
 
 
 
        if ((tdata>adcgate[ich][0])&&(tdata<adcgate[ich][1])) {
 
          qdc[ich]+=-chdat[ich][i]*(tdatap-tdatan)/2;
 
        }
 
        if ((tdata>twin[ich][0])&&(tdata<twin[ich][1])) {
 
          if ((chdat[ich][i]<threshold[ich][1])&&(letime[ich][1]<0.)) {
 
            letime[ich][1]=tdatan+(tdata-tdatan)*
 
                           (threshold[ich][1]-chdat[ich][i-1])/(chdat[ich][i]-chdat[ich][i-1]);
 
// preveri za 0!!                           
 
          }
 
        }
 
      }
 
    
 
 
 
      i=ich;
 
    
 
     
 
      for (j=0;j<DWIDTH;j++) {
 
        float tdata  = tcdata[i][j];
 
        float tdatap = tcdata[i][j+1];
 
        float tdatan = tcdata[i][j-1];
 
        h2_dpo[i]->Fill(tdata,chdat[i][j]);
 
        gr_wfm[i]->SetPoint(j,tdata,chdat[i][j]+0.05*(i-1));
 
        if ((tdata>twin[i][0])&&(tdata<twin[i][1])) {
 
          if ((chmin[i][0]<threshold[i][1])&&(chdat[i][j]<chmin[i][0]*CFRAC)&&(cftime[i][1]<0.)) {
 
            cftime[i][1]=tdatan+(tdata-tdatan)*
 
                           (chmin[i][0]*CFRAC-chdat[i][j-1])/(chdat[i][j]-chdat[i][j-1]);
 
// preveri za 0!!                           
 
          }
 
        }
 
      }
 
    
 
    
 
      if ((chmax[i][1]>twin[i][0])&&(chmax[i][1]<twin[i][1])) {
 
        h1_chmax[i]->Fill(chmax[i][0]);
 
        if (chmax[i][0]>threshold[i][0]) h1_chmaxt[i]->Fill(chmax[i][1]);
 
      }
 
      if ((chmin[i][1]>twin[i][0])&&(chmin[i][1]<twin[i][1])) {
 
        h1_chmin[i]->Fill(chmin[i][0]);
 
        if (chmin[i][0]<threshold[i][1]) h1_chmint[i]->Fill(chmin[i][1]);
 
      }
 
      h1_qdc[i]->Fill(qdc[i]);
 
      h1_letime[i][1]->Fill(letime[i][1]);
 
      h1_cftime[i][1]->Fill(cftime[i][1]);
 
//84,56
 
 
 
 
 
     
 
 
 
    if (!(iev%updfrq)) {
 
      c_wfm->Update();
 
      for (j=0;j<DWIDTH;j++) {
 
        achdat[i][j]/=nwfm[i];
 
        gr_wfm[i]->SetPoint(j,tcdata[i][j],achdat[i][j]+0.05*(i-1));
 
      }
 
      printf("processing event %d\n",iev
);  
    }
 
  
 
 
 
    
 
 
 
  }
 
 
 
 
 
  printf("number of events: %d\n",iev
);  
 
 
  TCanvas* c_dpo= new TCanvas("c_dpo","DPO",750,50,700,700);
 
  c_dpo->Divide(2,2);
 
  for (i=0;i<4;i++) {c_dpo->cd(i+1);h2_dpo[i]->Draw("colz");}
 
 
 
  TCanvas* c_cftdif= new TCanvas("c_cftdif","DPO",750,50,700,700);
 
  c_cftdif->Divide(2,2);
 
  for (i=0;i<4;i++) {c_cftdif->cd(i+1);h1_cftdif[i]->Draw();}
 
 
 
  return;
 
}