skia2/gm/destcolor.cpp
John Stiles 307f8f525a Simplify the boilerplate of cloning a fragment processor.
GrFragmentProcessor now provides an (explicit) copy constructor which
clones all child processors and flags from the passed-in FP. Since we no
longer have flags which propagate up to the root node of the FP tree,
all flags are now safe to copy, since a cloned FP also clones all of its
children.

Change-Id: Ia9f80e0ec540ed1056d25dbb1861a174a1d55f4b
Bug: skia:12299
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/437836
Commit-Queue: John Stiles <johnstiles@google.com>
Commit-Queue: Brian Salomon <bsalomon@google.com>
Auto-Submit: John Stiles <johnstiles@google.com>
Reviewed-by: Brian Salomon <bsalomon@google.com>
2021-08-09 20:36:14 +00:00

100 lines
3.3 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/SkCanvas.h"
#include "include/core/SkImageInfo.h"
#include "include/core/SkMatrix.h"
#include "include/core/SkRect.h"
#include "include/core/SkTypes.h"
#include "include/effects/SkImageFilters.h"
#include "src/core/SkCanvasPriv.h"
#include "src/gpu/GrDirectContextPriv.h"
#include "src/gpu/GrFragmentProcessor.h"
#include "src/gpu/GrStyle.h"
#include "src/gpu/SkGr.h"
#include "src/gpu/glsl/GrGLSLFragmentProcessor.h"
#include "src/gpu/glsl/GrGLSLFragmentShaderBuilder.h"
#include "src/gpu/v1/SurfaceDrawContext_v1.h"
#include "tools/Resources.h"
#include "tools/ToolUtils.h"
namespace {
class DestColorTestFP : public GrFragmentProcessor {
public:
static std::unique_ptr<GrFragmentProcessor> Make(std::unique_ptr<GrFragmentProcessor> child) {
return std::unique_ptr<GrFragmentProcessor>(new DestColorTestFP(std::move(child)));
}
std::unique_ptr<GrFragmentProcessor> clone() const override {
return std::unique_ptr<GrFragmentProcessor>(new DestColorTestFP(*this));
}
private:
DestColorTestFP(std::unique_ptr<GrFragmentProcessor> child)
: INHERITED(kTestFP_ClassID, kNone_OptimizationFlags) {
this->registerChild(std::move(child));
}
explicit DestColorTestFP(const DestColorTestFP& that)
: INHERITED(that) {}
const char* name() const override { return "DestColorTestFP"; }
void onAddToKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override {}
bool onIsEqual(const GrFragmentProcessor&) const override { return true; }
class Impl : public ProgramImpl {
void emitCode(EmitArgs& args) override {
SkString result = this->invokeChild(0, args);
args.fFragBuilder->codeAppendf("return (half4(1) - (%s)).rgb1;", result.c_str());
}
void onSetData(const GrGLSLProgramDataManager& pdman,
const GrFragmentProcessor& processor) override {
}
};
std::unique_ptr<ProgramImpl> onMakeProgramImpl() const override {
return std::make_unique<Impl>();
}
using INHERITED = GrFragmentProcessor;
};
} // namespace
namespace skiagm {
DEF_SIMPLE_GPU_GM_CAN_FAIL(destcolor, rContext, canvas, errorMsg, 640, 640) {
SkRect bounds = SkRect::MakeIWH(512, 512);
auto sdc = SkCanvasPriv::TopDeviceSurfaceDrawContext(canvas);
if (!sdc) {
*errorMsg = GM::kErrorMsg_DrawSkippedGpuOnly;
return DrawResult::kSkip;
}
// Draw the mandrill.
SkPaint p;
p.setImageFilter(SkImageFilters::Image(GetResourceAsImage("images/mandrill_512.png"),
bounds, bounds, SkSamplingOptions()));
canvas->drawPaint(p);
// Now let's add our test FP on top. It reads back the original image and inverts it.
GrPaint invertPaint;
invertPaint.setColor4f(SK_PMColor4fWHITE);
invertPaint.setPorterDuffXPFactory(SkBlendMode::kSrcOver);
invertPaint.setColorFragmentProcessor(
DestColorTestFP::Make(GrFragmentProcessor::SurfaceColor()));
sdc->drawOval(/*clip*/ nullptr, std::move(invertPaint), GrAA::kYes, SkMatrix::I(),
SkRect::MakeLTRB(128, 128, 640, 640), GrStyle::SimpleFill());
return DrawResult::kOk;
}
} // namespace skiagm