Index: gm/yuvtorgbeffect.cpp |
diff --git a/gm/yuvtorgbeffect.cpp b/gm/yuvtorgbeffect.cpp |
index 12676f262e2477f72cd3a19487ea2b7400e3a0fd..9750da813c4f32dfacf4b2656e798823167eec63 100644 |
--- a/gm/yuvtorgbeffect.cpp |
+++ b/gm/yuvtorgbeffect.cpp |
@@ -25,13 +25,23 @@ namespace skiagm { |
*/ |
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 { |
@@ -107,7 +117,8 @@ protected: |
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); |
@@ -127,11 +138,13 @@ protected: |
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 |