#include <cpPlugins/Extensions/Visualization/ImageInteractorStyle.h>

#include <vtkAnnotatedCubeActor.h>
#include <vtkAxesActor.h>
#include <vtkCamera.h>
#include <vtkCommand.h>
#include <vtkPropAssembly.h>
#include <vtkProperty.h>
#include <vtkRendererCollection.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>

#include <cpPlugins/Extensions/Visualization/ImageSliceActors.h>
#include <cpPlugins/Extensions/Visualization/MPRActors.h>

// -------------------------------------------------------------------------
const int cpPlugins::Extensions::Visualization::
ImageInteractorStyle::SliceEvent = vtkCommand::UserEvent + 1;

// -------------------------------------------------------------------------
cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Self* cpPlugins::Extensions::Visualization::ImageInteractorStyle::
New( )
{
  return( new Self( ) );
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Configure( ImageSliceActors* slice_actors, MPRActors* mpr_actors )
{
  this->m_SliceActors = slice_actors;
  this->m_MPRActors = mpr_actors;
  this->SetModeToNavigation( );
  this->PropPicker->AddPickList( slice_actors->GetImageActor( ) );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
SetModeToNavigation( )
{
  this->Mode = Self::NavigationMode;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
SetModeToDeformation( )
{
  this->Mode = Self::DeformationMode;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
SetInteractor( vtkRenderWindowInteractor* interactor, const int& axis )
{
  this->Superclass::SetInteractor( interactor );
  this->OrientationWidget->SetInteractor( interactor );
  interactor->SetInteractorStyle( this );
  if( interactor == NULL )
    return;

  // TODO: interactor->SetPicker( this->PropPicker );

  // Get camera, avoiding segfaults
  vtkRenderer* ren =
    interactor->GetRenderWindow( )->GetRenderers( )->GetFirstRenderer( );
  if( ren == NULL )
    return;
  vtkCamera* cam = ren->GetActiveCamera( );
  if( cam == NULL )
    return;

  // Parallel projections are better when displaying 2D images
  cam->ParallelProjectionOn( );
  cam->SetFocalPoint( double( 0 ), double( 0 ), double( 0 ) );
  if( axis == 0 )
  {
    cam->SetPosition( double( 1 ), double( 0 ), double( 0 ) );
    cam->SetViewUp  ( double( 0 ), double( 1 ), double( 0 ) );
  }
  else if( axis == 1 )
  {
    cam->SetPosition( double( 0 ), double( 1 ), double(  0 ) );
    cam->SetViewUp  ( double( 0 ), double( 0 ), double( -1 ) );
  }
  else // if( axis == 2 )
  {
    cam->SetPosition( double( 0 ), double( 0 ), double( 1 ) );
    cam->SetViewUp  ( double( 0 ), double( 1 ), double( 0 ) );

  } // fi
  ren->ResetCamera( );

  // Enable 2D orientation widget
  this->OrientationWidget->SetEnabled( 1 );
  this->OrientationWidget->InteractiveOff( );
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnMouseMove( )
{
  std::cout << "moumov" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnLeftButtonDown( )
{
  std::cout << "leftdown" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnLeftButtonUp( )
{
  std::cout << "leftup" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnMiddleButtonDown( )
{
  std::cout << "middown" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnMiddleButtonUp( )
{
  std::cout << "midup" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnRightButtonDown( )
{
  std::cout << "ridown" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnRightButtonUp( )
{
  std::cout << "riup" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnMouseWheelForward( )
{
  std::cout << "whfwd" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnMouseWheelBackward( )
{
  std::cout << "whbwd" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
OnChar( )
{
  std::cout << "char" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Rotate( )
{
  std::cout << "Rotate" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Pan( )
{
  std::cout << "pan" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Spin( )
{
  std::cout << "spin" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Zoom( )
{
  std::cout << "Zoom" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
WindowLevel( )
{
  std::cout << "wl" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Pick( )
{
  std::cout << "Pick" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
Slice( )
{
  std::cout << "Slice" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
StartWindowLevel( )
{
  std::cout << "swl" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
EndWindowLevel( )
{
  std::cout << "ewl" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
StartPick( )
{
  std::cout << "sp" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
EndPick( )
{
  std::cout << "ep" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
StartSlice( )
{
  std::cout << "ss" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
EndSlice( )
{
  std::cout << "es" << std::endl;
}

// -------------------------------------------------------------------------
cpPlugins::Extensions::Visualization::ImageInteractorStyle::
ImageInteractorStyle( )
  : Superclass( ),
    Mode( Self::NavigationMode ),
    m_SliceActors( NULL ),
    m_MPRActors( NULL )
{
  // Orientation marks
  vtkSmartPointer< vtkAnnotatedCubeActor > cube =
    vtkSmartPointer< vtkAnnotatedCubeActor >::New( );
  cube->GetCubeProperty( )->SetColor( 0.9, 0.7, 0.2 );
  cube->GetTextEdgesProperty( )->SetLineWidth( 1 );
  cube->GetTextEdgesProperty( )->SetDiffuse( 0 );
  cube->GetTextEdgesProperty( )->SetAmbient( 1 );
  cube->GetTextEdgesProperty( )->SetColor( 0.18, 0.28, 0.23 );
  cube->GetXPlusFaceProperty( )->SetColor( 1, 0, 0 );
  cube->GetXPlusFaceProperty( )->SetInterpolationToFlat( );
  cube->GetXMinusFaceProperty( )->SetColor( 1, 0, 0 );
  cube->GetXMinusFaceProperty( )->SetInterpolationToFlat( );
  cube->GetYPlusFaceProperty( )->SetColor( 0, 1, 0 );
  cube->GetYPlusFaceProperty( )->SetInterpolationToFlat( );
  cube->GetYMinusFaceProperty( )->SetColor( 0, 1, 0 );
  cube->GetYMinusFaceProperty( )->SetInterpolationToFlat( );
  cube->GetZPlusFaceProperty( )->SetColor( 0, 0, 1 );
  cube->GetZPlusFaceProperty( )->SetInterpolationToFlat( );
  cube->GetZMinusFaceProperty( )->SetColor( 0, 0, 1 );
  cube->GetZMinusFaceProperty( )->SetInterpolationToFlat( );

  vtkSmartPointer< vtkAxesActor > axes =
    vtkSmartPointer< vtkAxesActor >::New( );
  axes->AxisLabelsOff( );
  axes->SetShaftTypeToCylinder( );
  axes->SetTotalLength( 2, 2, 2 );

  vtkSmartPointer< vtkPropAssembly > actors =
    vtkSmartPointer< vtkPropAssembly >::New( );
  actors->AddPart( cube );
  actors->AddPart( axes );

  this->OrientationWidget =
    vtkSmartPointer< vtkOrientationMarkerWidget >::New( );
  this->OrientationWidget->SetOutlineColor( 0.93, 0.57, 0.13 );
  this->OrientationWidget->SetOrientationMarker( actors );
  this->OrientationWidget->SetViewport( 0.0, 0.0, 0.2, 0.2 );

  this->PropPicker = vtkSmartPointer< vtkPropPicker >::New( );
  this->PropPicker->PickFromListOn( );
}

// -------------------------------------------------------------------------
cpPlugins::Extensions::Visualization::ImageInteractorStyle::
~ImageInteractorStyle( )
{
  std::cout << "destructor" << std::endl;
}

// -------------------------------------------------------------------------
bool cpPlugins::Extensions::Visualization::ImageInteractorStyle::
_PickPosition( double pos[ 3 ] )
{
  std::cout << "pickpos" << std::endl;
}

// -------------------------------------------------------------------------
void cpPlugins::Extensions::Visualization::ImageInteractorStyle::
_UpdateCursor( )
{
  std::cout << "upcur" << std::endl;
}

// eof - $RCSfile$