123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- /*
- * Copyright 2017 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #ifndef SKSL_CPPCODEGENERATOR
- #define SKSL_CPPCODEGENERATOR
- #include "src/sksl/SkSLGLSLCodeGenerator.h"
- #include "src/sksl/SkSLSectionAndParameterHelper.h"
- #include <set>
- namespace SkSL {
- class CPPCodeGenerator : public GLSLCodeGenerator {
- public:
- CPPCodeGenerator(const Context* context, const Program* program, ErrorReporter* errors,
- String name, OutputStream* out);
- bool generateCode() override;
- private:
- void writef(const char* s, va_list va) SKSL_PRINTF_LIKE(2, 0);
- void writef(const char* s, ...) SKSL_PRINTF_LIKE(2, 3);
- bool writeSection(const char* name, const char* prefix = "");
- void writeHeader() override;
- bool usesPrecisionModifiers() const override;
- String getTypeName(const Type& type) override;
- void writeBinaryExpression(const BinaryExpression& b, Precedence parentPrecedence) override;
- void writeIndexExpression(const IndexExpression& i) override;
- void writeIntLiteral(const IntLiteral& i) override;
- void writeSwizzle(const Swizzle& swizzle) override;
- void writeFieldAccess(const FieldAccess& access) override;
- void writeVariableReference(const VariableReference& ref) override;
- String getSamplerHandle(const Variable& var);
- void writeIfStatement(const IfStatement& s) override;
- void writeReturnStatement(const ReturnStatement& s) override;
- void writeSwitchStatement(const SwitchStatement& s) override;
- void writeFunctionCall(const FunctionCall& c) override;
- void writeFunction(const FunctionDefinition& f) override;
- void writeSetting(const Setting& s) override;
- void writeProgramElement(const ProgramElement& p) override;
- void addUniform(const Variable& var);
- // writes a printf escape that will be filled in at runtime by the given C++ expression string
- void writeRuntimeValue(const Type& type, const Layout& layout, const String& cppCode);
- void writeVarInitializer(const Variable& var, const Expression& value) override;
- void writeInputVars() override;
- void writePrivateVars();
- void writePrivateVarValues();
- void writeCodeAppend(const String& code);
- bool writeEmitCode(std::vector<const Variable*>& uniforms);
- void writeSetData(std::vector<const Variable*>& uniforms);
- void writeGetKey();
- void writeOnTextureSampler();
- void writeClone();
- void writeTest();
- // If the returned C++ is included in the generated code, then the variable name stored in
- // cppVar will refer to a valid SkString that matches the Expression. Successful returns leave
- // the output buffer (and related state) unmodified.
- //
- // In the simplest cases, this will return "SkString {cppVar}(\"{e}\");", while more advanced
- // cases will properly insert format arguments.
- String convertSKSLExpressionToCPP(const Expression& e, const String& cppVar);
- // Process accumulated sksl to split it into appended code sections, properly interleaved with
- // the extra emit code blocks, based on statement/block locations and the inserted tokens
- // from newExtraEmitCodeBlock(). It is necessary to split the sksl after the program has been
- // fully walked since many elements redirect fOut to simultaneously build header sections and
- // bodies that are then concatenated; due to this it is not possible to split the sksl emission
- // on the fly.
- void flushEmittedCode();
- // Start a new extra emit code block for accumulating C++ code. This will insert a token into
- // the sksl stream to mark the fence between previous complete sksl statements and where the
- // C++ code added to the new block will be added to emitCode(). These tokens are removed by
- // flushEmittedCode() as it consumes them before passing pure sksl to writeCodeAppend().
- void newExtraEmitCodeBlock();
- // Append CPP code to the current extra emit code block.
- void addExtraEmitCodeLine(const String& toAppend);
- int getChildFPIndex(const Variable& var) const;
- String fName;
- String fFullName;
- SectionAndParameterHelper fSectionAndParameterHelper;
- std::vector<String> fExtraEmitCodeBlocks;
- std::vector<String> fFormatArgs;
- std::set<int> fWrittenTransformedCoords;
- // if true, we are writing a C++ expression instead of a GLSL expression
- bool fCPPMode = false;
- bool fInMain = false;
- // if not null, we are accumulating SkSL for emitCode into fOut, which
- // replaced the original buffer with a StringStream. The original buffer is
- // stored here for restoration.
- OutputStream* fCPPBuffer = nullptr;
- typedef GLSLCodeGenerator INHERITED;
- };
- }
- #endif
|