// compile with:
// g++ -o plotchipped.exe plotchipped.cpp `root-config --cflags --glibs`
// TODO move csvfile wrting before hsignal->setaxis.
//
#include "TH1F.h"
#include "TProfile.h"
#include "TMath.h"
#include "TF1.h"
#include "TLegend.h"
#include "TCanvas.h"
 //
#include "TROOT.h"
//#include "TStyle.h"
#include "TTree.h"
#include "TFile.h"
#include "TGraphErrors.h"
#include "TStyle.h"
#include <string>
#include <cmath>
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
//
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

using namespace std;

int plotchipped(int mcm = 0,const int iEvent = 1,int chip = 0, int saltronb = 0, int gain = 0,const char *dir = "",const char *filename = "", const char *title = "")
{
  char name[100];
  unsigned int ErrorCode[16] = {0};
  unsigned int WarningCode[16] = {0};
  int item = 0;
  Char_t iChan; // Channel number
  Char_t iChip; // Channel number
  UShort_t iMcm; // MCM/FEC number
  UShort_t iClTime; // Cluster time
  UShort_t iClLength; //number of samples
  UInt_t iEventnb;
  Int_t iSamples[1024]; // Samples
  Int_t iMax = 0;
  TH1F* hEvent[16];
  TH1F* hSignal[16];
  float ped_min_mean[3] = {0};
  float ped_max_mean[3] = {0};
  float ped_max_rms[3] = {0};
  TCanvas *c1 = new TCanvas("c1","Single cluster",600,600);

  sprintf(name, "%s/%s.root",dir,filename);
  TFile *inFile = new TFile(name);
  TTree *tree = (TTree*) inFile->Get("tree");
  if(!tree) {
    cerr << "Tree SALTRO events was not found: " << name << "\n";
    return 1;
  }

  FILE *limitfile;
  sprintf(name, "%s.txt","plotchipped");
  limitfile = fopen(name,"r");
  if (limitfile == NULL) {
    return 4;
  }
  while (fgets(name,sizeof(name),limitfile) != NULL) {
    if (name[0] != '#') {
      switch(item) {
      case 0: sscanf(name,"%f %f %f",&ped_min_mean[0],&ped_max_mean[0],&ped_max_rms[0]); break;
      case 1: sscanf(name,"%f %f %f",&ped_min_mean[1],&ped_max_mean[1],&ped_max_rms[1]); break;
      case 2: sscanf(name,"%f %f %f",&ped_min_mean[2],&ped_max_mean[2],&ped_max_rms[2]); break;
      case 3: sscanf(name,"%f %f %f",&ped_min_mean[3],&ped_max_mean[3],&ped_max_rms[3]); break;
      }
      item++;
    }
  }
  fclose(limitfile);
  if (item == 4) {
    printf("%f %f %f\n",ped_min_mean[0],ped_max_mean[0],ped_max_rms[0]);
    printf("%f %f %f\n",ped_min_mean[1],ped_max_mean[1],ped_max_rms[1]);
    printf("%f %f %f\n",ped_min_mean[2],ped_max_mean[2],ped_max_rms[2]);
    printf("%f %f %f\n",ped_min_mean[3],ped_max_mean[3],ped_max_rms[3]);
  }
  else return 5;

  //  tree->Show(iEntry);
  tree->SetBranchAddress("iEventnb",&iEventnb);
  tree->SetBranchAddress("iSamples",&iSamples);
  tree->SetBranchAddress("iChan",&iChan);
  tree->SetBranchAddress("iChip",&iChip);
  tree->SetBranchAddress("iMcm",&iMcm);
  tree->SetBranchAddress("iClLength",&iClLength);
  tree->SetBranchAddress("iClTime",&iClTime);

  Int_t nentries = tree->GetEntries(); 
  printf("ENTRIES %d\n",nentries);
  for (Int_t j = 0; j < 16; j++) {
    sprintf(name, "hEvent_%d", j);
    hEvent[j] = ((TH1F *)(gROOT->FindObject(name)));
    if (hEvent[j]) delete hEvent[j];
    hEvent[j] = new TH1F(name, "",1024,-0.5,1023.5);
    hEvent[j]->SetMinimum(0);
    hEvent[j]->SetStats(0);
  }

  for (Int_t j = 0; j < 16; j++) {
    sprintf(name, "hSignal_%d", j);
    hSignal[j] = ((TH1F *)(gROOT->FindObject(name)));
    if (hSignal[j]) delete hSignal[j];
    hSignal[j] = new TH1F(name, "",1024,-0.5,1023.5);
    hSignal[j]->SetMinimum(0);
  }
  gStyle->SetOptStat(1100);
  for (Int_t j = 0; j < nentries; j++) {
    tree->GetEntry(j);
    if ((iMcm == mcm) && (iChip == chip)) {
      if (iEventnb == iEvent) {
	for(Int_t l = 0; l < iClLength; l++) {
	  hEvent[iChan]->AddBinContent(iClTime-l, iSamples[l]);
	}
      }

      for(Int_t l = 20; l < iClLength - 20; l++) {
	hSignal[iChan]->Fill(iSamples[l]);
      }
    }
  }
  // Print in pedestal file format for chip=7
  FILE *fped;
  sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-%s.ped",dir,saltronb,saltronb,mcm,chip,title);
  fped = fopen(name,"w");
  for (Int_t j = 0; j < 16; j++) {  
    printf("%d\t%d\t%5.2f\t%5.3f\t%5.1f\n",
	   j+iChip*16,(int)hSignal[j]->GetMean(), hSignal[j]->GetMean(),hSignal[j]->GetRMS(),hSignal[j]->GetEntries());
    fprintf(fped,"%d\t%d\t%5.2f\t%5.3f\t%5.1f\n",
	    j+iChip*16,(int)hSignal[j]->GetMean(), hSignal[j]->GetMean(),hSignal[j]->GetRMS(),hSignal[j]->GetEntries());
    if ((hSignal[j]->GetMean() < ped_min_mean[gain]) || (hSignal[j]->GetMean() > ped_max_mean[gain])) ErrorCode[j] |= 0x0001;
    if (hSignal[j]->GetRMS() > ped_max_rms[gain]) ErrorCode[j] |= 0x0002;
  }
  fclose(fped);
  Int_t oldLevel = gErrorIgnoreLevel;
  gErrorIgnoreLevel = kWarning;
  for (Int_t j = 0; j < 16; j++) {
    c1->cd(iChan+1);
    sprintf(name, "Event %d Channel %d",iEvent,j);
    hEvent[j]->SetTitle(name);
    hEvent[j]->GetXaxis()->SetTitle("Sample #");
    hEvent[j]->GetYaxis()->SetTitle("ADC value");
    hEvent[j]->Draw();
    if (j == 0) sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-event%d-%s.pdf(",dir,saltronb,saltronb,mcm,chip,iEvent,title);
    else if ( j == 15) sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-event%d-%s.pdf)",dir,saltronb,saltronb,mcm,chip,iEvent,title);
    else  sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-event%d-%s.pdf",dir,saltronb,saltronb,mcm,chip,iEvent,title);
    c1->SaveAs(name,"pdf");
  }
  
  for (Int_t j = 0; j < 16; j++) {  
    c1->cd(j+1);
    sprintf(name, "Pedestals ch %d chip %d %s", j,chip,title);
    hSignal[j]->SetTitle(name);
    hSignal[j]->GetYaxis()->SetTitle("Entries");
    hSignal[j]->GetXaxis()->SetTitle("ADC value");
    //    hSignal[j]->SetAxisRange(0, 1023, "X");
    hSignal[j]->SetAxisRange(0, 200, "X");
    hSignal[j]->Draw();
    printf("%d\t%5.2f\t%5.3f\t%5.1f\n",j+iChip*16,hSignal[j]->GetMean(),hSignal[j]->GetRMS(),hSignal[j]->GetEntries());
    if (j == 0) sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-%s.pdf(",dir,saltronb,saltronb,mcm,chip,title);
    else if (j==15)  sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-%s.pdf)",dir,saltronb,saltronb,mcm,chip,title);
    else sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-%s.pdf",dir,saltronb,saltronb,mcm,chip,title);
    c1->SaveAs(name);
  }
  gErrorIgnoreLevel = oldLevel;
 
  FILE *csvfile;
  sprintf(name,"%s/chip%d/saltro%d-mcm%d-c%d-pedestal-%s.csv",dir,saltronb,saltronb,mcm,chip,title);
  //  sprintf(name, "%s/%s.csv",dir,filename);

  struct stat   fileBuffer;
  int csvexist = 0;
  if (stat (name, &fileBuffer) == 0) csvexist = 1;
  printf("CSV EXIST %d\n",csvexist);
  csvfile = fopen(name,"a");
  if (csvfile == NULL) {
    return 3;
  }

  if (csvexist == 0) {
    fprintf(csvfile,"Channel,PedMean,PedRMS,PedEntries,Error,Warning\n");
  }
  for (int j = 0; j < 16; j++) {
    hSignal[j]->SetAxisRange(0, 1023, "X");
    fprintf(csvfile,"%d,%6.2f,%6.3f,%6.1f,",j+iChip*16,hSignal[j]->GetMean(),hSignal[j]->GetRMS(),hSignal[j]->GetEntries());
    fprintf(csvfile,"%08X,%08X\n",ErrorCode[j],WarningCode[j]);
  }
  fclose(csvfile);

  
  return 0;
}

int main (int argc, char *argv[])
{
  int iret = 0;
  if (argc!=9) {
    printf("not enough arguments: argc = %d\nSyntax should be:\n./plotchipped.exe eventnb saltro mcm chip gain dir filename title\n",argc);
    return -1;
  }
  int eventnb = atoi(argv[1]);
  int saltronb = atoi(argv[2]);
  int mcm = atoi(argv[3]);
  int chip = atoi(argv[4]);
  int gain = atoi(argv[5]);
  iret = plotchipped(mcm,eventnb,chip,saltronb,gain,argv[6],argv[7],argv[8]);
  return iret;
}