src/PtView.cc

#include "PtView.hh"
#include "LogViewer.hh"

#include <TVirtualPad.h>
#include <TH1.h>
#include <TAxis.h>
#include <TLegend.h>
#include <TPgSQLServer.h>
#include <TSQLResult.h>
#include <TSQLRow.h>

#include <cmath>
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

using namespace std;

PtView::PtView() : 
  vector< PtViewEntry >( 0 ),
  name_( "" ), title_( "" ), lv_( NULL ),
  pad_( NULL ), frame_( NULL ),
  ymin_( 0.0 ), ymax_( 1.0 ), xmin_( 0.0 ), xmax_( 100.0 ),
  startIndex_( 0 )
{
  this->addTimeEntries();
}

PtView::PtView( const string& name ) : 
  vector< PtViewEntry >( 0 ),
  name_( name ), title_( "" ), lv_( NULL ),
  pad_( NULL ), frame_( NULL ),
  ymin_( 0.0 ), ymax_( 1.0 ), xmin_( 0.0 ), xmax_( 100.0 ),
  startIndex_( 0 )
{
  this->addTimeEntries();
}

PtView::PtView( const PtView& view ) : 
  vector< PtViewEntry >( view ),
  name_( view.name_ ), title_( view.title_ ), lv_( view.lv_ ),
  pad_( view.pad_ ), frame_( view.frame_ ),
  ymin_( view.ymin_ ), ymax_( view.ymax_ ), xmin_( view.xmin_ ), xmax_( view.xmax_ ),
  startIndex_( view.startIndex_ )
{
}

PtView::~PtView() {
}

PtView& PtView::operator=( const PtView& view ){
  if( this != &view ){
    name_  = view.name_;
    title_ = view.title_;
    lv_    = view.lv_;
    pad_   = view.pad_;
    frame_ = view.frame_;
    ymin_  = view.ymin_;
    ymax_  = view.ymax_;
    xmin_  = view.xmin_;
    xmax_  = view.xmax_;
    startIndex_ = view.startIndex_;
    this->resize( view.size() );
    for( int i = 0; i < view.size(); i++ ) (*this)[ i ] = view[ i ];
  }
  return *this;
}

void PtView::addTimeEntries(){
  
  this->add( ( name_ == LogViewer::tgCoolingData ?
               "EXTRACT( EPOCH FROM ( to_timestamp( tbl_coolingdata.datetime ) ) )" :
               LogViewer::sqlEpoch ),
             "epoch", kBlack );
  
  this->add( ( name_ == LogViewer::tgCoolingData ?
               "EXTRACT( TIMEZONE FROM ( to_timestamp( tbl_coolingdata.datetime ) ) )"  :
               LogViewer::sqlTimeZone ),
             "TZ", kBlack );
  
  startIndex_ = this->size();
}

void PtView::setRange(){
  
  double x;
  ymin_ = ( pad_->GetLogy() ? 0.1 : 0.0 );
  ymax_ = 1.0;
  
  if( this->size() > startIndex_ ){
    
    TGraph& g = (*this)[ startIndex_ ].graph();
    
    int nPoint = g.GetN();
    g.GetPoint( 0, x, ymin_ );
    ymax_ = ymin_;
    
    for( int i = startIndex_; i < this->size(); i++ ){
      for( int j = 0; j < nPoint; j++ ){
        double y;
        (*this)[ i ].graph().GetPoint( j, x, y );
        ymin_ = ( y < ymin_ ? y : ymin_ );
        ymax_ = ( y > ymax_ ? y : ymax_ );
      }
    }
  }
  
  double yh = ( pad_->GetLogy() ? 
                log10( ymax_ ) - log10( ymin_ ) :
                ymax_ - ymin_ );
  
  ymin_ = ( pad_->GetLogy() ?
            pow( 10, log10( ymin_ ) - 0.1 * yh ) : 
            ymin_ - 0.1 * yh );
  
  ymax_ = ( pad_->GetLogy() ?
            pow( 10, log10( ymax_ ) + 0.1 * yh ) : 
            ymax_ + 0.1 * ( ymax_ - ymin_ ) );
  
}

