skia2/tests/RuntimeBlendTest.cpp
Robert Phillips 5bde3f4b2f Add kSkBlendModeCount
I keep bumping into the need for this

Change-Id: I69384f7d590c163fd6244bdc64cc5e48450fecd4
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/540171
Commit-Queue: Robert Phillips <robertphillips@google.com>
Reviewed-by: Brian Osman <brianosman@google.com>
2022-05-13 18:56:38 +00:00

90 lines
3.7 KiB
C++

/*
* Copyright 2021 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "gm/gm.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkCanvas.h"
#include "include/core/SkPaint.h"
#include "include/core/SkSize.h"
#include "include/core/SkSurface.h"
#include "include/effects/SkRuntimeEffect.h"
#include "src/gpu/ganesh/GrCaps.h"
#include "src/gpu/ganesh/GrDirectContextPriv.h"
#include "tests/Test.h"
#include "tools/Resources.h"
#include "tools/RuntimeBlendUtils.h"
#include "tools/ToolUtils.h"
static bool nearly_equal(const SkColor& x, const SkColor& y) {
const int kTolerance = 1;
return abs((int)SkColorGetA(x) - (int)SkColorGetA(y)) <= kTolerance &&
abs((int)SkColorGetR(x) - (int)SkColorGetR(y)) <= kTolerance &&
abs((int)SkColorGetG(x) - (int)SkColorGetG(y)) <= kTolerance &&
abs((int)SkColorGetB(x) - (int)SkColorGetB(y)) <= kTolerance;
}
static void test_blend(skiatest::Reporter* r, SkSurface* surface) {
SkBitmap bitmap;
REPORTER_ASSERT(r, bitmap.tryAllocPixels(surface->imageInfo()));
for (int m = 0; m < kSkBlendModeCount; ++m) {
SkBlendMode mode = (SkBlendMode)m;
for (int alpha : {0x80, 0xFF}) {
std::vector<SkColor> colors;
for (bool useRuntimeBlend : {false, true}) {
// Draw a solid red pixel.
SkPaint paint;
paint.setColor(SK_ColorRED);
paint.setBlendMode(SkBlendMode::kSrc);
surface->getCanvas()->drawRect(SkRect::MakeWH(1, 1), paint);
// Draw a blue pixel on top of it, using the passed-in blend mode.
paint.setColor(SkColorSetARGB(alpha, 0x00, 0x00, 0xFF));
if (useRuntimeBlend) {
paint.setBlender(GetRuntimeBlendForBlendMode(mode));
} else {
paint.setBlendMode(mode);
}
surface->getCanvas()->drawRect(SkRect::MakeWH(1, 1), paint);
// Read back the red/blue blended pixel.
REPORTER_ASSERT(r, surface->readPixels(bitmap.info(), bitmap.getPixels(),
bitmap.rowBytes(), /*srcX=*/0, /*srcY=*/0));
colors.push_back(bitmap.getColor(/*x=*/0, /*y=*/0));
}
REPORTER_ASSERT(r, nearly_equal(colors[0], colors[1]),
"Expected: %s %s blend matches. Actual: Built-in "
"A=%02X R=%02X G=%02X B=%02X, Runtime A=%02X R=%02X G=%02X B=%02X",
SkBlendMode_Name(mode),
(alpha == 0xFF) ? "solid" : "transparent",
SkColorGetA(colors[0]),
SkColorGetR(colors[0]),
SkColorGetG(colors[0]),
SkColorGetB(colors[0]),
SkColorGetA(colors[1]),
SkColorGetR(colors[1]),
SkColorGetG(colors[1]),
SkColorGetB(colors[1]));
}
}
}
DEF_TEST(SkRuntimeBlender_CPU, r) {
const SkImageInfo info = SkImageInfo::MakeN32Premul(/*width=*/1, /*height=*/1);
sk_sp<SkSurface> surface(SkSurface::MakeRaster(info));
test_blend(r, surface.get());
}
DEF_GPUTEST_FOR_RENDERING_CONTEXTS(SkRuntimeBlender_GPU, r, ctxInfo) {
const SkImageInfo info = SkImageInfo::MakeN32Premul(/*width=*/1, /*height=*/1);
sk_sp<SkSurface> surface(SkSurface::MakeRenderTarget(ctxInfo.directContext(),
SkBudgeted::kNo, info));
test_blend(r, surface.get());
}