Add MCSymbol::{print, dump}
[oota-llvm.git] / lib / MC / MCSymbol.cpp
1 //===- lib/MC/MCSymbol.cpp - MCSymbol implementation ----------------------===//
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 #include "llvm/MC/MCSymbol.h"
11 #include "llvm/Support/raw_ostream.h"
12
13 using namespace llvm;
14
15 /// NeedsQuoting - Return true if the string \arg Str needs quoting, i.e., it
16 /// does not match [a-zA-Z_.][a-zA-Z0-9_.]*.
17 //
18 // FIXME: This could be more permissive, do we care?
19 static inline bool NeedsQuoting(const StringRef &Str) {
20   if (Str.empty())
21     return true;
22
23   // Check that first character is in [a-zA-Z_.].
24   if (!((Str[0] >= 'a' && Str[0] <= 'z') ||
25         (Str[0] >= 'A' && Str[0] <= 'Z') ||
26         (Str[0] == '_' || Str[0] == '.')))
27     return true;
28
29   // Check subsequent characters are in [a-zA-Z0-9_.].
30   for (unsigned i = 1, e = Str.size(); i != e; ++i)
31     if (!((Str[i] >= 'a' && Str[i] <= 'z') ||
32           (Str[i] >= 'A' && Str[i] <= 'Z') ||
33           (Str[i] >= '0' && Str[i] <= '9') ||
34           (Str[i] == '_' || Str[i] == '.')))
35       return true;
36
37   return false;
38 }
39
40 void MCSymbol::print(raw_ostream &OS) const {
41   if (NeedsQuoting(getName()))
42     OS << '"' << getName() << '"';
43   else
44     OS << getName();
45 }
46
47 void MCSymbol::dump() const {
48   print(errs());
49 }