#include "MeshReader.h"
#include <cpPlugins/Interface/Mesh.h>
+#include <cstring>
+
#include <vtkPolyData.h>
#include <vtkPolyDataReader.h>
-
-#ifdef cpPlugins_Interface_QT4
-#include <QFileDialog>
-#endif // cpPlugins_Interface_QT4
-
-// -------------------------------------------------------------------------
-bool cpPlugins::IO::MeshReader::
-ExecConfigurationDialog( QWidget* parent )
-{
- bool r = false;
-
-#ifdef cpPlugins_Interface_QT4
-
- // Show dialog and check if it was accepted
- QFileDialog dialog( parent );
- dialog.setFileMode( QFileDialog::ExistingFile );
- dialog.setDirectory( QFileDialog::tr( "." ) );
- dialog.setNameFilter( QFileDialog::tr( "All files (*)" ) );
- if( dialog.exec( ) )
- {
- this->m_Parameters = this->m_DefaultParameters;
- QStringList names = dialog.selectedFiles( );
- this->m_Parameters.SetValueAsString(
- "FileName", names[ 0 ].toStdString( )
- );
-
- /* TODO
- this->m_Parameters.SetValueAsString( "PixelType", "float" );
- this->m_Parameters.SetValueAsUint( "Dimension", 3 );
- */
-
- r = true;
-
- } // fi
-
-#endif // cpPlugins_Interface_QT4
-
- return( r );
-}
+#include <vtkSTLReader.h>
// -------------------------------------------------------------------------
cpPlugins::IO::MeshReader::
MeshReader( )
: Superclass( )
{
- this->m_ClassName = "cpPlugins::IO::MeshReader";
- this->m_ClassCategory = "MeshReader";
-
- this->SetNumberOfOutputs( 1 );
- this->_MakeOutput< cpPlugins::Interface::Mesh >( 0 );
-
- using namespace cpPlugins::Interface;
- this->m_DefaultParameters.Configure( Parameters::String, "FileName" );
- this->m_DefaultParameters.Configure( Parameters::String, "PixelType" );
- this->m_DefaultParameters.Configure( Parameters::Uint, "Dimension" );
- this->m_DefaultParameters.SetValueAsString( "PixelType", "float" );
- this->m_DefaultParameters.SetValueAsUint( "Dimension", 3 );
- this->m_Parameters = this->m_DefaultParameters;
+ this->_AddOutput< cpPlugins::Interface::Mesh >( "Output" );
+
+ std::vector< TParameters::TString > valid_types;
+ valid_types.push_back( "float" );
+ valid_types.push_back( "double" );
+ this->m_Parameters->ConfigureAsOpenFileName( "FileName" );
+ this->m_Parameters->ConfigureAsChoices( "PixelType", valid_types );
+ this->m_Parameters->ConfigureAsUint( "Dimension" );
+
+ this->m_Parameters->SetUint( "Dimension", 3 );
+ this->m_Parameters->SetAcceptedFileExtensions(
+ "FileName",
+ "Mesh files (*.vtk *.stl *.obj)"
+ );
}
// -------------------------------------------------------------------------
_GenerateData( )
{
using namespace cpPlugins::Interface;
- Parameters::TUint dim = this->m_Parameters.GetValueAsUint( "Dimension" );
+ Parameters::TUint dim = this->m_Parameters->GetUint( "Dimension" );
std::string r = "MeshReader: Mesh dimension not supported.";
if( dim == 2 )
r = this->_GD0< 2 >( );
_GD0( )
{
using namespace cpPlugins::Interface;
- Parameters::TString pt = this->m_Parameters.GetValueAsString( "PixelType" );
+ Parameters::TString pt =
+ this->m_Parameters->GetSelectedChoice( "PixelType" );
std::string r = "MeshReader: Mesh pixel type not supported";
if( pt == "float" ) r = this->_GD1< float, D >( );
else if( pt == "double" ) r = this->_GD1< double, D >( );
_GD1( )
{
// Get filename
- using namespace cpPlugins::Interface;
- Parameters::TString fname =
- this->m_Parameters.GetValueAsString( "FileName" );
+ std::string fname = this->m_Parameters->GetOpenFileName( "FileName" );
+
+ // Get file extension
+ std::istringstream fname_str( fname );
+ std::string token, ext;
+ while( std::getline( fname_str, token, '.' ) )
+ ext = token;
+ std::transform( ext.begin( ), ext.end( ), ext.begin( ), tolower );
+
+ // Real read
+ if( ext == "stl" )
+ {
+ vtkSTLReader* stlr = this->_CreateVTK< vtkSTLReader >( );
+ stlr->SetFileName( fname.c_str( ) );
+ stlr->Update( );
+
+ auto out = this->GetOutputData< cpPlugins::Interface::Mesh >( "Output" );
+ if( out != NULL )
+ out->SetVTK( stlr->GetOutput( ) );
+ else
+ return( "MeshReader: output not correctly created." );
+ return( "" );
+ }
+ else if( ext == "vtk" )
+ {
+ vtkPolyDataReader* pdr = this->_CreateVTK< vtkPolyDataReader >( );
+ pdr->SetFileName( fname.c_str( ) );
+ pdr->Update( );
- vtkPolyDataReader* pdr = this->_CreateVTK< vtkPolyDataReader >( );
- pdr->SetFileName( fname.c_str( ) );
- pdr->Update( );
+ auto out = this->GetOutputData< cpPlugins::Interface::Mesh >( "Output" );
+ if( out != NULL )
+ out->SetVTK( pdr->GetOutput( ) );
+ else
+ return( "MeshReader: output not correctly created." );
+ return( "" );
- cpPlugins::Interface::Mesh* out =
- this->GetOutput< cpPlugins::Interface::Mesh >( 0 );
- if( out != NULL )
- out->SetVTK( pdr->GetOutput( ) );
- else
- return( "MeshReader: output not correctly created." );
- return( "" );
+ } // fi
}
// eof - $RCSfile$