lib/cpPlugins/ProcessObject.hxx

   1 #ifndef __CPPLUGINS__PROCESSOBJECT__HXX__
   2 #define __CPPLUGINS__PROCESSOBJECT__HXX__
   3
   4 // -------------------------------------------------------------------------
   5 template< class _TType >
   6 _TType* cpPlugins::ProcessObject::
   7 GetInput( const std::string& id )
   8 {
   9   auto i = this->m_Inputs.find( id );
  10   if( i != this->m_Inputs.end( ) )
  11     return( dynamic_cast< _TType* >( i->second.GetPointer( ) ) );
  12   else
  13     return( NULL );
  14 }
  15
  16 // -------------------------------------------------------------------------
  17 template< class _TType >
  18 const _TType* cpPlugins::ProcessObject::
  19 GetInput( const std::string& id ) const
  20 {
  21   auto i = this->m_Inputs.find( id );
  22   if( i != this->m_Inputs.end( ) )
  23     return( dynamic_cast< const _TType* >( i->second.GetPointer( ) ) );
  24   else
  25     return( NULL );
  26 }
  27
  28 // -------------------------------------------------------------------------
  29 template< class _TType >
  30 _TType* cpPlugins::ProcessObject::
  31 GetOutput( const std::string& id )
  32 {
  33   auto i = this->m_Outputs.find( id );
  34   if( i != this->m_Outputs.end( ) )
  35     return( dynamic_cast< _TType* >( i->second.GetPointer( ) ) );
  36   else
  37     return( NULL );
  38 }
  39
  40 // -------------------------------------------------------------------------
  41 template< class _TType >
  42 const _TType* cpPlugins::ProcessObject::
  43 GetOutput( const std::string& id ) const
  44 {
  45   auto i = this->m_Outputs.find( id );
  46   if( i != this->m_Outputs.end( ) )
  47     return( dynamic_cast< const _TType* >( i->second.GetPointer( ) ) );
  48   else
  49     return( NULL );
  50 }
  51
  52 // -------------------------------------------------------------------------
  53 template< class _TType >
  54 _TType* cpPlugins::ProcessObject::
  55 GetInputData( const std::string& name )
  56 {
  57   auto in = this->GetInput( name );
  58   if( in != NULL )
  59   {
  60     auto i = in->GetITK< _TType >( );
  61     auto v = in->GetVTK< _TType >( );
  62     if     ( i != NULL ) return( i );
  63     else if( v != NULL ) return( v );
  64     else                 return( NULL );
  65   }
  66   else
  67     return( NULL );
  68 }
  69
  70 // -------------------------------------------------------------------------
  71 template< class _TType >
  72 _TType* cpPlugins::ProcessObject::
  73 GetOutputData( const std::string& name )
  74 {
  75   auto out = this->GetOutput( name );
  76   if( out != NULL )
  77   {
  78     auto i = out->GetITK< _TType >( );
  79     auto v = out->GetVTK< _TType >( );
  80     if     ( i != NULL ) return( i );
  81     else if( v != NULL ) return( v );
  82     else                 return( NULL );
  83   }
  84   else
  85     return( NULL );
  86 }
  87
  88 // -------------------------------------------------------------------------
  89 template< class O >
  90 void cpPlugins::ProcessObject::
  91 _AddOutput( const std::string& name )
  92 {
  93   auto i = this->m_Outputs.find( name );
  94   if( i == this->m_Outputs.end( ) )
  95   {
  96     typename O::Pointer o = O::New( );
  97     o->SetSource( this );
  98     this->m_Outputs[ name ] = o;
  99     this->Modified( );
 100
 101   } // fi
 102 }
 103
 104 // -------------------------------------------------------------------------
 105 template< class F >
 106 F* cpPlugins::ProcessObject::
 107 _CreateITK( )
 108 {
 109   F* filter = this->GetITK< F >( );
 110   if( filter == NULL )
 111   {
 112     typename F::Pointer filter_ptr = F::New( );
 113     this->m_ITKObject = filter_ptr;
 114     this->m_VTKObject = NULL;
 115     filter = filter_ptr.GetPointer( );
 116     this->Modified( );
 117
 118   } // fi
 119   return( filter );
 120 }
 121
 122 // -------------------------------------------------------------------------
 123 template< class F >
 124 F* cpPlugins::ProcessObject::
 125 _CreateVTK( )
 126 {
 127   F* filter = this->GetVTK< F >( );
 128   if( filter == NULL )
 129   {
 130     vtkSmartPointer< F > filter_ptr = vtkSmartPointer< F >::New( );
 131     this->m_ITKObject = NULL;
 132     this->m_VTKObject = filter_ptr;
 133     filter = filter_ptr.GetPointer( );
 134     this->Modified( );
 135
 136   } // fi
 137   return( filter );
 138 }
 139
 140 #endif // __CPPLUGINS__PROCESSOBJECT__HXX__
 141
 142 // eof - $RCSfile$