include/llvm/MC/MCParser/AsmLexer.h

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