saberbud/PoissonRecon
1
0
Fork 0
mirror of https://github.com/CloudCompare/PoissonRecon.git synced 2026-08-31 01:20:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Catch exception when failing to write image.

Browse Source 
Output transformation when writing grid.
This commit is contained in:
mkazhdan
2019-01-27 18:16:08 -05:00
parent f4f47883db
commit 3107d150fb
6 changed files with 83 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Src/PoissonRecon.cpp
+25 -16
View File
@@ -58,7 +58,7 @@ cmdLineParameter< char* >
In( "in" ) ,
Out( "out" ) ,
TempDir( "tempDir" ) ,
VoxelGrid( "voxel" ) ,
Grid( "grid" ) ,
Tree( "tree" ) ,
Transform( "xForm" );
@@ -71,7 +71,7 @@ cmdLineReadable
ASCII( "ascii" ) ,
Density( "density" ) ,
LinearFit( "linearFit" ) ,
PrimalVoxel( "primalVoxel" ) ,
PrimalGrid( "primalGrid" ) ,
ExactInterpolation( "exact" ) ,
Normals( "normals" ) ,
Colors( "colors" ) ,
@@ -116,7 +116,7 @@ cmdLineReadable* params[] =
&ConfidenceBias ,
&BaseDepth , &BaseVCycles ,
&PointWeight ,
&VoxelGrid , &Threads ,
&Grid , &Threads ,
&Tree ,
&Density ,
&FullDepth ,
@@ -125,7 +125,7 @@ cmdLineReadable* params[] =
&Colors ,
&Normals ,
&LinearFit ,
&PrimalVoxel ,
&PrimalGrid ,
&TempDir ,
&ExactInterpolation ,
&Performance ,
@@ -139,7 +139,7 @@ void ShowUsage(char* ex)
printf( "Usage: %s\n" , ex );
printf( "\t --%s <input points>\n" , In.name );
printf( "\t[--%s <ouput triangle mesh>]\n" , Out.name );
printf( "\t[--%s <ouput voxel grid>]\n" , VoxelGrid.name );
printf( "\t[--%s <ouput grid>]\n" , Grid.name );
printf( "\t[--%s <ouput fem tree>]\n" , Tree.name );
#ifndef FAST_COMPILE
printf( "\t[--%s <b-spline degree>=%d]\n" , Degree.name , Degree.value );
@@ -147,7 +147,7 @@ void ShowUsage(char* ex)
for( int i=0 ; i<BOUNDARY_COUNT ; i++ ) printf( "\t\t%d] %s\n" , i+1 , BoundaryNames[i] );
#endif // !FAST_COMPILE
printf( "\t[--%s <maximum reconstruction depth>=%d]\n" , Depth.name , Depth.value );
printf( "\t[--%s <voxel width>]\n" , Width.name );
printf( "\t[--%s <grid width>]\n" , Width.name );
printf( "\t[--%s <full depth>=%d]\n" , FullDepth.name , FullDepth.value );
printf( "\t[--%s <coarse MG solver depth>=%d]\n" , BaseDepth.name , BaseDepth.value );
printf( "\t[--%s <coarse MG solver v-cycles>=%d]\n" , BaseVCycles.name , BaseVCycles.value );
@@ -171,7 +171,7 @@ void ShowUsage(char* ex)
printf( "\t[--%s]\n" , Performance.name );
printf( "\t[--%s]\n" , Density.name );
printf( "\t[--%s]\n" , LinearFit.name );
printf( "\t[--%s]\n" , PrimalVoxel.name );
printf( "\t[--%s]\n" , PrimalGrid.name );
printf( "\t[--%s]\n" , ASCII.name );
printf( "\t[--%s]\n" , NoComments.name );
printf( "\t[--%s]\n" , TempDir.name );
@@ -380,7 +380,9 @@ bool WriteImage( const Real *values , int res , const char *fileName , bool verb
unsigned char color = (unsigned char )std::min< Real >( (Real)255. , std::max< Real >( (Real)0. , v ) );
for( int c=0 ; c<3 ; c++ ) pixels[i*3+c ] = color;
}
bool success = ImageWriter::Write( fileName , pixels , res , res , 3 );
bool success = true;
try{ ImageWriter::Write( fileName , pixels , res , res , 3 ); }
catch( MKExceptions::Exception & ){ success = false; }
delete[] pixels;
return success;
}
@@ -391,9 +393,8 @@ void WriteGrid( const Real *values , int res , const char *fileName )
int resolution = 1;
for( int d=0 ; d<Dim ; d++ ) resolution *= res;
char *ext = GetFileExtension( fileName );
FILE *fp = fopen( fileName , "wb" );
if( !fp ) ERROR_OUT( "Failed to open voxel file for writing: %s" , fileName );
if( !fp ) ERROR_OUT( "Failed to open grid file for writing: %s" , fileName );
else
{
fwrite( &res , sizeof(int) , 1 , fp );
@@ -406,9 +407,7 @@ void WriteGrid( const Real *values , int res , const char *fileName )
delete[] fValues;
}
fclose( fp );
DeletePointer( values );
}
delete[] ext;
}
template< class Real , typename ... SampleData , unsigned int ... FEMSigs >
@@ -660,18 +659,28 @@ void Execute( int argc , char* argv[] , UIntPack< FEMSigs ... > )
fclose( fp );
}
if( VoxelGrid.set )
if( Grid.set )
{
int res = 0;
profiler.start();
Pointer( Real ) values = tree.template regularGridEvaluate< true >( solution , res , -1 , PrimalVoxel.set );
Pointer( Real ) values = tree.template regularGridEvaluate< true >( solution , res , -1 , PrimalGrid.set );
int resolution = 1;
for( int d=0 ; d<Dim ; d++ ) resolution *= res;
#pragma omp parallel for
for( int i=0 ; i<resolution ; i++ ) values[i] -= isoValue;
profiler.dumpOutput( "Got voxel grid:" );
if( !WriteImage< Real , DIMENSION >( values , res , VoxelGrid.value , Verbose.set ) ) WriteGrid< Real , DIMENSION >( values , res , VoxelGrid.value );
profiler.dumpOutput( "Got grid:" );
if( !WriteImage< Real , DIMENSION >( values , res , Grid.value , Verbose.set ) ) WriteGrid< Real , DIMENSION >( values , res , Grid.value );
DeletePointer( values );
if( Verbose.set )
{
printf( "Transform:\n" );
for( int i=0 ; i<Dim+1 ; i++ )
{
printf( "\t" );
for( int j=0 ; j<Dim+1 ; j++ ) printf( " %f" , iXForm(j,i) );
printf( "\n" );
}
}
}
if( Out.set )