/* * 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 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& uniforms); void writeSetData(std::vector& 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 fExtraEmitCodeBlocks; std::vector fFormatArgs; std::set 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