blob: 04081c3e60421d4945fb13737653f436eb6e9578 [file] [log] [blame]
/*
* 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/gm.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkColor.h"
#include "include/core/SkImageInfo.h"
#include "include/core/SkMatrix.h"
#include "include/core/SkRect.h"
#include "include/core/SkScalar.h"
#include "include/core/SkSize.h"
#include "include/core/SkString.h"
#include "include/core/SkYUVAInfo.h"
#include "include/core/SkYUVAPixmaps.h"
#include "src/gpu/GrBitmapTextureMaker.h"
#include "src/gpu/GrDirectContextPriv.h"
#include "src/gpu/GrPaint.h"
#include "src/gpu/GrSamplerState.h"
#include "src/gpu/GrSurfaceDrawContext.h"
#include "src/gpu/GrTextureProxy.h"
#include "src/gpu/GrYUVATextureProxies.h"
#include "src/gpu/effects/GrYUVtoRGBEffect.h"
#include <memory>
#include <utility>
class SkCanvas;
namespace skiagm {
//////////////////////////////////////////////////////////////////////////////
// This GM tests subsetting YUV multiplanar images where the U and V
// planes have different resolution from Y. See skbug:8959
class YUVtoRGBSubsetEffect : public GpuGM {
public:
YUVtoRGBSubsetEffect() {
this->setBGColor(0xFFFFFFFF);
}
protected:
SkString onShortName() override {
return SkString("yuv_to_rgb_subset_effect");
}
SkISize onISize() override { return {1310, 540}; }
void makePixmaps() {
SkYUVAInfo yuvaInfo = SkYUVAInfo({8, 8},
SkYUVAInfo::PlaneConfig::kY_U_V,
SkYUVAInfo::Subsampling::k420,
kJPEG_Full_SkYUVColorSpace);
SkColorType colorTypes[] = {kAlpha_8_SkColorType,
kAlpha_8_SkColorType,
kAlpha_8_SkColorType};
SkYUVAPixmapInfo pmapInfo(yuvaInfo, colorTypes, nullptr);
fPixmaps = SkYUVAPixmaps::Allocate(pmapInfo);
unsigned char innerY[16] = {149, 160, 130, 105,
160, 130, 105, 149,
130, 105, 149, 160,
105, 149, 160, 130};
unsigned char innerU[4] = {43, 75, 145, 200};
unsigned char innerV[4] = {88, 180, 200, 43};
int outerYUV[] = {128, 128, 128};
SkBitmap bitmaps[3];
for (int i = 0; i < 3; ++i) {
bitmaps[i].installPixels(fPixmaps.plane(i));
bitmaps[i].eraseColor(SkColorSetARGB(outerYUV[i], 0, 0, 0));
}
SkPixmap innerYPM(SkImageInfo::MakeA8(4, 4), innerY, 4);
SkPixmap innerUPM(SkImageInfo::MakeA8(2, 2), innerU, 2);
SkPixmap innerVPM(SkImageInfo::MakeA8(2, 2), innerV, 2);
bitmaps[0].writePixels(innerYPM, 2, 2);
bitmaps[1].writePixels(innerUPM, 1, 1);
bitmaps[2].writePixels(innerVPM, 1, 1);
}
DrawResult onGpuSetup(GrDirectContext* context, SkString* errorMsg) override {
if (!context) {
return DrawResult::kSkip;
}
if (!fPixmaps.isValid()) {
this->makePixmaps();
}
GrSurfaceProxyView views[SkYUVAInfo::kMaxPlanes];
GrColorType colorTypes[SkYUVAInfo::kMaxPlanes];
for (int i = 0; i < fPixmaps.numPlanes(); ++i) {
SkBitmap bitmap;
bitmap.installPixels(fPixmaps.plane(i));
bitmap.setImmutable();
GrBitmapTextureMaker maker(
context, bitmap, GrImageTexGenPolicy::kNew_Uncached_Budgeted);
views[i] = maker.view(GrMipmapped::kNo);
if (!views[i]) {
*errorMsg = "Failed to create proxy";
return context->abandoned() ? DrawResult::kSkip : DrawResult::kFail;
}
colorTypes[i] = SkColorTypeToGrColorType(bitmap.colorType());
}
fProxies = GrYUVATextureProxies(fPixmaps.yuvaInfo(), views, colorTypes);
if (!fProxies.isValid()) {
*errorMsg = "Failed to create GrYUVATextureProxies";
return DrawResult::kFail;
}
return DrawResult::kOk;
}
void onGpuTeardown() override { fProxies = {}; }
DrawResult onDraw(GrRecordingContext* context,
GrSurfaceDrawContext* surfaceDrawContext,
SkCanvas* canvas,
SkString* errorMsg) override {
static const GrSamplerState::Filter kFilters[] = {GrSamplerState::Filter::kNearest,
GrSamplerState::Filter::kLinear};
static const SkRect kColorRect = SkRect::MakeLTRB(2.f, 2.f, 6.f, 6.f);
// Outset to visualize wrap modes.
SkRect rect = SkRect::Make(fProxies.yuvaInfo().dimensions());
rect = rect.makeOutset(fProxies.yuvaInfo().width()/2.f, fProxies.yuvaInfo().height()/2.f);
SkScalar y = kTestPad;
// Rows are filter modes.
for (uint32_t i = 0; i < SK_ARRAY_COUNT(kFilters); ++i) {
SkScalar x = kTestPad;
// Columns are non-subsetted followed by subsetted with each WrapMode in a row
for (uint32_t j = 0; j < GrSamplerState::kWrapModeCount + 1; ++j) {
SkMatrix ctm = SkMatrix::Translate(x, y);
ctm.postScale(10.f, 10.f);
const SkRect* subset = j > 0 ? &kColorRect : nullptr;
GrSamplerState samplerState;
samplerState.setFilterMode(kFilters[i]);
if (j > 0) {
auto wm = static_cast<GrSamplerState::WrapMode>(j - 1);
samplerState.setWrapModeX(wm);
samplerState.setWrapModeY(wm);
}
const auto& caps = *context->priv().caps();
std::unique_ptr<GrFragmentProcessor> fp =
GrYUVtoRGBEffect::Make(fProxies, samplerState, caps, SkMatrix::I(), subset);
if (fp) {
GrPaint grPaint;
grPaint.setColorFragmentProcessor(std::move(fp));
surfaceDrawContext->drawRect(
nullptr, std::move(grPaint), GrAA::kYes, ctm, rect);
}
x += rect.width() + kTestPad;
}
y += rect.height() + kTestPad;
}
return DrawResult::kOk;
}
private:
SkYUVAPixmaps fPixmaps;
GrYUVATextureProxies fProxies;
static constexpr SkScalar kTestPad = 10.f;
using INHERITED = GM;
};
DEF_GM(return new YUVtoRGBSubsetEffect;)
} // namespace skiagm