void PtView::drawLegend(){
  if( this->size() <= startIndex_ ) return;
  TLegend *leg = new TLegend( 0.92, 0.02, 1.0, 0.97 );
  leg->SetFillColor( kWhite );
  for( int i = startIndex_; i < this->size(); i++ ){
    TGraph& g = (*this)[ i ].graph();
    leg->AddEntry( &g, (*this)[ i ].name().c_str(), "L" );
  }
  leg->Draw();
}

void PtView::drawFrame(){

  xmin_ = lv_->tmin();
  xmax_ = lv_->tmax();
  
  frame_ = pad_->DrawFrame( xmin_, ymin_, xmax_, ymax_ );
  
  double lsize   = lv_->labelSize();
  double tsize   = lv_->titleSize();
  double toffset = lv_->titleOffset();
  
  TAxis* xaxis = frame_->GetXaxis();
  xaxis->SetTimeDisplay( 1 );
  xaxis->SetTimeFormat( "%H:%M" );
  xaxis->SetLabelSize( lsize );
  
  TAxis* yaxis = frame_->GetYaxis();
  yaxis->CenterTitle( true );
  yaxis->SetTitle( title_.c_str() );
  yaxis->SetTitleSize( tsize );
  yaxis->SetTitleOffset( toffset );
  yaxis->SetLabelSize( lsize );

}

void PtView::drawGraph(){
  for( int i = startIndex_; i < this->size(); i++ ){
    TGraph& g = (*this)[ i ].graph();
    g.Draw( "L" );
  }
}

void PtView::draw(){
  
  if( pad_ == NULL ) return;
  pad_->cd();
  
  this->setRange();
  this->drawFrame();
  this->drawLegend();
  this->drawGraph();
  
}


void PtView::query( LogViewer *lv ){
  
  lv_ = lv;
  
  TSQLServer* srv = lv_->server();
  if( srv == NULL ){
    cout << "server has not been initialized yet." << endl;
    return;
  }

  cout << "# loading data for " << this->name() << endl;
  
  int nField = this->size();
  if( this->size() < startIndex_ + 1 ) return;
  
  vector< PtViewEntry >& ent = (*this);
  
  
  string sql( "SELECT " );
  
  for( int i = 0; i < nField; i++ )
    sql += ent[ i ].field() + string( i != nField - 1 ? "," : " " );
  
  sql += "FROM " + this->name() + " ";
  
  string tstamp = 
    ( this->name() != LogViewer::tgCoolingData ? 
      "datetime" :
      "to_timestamp( tbl_coolingdata.datetime )" );
  
  sql += 
    string( "WHERE " + tstamp + " >='"  ) + lv_->startTime() +
    string( "' AND " + tstamp + " <= '" ) + lv_->endTime() +
    string( "'" );
  
  //  cout << "SQL: " << sql << endl;
  
  TSQLResult *res = srv->Query( sql.c_str() );
  if( res == NULL ) {
    cerr << "Failed to get Query results from Server." << endl;
    return;
  }
  
  int nRow = res->GetRowCount();
  for( int i = startIndex_; i < nField; i++ ) ent[ i ].graph().Set( nRow );

  // -- output file
  string output = string( lv_->output() + "_" + this->name() + ".dat" );
  cout << "Opening: " << output << endl;
  ofstream ofs( output.c_str() );
  
  // -- header
  ofs << "# Table View:   " << this->name()    << endl;
  ofs << "# Start:        " << lv_->startTime() << endl;
  ofs << "# End:          " << lv_->endTime()   << endl;
  ofs << "#" << flush;
  ofs << setw(12) << "datetime" << endl;
  for( int i = startIndex_; i < nField; i++ ) ofs << setw(12) << ent[ i ].name() << flush;
  ofs << endl;
  
  TSQLRow *row;
  int counter( 0 );
  while( row = res->Next() ){
    ofs << " " << flush;
    
    lv_->toffset() = atof( row->GetField( 1 ) );
    double t = atof( row->GetField( 0 ) ) - lv_->toffset();
    
    ofs << setw(12) << (int)t << flush;
    for( int i = 2; i < nField; i++ ){
      ofs << setw(12) << row->GetField( i ) << flush;
      ent[ i ].graph().SetPoint( counter, 
                                     t, atof( row->GetField( i ) ) );
    }
    ofs << endl;
    counter++;
  }
  
  ofs.close();
}

void PtView::analysis(){
}

void PtView::add( const string& field, const string& title, const int& color ){
  this->push_back( PtViewEntry() );
  this->back().set( field, title, color );
}