diff --git a/src/main/java/GraphDataObjectWriter.java b/src/main/java/GraphDataObjectWriter.java index 3cf79e8..6b6a832 100644 --- a/src/main/java/GraphDataObjectWriter.java +++ b/src/main/java/GraphDataObjectWriter.java @@ -1,3 +1,5 @@ +import org.jgrapht.Graph; + import java.io.BufferedOutputStream; import java.io.FileOutputStream; import java.io.IOException; @@ -7,6 +9,7 @@ public class GraphDataObjectWriter { private GraphWithMapData data; private String filename; + private boolean verbose = true; public GraphDataObjectWriter(String filename, GraphWithMapData data) { if(!filename.matches(".*\\.ser")){ @@ -16,13 +19,24 @@ public class GraphDataObjectWriter { this.data = data; } + public GraphDataObjectWriter(String filename, GraphWithMapData data, boolean verbose) { + this.verbose = verbose; + if(!filename.matches(".*\\.ser")){ + filename = filename + ".ser"; + } + this.filename = filename; + this.data = data; + } + public void writeDataToFile() { try (BufferedOutputStream bufferedOut = new BufferedOutputStream(new FileOutputStream(filename)); ObjectOutputStream out = new ObjectOutputStream(bufferedOut); ){ - System.out.println("Writing graph and occupancy data to file. This may take some time."); - System.out.println("File I/O time is not included in results."); + if(verbose) { + System.out.println("Writing graph and occupancy data to file. This may take some time."); + System.out.println("File I/O time is not included in results."); + } out.writeObject(data); } catch (IOException ex) { ex.printStackTrace();