Added result rendering via JFreeChart

Author: Dragos Manolescu

build.sbt

@@ -54,7 +54,8 @@ libraryDependencies ++= Seq (
     "nl.grons" %% "metrics-scala" % "2.2.0",
     "fr.janalyse" %% "janalyse-jmx" % "0.6.1" % "compile",
     "net.sf.jopt-simple" % "jopt-simple" % "4.4",
-    "play" % "play-json_2.10" % "2.2-SNAPSHOT"
+    "play" % "play-json_2.10" % "2.2-SNAPSHOT",
+     "org.jfree" % "jfreechart" % "1.0.14"
 )
 
 /* you may need these repos */

data/twitter/tweet.json

@@ -24,7 +24,7 @@ case class Experiment(exclude: Set[String], warmUpIterations: Int=5) {
       }))
     }
 
-    println("Parsing measurement (%d warmup, %d iterations)...".format(warmUpIterations, iterations))
+    println("Parsing measurement (%d warmup, %d iterations on %s)...".format(warmUpIterations, iterations, adaptorsToTest.map(_.getName).mkString(", ")))
     run(iterations, adaptorsToTest.toSeq)
   }
 
@@ -37,7 +37,7 @@ case class Experiment(exclude: Set[String], warmUpIterations: Int=5) {
     }
 
     val targetAdaptors = adaptorsToTest.filter(_.hasMap)
-    println("Mapping measurement (%d warmup, %d iterations)...".format(warmUpIterations, iterations))
+    println("Mapping measurement (%d warmup, %d iterations on %s)...".format(warmUpIterations, iterations, adaptorsToTest.map(_.getName).mkString(", ")))
     run(iterations, targetAdaptors)
   }
 

src/main/scala/com/microWorkflow/jsonScalaPerftest/Experiment.scala

@@ -7,7 +7,8 @@ package com.microWorkflow.jsonScalaPerftest
 
 import joptsimple._
 import scala.collection.JavaConversions._
