diff --git a/src/main/java/kieker/analysis/graph/export/dot/AbstractDotFileWriterStage.java b/src/main/java/kieker/analysis/graph/export/dot/AbstractDotFileWriterStage.java index fdd6eefc9d141d0bac3733453718b9a5672d4fe7..2368b3c8e659bb06ba011365cf88ad94c029c2e6 100644 --- a/src/main/java/kieker/analysis/graph/export/dot/AbstractDotFileWriterStage.java +++ b/src/main/java/kieker/analysis/graph/export/dot/AbstractDotFileWriterStage.java @@ -4,6 +4,8 @@ import java.io.FileWriter; import java.io.IOException; import java.io.Writer; +import kieker.analysis.graph.Graph; + public abstract class AbstractDotFileWriterStage extends AbstractDotWriterStage { public AbstractDotFileWriterStage() { @@ -11,14 +13,14 @@ public abstract class AbstractDotFileWriterStage extends AbstractDotWriterStage } @Override - protected Writer getWriter() { + protected Writer getWriter(final Graph graph) { try { - return new FileWriter(getFileName()); + return new FileWriter(getFileName(graph)); } catch (IOException e) { throw new IllegalStateException(e); } } - protected abstract String getFileName(); + protected abstract String getFileName(final Graph graph); } diff --git a/src/main/java/kieker/analysis/graph/export/dot/AbstractDotWriterStage.java b/src/main/java/kieker/analysis/graph/export/dot/AbstractDotWriterStage.java index 576314e124a4b1785ceb4bc6129fcdb3d432667e..37a206bda28713843209036f9eb2850d6449680b 100644 --- a/src/main/java/kieker/analysis/graph/export/dot/AbstractDotWriterStage.java +++ b/src/main/java/kieker/analysis/graph/export/dot/AbstractDotWriterStage.java @@ -14,10 +14,10 @@ public abstract class AbstractDotWriterStage extends AbstractConsumerStage<Graph @Override protected final void execute(final Graph graph) { - DotExporter dotExporter = new DotExporter(graph, getWriter()); + DotExporter dotExporter = new DotExporter(graph, getWriter(graph)); dotExporter.transform(); } - protected abstract Writer getWriter(); + protected abstract Writer getWriter(final Graph graph); }