yet another refactoring

[cpPlugins.git] / lib / cpExtensions / Visualization / LUTImageActor.cxx

diff --git a/lib/cpExtensions/Visualization/LUTImageActor.cxx b/lib/cpExtensions/Visualization/LUTImageActor.cxx
new file mode 100644 (file)
index 0000000..d377215
--- /dev/null
+++ b/lib/cpExtensions/Visualization/LUTImageActor.cxx
@@ -0,0 +1,160 @@
+#include <cpExtensions/Visualization/LUTImageActor.h>
+#include <cpExtensions/Visualization/ImageSliceMapper.h>
+
+#include <vtkImageData.h>
+#include <vtkImageMapToColors.h>
+#include <vtkImageProperty.h>
+#include <vtkLookupTable.h>
+#include <cpExtensions/Algorithms/ImageBlender.h>
+
+// -------------------------------------------------------------------------
+cpExtensions::Visualization::LUTImageActor::
+Self* cpExtensions::Visualization::LUTImageActor::
+New( )
+{
+  return( new Self( ) );
+}
+
+// -------------------------------------------------------------------------
+unsigned int cpExtensions::Visualization::LUTImageActor::
+GetNumberOfImages( ) const
+{
+  return( this->m_Images.size( ) );
+}
+
+// -------------------------------------------------------------------------
+vtkImageData* cpExtensions::Visualization::LUTImageActor::
+GetImage( unsigned int id )
+{
+  if( id < this->m_Images.size( ) )
+    return( this->m_Images[ id ] );
+  else
+    return( NULL );
+}
+
+// -------------------------------------------------------------------------
+const vtkImageData* cpExtensions::Visualization::LUTImageActor::
+GetImage( unsigned int id ) const
+{
+  if( id < this->m_Images.size( ) )
+    return( this->m_Images[ id ] );
+  else
+    return( NULL );
+}
+
+// -------------------------------------------------------------------------
+unsigned int cpExtensions::Visualization::LUTImageActor::
+GetImageId( vtkImageData* image ) const
+{
+  unsigned int i = 0;
+  bool f = false;
+  while( i < this->m_Images.size( ) && !f )
+  {
+    if( this->m_Images[ i ].GetPointer( ) == image )
+      f = true;
+    else
+      i++;
+
+  } // elihw
+  return( i );
+}
+
+// -------------------------------------------------------------------------
+void cpExtensions::Visualization::LUTImageActor::
+GetLUTColor(
+  unsigned int id, double& r, double& g, double& b, double& a
+  ) const
+{
+  double rgba[ 4 ];
+  this->m_LUT->GetTableValue( id, rgba );
+  r = rgba[ 0 ];
+  g = rgba[ 1 ];
+  b = rgba[ 2 ];
+  a = rgba[ 3 ];
+}
+
+// -------------------------------------------------------------------------
+void cpExtensions::Visualization::LUTImageActor::
+GetLUTColor(
+  vtkImageData* image, double& r, double& g, double& b, double& a
+  ) const
+{
+  this->GetLUTColor( this->GetImageId( image ), r, g, b, a );
+}
+
+// -------------------------------------------------------------------------
+void cpExtensions::Visualization::LUTImageActor::
+SetLUTColor( unsigned int id, double r, double g, double b, double a )
+{
+  if( id >= ( unsigned int )( this->m_LUT->GetNumberOfTableValues( ) ) )
+    this->m_LUT->SetNumberOfTableValues( id + 1 );
+  this->m_LUT->SetTableValue( id, r, g, b, a );
+  this->m_LUT->Modified( );
+}
+
+// -------------------------------------------------------------------------
+void cpExtensions::Visualization::LUTImageActor::
+SetLUTColor( vtkImageData* image, double r, double g, double b, double a )
+{
+  this->SetLUTColor( this->GetImageId( image ), r, g, b, a );
+}
+
+// TODO: !!!!!
+#include <vtkImageCast.h>
+
+// -------------------------------------------------------------------------
+unsigned int cpExtensions::Visualization::LUTImageActor::
+AddImage( vtkImageData* image, double r, double g, double b, double a )
+{
+  static vtkSmartPointer< vtkImageCast > cast;
+  cast = vtkSmartPointer< vtkImageCast >::New( );
+
+  if( image != NULL )
+  {
+    cast->SetOutputScalarTypeToUnsignedChar( );
+    cast->SetInputData( image );
+    cast->Update( );
+
+    this->m_Images.push_back( image );
+    this->SetLUTColor( this->m_Images.size( ), r, g, b, a );
+    this->m_Blender->AddInputData( cast->GetOutput( ) );
+    this->m_Blender->Update( );
+
+    if( this->m_Images.size( ) == 1 )
+    {
+      this->m_ImageMap->
+        SetInputConnection( this->m_Blender->GetOutputPort( ) );
+      this->m_Mapper->
+        SetInputConnection( this->m_ImageMap->GetOutputPort( ) );
+
+    } // fi
+    this->Modified( );
+
+    this->GetProperty( )->SetInterpolationTypeToNearest( );
+
+  } // fi
+  return( this->m_Images.size( ) );
+}
+
+// -------------------------------------------------------------------------
+cpExtensions::Visualization::LUTImageActor::
+LUTImageActor( )
+  : Superclass( )
+{
+  this->m_LUT = vtkSmartPointer< vtkLookupTable >::New( );
+  this->m_LUT->SetNumberOfTableValues( 1 );
+  this->m_LUT->SetNanColor( 0, 0, 0, 0 );
+  this->m_LUT->SetTableValue( 0, 0, 0, 0, 0 );
+
+  this->m_Blender = vtkSmartPointer< _TImageBlender >::New( );
+  this->m_ImageMap = vtkSmartPointer< vtkImageMapToColors >::New( );
+  this->m_ImageMap->SetLookupTable( this->m_LUT );
+}
+
+// -------------------------------------------------------------------------
+cpExtensions::Visualization::LUTImageActor::
+~LUTImageActor( )
+{
+}
+
+// eof - $RCSfile$