1 files changed, 12 insertions, 3 deletions
diff --git a/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/chartbuilder/ChartBuilderFactory.java b/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/chartbuilder/ChartBuilderFactory.java
index 4c5d4fa09f..f4e11a2c4d 100644
--- a/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/chartbuilder/ChartBuilderFactory.java
+++ b/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/chartbuilder/ChartBuilderFactory.java
@@ -20,18 +20,27 @@
 package org.apache.qpid.disttest.charting.chartbuilder;
 
 import org.apache.qpid.disttest.charting.ChartType;
+import org.apache.qpid.disttest.charting.seriesbuilder.SeriesBuilder;
 
 public class ChartBuilderFactory
 {
 
-    public static ChartBuilder createChartBuilder(ChartType chartType)
+    public static ChartBuilder createChartBuilder(ChartType chartType, SeriesBuilder seriesBuilder)
     {
         switch (chartType)
         {
         case LINE:
-            return new LineChartBuilder();
+            return new LineChartBuilder(seriesBuilder);
+        case LINE3D:
+            return new LineChart3DBuilder(seriesBuilder);
         case BAR:
-            return new BarChartBuilder();
+            return new BarChartBuilder(seriesBuilder);
+        case BAR3D:
+            return new BarChart3DBuilder(seriesBuilder);
+        case XYLINE:
+            return new XYLineChartBuilder(seriesBuilder);
+        case STATISTICAL_BAR:
+            return new StatisticalBarCharBuilder(seriesBuilder);
         default:
             throw new IllegalArgumentException("Unknown chart type " + chartType);
         }