Clean up test drawContext usage

gm/yuvtorgbeffect.cpp

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 // This test only works with the GPU backend.
 
 #include "gm.h"
@@ skipping 51 matching lines @@
         for (int i = 0; i < 3; ++i) {
             const size_t nbBytes = fBmp[i].rowBytes() * fBmp[i].height();
             for (size_t j = 0; j < nbBytes; ++j) {
                 pixels[i][j] = (unsigned char)color[i];
                 color[i] = (color[i] == limit[i]) ? invl[i] : color[i] + inc[i];
             }
         }
     }
 
     void onDraw(SkCanvas* canvas) override {
-        GrRenderTarget* rt = canvas->internal_private_accessTopLayerRenderTarget();
-        if (nullptr == rt) {
-            return;
+        GrDrawContext* drawContext = canvas->internal_private_accessTopLayerDrawContext();
+        if (!drawContext) {
+            skiagm::GM::DrawGpuOnlyMessage(canvas);
+            return;        
         }
-        GrContext* context = rt->getContext();
-        if (nullptr == context) {
-            skiagm::GM::DrawGpuOnlyMessage(canvas);
+
+        GrContext* context = canvas->getGrContext();
+        if (!context) {
             return;
         }
 
-        sk_sp<GrDrawContext> drawContext(context->drawContext(sk_ref_sp(rt)));
-        if (!drawContext) {
-            return;
-        }
-
         SkAutoTUnref<GrTexture> texture[3];
         texture[0].reset(GrRefCachedBitmapTexture(context, fBmp[0],
                                                   GrTextureParams::ClampBilerp()));
         texture[1].reset(GrRefCachedBitmapTexture(context, fBmp[1],
                                                   GrTextureParams::ClampBilerp()));
         texture[2].reset(GrRefCachedBitmapTexture(context, fBmp[2],
                                                   GrTextureParams::ClampBilerp()));
 
         if (!texture[0] || !texture[1] || !texture[2]) {
             return;
@@ skipping 22 matching lines @@
                     GrPorterDuffXPFactory::Create(SkXfermode::kSrc_Mode))->unref();
                 SkAutoTUnref<const GrFragmentProcessor> fp(
                             GrYUVEffect::CreateYUVToRGB(texture[indices[i][0]],
                                                         texture[indices[i][1]],
                                                         texture[indices[i][2]],
                                                         sizes,
                                                         static_cast<SkYUVColorSpace>(space)));
                 if (fp) {
                     SkMatrix viewMatrix;
                     viewMatrix.setTranslate(x, y);
-                    pipelineBuilder.setRenderTarget(rt);
+                    pipelineBuilder.setRenderTarget(drawContext->accessRenderTarget());
                     pipelineBuilder.addColorFragmentProcessor(fp);
                     SkAutoTUnref<GrDrawBatch> batch(
                             GrRectBatchFactory::CreateNonAAFill(GrColor_WHITE, viewMatrix,
                                                                 renderRect, nullptr, nullptr));
                     drawContext->drawContextPriv().testingOnly_drawBatch(pipelineBuilder, batch);
                 }
                 x += renderRect.width() + kTestPad;
             }
         }
      }
 
 private:
     SkBitmap fBmp[3];
 
     typedef GM INHERITED;
 };
 
 DEF_GM(return new YUVtoRGBEffect;)
 }
 
 #endif