#ifndef __CINT__
#include <TSQLServer.h>
#include <TSQLResult.h>
#include <TSQLRow.h>
#endif

#include <math>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
#include <string>
#include "TString.h"

using namespace std;

void cr8pideff(string arm = "left") {

  const char* dbname = "mysql://jmysql/hapvdis";
  const char* username = "hapvdis";
  const char* userpass = "to<9uLkp";

  TSQLServer *db = TSQLServer::Connect(dbname, username, userpass);

  printf("Server info: %s\n", db->ServerInfo());

  cout<<"Database is: "<<db->GetDB()<<endl;

  TSQLResult *res;

  vector <Float_t> DateOfChange, Correction;

  TString inputfile;
  TString outfile;

  if (arm == "left") {
    inputfile = "PIDEff_L.txt";
    outfile = "PIDCorL.txt";
  }
  else if (arm == "right") {
    inputfile = "PIDEff_R.txt";
    outfile = "PIDCorR.txt";
  }

  cout<<inputfile<<"\t"<<outfile<<endl;

  ifstream input(inputfile, ios::in);
  ofstream output(outfile, ios::out);

  if (!input.good()) {
    cout<<"Unable to open the input file: "<<inputfile<<endl;
  }


  Float_t datetmp, cortmp;

  while (input.good()) {
    input>>datetmp>>cortmp;
    DateOfChange.push_back(datetmp*100);
    Correction.push_back(cortmp);
  }

  Int_t vectorsize = Correction.size();

  vector <Float_t> Error (vectorsize);

  for (Int_t i = 0; i < (Int_t)  Correction.size(); i ++) {

    if (i == 0)
      Error[i] = fabs(Correction[i] - Correction[i+1]);
    else 
      Error[i] = fabs(Correction[i] - Correction[i-1]);

    if (Error[i] == 0)
      Error[i] = Error[i-1];
  }

  TString sql = "select run_num, date, start_time from runlist";

  res = db->Query(sql);

  const Int_t nrows = res->GetRowCount();
  cout<<"total rows: "<<nrows<<endl;


  Int_t *runnum = new Int_t [nrows];
  Float_t *PID = new Float_t [nrows];
  Float_t *PIDErr = new Float_t [nrows];
  string *date = new string [nrows];
  string *start_time = new string [nrows];

  for (Int_t i = 0; i < nrows; i++) {

    TSQLRow *row = res->Next();
    runnum[i] = atoi(row->GetField(0));
    date[i] = row->GetField(1);
    start_time[i] = row->GetField(2);

    Int_t monthtmp = atoi(((date[i]).substr(5,2)).c_str());
    Int_t daytmp = atoi((date[i]).substr(8,2).c_str());
    Int_t hourtmp = atoi((start_time[i]).substr(0,2).c_str());

    cout<<"run: "<<runnum[i]<<", date: "<<date[i]<<", start time: "<<start_time[i]<<endl;
    cout<<"month: "<<monthtmp<<", day: "<<daytmp<<", hour: "<<hourtmp<<endl;

    Float_t TheTime = monthtmp*10000 + daytmp*100 + hourtmp;

    Int_t index = 0;
    for (Int_t j = 0; j < (Int_t) DateOfChange.size(); j++) {
      if (TheTime > DateOfChange[j])
	index = j+1;
    }

    PID[i] = Correction[index];
    PIDErr[i] = Error[index];

    cout<<runnum[i]<<"\t"<<PID[i]<<"\t"<<PIDErr[i]<<endl;
    output<<fixed;
    output<<runnum[i]<<"\t"<<setprecision(5)<<PID[i]<<"\t"<<setprecision(4)<<PIDErr[i]*100<<endl;

    delete row;

  }

  delete res;
  delete [] runnum;
  delete [] PID;
  delete [] PIDErr;
  delete [] date;
  delete [] start_time;

}