1d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Copyright 2016 The SwiftShader Authors. All Rights Reserved. 2894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 3d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Licensed under the Apache License, Version 2.0 (the "License"); 4d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// you may not use this file except in compliance with the License. 5d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// You may obtain a copy of the License at 6894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 7d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// http://www.apache.org/licenses/LICENSE-2.0 8d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// 9d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Unless required by applicable law or agreed to in writing, software 10d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// distributed under the License is distributed on an "AS IS" BASIS, 11d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// See the License for the specific language governing permissions and 13d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// limitations under the License. 14894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 15894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman#ifndef _SYMBOL_TABLE_INCLUDED_ 16894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman#define _SYMBOL_TABLE_INCLUDED_ 17894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 18894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 19894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// Symbol table for parsing. Has these design characteristics: 20894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 21894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// * Same symbol table can be used to compile many shaders, to preserve 22894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// effort of creating and loading with the large numbers of built-in 23894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// symbols. 24894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 25894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// * Name mangling will be used to give each function a unique name 26894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// so that symbol table lookups are never ambiguous. This allows 27894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// a simpler symbol table structure. 28894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 29cc1e996235768650d38d4ad26d50a9c148434ee5Nicolas Capens// * Pushing and popping of scope, so symbol table will really be a stack 30894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// of symbol tables. Searched from the top, with new inserts going into 31894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// the top. 32894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 33894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// * Constants: Compile time constant symbols will keep their values 34894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// in the symbol table. The parser can substitute constants at parse 35894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// time, including doing constant folding and constant propagation. 36894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 37894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// * No temporaries: Temporaries made from operations (+, --, .xy, etc.) 38894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// are tracked in the intermediate representation, not the symbol table. 39894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 40894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 41d61ac5fa24a7968441f947c6f0b27dc5c0d3148bGreg Hartman#ifndef __ANDROID__ 42894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman#include <assert.h> 43d61ac5fa24a7968441f947c6f0b27dc5c0d3148bGreg Hartman#else 44d61ac5fa24a7968441f947c6f0b27dc5c0d3148bGreg Hartman#include "../../Common/DebugAndroid.hpp" 45d61ac5fa24a7968441f947c6f0b27dc5c0d3148bGreg Hartman#endif 46894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 47cc863da574ed5079b055574127fe5788a9a0fc33Nicolas Capens#include "InfoSink.h" 48cc863da574ed5079b055574127fe5788a9a0fc33Nicolas Capens#include "intermediate.h" 49dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu#include <set> 50894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 51894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 52894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// Symbol base class. (Can build functions or variables out of these...) 53894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 540a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensclass TSymbol 550a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 56894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanpublic: 57d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens POOL_ALLOCATOR_NEW_DELETE(); 58d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol(const TString *n) : name(n) { } 59d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual ~TSymbol() { /* don't delete name, it's from the pool */ } 60d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 61d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TString& getName() const { return *name; } 62d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual const TString& getMangledName() const { return getName(); } 63d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual bool isFunction() const { return false; } 64d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual bool isVariable() const { return false; } 65d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void setUniqueId(int id) { uniqueId = id; } 66d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens int getUniqueId() const { return uniqueId; } 67d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol(const TSymbol&); 68894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 69894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanprotected: 70d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TString *name; 71d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens unsigned int uniqueId; // For real comparing during code generation 72894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 73894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 74894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 75894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// Variable class, meaning a symbol that's not a function. 76cc1e996235768650d38d4ad26d50a9c148434ee5Nicolas Capens// 77894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// There could be a separate class heirarchy for Constant variables; 78894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// Only one of int, bool, or float, (or none) is correct for 79894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// any particular use, but it's easy to do this way, and doesn't 80894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// seem worth having separate classes, and "getConst" can't simply return 81cc1e996235768650d38d4ad26d50a9c148434ee5Nicolas Capens// different values for different types polymorphically, so this is 82894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// just simple and pragmatic. 83894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 840a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensclass TVariable : public TSymbol 850a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 86894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanpublic: 87d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TVariable(const TString *name, const TType& t, bool uT = false ) : TSymbol(name), type(t), userType(uT), unionArray(0), arrayInformationType(0) { } 88d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual ~TVariable() { } 89d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual bool isVariable() const { return true; } 90d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType& getType() { return type; } 91d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TType& getType() const { return type; } 92d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool isUserType() const { return userType; } 93d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void setQualifier(TQualifier qualifier) { type.setQualifier(qualifier); } 94d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void updateArrayInformationType(TType *t) { arrayInformationType = t; } 95d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType* getArrayInformationType() { return arrayInformationType; } 96d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 97d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ConstantUnion* getConstPointer() 98d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 99d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (!unionArray) 100d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens unionArray = new ConstantUnion[type.getObjectSize()]; 101d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 102d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return unionArray; 103d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 104d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 105d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ConstantUnion* getConstPointer() const { return unionArray; } 1062cb7452bae261744d6410a01a54fd398f7f0c2acNicolas Capens bool isConstant() const { return unionArray != nullptr; } 107894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 108d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void shareConstPointer( ConstantUnion *constArray) 109d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 110d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (unionArray == constArray) 111d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return; 112894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 113d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens delete[] unionArray; 114d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens unionArray = constArray; 115d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 116894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 117894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanprotected: 118d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType type; 119d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool userType; 120d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // we are assuming that Pool Allocator will free the memory allocated to unionArray 121d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // when this object is destroyed 122d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ConstantUnion *unionArray; 123d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType *arrayInformationType; // this is used for updating maxArraySize in all the references to a given symbol 124894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 125894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 126894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 127894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// The function sub-class of symbols and the parser will need to 128894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// share this definition of a function parameter. 129894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 1300a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensstruct TParameter 1310a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 132d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TString *name; 133d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType *type; 134894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 135894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 136894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 137cc1e996235768650d38d4ad26d50a9c148434ee5Nicolas Capens// The function sub-class of a symbol. 138894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman// 1390a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensclass TFunction : public TSymbol 1400a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 141894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanpublic: 142d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TFunction(TOperator o) : 143d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol(0), 144d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens returnType(TType(EbtVoid, EbpUndefined)), 145d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens op(o), 146d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens defined(false), 147242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu prototypeDeclaration(false) { } 148d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TFunction(const TString *name, const TType& retType, TOperator tOp = EOpNull, const char *ext = "") : 149d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol(name), 150d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens returnType(retType), 151d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens mangledName(TFunction::mangleName(*name)), 152d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens op(tOp), 153d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens extension(ext), 154242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu defined(false), 155242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu prototypeDeclaration(false) { } 156d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual ~TFunction(); 157d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual bool isFunction() const { return true; } 158894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 159d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens static TString mangleName(const TString& name) { return name + '('; } 160d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens static TString unmangleName(const TString& mangledName) 161d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 162d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return TString(mangledName.c_str(), mangledName.find_first_of('(')); 163d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 164894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 165d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void addParameter(TParameter& p) 166d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 167d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens parameters.push_back(p); 168d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens mangledName = mangledName + p.type->getMangledName(); 169d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 170894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 171d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TString& getMangledName() const { return mangledName; } 172d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TType& getReturnType() const { return returnType; } 173894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 174d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TOperator getBuiltInOp() const { return op; } 175d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TString& getExtension() const { return extension; } 176894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 177d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void setDefined() { defined = true; } 178d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool isDefined() { return defined; } 179242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu void setHasPrototypeDeclaration() { prototypeDeclaration = true; } 180242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu bool hasPrototypeDeclaration() const { return prototypeDeclaration; } 181894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 182d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens size_t getParamCount() const { return parameters.size(); } 183d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const TParameter& getParam(int i) const { return parameters[i]; } 184894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 185894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanprotected: 186d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef TVector<TParameter> TParamList; 187d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TParamList parameters; 188d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TType returnType; 189d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TString mangledName; 190d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TOperator op; 191d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TString extension; 192d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool defined; 193242eb5cf7aa683c909eb3add56839ac046312bceAlexis Hetu bool prototypeDeclaration; 194894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 195894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 196894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 1970a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensclass TSymbolTableLevel 1980a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 199894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanpublic: 200d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef TMap<TString, TSymbol*> tLevel; 201d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef tLevel::const_iterator const_iterator; 202d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef const tLevel::value_type tLevelPair; 203d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef std::pair<tLevel::iterator, bool> tInsertResult; 204894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 205d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens POOL_ALLOCATOR_NEW_DELETE(); 206d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbolTableLevel() { } 207d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ~TSymbolTableLevel(); 208894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 209d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool insert(TSymbol &symbol) 210d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 211ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu symbol.setUniqueId(nextUniqueId()); 212d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman 213d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // 214d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // returning true means symbol was added to the table 215d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // 216d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens tInsertResult result; 217d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens result = level.insert(tLevelPair(symbol.getMangledName(), &symbol)); 218894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 219d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return result.second; 220d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 221894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 222d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol* find(const TString& name) const 223d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 224d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens tLevel::const_iterator it = level.find(name); 225d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (it == level.end()) 226d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return 0; 227d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens else 228d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return (*it).second; 229d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 230894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 231ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu static int nextUniqueId() 232ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu { 233ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu return ++uniqueId; 234ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu } 235ad6b87569fb1d2e1705e6e4b090d209be3f68e68Alexis Hetu 236894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanprotected: 237d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens tLevel level; 238d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman static int uniqueId; // for unique identification in code generation 239894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 240894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 241e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capensenum ESymbolLevel 242e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens{ 243d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens COMMON_BUILTINS, 244d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ESSL1_BUILTINS, 245d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ESSL3_BUILTINS, 246d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens LAST_BUILTIN_LEVEL = ESSL3_BUILTINS, 247d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens GLOBAL_LEVEL 248e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens}; 249e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens 2508fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capensinline bool IsGenType(const TType *type) 2518fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens{ 2528fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens if(type) 2538fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 2548fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens TBasicType basicType = type->getBasicType(); 2558fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return basicType == EbtGenType || basicType == EbtGenIType || basicType == EbtGenUType || basicType == EbtGenBType; 2568fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 2578fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2588fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return false; 2598fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens} 2608fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2618fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capensinline bool IsVecType(const TType *type) 2628fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens{ 2638fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens if(type) 2648fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 2658fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens TBasicType basicType = type->getBasicType(); 2668fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return basicType == EbtVec || basicType == EbtIVec || basicType == EbtUVec || basicType == EbtBVec; 2678fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 2688fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2698fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return false; 2708fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens} 2718fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2728fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capensinline TType *GenType(TType *type, int size) 2738fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens{ 2748fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens ASSERT(size >= 1 && size <= 4); 2758fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2768fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens if(!type) 2778fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 2788fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return nullptr; 2798fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 2808fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2818fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens ASSERT(!IsVecType(type)); 2828fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2838fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens switch(type->getBasicType()) 2848fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 2858fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtGenType: return new TType(EbtFloat, size); 2868fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtGenIType: return new TType(EbtInt, size); 2878fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtGenUType: return new TType(EbtUInt, size); 2888fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtGenBType: return new TType(EbtBool, size); 2898fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens default: return type; 2908fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 2918fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens} 2928fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2938fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capensinline TType *VecType(TType *type, int size) 2948fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens{ 2958fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens ASSERT(size >= 2 && size <= 4); 2968fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 2978fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens if(!type) 2988fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 2998fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens return nullptr; 3008fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 3018fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 3028fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens ASSERT(!IsGenType(type)); 3038fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 3048fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens switch(type->getBasicType()) 3058fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 3068fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtVec: return new TType(EbtFloat, size); 3078fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtIVec: return new TType(EbtInt, size); 3088fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtUVec: return new TType(EbtUInt, size); 3098fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens case EbtBVec: return new TType(EbtBool, size); 3108fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens default: return type; 3118fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 3128fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens} 3138fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 3140a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensclass TSymbolTable 3150a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capens{ 316894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Baumanpublic: 317d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbolTable() 318d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens : mGlobalInvariant(false) 319d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 320d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // 321d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // The symbol table cannot be used until push() is called, but 322d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // the lack of an initial call to push() can be used to detect 323d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // that the symbol table has not been preloaded with built-ins. 324d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // 325d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 326d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 327d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ~TSymbolTable() 328d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 329e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens while(currentLevel() > LAST_BUILTIN_LEVEL) 330e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens { 331e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens pop(); 332e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens } 333d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 334d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 335d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool isEmpty() { return table.empty(); } 336d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool atBuiltInLevel() { return currentLevel() <= LAST_BUILTIN_LEVEL; } 337d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool atGlobalLevel() { return currentLevel() <= GLOBAL_LEVEL; } 338d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void push() 339d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 340d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens table.push_back(new TSymbolTableLevel); 341d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens precisionStack.push_back( PrecisionStackLevel() ); 342d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 343d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 344d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void pop() 345d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 346d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens delete table[currentLevel()]; 347d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens table.pop_back(); 348d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens precisionStack.pop_back(); 349d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 350d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 351d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool declare(TSymbol &symbol) 352d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 353d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return insert(currentLevel(), symbol); 354d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 355d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 356d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool insert(ESymbolLevel level, TSymbol &symbol) 357d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 358d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return table[level]->insert(symbol); 359d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 360894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 361e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens bool insertConstInt(ESymbolLevel level, const char *name, int value) 362d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman { 36331ad2aa7f12e804d6628fb70d6e945c419cb2c7dNicolas Capens TVariable *constant = new TVariable(NewPoolTString(name), TType(EbtInt, EbpUndefined, EvqConstExpr, 1)); 364d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman constant->getConstPointer()->setIConst(value); 365e2858656575d8bc1f88ad3c885ab3cedf29a482eNicolas Capens return insert(level, *constant); 366d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman } 367d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman 368b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu void insertBuiltIn(ESymbolLevel level, TOperator op, const char *ext, TType *rvalue, const char *name, TType *ptype1, TType *ptype2 = 0, TType *ptype3 = 0, TType *ptype4 = 0, TType *ptype5 = 0) 369d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman { 370f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens if(ptype1->getBasicType() == EbtGSampler2D) 371f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens { 372f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens bool gvec4 = (rvalue->getBasicType() == EbtGVec4); 373b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtFloat, 4) : rvalue, name, new TType(EbtSampler2D), ptype2, ptype3, ptype4, ptype5); 374b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtInt, 4) : rvalue, name, new TType(EbtISampler2D), ptype2, ptype3, ptype4, ptype5); 375b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtUInt, 4) : rvalue, name, new TType(EbtUSampler2D), ptype2, ptype3, ptype4, ptype5); 376f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens } 377f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens else if(ptype1->getBasicType() == EbtGSampler3D) 378f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens { 379f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens bool gvec4 = (rvalue->getBasicType() == EbtGVec4); 380b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtFloat, 4) : rvalue, name, new TType(EbtSampler3D), ptype2, ptype3, ptype4, ptype5); 381b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtInt, 4) : rvalue, name, new TType(EbtISampler3D), ptype2, ptype3, ptype4, ptype5); 382b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtUInt, 4) : rvalue, name, new TType(EbtUSampler3D), ptype2, ptype3, ptype4, ptype5); 383f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens } 384f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens else if(ptype1->getBasicType() == EbtGSamplerCube) 385f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens { 386f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens bool gvec4 = (rvalue->getBasicType() == EbtGVec4); 387b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtFloat, 4) : rvalue, name, new TType(EbtSamplerCube), ptype2, ptype3, ptype4, ptype5); 388b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtInt, 4) : rvalue, name, new TType(EbtISamplerCube), ptype2, ptype3, ptype4, ptype5); 389b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtUInt, 4) : rvalue, name, new TType(EbtUSamplerCube), ptype2, ptype3, ptype4, ptype5); 390f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens } 391f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens else if(ptype1->getBasicType() == EbtGSampler2DArray) 392f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens { 393f8a4cba007896482e793e4d84f034b601e775632Nicolas Capens bool gvec4 = (rvalue->getBasicType() == EbtGVec4); 394b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtFloat, 4) : rvalue, name, new TType(EbtSampler2DArray), ptype2, ptype3, ptype4, ptype5); 395b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtInt, 4) : rvalue, name, new TType(EbtISampler2DArray), ptype2, ptype3, ptype4, ptype5); 396b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, gvec4 ? new TType(EbtUInt, 4) : rvalue, name, new TType(EbtUSampler2DArray), ptype2, ptype3, ptype4, ptype5); 3978fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 39833cda11b476675feccf858c5baa22cbffef51597Nicolas Capens else if(IsGenType(rvalue) || IsGenType(ptype1) || IsGenType(ptype2) || IsGenType(ptype3)) 399d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman { 40033cda11b476675feccf858c5baa22cbffef51597Nicolas Capens ASSERT(!ptype4); 40118d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, GenType(rvalue, 1), name, GenType(ptype1, 1), GenType(ptype2, 1), GenType(ptype3, 1)); 40218d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, GenType(rvalue, 2), name, GenType(ptype1, 2), GenType(ptype2, 2), GenType(ptype3, 2)); 40318d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, GenType(rvalue, 3), name, GenType(ptype1, 3), GenType(ptype2, 3), GenType(ptype3, 3)); 40418d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, GenType(rvalue, 4), name, GenType(ptype1, 4), GenType(ptype2, 4), GenType(ptype3, 4)); 405d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman } 40633cda11b476675feccf858c5baa22cbffef51597Nicolas Capens else if(IsVecType(rvalue) || IsVecType(ptype1) || IsVecType(ptype2) || IsVecType(ptype3)) 407d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman { 40833cda11b476675feccf858c5baa22cbffef51597Nicolas Capens ASSERT(!ptype4); 40918d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, VecType(rvalue, 2), name, VecType(ptype1, 2), VecType(ptype2, 2), VecType(ptype3, 2)); 41018d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, VecType(rvalue, 3), name, VecType(ptype1, 3), VecType(ptype2, 3), VecType(ptype3, 3)); 41118d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens insertBuiltIn(level, op, ext, VecType(rvalue, 4), name, VecType(ptype1, 4), VecType(ptype2, 4), VecType(ptype3, 4)); 412d4ae863d01d5f448dbbba6be4ecc161971a2324fJohn Bauman } 41333cda11b476675feccf858c5baa22cbffef51597Nicolas Capens else 4148fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens { 41518d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens TFunction *function = new TFunction(NewPoolTString(name), *rvalue, op, ext); 41633cda11b476675feccf858c5baa22cbffef51597Nicolas Capens 41733cda11b476675feccf858c5baa22cbffef51597Nicolas Capens TParameter param1 = {0, ptype1}; 41833cda11b476675feccf858c5baa22cbffef51597Nicolas Capens function->addParameter(param1); 41933cda11b476675feccf858c5baa22cbffef51597Nicolas Capens 42033cda11b476675feccf858c5baa22cbffef51597Nicolas Capens if(ptype2) 42133cda11b476675feccf858c5baa22cbffef51597Nicolas Capens { 42233cda11b476675feccf858c5baa22cbffef51597Nicolas Capens TParameter param2 = {0, ptype2}; 42333cda11b476675feccf858c5baa22cbffef51597Nicolas Capens function->addParameter(param2); 42433cda11b476675feccf858c5baa22cbffef51597Nicolas Capens } 42533cda11b476675feccf858c5baa22cbffef51597Nicolas Capens 42633cda11b476675feccf858c5baa22cbffef51597Nicolas Capens if(ptype3) 42733cda11b476675feccf858c5baa22cbffef51597Nicolas Capens { 42833cda11b476675feccf858c5baa22cbffef51597Nicolas Capens TParameter param3 = {0, ptype3}; 42933cda11b476675feccf858c5baa22cbffef51597Nicolas Capens function->addParameter(param3); 43033cda11b476675feccf858c5baa22cbffef51597Nicolas Capens } 43133cda11b476675feccf858c5baa22cbffef51597Nicolas Capens 43233cda11b476675feccf858c5baa22cbffef51597Nicolas Capens if(ptype4) 43333cda11b476675feccf858c5baa22cbffef51597Nicolas Capens { 43433cda11b476675feccf858c5baa22cbffef51597Nicolas Capens TParameter param4 = {0, ptype4}; 43533cda11b476675feccf858c5baa22cbffef51597Nicolas Capens function->addParameter(param4); 43633cda11b476675feccf858c5baa22cbffef51597Nicolas Capens } 43733cda11b476675feccf858c5baa22cbffef51597Nicolas Capens 438b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu if(ptype5) 439b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu { 440b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu TParameter param5 = {0, ptype5}; 441b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu function->addParameter(param5); 442b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu } 443b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu 44433cda11b476675feccf858c5baa22cbffef51597Nicolas Capens insert(level, *function); 4458fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens } 446d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 4478fa9d23dda6436369ae1e16dbed3bf1425626ae0Nicolas Capens 448b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu void insertBuiltIn(ESymbolLevel level, TOperator op, TType *rvalue, const char *name, TType *ptype1, TType *ptype2 = 0, TType *ptype3 = 0, TType *ptype4 = 0, TType *ptype5 = 0) 44918d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens { 450b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, op, "", rvalue, name, ptype1, ptype2, ptype3, ptype4, ptype5); 451d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 45218d5ef97c6a5112b08c310606ef854cafae864bfNicolas Capens 453b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu void insertBuiltIn(ESymbolLevel level, TType *rvalue, const char *name, TType *ptype1, TType *ptype2 = 0, TType *ptype3 = 0, TType *ptype4 = 0, TType *ptype5 = 0) 45433cda11b476675feccf858c5baa22cbffef51597Nicolas Capens { 455b5332c54150a1d30991390f77c8a277fd2e90c21Alexis Hetu insertBuiltIn(level, EOpNull, rvalue, name, ptype1, ptype2, ptype3, ptype4, ptype5); 456d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 457d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 458d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol *find(const TString &name, int shaderVersion, bool *builtIn = nullptr, bool *sameScope = nullptr) const; 459d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbol *findBuiltIn(const TString &name, int shaderVersion) const; 460d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 461d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TSymbolTableLevel *getOuterLevel() const 462d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 463d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens assert(currentLevel() >= 1); 464d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return table[currentLevel() - 1]; 465d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 466d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 467d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool setDefaultPrecision(const TPublicType &type, TPrecision prec) 468d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 469d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (IsSampler(type.type)) 470d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return true; // Skip sampler types for the time being 471d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (type.type != EbtFloat && type.type != EbtInt) 472d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return false; // Only set default precision for int/float 473d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (type.primarySize > 1 || type.secondarySize > 1 || type.array) 474d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return false; // Not allowed to set for aggregate types 475d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens int indexOfLastElement = static_cast<int>(precisionStack.size()) - 1; 476d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens precisionStack[indexOfLastElement][type.type] = prec; // Uses map operator [], overwrites the current value 477d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return true; 478d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 479d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 480d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // Searches down the precisionStack for a precision qualifier for the specified TBasicType 481d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TPrecision getDefaultPrecision( TBasicType type) 482d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 483d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens // unsigned integers use the same precision as signed 484d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if (type == EbtUInt) type = EbtInt; 485d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 486d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if( type != EbtFloat && type != EbtInt ) return EbpUndefined; 487d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens int level = static_cast<int>(precisionStack.size()) - 1; 488d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens assert( level >= 0); // Just to be safe. Should not happen. 489d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens PrecisionStackLevel::iterator it; 490d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens TPrecision prec = EbpUndefined; // If we dont find anything we return this. Should we error check this? 491d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens while( level >= 0 ){ 492d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens it = precisionStack[level].find( type ); 493d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens if( it != precisionStack[level].end() ){ 494d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens prec = (*it).second; 495d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens break; 496d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 497d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens level--; 498d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 499d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens return prec; 500d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens } 501894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 502dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // This records invariant varyings declared through 503dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // "invariant varying_name;". 504dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu void addInvariantVarying(const std::string &originalName) 505dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu { 506dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu mInvariantVaryings.insert(originalName); 507dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu } 508dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // If this returns false, the varying could still be invariant 509dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // if it is set as invariant during the varying variable 510dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // declaration - this piece of information is stored in the 511dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu // variable's type, not here. 512dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu bool isVaryingInvariant(const std::string &originalName) const 513dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu { 514dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu return (mGlobalInvariant || 515dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu mInvariantVaryings.count(originalName) > 0); 516dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu } 517dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu 518dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu void setGlobalInvariant() { mGlobalInvariant = true; } 519dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu bool getGlobalInvariant() const { return mGlobalInvariant; } 520dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu 5210a7f0c214220bfc89b5e512688124a10fbe54a40Nicolas Capensprotected: 522d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens ESymbolLevel currentLevel() const { return static_cast<ESymbolLevel>(table.size() - 1); } 523894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 524d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<TSymbolTableLevel*> table; 525d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef std::map< TBasicType, TPrecision > PrecisionStackLevel; 526d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector< PrecisionStackLevel > precisionStack; 527dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu 528dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu std::set<std::string> mInvariantVaryings; 529dd7ff7a5695e28d5e1fe156349786639ee0783d7Alexis Hetu bool mGlobalInvariant; 530894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman}; 531894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman 532894018228b0e0bdbd7aa7e8f47d4a9458789ca82John Bauman#endif // _SYMBOL_TABLE_INCLUDED_ 533