diff --git a/src/main/java/kieker/analysis/util/graph/export/dot/DotFileWriterStage.java b/src/main/java/kieker/analysis/util/graph/export/dot/DotFileWriterStage.java index 645307a3af0254f7bc18956605056eb25e01a2b1..88ff5b7d63d43a60982d910ee14ceb764eb763bf 100644 --- a/src/main/java/kieker/analysis/util/graph/export/dot/DotFileWriterStage.java +++ b/src/main/java/kieker/analysis/util/graph/export/dot/DotFileWriterStage.java @@ -11,17 +11,25 @@ import kieker.analysis.util.graph.util.FileExtension; public class DotFileWriterStage extends DotWriterStage { public DotFileWriterStage(final Function<Graph, String> fileNameMapper) { + this(fileNameMapper, new SimpleDotExportConfiguration()); + } + + public DotFileWriterStage(final Function<Graph, String> fileNameMapper, final DotExportConfiguration exportConfiguration) { super(fileNameMapper.andThen(fileName -> { try { return new FileWriter(fileName); } catch (IOException e) { throw new IllegalArgumentException(e); } - })); + }), exportConfiguration); } public DotFileWriterStage(final String outputDirectory) { - this(new SimpleFileNameMapper(outputDirectory, FileExtension.DOT)); + this(outputDirectory, new SimpleDotExportConfiguration()); + } + + public DotFileWriterStage(final String outputDirectory, final DotExportConfiguration exportConfiguration) { + this(new SimpleFileNameMapper(outputDirectory, FileExtension.DOT), exportConfiguration); } }