src/PtEventView.cc

#include "PtEventView.hh"
#include "LogViewer.hh"

#include <TVirtualPad.h>
#include <TH1.h>
#include <TAxis.h>
#include <TLegend.h>
#include <TLatex.h>


#include <sqlite3.h>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

using namespace std;

map< double, std::string > PtEventView::events;

int PtEventView::parseEventLog( void *NotUsed, 
                               int argc, char **argv, char **azColName ){
  if( argc != 5 ) return 1;
  PtEventView::events[ atof( argv[ 0 ] ) / 1000.0 ] = argv[ 1 ];
  return 0;
}

PtEventView::PtEventView() :
  PtView( "ptlog" ), latex_( NULL )
{
  this->title() = "Event";
  latex_ = new TLatex;
}

PtEventView::~PtEventView(){
}

void PtEventView::setRange(){
  ymin_ = 0.0;
  ymax_ = 1.0;
}

void PtEventView::drawGraph(){
  string output = lv_->output() + "_event.dat";
  cout << "Opening: " << output << endl;
  ofstream ofs( output.c_str() );

  double toff = lv_->toffset();
  
  latex_->SetTextAngle( 60.0 );
  latex_->SetTextColor( kBlack );
  latex_->SetTextSize( 0.8 * lv_->titleSize() );

  for( map< double, string >::iterator itr = PtEventView::events.begin();
       itr != PtEventView::events.end();
       itr++ ){
    
    double t = itr->first - toff;
    latex_->DrawLatex( t, 0.05, itr->second.c_str() );
    ofs << setw(12) << (int) t
        << "  "
        << itr->second << endl;
  }
}

void PtEventView::drawLegend(){

  latex_->SetTextAngle( 0.0 );
  latex_->SetTextColor( kGray );
  latex_->SetTextSize( 0.8 * lv_->titleSize() );
  latex_->DrawLatex( lv_->tmin() + 0.1 * ( lv_->tmax() - lv_->tmin() ), 0.9, 
                     string( lv_->startTime() + " - " +
                             lv_->endTime() ).c_str() );
  
}

void PtEventView::query( LogViewer *lv ){
  
  lv_ = lv;

  // open event log file
  char *zErrMsg = 0;
  sqlite3 *ppDp;
  if( sqlite3_open( lv_->eventLog().c_str(), &ppDp ) != SQLITE_OK ) {
    cerr << "Failed to open and connect DB: " 
         << lv_->eventLog()
         << endl;
    return;
  } 
  
  ostringstream ostr;
  ostr << "select * from " << this->name() << " "
       << "where datetime > " << 1000 * ( lv_->tmin() + lv_->toffset() )
       << " and datetime < "  << 1000 * ( lv_->tmax() + lv_->toffset() );
  
  if( sqlite3_exec( ppDp, ostr.str().c_str(),
                    PtEventView::parseEventLog, 
                    0, 
                    &zErrMsg ) != SQLITE_OK ){
    cerr << "Failed to exec SQL to event log file" << endl;
  }
  
  // close log file
  if( sqlite3_close( ppDp ) != SQLITE_OK ){
    cerr << "Failed to close DB file" << endl;
  }

}