X-Git-Url: http://bilbo.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/4e8a86c1c16d0515922517efce7620528cde9903..35a644bdf9d0c5603c9483f03f913e4a649638d6:/tools/graphicator/graphicator.cpp

diff --git a/tools/graphicator/graphicator.cpp b/tools/graphicator/graphicator.cpp
index 2d5e63c0f5..368892bd08 100644
--- a/tools/graphicator/graphicator.cpp
+++ b/tools/graphicator/graphicator.cpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2008-2022. The SimGrid Team.
+/* Copyright (c) 2008-2023. The SimGrid Team.
  * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
@@ -10,9 +10,23 @@
 int main(int argc, char** argv)
 {
   simgrid::s4u::Engine e(&argc, argv);
-  xbt_assert(argc == 3, "Usage: %s <platform_file.xml> <graphviz_file.dot>", argv[0]);
+  xbt_assert(argc == 3, "Usage: %s <platform_file.xml> <graphviz_file.dot|graphviz_file.csv>", argv[0]);
 
   e.load_platform(argv[1]);
-  simgrid::instr::platform_graph_export_graphviz(argv[2]);
+  e.seal_platform();
+
+  const std::string outputfile(argv[2]);
+  const std::string extension = outputfile.substr(outputfile.find_last_of(".") + 1);
+  if(extension == "csv") {
+    printf("Dumping to CSV file\n");
+    simgrid::instr::platform_graph_export_csv(outputfile);
+  }
+  else if(extension == "dot") {
+    printf("Dumping to DOT file\n");
+    simgrid::instr::platform_graph_export_graphviz(outputfile);
+  }
+  else {
+    xbt_die("Unknown output file format, please use '.dot' or .csv' extension");
+  }
   return 0;
 }