From b12521ed664767ea12a2079f29d6b39320fe16db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B6ren=20Henning?= <stu114708@informatik.uni-kiel.de>
Date: Mon, 30 Nov 2015 13:03:56 +0100
Subject: [PATCH] Add stages for trace graph building to configuration

---
 .../TraceAggregationComposite.java            |  1 +
 .../TraceReconstructionComposite.java         | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
index 7cf2656b..72dd5d48 100644
--- a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
+++ b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
@@ -51,6 +51,7 @@ public final class TraceAggregationComposite extends AbstractCompositeStage {
 		// super.connectPorts(this.statisticsDecorator.getOutputPort(), this.tracesCollector.getInputPort());
 
 		// TODO New from Sören
+		// TODO Move to own composite
 
 		String graphFilesOutputDir = "example/event monitoring log/output"; // TODO Temp hard coded
 
diff --git a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceReconstructionComposite.java b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceReconstructionComposite.java
index 985fef20..cb5d9462 100644
--- a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceReconstructionComposite.java
+++ b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceReconstructionComposite.java
@@ -18,6 +18,10 @@ package kieker.analysis.stage.tracediagnosis;
 
 import java.util.List;
 
+import kieker.analysis.trace.graphoutput.DotGraphWriter;
+import kieker.analysis.trace.graphoutput.GraphMLWriter;
+import kieker.analysis.trace.traversal.NamedGraph;
+import kieker.analysis.trace.traversal.TraceTraverserStage;
 import kieker.analysis.traceanalysisdomain.Trace;
 import kieker.common.record.IMonitoringRecord;
 import kieker.common.record.controlflow.OperationExecutionRecord;
@@ -65,6 +69,22 @@ public final class TraceReconstructionComposite extends AbstractCompositeStage {
 		super.connectPorts(merger.getOutputPort(), distributor.getInputPort());
 		super.connectPorts(distributor.getNewOutputPort(), this.tracesCollector.getInputPort());
 		super.connectPorts(distributor.getNewOutputPort(), this.statisticsDecorator.getInputPort());
+
+		// TODO New from Sören
+		// TODO move to own composite
+
+		String graphFilesOutputDir = "example/event monitoring log/output"; // TODO Temp hard coded
+
+		TraceTraverserStage traceTraverserStage = new TraceTraverserStage();
+		final Distributor<NamedGraph> graphDistributor = new Distributor<>(new CopyByReferenceStrategy());
+		GraphMLWriter graphMLWriter = new GraphMLWriter(graphFilesOutputDir);
+		DotGraphWriter dotGraphWriter = new DotGraphWriter(graphFilesOutputDir);
+
+		super.connectPorts(distributor.getNewOutputPort(), traceTraverserStage.getInputPort());
+		super.connectPorts(traceTraverserStage.getOutputPort(), graphDistributor.getInputPort());
+		super.connectPorts(graphDistributor.getNewOutputPort(), graphMLWriter.getInputPort());
+		super.connectPorts(graphDistributor.getNewOutputPort(), dotGraphWriter.getInputPort());
+
 	}
 
 	public int countIncompleteTraces() {
-- 
GitLab