Add support for the Rec601 YUV color space to GrYUVtoRGBEffect.

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"
 
 namespace skiagm {
 /**
  * This GM directly exercises GrYUVtoRGBEffect.
  */
 class YUVtoRGBEffect : public GM {
 public:
-    YUVtoRGBEffect() {
+    YUVtoRGBEffect(SkPixelRef::YUVColorSpace colorSpace)
+    : fColorSpace(colorSpace) {
         this->setBGColor(0xFFFFFFFF);
     }
 
 protected:
     virtual SkString onShortName() SK_OVERRIDE {
-        return SkString("yuv_to_rgb_effect");
+        SkString out("yuv_to_rgb_effect_");
+        switch(fColorSpace) {
+            case SkPixelRef::kJPEG_YUVColorSpace:
+                out.append("jpeg");
+                break;
+            case SkPixelRef::kRec601_YUVColorSpace:
+                out.append("rec601");
+                break;
+        }
+        return out;
     }
 
     virtual SkISize onISize() SK_OVERRIDE {
         return SkISize::Make(334, 64);
     }
 
     virtual uint32_t onGetFlags() const SK_OVERRIDE {
         // This is a GPU-specific GM.
         return kGPUOnly_Flag;
     }
@@ skipping 55 matching lines @@
 
         SkScalar y = kDrawPad + kTestPad;
         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<GrEffect> effect(
                         GrYUVtoRGBEffect::Create(texture[indices[i][0]],
                                                  texture[indices[i][1]],
-                                                 texture[indices[i][2]]));
+                                                 texture[indices[i][2]],
+                                                 SkPixelRef::kJPEG_YUVColorSpace));
             if (effect) {
                 SkMatrix viewMatrix;
                 viewMatrix.setTranslate(x, y);
                 drawState->reset(viewMatrix);
                 drawState->setRenderTarget(rt);
                 drawState->setColor(0xffffffff);
                 drawState->addColorEffect(effect, 1);
                 tt.target()->drawSimpleRect(renderRect);
             }
             x += renderRect.width() + kTestPad;
         }
 
         GrUnlockAndUnrefCachedBitmapTexture(texture[0]);
         GrUnlockAndUnrefCachedBitmapTexture(texture[1]);
         GrUnlockAndUnrefCachedBitmapTexture(texture[2]);
     }
 
 private:
     SkBitmap fBmp[3];
+    SkPixelRef::YUVColorSpace fColorSpace;
 
     typedef GM INHERITED;
 };
 
-DEF_GM( return SkNEW(YUVtoRGBEffect); )
+DEF_GM( return SkNEW_ARGS(YUVtoRGBEffect, (SkPixelRef::kJPEG_YUVColorSpace)); )
+DEF_GM( return SkNEW_ARGS(YUVtoRGBEffect, (SkPixelRef::kRec601_YUVColorSpace)); )
 }
 
 #endif