/// Class to plot gas properties inside TPC
/// over a particular time period (or run).
/// Blair Jamieson (C) 2010
#include "ITPCGasAnalysis.hxx"
#include <list>
#include <TGraph.h>
#include <TFile.h>
#include <TCanvas.h>
#include <TDatime.h>

/// Initialize TPC gas properties averaging object using unix time
ITPCGasAnalysis::ITPCGasAnalysis( int maxsamples, long autmin,  long autmax) {

  fTimeMin = autmin;
  fTimeMax = autmax;
  fNSam = maxsamples;
  Process();
}
  
/// Initialize TPC gas properties averaging object using year,
/// month(0-11), day (1-31), ...
ITPCGasAnalysis:: ITPCGasAnalysis(int maxsamples,
				  int ayearmin, int amonthmin, int adaymin, int ahourmin, int aminmin, int asecmin,
				  int ayearmax, int amonthmax, int adaymax, int ahourmax, int aminmax, int asecmax ) {
  fNSam = maxsamples;

  tm mt;  
  // use time structure from time.h
  // to build unix time from year,month, etc.
  mt.tm_year = ayearmin - 1900;
  mt.tm_mon  = amonthmin-1;
  mt.tm_mday = adaymin;
  mt.tm_hour = ahourmin;
  mt.tm_min  = aminmin;
  mt.tm_sec  = asecmin;
  mt.tm_isdst = -1;
  fTimeMin = (long)mktime(&mt);
  
  mt.tm_year = ayearmax - 1900;
  mt.tm_mon  = amonthmax-1;
  mt.tm_mday = adaymax;
  mt.tm_hour = ahourmax;
  mt.tm_min  = aminmax;
  mt.tm_sec  = asecmax;
  mt.tm_isdst = -1;
  fTimeMax = (long)mktime(&mt);

  Process();
}


///----------------------------------------------------------------
/// Get TGraph time offset due to day light savings time
/// messing things up
int ITPCGasAnalysis::GetDSTOffset( long atime ){
  // convert atime to the kind needed by TGraph
  // set conversion time. Values of 1 hour
  // and 2 hour offset seemed to work.
  tm mt;
  localtime_r( (const time_t*) &atime, &mt );
  //std::cout<<"GetDSTOffset atime="<<atime<<" dst="<<mt.tm_isdst<<std::endl;
  int aTOffset=0;
  if ( mt.tm_isdst == 0 ) aTOffset=3600;
  else if ( mt.tm_isdst == 1 ) aTOffset=7200;

  return aTOffset;
}

