PickTable.cxx

Go to the documentation of this file.

#include "PickTable.h"

#include "controllers/DisplayController.h"
#include "datareps/DataRep.h"
#include "datasrcs/DataSourceController.h"
#include "datasrcs/NTuple.h"
#include "datasrcs/NTupleSorter.h"
#include "graphics/Color.h"
#include "graphics/SymbolType.h"
#include "plotters/PlotterBase.h"

#if QT_VERSION < 0x040000
#include <qheader.h>
#include <qlistview.h>
#else
#include <q3header.h>
#include <q3listview.h>
#endif

#include <qpushbutton.h>
#include <qlabel.h>

using namespace hippodraw;

using std::string;
using std::vector;


PickTable::
PickTable ( PlotterBase * plotter )
{
  m_target = plotter;

  NTuple * ntuple = plotter -> createPickTuple ();
  m_sorter = new NTupleSorter ( ntuple );

  const string & name = ntuple -> getName ();
  m_title->setText ( name.c_str() );

  const DataRep * rep = plotter->getDataRep ( 0 );
  const string & type = rep -> name ();

  string text = "Plot type : ";
  text += type.c_str();
  m_type->setText ( text.c_str() );


#if QT_VERSION < 0x040000
  const vector < string > & labels = ntuple -> getLabels ();
  m_pick_table->setColumnText ( 1, labels[1].c_str() );
  m_pick_table->setColumnText ( 2, labels[2].c_str() );
  unsigned int size = ntuple -> columns ();
  if ( size == 3 ) {
    m_pick_table -> removeColumn ( 3 );
  }
  else {
    m_pick_table -> setColumnText ( 3, labels[3].c_str() );
    if ( size == 5 ) {
      const QString & str ( labels[4].c_str() );
      m_pick_table -> addColumn ( str );
    }
  }

#else
//   const vector < string > & labels = ntuple -> getLabels ();
//   m_pick_table->setColumnText ( 1, labels[1].c_str() );
//   m_pick_table->setColumnText ( 2, labels[2].c_str() );
//   unsigned int size = ntuple -> columns ();
//   if ( size == 3 ) {
//     m_pick_table -> removeColumn ( 3 );
//   }
//   else {
//     m_pick_table -> setColumnText ( 3, labels[3].c_str() );
//     if ( size == 5 ) {
//       const QString & str ( labels[4].c_str() );
//       m_pick_table -> addColumn ( str );
//     }
//   }

#endif
  DataSourceController::instance()->registerNTuple ( ntuple );

#if QT_VERSION < 0x040000
  QHeader * header = m_pick_table->header();
  connect ( header, SIGNAL ( clicked (int) ), 
            this, SLOT( listSorted ( int ) ) );
  m_pick_table->setSorting ( -1 );
  m_pick_table->setShowSortIndicator ( true );
  m_delete -> setEnabled ( false );
#else
//   Q3Header * header = m_pick_table->header();
//   connect ( header, SIGNAL ( clicked (int) ), 
//          this, SLOT( listSorted ( int ) ) );
//   m_pick_table->setSorting ( -1 );
//   m_pick_table->setShowSortIndicator ( true );
//   m_delete -> setEnabled ( false );
#endif

}

void
PickTable::
listSorted ( int i )
{
  m_sorter -> setSorting ( i );
  m_sorter -> sort ( );

#if QT_VERSION < 0x040000
  QListViewItem * item = m_pick_table ->selectedItem ();
  QString text = item -> text ( 0 );
  unsigned int selected = text.toUInt ();
  refreshItems ( selected );
#else
//   Q3ListViewItem * item = m_pick_table ->selectedItem ();
//   QString text = item -> text ( 0 );
//   unsigned int selected = text.toUInt ();
//   refreshItems ( selected );
#endif

}

void PickTable::clear()
{
  m_pick_table->clear();
  m_sorter->clear();
}

