123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168 |
- /*
- * Copyright 2016 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #include "include/core/SkBitmap.h"
- #include "include/core/SkBlendMode.h"
- #include "include/core/SkCanvas.h"
- #include "include/core/SkColor.h"
- #include "include/core/SkColorFilter.h"
- #include "include/core/SkColorPriv.h"
- #include "include/core/SkImageInfo.h"
- #include "include/core/SkPaint.h"
- #include "include/core/SkRefCnt.h"
- #include "include/core/SkScalar.h"
- #include "include/core/SkSurface.h"
- #include "include/core/SkTypes.h"
- #include "include/gpu/GrContext.h"
- #include "include/private/GrTypesPriv.h"
- #include "include/private/SkTemplates.h"
- #include "src/core/SkUtils.h"
- #include "src/gpu/GrCaps.h"
- #include "src/gpu/GrContextPriv.h"
- #include "src/gpu/GrShaderCaps.h"
- #include "tests/Test.h"
- #include "tools/gpu/GrContextFactory.h"
- #include <math.h>
- #include <initializer_list>
- /** convert 0..1 linear value to 0..1 srgb */
- static float linear_to_srgb(float linear) {
- if (linear <= 0.0031308) {
- return linear * 12.92f;
- } else {
- return 1.055f * powf(linear, 1.f / 2.4f) - 0.055f;
- }
- }
- /** convert 0..1 srgb value to 0..1 linear */
- static float srgb_to_linear(float srgb) {
- if (srgb <= 0.04045f) {
- return srgb / 12.92f;
- } else {
- return powf((srgb + 0.055f) / 1.055f, 2.4f);
- }
- }
- bool check_gamma(uint32_t src, uint32_t dst, bool toSRGB, float error,
- uint32_t* expected) {
- bool result = true;
- uint32_t expectedColor = src & 0xff000000;
- // Alpha should always be exactly preserved.
- if ((src & 0xff000000) != (dst & 0xff000000)) {
- result = false;
- }
- // need to unpremul before we can perform srgb magic
- float invScale = 0;
- float alpha = SkGetPackedA32(src);
- if (alpha) {
- invScale = 255.0f / alpha;
- }
- for (int c = 0; c < 3; ++c) {
- float srcComponent = ((src & (0xff << (c * 8))) >> (c * 8)) * invScale;
- float lower = SkTMax(0.f, srcComponent - error);
- float upper = SkTMin(255.f, srcComponent + error);
- if (toSRGB) {
- lower = linear_to_srgb(lower / 255.f);
- upper = linear_to_srgb(upper / 255.f);
- } else {
- lower = srgb_to_linear(lower / 255.f);
- upper = srgb_to_linear(upper / 255.f);
- }
- lower *= alpha;
- upper *= alpha;
- SkASSERT(lower >= 0.f && lower <= 255.f);
- SkASSERT(upper >= 0.f && upper <= 255.f);
- uint8_t dstComponent = (dst & (0xff << (c * 8))) >> (c * 8);
- if (dstComponent < SkScalarFloorToInt(lower) ||
- dstComponent > SkScalarCeilToInt(upper)) {
- result = false;
- }
- uint8_t expectedComponent = SkScalarRoundToInt((lower + upper) * 0.5f);
- expectedColor |= expectedComponent << (c * 8);
- }
- *expected = expectedColor;
- return result;
- }
- DEF_GPUTEST_FOR_RENDERING_CONTEXTS(ApplyGamma, reporter, ctxInfo) {
- GrContext* context = ctxInfo.grContext();
- static const int kW = 256;
- static const int kH = 256;
- static const size_t kRowBytes = sizeof(uint32_t) * kW;
- GrSurfaceDesc baseDesc;
- baseDesc.fConfig = kRGBA_8888_GrPixelConfig;
- baseDesc.fWidth = kW;
- baseDesc.fHeight = kH;
- const SkImageInfo ii = SkImageInfo::MakeN32Premul(kW, kH);
- SkAutoTMalloc<uint32_t> srcPixels(kW * kH);
- for (int y = 0; y < kH; ++y) {
- for (int x = 0; x < kW; ++x) {
- srcPixels.get()[y*kW+x] = SkPreMultiplyARGB(x, y, x, 0xFF);
- }
- }
- SkBitmap bm;
- bm.installPixels(ii, srcPixels.get(), kRowBytes);
- SkAutoTMalloc<uint32_t> read(kW * kH);
- // We allow more error on GPUs with lower precision shader variables.
- float error = context->priv().caps()->shaderCaps()->halfIs32Bits() ? 0.5f : 1.2f;
- for (auto toSRGB : { false, true }) {
- sk_sp<SkSurface> dst(SkSurface::MakeRenderTarget(context, SkBudgeted::kNo, ii));
- if (!dst) {
- ERRORF(reporter, "Could not create surfaces for copy surface test.");
- continue;
- }
- SkCanvas* dstCanvas = dst->getCanvas();
- dstCanvas->clear(SK_ColorRED);
- dst->flush();
- SkPaint gammaPaint;
- gammaPaint.setBlendMode(SkBlendMode::kSrc);
- gammaPaint.setColorFilter(toSRGB ? SkColorFilters::LinearToSRGBGamma()
- : SkColorFilters::SRGBToLinearGamma());
- dstCanvas->drawBitmap(bm, 0, 0, &gammaPaint);
- dst->flush();
- sk_memset32(read.get(), 0, kW * kH);
- if (!dst->readPixels(ii, read.get(), kRowBytes, 0, 0)) {
- ERRORF(reporter, "Error calling readPixels");
- continue;
- }
- bool abort = false;
- // Validate that pixels were copied/transformed correctly.
- for (int y = 0; y < kH && !abort; ++y) {
- for (int x = 0; x < kW && !abort; ++x) {
- uint32_t r = read.get()[y * kW + x];
- uint32_t s = srcPixels.get()[y * kW + x];
- uint32_t expected;
- if (!check_gamma(s, r, toSRGB, error, &expected)) {
- ERRORF(reporter, "Expected dst %d,%d to contain 0x%08x "
- "from src 0x%08x and mode %s. Got %08x", x, y, expected, s,
- toSRGB ? "ToSRGB" : "ToLinear", r);
- abort = true;
- break;
- }
- }
- }
- }
- }
|