#include <cpExtensions/QT/MPR3DWidget.h>

#ifdef cpExtensions_QT4

#include <cpExtensions/Visualization/ImageOutlineActor.h>
#include <cpExtensions/Visualization/MPR3DActors.h>
#include <cpExtensions/Visualization/LUTImageActor.h>
#include <cpExtensions/Visualization/WindowLevelImageActor.h>

#include <vtkActor.h>
#include <vtkImageData.h>
#include <vtkPolyData.h>

// -------------------------------------------------------------------------
cpExtensions::QT::MPR3DWidget::
MPR3DWidget( QWidget* parent, Qt::WindowFlags f )
  : Superclass( parent, f ),
    m_ImageName( "" )
{
  this->m_Actors =
    vtkSmartPointer< cpExtensions::Visualization::MPR3DActors >::New( );
}

// -------------------------------------------------------------------------
cpExtensions::QT::MPR3DWidget::
~MPR3DWidget( )
{
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetImage( vtkImageData* image, const std::string& name )
{
  if( this->m_ImageName != "" )
  {
    // TODO: Clear visualization

  } // fi

  this->m_Actors->SetImage( image );
  this->m_ImageName = name;

  vtkSmartPointer< vtkPropCollection > props =
    vtkSmartPointer< vtkPropCollection >::New( );
  props->AddItem( this->m_Actors->GetWindowLevelImageActor( 0 ) );
  props->AddItem( this->m_Actors->GetWindowLevelImageActor( 1 ) );
  props->AddItem( this->m_Actors->GetWindowLevelImageActor( 2 ) );
  this->AddViewProps( props, name );
  this->AddAuxViewProp( this->m_Actors->GetImageOutlineActor( 0 ), name );
  this->AddAuxViewProp( this->m_Actors->GetImageOutlineActor( 1 ), name );
  this->AddAuxViewProp( this->m_Actors->GetImageOutlineActor( 2 ), name );
  this->ResetCamera( );
  this->Render( );
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
Add( vtkDataSet* data, const std::string& name )
{
  auto image = dynamic_cast< vtkImageData* >( data );
  auto pdata = dynamic_cast< vtkPolyData* >( data );
  if( image != NULL )
  {
    if( this->m_ImageName != "" )
    {
    }
    else
      this->SetImage( image, name );
  }
  else if( pdata != NULL )
  {
    this->AddViewProp( this->m_Actors->AddMesh( pdata ), name );
    this->Render( );
    if( this->m_ImageName == "" )
      this->ResetCamera( );

  } // fi

  /* TODO
     return;

     auto image = dynamic_cast< vtkImageData* >( data );
     auto pdata = dynamic_cast< vtkPolyData* >( data );
     if( image != NULL )
     {
     this->m_Actors->AddLUTImage( image );
     vtkSmartPointer< vtkPropCollection > coll =
     vtkSmartPointer< vtkPropCollection >::New( );
     coll->AddItem( this->m_Actors->GetLUTImageActor( 0 ) );
     coll->AddItem( this->m_Actors->GetLUTImageActor( 1 ) );
     coll->AddItem( this->m_Actors->GetLUTImageActor( 2 ) );
     this->AddViewProps( coll, name );
     this->Render( );
     }
     else if( pdata != NULL )
     {

     } // fi
  */
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetSliceNumber( int orientation, int slice )
{
  this->m_Actors->SetSliceNumber( orientation, slice );
  this->Render( );
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetSlicesNumbers( int x, int y, int z )
{
  this->m_Actors->SetSliceNumber( 0, x );
  this->m_Actors->SetSliceNumber( 1, y );
  this->m_Actors->SetSliceNumber( 2, z );
  this->Render( );
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetScalarRange( double r[ 2 ] )
{
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetWindowLevel( double wl[ 2 ] )
{
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetImageOpacity( double o )
{
}

// -------------------------------------------------------------------------
void cpExtensions::QT::MPR3DWidget::
SetImageInterpolation( unsigned char i )
{
}

#endif // cpExtensions_QT4

// eof - $RCSfile$