void
PickTable::
addItem ( std::vector < double > & v )
{
  unsigned int row = m_sorter -> rows ();
  v[0] = row;
  m_sorter -> addRow ( v );

  refreshItems ( row );
}

void
PickTable::
refreshItems ( unsigned int select )
{
  m_pick_table -> clear();

  unsigned int row = m_sorter->rows ();
  bool yes = row > 0;
  m_delete -> setEnabled ( yes );

#if QT_VERSION < 0x040000
  while ( row-- != 0 ) {
    const vector < double > & vec = m_sorter->getRow ( row );
    unsigned int current = static_cast < unsigned int > ( vec[0] );
    QListViewItem * item = new QListViewItem ( m_pick_table );
    unsigned int size = vec.size();

    for ( unsigned int i = 0; i < size; i++ ) {
      item -> setText ( i, QString ( "%1" ).arg ( vec[i] ) );
    }

    m_pick_table -> insertItem ( item );

    if ( select == current ) {
      m_pick_table -> setSelected ( item, true );
      m_pick_table -> ensureItemVisible ( item );
    }
  }
#else
//   while ( row-- != 0 ) {
//     const vector < double > & vec = m_sorter->getRow ( row );
//     unsigned int current = static_cast < unsigned int > ( vec[0] );
//     Q3ListViewItem * item = new Q3ListViewItem ( m_pick_table );
//     unsigned int size = vec.size();

//     for ( unsigned int i = 0; i < size; i++ ) {
//       item -> setText ( i, QString ( "%1" ).arg ( vec[i] ) );
//     }

//     m_pick_table -> insertItem ( item );

//     if ( select == current ) {
//       m_pick_table -> setSelected ( item, true );
//       m_pick_table -> ensureItemVisible ( item );
//     }
//   }
#endif
}

unsigned int
PickTable::
#if QT_VERSION < 0x040000
indexOf ( QListViewItem * target )
#else
indexOf ( Q3ListViewItem * target )
#endif
{
  unsigned int index = 0;
#if QT_VERSION < 0x040000
  QListViewItem * item = m_pick_table -> firstChild ();
  while ( item != target ) {
    index++;
    item = item->nextSibling();
    if ( item == 0 ) break;
  }

#else
//   Q3ListViewItem * item = m_pick_table -> firstChild ();
//   while ( item != target ) {
//     index++;
//     item = item->nextSibling();
//     if ( item == 0 ) break;
//   }

#endif

  return index;
}

void PickTable::deleteSelectedItem ()
{
#if QT_VERSION < 0x040000
  QListViewItem * item = m_pick_table->selectedItem ();
  if ( item != 0 ) {
    unsigned int index = indexOf ( item );

    m_sorter->eraseRow ( index );

    unsigned int row = index == 0 ? 0 : index - 1;
    refreshItems ( row );
  }
#else
//   Q3ListViewItem * item = m_pick_table->selectedItem ();
#endif
}

void
PickTable::
addDataRepButton ()
{
  const string plotTypeStr ( "Scatter Plot" );  
  NTuple * ntuple = m_sorter -> getNTuple ();
  const vector < string > & labels = ntuple -> getLabels ();
  vector < string > bindings;
  bindings.push_back ( labels[1] );
  bindings.push_back ( labels[2] );
  if ( labels.size() == 4 ) {
    if ( labels[3] == "Density" ) {
      bindings[1] = "Density";
    }
  }

  DisplayController * controller = DisplayController::instance();
  DataRep * rep 
    = controller -> addDataRep ( m_target, plotTypeStr, 
                                 ntuple, bindings );
  rep -> setRepColor ( Color::red );
  rep -> setRepStyle ( Symbol::CIRCLE );
  rep -> setRepSize ( 6.0 );
}

void
PickTable::
on_m_pick_table_selectionChanged ()
{
  QListViewItem * item = m_pick_table -> selectedItem ();
  bool yes = item != 0;

  m_delete -> setEnabled ( yes );
}

---