1 files changed, 15 insertions, 7 deletions
diff --git a/example/Bitmap_cubical_complex/Bitmap_cubical_complex_periodic_boundary_conditions.cpp b/example/Bitmap_cubical_complex/Bitmap_cubical_complex_periodic_boundary_conditions.cpp
index 839a4c89..122160a2 100644
--- a/example/Bitmap_cubical_complex/Bitmap_cubical_complex_periodic_boundary_conditions.cpp
+++ b/example/Bitmap_cubical_complex/Bitmap_cubical_complex_periodic_boundary_conditions.cpp
@@ -30,6 +30,7 @@
 #include <iostream>
 #include <sstream>
 #include <vector>
+#include <string>
 
 int main(int argc, char** argv) {
   std::cout << "This program computes persistent homology, by using " <<
@@ -40,9 +41,6 @@ int main(int argc, char** argv) {
       "assume that the cells are in the lexicographical order. See CubicalOneSphere.txt or CubicalTwoSphere.txt for" <<
       " example.\n" << std::endl;
 
-  int p = 2;
-  double min_persistence = 0;
-
   if (argc != 2) {
     std::cerr << "Wrong number of parameters. Please provide the name of a file with a Perseus style bitmap at " <<
         "the input. The program will now terminate.\n";
@@ -58,16 +56,26 @@ int main(int argc, char** argv) {
   typedef Gudhi::persistent_cohomology::Persistent_cohomology<Bitmap_cubical_complex, Field_Zp> Persistent_cohomology;
   // Compute the persistence diagram of the complex
   Persistent_cohomology pcoh(b, true);
+
+  int p = 2;
+  double min_persistence = 0;
   pcoh.init_coefficients(p);  // initializes the coefficient field for homology
   pcoh.compute_persistent_cohomology(min_persistence);
 
-  std::stringstream ss;
-  ss << argv[1] << "_persistence";
-  std::ofstream out(ss.str().c_str());
+  std::string output_file_name(argv[1]);
+  output_file_name += "_persistence";
+
+  std::size_t last_in_path = output_file_name.find_last_of("/\\");
+
+  if (last_in_path != std::string::npos) {
+    output_file_name = output_file_name.substr(last_in_path+1);
+  }
+
+  std::ofstream out(output_file_name.c_str());
   pcoh.output_diagram(out);
   out.close();
 
-  std::cout << "Result in file: " << ss.str().c_str() << "\n";
+  std::cout << "Result in file: " << output_file_name << "\n";
 
   return 0;
 }