// -*- C++ -*-
//
// This file is part of the Coriolis Software.
// Copyright (c) UPMC/LIP6 2008-2008, All Rights Reserved
//
// ===================================================================
//
// $Id$
//
// x-----------------------------------------------------------------x 
// |                                                                 |
// |                  H U R R I C A N E                              |
// |                                                                 |
// |  Author      :                    Jean-Paul CHAPUT              |
// |  E-mail      :       Jean-Paul.Chaput@asim.lip6.fr              |
// | =============================================================== |
// |  C++ Module  :       "./SelectionWidget.cpp"                    |
// | *************************************************************** |
// |  U p d a t e s                                                  |
// |                                                                 |
// x-----------------------------------------------------------------x


#include  <QFontMetrics>
#include  <QFrame>
#include  <QLabel>
#include  <QCheckBox>
#include  <QPushButton>
#include  <QLineEdit>
#include  <QHeaderView>
#include  <QKeyEvent>
#include  <QGroupBox>
#include  <QHBoxLayout>
#include  <QVBoxLayout>
#include  <QAction>

#include "hurricane/Commons.h"
#include "hurricane/Cell.h"

#include "hurricane/viewer/Graphics.h"
#include "hurricane/viewer/SelectionModel.h"
#include "hurricane/viewer/SelectionWidget.h"
#include "hurricane/viewer/CellWidget.h"


namespace Hurricane {



  SelectionWidget::SelectionWidget ( QWidget* parent )
    : QWidget               (parent)
    , _cellWidget           (NULL)
    , _baseModel            (new SelectionModel(this))
    , _sortModel            (new QSortFilterProxyModel(this))
    , _view                 (new QTableView(this))
    , _filterPatternLineEdit(new QLineEdit(this))
    , _cumulative           (new QCheckBox())
    , _showSelection        (new QCheckBox())
    , _rowHeight            (20)
    , _updateState          (ExternalEmit)
  {
    setAttribute ( Qt::WA_DeleteOnClose );
    setAttribute ( Qt::WA_QuitOnClose, false );
    setContextMenuPolicy ( Qt::ActionsContextMenu );

    _rowHeight = QFontMetrics(Graphics::getFixedFont()).height() + 4;

    QLabel* filterPatternLabel = new QLabel(tr("&Filter pattern:"), this);
    filterPatternLabel->setBuddy(_filterPatternLineEdit);

    QHBoxLayout* hLayout1 = new QHBoxLayout ();
    hLayout1->addWidget ( filterPatternLabel );
    hLayout1->addWidget ( _filterPatternLineEdit );

    QFrame* separator = new QFrame ();
    separator->setFrameShape  ( QFrame::HLine );
    separator->setFrameShadow ( QFrame::Sunken );

    _cumulative->setText    ( tr("Cumulative Selection") );
    _cumulative->setChecked ( false );

    _showSelection->setText    ( tr("Show Selection") );
    _showSelection->setChecked ( false );

    QPushButton* clear = new QPushButton ();
    clear->setText ( tr("Clear") );

    QHBoxLayout* hLayout2 = new QHBoxLayout ();
    hLayout2->addWidget  ( _cumulative );
    hLayout2->addStretch ();
    hLayout2->addWidget  ( _showSelection );
    hLayout2->addStretch ();
    hLayout2->addWidget  ( clear );

    _sortModel->setSourceModel       ( _baseModel );
    _sortModel->setDynamicSortFilter ( true );
    _sortModel->setFilterKeyColumn   ( 1 );

    _view->setShowGrid(false);
    _view->setAlternatingRowColors(true);
    _view->setSelectionBehavior(QAbstractItemView::SelectRows);
    _view->setSortingEnabled(true);
    _view->setModel ( _sortModel );
    _view->horizontalHeader()->setStretchLastSection ( true );
    _view->installEventFilter(this);

    QHeaderView* horizontalHeader = _view->horizontalHeader ();
    horizontalHeader->setStretchLastSection ( true );
  //horizontalHeader->setMinimumSectionSize ( 200 );

    QHeaderView* verticalHeader = _view->verticalHeader ();
    verticalHeader->setVisible ( false );

    QVBoxLayout* vLayout = new QVBoxLayout ();
  //vLayout->setSpacing ( 0 );
    vLayout->addLayout ( hLayout2 );
    vLayout->addWidget ( separator );
    vLayout->addWidget ( _view );
    vLayout->addLayout ( hLayout1 );

    setLayout ( vLayout );

    connect ( _filterPatternLineEdit, SIGNAL(textChanged(const QString &))
            , this                  , SLOT  (textFilterChanged()) );

    connect ( _showSelection, SIGNAL(toggled(bool))  , this      , SLOT(setShowSelection(bool)) );
    connect ( _cumulative   , SIGNAL(toggled(bool))  , this      , SLOT(setCumulativeSelection(bool)) );
    connect ( clear         , SIGNAL(clicked())      , _baseModel, SLOT(clear()) );
    connect ( clear         , SIGNAL(clicked())      , this      , SLOT(clear()) );

    connect ( _view->selectionModel(), SIGNAL(currentChanged(const QModelIndex&,const QModelIndex&))
            , this                   , SLOT  (selectCurrent (const QModelIndex&,const QModelIndex&)) );

    QAction* toggleAction = new QAction  ( tr("&Toggle Selection"), this );
    toggleAction->setShortcut   ( QKeySequence(tr("T")) );
    toggleAction->setStatusTip  ( tr("Toggle the selection state of this Occurrence") );
    addAction ( toggleAction );

    QAction* inspectAction = new QAction  ( tr("&Inspect"), this );
    inspectAction->setShortcut   ( QKeySequence(tr("I")) );
    inspectAction->setStatusTip  ( tr("Load this Occurrence in the Inspector") );
    addAction ( inspectAction );

    connect ( toggleAction , SIGNAL(triggered()), this, SLOT(toggleSelection ()) );
    connect ( inspectAction, SIGNAL(triggered()), this, SLOT(inspect()) );

    setWindowTitle ( tr("Selection<None>") );
    resize ( 500, 300 );
  }


