1d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Copyright 2016 The SwiftShader Authors. All Rights Reserved. 2d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// 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 6d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// 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. 14d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 15d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#ifndef COMPILER_PREPROCESSOR_MACRO_EXPANDER_H_ 16d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#define COMPILER_PREPROCESSOR_MACRO_EXPANDER_H_ 17d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 18d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include <cassert> 19d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include <memory> 20d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include <vector> 21d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 22d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include "Lexer.h" 23d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include "Macro.h" 24d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include "pp_utils.h" 25d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 26d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capensnamespace pp 27d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens{ 28d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 29d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capensclass Diagnostics; 30d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 31d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capensclass MacroExpander : public Lexer 32d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens{ 33d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capenspublic: 34d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens MacroExpander(Lexer* lexer, MacroSet* macroSet, Diagnostics* diagnostics, bool parseDefined); 35d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual ~MacroExpander(); 36d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 37d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens virtual void lex(Token* token); 38d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 39d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capensprivate: 40d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens PP_DISALLOW_COPY_AND_ASSIGN(MacroExpander); 41d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 42d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void getToken(Token* token); 43d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void ungetToken(const Token& token); 44d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool isNextTokenLeftParen(); 45d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 46d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool pushMacro(const Macro& macro, const Token& identifier); 47d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void popMacro(); 48d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 49d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool expandMacro(const Macro& macro, 50d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const Token& identifier, 51d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<Token>* replacements); 52d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 53d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens typedef std::vector<Token> MacroArg; 54d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool collectMacroArgs(const Macro& macro, 55d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const Token& identifier, 56d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<MacroArg>* args); 57d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void replaceMacroParams(const Macro& macro, 58d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const std::vector<MacroArg>& args, 59d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<Token>* replacements); 60d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 61d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens struct MacroContext 62d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens { 63d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const Macro* macro; 64d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens size_t index; 65d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<Token> replacements; 66d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 67d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens MacroContext() : macro(0), index(0) { } 68d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens bool empty() const { return index == replacements.size(); } 69d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const Token& get() { return replacements[index++]; } 70d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens void unget() { assert(index > 0); --index; } 71d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens }; 72d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 73d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens Lexer* mLexer; 74d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens MacroSet* mMacroSet; 75d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens Diagnostics* mDiagnostics; 76d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens const bool mParseDefined; 77d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 78d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::auto_ptr<Token> mReserveToken; 79d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens std::vector<MacroContext*> mContextStack; 80d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens}; 81d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 82d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens} // namespace pp 83d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#endif // COMPILER_PREPROCESSOR_MACRO_EXPANDER_H_ 84d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens 85