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 LLVM_MC_MCPARSER_ASMLEXER_H
15 #define LLVM_MC_MCPARSER_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   StringRef CurBuf;
32   bool isAtStartOfLine;
33 
34   void operator=(const AsmLexer&) = delete;
35   AsmLexer(const AsmLexer&) = delete;
36 
37 protected:
38   /// LexToken - Read the next token and return its code.
39   AsmToken LexToken() override;
40 
41 public:
42   AsmLexer(const MCAsmInfo &MAI);
43   ~AsmLexer() override;
44 
45   void setBuffer(StringRef Buf, const char *ptr = nullptr);
46 
47   StringRef LexUntilEndOfStatement() override;
48   StringRef LexUntilEndOfLine();
49 
50   const AsmToken peekTok(bool ShouldSkipSpace = true) override;
51 
52   bool isAtStartOfComment(const char *Ptr);
53   bool isAtStatementSeparator(const char *Ptr);
54 
getMAI()55   const MCAsmInfo &getMAI() const { return MAI; }
56 
57 private:
58   int getNextChar();
59   AsmToken ReturnError(const char *Loc, const std::string &Msg);
60 
61   AsmToken LexIdentifier();
62   AsmToken LexSlash();
63   AsmToken LexLineComment();
64   AsmToken LexDigit();
65   AsmToken LexSingleQuote();
66   AsmToken LexQuote();
67   AsmToken LexFloatLiteral();
68   AsmToken LexHexFloatLiteral(bool NoIntDigits);
69 };
70 
71 } // end namespace llvm
72 
73 #endif
74