DM: add --pipe

dm/DMPipeTask.cpp

Index: dm/DMPipeTask.cpp
diff --git a/dm/DMPipeTask.cpp b/dm/DMPipeTask.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..8a8fe62000ad53870d970e8693e127598bac888a
--- /dev/null
+++ b/dm/DMPipeTask.cpp
@@ -0,0 +1,84 @@
+#include "DMPipeTask.h"
+#include "DMUtil.h"
+#include "DMWriteTask.h"
+
+#include "SamplePipeControllers.h"
+#include "SkCommandLineFlags.h"
+#include "SkGPipe.h"
+
+DEFINE_bool(pipe, false, "If true, check several pipe variants against the reference bitmap.");
+
+namespace DM {
+
+static uint32_t get_flags(bool crossProcess, bool sharedAddressSpace) {
+    SkASSERT(!(!crossProcess && sharedAddressSpace));
+    uint32_t flags = 0;
+    if (crossProcess) {
+        flags |= SkGPipeWriter::kCrossProcess_Flag;
+        if (sharedAddressSpace) {
+            flags |= SkGPipeWriter::kSharedAddressSpace_Flag;
+        }
+    }
+    return flags;
+}
+
+static const char* get_name(const uint32_t flags) {
+    if (flags & SkGPipeWriter::kCrossProcess_Flag &&
+        flags & SkGPipeWriter::kSharedAddressSpace_Flag) {
+        return "cross_process_shared_address_space_pipe";
+    } else if (flags & SkGPipeWriter::kCrossProcess_Flag) {
+        return "cross_process_pipe";
+    } else {
+        return "pipe";
+    }
+}
+
+PipeTask::PipeTask(const Task& parent,
+                   skiagm::GM* gm,
+                   SkBitmap reference,
+                   bool crossProcess,
+                   bool sharedAddressSpace)
+    : Task(parent)
+    , fFlags(get_flags(crossProcess, sharedAddressSpace))
+    , fName(UnderJoin(parent.name().c_str(), get_name(fFlags)))
+    , fGM(gm)
+    , fReference(reference)
+    {}
+
+void PipeTask::draw() {
+    SkBitmap bitmap;
+    SetupBitmap(fReference.config(), fGM.get(), &bitmap);
+
+    SkCanvas canvas(bitmap);
+    PipeController pipeController(&canvas, &SkImageDecoder::DecodeMemory);
+    SkGPipeWriter writer;
+
+    SkCanvas* pipeCanvas = writer.startRecording(&pipeController,
+                                                 fFlags,
+                                                 bitmap.width(),
+                                                 bitmap.height());
+    pipeCanvas->concat(fGM->getInitialTransform());
+    fGM->draw(pipeCanvas);
+    writer.endRecording();
+
+    if (!BitmapsEqual(bitmap, fReference)) {
+        this->fail();
+        this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap)));
+    }
+}
+
+bool PipeTask::shouldSkip() const {
+    if (!FLAGS_pipe) {
+        return true;
+    }
+    if (fGM->getFlags() & skiagm::GM::kSkipPipe_Flag) {
+        return true;
+    }
+    if (fFlags == SkGPipeWriter::kCrossProcess_Flag &&
+        fGM->getFlags() & skiagm::GM::kSkipPipeCrossProcess_Flag) {
+        return true;
+    }
+    return false;
+}
+
+}  // namespace DM