-import collection.immutable.HashSet
+import scala.collection.immutable.HashSet
+import com.microWorkflow.jsonScalaPerftest.output.{ConsoleReporter, ChartReporter}
 
 
 object Main {
@@ -34,6 +35,12 @@ object Main {
       .withValuesSeparatedBy(',')
       .describedAs("exclude")
       .ofType(classOf[String])
+    val reportOpt = argParser
+      .accepts("report", "Result generation, c=console, b=bar chart.")
+      .withRequiredArg()
+      .describedAs("Character; default is 'c'")
+      .ofType(classOf[String])
+      .defaultsTo("c")
 
     val options = try {
       argParser.parse(args: _*)
@@ -57,12 +64,15 @@ object Main {
     println("Running %d iterations %s object mapping".format(iterations, if (doMap) "with" else "without"))
     val experiment = Experiment(exclude, warmUpIterations)
     val ms = if (doMap) experiment.measureMapping(iterations) else experiment.measureParsing(iterations)
-    for (m <- ms) {
-      print("Category: %s\n".format(m._1))
-      for (d <- m._2.iterator)
-      print("\t dataset '%s', measurement: %s\n".format(d._1, d._2))
-    }
 
+    if (options.has(reportOpt) && options.valueOf[String](reportOpt) == "b") {
+      val chart = new ChartReporter(if (doMap) "JSON Parsing and Mapping" else "JSON Parsing", ms)
+      chart.view
+      System.in.read()
+    } else {
+      val reporter = new ConsoleReporter(ms)
+      reporter.printResults()
+    }
     sys.exit()
   }
 }

src/main/scala/com/microWorkflow/jsonScalaPerftest/Main.scala

@@ -0,0 +1,116 @@
+package com.microWorkflow.jsonScalaPerftest.output
+
+import scala.collection.immutable.HashMap
+import com.microWorkflow.jsonScalaPerftest.Measurement
+import org.jfree.data.category.{CategoryDataset, DefaultCategoryDataset}
+import org.jfree.ui.{RefineryUtilities, ApplicationFrame}
+import org.jfree.chart.{ChartFactory, JFreeChart, ChartPanel}
+import scala.Predef._
+import org.jfree.chart.plot.PlotOrientation
+import org.jfree.chart.axis.{CategoryLabelPositions, NumberAxis}
+import org.jfree.chart.renderer.category.BarRenderer
+import java.awt.image.BufferedImage
+import java.io.FileOutputStream
+import java.awt.geom.Rectangle2D
+import com.sun.image.codec.jpeg.JPEGCodec
+
+
+class ChartReporter(title: String, results: Array[(String, HashMap[String, Measurement])]) extends ApplicationFrame(title) {
+
+  import java.awt.Color
+  import java.awt.Dimension
+
+  val graphDataSet = createDataset()
+
+  val chart = createChart(graphDataSet)
+  val chartPanel = new ChartPanel(chart)
+  chartPanel.setPreferredSize(new Dimension(500, 270))
+  setContentPane(chartPanel)
+
+  def createDataset(): DefaultCategoryDataset = {
+    val categoryDataSet = new DefaultCategoryDataset()
+
+    for ((category, datasets) <- results) {
+      for ((dataset, measurement) <- datasets) {
+        categoryDataSet.addValue(measurement.value, measurement.name, dataset)
+      }
+    }
+    categoryDataSet
+  }
+
+  def createChart(dataset: CategoryDataset): JFreeChart = {
+
+    // create the chart...
+    val chart = ChartFactory.createBarChart(
+      title, // chart title
+      "JSON Data set", // domain axis label
+      "Time (ms)", // range axis label
+      dataset, // data
+      PlotOrientation.VERTICAL, // orientation
+      true, // include legend
+      true, // tooltips?
+      false // URLs?
+    )
+
+    // NOW DO SOME OPTIONAL CUSTOMISATION OF THE CHART...
+
+    // set the background color for the chart...
+    chart.setBackgroundPaint(Color.white)
+
+    // get a reference to the plot for further customisation...
+    val plot = chart.getCategoryPlot()
+    plot.setBackgroundPaint(Color.lightGray)
+    plot.setDomainGridlinePaint(Color.white)
+    plot.setRangeGridlinePaint(Color.white)
+
+    // set the range axis to display integers only...
+    val rangeAxis = plot.getRangeAxis().asInstanceOf[NumberAxis]
+    rangeAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits())
+
+    // disable bar outlines...
+    val renderer = plot.getRenderer().asInstanceOf[BarRenderer]
+    renderer.setDrawBarOutline(false)
+    renderer.setShadowVisible(false)
+
+    val domainAxis = plot.getDomainAxis()
+    domainAxis.setCategoryLabelPositions(
+      CategoryLabelPositions.createUpRotationLabelPositions(Math.PI / 6.0)
+    )
+    chart
+  }
+
+
+  def view {
+    this.pack()
+    RefineryUtilities.centerFrameOnScreen(this)
+    this.setVisible(true)
+  }
+
+  def draw(chart: JFreeChart, width: Int, height: Int): BufferedImage = {
+    val img =
+      new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB)
+    val g2 = img.createGraphics()
+
+    chart.draw(g2, new Rectangle2D.Double(0, 0, width, height))
+
+    g2.dispose()
+    img
+  }
+
+  def saveToFile(
+                  aFileName: String,
+                  width: Int,
+                  height: Int,
+                  quality: Double) {
+    val img = draw(chart, width, height)
+
+    val fos = new FileOutputStream(aFileName)
+    val encoder2 =
+      JPEGCodec.createJPEGEncoder(fos)
+    val param2 = encoder2.getDefaultJPEGEncodeParam(img)
+    param2.setQuality(quality.toFloat, true)
+    encoder2.encode(img, param2)
+    fos.close()
+  }
+
+}

src/main/scala/com/microWorkflow/jsonScalaPerftest/output/ChartReporter.scala

@@ -0,0 +1,16 @@
+package com.microWorkflow.jsonScalaPerftest.output
+
+import scala.collection.immutable.HashMap
+import com.microWorkflow.jsonScalaPerftest.Measurement
+
+class ConsoleReporter(ms: Array[(String, HashMap[String, Measurement])]) {
+
+  def printResults() {
+    for ((category, measurements) <- ms) {
+      println("Category: %s".format(category))
+      for ((dataset, measurement) <- measurements)
+      print("\t dataset '%s', measurement: %s\n".format(dataset, measurement))
+    }
+
+  }
+}