1 files changed, 14 insertions, 4 deletions

diff --git a/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps_weighted_by_arctan_of_their_persistence.cpp b/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps_weighted_by_arctan_of_their_persistence.cpp
index 8496e33e..7cb7eeb7 100644
--- a/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps_weighted_by_arctan_of_their_persistence.cpp
+++ b/src/Gudhi_stat/utilities/persistence_heat_maps/create_persistence_heat_maps_weighted_by_arctan_of_their_persistence.cpp
@@ -1,4 +1,4 @@
-/*    This file is part of the Gudhi Library. The Gudhi library
+	/*    This file is part of the Gudhi Library. The Gudhi library
  *    (Geometric Understanding in Higher Dimensions) is a generic C++
  *    library for computational topology.
  *
@@ -40,8 +40,11 @@ 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 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";
@@ -53,8 +56,15 @@ int main( int argc , char** argv )
 	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] );
 	}
@@ -67,7 +77,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<arc_tan_of_persistence_of_point> l( filenames[i] , filter, false , size_of_grid , min_ , max_ );		
+		Persistence_heat_maps<arc_tan_of_persistence_of_point> l( filenames[i] , filter, false , size_of_grid , min_ , max_ , dimension );		
 		
 		std::stringstream ss;
 		ss << filenames[i] << ".mps";