1 files changed, 99 insertions, 0 deletions
diff --git a/tools/skiaserve/urlhandlers/BreakHandler.cpp b/tools/skiaserve/urlhandlers/BreakHandler.cpp
new file mode 100644
index 0000000000..9e01c0f509
--- /dev/null
+++ b/tools/skiaserve/urlhandlers/BreakHandler.cpp
@@ -0,0 +1,99 @@
+/*
+ * Copyright 2016 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#include "UrlHandler.h"
+
+#include "microhttpd.h"
+#include "../Request.h"
+#include "../Response.h"
+
+using namespace Response;
+
+bool BreakHandler::canHandle(const char* method, const char* url) {
+    static const char* kBasePath = "/break";
+    return 0 == strcmp(method, MHD_HTTP_METHOD_GET) &&
+           0 == strncmp(url, kBasePath, strlen(kBasePath));
+}
+
+SkColor BreakHandler::GetPixel(Request* request, int x, int y) {
+    SkCanvas* canvas = request->getCanvas();
+    canvas->flush();
+    SkAutoTDelete<SkBitmap> bitmap(request->getBitmapFromCanvas(canvas));
+    SkASSERT(bitmap);
+    bitmap->lockPixels();
+    uint8_t* start = ((uint8_t*) bitmap->getPixels()) + (y * Request::kImageWidth + x) * 4;
+    SkColor result = SkColorSetARGB(start[3], start[0], start[1], start[2]);
+    bitmap->unlockPixels();
+    return result;
+}
+
+
+int BreakHandler::handle(Request* request, MHD_Connection* connection,
+                         const char* url, const char* method,
+                         const char* upload_data, size_t* upload_data_size) {
+    SkTArray<SkString> commands;
+    SkStrSplit(url, "/", &commands);
+
+    if (!request->fPicture.get() || commands.count() != 4) {
+        return MHD_NO;
+    }
+
+    // /break/<n>/<x>/<y>
+    int n;
+    sscanf(commands[1].c_str(), "%d", &n);
+    int x;
+    sscanf(commands[2].c_str(), "%d", &x);
+    int y;
+    sscanf(commands[3].c_str(), "%d", &y);
+
+    int count = request->fDebugCanvas->getSize();
+    SkASSERT(n < count);
+
+    SkCanvas* canvas = request->getCanvas();
+    canvas->clear(SK_ColorWHITE);
+    int saveCount = canvas->save();
+    for (int i = 0; i <= n; ++i) {
+        request->fDebugCanvas->getDrawCommandAt(i)->execute(canvas);
+    }
+    SkColor target = GetPixel(request, x, y);
+    Json::Value response(Json::objectValue);
+    Json::Value startColor(Json::arrayValue);
+    startColor.append(Json::Value(SkColorGetR(target)));
+    startColor.append(Json::Value(SkColorGetG(target)));
+    startColor.append(Json::Value(SkColorGetB(target)));
+    startColor.append(Json::Value(SkColorGetA(target)));
+    response["startColor"] = startColor;
+    response["endColor"] = startColor;
+    response["endOp"] = Json::Value(n);
+    for (int i = n + 1; i < n + count; ++i) {
+        int index = i % count;
+        if (index == 0) {
+            // reset canvas for wraparound
+            canvas->restoreToCount(saveCount);
+            canvas->clear(SK_ColorWHITE);
+            saveCount = canvas->save();
+        }
+        request->fDebugCanvas->getDrawCommandAt(index)->execute(canvas);
+        SkColor current = GetPixel(request, x, y);
+        if (current != target) {
+            Json::Value endColor(Json::arrayValue);
+            endColor.append(Json::Value(SkColorGetR(current)));
+            endColor.append(Json::Value(SkColorGetG(current)));
+            endColor.append(Json::Value(SkColorGetB(current)));
+            endColor.append(Json::Value(SkColorGetA(current)));
+            response["endColor"] = endColor;
+            response["endOp"] = Json::Value(index);
+            break;
+        }
+    }
+    canvas->restoreToCount(saveCount);
+    SkDynamicMemoryWStream stream;
+    stream.writeText(Json::FastWriter().write(response).c_str());
+    SkAutoTUnref<SkData> data(stream.copyToData());
+    return SendData(connection, data, "application/json");
+}
+