#include <cpExtensions/Visualization/ImageSliceActors.h>

#include <cmath>
#include <sstream>

#include <vtkAlgorithmOutput.h>
#include <vtkCamera.h>
#include <vtkCellArray.h>
#include <vtkImageData.h>
#include <vtkInformation.h>
#include <vtkPlane.h>
#include <vtkPoints.h>
#include <vtkProperty.h>
#include <vtkRenderer.h>
#include <vtkRendererCollection.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkStreamingDemandDrivenPipeline.h>
#include <vtkTextProperty.h>
#include <vtkWindowLevelLookupTable.h>

// -------------------------------------------------------------------------
cpExtensions::Visualization::ImageSliceActors*
cpExtensions::Visualization::ImageSliceActors::
New( )
{
  return( new Self( ) );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
AddInputConnection( vtkAlgorithmOutput* aout, int axis )
{
  // Get input vtkImageData
  if( aout == NULL )
    return;
  vtkAlgorithm* producer = aout->GetProducer( );
  vtkImageData* data = dynamic_cast< vtkImageData* >(
    producer->GetOutputDataObject( aout->GetIndex( ) )
    );
  if( data == NULL )
    return;

  // Try to infere if input comes from a color mapping filter
  vtkImageMapToColors* new_map =
    dynamic_cast< vtkImageMapToColors* >( producer );
  if( new_map == NULL )
  {
    // Configure LUT, if possible (NULL is returned if not)
    this->_ConfigureNewLUT( data );
    new_map = *( this->m_ImageMaps.rbegin( ) );
    if( new_map != NULL )
    {
      new_map->SetInputConnection( aout );
      new_map->Update( );

    } // fi
  }
  else
    this->m_ImageMaps.push_back( new_map );

  // Update window level values
  if( new_map != NULL )
  {
    double range[ 2 ];
    dynamic_cast< vtkImageData* >( new_map->GetInput( ) )->
      GetScalarRange( range );
    this->m_MinWindow = double( 0 );
    this->m_MaxWindow = range[ 1 ] - range[ 0 ];
    this->m_MinLevel = range[ 0 ];
    this->m_MaxLevel = range[ 1 ];

  } // fi

  // Create mapper and actors
  vtkSmartPointer< vtkImageSliceMapper > mapper =
    vtkSmartPointer< vtkImageSliceMapper >::New( );
  if( new_map != NULL )
    mapper->SetInputConnection( new_map->GetOutputPort( ) );
  else
    mapper->SetInputConnection( aout );
  this->m_SliceMappers.push_back( mapper );
  this->_ConfigureNewInput( axis );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
AddInputData( vtkImageData* data, int axis )
{
  // Update window level values
  if( this->m_ImageMaps.size( ) == 0 )
  {
    double range[ 2 ];
    data->GetScalarRange( range );
    this->m_MinWindow = double( 0 );
    this->m_MaxWindow = range[ 1 ] - range[ 0 ];
    this->m_MinLevel = range[ 0 ];
    this->m_MaxLevel = range[ 1 ];

  } // fi

  // Configure LUT, if possible (NULL is returned if not)
  this->_ConfigureNewLUT( data );
  vtkImageMapToColors* new_map = *( this->m_ImageMaps.rbegin( ) );
  if( new_map != NULL )
  {
    new_map->SetInputData( data );
    new_map->Update( );

  } // fi

  // Create mapper and actors
  vtkSmartPointer< vtkImageSliceMapper > mapper =
    vtkSmartPointer< vtkImageSliceMapper >::New( );
  if( new_map != NULL )
    mapper->SetInputConnection( new_map->GetOutputPort( ) );
  else
    mapper->SetInputData( data );
  this->m_SliceMappers.push_back( mapper );
  this->_ConfigureNewInput( axis );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
Clear( )
{
  // Reset values
  this->m_MinWindow = double( 0 );
  this->m_MaxWindow = double( 0 );
  this->m_MinLevel = double( 0 );
  this->m_MaxLevel = double( 0 );

  // Unbind from container
  this->RemoveAllItems( );

  // Delete all images
  this->m_ImageMaps.clear( );
  this->m_SliceMappers.clear( );
  this->m_ImageActors.clear( );

  // Reconfigure unique objects
  this->m_Cursor        = vtkSmartPointer< vtkPolyData >::New( );
  this->m_CursorMapper  = vtkSmartPointer< vtkPolyDataMapper >::New( );
  this->m_CursorActor   = vtkSmartPointer< vtkActor >::New( );
  this->m_Plane         = vtkSmartPointer< vtkPolyData >::New( );
  this->m_PlaneMapper   = vtkSmartPointer< vtkPolyDataMapper >::New( );
  this->m_TextActor     = vtkSmartPointer< vtkTextActor >::New( );
  this->m_PlaneActor    = vtkSmartPointer< vtkActor >::New( );
  this->m_TextBuffer[ 0 ] = '\0';

  // Unique objects configuration
  vtkSmartPointer< vtkPoints > cursor_points =
    vtkSmartPointer< vtkPoints >::New( );
  vtkSmartPointer< vtkCellArray > cursor_lines =
    vtkSmartPointer< vtkCellArray >::New( );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_points->InsertNextPoint( 0, 0, 0 );
  cursor_lines->InsertNextCell( 2 );
  cursor_lines->InsertCellPoint( 0 );
  cursor_lines->InsertCellPoint( 1 );
  cursor_lines->InsertNextCell( 2 );
  cursor_lines->InsertCellPoint( 2 );
  cursor_lines->InsertCellPoint( 3 );
  cursor_lines->InsertNextCell( 2 );
  cursor_lines->InsertCellPoint( 4 );
  cursor_lines->InsertCellPoint( 5 );
  cursor_lines->InsertNextCell( 2 );
  cursor_lines->InsertCellPoint( 6 );
  cursor_lines->InsertCellPoint( 7 );
  this->m_Cursor->SetPoints( cursor_points );
  this->m_Cursor->SetLines( cursor_lines );
  this->m_CursorMapper->SetInputData( this->m_Cursor );
  this->m_CursorActor->SetMapper( this->m_CursorMapper );

  vtkSmartPointer< vtkPoints > plane_points =
    vtkSmartPointer< vtkPoints >::New( );
  vtkSmartPointer< vtkCellArray > plane_lines =
    vtkSmartPointer< vtkCellArray >::New( );

  plane_points->InsertNextPoint( 0, 0, 0 );
  plane_points->InsertNextPoint( 0, 1, 0 );
  plane_points->InsertNextPoint( 1, 1, 0 );
  plane_points->InsertNextPoint( 1, 0, 0 );
  plane_lines->InsertNextCell( 5 );
  plane_lines->InsertCellPoint( 0 );
  plane_lines->InsertCellPoint( 1 );
  plane_lines->InsertCellPoint( 2 );
  plane_lines->InsertCellPoint( 3 );
  plane_lines->InsertCellPoint( 0 );
  this->m_Plane->SetPoints( plane_points );
  this->m_Plane->SetLines( plane_lines );

  this->m_PlaneMapper->SetInputData( this->m_Plane );
  this->m_PlaneActor->SetMapper( this->m_PlaneMapper );

  this->m_TextActor->SetTextScaleModeToNone( );
  vtkTextProperty* textprop = this->m_TextActor->GetTextProperty( );
  textprop->SetColor( 1, 1, 1 );
  textprop->SetFontFamilyToCourier( );
  textprop->SetFontSize( 18 );
  textprop->BoldOff( );
  textprop->ItalicOff( );
  textprop->ShadowOff( );
  textprop->SetJustificationToLeft( );
  textprop->SetVerticalJustificationToBottom( );
  vtkCoordinate* coord = this->m_TextActor->GetPositionCoordinate( );
  coord->SetCoordinateSystemToNormalizedViewport( );
  coord->SetValue( 0.01, 0.01 );

  // Update actor collection
  this->AddItem( this->m_CursorActor );
  this->AddItem( this->m_TextActor );
  this->AddItem( this->m_PlaneActor );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
AssociateSlice( Self* slice )
{
  this->m_AssociatedSlices.push_back( slice );
  this->Modified( );
}

// -------------------------------------------------------------------------
vtkInteractorStyle* cpExtensions::Visualization::ImageSliceActors::
GetStyle( )
{
  return( this->m_Style.GetPointer( ) );
}

// -------------------------------------------------------------------------
const vtkInteractorStyle* cpExtensions::Visualization::ImageSliceActors::
GetStyle( ) const
{
  return( this->m_Style.GetPointer( ) );
}

// -------------------------------------------------------------------------
vtkImageData* cpExtensions::Visualization::ImageSliceActors::
GetInputImage( unsigned int id )
{
  vtkAlgorithmOutput* aout = this->m_ImageMaps[ id ]->GetOutputPort( );
  vtkImageData* image = dynamic_cast< vtkImageData* >(
    aout->GetProducer( )->GetOutputDataObject( aout->GetIndex( ) )
    );
  return( image );
}

// -------------------------------------------------------------------------
const vtkImageData* cpExtensions::Visualization::ImageSliceActors::
GetInputImage( unsigned int id ) const
{
  vtkAlgorithmOutput* aout = this->m_ImageMaps[ id ]->GetOutputPort( );
  const vtkImageData* image = dynamic_cast< const vtkImageData* >(
    aout->GetProducer( )->GetOutputDataObject( aout->GetIndex( ) )
    );
  return( image );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
PushActorsInto( vtkRenderWindow* window, bool force_style )
{
  this->m_Window = window;
  if( window == NULL )
    return;
  vtkRenderWindowInteractor* rwi = window->GetInteractor( );
  vtkRenderer* renderer = window->GetRenderers( )->GetFirstRenderer( );
  if( rwi == NULL || renderer == NULL )
    return;

  // Update style
  if( this->m_Style.GetPointer( ) != NULL && force_style )
    rwi->SetInteractorStyle( this->m_Style );

  // Update actors
  vtkProp* prop;
  this->InitTraversal( );
  while( prop = this->GetNextProp( ) )
    renderer->AddViewProp( prop );
  renderer->Modified( );
  if( !force_style )
  {
    renderer->RemoveViewProp( this->m_CursorActor );
    renderer->RemoveViewProp( this->m_TextActor );

  } // fi

  // Configure camera
  vtkCamera* camera = renderer->GetActiveCamera( );
  if( camera != NULL && force_style )
  {
    // Parallel projections are better when displaying 2D images
    int axis = this->GetAxis( );
    camera->ParallelProjectionOn( );
    camera->SetFocalPoint( double( 0 ), double( 0 ), double( 0 ) );
    if( axis == 0 )
    {
      camera->SetPosition( double( 1 ), double( 0 ), double( 0 ) );
      camera->SetViewUp  ( double( 0 ), double( 0 ), double( 1 ) );
    }
    else if( axis == 1 )
    {
      camera->SetPosition( double( 0 ), double( -1 ), double( 0 ) );
      camera->SetViewUp  ( double( 0 ), double(  0 ), double( 1 ) );
    }
    else // if( axis == 2 )
    {
      camera->SetPosition( double( 0 ), double(  0 ), double( -1 ) );
      camera->SetViewUp  ( double( 0 ), double( -1 ), double(  0 ) );

    } // fi

  } // fi
  this->ResetCamera( );
  rwi->Render( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
PopActorsFrom( vtkRenderWindow* window )
{
  vtkRenderWindowInteractor* rwi = window->GetInteractor( );
  vtkRenderer* renderer = window->GetRenderers( )->GetFirstRenderer( );

  if( renderer != NULL )
  {
    // Update actors
    vtkProp* prop;
    this->InitTraversal( );
    while( prop = this->GetNextProp( ) )
      renderer->RemoveViewProp( prop );
    renderer->Modified( );

  } // fi
  if( rwi != NULL )
    rwi->Render( );
}

// -------------------------------------------------------------------------
unsigned int cpExtensions::Visualization::ImageSliceActors::
GetNumberOfImageActors( ) const
{
  return( this->m_ImageActors.size( ) );
}

// -------------------------------------------------------------------------
vtkImageActor* cpExtensions::Visualization::ImageSliceActors::
GetImageActor( unsigned int id )
{
  if( id < this->m_ImageActors.size( ) )
    return( this->m_ImageActors[ id ] );
  else
    return( NULL );
}

// -------------------------------------------------------------------------
const vtkImageActor* cpExtensions::Visualization::ImageSliceActors::
GetImageActor( unsigned int id ) const
{
  if( id < this->m_ImageActors.size( ) )
    return( this->m_ImageActors[ id ] );
  else
    return( NULL );
}

// -------------------------------------------------------------------------
vtkTextActor* cpExtensions::Visualization::ImageSliceActors::
GetTextActor( )
{
  return( this->m_TextActor );
}

// -------------------------------------------------------------------------
const vtkTextActor* cpExtensions::Visualization::ImageSliceActors::
GetTextActor( ) const
{
  return( this->m_TextActor );
}

// -------------------------------------------------------------------------
vtkActor* cpExtensions::Visualization::ImageSliceActors::
GetPlaneActor( )
{
  return( this->m_PlaneActor );
}

// -------------------------------------------------------------------------
const vtkActor* cpExtensions::Visualization::ImageSliceActors::
GetPlaneActor( ) const
{
  return( this->m_PlaneActor );
}

// -------------------------------------------------------------------------
vtkPlane* cpExtensions::Visualization::ImageSliceActors::
GetPlaneFunction( )
{
  return( this->m_SliceMappers[ 0 ]->GetSlicePlane( ) );
}

// -------------------------------------------------------------------------
const vtkPlane* cpExtensions::Visualization::ImageSliceActors::
GetPlaneFunction( ) const
{
  return( this->m_SliceMappers[ 0 ]->GetSlicePlane( ) );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetInterpolate( bool v )
{
  if( this->m_Interpolate != v )
  {
    for( unsigned int i = 0; i < this->m_ImageActors.size( ); ++i )
      this->m_ImageActors[ i ]->SetInterpolate( v );
    this->m_Interpolate = v;
    this->Modified( );

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
InterpolateOn( )
{
  this->SetInterpolate( true );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
InterpolateOff( )
{
  this->SetInterpolate( false );
}

// -------------------------------------------------------------------------
double* cpExtensions::Visualization::ImageSliceActors::
GetDisplayBounds( ) const
{
  if( this->m_ImageActors.size( ) > 0 )
    return( this->m_ImageActors[ 0 ]->GetDisplayBounds( ) );
  else
    return( NULL );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
GetDisplayBounds( double bounds[ 6 ] ) const
{
  if( this->m_ImageActors.size( ) == 0 )
  {
    bounds[ 0 ] = bounds[ 2 ] = bounds[ 4 ] = double( -1 );
    bounds[ 1 ] = bounds[ 3 ] = bounds[ 5 ] = double( -1 );
  }
  else
    this->m_ImageActors[ 0 ]->GetDisplayBounds( bounds );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
ResetCursor( )
{
  if( this->m_SliceMappers.size( ) > 0 )
  {
    double bounds[ 6 ];
    this->m_SliceMappers[ 0 ]->GetInput( )->GetBounds( bounds );
    double pos[] = { bounds[ 0 ], bounds[ 2 ], bounds[ 4 ] };
    this->SetCursor( pos );
  }
  else
  {
    vtkPoints* points = this->m_Cursor->GetPoints( );
    points->SetPoint( 0, 0, 0, 0 );
    points->SetPoint( 1, 0, 0, 0 );
    points->SetPoint( 2, 0, 0, 0 );
    points->SetPoint( 3, 0, 0, 0 );
    points->SetPoint( 4, 0, 0, 0 );
    points->SetPoint( 5, 0, 0, 0 );
    points->SetPoint( 6, 0, 0, 0 );
    points->SetPoint( 7, 0, 0, 0 );
    this->m_Cursor->Modified( );
    this->m_CursorMapper->Modified( );
    this->m_CursorActor->Modified( );

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetCursor( double pos[ 3 ] )
{
  if( this->m_SliceMappers.size( ) == 0 )
    return;

  // Get ordered axes
  int a0 = this->GetAxis( );
  int a1 = ( a0 + 1 ) % 3;
  int a2 = ( a0 + 2 ) % 3;
  int ma0 = a0 << 1;
  int ma1 = a1 << 1;
  int ma2 = a2 << 1;

  double bounds[ 6 ];
  this->m_SliceMappers[ 0 ]->GetInput( )->GetBounds( bounds );

  double
    p0[ 3 ], p1[ 3 ], p2[ 3 ], p3[ 3 ],
    p4[ 3 ], p5[ 3 ], p6[ 3 ], p7[ 3 ];

  p0[ a2 ] = p1[ a2 ] = p4[ a2 ] = p5[ a2 ] = pos[ a2 ];
  p2[ a1 ] = p3[ a1 ] = p6[ a1 ] = p7[ a1 ] = pos[ a1 ];
  p0[ a0 ] = p1[ a0 ] = p2[ a0 ] = p3[ a0 ] = bounds[ ma0 ];
  p4[ a0 ] = p5[ a0 ] = p6[ a0 ] = p7[ a0 ] = bounds[ ma0 + 1 ];
  p0[ a1 ] = p4[ a1 ] = bounds[ ma1 ];
  p1[ a1 ] = p5[ a1 ] = bounds[ ma1 + 1 ];
  p2[ a2 ] = p6[ a2 ] = bounds[ ma2 ];
  p3[ a2 ] = p7[ a2 ] = bounds[ ma2 + 1 ];

  vtkPoints* points = this->m_Cursor->GetPoints( );
  points->SetPoint( 0, p0 );
  points->SetPoint( 1, p1 );
  points->SetPoint( 2, p2 );
  points->SetPoint( 3, p3 );
  points->SetPoint( 4, p4 );
  points->SetPoint( 5, p5 );
  points->SetPoint( 6, p6 );
  points->SetPoint( 7, p7 );
  this->m_Cursor->Modified( );
  this->m_CursorMapper->Modified( );
  this->m_CursorActor->Modified( );
}

// -------------------------------------------------------------------------
vtkImageMapToColors* cpExtensions::Visualization::ImageSliceActors::
GetImageMap( unsigned int id )
{
  if( id < this->m_ImageMaps.size( ) )
    return( this->m_ImageMaps[ id ].GetPointer( ) );
  else
    return( NULL );
}

// -------------------------------------------------------------------------
const vtkImageMapToColors* cpExtensions::Visualization::ImageSliceActors::
GetImageMap( unsigned int id ) const
{
  if( id < this->m_ImageMaps.size( ) )
    return( this->m_ImageMaps[ id ].GetPointer( ) );
  else
    return( NULL );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetMinWindow( ) const
{
  return( this->m_MinWindow );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetMaxWindow( ) const
{
  return( this->m_MaxWindow );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetMinLevel( ) const
{
  return( this->m_MinLevel );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetMaxLevel( ) const
{
  return( this->m_MaxLevel );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetWindow( ) const
{
  if( this->m_ImageMaps.size( ) == 0 )
    return( double( 0 ) );
  if( this->m_ImageMaps[ 0 ].GetPointer( ) == NULL )
    return( double( 0 ) );

  vtkWindowLevelLookupTable* lut =
    dynamic_cast< vtkWindowLevelLookupTable* >(
      this->m_ImageMaps[ 0 ]->GetLookupTable( )
      );
  if( lut != NULL )
    return( lut->GetWindow( ) );
  else
    return( double( 0 ) );
}

// -------------------------------------------------------------------------
double cpExtensions::Visualization::ImageSliceActors::
GetLevel( ) const
{
  if( this->m_ImageMaps.size( ) == 0 )
    return( double( 0 ) );
  if( this->m_ImageMaps[ 0 ].GetPointer( ) == NULL )
    return( double( 0 ) );

  vtkWindowLevelLookupTable* lut =
    dynamic_cast< vtkWindowLevelLookupTable* >(
      this->m_ImageMaps[ 0 ]->GetLookupTable( )
      );
  if( lut != NULL )
    return( lut->GetLevel( ) );
  else
    return( double( 0 ) );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetWindow( double w )
{
  if( this->m_ImageMaps.size( ) == 0 )
    return;
  if( this->m_ImageMaps[ 0 ].GetPointer( ) == NULL )
    return;
  vtkWindowLevelLookupTable* lut =
    dynamic_cast< vtkWindowLevelLookupTable* >(
      this->m_ImageMaps[ 0 ]->GetLookupTable( )
      );
  if( lut == NULL )
    return;

  if( w < this->m_MinWindow )
    w = this->m_MinWindow;
  if( w > this->m_MaxWindow )
    w = this->m_MaxWindow;

  lut->SetWindow( w );
  lut->Build( );
  this->m_ImageMaps[ 0 ]->Modified( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetLevel( double l )
{
  if( this->m_ImageMaps.size( ) == 0 )
    return;
  if( this->m_ImageMaps[ 0 ].GetPointer( ) == NULL )
    return;
  vtkWindowLevelLookupTable* lut =
    dynamic_cast< vtkWindowLevelLookupTable* >(
      this->m_ImageMaps[ 0 ]->GetLookupTable( )
      );
  if( lut == NULL )
    return;

  if( l < this->m_MinLevel )
    l = this->m_MinLevel;
  if( l > this->m_MaxLevel )
    l = this->m_MaxLevel;

  lut->SetLevel( l );
  lut->Build( );
  this->m_ImageMaps[ 0 ]->Modified( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetWindowLevel( double w, double l )
{
  if( this->m_ImageMaps.size( ) == 0 )
    return;
  if( this->m_ImageMaps[ 0 ].GetPointer( ) == NULL )
    return;
  vtkWindowLevelLookupTable* lut =
    dynamic_cast< vtkWindowLevelLookupTable* >(
      this->m_ImageMaps[ 0 ]->GetLookupTable( )
      );
  if( lut == NULL )
    return;

  if( w < this->m_MinWindow )
    w = this->m_MinWindow;
  if( w > this->m_MaxWindow )
    w = this->m_MaxWindow;
  if( l < this->m_MinLevel )
    l = this->m_MinLevel;
  if( l > this->m_MaxLevel )
    l = this->m_MaxLevel;

  lut->SetWindow( w );
  lut->SetLevel( l );
  lut->Build( );
  this->m_ImageMaps[ 0 ]->Modified( );
  this->UpdateText( w, l );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
ResetWindowLevel( )
{
  static const double _2 = double( 2 );
  this->SetWindowLevel(
    this->m_MaxWindow, ( this->m_MaxLevel + this->m_MinLevel ) / _2
    );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetLookupTable( unsigned int id, vtkLookupTable* lut )
{
  if( id < this->m_ImageMaps.size( ) && id > 0 )
  {
    if( this->m_ImageMaps[ id ].GetPointer( ) != NULL )
    {
      this->m_ImageMaps[ id ]->SetLookupTable( lut );
      this->m_ImageMaps[ id ]->Modified( );
      this->Modified( );

    } // fi

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetLookupTableAsColor( unsigned int id, double r, double g, double b )
{
  static const double _0 = double( 0 );
  static const double _2 = double( 2 );
  static const double _4 = double( 4 );
  static const double _6 = double( 6 );
  static const double _OPACITY = double( 0.6 );

  // Check ID consistency
  if( id == 0 || id >= this->m_ImageMaps.size( ) )
    return;

  // Get image scalar range
  vtkAlgorithmOutput* aout = this->m_ImageMaps[ id ]->GetOutputPort( );
  vtkImageData* image = dynamic_cast< vtkImageData* >(
    aout->GetProducer( )->GetOutputDataObject( aout->GetIndex( ) )
    );
  double range[ 2 ];
  image->GetScalarRange( range );

  // Get HSV from display color
  double cmax = ( r > g )? r: g; cmax = ( b > cmax )? b: cmax;
  double cmin = ( r < g )? r: g; cmin = ( b < cmin )? b: cmin;
  double d = cmax - cmin;

  double saturation = ( std::fabs( cmax ) > _0 )? d / cmax: _0;
  double value = cmax;
  double hue = _0;
  if( d > _0 )
  {
    if( r == cmax )
      hue = std::fmod( ( g - b ) / d, _6 );
    else if( g == cmax )
      hue = ( ( b - r ) / d ) + _2;
    else if( b == cmax )
      hue = ( ( r - g ) / d ) + _4;
    hue /= _6;

  } // fi

  // Define new lookup table
  vtkSmartPointer< vtkLookupTable > lut =
    vtkSmartPointer< vtkLookupTable >::New( );
  lut->SetScaleToLinear( );
  lut->SetNanColor( _0, _0, _0, _0 );
  lut->SetTableRange( range[ 0 ], range[ 1 ] );
  lut->SetAlphaRange( _0, _OPACITY );
  lut->SetHueRange( _0, hue );
  lut->SetSaturationRange( _0, saturation );
  lut->SetValueRange( _0, value );
  lut->Build( );
  this->SetLookupTable( id, lut );
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::ImageSliceActors::
GetAxis( ) const
{
  if( this->m_SliceMappers.size( ) > 0 )
    return( this->m_SliceMappers[ 0 ]->GetOrientation( ) );
  else
    return( -1 );
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::ImageSliceActors::
GetSliceNumber( ) const
{
  if( this->m_SliceMappers.size( ) > 0 )
    return( this->m_SliceMappers[ 0 ]->GetSliceNumber( ) );
  else
    return( -1 );
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::ImageSliceActors::
GetSliceNumberMinValue( ) const
{
  if( this->m_SliceMappers.size( ) > 0 )
    return( this->m_SliceMappers[ 0 ]->GetSliceNumberMinValue( ) );
  else
    return( -1 );
}

// -------------------------------------------------------------------------
int cpExtensions::Visualization::ImageSliceActors::
GetSliceNumberMaxValue( ) const
{
  if( this->m_SliceMappers.size( ) > 0 )
    return( this->m_SliceMappers[ 0 ]->GetSliceNumberMaxValue( ) );
  else
    return( -1 );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetSliceNumber( const int& slice )
{
  unsigned int nImages = this->m_SliceMappers.size( );
  if( nImages == 0 )
    return;

  // Change visualization extent
  int axis = this->GetAxis( );
  for( unsigned int i = 0; i < nImages; ++i )
  {
    // Update mappers and display bounds
    this->m_SliceMappers[ i ]->SetSliceNumber( slice );
    this->m_SliceMappers[ i ]->Modified( );
    this->m_ImageActors[ i ]->Modified( );
    this->m_SliceMappers[ i ]->Update( );

    // Update display extent (this isn't done automatically)
    vtkImageData* image = this->m_SliceMappers[ i ]->GetInput( );
    int ext[ 6 ];
    image->GetExtent( ext );
    ext[ axis << 1 ] = slice;
    ext[ ( axis << 1 ) + 1 ] = slice;
    this->m_ImageActors[ i ]->SetDisplayExtent( ext );

  } // rof

  double bnds[ 6 ];
  this->m_SliceMappers[ 0 ]->GetBounds( bnds );
  double x0[][ 3 ] =
  {
    { bnds[ 0 ], bnds[ 2 ], bnds[ 4 ] },
    { bnds[ 1 ], bnds[ 3 ], bnds[ 5 ] }
  };
  double p0[ 2 ][ 3 ];

  vtkPlane* plane = this->m_SliceMappers[ 0 ]->GetSlicePlane( );
  plane->GeneralizedProjectPoint( x0[ 0 ], p0[ 0 ] );
  plane->GeneralizedProjectPoint( x0[ 1 ], p0[ 1 ] );

  bnds[ 0 ] = p0[ 0 ][ 0 ];
  bnds[ 1 ] = p0[ 1 ][ 0 ];
  bnds[ 2 ] = p0[ 0 ][ 1 ];
  bnds[ 3 ] = p0[ 1 ][ 1 ];
  bnds[ 4 ] = p0[ 0 ][ 2 ];
  bnds[ 5 ] = p0[ 1 ][ 2 ];

  // Configure visualization and implicit plane orientation
  this->m_PlaneActor->GetProperty( )->SetRepresentationToWireframe( );
  this->m_PlaneActor->GetProperty( )->SetLineWidth( 3 );
  vtkPoints* plane_points = this->m_Plane->GetPoints( );
  plane_points->SetPoint( 0, bnds[ 0 ], bnds[ 2 ], bnds[ 4 ] );
  plane_points->SetPoint( 2, bnds[ 1 ], bnds[ 3 ], bnds[ 5 ] );
  if( axis == 0 || axis == 2 ) // YZ, x-normal
  {
    plane_points->SetPoint( 1, bnds[ 1 ], bnds[ 2 ], bnds[ 5 ] );
    plane_points->SetPoint( 3, bnds[ 0 ], bnds[ 3 ], bnds[ 4 ] );
    if( axis == 0 )
      this->m_PlaneActor->GetProperty( )->SetColor( 1, 0, 0 );
    else
      this->m_PlaneActor->GetProperty( )->SetColor( 0, 0, 1 );
  }
  else if( axis == 1 ) // ZX, y-normal
  {
    plane_points->SetPoint( 1, bnds[ 0 ], bnds[ 2 ], bnds[ 5 ] );
    plane_points->SetPoint( 3, bnds[ 1 ], bnds[ 3 ], bnds[ 4 ] );
    this->m_PlaneActor->GetProperty( )->SetColor( 0, 1, 0 );

  } // fi
  this->m_Plane->Modified( );
  this->m_PlaneMapper->Modified( );
  this->m_PlaneActor->Modified( );

  // Update text
  this->UpdateText( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
SetSlice( double* pos )
{
  vtkImageData* image;
  if( this->m_ImageMaps[ 0 ] != NULL )
    image =
      dynamic_cast< vtkImageData* >( this->m_ImageMaps[ 0 ]->GetInput( ) );
  else
    image = this->m_SliceMappers[ 0 ]->GetInput( );

  int ijk[ 3 ];
  double pcoords[ 3 ];
  image->ComputeStructuredCoordinates( pos, ijk, pcoords );
  this->SetSliceNumber( ijk[ this->GetAxis( ) ] );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
UpdateText( )
{
  if( this->m_SliceMappers.size( ) > 0 )
  {
    char axis;
    int axId = this->m_SliceMappers[ 0 ]->GetOrientation( );
    if     ( axId == 0 ) axis = 'X';
    else if( axId == 1 ) axis = 'Y';
    else if( axId == 2 ) axis = 'Z';

    std::sprintf(
      this->m_TextBuffer, "Axis: %c (%d)",
      axis, this->m_SliceMappers[ 0 ]->GetSliceNumber( )
      );
  }
  else
    this->m_TextBuffer[ 0 ] = '\0';
  this->m_TextActor->SetInput( this->m_TextBuffer );
  this->m_TextActor->Modified( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
UpdateText( double pos[ 3 ] )
{
  if( this->m_SliceMappers.size( ) > 0 )
  {
    char axis;
    int axId = this->m_SliceMappers[ 0 ]->GetOrientation( );
    if     ( axId == 0 ) axis = 'X';
    else if( axId == 1 ) axis = 'Y';
    else if( axId == 2 ) axis = 'Z';
    int slice = this->GetSliceNumber( );

    vtkImageData* image;
    if( this->m_ImageMaps[ 0 ] != NULL )
      image =
        dynamic_cast< vtkImageData* >( this->m_ImageMaps[ 0 ]->GetInput( ) );
    else
      image = this->m_SliceMappers[ 0 ]->GetInput( );

    int ijk[ 3 ];
    double pcoords[ 3 ];
    image->ComputeStructuredCoordinates( pos, ijk, pcoords );
    ijk[ axId ] = slice;

    int ext[ 6 ];
    image->GetExtent( ext );
    if(
      ext[ 0 ] <= ijk[ 0 ] && ijk[ 0 ] <= ext[ 1 ] &&
      ext[ 2 ] <= ijk[ 1 ] && ijk[ 1 ] <= ext[ 3 ] &&
      ext[ 4 ] <= ijk[ 2 ] && ijk[ 2 ] <= ext[ 5 ]
      )
    {
      int nScl = image->GetNumberOfScalarComponents( );
      std::stringstream str;
      str
        << "[" << ijk[ 0 ]
        << "," << ijk[ 1 ]
        << "," << ijk[ 2 ] << "]=(";
      str <<
        image->GetScalarComponentAsFloat( ijk[ 0 ], ijk[ 1 ], ijk[ 2 ], 0 );
      for( int n = 1; n < nScl; ++n )
        str
          << " "
          << image->GetScalarComponentAsFloat(
            ijk[ 0 ], ijk[ 1 ], ijk[ 2 ], n
            );
      str << ")";
      std::sprintf(
        this->m_TextBuffer, "Axis: %c (%d)\nPixel %s",
        axis, slice, str.str( ).c_str( )
        );

    } // fi
  }
  else
    this->m_TextBuffer[ 0 ] = '\0';
  this->m_TextActor->SetInput( this->m_TextBuffer );
  this->m_TextActor->Modified( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
UpdateText( const double& w, const double& l )
{
  if( this->m_SliceMappers.size( ) > 0 )
  {
    char axis;
    int axId = this->m_SliceMappers[ 0 ]->GetOrientation( );
    if     ( axId == 0 ) axis = 'X';
    else if( axId == 1 ) axis = 'Y';
    else if( axId == 2 ) axis = 'Z';

    std::sprintf(
      this->m_TextBuffer, "Axis: %c (%d)\nW/L (%.2f/%.2f)",
      axis, this->m_SliceMappers[ 0 ]->GetSliceNumber( ), w, l
      );
  }
  else
    this->m_TextBuffer[ 0 ] = '\0';
  this->m_TextActor->SetInput( this->m_TextBuffer );
  this->m_TextActor->Modified( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
Render( const double& t )
{
  if( this->m_Window != NULL )
  {
    vtkRenderer* renderer =
      this->m_Window->GetRenderers( )->GetFirstRenderer( );
    if( renderer != NULL )
      renderer->SetAllocatedRenderTime( t );
    this->m_Window->Render( );

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
ResetCamera( )
{
  if( this->m_Window == NULL )
    return;
  vtkRenderer* renderer =
    this->m_Window->GetRenderers( )->GetFirstRenderer( );
  if( renderer != NULL )
    renderer->ResetCamera( );
#error ACA VOY: reconfigurar la camara alrededor únicamente del plano.
}

// -------------------------------------------------------------------------
cpExtensions::Visualization::ImageSliceActors::
ImageSliceActors( )
  : Superclass( ),
    m_Window( NULL ),
    m_Interpolate( false )
{
  this->Clear( );
  this->_ConfigureStyle( );
}

// -------------------------------------------------------------------------
cpExtensions::Visualization::ImageSliceActors::
~ImageSliceActors( )
{
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_ConfigureStyle( )
{
  // Connect this view with a controller
  this->m_Style = vtkSmartPointer< TStyle >::New( );
  this->m_Style->AddMouseMoveCommand( Self::_MouseMoveCommand, this );
  this->m_Style->AddMouseClickCommand( Self::_MouseClickCommand, this );
  this->m_Style->AddMouseWheelCommand( Self::_MouseWheelCommand, this );
  this->m_Style->AddKeyCommand( Self::_KeyCommand, this );
  this->m_Style->AddEnterCommand( Self::_EnterCommand, this );
  this->m_Style->AddLeaveCommand( Self::_LeaveCommand, this );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_ConfigureNewLUT( vtkImageData* data )
{
  // Does the new LUT is a window-level one?
  unsigned int nImgs = this->m_ImageMaps.size( );
  unsigned int nCmps = data->GetNumberOfScalarComponents( );

  if( nCmps > 1 )
  {
    this->m_ImageMaps.push_back( NULL );
    return;

  } // fi

  // Create LUT filter
  this->m_ImageMaps.push_back( vtkSmartPointer< vtkImageMapToColors >::New( ) );
  if( nImgs == 0 )
  {
    double range[ 2 ];
    data->GetScalarRange( range );
    vtkSmartPointer< vtkWindowLevelLookupTable > lut =
      vtkSmartPointer< vtkWindowLevelLookupTable >::New( );
    lut->SetScaleToLinear( );
    lut->SetTableRange( range );
    lut->SetWindow( range[ 1 ] - range[ 0 ] );
    lut->SetLevel( ( range[ 1 ] + range[ 0 ] ) / double( 2 ) );
    lut->Build( );
    this->m_ImageMaps[ 0 ]->SetLookupTable( lut );
  }
  else
    this->SetLookupTableAsColor( nImgs, 1, 0, 0 );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_ConfigureNewInput( int axis )
{
  unsigned int nImages = this->m_ImageActors.size( );

  // Configure mapper
  vtkImageSliceMapper* mapper = this->m_SliceMappers[ nImages ];
  if( nImages == 0 )
    mapper->SetOrientation( axis );
  else
    mapper->SetOrientation( this->m_SliceMappers[ 0 ]->GetOrientation( ) );
  mapper->Update( );

  // Create actor
  vtkSmartPointer< vtkImageActor > actor =
    vtkSmartPointer< vtkImageActor >::New( );
  this->m_ImageActors.push_back( actor );
  actor->SetMapper( mapper );
  actor->SetInterpolate( this->m_Interpolate );
  actor->Modified( );

  if( nImages == 0 )
    this->m_Style->AssociateImageActor( actor );
  this->AddItem( actor );

  if( nImages > 1 )
    this->SetSliceNumber( this->GetSliceNumber( ) );
  else
    this->SetSliceNumber( this->GetSliceNumberMaxValue( ) );
  this->ResetCursor( );
  this->Modified( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_MouseMoveCommand(
  void* data, const TStyle::ButtonID& btn, int* idx, double* pos,
  bool alt, bool ctr, bool sft
  )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  if( btn == TStyle::ButtonID_None )
  {
    // Just show the pixel information
    actors->SetCursor( pos );
    actors->UpdateText( pos );
    actors->Render( 1e-3 );
  }
  else if( btn == TStyle::ButtonID_Left )
  {
    if( !alt && ctr && !sft )
    {
      // Interactively move slices
      auto i = actors->m_SlicesCommands.begin( );
      for( ; i != actors->m_SlicesCommands.end( ); ++i )
        i->first( pos, actors->GetAxis( ), i->second );
      actors->Render( 1e-3 );

    } // fi
  }
  else if( btn == TStyle::ButtonID_Right )
  {
    if( !alt && !ctr && sft )
    {
      // Change image window level
      double bounds[ 6 ];
      actors->m_SliceMappers[ 0 ]->GetBounds( bounds );

      int a0 = actors->GetAxis( );
      int a1 = ( a0 + 1 ) % 3;
      int a2 = ( a0 + 2 ) % 3;
      double dx = pos[ a1 ] - actors->m_StartWindowLevelPos[ a1 ];
      double dy = pos[ a2 ] - actors->m_StartWindowLevelPos[ a2 ];
      dx /= bounds[ ( a1 << 1 ) + 1 ] - bounds[ a1 << 1 ];
      dy /= bounds[ ( a2 << 1 ) + 1 ] - bounds[ a2 << 1 ];

      dx *= actors->m_StartWindowLevel[ 0 ];
      dy *= actors->m_StartWindowLevel[ 1 ];
      dx += actors->m_StartWindowLevel[ 0 ];
      dy += actors->m_StartWindowLevel[ 1 ];
      actors->SetWindowLevel( dx, dy );
      actors->Render( 1e-3 );

      // Associate objects
      auto i = actors->m_WindowLevelCommands.begin( );
      for( ; i != actors->m_WindowLevelCommands.end( ); ++i )
        i->first( dx, dy, i->second );
      
    } // fi

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_MouseClickCommand(
  void* data, const TStyle::ButtonID& btn, int* idx, double* pos,
  bool alt, bool ctr, bool sft
  )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  actors->m_StartWindowLevelPos[ 0 ] = pos[ 0 ];
  actors->m_StartWindowLevelPos[ 1 ] = pos[ 1 ];
  actors->m_StartWindowLevelPos[ 2 ] = pos[ 2 ];
  actors->m_StartWindowLevel[ 0 ] = actors->GetWindow( );
  actors->m_StartWindowLevel[ 1 ] = actors->GetLevel( );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_MouseWheelCommand(
  void* data, const int& dir,
  bool alt, bool ctr, bool sft
  )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  if( !alt && !ctr )
  {
    int slice = actors->GetSliceNumber( ) + ( dir * ( ( sft )? 10: 1 ) );
    if( slice < actors->GetSliceNumberMinValue( ) )
      slice = actors->GetSliceNumberMinValue( );
    if( slice > actors->GetSliceNumberMaxValue( ) )
      slice = actors->GetSliceNumberMaxValue( );
    actors->SetSliceNumber( slice );

    auto a = actors->m_AssociatedSlices.begin( );
    for( ; a != actors->m_AssociatedSlices.end( ); ++a )
    {
      ( *a )->SetSliceNumber( slice );
      ( *a )->Render( 1e-3 );

    } // rof
    actors->Render( 1e-3 );
    
    // Associate objects
    auto i = actors->m_RenderCommands.begin( );
    for( ; i != actors->m_RenderCommands.end( ); ++i )
      i->first( i->second );

  } // fi
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_KeyCommand( void* data, const char& key )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  switch( key )
  {
  case 'r': case 'R':
  {
    actors->ResetCamera( );
    actors->Render( 1e-3 );

    // Associate objects
    auto i = actors->m_RenderCommands.begin( );
    for( ; i != actors->m_RenderCommands.end( ); ++i )
      i->first( i->second );
  }
  break;
  case 'w': case 'W':
  {
    actors->ResetWindowLevel( );
    actors->Render( 1e-3 );

    // Associate objects
    auto i = actors->m_RenderCommands.begin( );
    for( ; i != actors->m_RenderCommands.end( ); ++i )
      i->first( i->second );
  }
  break;
  default:
    break;
  } // hctiws
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_EnterCommand( void* data )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  actors->ResetCursor( );
  actors->m_CursorActor->VisibilityOn( );
  actors->Render( 1e-3 );
}

// -------------------------------------------------------------------------
void cpExtensions::Visualization::ImageSliceActors::
_LeaveCommand( void* data )
{
  ImageSliceActors* actors = reinterpret_cast< ImageSliceActors* >( data );
  if( actors == NULL )
    return;

  actors->ResetCursor( );
  actors->m_CursorActor->VisibilityOff( );
  actors->Render( 1e-3 );
}

// eof - $RCSfile$