Skip to content

Commit b4f5089

Browse files
committed
ENH: Example: check command line arguments
1 parent 51adc6b commit b4f5089

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

examples/example.cxx

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,21 @@ int example_regularstepgradientdescent( int argc, char* argv[] )
230230

231231
int main( int argc, char* argv[] )
232232
{
233-
return example_gradientdescent( argc, argv );
234-
//return example_regularstepgradientdescent( argc, argv );
233+
if( argc < 3 )
234+
{
235+
std::cerr << "Usage: MinimalPathExamples <SpeedImage> <OutputImage> [UseRegularStepGradientDescentOptimizer]" << std::endl;
236+
return EXIT_FAILURE;
237+
}
238+
239+
int useRegularStepGradientDescentOptimizer = 0;
240+
if ( argc > 3 )
241+
{
242+
useRegularStepGradientDescentOptimizer = std::atoi( argv[3] );
243+
}
244+
245+
if ( useRegularStepGradientDescentOptimizer )
246+
{
247+
return example_regularstepgradientdescent( argc, argv );
248+
}
249+
return example_gradientdescent( argc, argv );
235250
}

0 commit comments

Comments
 (0)