/// Do the analysis of gas system quantities.
void ITPCGasAnalysis::Process(){
  // setup simple arrays of the quantities to plot

  // times relative to 2008 01 01 00 00 00 
  TDatime T0(2008,01,01,00,00,00);
  int X0 = T0.Convert();
  gStyle->SetTimeOffset(X0);
  TDatime tdtime;
  
  std::vector< float > alltimes;
  // temperatures
  int nTvars = 8;
  const char * Tnames[] = {    "O2 Analyzer T(degC)",
			       "D0TC3 Purifier 1 T(degC)",
			       "D0TC3 Purifier 2 T(degC)",
			       "Patm+dP4(mbar)",
			       "P4/TO2 (mbar/K)",
			       "Patm(mbar)",
			       "Avg TPC Gas Manifold T(degC)",
			       "Patm/TO2(mbar/K)"};
  
  std::vector< float > allT[ 8 ];

  // gas analysis quantities
  int nAnVars = 10;
  const char * Annames[] = { "H2O (ppm)", 
			     "O2 (ppm)",
			     "Ibu Mon (%)",
			     "CO2 (ppm)",
			     "CF4 (%)",
			     "VdIn",
			     "VdOut",
			     "GainIn",
			     "GainOut",
			     "VdInUnc"};

  std::vector< float > allAn[ 10 ];

  // flow fractions
  int nFlows = 3;
  const char * Flownames [] = { "Fraction Ar (%)",
				"Fraction CF4 (%)",
				"Fraction iC4H10 (%)"};
  std::vector< float > allFlow[3];


  // first calculate delta-time for maximum number of samples:
  long adt;
  if ( fNSam >0 ){
    adt = (fTimeMax-fTimeMin)/fNSam;
    if (adt<30) adt = 30; // readings are only every so often anyway so set minimum to every 30 seconds
  } else {
    // use a default of every 30 sec?
    adt = 30; 
  }

  // Build Gas property reader
  ITPCGas aGas( fTimeMin, fTimeMax );
  ITPCGasSmooth aGasSm( fTimeMin, fTimeMax, 7200 ); // use some smoothed quantities
  long atime = aGas.GetTime();
  long astart_time;
  long aend_time;
  aGas.GetTimeInterval( astart_time, aend_time );
  int nquery = 0;
  do {
    if (nquery%1 == 0 ){
      std::cout<<"DB Query"<<nquery
	       <<" time = "<<aGas.GetTimeStamp();
    }

    // Fill vectors with the values being plotted
    alltimes.push_back( float( int(atime-X0+GetDSTOffset( atime ) )) );
    (allT[0]).push_back( aGasSm.T(atime) );
    (allT[1]).push_back( aGas.P0TC3() );
    (allT[2]).push_back( aGas.P0TC6() );
    (allT[3]).push_back( aGas.PD4()+aGas.Patm()*1000.0  )    ;
    (allT[4]).push_back( (aGas.PD4()+aGas.Patm()*1000.0)/(273.15+aGas.TO2()) );
    (allT[5]).push_back( aGasSm.P(atime) );
    (allT[6]).push_back( aGas.TTPC(-1) );
    (allT[7]).push_back( aGasSm.Density(atime) );
    (allAn[0]).push_back( aGasSm.H2O(atime) );
    (allAn[1]).push_back( aGasSm.O2(atime) );
    (allAn[2]).push_back( aGasSm.IbuMon(atime) );
    (allAn[3]).push_back( aGasSm.CO2(atime) );
    (allAn[4]).push_back( aGasSm.CF4(atime) );
    (allAn[5]).push_back( aGasSm.VdIn(atime) );
    (allAn[6]).push_back( aGasSm.VdOut(atime) );
    (allAn[7]).push_back( aGasSm.GainIn(atime) );
    (allAn[8]).push_back( aGasSm.GainOut(atime) );    
    (allAn[9]).push_back( aGas.VdInUnc() );    
    (allFlow[0]).push_back( 100.0 * aGasSm.ArFrac(atime) );
    (allFlow[1]).push_back( 100.0 * aGasSm.CF4Frac(atime) );
    (allFlow[2]).push_back( 100.0 * aGasSm.IbuFrac(atime) );

    aGas.ReadNext( adt );
    atime = aGas.GetTime();
    aGas.GetTimeInterval( astart_time, aend_time );
    nquery++;
  } while ( aend_time < fTimeMax );

  TFile fout("ttpcgasanalysis.root","recreate");
  fout.cd();
  
  std::list<TGraph*> tgs;
  std::list<TCanvas*> tcs;

  // Graphs vs time
  for (int i=0; i<nTvars; i++){
    TGraph * tg = new TGraph( alltimes.size(), &(alltimes[0]), &(allT[i][0]));
    tg->SetMarkerStyle(7);
    tg->SetMarkerColor(kBlue);
    char tgname[256];
    sprintf( tgname, "%s vs Time; Time; %s",Tnames[i],Tnames[i]);
    tg->SetTitle(tgname);
    sprintf( tgname,"Tvst%02d",i);
    tg->SetName(tgname);
    tg->GetXaxis()->SetTimeDisplay(1);
    tg->GetXaxis()->SetTimeFormat("#splitline{%b-%d}{%H:%M}");
    tg->GetXaxis()->SetLabelOffset(0.03);
    tg->GetXaxis()->SetTimeOffset( X0, "gmt");
    tg->GetXaxis()->SetNdivisions(507);
    tg->GetXaxis()->SetLabelSize(0.04);
    tgs.push_back( tg );
    sprintf( tgname,"tcTvst%02d",i);
    TCanvas * tc = new TCanvas(tgname, tgname);
    tc->cd();
    tg->Draw("AP");
    tcs.push_back( tc );
  }
  for (int i=0; i<nAnVars; i++){
    TGraph * tg = new TGraph( alltimes.size(), &(alltimes[0]), &(allAn[i][0]) );
    tg->SetMarkerStyle(7);
    tg->SetMarkerColor(kBlue);
    char tgname[256];
    sprintf( tgname, "%s vs Time; Time; %s",Annames[i],Annames[i]);
    tg->SetTitle(tgname);
    sprintf( tgname,"Vvst%02d",i);
    tg->SetName(tgname);;
    tg->GetXaxis()->SetTimeDisplay(1);
    tg->GetXaxis()->SetTimeFormat("#splitline{%b-%d}{%H:%M}");
    tg->GetXaxis()->SetLabelOffset(0.03);
    tg->GetXaxis()->SetTimeOffset( X0, "gmt");
    tg->GetXaxis()->SetNdivisions(507);
    tg->GetXaxis()->SetLabelSize(0.04);
    tgs.push_back( tg );

    sprintf( tgname,"tcVvst%02d",i);
    TCanvas * tc = new TCanvas(tgname,tgname);
    tc->cd();
    tg->Draw("AP");
    tcs.push_back( tc );
  }

  // Graphs vs temperatures
  for (int itemp=0; itemp<nTvars; itemp++){
    for (int ian=0; ian<nAnVars; ian++){
      TGraph * tg = new TGraph( (allT[itemp]).size(), &(allT[itemp][0]), &(allAn[ian][0]) );
      tg->SetMarkerStyle(7);
      tg->SetMarkerColor(kBlue);
      char tgname[256];
      sprintf( tgname, "%s vs %s; %s; %s", Annames[ian], Tnames[itemp], Tnames[itemp], Annames[ian] );
      tg->SetTitle(tgname);
      sprintf( tgname,"V%02dvsT%02d",ian,itemp);
      tg->SetName(tgname);
      tgs.push_back( tg );
      sprintf( tgname,"tcV%02dvsT%02d",ian,itemp);
      TCanvas * tc = new TCanvas(tgname,tgname);
      tc->cd();
      tg->Draw("AP");
      tcs.push_back( tc );
    }
  }


  // Last added plots of fractional Flows vs time
  for (int i=0; i<nFlows; i++){
    TGraph * tg = new TGraph( alltimes.size(), &(alltimes[0]), &(allFlow[i][0]) );
    tg->SetMarkerStyle(7);
    tg->SetMarkerColor(kBlue);
    char tgname[256];
    sprintf( tgname, "%s vs Time; Time; %s",Flownames[i],Flownames[i]);
    tg->SetTitle(tgname);
    sprintf( tgname,"Flowvst%02d",i);
    tg->SetName(tgname);
    tg->GetXaxis()->SetTimeDisplay(1);
    tg->GetXaxis()->SetTimeFormat("#splitline{%b-%d}{%H:%M}");
    tg->GetXaxis()->SetLabelOffset(0.03);
    tg->GetXaxis()->SetTimeOffset( X0, "gmt");
    tg->GetXaxis()->SetNdivisions(507);
    tg->GetXaxis()->SetLabelSize(0.04);
    tgs.push_back( tg );
    sprintf( tgname,"tcFlowvst%02d",i);
    TCanvas * tc = new TCanvas(tgname,tgname);
    tc->cd();
    tg->Draw("AP");
    tcs.push_back( tc );
  }


  // write plots to file
  fout.cd();
  for ( list<TGraph*>::iterator it = tgs.begin(); it != tgs.end(); it++) {
    (*it)->Write();
  }
  char apngname[256];
  for ( list<TCanvas*>::iterator it = tcs.begin(); it != tcs.end(); it++) {
    sprintf(apngname,"%s.png",(*it)->GetName());
    (*it)->Print(apngname);
    (*it)->Write();
  }

  // close file
  fout.Close();

  // make second file of plots vs time
  aGasSm.Plot();


}