Line data    Source code

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