private def writeHeatmapDef()

in atlas-chart/src/main/scala/com/netflix/atlas/chart/JsonCodec.scala [192:260]
61 lines of code
5 McCabe index (conditional complexity)

  private def writeHeatmapDef(gen: JsonGenerator, graph: GraphDef, plot: PlotDef, id: Int): Unit = {
    plot.heatmapData(graph).foreach { heatmap =>
      gen.writeStartObject()
      gen.writeStringField("type", "heatmap")
      gen.writeNumberField("plot", id)
      gen.writeStringField("colorScale", heatmap.settings.colorScale.name())
      gen.writeStringField("upper", heatmap.settings.upper.toString)
      gen.writeStringField("lower", heatmap.settings.lower.toString)
      heatmap.settings.label.foreach { label =>
        gen.writeStringField("label", label)
      }

      // Y-tick information, used to define the vertical buckets for heatmap counts. Included
      // so the result can be reproduced in a dynamic rendering.
      gen.writeArrayFieldStart("yTicks")
      var min = heatmap.yaxis.min
      var i = 0
      while (i < heatmap.yTicks.size) {
        val max = heatmap.yTicks(i).v
        gen.writeStartObject()
        gen.writeNumberField("min", min)
        gen.writeNumberField("max", max)
        gen.writeStringField("label", heatmap.yTicks(i).label)
        gen.writeEndObject()
        min = max
        i += 1
      }
      gen.writeEndArray()

      // Color ticks used to map counts to a color
      gen.writeArrayFieldStart("colorTicks")
      val colorTicks = heatmap.colorTicks
      min = heatmap.minCount
      i = 1
      while (i < colorTicks.size) {
        val max = colorTicks(i).v
        gen.writeStartObject()
        gen.writeFieldName("color")
        writeColor(gen, heatmap.palette.colors(i - 1))
        gen.writeNumberField("min", min)
        gen.writeNumberField("max", max)
        gen.writeStringField("label", colorTicks(i).label)
        gen.writeEndObject()
        min = max
        i += 1
      }
      gen.writeEndArray()

      // Output the counts associated with each cell
      gen.writeObjectFieldStart("data")
      gen.writeStringField("type", "heatmap")
      gen.writeArrayFieldStart("values")
      var t = heatmap.xaxis.start
      while (t < heatmap.xaxis.end) {
        gen.writeStartArray()
        var y = 0
        while (y < heatmap.numberOfValueBuckets) {
          gen.writeNumber(heatmap.count(t, y))
          y += 1
        }
        gen.writeEndArray()
        t += graph.step
      }
      gen.writeEndArray()
      gen.writeEndObject()

      gen.writeEndObject()
    }
  }