YUV scale fix

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"
 
 #if SK_SUPPORT_GPU
 
 #include "GrContext.h"
 #include "GrTest.h"
 #include "effects/GrYUVtoRGBEffect.h"
 #include "SkBitmap.h"
 #include "SkGr.h"
 #include "SkGradientShader.h"
 
+#define IMAGE_SIZE 24
+
 namespace skiagm {
 /**
  * This GM directly exercises GrYUVtoRGBEffect.
  */
 class YUVtoRGBEffect : public GM {
 public:
     YUVtoRGBEffect() {
         this->setBGColor(0xFFFFFFFF);
     }
 
 protected:
     SkString onShortName() SK_OVERRIDE {
         return SkString("yuv_to_rgb_effect");
     }
 
     SkISize onISize() SK_OVERRIDE {
         return SkISize::Make(334, 128);
     }
 
     void onOnceBeforeDraw() SK_OVERRIDE {
-        SkImageInfo info = SkImageInfo::MakeA8(24, 24);
+        SkImageInfo info = SkImageInfo::MakeA8(IMAGE_SIZE, IMAGE_SIZE);
         fBmp[0].allocPixels(info);
         fBmp[1].allocPixels(info);
         fBmp[2].allocPixels(info);
         unsigned char* pixels[3];
         for (int i = 0; i < 3; ++i) {
             pixels[i] = (unsigned char*)fBmp[i].getPixels();
         }
         int color[] = {0, 85, 170};
         const int limit[] = {255, 0, 255};
         const int invl[]  = {0, 255, 0};
@@ skipping 29 matching lines @@
         texture[1].reset(GrRefCachedBitmapTexture(context, fBmp[1], NULL));
         texture[2].reset(GrRefCachedBitmapTexture(context, fBmp[2], NULL));
 
         if (!texture[0] || !texture[1] || !texture[2]) {
             return;
         }
 
         static const SkScalar kDrawPad = 10.f;
         static const SkScalar kTestPad = 10.f;
         static const SkScalar kColorSpaceOffset = 64.f;
+        SkISize sizes[3] = {{IMAGE_SIZE, IMAGE_SIZE},
+                            {IMAGE_SIZE, IMAGE_SIZE},
+                            {IMAGE_SIZE, IMAGE_SIZE}};
 
         for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace;
              ++space) {
             SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fBmp[0].width()),
                                                SkIntToScalar(fBmp[0].height()));
             renderRect.outset(kDrawPad, kDrawPad);
 
             SkScalar y = kDrawPad + kTestPad + space * kColorSpaceOffset;
             SkScalar x = kDrawPad + kTestPad;
 
             const int indices[6][3] = {{0, 1, 2}, {0, 2, 1}, {1, 0, 2},
                                        {1, 2, 0}, {2, 0, 1}, {2, 1, 0}};
 
             for (int i = 0; i < 6; ++i) {
                 SkAutoTUnref<GrFragmentProcessor> fp(
                             GrYUVtoRGBEffect::Create(texture[indices[i][0]],
-                                                    texture[indices[i][1]],
-                                                    texture[indices[i][2]],
-                                                    static_cast<SkYUVColorSpace>(space)));
+                                                     texture[indices[i][1]],
+                                                     texture[indices[i][2]],
+                                                     sizes,
+                                                     static_cast<SkYUVColorSpace>(space)));
                 if (fp) {
                     SkMatrix viewMatrix;
                     viewMatrix.setTranslate(x, y);
                     GrPipelineBuilder pipelineBuilder;
                     pipelineBuilder.setRenderTarget(rt);
                     pipelineBuilder.addColorProcessor(fp);
                     tt.target()->drawSimpleRect(&pipelineBuilder, GrColor_WHITE, viewMatrix,
                                                 renderRect);
                 }
                 x += renderRect.width() + kTestPad;
             }
         }
      }
 
 private:
     SkBitmap fBmp[3];
 
     typedef GM INHERITED;
 };
 
 DEF_GM( return SkNEW(YUVtoRGBEffect); )
 }
 
 #endif