Skip to content

GitLab

Commit 69822b6b authored by Sebastian Eichelbaum's avatar Sebastian Eichelbaum
Browse files

[STYLE]

parent 3c3fa480
Hide whitespace changes
Inline Side-by-side
Showing with 468 additions and 469 deletions
src/modules/data/WMData.cpp
View file @ 69822b6b
......@@ -286,7 +286,7 @@ void WMData::moduleMain()
WReaderFiberVTK fibReader( fileName );
m_dataSet = fibReader.read();
}
else if ( suffix == ".vtk" )
else if( suffix == ".vtk" )
{
WReaderVTK vtkReader( fileName );
m_dataSet = vtkReader.read();
......
src/modules/data/io/WReaderVTK.cpp
View file @ 69822b6b
//---------------------------------------------------------------------------
//
// Project: OpenWalnut ( http://www.openwalnut.org )
//
// Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
// For more information see http://www.openwalnut.org/copying
//
// This file is part of OpenWalnut.
//
// OpenWalnut is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// OpenWalnut is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
//
//---------------------------------------------------------------------------
#include <stdint.h>
#include <fstream>
#include <iosfwd>
#include <string>
#include <vector>
#include <boost/shared_ptr.hpp>
#include "core/common/WAssert.h"
#include "core/common/WIOTools.h"
#include "core/common/WLimits.h"
#include "core/common/WLogger.h"
#include "core/common/WStringUtils.h"
#include "core/dataHandler/exceptions/WDHIOFailure.h"
#include "core/dataHandler/exceptions/WDHNoSuchFile.h"
#include "core/dataHandler/exceptions/WDHParseError.h"
#include "core/dataHandler/WDataSet.h"
#include "core/dataHandler/WDataSetScalar.h"
#include "core/dataHandler/WDataSetVector.h"
#include "core/dataHandler/WDataSetDTI.h"
#include "core/dataHandler/WGridRegular3D.h"
#include "WReaderVTK.h"
WReaderVTK::WReaderVTK( std::string fname )
: WReader( fname )
{
}
WReaderVTK::~WReaderVTK() throw()
{
}
boost::shared_ptr< WDataSet > WReaderVTK::read()
{
boost::shared_ptr< WDataSet > ds;
m_ifs = boost::shared_ptr< std::ifstream >( new std::ifstream() );
m_ifs->open( m_fname.c_str(), std::ifstream::in | std::ifstream::binary );
if( !m_ifs || m_ifs->bad() )
{
throw WDHIOFailure( std::string( "internal error while opening" ) );
}
if( !readHeader() )
{
return ds;
}
// constructs the grid
boost::shared_ptr< WGridRegular3D > grid;
switch( m_domainType )
{
case STRUCTURED_POINTS:
grid = readStructuredPoints();
break;
case RECTILINEAR_GRID:
grid = readRectilinearGrid();
break;
default:
break;
}
if( !grid )
{
wlog::error( "WReaderVTK" ) << "Invalid grid!";
return ds;
}
boost::shared_ptr< WValueSetBase > values = readData( grid );
if( !values )
{
wlog::error( "WReaderVTK" ) << "Invalid values!";
return ds;
}
m_ifs->close();
switch( m_attributeType )
{
case SCALARS:
ds = boost::shared_ptr< WDataSet >( new WDataSetScalar( values, grid ) );
break;
case VECTORS:
ds = boost::shared_ptr< WDataSet >( new WDataSetVector( values, grid ) );
break;
case TENSORS:
case ARRAYS:
ds = boost::shared_ptr< WDataSet >( new WDataSetDTI( values, grid ) );
break;
default:
break;
}
return ds;
}
bool WReaderVTK::readHeader()
{
for( int i = 0; i < 4; ++i ) // strip first four lines
{
m_header.push_back( getLine( "reading first 4 lines (aka header)" ) );
}
// check if the header may be valid
if( ( m_header.at( 0 ) != "# vtk DataFile Version 3.0" )
&& ( m_header.at( 0 ) != "# vtk DataFile Version 2.0" ) )
{
wlog::error( "WReaderVTK" ) << "Unsupported format version string in VTK file: "
<< m_fname
<< ", "
<< m_header.at( 0 );
return false;
}
if( m_header.at( 1 ).size() > 256 )
{
wlog::warn( "WReaderVTK" ) << "Invalid header size of VTK file: "
<< m_fname
<< ", max. 256 but got: "
<< m_header.at( 1 ).size();
}
namespace su = string_utils;
if( ( su::toUpper( su::trim( m_header.at( 2 ) ) ) != "BINARY" )
&&( su::toUpper( su::trim( m_header.at( 2 ) ) ) != "ASCII" ) )
{
wlog::error( "WReaderVTK" ) << "VTK files in '" << m_header.at( 2 ) << "' format are not yet supported. Must be BINARY or ASCII";
return false;
}
m_isASCII = ( su::toUpper( su::trim( m_header.at( 2 ) ) ) == "ASCII" );
if( su::tokenize( m_header.at( 3 ) ).size() == 2 )
{
if( su::toUpper( su::tokenize( m_header.at( 3 ) )[1] ) == "STRUCTURED_POINTS" )
{
m_domainType = STRUCTURED_POINTS;
}
else if( su::toUpper( su::tokenize( m_header.at( 3 ) )[1] ) == "RECTILINEAR_GRID" )
{
m_domainType = RECTILINEAR_GRID;
}
else
{
wlog::error( "WReaderVTK" ) << "Cannot read this VTK DATASET type: " << su::tokenize( m_header.back() )[1];
return false;
}
}
else
{
wlog::error( "WReaderVTK" ) << "Cannot read this VTK DATASET type: " << su::tokenize( m_header.back() )[1];
return false;
}
return true;
}
boost::shared_ptr< WGridRegular3D > WReaderVTK::readStructuredPoints()
{
std::string line = getLine( "reading DIMENSIONS" );
std::vector< std::string > dimensions_line = string_utils::tokenize( line );
if( ( dimensions_line.size() != 4 ) && string_utils::toLower( dimensions_line.at( 0 ) ) != "dimensions" )
{
throw WDHParseError( std::string( "invalid DIMENSIONS token: " + line + ", expected DIMENSIONS." ) );
}
size_t dimensions[ 3 ];
dimensions[ 0 ] = getLexicalCast< size_t >( dimensions_line.at( 1 ), "" );
dimensions[ 1 ] = getLexicalCast< size_t >( dimensions_line.at( 2 ), "" );
dimensions[ 2 ] = getLexicalCast< size_t >( dimensions_line.at( 3 ), "" );
line = getLine( "reading ORIGIN" );
std::vector< std::string > origin_line = string_utils::tokenize( line );
if( ( origin_line.size() != 4 ) && string_utils::toLower( origin_line.at( 0 ) ) != "origin" )
{
throw WDHParseError( std::string( "invalid ORIGIN token: " + line + ", expected ORIGIN." ) );
}
double origin[ 3 ];
origin[ 0 ] = getLexicalCast< float >( origin_line.at( 1 ), "" );
origin[ 1 ] = getLexicalCast< float >( origin_line.at( 2 ), "" );
origin[ 2 ] = getLexicalCast< float >( origin_line.at( 3 ), "" );
line = getLine( "reading SPACING" );
std::vector< std::string > spacings_line = string_utils::tokenize( line );
if( ( spacings_line.size() != 4 ) && string_utils::toLower( spacings_line.at( 0 ) ) != "spacing" )
{
throw WDHParseError( std::string( "invalid DIMENSIONS token: " + line + ", expected DIMENSIONS." ) );
}
double spacings[ 3 ];
spacings[ 0 ] = getLexicalCast< float >( spacings_line.at( 1 ), "" );
spacings[ 1 ] = getLexicalCast< float >( spacings_line.at( 2 ), "" );
spacings[ 2 ] = getLexicalCast< float >( spacings_line.at( 3 ), "" );
WGridTransformOrtho transform( spacings[ 0 ], spacings[ 1 ], spacings[ 2 ] );
transform.translate( origin );
return boost::shared_ptr< WGridRegular3D >( new WGridRegular3D( dimensions[ 0 ], dimensions[ 1 ], dimensions[ 2 ], transform ) );
}
boost::shared_ptr< WGridRegular3D > WReaderVTK::readRectilinearGrid()
{
std::string line = getLine( "reading DIMENSIONS" );
std::vector< std::string > dimensions_line = string_utils::tokenize( line );
if( ( dimensions_line.size() != 4 ) && string_utils::toLower( dimensions_line.at( 0 ) ) != "dimensions" )
{
throw WDHParseError( std::string( "invalid DIMENSIONS token: " + line + ", expected DIMENSIONS." ) );
}
size_t dimensions[ 3 ];
dimensions[ 0 ] = getLexicalCast< size_t >( dimensions_line.at( 1 ), "" );
dimensions[ 1 ] = getLexicalCast< size_t >( dimensions_line.at( 2 ), "" );
dimensions[ 2 ] = getLexicalCast< size_t >( dimensions_line.at( 3 ), "" );
std::vector< float > xcoords;
readCoords( "X_COORDINATES", dimensions[ 0 ], xcoords );
std::vector< float > ycoords;
readCoords( "Y_COORDINATES", dimensions[ 1 ], ycoords );
std::vector< float > zcoords;
readCoords( "Z_COORDINATES", dimensions[ 2 ], zcoords );
wlog::warn( "WReaderVTK" ) << "Assuming evenly spaced rectilinear grid! This may not adhere to the domain data in the file!";
WGridTransformOrtho transform( xcoords.at( 1 ) - xcoords.at( 0 ), ycoords.at( 1 ) - ycoords.at( 0 ), zcoords.at( 1 ) - zcoords.at( 0 ) );
return boost::shared_ptr< WGridRegular3D >( new WGridRegular3D( dimensions[ 0 ], dimensions[ 1 ], dimensions[ 2 ], transform ) );
}
void WReaderVTK::readCoords( std::string const& name, std::size_t dim, std::vector< float >& coords )
{
std::string line = getLine( std::string( "reading " ) + name );
std::vector< std::string > l = string_utils::tokenize( line );
if( ( l.size() != 3 ) && string_utils::toLower( l.at( 0 ) ) != name )
{
throw WDHParseError( std::string( "invalid ") + name + " token: " + line + ", expected " + name + "!" );
}
readValuesFromFile( coords, dim );
}
void WReaderVTK::readValuesFromFile( std::vector< float >& values, std::size_t numValues )
{
values.clear();
if( this->m_isASCII )
{
wlog::debug( "WReaderVTK" ) << "Reading ASCII";
std::string line = getLine( "Data" );
std::vector< std::string > tokens = string_utils::tokenize( line );
size_t point = 0;
while( point < numValues )
{
for( size_t token = 0; token < tokens.size(); ++token )
{
values.push_back( getLexicalCast< float >( tokens.at( token ), "invalid data point" ) );
++point;
if( point > numValues )
{
throw WDHException( std::string( "Too many points in file" ) );
}
}
if( point < numValues )
{
line = getLine( "Data Loop" );
tokens = string_utils::tokenize( line );
}
}
}
else
{
wlog::debug( "WReaderVTK" ) << "Reading BINARY";
m_ifs->read( reinterpret_cast<char*>( &values[ 0 ] ), values.size() * sizeof( values[ 0 ] ) );
// TODO( hlawitschka ): do endianess stuff!!! switchByteOrderOfArray( data, data->size() );
}
}
// we currently do not handle the name flag but should set the name of the data set somewhere
boost::shared_ptr< WValueSetBase > WReaderVTK::readScalars( size_t nbPoints, const std::string& /*name*/ )
{
std::size_t pos = m_ifs->tellg();
std::string line = getLine( "LookupTableOrData" );
std::vector< std::string > tokens = string_utils::tokenize( line );
if( string_utils::toUpper( tokens.at( 0 ) ) == "LOOKUP_TABLE" )
{
pos = m_ifs->tellg(); // remember where we are for binary files
}
m_ifs->seekg( pos, std::ios::beg );
boost::shared_ptr< std::vector< float > > data( new std::vector< float >() );
readValuesFromFile( *data, nbPoints );
return boost::shared_ptr< WValueSetBase >( new WValueSet< float >( 0, 1, data, W_DT_FLOAT ) );
}
boost::shared_ptr< WValueSetBase > WReaderVTK::readVectors( size_t nbPoints, const std::string& /*name*/ )
{
int pos = m_ifs->tellg(); // remember where we are for binary files
// some files even have a lookup-table tag for vector-valued data
std::string line = getLine( "LookupTableOrData" );
std::vector< std::string > tokens = string_utils::tokenize( line );
if( string_utils::toUpper( tokens.at( 0 ) ) == "LOOKUP_TABLE" )
{
line = getLine( "Data" );
tokens = string_utils::tokenize( line );
pos = m_ifs->tellg(); // remember where we are for binary files
}
m_ifs->seekg( pos, std::ios::beg );
// We assume that all VTK fields store 3-dimensional vectors.
// There are other fields around, but there is no official VTK documentation dealing
// with 2D vectors and no unique interpretation of the data
const size_t dimension = 3;
boost::shared_ptr< std::vector< float > > data( new std::vector< float >( nbPoints * dimension ) );
readValuesFromFile( *data, nbPoints * dimension );
return boost::shared_ptr< WValueSetBase >( new WValueSet< float >( 1, 3, data, W_DT_FLOAT ) );
// line = getLine( "also eat the remaining newline after lines declaration" );
// WAssert( std::string( "" ) == line, "Found characters in file where nothing was expected." );
}
boost::shared_ptr< WValueSetBase > WReaderVTK::readTensors( size_t nbPoints, const std::string& /*name*/ )
{
int pos = m_ifs->tellg(); // remember where we are for binary files
// some files even have a lookup-table tag for vector-valued data
std::string line = getLine( "LookupTableOrData" );
std::vector< std::string > tokens = string_utils::tokenize( line );
if( string_utils::toUpper( tokens.at( 0 ) ) == "LOOKUP_TABLE" )
{
pos = m_ifs->tellg(); // remember where we are for binary files
}
m_ifs->seekg( pos, std::ios::beg );
std::size_t dimension = 9;
if( m_attributeType == ARRAYS )
{
dimension = 6;
}
boost::shared_ptr< std::vector< float > > data( new std::vector< float >( nbPoints * dimension ) );
readValuesFromFile( *data, nbPoints * dimension );
if( dimension == 9 )
{
std::size_t idx = 0;
for( std::size_t k = 0; k < data->size(); ++k )
{
std::size_t m = k % 9;
if( m != 3 && m != 6 && m != 7 )
{
( *data )[ idx ] = ( *data )[ k ];
++idx;
}
}
data->resize( nbPoints * 6 );
}
return boost::shared_ptr< WValueSetBase >( new WValueSet< float >( 1, 6, data, W_DT_FLOAT ) );
// line = getLine( "also eat the remaining newline after lines declaration" );
// WAssert( std::string( "" ) == line, "Found characters in file where nothing was expected." );
}
boost::shared_ptr< WValueSetBase > WReaderVTK::readData( boost::shared_ptr< WGridRegular3D > const& /* grid */ )
{
std::string line = getLine( "POINT_DATA declaration" );
std::vector< std::string > tokens = string_utils::tokenize( line );
if( tokens.size() != 2 || string_utils::toUpper( tokens.at( 0 ) ) != "POINT_DATA" )
{
throw WDHException( std::string( "Invalid SCALARS tag" ) );
}
size_t nbPoints = getLexicalCast< size_t >( tokens.at( 1 ), "parsing number of points" );
line = getLine( "SCALARS declaration" );
tokens = string_utils::tokenize( line );
std::string name = "Default";
if( tokens.size() >= 2 )
{
name = tokens.at( 1 );
}
if( tokens.size() == 3 && string_utils::toUpper( tokens.at( 0 ) ) == "SCALARS" )
{
m_attributeType = SCALARS;
return readScalars( nbPoints, name );
}
else if( tokens.size() == 3 && string_utils::toUpper( tokens.at( 0 ) ) == "VECTORS" )
{
m_attributeType = VECTORS;
return readVectors( nbPoints, name );
}
else if( tokens.size() == 3 && string_utils::toUpper( tokens.at( 0 ) ) == "TENSORS" )
{
m_attributeType = TENSORS;
return readTensors( nbPoints, name );
}
else if( tokens.size() == 4 && string_utils::toUpper( tokens.at( 0 ) ) == "ARRAYS" && tokens.at( 2 ) == "6" )
{
m_attributeType = ARRAYS;
return readTensors( nbPoints, name );
}
else
{
throw WDHException( std::string( "Invalid SCALARS or VECTORS tag" ) );
}
}
std::string WReaderVTK::getLine( const std::string& desc )
{
std::string line;
try
{
// we use '\n' as line termination under every platform so our files (which are most likely to be generated on Unix systems)
// can be read from all platforms not having those line termination symbols like e.g. windows ('\r\n').
std::getline( *m_ifs, line, '\n' );
}
catch( const std::ios_base::failure &e )
{
throw WDHIOFailure( std::string( "IO error while " + desc + " of VTK fiber file: " + m_fname + ", " + e.what() ) );
}
if( !m_ifs->good() )
{
throw WDHParseError( std::string( "Unexpected end of VTK fiber file: " + m_fname ) );
}
return line;
}
//---------------------------------------------------------------------------
//
// Project: OpenWalnut ( http://www.openwalnut.org )
//
// Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
// For more information see http://www.openwalnut.org/copying
//
// This file is part of OpenWalnut.
//
// OpenWalnut is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// OpenWalnut is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
//
//---------------------------------------------------------------------------
#include <stdint.h>
#include <fstream>
#include <iosfwd>
#include <string>
#include <vector>
#include <boost/shared_ptr.hpp>
#include "core/common/WAssert.h"
#include "core/common/WIOTools.h"
#include "core/common/WLimits.h"
#include "core/common/WLogger.h"
#include "core/common/WStringUtils.h"
#include "core/dataHandler/exceptions/WDHIOFailure.h"
#include "core/dataHandler/exceptions/WDHNoSuchFile.h"
#include "core/dataHandler/exceptions/WDHParseError.h"
#include "core/dataHandler/WDataSet.h"
#include "core/dataHandler/WDataSetScalar.h"
#include "core/dataHandler/WDataSetVector.h"
#include "core/dataHandler/WDataSetDTI.h"
#include "core/dataHandler/WGridRegular3D.h"
#include "WReaderVTK.h"
WReaderVTK::WReaderVTK( std::string fname )
: WReader( fname )
{
}
WReaderVTK::~WReaderVTK() throw()
{
}
boost::shared_ptr< WDataSet > WReaderVTK::read()
{
boost::shared_ptr< WDataSet > ds;
m_ifs = boost::shared_ptr< std::ifstream >( new std::ifstream() );
m_ifs->open( m_fname.c_str(), std::ifstream::in | std::ifstream::binary );
if( !m_ifs || m_ifs->bad() )
{
throw WDHIOFailure( std::string( "internal error while opening" ) );
}
if( !readHeader() )
{
return ds;
}
// constructs the grid
boost::shared_ptr< WGridRegular3D > grid;
switch( m_domainType )
{
case STRUCTURED_POINTS:
grid = readStructuredPoints();
break;
case RECTILINEAR_GRID:
grid = readRectilinearGrid();
break;
default:
break;
}
if( !grid )
{
wlog::error( "WReaderVTK" ) << "Invalid grid!";
return ds;
}
boost::shared_ptr< WValueSetBase > values = readData( grid );
if( !values )
{
wlog::error( "WReaderVTK" ) << "Invalid values!";
return ds;
}
m_ifs->close();
switch( m_attributeType )
{
case SCALARS:
ds = boost::shared_ptr< WDataSet >( new WDataSetScalar( values, grid ) );
break;
case VECTORS:
ds = boost::shared_ptr< WDataSet >( new WDataSetVector( values, grid ) );
break;
case TENSORS:
case ARRAYS:
ds = boost::shared_ptr< WDataSet >( new WDataSetDTI( values, grid ) );
break;