Revert "added support for SkSL unpremul function"

This reverts commit ffdc3e6680.

Reason for revert: breaking Chrome roll

Original change's description:
> added support for SkSL unpremul function
> 
> Change-Id: I970f1ad0dd0859448c874498fe02342f8abc3aa3
> Reviewed-on: https://skia-review.googlesource.com/c/skia/+/242897
> Reviewed-by: Brian Salomon <bsalomon@google.com>
> Commit-Queue: Ethan Nicholas <ethannicholas@google.com>

TBR=bsalomon@google.com,brianosman@google.com,ethannicholas@google.com

Change-Id: I72dacea82b1f4ea2aa6cd600122a4810657aa32e
No-Presubmit: true
No-Tree-Checks: true
No-Try: true
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/243117
Reviewed-by: Ethan Nicholas <ethannicholas@google.com>
Commit-Queue: Ethan Nicholas <ethannicholas@google.com>
This commit is contained in:
Ethan Nicholas 2019-09-20 12:59:22 -04:00 committed by Skia Commit-Bot
parent 6934d75b55
commit 5a9a9b8d5b
14 changed files with 62 additions and 132 deletions

View File

@ -245,7 +245,11 @@ void GLHighContrastFilterEffect::emitCode(EmitArgs& args) {
GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
fragBuilder->codeAppendf("half4 color = unpremul(%s);", args.fInputColor);
fragBuilder->codeAppendf("half4 color = %s;", args.fInputColor);
// Unpremultiply. The max() is to guard against 0 / 0.
fragBuilder->codeAppendf("half nonZeroAlpha = max(color.a, 0.0001);");
fragBuilder->codeAppendf("color = half4(color.rgb / nonZeroAlpha, nonZeroAlpha);");
if (hcfe.linearize()) {
fragBuilder->codeAppend("color.rgb = color.rgb * color.rgb;");

View File

@ -301,7 +301,8 @@ void GLColorTableEffect::emitCode(EmitArgs& args) {
kMaxValue, kMaxValue, kMaxValue, kMaxValue);
} else {
fragBuilder->codeAppendf("\t\thalf4 coord = unpremul(%s);\n",
fragBuilder->codeAppendf("\t\thalf nonZeroAlpha = max(%s.a, .0001);\n", args.fInputColor);
fragBuilder->codeAppendf("\t\thalf4 coord = half4(%s.rgb / nonZeroAlpha, nonZeroAlpha);\n",
args.fInputColor);
fragBuilder->codeAppendf("\t\tcoord = coord * %f + half4(%f, %f, %f, %f);\n",
kColorScaleFactor,

View File

@ -18,7 +18,10 @@ layout(key) in bool premulOutput;
void main() {
half4 inputColor = sk_InColor;
@if (unpremulInput) {
inputColor = unpremul(inputColor);
// The max() is to guard against 0 / 0 during unpremul when the incoming color is
// transparent black.
half nonZeroAlpha = max(inputColor.a, 0.0001);
inputColor = half4(inputColor.rgb / nonZeroAlpha, nonZeroAlpha);
}
sk_OutColor = m * inputColor + v;
@if (clampRGBOutput) {

View File

@ -47,7 +47,8 @@ public:
// calculations to be performed with sufficient precision.
fragBuilder->codeAppendf("float4 color = %s;", args.fInputColor);
if (srgbe.alpha() == GrSRGBEffect::Alpha::kPremul) {
fragBuilder->codeAppendf("color = unpremul(color);");
fragBuilder->codeAppendf("float nonZeroAlpha = max(color.a, 0.0001);");
fragBuilder->codeAppendf("color = float4(color.rgb / nonZeroAlpha, color.a);");
}
fragBuilder->codeAppendf("color = float4(%s(half(color.r)), %s(half(color.g)), "
"%s(half(color.b)), color.a);",

View File

@ -37,12 +37,11 @@ public:
mVar = args.fUniformHandler->addUniform(kFragment_GrShaderFlag, kHalf4x4_GrSLType, "m");
vVar = args.fUniformHandler->addUniform(kFragment_GrShaderFlag, kHalf4_GrSLType, "v");
fragBuilder->codeAppendf(
" half4 unpremul0; half unpremulNonZeroAlpha1;half4 inputColor = %s;\n@if "
"(%s) {\n inputColor = (unpremul0 = inputColor, unpremulNonZeroAlpha1 = "
"max(unpremul0.a, 9.9999997473787516e-05), half4(unpremul0.rgb / "
"unpremulNonZeroAlpha1, unpremulNonZeroAlpha1));\n}\n%s = %s * inputColor + "
"%s;\n@if (%s) {\n %s = clamp(%s, 0.0, 1.0);\n} else {\n %s.w = clamp(%s.w, "
"0.0, 1.0);\n}\n@if (%s) {\n %s.xyz *= %s.w;\n}\n",
"half4 inputColor = %s;\n@if (%s) {\n half nonZeroAlpha = max(inputColor.w, "
"9.9999997473787516e-05);\n inputColor = half4(inputColor.xyz / nonZeroAlpha, "
"nonZeroAlpha);\n}\n%s = %s * inputColor + %s;\n@if (%s) {\n %s = clamp(%s, "
"0.0, 1.0);\n} else {\n %s.w = clamp(%s.w, 0.0, 1.0);\n}\n@if (%s) {\n "
"%s.xyz *= %s.w;\n}\n",
args.fInputColor, (_outer.unpremulInput ? "true" : "false"), args.fOutputColor,
args.fUniformHandler->getUniformCStr(mVar),
args.fUniformHandler->getUniformCStr(vVar),

View File

@ -159,7 +159,8 @@ void GrGLSLShaderBuilder::appendColorGamutXform(SkString* out,
const GrShaderVar gColorXformArgs[] = { GrShaderVar("color", kHalf4_GrSLType) };
SkString body;
if (colorXformHelper->applyUnpremul()) {
body.append("color = unpremul(color);");
body.append("half nonZeroAlpha = max(color.a, 0.0001);");
body.append("color = half4(color.rgb / nonZeroAlpha, nonZeroAlpha);");
}
if (colorXformHelper->applySrcTF()) {
body.appendf("color.r = %s(color.r);", srcTFFuncName.c_str());

View File

@ -472,7 +472,6 @@ void GLSLCodeGenerator::writeFunctionCall(const FunctionCall& c) {
(*fFunctionClasses)["saturate"] = FunctionClass::kSaturate;
(*fFunctionClasses)["sample"] = FunctionClass::kTexture;
(*fFunctionClasses)["transpose"] = FunctionClass::kTranspose;
(*fFunctionClasses)["unpremul"] = FunctionClass::kUnpremul;
}
#ifndef SKSL_STANDALONE
);
@ -692,24 +691,6 @@ void GLSLCodeGenerator::writeFunctionCall(const FunctionCall& c) {
return;
}
break;
case FunctionClass::kUnpremul:
String tmpVar1 = "unpremul" + to_string(fVarCount++);
this->fFunctionHeader += String(" ") +
this->getTypePrecision(c.fArguments[0]->fType) +
this->getTypeName(c.fArguments[0]->fType) + " " + tmpVar1 +
";";
String tmpVar2 = "unpremulNonZeroAlpha" + to_string(fVarCount++);
this->fFunctionHeader += String(" ") +
this->getTypePrecision(c.fArguments[0]->fType) + " " +
this->getTypeName(c.fArguments[0]->fType.componentType()) +
" " + tmpVar2 + ";";
this->write("(" + tmpVar1 + " = ");
this->writeExpression(*c.fArguments[0], kSequence_Precedence);
this->write(", " + tmpVar2 + " = max(" + tmpVar1 + ".a, " +
to_string(SKSL_UNPREMUL_MIN) + "), " +
this->getTypeName(*fContext.fHalf4_Type) + "(" + tmpVar1 + ".rgb / " +
tmpVar2 + ", " + tmpVar2 + "))");
return;
}
}
if (!nameWritten) {

View File

@ -254,8 +254,7 @@ protected:
kPow,
kSaturate,
kTexture,
kTranspose,
kUnpremul
kTranspose
};
static std::unordered_map<StringFragment, FunctionClass>* fFunctionClasses;

View File

@ -32,7 +32,6 @@ void MetalCodeGenerator::setupIntrinsics() {
fIntrinsicMap[String("lessThanEqual")] = METAL(LessThanEqual);
fIntrinsicMap[String("greaterThan")] = METAL(GreaterThan);
fIntrinsicMap[String("greaterThanEqual")] = METAL(GreaterThanEqual);
fIntrinsicMap[String("unpremul")] = SPECIAL(Unpremul);
}
void MetalCodeGenerator::write(const char* s) {
@ -70,51 +69,50 @@ void MetalCodeGenerator::writeExtension(const Extension& ext) {
this->writeLine("#extension " + ext.fName + " : enable");
}
String MetalCodeGenerator::getTypeName(const Type& type) {
void MetalCodeGenerator::writeType(const Type& type) {
switch (type.kind()) {
case Type::kStruct_Kind:
return type.name();
for (const Type* search : fWrittenStructs) {
if (*search == type) {
// already written
this->write(type.name());
return;
}
}
fWrittenStructs.push_back(&type);
this->writeLine("struct " + type.name() + " {");
fIndentation++;
this->writeFields(type.fields(), type.fOffset);
fIndentation--;
this->write("}");
break;
case Type::kVector_Kind:
return this->getTypeName(type.componentType()) + to_string(type.columns());
this->writeType(type.componentType());
this->write(to_string(type.columns()));
break;
case Type::kMatrix_Kind:
return this->getTypeName(type.componentType()) + to_string(type.columns()) + "x" +
to_string(type.rows());
this->writeType(type.componentType());
this->write(to_string(type.columns()));
this->write("x");
this->write(to_string(type.rows()));
break;
case Type::kSampler_Kind:
return "texture2d<float>"; // FIXME - support other texture types;
this->write("texture2d<float> "); // FIXME - support other texture types;
break;
default:
if (type == *fContext.fHalf_Type) {
// FIXME - Currently only supporting floats in MSL to avoid type coercion issues.
return fContext.fFloat_Type->name();
this->write(fContext.fFloat_Type->name());
} else if (type == *fContext.fByte_Type) {
return "char";
this->write("char");
} else if (type == *fContext.fUByte_Type) {
return "uchar";
this->write("uchar");
} else {
return type.name();
this->write(type.name());
}
}
}
void MetalCodeGenerator::writeType(const Type& type) {
if (type.kind() == Type::kStruct_Kind) {
for (const Type* search : fWrittenStructs) {
if (*search == type) {
// already written
this->write(this->getTypeName(type));
return;
}
}
fWrittenStructs.push_back(&type);
this->writeLine("struct " + type.name() + " {");
fIndentation++;
this->writeFields(type.fields(), type.fOffset);
fIndentation--;
this->write("}");
} else {
this->write(this->getTypeName(type));
}
}
void MetalCodeGenerator::writeExpression(const Expression& expr, Precedence parentPrecedence) {
switch (expr.fKind) {
case Expression::kBinary_Kind:
@ -371,23 +369,6 @@ void MetalCodeGenerator::writeSpecialIntrinsic(const FunctionCall & c, SpecialIn
this->writeExpression(*c.fArguments[1], kSequence_Precedence);
this->write(")))");
break;
case kUnpremul_SpecialIntrinsic: {
String tmpVar1 = "unpremul" + to_string(fVarCount++);
this->fFunctionHeader += String(" ") +
this->getTypeName(c.fArguments[0]->fType) + " " + tmpVar1 +
";";
String tmpVar2 = "unpremulNonZeroAlpha" + to_string(fVarCount++);
this->fFunctionHeader += String(" ") +
this->getTypeName(c.fArguments[0]->fType.componentType()) +
" " + tmpVar2 + ";";
this->write("(" + tmpVar1 + " = ");
this->writeExpression(*c.fArguments[0], kSequence_Precedence);
this->write(", " + tmpVar2 + " = max(" + tmpVar1 + ".a, " +
to_string(SKSL_UNPREMUL_MIN) + "), " +
this->getTypeName(*fContext.fHalf4_Type) + "(" + tmpVar1 +
".rgb / " + tmpVar2 + ", " + tmpVar2 + "))");
return;
}
default:
ABORT("unsupported special intrinsic kind");
}

View File

@ -78,7 +78,7 @@ public:
};
MetalCodeGenerator(const Context* context, const Program* program, ErrorReporter* errors,
OutputStream* out)
OutputStream* out)
: INHERITED(program, errors, out)
, fReservedWords({"atan2", "rsqrt", "dfdx", "dfdy", "vertex", "fragment"})
, fLineEnding("\n")
@ -105,7 +105,6 @@ protected:
enum SpecialIntrinsic {
kTexture_SpecialIntrinsic,
kMod_SpecialIntrinsic,
kUnpremul_SpecialIntrinsic,
};
enum MetalIntrinsic {
@ -150,8 +149,6 @@ protected:
void writePrecisionModifier();
String getTypeName(const Type& type);
void writeType(const Type& type);
void writeExtension(const Extension& ext);

View File

@ -141,7 +141,6 @@ void SPIRVCodeGenerator::setupIntrinsics() {
SpvOpUndef);
fIntrinsicMap[String("EmitVertex")] = ALL_SPIRV(EmitVertex);
fIntrinsicMap[String("EndPrimitive")] = ALL_SPIRV(EndPrimitive);
fIntrinsicMap[String("unpremul")] = SPECIAL(Unpremul);
// interpolateAt* not yet supported...
}
@ -1003,26 +1002,6 @@ SpvId SPIRVCodeGenerator::writeSpecialIntrinsic(const FunctionCall& c, SpecialIn
GLSLstd450UClamp, spvArgs, out);
break;
}
case kUnpremul_SpecialIntrinsic: {
SpvId color = this->writeExpression(*c.fArguments[0], out);
SpvId a = this->writeSwizzle(*fContext.fHalf_Type, c.fArguments[0]->fType, color, { 3 },
out);
FloatLiteral min(fContext, -1, SKSL_UNPREMUL_MIN);
SpvId minId = this->writeFloatLiteral(min);
SpvId nonZeroAlpha = this->nextId();
this->writeGLSLExtendedInstruction(*fContext.fHalf_Type, nonZeroAlpha, GLSLstd450FMax,
SpvOpUndef, SpvOpUndef, { a, minId }, out);
SpvId rgb = this->writeSwizzle(*fContext.fHalf3_Type, *fContext.fHalf4_Type, color,
{ 0, 1, 2 }, out);
SpvId scaled = this->writeBinaryExpression(*fContext.fHalf3_Type, rgb, Token::SLASH,
*fContext.fFloat_Type, nonZeroAlpha,
*fContext.fHalf3_Type, out);
this->writeOpCode(SpvOpCompositeConstruct, 5, out);
this->writeWord(this->getType(c.fType), out);
this->writeWord(result, out);
this->writeWord(scaled, out);
this->writeWord(nonZeroAlpha, out);
}
}
return result;
}
@ -1933,24 +1912,19 @@ SpvId SPIRVCodeGenerator::writeFieldAccess(const FieldAccess& f, OutputStream& o
}
SpvId SPIRVCodeGenerator::writeSwizzle(const Swizzle& swizzle, OutputStream& out) {
return this->writeSwizzle(swizzle.fType, swizzle.fBase->fType,
this->writeExpression(*swizzle.fBase, out), swizzle.fComponents, out);
}
SpvId SPIRVCodeGenerator::writeSwizzle(const Type& type, const Type& baseType, SpvId base,
const std::vector<int> components, OutputStream& out) {
SpvId base = this->writeExpression(*swizzle.fBase, out);
SpvId result = this->nextId();
size_t count = components.size();
size_t count = swizzle.fComponents.size();
if (count == 1) {
this->writeInstruction(SpvOpCompositeExtract, this->getType(type), result, base,
components[0], out);
this->writeInstruction(SpvOpCompositeExtract, this->getType(swizzle.fType), result, base,
swizzle.fComponents[0], out);
} else {
this->writeOpCode(SpvOpVectorShuffle, 5 + (int32_t) count, out);
this->writeWord(this->getType(type), out);
this->writeWord(this->getType(swizzle.fType), out);
this->writeWord(result, out);
this->writeWord(base, out);
SpvId other = base;
for (int c : components) {
for (int c : swizzle.fComponents) {
if (c < 0) {
if (!fConstantZeroOneVector) {
FloatLiteral zero(fContext, -1, 0);
@ -1970,11 +1944,11 @@ SpvId SPIRVCodeGenerator::writeSwizzle(const Type& type, const Type& baseType, S
}
}
this->writeWord(other, out);
for (int component : components) {
for (int component : swizzle.fComponents) {
if (component == SKSL_SWIZZLE_0) {
this->writeWord(baseType.columns(), out);
this->writeWord(swizzle.fBase->fType.columns(), out);
} else if (component == SKSL_SWIZZLE_1) {
this->writeWord(baseType.columns() + 1, out);
this->writeWord(swizzle.fBase->fType.columns() + 1, out);
} else {
this->writeWord(component, out);
}
@ -2094,11 +2068,6 @@ std::unique_ptr<Expression> create_literal_1(const Context& context, const Type&
SpvId SPIRVCodeGenerator::writeBinaryExpression(const Type& leftType, SpvId lhs, Token::Kind op,
const Type& rightType, SpvId rhs,
const Type& resultType, OutputStream& out) {
// it's important to handle comma early, so we don't end up vectorizing the operands
if (op == Token::COMMA) {
return rhs;
}
Type tmp("<invalid>");
// overall type we are operating on: float2, int, uint4...
const Type* operandType;
@ -2291,6 +2260,8 @@ SpvId SPIRVCodeGenerator::writeBinaryExpression(const Type& leftType, SpvId lhs,
case Token::BITWISEXOR:
return this->writeBinaryOperation(resultType, *operandType, lhs, rhs, SpvOpUndef,
SpvOpBitwiseXor, SpvOpBitwiseXor, SpvOpUndef, out);
case Token::COMMA:
return rhs;
default:
SkASSERT(false);
return -1;

View File

@ -137,7 +137,6 @@ private:
kSampledImage_SpecialIntrinsic,
kSubpassLoad_SpecialIntrinsic,
kTexture_SpecialIntrinsic,
kUnpremul_SpecialIntrinsic,
};
enum class Precision {
@ -254,9 +253,6 @@ private:
SpvId writeSwizzle(const Swizzle& swizzle, OutputStream& out);
SpvId writeSwizzle(const Type& type, const Type& basetype, SpvId base,
const std::vector<int> components, OutputStream& out);
/**
* Folds the potentially-vector result of a logical operation down to a single bool. If
* operandType is a vector type, assumes that the intermediate result in id is a bvec of the

View File

@ -27,8 +27,6 @@ class GrShaderCaps;
namespace SkSL {
static constexpr float SKSL_UNPREMUL_MIN = 0.0001f;
class OutputStream;
class StringStream;

View File

@ -102,8 +102,6 @@ $genType smoothstep($genType edge0, $genType edge1, $genType x);
$genType smoothstep(float edge0, float edge1, $genType x);
$genHType smoothstep($genHType edge0, $genHType edge1, $genHType x);
$genHType smoothstep(half edge0, half edge1, $genHType x);
half4 unpremul(half4 c);
float4 unpremul(float4 c);
$genBType isnan($genType x);
$genBType isnan($genDType x);
$genBType isinf($genType x);