Make SKP bench JSON ouput better

tools/PictureResultsWriter.h

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  *
  * Classes for writing out bench results in various formats.
  */
 
 #ifndef SkPictureResultsWriter_DEFINED
 #define SkPictureResultsWriter_DEFINED
 
+
+#include "PictureRenderer.h"
 #include "BenchLogger.h"
 #include "ResultsWriter.h"
 #include "SkJSONCPP.h"
 #include "SkStream.h"
 #include "SkString.h"
 #include "SkTArray.h"
 #include "TimerData.h"
 
 /**
  * Base class for writing picture bench results.
  */
 class PictureResultsWriter : SkNoncopyable {
 public:
     enum TileFlags {kPurging, kAvg};
 
     PictureResultsWriter() {}
     virtual ~PictureResultsWriter() {}
 
     virtual void bench(const char name[], int32_t x, int32_t y) = 0;
-    virtual void tileConfig(SkString configName) = 0;
+    virtual void logRenderer(sk_tools::PictureRenderer *pr) = 0;
     virtual void tileMeta(int x, int y, int tx, int ty) = 0;
     virtual void addTileFlag(PictureResultsWriter::TileFlags flag) = 0;
     virtual void tileData(
             TimerData* data,
             const char format[],
             const TimerData::Result result,
             uint32_t timerTypes,
             int numInnerLoops = 1) = 0;
    virtual void end() = 0;
 };
 
 /**
  * This class allows bench data to be piped into multiple
  * PictureResultWriter classes. It does not own any classes
  * passed to it, so the owner is required to manage any classes
  * passed to PictureResultsMultiWriter */
 class PictureResultsMultiWriter : public PictureResultsWriter {
 public:
     PictureResultsMultiWriter()
         : fWriters() {}
     void add(PictureResultsWriter* newWriter) {
         fWriters.push_back(newWriter);
     }
     virtual ~PictureResultsMultiWriter() {}
     virtual void bench(const char name[], int32_t x, int32_t y) {
         for(int i=0; i<fWriters.count(); ++i) {
             fWriters[i]->bench(name, x, y);
         }
     }
-    virtual void tileConfig(SkString configName) {
+    virtual void logRenderer(sk_tools::PictureRenderer *pr) {
         for(int i=0; i<fWriters.count(); ++i) {
-            fWriters[i]->tileConfig(configName);
+            fWriters[i]->logRenderer(pr);
         }
     }
     virtual void tileMeta(int x, int y, int tx, int ty) {
         for(int i=0; i<fWriters.count(); ++i) {
             fWriters[i]->tileMeta(x, y, tx, ty);
         }
     }
     virtual void addTileFlag(PictureResultsWriter::TileFlags flag) {
         for(int i=0; i<fWriters.count(); ++i) {
             fWriters[i]->addTileFlag(flag);
@@ skipping 29 matching lines @@
             fLogger->logProgress(str);
         }
     }
 public:
     PictureResultsLoggerWriter(BenchLogger* log)
           : fLogger(log), currentLine() {}
     virtual void bench(const char name[], int32_t x, int32_t y) {
         SkString result;
         result.printf("running bench [%i %i] %s ", x, y, name);
         this->logProgress(result.c_str());
     }

[robertphillips, 2014/06/24 20:49:52]

OVERRIDE here ?

[kelvinly, 2014/06/24 22:25:19]

Ah, sorry, didn't know that was a thing. Fixed

-    virtual void tileConfig(SkString configName) {
-        currentLine = configName;
+    virtual void logRenderer(sk_tools::PictureRenderer* renderer) {

[robertphillips, 2014/06/24 20:49:51]

fCurrentLine ?

[kelvinly, 2014/06/24 22:25:19]

Done.

+        currentLine = renderer->getConfigName();
     }
     virtual void tileMeta(int x, int y, int tx, int ty) {
         currentLine.appendf(": tile [%i,%i] out of [%i,%i]", x, y, tx, ty);
     }
     virtual void addTileFlag(PictureResultsWriter::TileFlags flag) {
         if(flag == PictureResultsWriter::kPurging) {
             currentLine.append(" <withPurging>");
         } else if(flag == PictureResultsWriter::kAvg) {
             currentLine.append(" <averaged>");
         }
     }
     virtual void tileData(
             TimerData* data,
             const char format[],
             const TimerData::Result result,
             uint32_t timerTypes,
             int numInnerLoops = 1) {
         SkString results = data->getResult(format, result,
                 currentLine.c_str(), timerTypes, numInnerLoops);
         results.append("\n");
         this->logProgress(results.c_str());
     }
     virtual void end() {}
 private:
     BenchLogger* fLogger;

[robertphillips, 2014/06/24 20:49:52]

fCurrentLine ?

[kelvinly, 2014/06/24 22:25:19]

Done.

     SkString currentLine;
 };
 
 /**
  * This PictureResultsWriter collects data in a JSON node
  *
  * The format is something like
  * {
  *      benches: [
  *          {
@@ skipping 13 matching lines @@
  *                          }
  *                      ]
  *                  }
  *              ]
  *          }
  *      ]
  * }*/
 
 class PictureJSONResultsWriter : public PictureResultsWriter {
 public:
-    PictureJSONResultsWriter(const char filename[])
-        : fFilename(filename),
-          fRoot(),
-          fCurrentBench(NULL),
-          fCurrentTileSet(NULL),
-          fCurrentTile(NULL) {}
+    PictureJSONResultsWriter(const char filename[], 
+                             const char builderName[],
+                             int buildNumber,
+                             int timestamp,
+                             const char gitHash[],
+                             int gitNumber)
+        : fStream(filename)
+        , fParams()
+        , fConfigString() {
+        fBuilderName = SkString(builderName);
+        fBuildNumber = buildNumber;
+        fTimestamp = timestamp;
+        fGitHash = SkString(gitHash);
+        fGitNumber = gitNumber;
+        fBuilderData = makeBuilderJSON(fBuilderName);
+    }
 
     virtual void bench(const char name[], int32_t x, int32_t y) {
-        SkString sk_name(name);
-        sk_name.append("_");
-        sk_name.appendS32(x);
-        sk_name.append("_");
-        sk_name.appendS32(y);
-        Json::Value* bench_node = SkFindNamedNode(&fRoot["benches"], sk_name.c_str());
-        fCurrentBench = &(*bench_node)["tileSets"];
+        fBenchName = SkString(name);
     }

[robertphillips, 2014/06/24 20:49:51]

OVERRIDE here ?

[kelvinly, 2014/06/24 22:25:19]

Done.

-    virtual void tileConfig(SkString configName) {
-        SkASSERT(fCurrentBench != NULL);
-        fCurrentTileSet = SkFindNamedNode(fCurrentBench, configName.c_str());
-        fCurrentTile = &(*fCurrentTileSet)["tiles"][0];
+    virtual void logRenderer(sk_tools::PictureRenderer* pr) {
+        fParams = pr->getJSONConfig();
+        fConfigString = pr->getConfigName();
     }
-    virtual void tileMeta(int x, int y, int tx, int ty) {
-        SkASSERT(fCurrentTileSet != NULL);
-        (*fCurrentTileSet)["tx"] = tx;
-        (*fCurrentTileSet)["ty"] = ty;
-        fCurrentTile = &(*fCurrentTileSet)["tiles"][x+tx*y];
-    }
-    virtual void addTileFlag(PictureResultsWriter::TileFlags flag) {
-        SkASSERT(fCurrentTile != NULL);
-        if(flag == PictureResultsWriter::kPurging) {
-            (*fCurrentTile)["flags"]["purging"] = true;
-        } else if(flag == PictureResultsWriter::kAvg) {
-            (*fCurrentTile)["flags"]["averaged"] = true;
-        }
-    }
+    // Apparently tiles aren't used, so tileMeta is empty

[robertphillips, 2014/06/24 20:49:51]

OVERRIDE here ? 
Don't need ';' here

[kelvinly, 2014/06/24 22:25:19]

Done.

+    virtual void tileMeta(int x, int y, int tx, int ty) { ; }
+    // Flags aren't used, so addTileFlag is empty

[robertphillips, 2014/06/24 20:49:52]

Same here

[kelvinly, 2014/06/24 22:25:19]

Done.

+    virtual void addTileFlag(PictureResultsWriter::TileFlags flag) { ; }
     virtual void tileData(
             TimerData* data,
             const char format[],
             const TimerData::Result result,
             uint32_t timerTypes,
             int numInnerLoops = 1) {
-        SkASSERT(fCurrentTile != NULL);
-        (*fCurrentTile)["data"] = data->getJSON(timerTypes, result, numInnerLoops);
+        Json::Value newData = data->getJSON(timerTypes, result, numInnerLoops);
+        Json::Value combinedParams(fBuilderData);
+        for(Json::ValueIterator iter = fParams.begin(); iter != fParams.end();
+                iter++) {
+            combinedParams[iter.key().asString()]= *iter;
+        }
+        // For each set of timer data
+        for(Json::ValueIterator iter = newData.begin(); iter != newData.end();
+                iter++) {
+            Json::Value data;
+            data["buildNumber"] = fBuildNumber;
+            data["timestamp"] = fTimestamp;
+            data["gitHash"] = fGitHash.c_str();
+            data["gitNumber"] = fGitNumber;
+            data["isTrybot"] = fBuilderName.endsWith("Trybot");
+
+            data["params"] = combinedParams;
+            data["params"]["benchName"] = fBenchName.c_str();
+
+            // Not including skpSize because that's deprecated?
+            data["key"] = makeKey(iter.key().asString().c_str()).c_str();
+            // Get the data
+            SkTArray<double> times;
+            Json::Value val = *iter;
+            for(Json::ValueIterator vals = val.begin(); vals != val.end();
+                    vals++) {
+                times.push_back((*vals).asDouble());
+            }
+            // Because there's no sort I'll just remove the smallest 
+            // 25% of values
+            int numValues = times.count();
+            double currentMin = 10000000.0f;
+            for(int i = 0; i < numValues; i++) {
+                if(times[i] < currentMin) {
+                    currentMin = times[i];
+                }
+            }
+            for(int i = 0; i < numValues*0.25f; i++) {
+                double localMin = currentMin;
+                for(int j = 0; j < numValues; j++) {
+                    if(times[i] > currentMin && times[i] < localMin) {
+                        localMin = times[i];
+                    }
+                }
+                currentMin = localMin;
+            }
+            data["value"] = currentMin;
+            data["params"]["measurementType"] = iter.key().asString();
+            fStream.writeText(Json::FastWriter().write(data).c_str());
+        }
     }
     virtual void end() {
-       SkFILEWStream stream(fFilename.c_str());
-       stream.writeText(Json::FastWriter().write(fRoot).c_str());
-       stream.flush();
+       fStream.flush();
     }
 private:
-    SkString fFilename;
-    Json::Value fRoot;
-    Json::Value *fCurrentBench;
-    Json::Value *fCurrentTileSet;
-    Json::Value *fCurrentTile;
+    SkString makeKey(const char measurementType[]) {
+        SkString tmp(fBuilderName);
+        tmp.append("_");
+        tmp.append(fBenchName);
+        tmp.append("_");
+        tmp.append(fConfigString);
+        tmp.append("_");
+        tmp.append(measurementType);
+        return tmp;
+    }
+
+    SkFILEWStream fStream;
+    Json::Value fBuilderData;
+    SkString fBenchName;
+    Json::Value fParams;
+
+    SkString fConfigString;
+    SkString fBuilderName;
+    int fBuildNumber;
+    int fTimestamp;
+    SkString fGitHash;
+    int fGitNumber;
 };
 
 #endif