#include "ImageMPR.h"
#include "ui_ImageMPR.h"

#include <vtkProperty.h>
#include <vtkRenderWindow.h>

#include <QFileDialog>
#include <QMessageBox>

#ifdef _WIN32
#  define PLUGIN_PREFIX ""
#  define PLUGIN_EXT "dll"
#  define PLUGIN_REGEX "Plugins file (*.dll);;All files (*)"
#else
#  define PLUGIN_PREFIX "lib"
#  define PLUGIN_EXT "so"
#  define PLUGIN_REGEX "Plugins file (*.so);;All files (*)"
#endif // _WIN32

// -------------------------------------------------------------------------
ImageMPR::ImageMPR( QWidget* parent )
  : QMainWindow( parent ),
    m_UI( new Ui::ImageMPR ),
    m_ImageReaderClass( "" ),
    m_ImageWriterClass( "" ),
    m_MeshReaderClass( "" ),
    m_MeshWriterClass( "" ),
    m_MeshCutterClass( "" ),
    m_Image( NULL )
{
  this->m_UI->setupUi( this );

  // Create and associate renderers
  this->m_MPRObjects = vtkSmartPointer< TMPRObjects >::New( );
  this->m_MPRObjects->SetRenderWindows(
    this->m_UI->m_XPlaneVTK->GetRenderWindow( ),
    this->m_UI->m_YPlaneVTK->GetRenderWindow( ),
    this->m_UI->m_ZPlaneVTK->GetRenderWindow( ),
    this->m_UI->m_3DVTK->GetRenderWindow( )
    );

  // signals <-> slots
  QObject::connect(
    this->m_UI->actionOpenPlugins, SIGNAL( triggered( ) ),
    this, SLOT( _triggered_actionOpenPlugins( ) )
    );
  QObject::connect(
    this->m_UI->actionOpenInputImage, SIGNAL( triggered( ) ),
    this, SLOT( _triggered_actionOpenInputImage( ) )
    );
  QObject::connect(
    this->m_UI->actionOpenSegmentation, SIGNAL( triggered( ) ),
    this, SLOT( _triggered_actionOpenSegmentation( ) )
    );
  QObject::connect(
    this->m_UI->actionOpenInputPolyData, SIGNAL( triggered( ) ),
    this, SLOT( _triggered_actionOpenInputPolyData( ) )
    );

  // Start: load all disponible plugins
  this->_LoadPlugins(
    std::string( PLUGIN_PREFIX ) +
    std::string( "cpPluginsIO." ) +
    std::string( PLUGIN_EXT )
    );
  this->_LoadPlugins(
    std::string( PLUGIN_PREFIX ) +
    std::string( "cpPluginsBasicFilters." ) +
    std::string( PLUGIN_EXT )
    );
}

// -------------------------------------------------------------------------
ImageMPR::
~ImageMPR( )
{
  // Close all connections
  this->m_Plugins.UnloadAll( );

  // Delete objects
  delete this->m_UI;
}

// -------------------------------------------------------------------------
bool ImageMPR::
_LoadPlugins( const std::string& filename )
{
  QApplication::setOverrideCursor( Qt::WaitCursor );
  this->setEnabled( false );

  this->m_ImageReaderClass = "";
  this->m_ImageWriterClass = "";
  this->m_MeshReaderClass = "";
  this->m_MeshWriterClass = "";
  this->m_MeshCutterClass = "";
  this->m_UI->MenuImageToImage->clear( );
  this->m_UI->MenuImageToMesh->clear( );

  if( !( this->m_Plugins.Load( filename ) ) )
    return( false );

  typedef TPluginsInterface::TClasses _TClasses;
  _TClasses::const_iterator cIt = this->m_Plugins.GetClasses( ).begin( );
  for( ; cIt != this->m_Plugins.GetClasses( ).end( ); ++cIt )
  {
    TPluginFilter::Pointer o =
      this->m_Plugins.CreateProcessObject( cIt->first );
    std::string name = o->GetClassName( );
    std::string category = o->GetClassCategory( );
    if( category == "ImageReader" )
      this->m_ImageReaderClass = name;
    else if( category == "ImageWriter" )
      this->m_ImageWriterClass = name;
    else if( category == "MeshReader" )
      this->m_MeshReaderClass = name;
    else if( category == "MeshWriter" )
      this->m_MeshWriterClass = name;
    else if( category == "MeshToMeshFilter" )
    {
      if( name.find_last_of( "Cutter" ) != std::string::npos )
        this->m_MeshCutterClass = name;
    }
    else if( category == "ImageToImageFilter" )
    {
      QAction* action =
        this->m_UI->MenuImageToImage->addAction( QString( name.c_str( ) ) );
      QObject::connect(
        action, SIGNAL( triggered( ) ),
        this, SLOT( _triggered_actionImageToImage( ) )
        );
    }
    else if( category == "ImageToMeshFilter" )
    {
      QAction* action =
        this->m_UI->MenuImageToMesh->addAction( QString( name.c_str( ) ) );
      QObject::connect(
        action, SIGNAL( triggered( ) ),
        this, SLOT( _triggered_actionImageToMesh( ) )
        );

    } // fi

  } // rof
  QApplication::restoreOverrideCursor( );
  this->setEnabled( true );

  return( true );
}

