Refactor CSV and Net loading to use InputStream for better flexibility

src/main/java/net/berack/upo/valpre/Main.java

@@ -3,31 +3,83 @@ package net.berack.upo.valpre;
 import java.io.File;
 import java.net.URISyntaxException;
 import java.util.Arrays;
+import java.util.HashMap;
 
 public class Main {
-    public static void main(String[] args) throws Exception {
+    public static void main(String[] args) {
         if (args.length == 0)
-            exit();
+            exit("No program specified!");
 
-        var subArgs = Arrays.copyOfRange(args, 1, args.length);
-        switch (args[0]) {
-            case "simulation":
-                var sim = new Simulation(subArgs);
-                sim.run();
-                break;
-            case "plot":
-                var plot = new Plot(subArgs);
-                plot.show();
-                break;
-            default:
-                exit();
+        try {
+            var program = args[0];
+            var subArgs = Arrays.copyOfRange(args, 1, args.length);
+            var param = Main.getParameters(program, subArgs);
+            switch (program) {
+                case "simulation" -> {
+                    var net = param.get("net");
+                    var csv = param.get("csv");
+                    var runs = param.getOrDefault("runs", Integer::parseInt, 100);
+                    var seed = param.getOrDefault("seed", Long::parseLong, 2007539552L);
+                    var p = param.getOrDefault("p", Boolean::parseBoolean, false);
+
+                    var sim = new Simulation(net, seed, runs, p, csv);
+                    sim.run();
+                }
+                case "plot" -> {
+                    var csv = param.get("csv");
+                    var plot = new Plot(csv);
+                    plot.show();
+                }
+                default -> exit("Invalid program!");
+            }
+        } catch (Exception e) {
+            exit(e.getMessage());
         }
     }
 
-    public static void exit() throws URISyntaxException {
-        var uri = Main.class.getProtectionDomain().getCodeSource().getLocation().toURI();
-        var name = new File(uri).getName();
-        System.out.println("Usage: java -jar " + name + ".jar [simulation|plot] [args]");
-        System.exit(1);
+    /**
+     * Get the parameters from the arguments.
+     * 
+     * @param program the program to run
+     * @param args    the arguments to parse
+     * @return the parameters
+     */
+    private static Parameters getParameters(String program, String[] args) {
+        var arguments = new HashMap<String, Boolean>();
+        arguments.put("p", false);
+        arguments.put("seed", true);
+        arguments.put("runs", true);
+        arguments.put("net", true);
+        arguments.put("csv", true);
+
+        var descriptions = new HashMap<String, String>();
+        descriptions.put("p", "Add this if you want the simulation to use threads (one each run).");
+        descriptions.put("seed", "The seed of the simulation.");
+        descriptions.put("runs", "How many runs the simulator should run.");
+        descriptions.put("net", "The file net to use. Use example1.net or example2.net for the provided ones.");
+
+        var csvDesc = switch (program) {
+            case "simulation" -> "The filename for saving every run statistics.";
+            case "plot" -> "The filename that contains the previous saved runs.";
+            default -> "";
+        };
+        descriptions.put("csv", csvDesc);
+
+        return Parameters.getArgsOrHelper(args, "-", arguments, descriptions);
+    }
+
+    /**
+     * Exit the program with an error message.
+     */
+    public static void exit(String message) {
+        try {
+            var uri = Main.class.getProtectionDomain().getCodeSource().getLocation().toURI();
+            var name = new File(uri).getName();
+            System.out.println(message);
+            System.out.println("Usage: java -jar " + name + ".jar [simulation|plot] [args]");
+            System.exit(1);
+        } catch (URISyntaxException e) {
+            e.printStackTrace();
+        }
     }
 }