| /* |
| * Copyright 2018 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "src/sksl/ir/SkSLVariableReference.h" |
| |
| #include "src/sksl/SkSLIRGenerator.h" |
| #include "src/sksl/ir/SkSLConstructor.h" |
| #include "src/sksl/ir/SkSLFloatLiteral.h" |
| #include "src/sksl/ir/SkSLSetting.h" |
| |
| namespace SkSL { |
| |
| VariableReference::VariableReference(int offset, const Variable* variable, RefKind refKind) |
| : INHERITED(offset, kExpressionKind, &variable->type()) |
| , fVariable(variable) |
| , fRefKind(refKind) { |
| SkASSERT(fVariable); |
| if (refKind != kRead_RefKind) { |
| fVariable->fWriteCount++; |
| } |
| if (refKind != kWrite_RefKind) { |
| fVariable->fReadCount++; |
| } |
| } |
| |
| VariableReference::~VariableReference() { |
| if (fRefKind != kRead_RefKind) { |
| fVariable->fWriteCount--; |
| } |
| if (fRefKind != kWrite_RefKind) { |
| fVariable->fReadCount--; |
| } |
| } |
| |
| void VariableReference::setRefKind(RefKind refKind) { |
| if (fRefKind != kRead_RefKind) { |
| fVariable->fWriteCount--; |
| } |
| if (fRefKind != kWrite_RefKind) { |
| fVariable->fReadCount--; |
| } |
| if (refKind != kRead_RefKind) { |
| fVariable->fWriteCount++; |
| } |
| if (refKind != kWrite_RefKind) { |
| fVariable->fReadCount++; |
| } |
| fRefKind = refKind; |
| } |
| |
| std::unique_ptr<Expression> VariableReference::constantPropagate(const IRGenerator& irGenerator, |
| const DefinitionMap& definitions) { |
| if (fRefKind != kRead_RefKind) { |
| return nullptr; |
| } |
| if ((fVariable->fModifiers.fFlags & Modifiers::kConst_Flag) && fVariable->fInitialValue && |
| fVariable->fInitialValue->isCompileTimeConstant() && |
| this->type().typeKind() != Type::TypeKind::kArray) { |
| return fVariable->fInitialValue->clone(); |
| } |
| auto exprIter = definitions.find(fVariable); |
| if (exprIter != definitions.end() && exprIter->second && |
| (*exprIter->second)->isCompileTimeConstant()) { |
| return (*exprIter->second)->clone(); |
| } |
| return nullptr; |
| } |
| |
| } // namespace SkSL |