MIR Parser: Simplify the token's string value handling.
[oota-llvm.git] / lib / CodeGen / MIRParser / MILexer.h
1 //===- MILexer.h - Lexer for machine instructions -------------------------===//
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 file declares the function that lexes the machine instruction source
11 // string.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef LLVM_LIB_CODEGEN_MIRPARSER_MILEXER_H
16 #define LLVM_LIB_CODEGEN_MIRPARSER_MILEXER_H
17
18 #include "llvm/ADT/APSInt.h"
19 #include "llvm/ADT/StringRef.h"
20 #include "llvm/ADT/STLExtras.h"
21 #include <functional>
22
23 namespace llvm {
24
25 class Twine;
26
27 /// A token produced by the machine instruction lexer.
28 struct MIToken {
29   enum TokenKind {
30     // Markers
31     Eof,
32     Error,
33
34     // Tokens with no info.
35     comma,
36     equal,
37     underscore,
38     colon,
39     coloncolon,
40     exclaim,
41     lparen,
42     rparen,
43     plus,
44     minus,
45
46     // Keywords
47     kw_implicit,
48     kw_implicit_define,
49     kw_dead,
50     kw_killed,
51     kw_undef,
52     kw_early_clobber,
53     kw_debug_use,
54     kw_frame_setup,
55     kw_debug_location,
56     kw_cfi_offset,
57     kw_cfi_def_cfa_register,
58     kw_cfi_def_cfa_offset,
59     kw_cfi_def_cfa,
60     kw_blockaddress,
61     kw_target_index,
62     kw_half,
63     kw_float,
64     kw_double,
65     kw_x86_fp80,
66     kw_fp128,
67     kw_ppc_fp128,
68     kw_target_flags,
69     kw_volatile,
70     kw_non_temporal,
71     kw_invariant,
72
73     // Identifier tokens
74     Identifier,
75     IntegerType,
76     NamedRegister,
77     MachineBasicBlock,
78     StackObject,
79     FixedStackObject,
80     NamedGlobalValue,
81     GlobalValue,
82     ExternalSymbol,
83
84     // Other tokens
85     IntegerLiteral,
86     FloatingPointLiteral,
87     VirtualRegister,
88     ConstantPoolItem,
89     JumpTableIndex,
90     NamedIRBlock,
91     IRBlock,
92     NamedIRValue,
93   };
94
95 private:
96   TokenKind Kind;
97   StringRef Range;
98   StringRef StringValue;
99   std::string StringValueStorage;
100   APSInt IntVal;
101
102 public:
103   MIToken() : Kind(Error) {}
104
105   MIToken &reset(TokenKind Kind, StringRef Range);
106
107   MIToken &setStringValue(StringRef StrVal);
108   MIToken &setOwnedStringValue(std::string StrVal);
109   MIToken &setIntegerValue(APSInt IntVal);
110
111   TokenKind kind() const { return Kind; }
112
113   bool isError() const { return Kind == Error; }
114
115   bool isRegister() const {
116     return Kind == NamedRegister || Kind == underscore ||
117            Kind == VirtualRegister;
118   }
119
120   bool isRegisterFlag() const {
121     return Kind == kw_implicit || Kind == kw_implicit_define ||
122            Kind == kw_dead || Kind == kw_killed || Kind == kw_undef ||
123            Kind == kw_early_clobber || Kind == kw_debug_use;
124   }
125
126   bool isMemoryOperandFlag() const {
127     return Kind == kw_volatile || Kind == kw_non_temporal ||
128            Kind == kw_invariant;
129   }
130
131   bool is(TokenKind K) const { return Kind == K; }
132
133   bool isNot(TokenKind K) const { return Kind != K; }
134
135   StringRef::iterator location() const { return Range.begin(); }
136
137   StringRef range() const { return Range; }
138
139   /// Return the token's string value.
140   StringRef stringValue() const { return StringValue; }
141
142   const APSInt &integerValue() const { return IntVal; }
143
144   bool hasIntegerValue() const {
145     return Kind == IntegerLiteral || Kind == MachineBasicBlock ||
146            Kind == StackObject || Kind == FixedStackObject ||
147            Kind == GlobalValue || Kind == VirtualRegister ||
148            Kind == ConstantPoolItem || Kind == JumpTableIndex ||
149            Kind == IRBlock;
150   }
151 };
152
153 /// Consume a single machine instruction token in the given source and return
154 /// the remaining source string.
155 StringRef lexMIToken(
156     StringRef Source, MIToken &Token,
157     function_ref<void(StringRef::iterator, const Twine &)> ErrorCallback);
158
159 } // end namespace llvm
160
161 #endif