SkRasterPipeline preliminaries

tests/SkRasterPipelineTest.cpp

+/*
+ * 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 "Test.h"
+#include "SkRasterPipeline.h"
+
+DEF_TEST(SkRasterPipeline, r) {
+    // We'll build up and run a simple pipeline that exercises the salient
+    // mechanics of SkRasterPipeline:
+    //    - context pointers
+    //    - stages sensitive to the number of pixels
+    //    - stages insensitive to the number of pixels
+    //
+    // This pipeline loads up some values, squares them, then writes them back to memory.
+
+    SkRasterPipeline p;
+
+    // Load needs two variants, one to load 4 values...
+    auto load_4 = [](SkRasterPipeline::Stage* st, size_t x, Sk4f v0, Sk4f v1, Sk4f v2, Sk4f v3,
+                                                            Sk4f v4, Sk4f v5, Sk4f v6, Sk4f v7) {
+        auto ptr = st->ctx<const float*>();
+        v0 = Sk4f{ptr[x+0]};
+        v1 = Sk4f{ptr[x+1]};
+        v2 = Sk4f{ptr[x+2]};
+        v3 = Sk4f{ptr[x+3]};
+
+        st->next(x, v0,v1,v2,v3, v4,v5,v6,v7);
+    };
+    // ...and one to load a single value.
+    auto load_1 = [](SkRasterPipeline::Stage* st, size_t x, Sk4f v0, Sk4f v1, Sk4f v2, Sk4f v3,
+                                                            Sk4f v4, Sk4f v5, Sk4f v6, Sk4f v7) {
+        auto ptr = st->ctx<const float*>();
+        v0 = Sk4f{ptr[x]};
+
+        st->next(x, v0,v1,v2,v3, v4,v5,v6,v7);
+    };
+
+    // square doesn't really care how many of its inputs are active, nor does it need a context.
+    auto square = [](SkRasterPipeline::Stage* st, size_t x, Sk4f v0, Sk4f v1, Sk4f v2, Sk4f v3,
+                                                            Sk4f v4, Sk4f v5, Sk4f v6, Sk4f v7) {
+        v0 *= v0;
+        v1 *= v1;
+        v2 *= v2;
+        v3 *= v3;
+        st->next(x, v0,v1,v2,v3, v4,v5,v6,v7);
+    };
+
+    // Like load, store has _4 and _1 variants.  It ends the pipeline by returning.
+    auto store_4 = [](SkRasterPipeline::Stage* st, size_t x, Sk4f v0, Sk4f v1, Sk4f v2, Sk4f v3,
+                                                             Sk4f v4, Sk4f v5, Sk4f v6, Sk4f v7) {
+        auto ptr = st->ctx<float*>();
+        ptr[x+0] = v0[0];
+        ptr[x+1] = v1[0];
+        ptr[x+2] = v2[0];
+        ptr[x+3] = v3[0];
+    };
+    auto store_1 = [](SkRasterPipeline::Stage* st, size_t x, Sk4f v0, Sk4f v1, Sk4f v2, Sk4f v3,
+                                                             Sk4f v4, Sk4f v5, Sk4f v6, Sk4f v7) {
+        auto ptr = st->ctx<float*>();
+        ptr[x+0] = v0[0];
+    };
+
+    const float src_vals[] = { 1,2,3,4,5 };
+    float       dst_vals[] = { 0,0,0,0,0 };
+    p.append(load_4, load_1, src_vals);
+    p.append(square);
+    p.append(store_4, store_1, dst_vals);
+
+    // Run the pipeline.
+    p.run(5);
+
+    REPORTER_ASSERT(r, dst_vals[0] ==  1);
+    REPORTER_ASSERT(r, dst_vals[1] ==  4);
+    REPORTER_ASSERT(r, dst_vals[2] ==  9);
+    REPORTER_ASSERT(r, dst_vals[3] == 16);
+    REPORTER_ASSERT(r, dst_vals[4] == 25);
+}