You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
269 lines
9.0 KiB
269 lines
9.0 KiB
/*
|
|
* 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"
|
|
|
|
#include "GrContext.h"
|
|
#include "GrContextPriv.h"
|
|
#include "GrProxyProvider.h"
|
|
#include "GrRenderTargetContextPriv.h"
|
|
#include "GrTextureProxy.h"
|
|
#include "SkBitmap.h"
|
|
#include "SkGr.h"
|
|
#include "SkGradientShader.h"
|
|
#include "effects/GrYUVtoRGBEffect.h"
|
|
#include "ops/GrDrawOp.h"
|
|
#include "ops/GrFillRectOp.h"
|
|
|
|
#define YSIZE 8
|
|
#define USIZE 4
|
|
#define VSIZE 4
|
|
|
|
namespace skiagm {
|
|
/**
|
|
* This GM directly exercises GrYUVtoRGBEffect.
|
|
*/
|
|
class YUVtoRGBEffect : public GM {
|
|
public:
|
|
YUVtoRGBEffect() {
|
|
this->setBGColor(0xFFFFFFFF);
|
|
}
|
|
|
|
protected:
|
|
SkString onShortName() override {
|
|
return SkString("yuv_to_rgb_effect");
|
|
}
|
|
|
|
SkISize onISize() override {
|
|
return SkISize::Make(238, 120);
|
|
}
|
|
|
|
void onOnceBeforeDraw() override {
|
|
SkBitmap bmp[3];
|
|
SkImageInfo yinfo = SkImageInfo::MakeA8(YSIZE, YSIZE);
|
|
bmp[0].allocPixels(yinfo);
|
|
SkImageInfo uinfo = SkImageInfo::MakeA8(USIZE, USIZE);
|
|
bmp[1].allocPixels(uinfo);
|
|
SkImageInfo vinfo = SkImageInfo::MakeA8(VSIZE, VSIZE);
|
|
bmp[2].allocPixels(vinfo);
|
|
unsigned char* pixels[3];
|
|
for (int i = 0; i < 3; ++i) {
|
|
pixels[i] = (unsigned char*)bmp[i].getPixels();
|
|
}
|
|
int color[] = {0, 85, 170};
|
|
const int limit[] = {255, 0, 255};
|
|
const int invl[] = {0, 255, 0};
|
|
const int inc[] = {1, -1, 1};
|
|
for (int i = 0; i < 3; ++i) {
|
|
const size_t nbBytes = bmp[i].rowBytes() * bmp[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];
|
|
}
|
|
}
|
|
for (int i = 0; i < 3; ++i) {
|
|
fImage[i] = SkImage::MakeFromBitmap(bmp[i]);
|
|
}
|
|
}
|
|
|
|
void onDraw(SkCanvas* canvas) override {
|
|
GrRenderTargetContext* renderTargetContext =
|
|
canvas->internal_private_accessTopLayerRenderTargetContext();
|
|
if (!renderTargetContext) {
|
|
skiagm::GM::DrawGpuOnlyMessage(canvas);
|
|
return;
|
|
}
|
|
|
|
GrContext* context = canvas->getGrContext();
|
|
if (!context) {
|
|
return;
|
|
}
|
|
|
|
GrProxyProvider* proxyProvider = context->contextPriv().proxyProvider();
|
|
sk_sp<GrTextureProxy> proxies[3];
|
|
|
|
for (int i = 0; i < 3; ++i) {
|
|
proxies[i] = proxyProvider->createTextureProxy(fImage[i], kNone_GrSurfaceFlags, 1,
|
|
SkBudgeted::kYes, SkBackingFit::kExact);
|
|
if (!proxies[i]) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
constexpr SkScalar kDrawPad = 10.f;
|
|
constexpr SkScalar kTestPad = 10.f;
|
|
constexpr SkScalar kColorSpaceOffset = 36.f;
|
|
|
|
for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace; ++space) {
|
|
SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fImage[0]->width()),
|
|
SkIntToScalar(fImage[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) {
|
|
SkYUVAIndex yuvaIndices[4] = {
|
|
{ indices[i][0], SkColorChannel::kR },
|
|
{ indices[i][1], SkColorChannel::kR },
|
|
{ indices[i][2], SkColorChannel::kR },
|
|
{ -1, SkColorChannel::kA }
|
|
};
|
|
|
|
std::unique_ptr<GrFragmentProcessor> fp(
|
|
GrYUVtoRGBEffect::Make(proxies, yuvaIndices,
|
|
static_cast<SkYUVColorSpace>(space),
|
|
GrSamplerState::Filter::kNearest));
|
|
if (fp) {
|
|
GrPaint grPaint;
|
|
grPaint.setXPFactory(GrPorterDuffXPFactory::Get(SkBlendMode::kSrc));
|
|
grPaint.addColorFragmentProcessor(std::move(fp));
|
|
SkMatrix viewMatrix;
|
|
viewMatrix.setTranslate(x, y);
|
|
renderTargetContext->priv().testingOnly_addDrawOp(
|
|
GrFillRectOp::Make(context, std::move(grPaint), GrAAType::kNone,
|
|
viewMatrix, renderRect));
|
|
}
|
|
x += renderRect.width() + kTestPad;
|
|
}
|
|
}
|
|
}
|
|
|
|
private:
|
|
sk_sp<SkImage> fImage[3];
|
|
|
|
typedef GM INHERITED;
|
|
};
|
|
|
|
DEF_GM(return new YUVtoRGBEffect;)
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
class YUVNV12toRGBEffect : public GM {
|
|
public:
|
|
YUVNV12toRGBEffect() {
|
|
this->setBGColor(0xFFFFFFFF);
|
|
}
|
|
|
|
protected:
|
|
SkString onShortName() override {
|
|
return SkString("yuv_nv12_to_rgb_effect");
|
|
}
|
|
|
|
SkISize onISize() override {
|
|
return SkISize::Make(48, 120);
|
|
}
|
|
|
|
void onOnceBeforeDraw() override {
|
|
SkBitmap bmp[2];
|
|
SkImageInfo yinfo = SkImageInfo::MakeA8(YSIZE, YSIZE);
|
|
bmp[0].allocPixels(yinfo);
|
|
SkImageInfo uvinfo = SkImageInfo::MakeN32Premul(USIZE, USIZE);
|
|
bmp[1].allocPixels(uvinfo);
|
|
int color[] = {0, 85, 170};
|
|
const int limit[] = {255, 0, 255};
|
|
const int invl[] = {0, 255, 0};
|
|
const int inc[] = {1, -1, 1};
|
|
|
|
{
|
|
unsigned char* pixels = (unsigned char*)bmp[0].getPixels();
|
|
const size_t nbBytes = bmp[0].rowBytes() * bmp[0].height();
|
|
for (size_t j = 0; j < nbBytes; ++j) {
|
|
pixels[j] = (unsigned char)color[0];
|
|
color[0] = (color[0] == limit[0]) ? invl[0] : color[0] + inc[0];
|
|
}
|
|
}
|
|
|
|
{
|
|
for (int y = 0; y < bmp[1].height(); ++y) {
|
|
uint32_t* pixels = bmp[1].getAddr32(0, y);
|
|
for (int j = 0; j < bmp[1].width(); ++j) {
|
|
pixels[j] = SkColorSetARGB(0, color[1], color[2], 0);
|
|
color[1] = (color[1] == limit[1]) ? invl[1] : color[1] + inc[1];
|
|
color[2] = (color[2] == limit[2]) ? invl[2] : color[2] + inc[2];
|
|
}
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < 2; ++i) {
|
|
fImage[i] = SkImage::MakeFromBitmap(bmp[i]);
|
|
}
|
|
}
|
|
|
|
void onDraw(SkCanvas* canvas) override {
|
|
GrRenderTargetContext* renderTargetContext =
|
|
canvas->internal_private_accessTopLayerRenderTargetContext();
|
|
if (!renderTargetContext) {
|
|
skiagm::GM::DrawGpuOnlyMessage(canvas);
|
|
return;
|
|
}
|
|
|
|
GrContext* context = canvas->getGrContext();
|
|
if (!context) {
|
|
return;
|
|
}
|
|
|
|
GrProxyProvider* proxyProvider = context->contextPriv().proxyProvider();
|
|
sk_sp<GrTextureProxy> proxies[2];
|
|
|
|
for (int i = 0; i < 2; ++i) {
|
|
proxies[i] = proxyProvider->createTextureProxy(fImage[i], kNone_GrSurfaceFlags, 1,
|
|
SkBudgeted::kYes, SkBackingFit::kExact);
|
|
if (!proxies[i]) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
SkYUVAIndex yuvaIndices[4] = {
|
|
{ 0, SkColorChannel::kR },
|
|
{ 1, SkColorChannel::kR },
|
|
{ 1, SkColorChannel::kG },
|
|
{ -1, SkColorChannel::kA }
|
|
};
|
|
|
|
constexpr SkScalar kDrawPad = 10.f;
|
|
constexpr SkScalar kTestPad = 10.f;
|
|
constexpr SkScalar kColorSpaceOffset = 36.f;
|
|
|
|
for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace; ++space) {
|
|
SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fImage[0]->width()),
|
|
SkIntToScalar(fImage[0]->height()));
|
|
renderRect.outset(kDrawPad, kDrawPad);
|
|
|
|
SkScalar y = kDrawPad + kTestPad + space * kColorSpaceOffset;
|
|
SkScalar x = kDrawPad + kTestPad;
|
|
|
|
GrPaint grPaint;
|
|
grPaint.setXPFactory(GrPorterDuffXPFactory::Get(SkBlendMode::kSrc));
|
|
auto fp = GrYUVtoRGBEffect::Make(proxies, yuvaIndices,
|
|
static_cast<SkYUVColorSpace>(space),
|
|
GrSamplerState::Filter::kNearest);
|
|
if (fp) {
|
|
SkMatrix viewMatrix;
|
|
viewMatrix.setTranslate(x, y);
|
|
grPaint.addColorFragmentProcessor(std::move(fp));
|
|
std::unique_ptr<GrDrawOp> op(GrFillRectOp::Make(context, std::move(grPaint),
|
|
GrAAType::kNone, viewMatrix, renderRect));
|
|
renderTargetContext->priv().testingOnly_addDrawOp(std::move(op));
|
|
}
|
|
}
|
|
}
|
|
|
|
private:
|
|
sk_sp<SkImage> fImage[2];
|
|
|
|
typedef GM INHERITED;
|
|
};
|
|
|
|
DEF_GM(return new YUVNV12toRGBEffect;)
|
|
}
|