AsmLexer.h revision f1c21a8da6ed27a6ab4944e30bbeb4bd3ee08a71
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/Support/DataTypes.h" 21#include <string> 22#include <cassert> 23 24namespace llvm { 25class MemoryBuffer; 26class SMLoc; 27class MCAsmInfo; 28 29/// AsmLexer - Lexer class for assembly files. 30class AsmLexer : public MCAsmLexer { 31 const MCAsmInfo &MAI; 32 33 const char *CurPtr; 34 const MemoryBuffer *CurBuf; 35 36 void operator=(const AsmLexer&); // DO NOT IMPLEMENT 37 AsmLexer(const AsmLexer&); // DO NOT IMPLEMENT 38 39protected: 40 /// LexToken - Read the next token and return its code. 41 virtual AsmToken LexToken(); 42 43public: 44 AsmLexer(const MCAsmInfo &MAI); 45 ~AsmLexer(); 46 47 void setBuffer(const MemoryBuffer *buf, const char *ptr = NULL); 48 49 virtual StringRef LexUntilEndOfStatement(); 50 StringRef LexUntilEndOfLine(); 51 52 bool isAtStartOfComment(char Char); 53 bool isAtStatementSeparator(const char *Ptr); 54 55 const MCAsmInfo &getMAI() const { return MAI; } 56 57private: 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}; 69 70} // end namespace llvm 71 72#endif 73