blob: 2ff8c41780a7c16b429b21c2323e38ff9ba3339e [file] [log] [blame]
/*
* Copyright 2019 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "src/gpu/ganesh/GrDirectContextPriv.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkTypes.h"
#include "include/gpu/GrContextThreadSafeProxy.h"
#include "include/gpu/GrDirectContext.h"
#include "src/core/SkRuntimeEffectPriv.h"
#include "src/gpu/ganesh/GrContextThreadSafeProxyPriv.h"
#include "src/gpu/ganesh/GrDrawingManager.h"
#include "src/gpu/ganesh/GrGpu.h"
#include "src/gpu/ganesh/GrMemoryPool.h"
#include "src/gpu/ganesh/GrRecordingContextPriv.h"
#include "src/gpu/ganesh/GrTexture.h"
#include "src/gpu/ganesh/GrThreadSafePipelineBuilder.h"
#include "src/gpu/ganesh/GrTracing.h"
#include "src/gpu/ganesh/SkGr.h"
#include "src/gpu/ganesh/SurfaceContext.h"
#include "src/gpu/ganesh/SurfaceFillContext.h"
#include "src/gpu/ganesh/effects/GrSkSLFP.h"
#include "src/gpu/ganesh/effects/GrTextureEffect.h"
#include "src/gpu/ganesh/text/GrAtlasManager.h"
#include "src/image/SkImage_Base.h"
#include "src/image/SkImage_Gpu.h"
#include "src/text/gpu/TextBlobRedrawCoordinator.h"
using namespace skia_private;
using MaskFormat = skgpu::MaskFormat;
#define ASSERT_OWNED_PROXY(P) \
SkASSERT(!(P) || !((P)->peekTexture()) || (P)->peekTexture()->getContext() == this->context())
#define ASSERT_SINGLE_OWNER SKGPU_ASSERT_SINGLE_OWNER(this->context()->singleOwner())
#define RETURN_VALUE_IF_ABANDONED(value) if (this->context()->abandoned()) { return (value); }
GrSemaphoresSubmitted GrDirectContextPriv::flushSurfaces(
SkSpan<GrSurfaceProxy*> proxies,
SkSurface::BackendSurfaceAccess access,
const GrFlushInfo& info,
const skgpu::MutableTextureState* newState) {
ASSERT_SINGLE_OWNER
GR_CREATE_TRACE_MARKER_CONTEXT("GrDirectContextPriv", "flushSurfaces", this->context());
if (this->context()->abandoned()) {
if (info.fSubmittedProc) {
info.fSubmittedProc(info.fSubmittedContext, false);
}
if (info.fFinishedProc) {
info.fFinishedProc(info.fFinishedContext);
}
return GrSemaphoresSubmitted::kNo;
}
#ifdef SK_DEBUG
for (GrSurfaceProxy* proxy : proxies) {
SkASSERT(proxy);
ASSERT_OWNED_PROXY(proxy);
}
#endif
return this->context()->drawingManager()->flushSurfaces(proxies, access, info, newState);
}
void GrDirectContextPriv::createDDLTask(sk_sp<const SkDeferredDisplayList> ddl,
sk_sp<GrRenderTargetProxy> newDest,
SkIPoint offset) {
this->context()->drawingManager()->createDDLTask(std::move(ddl), std::move(newDest), offset);
}
bool GrDirectContextPriv::compile(const GrProgramDesc& desc, const GrProgramInfo& info) {
GrGpu* gpu = this->getGpu();
if (!gpu) {
return false;
}
return gpu->compile(desc, info);
}
//////////////////////////////////////////////////////////////////////////////
#if GR_TEST_UTILS
void GrDirectContextPriv::dumpCacheStats(SkString* out) const {
#if GR_CACHE_STATS
this->context()->fResourceCache->dumpStats(out);
#endif
}
void GrDirectContextPriv::dumpCacheStatsKeyValuePairs(SkTArray<SkString>* keys,
SkTArray<double>* values) const {
#if GR_CACHE_STATS
this->context()->fResourceCache->dumpStatsKeyValuePairs(keys, values);
#endif
}
void GrDirectContextPriv::printCacheStats() const {
SkString out;
this->dumpCacheStats(&out);
SkDebugf("%s", out.c_str());
}
/////////////////////////////////////////////////
void GrDirectContextPriv::resetGpuStats() const {
#if GR_GPU_STATS
this->context()->fGpu->stats()->reset();
#endif
}
void GrDirectContextPriv::dumpGpuStats(SkString* out) const {
#if GR_GPU_STATS
this->context()->fGpu->stats()->dump(out);
if (auto builder = this->context()->fGpu->pipelineBuilder()) {
builder->stats()->dump(out);
}
#endif
}
void GrDirectContextPriv::dumpGpuStatsKeyValuePairs(SkTArray<SkString>* keys,
SkTArray<double>* values) const {
#if GR_GPU_STATS
this->context()->fGpu->stats()->dumpKeyValuePairs(keys, values);
if (auto builder = this->context()->fGpu->pipelineBuilder()) {
builder->stats()->dumpKeyValuePairs(keys, values);
}
#endif
}
void GrDirectContextPriv::printGpuStats() const {
SkString out;
this->dumpGpuStats(&out);
SkDebugf("%s", out.c_str());
}
/////////////////////////////////////////////////
void GrDirectContextPriv::resetContextStats() {
#if GR_GPU_STATS
this->context()->stats()->reset();
#endif
}
void GrDirectContextPriv::dumpContextStats(SkString* out) const {
#if GR_GPU_STATS
this->context()->stats()->dump(out);
#endif
}
void GrDirectContextPriv::dumpContextStatsKeyValuePairs(SkTArray<SkString>* keys,
SkTArray<double>* values) const {
#if GR_GPU_STATS
this->context()->stats()->dumpKeyValuePairs(keys, values);
#endif
}
void GrDirectContextPriv::printContextStats() const {
SkString out;
this->dumpContextStats(&out);
SkDebugf("%s", out.c_str());
}
/////////////////////////////////////////////////
sk_sp<SkImage> GrDirectContextPriv::testingOnly_getFontAtlasImage(MaskFormat format,
unsigned int index) {
auto atlasManager = this->getAtlasManager();
if (!atlasManager) {
return nullptr;
}
unsigned int numActiveProxies;
const GrSurfaceProxyView* views = atlasManager->getViews(format, &numActiveProxies);
if (index >= numActiveProxies || !views || !views[index].proxy()) {
return nullptr;
}
SkColorType colorType = skgpu::MaskFormatToColorType(format);
SkASSERT(views[index].proxy()->priv().isExact());
return sk_make_sp<SkImage_Gpu>(sk_ref_sp(this->context()),
kNeedNewImageUniqueID,
views[index],
SkColorInfo(colorType, kPremul_SkAlphaType, nullptr));
}
void GrDirectContextPriv::testingOnly_flushAndRemoveOnFlushCallbackObject(
GrOnFlushCallbackObject* cb) {
this->context()->flushAndSubmit();
this->context()->drawingManager()->testingOnly_removeOnFlushCallbackObject(cb);
}
#endif
// Both of these effects aggressively round to the nearest exact (N / 255) floating point values.
// This lets us find a round-trip preserving pair on some GPUs that do odd byte to float conversion.
static std::unique_ptr<GrFragmentProcessor> make_premul_effect(
std::unique_ptr<GrFragmentProcessor> fp) {
if (!fp) {
return nullptr;
}
static const SkRuntimeEffect* effect = SkMakeRuntimeEffect(SkRuntimeEffect::MakeForColorFilter,
"half4 main(half4 halfColor) {"
"float4 color = float4(halfColor);"
"color = floor(color * 255 + 0.5) / 255;"
"color.rgb = floor(color.rgb * color.a * 255 + 0.5) / 255;"
"return color;"
"}"
);
fp = GrSkSLFP::Make(effect, "ToPremul", std::move(fp), GrSkSLFP::OptFlags::kNone);
return GrFragmentProcessor::HighPrecision(std::move(fp));
}
static std::unique_ptr<GrFragmentProcessor> make_unpremul_effect(
std::unique_ptr<GrFragmentProcessor> fp) {
if (!fp) {
return nullptr;
}
static const SkRuntimeEffect* effect = SkMakeRuntimeEffect(SkRuntimeEffect::MakeForColorFilter,
"half4 main(half4 halfColor) {"
"float4 color = float4(halfColor);"
"color = floor(color * 255 + 0.5) / 255;"
"color.rgb = color.a <= 0 ? half3(0) : floor(color.rgb / color.a * 255 + 0.5) / 255;"
"return color;"
"}"
);
fp = GrSkSLFP::Make(effect, "ToUnpremul", std::move(fp), GrSkSLFP::OptFlags::kNone);
return GrFragmentProcessor::HighPrecision(std::move(fp));
}
static bool test_for_preserving_PM_conversions(GrDirectContext* dContext) {
static constexpr int kSize = 256;
AutoTMalloc<uint32_t> data(kSize * kSize * 3);
uint32_t* srcData = data.get();
// Fill with every possible premultiplied A, color channel value. There will be 256-y duplicate
// values in row y. We set r, g, and b to the same value since they are handled identically.
for (int y = 0; y < kSize; ++y) {
for (int x = 0; x < kSize; ++x) {
uint8_t* color = reinterpret_cast<uint8_t*>(&srcData[kSize*y + x]);
color[3] = y;
color[2] = std::min(x, y);
color[1] = std::min(x, y);
color[0] = std::min(x, y);
}
}
const SkImageInfo pmII =
SkImageInfo::Make(kSize, kSize, kRGBA_8888_SkColorType, kPremul_SkAlphaType);
const SkImageInfo upmII = pmII.makeAlphaType(kUnpremul_SkAlphaType);
auto readSFC =
dContext->priv().makeSFC(upmII, "ReadSfcForPMUPMConversion", SkBackingFit::kExact);
auto tempSFC =
dContext->priv().makeSFC(pmII, "TempSfcForPMUPMConversion", SkBackingFit::kExact);
if (!readSFC || !tempSFC) {
return false;
}
// This function is only ever called if we are in a GrDirectContext since we are calling read
// pixels here. Thus the pixel data will be uploaded immediately and we don't need to keep the
// pixel data alive in the proxy. Therefore the ReleaseProc is nullptr.
SkBitmap bitmap;
bitmap.installPixels(pmII, srcData, 4 * kSize);
bitmap.setImmutable();
auto dataView = std::get<0>(GrMakeUncachedBitmapProxyView(dContext, bitmap));
if (!dataView) {
return false;
}
uint32_t* firstRead = data.get() + kSize*kSize;
uint32_t* secondRead = data.get() + 2*kSize*kSize;
std::fill_n( firstRead, kSize*kSize, 0);
std::fill_n(secondRead, kSize*kSize, 0);
GrPixmap firstReadPM( upmII, firstRead, kSize*sizeof(uint32_t));
GrPixmap secondReadPM(upmII, secondRead, kSize*sizeof(uint32_t));
// We do a PM->UPM draw from dataTex to readTex and read the data. Then we do a UPM->PM draw
// from readTex to tempTex followed by a PM->UPM draw to readTex and finally read the data.
// We then verify that two reads produced the same values.
auto fp1 = make_unpremul_effect(GrTextureEffect::Make(std::move(dataView), bitmap.alphaType()));
readSFC->fillRectWithFP(SkIRect::MakeWH(kSize, kSize), std::move(fp1));
if (!readSFC->readPixels(dContext, firstReadPM, {0, 0})) {
return false;
}
auto fp2 = make_premul_effect(
GrTextureEffect::Make(readSFC->readSurfaceView(), readSFC->colorInfo().alphaType()));
tempSFC->fillRectWithFP(SkIRect::MakeWH(kSize, kSize), std::move(fp2));
auto fp3 = make_unpremul_effect(
GrTextureEffect::Make(tempSFC->readSurfaceView(), tempSFC->colorInfo().alphaType()));
readSFC->fillRectWithFP(SkIRect::MakeWH(kSize, kSize), std::move(fp3));
if (!readSFC->readPixels(dContext, secondReadPM, {0, 0})) {
return false;
}
for (int y = 0; y < kSize; ++y) {
for (int x = 0; x <= y; ++x) {
if (firstRead[kSize*y + x] != secondRead[kSize*y + x]) {
return false;
}
}
}
return true;
}
bool GrDirectContextPriv::validPMUPMConversionExists() {
ASSERT_SINGLE_OWNER
auto dContext = this->context();
if (!dContext->fDidTestPMConversions) {
dContext->fPMUPMConversionsRoundTrip = test_for_preserving_PM_conversions(dContext);
dContext->fDidTestPMConversions = true;
}
// The PM<->UPM tests fail or succeed together so we only need to check one.
return dContext->fPMUPMConversionsRoundTrip;
}
std::unique_ptr<GrFragmentProcessor> GrDirectContextPriv::createPMToUPMEffect(
std::unique_ptr<GrFragmentProcessor> fp) {
ASSERT_SINGLE_OWNER
// We should have already called this->priv().validPMUPMConversionExists() in this case
SkASSERT(this->context()->fDidTestPMConversions);
// ...and it should have succeeded
SkASSERT(this->validPMUPMConversionExists());
return make_unpremul_effect(std::move(fp));
}
std::unique_ptr<GrFragmentProcessor> GrDirectContextPriv::createUPMToPMEffect(
std::unique_ptr<GrFragmentProcessor> fp) {
ASSERT_SINGLE_OWNER
// We should have already called this->priv().validPMUPMConversionExists() in this case
SkASSERT(this->context()->fDidTestPMConversions);
// ...and it should have succeeded
SkASSERT(this->validPMUPMConversionExists());
return make_premul_effect(std::move(fp));
}