123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410 |
- /*
- * Copyright 2016 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #ifndef SKSL_SPIRVCODEGENERATOR
- #define SKSL_SPIRVCODEGENERATOR
- #include <stack>
- #include <tuple>
- #include <unordered_map>
- #include "src/sksl/SkSLCodeGenerator.h"
- #include "src/sksl/SkSLMemoryLayout.h"
- #include "src/sksl/SkSLStringStream.h"
- #include "src/sksl/ir/SkSLBinaryExpression.h"
- #include "src/sksl/ir/SkSLBoolLiteral.h"
- #include "src/sksl/ir/SkSLConstructor.h"
- #include "src/sksl/ir/SkSLDoStatement.h"
- #include "src/sksl/ir/SkSLFieldAccess.h"
- #include "src/sksl/ir/SkSLFloatLiteral.h"
- #include "src/sksl/ir/SkSLForStatement.h"
- #include "src/sksl/ir/SkSLFunctionCall.h"
- #include "src/sksl/ir/SkSLFunctionDeclaration.h"
- #include "src/sksl/ir/SkSLFunctionDefinition.h"
- #include "src/sksl/ir/SkSLIfStatement.h"
- #include "src/sksl/ir/SkSLIndexExpression.h"
- #include "src/sksl/ir/SkSLIntLiteral.h"
- #include "src/sksl/ir/SkSLInterfaceBlock.h"
- #include "src/sksl/ir/SkSLPostfixExpression.h"
- #include "src/sksl/ir/SkSLPrefixExpression.h"
- #include "src/sksl/ir/SkSLProgramElement.h"
- #include "src/sksl/ir/SkSLReturnStatement.h"
- #include "src/sksl/ir/SkSLStatement.h"
- #include "src/sksl/ir/SkSLSwitchStatement.h"
- #include "src/sksl/ir/SkSLSwizzle.h"
- #include "src/sksl/ir/SkSLTernaryExpression.h"
- #include "src/sksl/ir/SkSLVarDeclarations.h"
- #include "src/sksl/ir/SkSLVarDeclarationsStatement.h"
- #include "src/sksl/ir/SkSLVariableReference.h"
- #include "src/sksl/ir/SkSLWhileStatement.h"
- #include "src/sksl/spirv.h"
- union ConstantValue {
- ConstantValue(int64_t i)
- : fInt(i) {}
- ConstantValue(double d)
- : fDouble(d) {}
- bool operator==(const ConstantValue& other) const {
- return fInt == other.fInt;
- }
- int64_t fInt;
- double fDouble;
- };
- enum class ConstantType {
- kInt,
- kUInt,
- kShort,
- kUShort,
- kFloat,
- kDouble,
- kHalf,
- };
- namespace std {
- template <>
- struct hash<std::pair<ConstantValue, ConstantType>> {
- size_t operator()(const std::pair<ConstantValue, ConstantType>& key) const {
- return key.first.fInt ^ (int) key.second;
- }
- };
- }
- namespace SkSL {
- #define kLast_Capability SpvCapabilityMultiViewport
- /**
- * Converts a Program into a SPIR-V binary.
- */
- class SPIRVCodeGenerator : public CodeGenerator {
- public:
- class LValue {
- public:
- virtual ~LValue() {}
- // returns a pointer to the lvalue, if possible. If the lvalue cannot be directly referenced
- // by a pointer (e.g. vector swizzles), returns 0.
- virtual SpvId getPointer() = 0;
- virtual SpvId load(OutputStream& out) = 0;
- virtual void store(SpvId value, OutputStream& out) = 0;
- };
- SPIRVCodeGenerator(const Context* context, const Program* program, ErrorReporter* errors,
- OutputStream* out)
- : INHERITED(program, errors, out)
- , fContext(*context)
- , fDefaultLayout(MemoryLayout::k140_Standard)
- , fCapabilities(0)
- , fIdCount(1)
- , fBoolTrue(0)
- , fBoolFalse(0)
- , fSetupFragPosition(false)
- , fCurrentBlock(0)
- , fSynthetics(nullptr, errors) {
- this->setupIntrinsics();
- }
- bool generateCode() override;
- private:
- enum IntrinsicKind {
- kGLSL_STD_450_IntrinsicKind,
- kSPIRV_IntrinsicKind,
- kSpecial_IntrinsicKind
- };
- enum SpecialIntrinsic {
- kAtan_SpecialIntrinsic,
- kClamp_SpecialIntrinsic,
- kMax_SpecialIntrinsic,
- kMin_SpecialIntrinsic,
- kMix_SpecialIntrinsic,
- kMod_SpecialIntrinsic,
- kDFdy_SpecialIntrinsic,
- kSaturate_SpecialIntrinsic,
- kSubpassLoad_SpecialIntrinsic,
- kTexture_SpecialIntrinsic,
- };
- enum class Precision {
- kLow,
- kHigh,
- };
- void setupIntrinsics();
- SpvId nextId();
- Type getActualType(const Type& type);
- SpvId getType(const Type& type);
- SpvId getType(const Type& type, const MemoryLayout& layout);
- SpvId getImageType(const Type& type);
- SpvId getFunctionType(const FunctionDeclaration& function);
- SpvId getPointerType(const Type& type, SpvStorageClass_ storageClass);
- SpvId getPointerType(const Type& type, const MemoryLayout& layout,
- SpvStorageClass_ storageClass);
- void writePrecisionModifier(Precision precision, SpvId id);
- void writePrecisionModifier(const Type& type, SpvId id);
- std::vector<SpvId> getAccessChain(const Expression& expr, OutputStream& out);
- void writeLayout(const Layout& layout, SpvId target);
- void writeLayout(const Layout& layout, SpvId target, int member);
- void writeStruct(const Type& type, const MemoryLayout& layout, SpvId resultId);
- void writeProgramElement(const ProgramElement& pe, OutputStream& out);
- SpvId writeInterfaceBlock(const InterfaceBlock& intf);
- SpvId writeFunctionStart(const FunctionDeclaration& f, OutputStream& out);
- SpvId writeFunctionDeclaration(const FunctionDeclaration& f, OutputStream& out);
- SpvId writeFunction(const FunctionDefinition& f, OutputStream& out);
- void writeGlobalVars(Program::Kind kind, const VarDeclarations& v, OutputStream& out);
- void writeVarDeclarations(const VarDeclarations& decl, OutputStream& out);
- SpvId writeVariableReference(const VariableReference& ref, OutputStream& out);
- std::unique_ptr<LValue> getLValue(const Expression& value, OutputStream& out);
- SpvId writeExpression(const Expression& expr, OutputStream& out);
- SpvId writeIntrinsicCall(const FunctionCall& c, OutputStream& out);
- SpvId writeFunctionCall(const FunctionCall& c, OutputStream& out);
- void writeGLSLExtendedInstruction(const Type& type, SpvId id, SpvId floatInst,
- SpvId signedInst, SpvId unsignedInst,
- const std::vector<SpvId>& args, OutputStream& out);
- /**
- * Given a list of potentially mixed scalars and vectors, promotes the scalars to match the
- * size of the vectors and returns the ids of the written expressions. e.g. given (float, vec2),
- * returns (vec2(float), vec2). It is an error to use mismatched vector sizes, e.g. (float,
- * vec2, vec3).
- */
- std::vector<SpvId> vectorize(const std::vector<std::unique_ptr<Expression>>& args,
- OutputStream& out);
- SpvId writeSpecialIntrinsic(const FunctionCall& c, SpecialIntrinsic kind, OutputStream& out);
- SpvId writeConstantVector(const Constructor& c);
- SpvId writeFloatConstructor(const Constructor& c, OutputStream& out);
- SpvId writeIntConstructor(const Constructor& c, OutputStream& out);
- SpvId writeUIntConstructor(const Constructor& c, OutputStream& out);
- /**
- * Writes a matrix with the diagonal entries all equal to the provided expression, and all other
- * entries equal to zero.
- */
- void writeUniformScaleMatrix(SpvId id, SpvId diagonal, const Type& type, OutputStream& out);
- /**
- * Writes a potentially-different-sized copy of a matrix. Entries which do not exist in the
- * source matrix are filled with zero; entries which do not exist in the destination matrix are
- * ignored.
- */
- void writeMatrixCopy(SpvId id, SpvId src, const Type& srcType, const Type& dstType,
- OutputStream& out);
- void addColumnEntry(SpvId columnType, Precision precision, std::vector<SpvId>* currentColumn,
- std::vector<SpvId>* columnIds, int* currentCount, int rows, SpvId entry,
- OutputStream& out);
- SpvId writeMatrixConstructor(const Constructor& c, OutputStream& out);
- SpvId writeVectorConstructor(const Constructor& c, OutputStream& out);
- SpvId writeArrayConstructor(const Constructor& c, OutputStream& out);
- SpvId writeConstructor(const Constructor& c, OutputStream& out);
- SpvId writeFieldAccess(const FieldAccess& f, OutputStream& out);
- SpvId writeSwizzle(const Swizzle& swizzle, 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
- * same dimensions, and applys all() to it to fold it down to a single bool value. Otherwise,
- * returns the original id value.
- */
- SpvId foldToBool(SpvId id, const Type& operandType, SpvOp op, OutputStream& out);
- SpvId writeMatrixComparison(const Type& operandType, SpvId lhs, SpvId rhs, SpvOp_ floatOperator,
- SpvOp_ intOperator, SpvOp_ vectorMergeOperator,
- SpvOp_ mergeOperator, OutputStream& out);
- SpvId writeComponentwiseMatrixBinary(const Type& operandType, SpvId lhs, SpvId rhs,
- SpvOp_ floatOperator, SpvOp_ intOperator,
- OutputStream& out);
- SpvId writeBinaryOperation(const Type& resultType, const Type& operandType, SpvId lhs,
- SpvId rhs, SpvOp_ ifFloat, SpvOp_ ifInt, SpvOp_ ifUInt,
- SpvOp_ ifBool, OutputStream& out);
- SpvId writeBinaryOperation(const BinaryExpression& expr, SpvOp_ ifFloat, SpvOp_ ifInt,
- SpvOp_ ifUInt, OutputStream& out);
- SpvId writeBinaryExpression(const Type& leftType, SpvId lhs, Token::Kind op,
- const Type& rightType, SpvId rhs, const Type& resultType,
- OutputStream& out);
- SpvId writeBinaryExpression(const BinaryExpression& b, OutputStream& out);
- SpvId writeTernaryExpression(const TernaryExpression& t, OutputStream& out);
- SpvId writeIndexExpression(const IndexExpression& expr, OutputStream& out);
- SpvId writeLogicalAnd(const BinaryExpression& b, OutputStream& out);
- SpvId writeLogicalOr(const BinaryExpression& o, OutputStream& out);
- SpvId writePrefixExpression(const PrefixExpression& p, OutputStream& out);
- SpvId writePostfixExpression(const PostfixExpression& p, OutputStream& out);
- SpvId writeBoolLiteral(const BoolLiteral& b);
- SpvId writeIntLiteral(const IntLiteral& i);
- SpvId writeFloatLiteral(const FloatLiteral& f);
- void writeStatement(const Statement& s, OutputStream& out);
- void writeBlock(const Block& b, OutputStream& out);
- void writeIfStatement(const IfStatement& stmt, OutputStream& out);
- void writeForStatement(const ForStatement& f, OutputStream& out);
- void writeWhileStatement(const WhileStatement& w, OutputStream& out);
- void writeDoStatement(const DoStatement& d, OutputStream& out);
- void writeSwitchStatement(const SwitchStatement& s, OutputStream& out);
- void writeReturnStatement(const ReturnStatement& r, OutputStream& out);
- void writeCapabilities(OutputStream& out);
- void writeInstructions(const Program& program, OutputStream& out);
- void writeOpCode(SpvOp_ opCode, int length, OutputStream& out);
- void writeWord(int32_t word, OutputStream& out);
- void writeString(const char* string, size_t length, OutputStream& out);
- void writeLabel(SpvId id, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, StringFragment string, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, StringFragment string, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, StringFragment string,
- OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3,
- OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3, int32_t word4,
- OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3, int32_t word4,
- int32_t word5, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3, int32_t word4,
- int32_t word5, int32_t word6, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3, int32_t word4,
- int32_t word5, int32_t word6, int32_t word7, OutputStream& out);
- void writeInstruction(SpvOp_ opCode, int32_t word1, int32_t word2, int32_t word3, int32_t word4,
- int32_t word5, int32_t word6, int32_t word7, int32_t word8,
- OutputStream& out);
- void writeGeometryShaderExecutionMode(SpvId entryPoint, OutputStream& out);
- const Context& fContext;
- const MemoryLayout fDefaultLayout;
- uint64_t fCapabilities;
- SpvId fIdCount;
- SpvId fGLSLExtendedInstructions;
- typedef std::tuple<IntrinsicKind, int32_t, int32_t, int32_t, int32_t> Intrinsic;
- std::unordered_map<String, Intrinsic> fIntrinsicMap;
- std::unordered_map<const FunctionDeclaration*, SpvId> fFunctionMap;
- std::unordered_map<const Variable*, SpvId> fVariableMap;
- std::unordered_map<const Variable*, int32_t> fInterfaceBlockMap;
- std::unordered_map<String, SpvId> fImageTypeMap;
- std::unordered_map<String, SpvId> fTypeMap;
- StringStream fCapabilitiesBuffer;
- StringStream fGlobalInitializersBuffer;
- StringStream fConstantBuffer;
- StringStream fExtraGlobalsBuffer;
- StringStream fExternalFunctionsBuffer;
- StringStream fVariableBuffer;
- StringStream fNameBuffer;
- StringStream fDecorationBuffer;
- SpvId fBoolTrue;
- SpvId fBoolFalse;
- std::unordered_map<std::pair<ConstantValue, ConstantType>, SpvId> fNumberConstants;
- // The constant float2(0, 1), used in swizzling
- SpvId fConstantZeroOneVector = 0;
- bool fSetupFragPosition;
- // label of the current block, or 0 if we are not in a block
- SpvId fCurrentBlock;
- std::stack<SpvId> fBreakTarget;
- std::stack<SpvId> fContinueTarget;
- SpvId fRTHeightStructId = (SpvId) -1;
- SpvId fRTHeightFieldIndex = (SpvId) -1;
- // holds variables synthesized during output, for lifetime purposes
- SymbolTable fSynthetics;
- int fSkInCount = 1;
- friend class PointerLValue;
- friend class SwizzleLValue;
- typedef CodeGenerator INHERITED;
- };
- }
- #endif
|