1 //===- AsmLexer.h - Lexer for Assembly Files --------------------*- 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 // This class declares the lexer for assembly files. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef ASMLEXER_H 15 #define ASMLEXER_H 16 17 #include "llvm/ADT/StringRef.h" 18 #include "llvm/MC/MCParser/MCAsmLexer.h" 19 #include "llvm/Support/DataTypes.h" 20 #include <string> 21 22 namespace llvm { 23 class MemoryBuffer; 24 class MCAsmInfo; 25 26 /// AsmLexer - Lexer class for assembly files. 27 class AsmLexer : public MCAsmLexer { 28 const MCAsmInfo &MAI; 29 30 const char *CurPtr; 31 const MemoryBuffer *CurBuf; 32 bool isAtStartOfLine; 33 34 void operator=(const AsmLexer&); // DO NOT IMPLEMENT 35 AsmLexer(const AsmLexer&); // DO NOT IMPLEMENT 36 37 protected: 38 /// LexToken - Read the next token and return its code. 39 virtual AsmToken LexToken(); 40 41 public: 42 AsmLexer(const MCAsmInfo &MAI); 43 ~AsmLexer(); 44 45 void setBuffer(const MemoryBuffer *buf, const char *ptr = NULL); 46 47 virtual StringRef LexUntilEndOfStatement(); 48 StringRef LexUntilEndOfLine(); 49 50 bool isAtStartOfComment(char Char); 51 bool isAtStatementSeparator(const char *Ptr); 52 getMAI()53 const MCAsmInfo &getMAI() const { return MAI; } 54 55 private: 56 int getNextChar(); 57 AsmToken ReturnError(const char *Loc, const std::string &Msg); 58 59 AsmToken LexIdentifier(); 60 AsmToken LexSlash(); 61 AsmToken LexLineComment(); 62 AsmToken LexDigit(); 63 AsmToken LexSingleQuote(); 64 AsmToken LexQuote(); 65 AsmToken LexFloatLiteral(); 66 }; 67 68 } // end namespace llvm 69 70 #endif 71