1 //===- llvm/MC/MCAsmParserExtension.h - Asm Parser Hooks --------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #ifndef LLVM_MC_MCPARSER_MCASMPARSEREXTENSION_H 10 #define LLVM_MC_MCPARSER_MCASMPARSEREXTENSION_H 11 12 #include "llvm/ADT/STLFunctionalExtras.h" 13 #include "llvm/ADT/StringRef.h" 14 #include "llvm/MC/MCParser/MCAsmParser.h" 15 #include "llvm/Support/SMLoc.h" 16 17 namespace llvm { 18 19 class Twine; 20 21 /// Generic interface for extending the MCAsmParser, 22 /// which is implemented by target and object file assembly parser 23 /// implementations. 24 class MCAsmParserExtension { 25 MCAsmParser *Parser = nullptr; 26 27 protected: 28 MCAsmParserExtension(); 29 30 // Helper template for implementing static dispatch functions. 31 template<typename T, bool (T::*Handler)(StringRef, SMLoc)> HandleDirective(MCAsmParserExtension * Target,StringRef Directive,SMLoc DirectiveLoc)32 static bool HandleDirective(MCAsmParserExtension *Target, 33 StringRef Directive, 34 SMLoc DirectiveLoc) { 35 T *Obj = static_cast<T*>(Target); 36 return (Obj->*Handler)(Directive, DirectiveLoc); 37 } 38 39 bool BracketExpressionsSupported = false; 40 41 public: 42 MCAsmParserExtension(const MCAsmParserExtension &) = delete; 43 MCAsmParserExtension &operator=(const MCAsmParserExtension &) = delete; 44 virtual ~MCAsmParserExtension(); 45 46 /// Initialize the extension for parsing using the given \p Parser. 47 /// The extension should use the AsmParser interfaces to register its 48 /// parsing routines. 49 virtual void Initialize(MCAsmParser &Parser); 50 51 /// \name MCAsmParser Proxy Interfaces 52 /// @{ 53 getContext()54 MCContext &getContext() { return getParser().getContext(); } 55 getLexer()56 MCAsmLexer &getLexer() { return getParser().getLexer(); } getLexer()57 const MCAsmLexer &getLexer() const { 58 return const_cast<MCAsmParserExtension *>(this)->getLexer(); 59 } 60 getParser()61 MCAsmParser &getParser() { return *Parser; } getParser()62 const MCAsmParser &getParser() const { 63 return const_cast<MCAsmParserExtension*>(this)->getParser(); 64 } 65 getSourceManager()66 SourceMgr &getSourceManager() { return getParser().getSourceManager(); } getStreamer()67 MCStreamer &getStreamer() { return getParser().getStreamer(); } 68 Warning(SMLoc L,const Twine & Msg)69 bool Warning(SMLoc L, const Twine &Msg) { 70 return getParser().Warning(L, Msg); 71 } 72 73 bool Error(SMLoc L, const Twine &Msg, SMRange Range = SMRange()) { 74 return getParser().Error(L, Msg, Range); 75 } 76 Note(SMLoc L,const Twine & Msg)77 void Note(SMLoc L, const Twine &Msg) { 78 getParser().Note(L, Msg); 79 } 80 TokError(const Twine & Msg)81 bool TokError(const Twine &Msg) { 82 return getParser().TokError(Msg); 83 } 84 Lex()85 const AsmToken &Lex() { return getParser().Lex(); } getTok()86 const AsmToken &getTok() { return getParser().getTok(); } 87 bool parseToken(AsmToken::TokenKind T, 88 const Twine &Msg = "unexpected token") { 89 return getParser().parseToken(T, Msg); 90 } parseEOL()91 bool parseEOL() { return getParser().parseEOL(); } 92 93 bool parseMany(function_ref<bool()> parseOne, bool hasComma = true) { 94 return getParser().parseMany(parseOne, hasComma); 95 } 96 parseOptionalToken(AsmToken::TokenKind T)97 bool parseOptionalToken(AsmToken::TokenKind T) { 98 return getParser().parseOptionalToken(T); 99 } 100 101 bool ParseDirectiveCGProfile(StringRef, SMLoc); 102 check(bool P,const Twine & Msg)103 bool check(bool P, const Twine &Msg) { 104 return getParser().check(P, Msg); 105 } 106 check(bool P,SMLoc Loc,const Twine & Msg)107 bool check(bool P, SMLoc Loc, const Twine &Msg) { 108 return getParser().check(P, Loc, Msg); 109 } 110 addErrorSuffix(const Twine & Suffix)111 bool addErrorSuffix(const Twine &Suffix) { 112 return getParser().addErrorSuffix(Suffix); 113 } 114 HasBracketExpressions()115 bool HasBracketExpressions() const { return BracketExpressionsSupported; } 116 117 /// @} 118 }; 119 120 } // end namespace llvm 121 122 #endif // LLVM_MC_MCPARSER_MCASMPARSEREXTENSION_H 123