Added TestCreate for GrComposeEffect

src/core/SkComposeShader.cpp

 
 /*
  * Copyright 2006 The Android Open Source Project
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "SkComposeShader.h"
@@ skipping 180 matching lines @@
             x += n;
             count -= n;
         } while (count > 0);
     }
 }
 
 #if SK_SUPPORT_GPU
 
 #include "SkGr.h"
 #include "GrProcessor.h"
+#include "effects/GrConstColorProcessor.h"
 #include "gl/GrGLBlend.h"
 #include "gl/builders/GrGLProgramBuilder.h"
-#include "effects/GrConstColorProcessor.h"
 
 /////////////////////////////////////////////////////////////////////
 
 class GrComposeEffect : public GrFragmentProcessor {
 public:
 
     static GrFragmentProcessor* Create(const GrFragmentProcessor* fpA,
                                        const GrFragmentProcessor* fpB, SkXfermode::Mode mode) {
         return SkNEW_ARGS(GrComposeEffect, (fpA, fpB, mode));
     }
@@ skipping 14 matching lines @@
         SkDEBUGCODE(int shaderAChildIndex = )this->registerChildProcessor(fpA);
         SkDEBUGCODE(int shaderBChildIndex = )this->registerChildProcessor(fpB);
         SkASSERT(0 == shaderAChildIndex);
         SkASSERT(1 == shaderBChildIndex);
     }
 
     GrGLFragmentProcessor* onCreateGLInstance() const override;
 
     SkXfermode::Mode fMode;
 
+    GR_DECLARE_FRAGMENT_PROCESSOR_TEST;
+
     typedef GrFragmentProcessor INHERITED;
 };
 
 /////////////////////////////////////////////////////////////////////
 
 class GrGLComposeEffect : public GrGLFragmentProcessor {
 public:
     GrGLComposeEffect(const GrProcessor& processor) {}
 
     void emitCode(EmitArgs&) override;
 
 private:
     typedef GrGLFragmentProcessor INHERITED;
 };
 
+/////////////////////////////////////////////////////////////////////
+
+GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrComposeEffect);
+
+const GrFragmentProcessor* GrComposeEffect::TestCreate(GrProcessorTestData* d) {
+#if SK_ALLOW_STATIC_GLOBAL_INITIALIZERS
+    // Create two random frag procs.
+    // For now, we'll prevent either children from being a shader with children to prevent the
+    // possibility of an arbitrarily large tree of procs.
+    SkAutoTUnref<const GrFragmentProcessor> fpA;
+    do {
+        fpA.reset(GrProcessorTestFactory<GrFragmentProcessor>::CreateStage(d));
+        SkASSERT(fpA);
+    } while (fpA->numChildProcessors() != 0);
+    SkAutoTUnref<const GrFragmentProcessor> fpB;
+    do {
+        fpB.reset(GrProcessorTestFactory<GrFragmentProcessor>::CreateStage(d));
+        SkASSERT(fpB);
+    } while (fpB->numChildProcessors() != 0);
+
+    SkXfermode::Mode mode = static_cast<SkXfermode::Mode>(
+            d->fRandom->nextRangeU(0, SkXfermode::kLastCoeffMode));
+    return SkNEW_ARGS(GrComposeEffect, (fpA, fpB, mode));
+#else
+    SkFAIL("Should not be called if !SK_ALLOW_STATIC_GLOBAL_INITIALIZERS");
+    return nullptr;
+#endif
+}
+
 bool GrComposeEffect::onIsEqual(const GrFragmentProcessor& other) const {
     const GrComposeEffect& cs = other.cast<GrComposeEffect>();
     return fMode == cs.fMode;
 }
 
 void GrComposeEffect::onComputeInvariantOutput(GrInvariantOutput* inout) const {
     inout->setToUnknown(GrInvariantOutput::kWill_ReadInput);
 }
 
 void GrComposeEffect::onGetGLProcessorKey(const GrGLSLCaps& caps, GrProcessorKeyBuilder* b) const {
@@ skipping 97 matching lines @@
     if (fMode) {
         str->append(" Xfermode: ");
         fMode->toString(str);
     }
 
     this->INHERITED::toString(str);
 
     str->append(")");
 }
 #endif