diff options
Diffstat (limited to 'src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp')
-rw-r--r-- | src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp b/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp index 5d2100aa..d9c7aded 100644 --- a/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp +++ b/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps.cpp @@ -40,26 +40,32 @@ int main( int argc , char** argv ) std::cout << "The first parameter of a program is an integer, a size of a grid.\n"; std::cout << "The second and third parameters are min and max of the grid. If you want those numbers to be computed based on the data, set them both to -1 \n"; std::cerr << "The fourth parameter is an integer, the standard deviation of a gaussian kernel expressed in a number of pixels \n"; - std::cout << "The remaining parameters are the names of files with persistence diagrams. \n"; + std::cout << "The fifth parameter of this program is a dimension of persistence that will be used in creation of the persistence heat maps."; + std::cout << "If our input files contain persistence pairs of various dimension, as a fifth parameter of the procedure please provide the dimension of persistence you want to use."; + std::cout << "If in your file there are only birth-death pairs of the same dimension, set the first parameter to -1." << std::endl; + std::cout << "The remaining parameters are the names of files with persistence diagrams. \n"; if ( argc < 5 ) { std::cout << "Wrong parameter list, the program will now terminate \n"; return 1; - } - + } size_t size_of_grid = (size_t)atoi( argv[1] ); double min_ = atof( argv[2] ); double max_ = atof( argv[3] ); size_t stdiv = atof( argv[4] ); + unsigned dimension = std::numeric_limits<unsigned>::max(); + int dim = atoi( argv[5] ); + if ( (dim != -1) && (dim >= 0) ) + { + dimension = (unsigned)dim; + } std::vector< const char* > filenames; - for ( int i = 5 ; i < argc ; ++i ) + for ( int i = 6 ; i < argc ; ++i ) { filenames.push_back( argv[i] ); } - - std::cout << "Creating persistence heat maps...\n"; std::vector< std::vector<double> > filter = create_Gaussian_filter(stdiv,1); @@ -67,7 +73,7 @@ int main( int argc , char** argv ) for ( size_t i = 0 ; i != filenames.size() ; ++i ) { std::cout << "Creating a heat map based on a file : " << filenames[i] << std::endl; - Persistence_heat_maps<constant_scaling_function> l( filenames[i] , filter, false , size_of_grid , min_ , max_ ); + Persistence_heat_maps<constant_scaling_function> l( filenames[i] , filter, false , size_of_grid , min_ , max_ , dimension ); std::stringstream ss; ss << filenames[i] << ".mps"; |