  void  SelectionWidget::setCellWidget ( CellWidget* cw )
  {
    if ( _cellWidget ) {
      disconnect ( _cellWidget, 0, this       , 0 );
      disconnect ( this       , 0, _cellWidget, 0 );
    }

    _cellWidget = cw;
    if ( !_cellWidget ) return;

    connect ( _cellWidget, SIGNAL(selectionModeChanged()), this, SLOT(changeSelectionMode()) );

    connect ( _cellWidget, SIGNAL(selectionChanged(const SelectorSet&))
            , this       , SLOT  (setSelection    (const SelectorSet&)) );

    connect ( _cellWidget, SIGNAL(selectionToggled(Occurrence)), this, SLOT(toggleSelection(Occurrence)) );

    _updateState = ExternalEmit;
    changeSelectionMode ();
  }


  void  SelectionWidget::changeSelectionMode ()
  {
    if ( !_cellWidget ) return;

    if ( _updateState == InternalEmit ) {
      _updateState = InternalReceive;
      emit selectionModeChanged ();
    } else {
      if ( _updateState == ExternalEmit ) {
        blockAllSignals ( true );

        _showSelection->setChecked ( _cellWidget->getState()->showSelection      () );
        _cumulative   ->setChecked ( _cellWidget->getState()->cumulativeSelection() );

        blockAllSignals ( false );
      }
      _updateState = ExternalEmit;
    }
  }


  void  SelectionWidget::blockAllSignals ( bool state )
  {
    _showSelection->blockSignals ( state );
    _cumulative   ->blockSignals ( state );
    _baseModel    ->blockSignals ( state );
  }


  void  SelectionWidget::setShowSelection ( bool state )
  {
    _updateState = InternalEmit;
    _cellWidget->setShowSelection ( state );
  }


  void  SelectionWidget::setCumulativeSelection ( bool state )
  {
    _updateState = InternalEmit;
    _cellWidget->setCumulativeSelection ( state );
  }


  bool  SelectionWidget::cumulativeSelection () const
  {
    return _cumulative->isChecked ();
  }


  void  SelectionWidget::textFilterChanged ()
  {
    _sortModel->setFilterRegExp ( _filterPatternLineEdit->text() );
  }


  void  SelectionWidget::toggleSelection ()
  {
    toggleSelection ( _view->currentIndex() );
  }


  void  SelectionWidget::toggleSelection ( const QModelIndex& index )
  {
    if ( index.isValid() ) {
      Occurrence occurrence = _baseModel->toggleSelection ( _sortModel->mapToSource(index) );
      if ( occurrence.isValid() ) {
        _updateState = InternalEmit;
        _cellWidget->toggleSelection ( occurrence );
      }
    }
  }


  void  SelectionWidget::toggleSelection ( Occurrence occurrence )
  {
    if ( _updateState != InternalEmit ) {
      blockAllSignals ( true );
      _baseModel->toggleSelection ( occurrence );
      blockAllSignals ( false );
    }
    _updateState = ExternalEmit;
  }


  void  SelectionWidget::setSelection ( const SelectorSet& selection )
  {
    _baseModel->setSelection ( selection );
     
    string windowTitle = "Selection";
    Cell*  cell = (_cellWidget) ? _cellWidget->getCell() : NULL;
    if ( cell ) windowTitle += getString(cell);
    else        windowTitle += "<None>";
    setWindowTitle ( tr(windowTitle.c_str()) );

    _view->selectRow ( 0 );
    _view->resizeColumnToContents ( 0 );
  }


  void  SelectionWidget::clear ()
  {
    _baseModel->clear ();
    _view->selectionModel()->clearSelection ();
    if ( _cellWidget )
      _cellWidget->unselectAll();
  }


  void  SelectionWidget::selectCurrent ( const QModelIndex& current, const QModelIndex& )
  {
    inspect ( current );
  }


  void  SelectionWidget::inspect ()
  {
    inspect ( _view->currentIndex() );
  }


  void  SelectionWidget::inspect ( const QModelIndex& index  )
  {
    if ( index.isValid() ) {
      Occurrence occurrence = _baseModel->getOccurrence ( _sortModel->mapToSource(index).row() );
      emit inspect ( occurrence );
    } else
      emit inspect ( NULL );
  }


} // End of Hurricane namespace.