tests/GrMeshTest.cpp - platform/external/skia - Gitiles

 /*
  * Copyright 2017 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */

 #include "SkTypes.h"
 #include "Test.h"

 #include <array>
 #include <vector>
 #include "GrCaps.h"
 #include "GrContext.h"
 #include "GrContextPriv.h"
 #include "GrGeometryProcessor.h"
 #include "GrGpuCommandBuffer.h"
 #include "GrMemoryPool.h"
 #include "GrOpFlushState.h"
 #include "GrRenderTargetContext.h"
 #include "GrRenderTargetContextPriv.h"
 #include "GrResourceKey.h"
 #include "GrResourceProvider.h"
 #include "SkBitmap.h"
 #include "SkMakeUnique.h"
 #include "glsl/GrGLSLFragmentShaderBuilder.h"
 #include "glsl/GrGLSLGeometryProcessor.h"
 #include "glsl/GrGLSLVarying.h"
 #include "glsl/GrGLSLVertexGeoBuilder.h"

 GR_DECLARE_STATIC_UNIQUE_KEY(gIndexBufferKey);

 static constexpr int kBoxSize = 2;
 static constexpr int kBoxCountY = 8;
 static constexpr int kBoxCountX = 8;
 static constexpr int kBoxCount = kBoxCountY * kBoxCountX;

 static constexpr int kImageWidth = kBoxCountY * kBoxSize;
 static constexpr int kImageHeight = kBoxCountX * kBoxSize;

 static constexpr int kIndexPatternRepeatCount = 3;
 constexpr uint16_t kIndexPattern[6] = {0, 1, 2, 1, 2, 3};


 class DrawMeshHelper {
 public:
     DrawMeshHelper(GrOpFlushState* state) : fState(state) {}

     sk_sp<const GrBuffer> getIndexBuffer();

     template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const SkTArray<T>& data) {
         return this->makeVertexBuffer(data.begin(), data.count());
     }
     template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const std::vector<T>& data) {
         return this->makeVertexBuffer(data.data(), data.size());
     }
     template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const T* data, int count);

     void drawMesh(const GrMesh& mesh);

 private:
     GrOpFlushState* fState;
 };

 struct Box {
     float fX, fY;
     GrColor fColor;
 };

 ////////////////////////////////////////////////////////////////////////////////////////////////////

 /**
  * This is a GPU-backend specific test. It tries to test all possible usecases of GrMesh. The test
  * works by drawing checkerboards of colored boxes, reading back the pixels, and comparing with
  * expected results. The boxes are drawn on integer boundaries and the (opaque) colors are chosen
  * from the set (r,g,b) = (0,255)^3, so the GPU renderings ought to produce exact matches.
  */

 static void run_test(GrContext* context, const char* testName, skiatest::Reporter*,
                      const sk_sp<GrRenderTargetContext>&, const SkBitmap& gold,
                      std::function<void(DrawMeshHelper*)> testFn);

 DEF_GPUTEST_FOR_RENDERING_CONTEXTS(GrMeshTest, reporter, ctxInfo) {
     GrContext* context = ctxInfo.grContext();

     const GrBackendFormat format =
             context->contextPriv().caps()->getBackendFormatFromColorType(kRGBA_8888_SkColorType);

     sk_sp<GrRenderTargetContext> rtc(context->contextPriv().makeDeferredRenderTargetContext(
             format, SkBackingFit::kExact, kImageWidth, kImageHeight, kRGBA_8888_GrPixelConfig,
             nullptr));
     if (!rtc) {
         ERRORF(reporter, "could not create render target context.");
         return;
     }

     SkTArray<Box> boxes;
     SkTArray<std::array<Box, 4>> vertexData;
     SkBitmap gold;

     // ---- setup ----------

     SkPaint paint;
     paint.setBlendMode(SkBlendMode::kSrc);
     gold.allocN32Pixels(kImageWidth, kImageHeight);

     SkCanvas goldCanvas(gold);

     for (int y = 0; y < kBoxCountY; ++y) {
         for (int x = 0; x < kBoxCountX; ++x) {
             int c = y + x;
             int rgb[3] = {-(c & 1) & 0xff, -((c >> 1) & 1) & 0xff, -((c >> 2) & 1) & 0xff};

             const Box box = boxes.push_back() = {
                     float(x * kBoxSize),
                     float(y * kBoxSize),
                     GrColorPackRGBA(rgb[0], rgb[1], rgb[2], 255)
             };

             std::array<Box, 4>& boxVertices = vertexData.push_back();
             for (int i = 0; i < 4; ++i) {
                 boxVertices[i] = {
                         box.fX + (i / 2) * kBoxSize,
                         box.fY + (i % 2) * kBoxSize,
                         box.fColor
                 };
             }

             paint.setARGB(255, rgb[0], rgb[1], rgb[2]);
             goldCanvas.drawRect(SkRect::MakeXYWH(box.fX, box.fY, kBoxSize, kBoxSize), paint);
         }
     }

     goldCanvas.flush();

     // ---- tests ----------

 #define VALIDATE(buff)                           \
     do {                                         \
         if (!buff) {                             \
             ERRORF(reporter, #buff " is null."); \
             return;                              \
         }                                        \
     } while (0)

     run_test(context, "setNonIndexedNonInstanced", reporter, rtc, gold,
              [&](DrawMeshHelper* helper) {
                  SkTArray<Box> expandedVertexData;
                  for (int i = 0; i < kBoxCount; ++i) {
                      for (int j = 0; j < 6; ++j) {
                          expandedVertexData.push_back(vertexData[i][kIndexPattern[j]]);
                      }
                  }

                  // Draw boxes one line at a time to exercise base vertex.
                  auto vbuff = helper->makeVertexBuffer(expandedVertexData);
                  VALIDATE(vbuff);
                  for (int y = 0; y < kBoxCountY; ++y) {
                      GrMesh mesh(GrPrimitiveType::kTriangles);
                      mesh.setNonIndexedNonInstanced(kBoxCountX * 6);
                      mesh.setVertexData(vbuff, y * kBoxCountX * 6);
                      helper->drawMesh(mesh);
                  }
              });

     run_test(context, "setIndexed", reporter, rtc, gold, [&](DrawMeshHelper* helper) {
         auto ibuff = helper->getIndexBuffer();
         VALIDATE(ibuff);
         auto vbuff = helper->makeVertexBuffer(vertexData);
         VALIDATE(vbuff);
         int baseRepetition = 0;
         int i = 0;

         // Start at various repetitions within the patterned index buffer to exercise base index.
         while (i < kBoxCount) {
             GR_STATIC_ASSERT(kIndexPatternRepeatCount >= 3);
             int repetitionCount = SkTMin(3 - baseRepetition, kBoxCount - i);

             GrMesh mesh(GrPrimitiveType::kTriangles);
             mesh.setIndexed(ibuff, repetitionCount * 6, baseRepetition * 6, baseRepetition * 4,
                             (baseRepetition + repetitionCount) * 4 - 1, GrPrimitiveRestart::kNo);
             mesh.setVertexData(vbuff, (i - baseRepetition) * 4);
             helper->drawMesh(mesh);

             baseRepetition = (baseRepetition + 1) % 3;
             i += repetitionCount;
         }
     });

     run_test(context, "setIndexedPatterned", reporter, rtc, gold, [&](DrawMeshHelper* helper) {
         auto ibuff = helper->getIndexBuffer();
         VALIDATE(ibuff);
         auto vbuff = helper->makeVertexBuffer(vertexData);
         VALIDATE(vbuff);

         // Draw boxes one line at a time to exercise base vertex. setIndexedPatterned does not
         // support a base index.
         for (int y = 0; y < kBoxCountY; ++y) {
             GrMesh mesh(GrPrimitiveType::kTriangles);
             mesh.setIndexedPatterned(ibuff, 6, 4, kBoxCountX, kIndexPatternRepeatCount);
             mesh.setVertexData(vbuff, y * kBoxCountX * 4);
             helper->drawMesh(mesh);
         }
     });

     for (bool indexed : {false, true}) {
         if (!context->contextPriv().caps()->instanceAttribSupport()) {
             break;
         }

         run_test(context, indexed ? "setIndexedInstanced" : "setInstanced",
                  reporter, rtc, gold, [&](DrawMeshHelper* helper) {
             auto idxbuff = indexed ? helper->getIndexBuffer() : nullptr;
             auto instbuff = helper->makeVertexBuffer(boxes);
             VALIDATE(instbuff);
             auto vbuff = helper->makeVertexBuffer(std::vector<float>{0,0, 0,1, 1,0, 1,1});
             VALIDATE(vbuff);
             auto vbuff2 = helper->makeVertexBuffer( // for testing base vertex.
                               std::vector<float>{-1,-1, -1,-1, 0,0, 0,1, 1,0, 1,1});
             VALIDATE(vbuff2);

             // Draw boxes one line at a time to exercise base instance, base vertex, and null vertex
             // buffer. setIndexedInstanced intentionally does not support a base index.
             for (int y = 0; y < kBoxCountY; ++y) {
                 GrMesh mesh(indexed ? GrPrimitiveType::kTriangles
                                     : GrPrimitiveType::kTriangleStrip);
                 if (indexed) {
                     VALIDATE(idxbuff);
                     mesh.setIndexedInstanced(idxbuff, 6, instbuff, kBoxCountX, y * kBoxCountX,
                                              GrPrimitiveRestart::kNo);
                 } else {
                     mesh.setInstanced(instbuff, kBoxCountX, y * kBoxCountX, 4);
                 }
                 switch (y % 3) {
                     case 0:
                         if (context->contextPriv().caps()->shaderCaps()->vertexIDSupport()) {
                             if (y % 2) {
                                 // We don't need this call because it's the initial state of GrMesh.
                                 mesh.setVertexData(nullptr);
                             }
                             break;
                         }
                         // Fallthru.
                     case 1:
                         mesh.setVertexData(vbuff);
                         break;
                     case 2:
                         mesh.setVertexData(vbuff2, 2);
                         break;
                 }
                 helper->drawMesh(mesh);
             }
         });
     }
 }

 ////////////////////////////////////////////////////////////////////////////////////////////////////

 class GrMeshTestOp : public GrDrawOp {
 public:
     DEFINE_OP_CLASS_ID

     static std::unique_ptr<GrDrawOp> Make(GrContext* context,
                                           std::function<void(DrawMeshHelper*)> testFn) {
         GrOpMemoryPool* pool = context->contextPriv().opMemoryPool();

         return pool->allocate<GrMeshTestOp>(testFn);
     }

 private:
     friend class GrOpMemoryPool; // for ctor

     GrMeshTestOp(std::function<void(DrawMeshHelper*)> testFn)
         : INHERITED(ClassID())
         , fTestFn(testFn) {
         this->setBounds(SkRect::MakeIWH(kImageWidth, kImageHeight),
                         HasAABloat::kNo, IsZeroArea::kNo);
     }

     const char* name() const override { return "GrMeshTestOp"; }
     FixedFunctionFlags fixedFunctionFlags() const override { return FixedFunctionFlags::kNone; }
     GrProcessorSet::Analysis finalize(const GrCaps&, const GrAppliedClip*) override {
         return GrProcessorSet::EmptySetAnalysis();
     }
     void onPrepare(GrOpFlushState*) override {}
     void onExecute(GrOpFlushState* state, const SkRect& chainBounds) override {
         DrawMeshHelper helper(state);
         fTestFn(&helper);
     }

     std::function<void(DrawMeshHelper*)> fTestFn;

     typedef GrDrawOp INHERITED;
 };

 class GrMeshTestProcessor : public GrGeometryProcessor {
 public:
     GrMeshTestProcessor(bool instanced, bool hasVertexBuffer)
             : INHERITED(kGrMeshTestProcessor_ClassID) {
         if (instanced) {
             fInstanceLocation = {"location", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
             fInstanceColor = {"color", kUByte4_norm_GrVertexAttribType, kHalf4_GrSLType};
             this->setInstanceAttributes(&fInstanceLocation, 2);
             if (hasVertexBuffer) {
                 fVertexPosition = {"vertex", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
                 this->setVertexAttributes(&fVertexPosition, 1);
             }
         } else {
             fVertexPosition = {"vertex", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
             fVertexColor = {"color", kUByte4_norm_GrVertexAttribType, kHalf4_GrSLType};
             this->setVertexAttributes(&fVertexPosition, 2);
         }
     }

     const char* name() const override { return "GrMeshTest Processor"; }

     const Attribute& inColor() const {
         return fVertexColor.isInitialized() ? fVertexColor : fInstanceColor;
     }

     void getGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder* b) const final {
         b->add32(fInstanceLocation.isInitialized());
         b->add32(fVertexPosition.isInitialized());
     }

     GrGLSLPrimitiveProcessor* createGLSLInstance(const GrShaderCaps&) const final;

 private:
     Attribute fVertexPosition;
     Attribute fVertexColor;

     Attribute fInstanceLocation;
     Attribute fInstanceColor;

     friend class GLSLMeshTestProcessor;
     typedef GrGeometryProcessor INHERITED;
 };

 class GLSLMeshTestProcessor : public GrGLSLGeometryProcessor {
     void setData(const GrGLSLProgramDataManager& pdman, const GrPrimitiveProcessor&,
                  FPCoordTransformIter&& transformIter) final {}

     void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) final {
         const GrMeshTestProcessor& mp = args.fGP.cast<GrMeshTestProcessor>();

         GrGLSLVaryingHandler* varyingHandler = args.fVaryingHandler;
         varyingHandler->emitAttributes(mp);
         varyingHandler->addPassThroughAttribute(mp.inColor(), args.fOutputColor);

         GrGLSLVertexBuilder* v = args.fVertBuilder;
         if (!mp.fInstanceLocation.isInitialized()) {
             v->codeAppendf("float2 vertex = %s;", mp.fVertexPosition.name());
         } else {
             if (mp.fVertexPosition.isInitialized()) {
                 v->codeAppendf("float2 offset = %s;", mp.fVertexPosition.name());
             } else {
                 v->codeAppend ("float2 offset = float2(sk_VertexID / 2, sk_VertexID % 2);");
             }
             v->codeAppendf("float2 vertex = %s + offset * %i;", mp.fInstanceLocation.name(),
                            kBoxSize);
         }
         gpArgs->fPositionVar.set(kFloat2_GrSLType, "vertex");

         GrGLSLFPFragmentBuilder* f = args.fFragBuilder;
         f->codeAppendf("%s = half4(1);", args.fOutputCoverage);
     }
 };

 GrGLSLPrimitiveProcessor* GrMeshTestProcessor::createGLSLInstance(const GrShaderCaps&) const {
     return new GLSLMeshTestProcessor;
 }

 ////////////////////////////////////////////////////////////////////////////////////////////////////

 template<typename T>
 sk_sp<const GrBuffer> DrawMeshHelper::makeVertexBuffer(const T* data, int count) {
     return sk_sp<const GrBuffer>(
         fState->resourceProvider()->createBuffer(
             count * sizeof(T), kVertex_GrBufferType, kDynamic_GrAccessPattern,
             GrResourceProvider::Flags::kNoPendingIO |
             GrResourceProvider::Flags::kRequireGpuMemory, data));
 }

 sk_sp<const GrBuffer> DrawMeshHelper::getIndexBuffer() {
     GR_DEFINE_STATIC_UNIQUE_KEY(gIndexBufferKey);
     return fState->resourceProvider()->findOrCreatePatternedIndexBuffer(
             kIndexPattern, 6, kIndexPatternRepeatCount, 4, gIndexBufferKey);
 }

 void DrawMeshHelper::drawMesh(const GrMesh& mesh) {
     GrPipeline pipeline(GrScissorTest::kDisabled, SkBlendMode::kSrc);
     GrMeshTestProcessor mtp(mesh.isInstanced(), mesh.hasVertexData());
     fState->rtCommandBuffer()->draw(mtp, pipeline, nullptr, nullptr, &mesh, 1,
                                     SkRect::MakeIWH(kImageWidth, kImageHeight));
 }

 static void run_test(GrContext* context, const char* testName, skiatest::Reporter* reporter,
                      const sk_sp<GrRenderTargetContext>& rtc, const SkBitmap& gold,
                      std::function<void(DrawMeshHelper*)> testFn) {
     const int w = gold.width(), h = gold.height(), rowBytes = gold.rowBytes();
     const uint32_t* goldPx = reinterpret_cast<const uint32_t*>(gold.getPixels());
     if (h != rtc->height() || w != rtc->width()) {
         ERRORF(reporter, "[%s] expectation and rtc not compatible (?).", testName);
         return;
     }
     if (sizeof(uint32_t) * kImageWidth != gold.rowBytes()) {
         ERRORF(reporter, "unexpected row bytes in gold image.", testName);
         return;
     }

     SkAutoSTMalloc<kImageHeight * kImageWidth, uint32_t> resultPx(h * rowBytes);
     rtc->clear(nullptr, SkPMColor4f::FromBytes_RGBA(0xbaaaaaad),
                GrRenderTargetContext::CanClearFullscreen::kYes);
     rtc->priv().testingOnly_addDrawOp(GrMeshTestOp::Make(context, testFn));
     rtc->readPixels(gold.info(), resultPx, rowBytes, 0, 0, 0);
     for (int y = 0; y < h; ++y) {
         for (int x = 0; x < w; ++x) {
             uint32_t expected = goldPx[y * kImageWidth + x];
             uint32_t actual = resultPx[y * kImageWidth + x];
             if (expected != actual) {
                 ERRORF(reporter, "[%s] pixel (%i,%i): got 0x%x expected 0x%x",
                        testName, x, y, actual, expected);
                 return;
             }
         }
     }
 }
	/*
	* Copyright 2017 Google Inc.
	*
	* Use of this source code is governed by a BSD-style license that can be
	* found in the LICENSE file.
	*/

	#include "SkTypes.h"
	#include "Test.h"

	#include <array>
	#include <vector>
	#include "GrCaps.h"
	#include "GrContext.h"
	#include "GrContextPriv.h"
	#include "GrGeometryProcessor.h"
	#include "GrGpuCommandBuffer.h"
	#include "GrMemoryPool.h"
	#include "GrOpFlushState.h"
	#include "GrRenderTargetContext.h"
	#include "GrRenderTargetContextPriv.h"
	#include "GrResourceKey.h"
	#include "GrResourceProvider.h"
	#include "SkBitmap.h"
	#include "SkMakeUnique.h"
	#include "glsl/GrGLSLFragmentShaderBuilder.h"
	#include "glsl/GrGLSLGeometryProcessor.h"
	#include "glsl/GrGLSLVarying.h"
	#include "glsl/GrGLSLVertexGeoBuilder.h"

	GR_DECLARE_STATIC_UNIQUE_KEY(gIndexBufferKey);

	static constexpr int kBoxSize = 2;
	static constexpr int kBoxCountY = 8;
	static constexpr int kBoxCountX = 8;
	static constexpr int kBoxCount = kBoxCountY * kBoxCountX;

	static constexpr int kImageWidth = kBoxCountY * kBoxSize;
	static constexpr int kImageHeight = kBoxCountX * kBoxSize;

	static constexpr int kIndexPatternRepeatCount = 3;
	constexpr uint16_t kIndexPattern[6] = {0, 1, 2, 1, 2, 3};


	class DrawMeshHelper {
	public:
	DrawMeshHelper(GrOpFlushState* state) : fState(state) {}

	sk_sp<const GrBuffer> getIndexBuffer();

	template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const SkTArray<T>& data) {
	return this->makeVertexBuffer(data.begin(), data.count());
	}
	template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const std::vector<T>& data) {
	return this->makeVertexBuffer(data.data(), data.size());
	}
	template<typename T> sk_sp<const GrBuffer> makeVertexBuffer(const T* data, int count);

	void drawMesh(const GrMesh& mesh);

	private:
	GrOpFlushState* fState;
	};

	struct Box {
	float fX, fY;
	GrColor fColor;
	};

	////////////////////////////////////////////////////////////////////////////////////////////////////

	/**
	* This is a GPU-backend specific test. It tries to test all possible usecases of GrMesh. The test
	* works by drawing checkerboards of colored boxes, reading back the pixels, and comparing with
	* expected results. The boxes are drawn on integer boundaries and the (opaque) colors are chosen
	* from the set (r,g,b) = (0,255)^3, so the GPU renderings ought to produce exact matches.
	*/

	static void run_test(GrContext* context, const char* testName, skiatest::Reporter*,
	const sk_sp<GrRenderTargetContext>&, const SkBitmap& gold,
	std::function<void(DrawMeshHelper*)> testFn);

	DEF_GPUTEST_FOR_RENDERING_CONTEXTS(GrMeshTest, reporter, ctxInfo) {
	GrContext* context = ctxInfo.grContext();

	const GrBackendFormat format =
	context->contextPriv().caps()->getBackendFormatFromColorType(kRGBA_8888_SkColorType);

	sk_sp<GrRenderTargetContext> rtc(context->contextPriv().makeDeferredRenderTargetContext(
	format, SkBackingFit::kExact, kImageWidth, kImageHeight, kRGBA_8888_GrPixelConfig,
	nullptr));
	if (!rtc) {
	ERRORF(reporter, "could not create render target context.");
	return;
	}

	SkTArray<Box> boxes;
	SkTArray<std::array<Box, 4>> vertexData;
	SkBitmap gold;

	// ---- setup ----------

	SkPaint paint;
	paint.setBlendMode(SkBlendMode::kSrc);
	gold.allocN32Pixels(kImageWidth, kImageHeight);

	SkCanvas goldCanvas(gold);

	for (int y = 0; y < kBoxCountY; ++y) {
	for (int x = 0; x < kBoxCountX; ++x) {
	int c = y + x;
	int rgb[3] = {-(c & 1) & 0xff, -((c >> 1) & 1) & 0xff, -((c >> 2) & 1) & 0xff};

	const Box box = boxes.push_back() = {
	float(x * kBoxSize),
	float(y * kBoxSize),
	GrColorPackRGBA(rgb[0], rgb[1], rgb[2], 255)
	};

	std::array<Box, 4>& boxVertices = vertexData.push_back();
	for (int i = 0; i < 4; ++i) {
	boxVertices[i] = {
	box.fX + (i / 2) * kBoxSize,
	box.fY + (i % 2) * kBoxSize,
	box.fColor
	};
	}

	paint.setARGB(255, rgb[0], rgb[1], rgb[2]);
	goldCanvas.drawRect(SkRect::MakeXYWH(box.fX, box.fY, kBoxSize, kBoxSize), paint);
	}
	}

	goldCanvas.flush();

	// ---- tests ----------

	#define VALIDATE(buff) \
	do { \
	if (!buff) { \
	ERRORF(reporter, #buff " is null."); \
	return; \
	} \
	} while (0)

	run_test(context, "setNonIndexedNonInstanced", reporter, rtc, gold,
	[&](DrawMeshHelper* helper) {
	SkTArray<Box> expandedVertexData;
	for (int i = 0; i < kBoxCount; ++i) {
	for (int j = 0; j < 6; ++j) {
	expandedVertexData.push_back(vertexData[i][kIndexPattern[j]]);
	}
	}

	// Draw boxes one line at a time to exercise base vertex.
	auto vbuff = helper->makeVertexBuffer(expandedVertexData);
	VALIDATE(vbuff);
	for (int y = 0; y < kBoxCountY; ++y) {
	GrMesh mesh(GrPrimitiveType::kTriangles);
	mesh.setNonIndexedNonInstanced(kBoxCountX * 6);
	mesh.setVertexData(vbuff, y * kBoxCountX * 6);
	helper->drawMesh(mesh);
	}
	});

	run_test(context, "setIndexed", reporter, rtc, gold, [&](DrawMeshHelper* helper) {
	auto ibuff = helper->getIndexBuffer();
	VALIDATE(ibuff);
	auto vbuff = helper->makeVertexBuffer(vertexData);
	VALIDATE(vbuff);
	int baseRepetition = 0;
	int i = 0;

	// Start at various repetitions within the patterned index buffer to exercise base index.
	while (i < kBoxCount) {
	GR_STATIC_ASSERT(kIndexPatternRepeatCount >= 3);
	int repetitionCount = SkTMin(3 - baseRepetition, kBoxCount - i);

	GrMesh mesh(GrPrimitiveType::kTriangles);
	mesh.setIndexed(ibuff, repetitionCount * 6, baseRepetition * 6, baseRepetition * 4,
	(baseRepetition + repetitionCount) * 4 - 1, GrPrimitiveRestart::kNo);
	mesh.setVertexData(vbuff, (i - baseRepetition) * 4);
	helper->drawMesh(mesh);

	baseRepetition = (baseRepetition + 1) % 3;
	i += repetitionCount;
	}
	});

	run_test(context, "setIndexedPatterned", reporter, rtc, gold, [&](DrawMeshHelper* helper) {
	auto ibuff = helper->getIndexBuffer();
	VALIDATE(ibuff);
	auto vbuff = helper->makeVertexBuffer(vertexData);
	VALIDATE(vbuff);

	// Draw boxes one line at a time to exercise base vertex. setIndexedPatterned does not
	// support a base index.
	for (int y = 0; y < kBoxCountY; ++y) {
	GrMesh mesh(GrPrimitiveType::kTriangles);
	mesh.setIndexedPatterned(ibuff, 6, 4, kBoxCountX, kIndexPatternRepeatCount);
	mesh.setVertexData(vbuff, y * kBoxCountX * 4);
	helper->drawMesh(mesh);
	}
	});

	for (bool indexed : {false, true}) {
	if (!context->contextPriv().caps()->instanceAttribSupport()) {
	break;
	}

	run_test(context, indexed ? "setIndexedInstanced" : "setInstanced",
	reporter, rtc, gold, [&](DrawMeshHelper* helper) {
	auto idxbuff = indexed ? helper->getIndexBuffer() : nullptr;
	auto instbuff = helper->makeVertexBuffer(boxes);
	VALIDATE(instbuff);
	auto vbuff = helper->makeVertexBuffer(std::vector<float>{0,0, 0,1, 1,0, 1,1});
	VALIDATE(vbuff);
	auto vbuff2 = helper->makeVertexBuffer( // for testing base vertex.
	std::vector<float>{-1,-1, -1,-1, 0,0, 0,1, 1,0, 1,1});
	VALIDATE(vbuff2);

	// Draw boxes one line at a time to exercise base instance, base vertex, and null vertex
	// buffer. setIndexedInstanced intentionally does not support a base index.
	for (int y = 0; y < kBoxCountY; ++y) {
	GrMesh mesh(indexed ? GrPrimitiveType::kTriangles
	: GrPrimitiveType::kTriangleStrip);
	if (indexed) {
	VALIDATE(idxbuff);
	mesh.setIndexedInstanced(idxbuff, 6, instbuff, kBoxCountX, y * kBoxCountX,
	GrPrimitiveRestart::kNo);
	} else {
	mesh.setInstanced(instbuff, kBoxCountX, y * kBoxCountX, 4);
	}
	switch (y % 3) {
	case 0:
	if (context->contextPriv().caps()->shaderCaps()->vertexIDSupport()) {
	if (y % 2) {
	// We don't need this call because it's the initial state of GrMesh.
	mesh.setVertexData(nullptr);
	}
	break;
	}
	// Fallthru.
	case 1:
	mesh.setVertexData(vbuff);
	break;
	case 2:
	mesh.setVertexData(vbuff2, 2);
	break;
	}
	helper->drawMesh(mesh);
	}
	});
	}
	}

	////////////////////////////////////////////////////////////////////////////////////////////////////

	class GrMeshTestOp : public GrDrawOp {
	public:
	DEFINE_OP_CLASS_ID

	static std::unique_ptr<GrDrawOp> Make(GrContext* context,
	std::function<void(DrawMeshHelper*)> testFn) {
	GrOpMemoryPool* pool = context->contextPriv().opMemoryPool();

	return pool->allocate<GrMeshTestOp>(testFn);
	}

	private:
	friend class GrOpMemoryPool; // for ctor

	GrMeshTestOp(std::function<void(DrawMeshHelper*)> testFn)
	: INHERITED(ClassID())
	, fTestFn(testFn) {
	this->setBounds(SkRect::MakeIWH(kImageWidth, kImageHeight),
	HasAABloat::kNo, IsZeroArea::kNo);
	}

	const char* name() const override { return "GrMeshTestOp"; }
	FixedFunctionFlags fixedFunctionFlags() const override { return FixedFunctionFlags::kNone; }
	GrProcessorSet::Analysis finalize(const GrCaps&, const GrAppliedClip*) override {
	return GrProcessorSet::EmptySetAnalysis();
	}
	void onPrepare(GrOpFlushState*) override {}
	void onExecute(GrOpFlushState* state, const SkRect& chainBounds) override {
	DrawMeshHelper helper(state);
	fTestFn(&helper);
	}

	std::function<void(DrawMeshHelper*)> fTestFn;

	typedef GrDrawOp INHERITED;
	};

	class GrMeshTestProcessor : public GrGeometryProcessor {
	public:
	GrMeshTestProcessor(bool instanced, bool hasVertexBuffer)
	: INHERITED(kGrMeshTestProcessor_ClassID) {
	if (instanced) {
	fInstanceLocation = {"location", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
	fInstanceColor = {"color", kUByte4_norm_GrVertexAttribType, kHalf4_GrSLType};
	this->setInstanceAttributes(&fInstanceLocation, 2);
	if (hasVertexBuffer) {
	fVertexPosition = {"vertex", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
	this->setVertexAttributes(&fVertexPosition, 1);
	}
	} else {
	fVertexPosition = {"vertex", kFloat2_GrVertexAttribType, kHalf2_GrSLType};
	fVertexColor = {"color", kUByte4_norm_GrVertexAttribType, kHalf4_GrSLType};
	this->setVertexAttributes(&fVertexPosition, 2);
	}
	}

	const char* name() const override { return "GrMeshTest Processor"; }

	const Attribute& inColor() const {
	return fVertexColor.isInitialized() ? fVertexColor : fInstanceColor;
	}

	void getGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder* b) const final {
	b->add32(fInstanceLocation.isInitialized());
	b->add32(fVertexPosition.isInitialized());
	}

	GrGLSLPrimitiveProcessor* createGLSLInstance(const GrShaderCaps&) const final;

	private:
	Attribute fVertexPosition;
	Attribute fVertexColor;

	Attribute fInstanceLocation;
	Attribute fInstanceColor;

	friend class GLSLMeshTestProcessor;
	typedef GrGeometryProcessor INHERITED;
	};

	class GLSLMeshTestProcessor : public GrGLSLGeometryProcessor {
	void setData(const GrGLSLProgramDataManager& pdman, const GrPrimitiveProcessor&,
	FPCoordTransformIter&& transformIter) final {}

	void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) final {
	const GrMeshTestProcessor& mp = args.fGP.cast<GrMeshTestProcessor>();

	GrGLSLVaryingHandler* varyingHandler = args.fVaryingHandler;
	varyingHandler->emitAttributes(mp);
	varyingHandler->addPassThroughAttribute(mp.inColor(), args.fOutputColor);

	GrGLSLVertexBuilder* v = args.fVertBuilder;
	if (!mp.fInstanceLocation.isInitialized()) {
	v->codeAppendf("float2 vertex = %s;", mp.fVertexPosition.name());
	} else {
	if (mp.fVertexPosition.isInitialized()) {
	v->codeAppendf("float2 offset = %s;", mp.fVertexPosition.name());
	} else {
	v->codeAppend ("float2 offset = float2(sk_VertexID / 2, sk_VertexID % 2);");
	}
	v->codeAppendf("float2 vertex = %s + offset * %i;", mp.fInstanceLocation.name(),
	kBoxSize);
	}
	gpArgs->fPositionVar.set(kFloat2_GrSLType, "vertex");

	GrGLSLFPFragmentBuilder* f = args.fFragBuilder;
	f->codeAppendf("%s = half4(1);", args.fOutputCoverage);
	}
	};

	GrGLSLPrimitiveProcessor* GrMeshTestProcessor::createGLSLInstance(const GrShaderCaps&) const {
	return new GLSLMeshTestProcessor;
	}

	////////////////////////////////////////////////////////////////////////////////////////////////////

	template<typename T>
	sk_sp<const GrBuffer> DrawMeshHelper::makeVertexBuffer(const T* data, int count) {
	return sk_sp<const GrBuffer>(
	fState->resourceProvider()->createBuffer(
	count * sizeof(T), kVertex_GrBufferType, kDynamic_GrAccessPattern,
	GrResourceProvider::Flags::kNoPendingIO \|
	GrResourceProvider::Flags::kRequireGpuMemory, data));
	}

	sk_sp<const GrBuffer> DrawMeshHelper::getIndexBuffer() {
	GR_DEFINE_STATIC_UNIQUE_KEY(gIndexBufferKey);
	return fState->resourceProvider()->findOrCreatePatternedIndexBuffer(
	kIndexPattern, 6, kIndexPatternRepeatCount, 4, gIndexBufferKey);
	}

	void DrawMeshHelper::drawMesh(const GrMesh& mesh) {
	GrPipeline pipeline(GrScissorTest::kDisabled, SkBlendMode::kSrc);
	GrMeshTestProcessor mtp(mesh.isInstanced(), mesh.hasVertexData());
	fState->rtCommandBuffer()->draw(mtp, pipeline, nullptr, nullptr, &mesh, 1,
	SkRect::MakeIWH(kImageWidth, kImageHeight));
	}

	static void run_test(GrContext* context, const char* testName, skiatest::Reporter* reporter,
	const sk_sp<GrRenderTargetContext>& rtc, const SkBitmap& gold,
	std::function<void(DrawMeshHelper*)> testFn) {
	const int w = gold.width(), h = gold.height(), rowBytes = gold.rowBytes();
	const uint32_t* goldPx = reinterpret_cast<const uint32_t*>(gold.getPixels());
	if (h != rtc->height() \|\| w != rtc->width()) {
	ERRORF(reporter, "[%s] expectation and rtc not compatible (?).", testName);
	return;
	}
	if (sizeof(uint32_t) * kImageWidth != gold.rowBytes()) {
	ERRORF(reporter, "unexpected row bytes in gold image.", testName);
	return;
	}

	SkAutoSTMalloc<kImageHeight * kImageWidth, uint32_t> resultPx(h * rowBytes);
	rtc->clear(nullptr, SkPMColor4f::FromBytes_RGBA(0xbaaaaaad),
	GrRenderTargetContext::CanClearFullscreen::kYes);
	rtc->priv().testingOnly_addDrawOp(GrMeshTestOp::Make(context, testFn));
	rtc->readPixels(gold.info(), resultPx, rowBytes, 0, 0, 0);
	for (int y = 0; y < h; ++y) {
	for (int x = 0; x < w; ++x) {
	uint32_t expected = goldPx[y * kImageWidth + x];
	uint32_t actual = resultPx[y * kImageWidth + x];
	if (expected != actual) {
	ERRORF(reporter, "[%s] pixel (%i,%i): got 0x%x expected 0x%x",
	testName, x, y, actual, expected);
	return;
	}
	}
	}
	}