// -------------------------------------------------------------------------
std::string ImageMPR::
_LoadImage( TPluginImage::Pointer& image )
{
  std::string ret = "";
  image = NULL;

  // Get a reader from loaded plugins
  TPluginFilter::Pointer reader =
    this->m_Plugins.CreateProcessObject( this->m_ImageReaderClass );
  if( reader.IsNotNull( ) )
  {
    if( reader->ExecConfigurationDialog( this ) )
    {
      // Block application
      QApplication::setOverrideCursor( Qt::WaitCursor );
      this->setEnabled( false );

      // Execute and get error message, if any
      ret = reader->Update( );

      // Assign fresh image, if any
      if( ret == "" )
      {
        image = reader->GetOutput< TPluginImage >( 0 );
        reader->DisconnectOutputs( );

      } // fi

      // Unblock application
      QApplication::restoreOverrideCursor( );
      this->setEnabled( true );

    } // fi
  }
  else
    ret = "No suitable reader object found in loaded plugins.";
  
  return( ret );
}

// -------------------------------------------------------------------------
std::string ImageMPR::
_ConfigureMeshActors( )
{
  if( this->m_Mesh.IsNull( ) )
    return( "Valid mesh not found." );

  this->m_Mesh->CreateVTKActor( );
  vtkActor* vtk_actor = this->m_Mesh->GetVTKActor( );
  if( vtk_actor != NULL )
  {
    this->m_MPRObjects->Get3DRenderer( )->AddActor( vtk_actor );
    this->m_MPRObjects->Render( 4 );

  } // fi

  TMPRObjects::TMPRActors* mprActors = this->m_MPRObjects->GetMPRActors( );

  std::string err = "";
  for( unsigned int i = 0; i < 3; ++i )
  {
    this->m_Cutters[ i ] = this->m_Plugins.CreateProcessObject( this->m_MeshCutterClass );
    this->m_Planes[ i ] = TPluginImplicitFunction::New( );
    this->m_Planes[ i ]->SetFunction( mprActors->GetSliceActors( i )->GetPlaneFunction( ) );
    this->m_Cutters[ i ]->SetInput( 0, this->m_Mesh );
    this->m_Cutters[ i ]->SetInput( 1, this->m_Planes[ i ] );
    std::string lerr = this->m_Cutters[ i ]->Update( );
    if( lerr == "" )
    {
      this->m_Cutters[ i ]->GetOutput< TPluginMesh >( 0 )->CreateVTKActor( );
      vtkActor* actor = this->m_Cutters[ i ]->GetOutput< TPluginMesh >( 0 )->GetVTKActor( );
      mprActors->GetSliceActors( i )->AddActor( this->m_Cutters[ i ]->GetVTK< vtkAlgorithm >( ), actor );
      if( i == 0 )
        this->m_MPRObjects->GetXRenderer( )->AddActor( actor );
      else if( i == 1 )
        this->m_MPRObjects->GetYRenderer( )->AddActor( actor );
      else if( i == 2 )
        this->m_MPRObjects->GetZRenderer( )->AddActor( actor );

    } // fi
    err += lerr;

  } // rof
  this->m_MPRObjects->RenderAll( );
  return( err );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionOpenPlugins( )
{
  // Show dialog and check if it was accepted
  QFileDialog dialog( this );
  dialog.setFileMode( QFileDialog::ExistingFile );
  dialog.setDirectory( "." );
  dialog.setNameFilter( tr( PLUGIN_REGEX ) );
  dialog.setDefaultSuffix( tr( PLUGIN_EXT ) );
  if( !( dialog.exec( ) ) )
    return;
  
  std::string fname = dialog.selectedFiles( ).at( 0 ).toStdString( );
  if( !( _LoadPlugins( fname ) ) )
    QMessageBox::critical(
      this,
      tr( "Ignoring plugin" ),
      tr( fname.c_str( ) )
      );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionOpenInputImage( )
{
  // Read image
  std::string err = this->_LoadImage( this->m_Image );
  if( err == "" )
  {
    vtkImageData* vtk_id = this->m_Image->GetVTK< vtkImageData >( );
    if( vtk_id != NULL )
    {
      this->m_MPRObjects->SetImage( vtk_id );
      this->m_MPRObjects->ActivateInteractors( );
      this->m_MPRObjects->ResetCameras( );
      this->m_MPRObjects->RenderAll( );
    }
    else
      QMessageBox::critical(
        this,
        tr( "Error message" ),
        tr( "Read image does not have a valid VTK converter." )
        );
  }
  else
    QMessageBox::critical(
      this,
      tr( "Error reading single image" ),
      tr( err.c_str( ) )
      );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionOpenSegmentation( )
{
  if( this->m_Image.IsNull( ) )
  {
    QMessageBox::critical(
      this,
      tr( "Error message" ),
      tr( "Before reading a segmentation, first load a raw image." )
      );
    return;

  } // fi

  // Read image
  std::string err = this->_LoadImage( this->m_Segmentation );
  if( err == "" )
  {
    vtkImageData* vtk_id = this->m_Segmentation->GetVTK< vtkImageData >( );
    if( vtk_id != NULL )
    {
      this->m_MPRObjects->AddAuxiliaryImage( vtk_id );
      this->m_MPRObjects->RenderAll( );
    }
    else
      QMessageBox::critical(
        this,
        tr( "Error message" ),
        tr( "Read image does not have a valid VTK converter." )
        );
  }
  else
    QMessageBox::critical(
      this,
      tr( "Error reading single image" ),
      tr( err.c_str( ) )
      );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionOpenInputPolyData( )
{
  this->m_Mesh = NULL;

  // Get a reader from plugins
  TPluginFilter::Pointer reader =
    this->m_Plugins.CreateProcessObject( this->m_MeshReaderClass );

  if( reader.IsNotNull( ) )
  {
    // Configure reader
    if( reader->ExecConfigurationDialog( this ) )
    {
      // Execute and get error message, if any
      QApplication::setOverrideCursor( Qt::WaitCursor );
      this->setEnabled( false );
      std::string err = reader->Update( );
      QApplication::restoreOverrideCursor( );
      this->setEnabled( true );

      // Assign fresh mesh, if any
      if( err == "" )
      {
        this->m_Mesh = reader->GetOutput< TPluginMesh >( 0 );
        reader->DisconnectOutputs( );
        err = this->_ConfigureMeshActors( );
        if( err != "" )
          QMessageBox::critical(
            this,
            tr( "Error message" ),
            tr( err.c_str( ) )
            );
      }
      else
        QMessageBox::critical(
          this,
          tr( "Error reading mesh" ),
          tr( err.c_str( ) )
          );

    } // fi
  }
  else
    QMessageBox::critical(
      this,
      tr( "Error reading single mesh" ),
      tr( "No suitable mesh reader found in loaded plugins." )
      );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionImageToImage( )
{
  if( this->m_Image.IsNull( ) )
    return;

  // Get filter name
  QAction* action = dynamic_cast< QAction* >( this->sender( ) );
  if( action == NULL )
    return;
  std::string name = action->text( ).toStdString( );

  // Configure filter
  TPluginFilter::Pointer filter =
    this->m_Plugins.CreateProcessObject( name );
  bool dlg_ok = filter->ExecConfigurationDialog( NULL );
  if( !dlg_ok )
    return;

  // Execute filter
  QApplication::setOverrideCursor( Qt::WaitCursor );
  this->setEnabled( false );
  filter->SetInput( 0, this->m_Image );
  std::string err = filter->Update( );
  QApplication::restoreOverrideCursor( );
  this->setEnabled( true );

  // Update image
  if( err == "" )
  {
    TPluginImage* result = filter->GetOutput< TPluginImage >( 0 );
    result->DisconnectPipeline( );
    this->m_Image = result;
    if( this->m_Image.IsNotNull( ) )
      this->m_MPRObjects->SetImage(
        this->m_Image->GetVTK< vtkImageData >( )
        );
  }
  else
    QMessageBox::critical(
      this,
      tr( "Error executing filter" ),
      tr( err.c_str( ) )
      );
}

// -------------------------------------------------------------------------
void ImageMPR::
_triggered_actionImageToMesh( )
{
  if( this->m_Image.IsNull( ) )
    return;

  // Get filter name
  QAction* action = dynamic_cast< QAction* >( this->sender( ) );
  if( action == NULL )
    return;
  std::string name = action->text( ).toStdString( );

  // Configure filter
  TPluginFilter::Pointer filter =
    this->m_Plugins.CreateProcessObject( name );
  bool dlg_ok = filter->ExecConfigurationDialog( NULL );
  if( !dlg_ok )
    return;

  // Execute filter
  QApplication::setOverrideCursor( Qt::WaitCursor );
  this->setEnabled( false );
  filter->SetInput( 0, this->m_Image );
  std::string err = filter->Update( );
  QApplication::restoreOverrideCursor( );
  this->setEnabled( true );

  // Update image
  if( err == "" )
  {
    TPluginMesh* result = filter->GetOutput< TPluginMesh >( 0 );
    result->DisconnectPipeline( );
    this->m_Mesh = result;
    err = this->_ConfigureMeshActors( );
    if( err != "" )
      QMessageBox::critical(
        this,
        tr( "Error message" ),
        tr( err.c_str( ) )
        );
  }
  else
    QMessageBox::critical(
      this,
      tr( "Error executing filter" ),
      tr( err.c_str( ) )
      );
}

// eof - $RCSfile$