1 //==--- SourceLocation.cpp - Compact identifier for Source 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 file defines accessor methods for the FullSourceLoc class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "clang/Basic/SourceLocation.h"
15 #include "clang/Basic/PrettyStackTrace.h"
16 #include "clang/Basic/SourceManager.h"
17 #include "llvm/Support/MemoryBuffer.h"
18 #include "llvm/Support/raw_ostream.h"
19 #include <cstdio>
20 using namespace clang;
21
22 //===----------------------------------------------------------------------===//
23 // PrettyStackTraceLoc
24 //===----------------------------------------------------------------------===//
25
print(raw_ostream & OS) const26 void PrettyStackTraceLoc::print(raw_ostream &OS) const {
27 if (Loc.isValid()) {
28 Loc.print(OS, SM);
29 OS << ": ";
30 }
31 OS << Message << '\n';
32 }
33
34 //===----------------------------------------------------------------------===//
35 // SourceLocation
36 //===----------------------------------------------------------------------===//
37
print(raw_ostream & OS,const SourceManager & SM) const38 void SourceLocation::print(raw_ostream &OS, const SourceManager &SM)const{
39 if (!isValid()) {
40 OS << "<invalid loc>";
41 return;
42 }
43
44 if (isFileID()) {
45 PresumedLoc PLoc = SM.getPresumedLoc(*this);
46
47 if (PLoc.isInvalid()) {
48 OS << "<invalid>";
49 return;
50 }
51 // The macro expansion and spelling pos is identical for file locs.
52 OS << PLoc.getFilename() << ':' << PLoc.getLine()
53 << ':' << PLoc.getColumn();
54 return;
55 }
56
57 SM.getExpansionLoc(*this).print(OS, SM);
58
59 OS << " <Spelling=";
60 SM.getSpellingLoc(*this).print(OS, SM);
61 OS << '>';
62 }
63
printToString(const SourceManager & SM) const64 std::string SourceLocation::printToString(const SourceManager &SM) const {
65 std::string S;
66 llvm::raw_string_ostream OS(S);
67 print(OS, SM);
68 return OS.str();
69 }
70
dump(const SourceManager & SM) const71 void SourceLocation::dump(const SourceManager &SM) const {
72 print(llvm::errs(), SM);
73 }
74
75 //===----------------------------------------------------------------------===//
76 // FullSourceLoc
77 //===----------------------------------------------------------------------===//
78
getFileID() const79 FileID FullSourceLoc::getFileID() const {
80 assert(isValid());
81 return SrcMgr->getFileID(*this);
82 }
83
84
getExpansionLoc() const85 FullSourceLoc FullSourceLoc::getExpansionLoc() const {
86 assert(isValid());
87 return FullSourceLoc(SrcMgr->getExpansionLoc(*this), *SrcMgr);
88 }
89
getSpellingLoc() const90 FullSourceLoc FullSourceLoc::getSpellingLoc() const {
91 assert(isValid());
92 return FullSourceLoc(SrcMgr->getSpellingLoc(*this), *SrcMgr);
93 }
94
getExpansionLineNumber(bool * Invalid) const95 unsigned FullSourceLoc::getExpansionLineNumber(bool *Invalid) const {
96 assert(isValid());
97 return SrcMgr->getExpansionLineNumber(*this, Invalid);
98 }
99
getExpansionColumnNumber(bool * Invalid) const100 unsigned FullSourceLoc::getExpansionColumnNumber(bool *Invalid) const {
101 assert(isValid());
102 return SrcMgr->getExpansionColumnNumber(*this, Invalid);
103 }
104
getSpellingLineNumber(bool * Invalid) const105 unsigned FullSourceLoc::getSpellingLineNumber(bool *Invalid) const {
106 assert(isValid());
107 return SrcMgr->getSpellingLineNumber(*this, Invalid);
108 }
109
getSpellingColumnNumber(bool * Invalid) const110 unsigned FullSourceLoc::getSpellingColumnNumber(bool *Invalid) const {
111 assert(isValid());
112 return SrcMgr->getSpellingColumnNumber(*this, Invalid);
113 }
114
isInSystemHeader() const115 bool FullSourceLoc::isInSystemHeader() const {
116 assert(isValid());
117 return SrcMgr->isInSystemHeader(*this);
118 }
119
isBeforeInTranslationUnitThan(SourceLocation Loc) const120 bool FullSourceLoc::isBeforeInTranslationUnitThan(SourceLocation Loc) const {
121 assert(isValid());
122 return SrcMgr->isBeforeInTranslationUnit(*this, Loc);
123 }
124
dump() const125 void FullSourceLoc::dump() const {
126 SourceLocation::dump(*SrcMgr);
127 }
128
getCharacterData(bool * Invalid) const129 const char *FullSourceLoc::getCharacterData(bool *Invalid) const {
130 assert(isValid());
131 return SrcMgr->getCharacterData(*this, Invalid);
132 }
133
getBuffer(bool * Invalid) const134 const llvm::MemoryBuffer* FullSourceLoc::getBuffer(bool *Invalid) const {
135 assert(isValid());
136 return SrcMgr->getBuffer(SrcMgr->getFileID(*this), Invalid);
137 }
138
getBufferData(bool * Invalid) const139 StringRef FullSourceLoc::getBufferData(bool *Invalid) const {
140 return getBuffer(Invalid)->getBuffer();
141 }
142
getDecomposedLoc() const143 std::pair<FileID, unsigned> FullSourceLoc::getDecomposedLoc() const {
144 return SrcMgr->getDecomposedLoc(*this);
145 }
146