Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1 | //===--- SourceManager.h - Track and cache source files ---------*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 0bc735f | 2007-12-29 19:59:25 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file defines the SourceManager interface. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #ifndef LLVM_CLANG_SOURCEMANAGER_H |
| 15 | #define LLVM_CLANG_SOURCEMANAGER_H |
| 16 | |
Chris Lattner | d47d3b0 | 2011-07-23 10:35:09 +0000 | [diff] [blame] | 17 | #include "clang/Basic/LLVM.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 18 | #include "clang/Basic/SourceLocation.h" |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 19 | #include "llvm/Support/Allocator.h" |
Michael J. Spencer | 03013fa | 2010-11-29 18:12:39 +0000 | [diff] [blame] | 20 | #include "llvm/Support/DataTypes.h" |
Douglas Gregor | c815108 | 2010-03-16 22:53:51 +0000 | [diff] [blame] | 21 | #include "llvm/ADT/PointerIntPair.h" |
Douglas Gregor | aea67db | 2010-03-15 22:54:52 +0000 | [diff] [blame] | 22 | #include "llvm/ADT/PointerUnion.h" |
Ted Kremenek | 4f32786 | 2011-03-21 18:40:17 +0000 | [diff] [blame] | 23 | #include "llvm/ADT/IntrusiveRefCntPtr.h" |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 24 | #include "llvm/ADT/DenseMap.h" |
Ted Kremenek | f61b831 | 2011-04-28 20:36:42 +0000 | [diff] [blame] | 25 | #include "llvm/Support/MemoryBuffer.h" |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 26 | #include <map> |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 27 | #include <vector> |
Chris Lattner | 9dc62f0 | 2007-07-12 15:32:57 +0000 | [diff] [blame] | 28 | #include <cassert> |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 29 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 30 | namespace clang { |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 31 | |
Douglas Gregor | aea67db | 2010-03-15 22:54:52 +0000 | [diff] [blame] | 32 | class Diagnostic; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 33 | class SourceManager; |
Ted Kremenek | 099b474 | 2007-12-05 00:14:18 +0000 | [diff] [blame] | 34 | class FileManager; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 35 | class FileEntry; |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 36 | class LineTableInfo; |
Argyrios Kyrtzidis | b73377e | 2011-07-07 03:40:34 +0000 | [diff] [blame] | 37 | class LangOptions; |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 38 | class ASTWriter; |
| 39 | class ASTReader; |
Douglas Gregor | aea67db | 2010-03-15 22:54:52 +0000 | [diff] [blame] | 40 | |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 41 | /// SrcMgr - Public enums and private classes that are part of the |
| 42 | /// SourceManager implementation. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 43 | /// |
| 44 | namespace SrcMgr { |
Chris Lattner | 9d72851 | 2008-10-27 01:19:25 +0000 | [diff] [blame] | 45 | /// CharacteristicKind - This is used to represent whether a file or directory |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 46 | /// holds normal user code, system code, or system code which is implicitly |
| 47 | /// 'extern "C"' in C++ mode. Entire directories can be tagged with this |
| 48 | /// (this is maintained by DirectoryLookup and friends) as can specific |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 49 | /// FileInfos when a #pragma system_header is seen or various other cases. |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 50 | /// |
Chris Lattner | 9d72851 | 2008-10-27 01:19:25 +0000 | [diff] [blame] | 51 | enum CharacteristicKind { |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 52 | C_User, C_System, C_ExternCSystem |
| 53 | }; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 54 | |
Dan Gohman | 4710a8e | 2010-08-25 21:59:25 +0000 | [diff] [blame] | 55 | /// ContentCache - One instance of this struct is kept for every file |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 56 | /// loaded or used. This object owns the MemoryBuffer object. |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 57 | class ContentCache { |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 58 | enum CCFlags { |
| 59 | /// \brief Whether the buffer is invalid. |
| 60 | InvalidFlag = 0x01, |
| 61 | /// \brief Whether the buffer should not be freed on destruction. |
| 62 | DoNotFreeFlag = 0x02 |
| 63 | }; |
| 64 | |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 65 | /// Buffer - The actual buffer containing the characters from the input |
| 66 | /// file. This is owned by the ContentCache object. |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 67 | /// The bits indicate indicates whether the buffer is invalid. |
| 68 | mutable llvm::PointerIntPair<const llvm::MemoryBuffer *, 2> Buffer; |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 69 | |
| 70 | public: |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 71 | /// Reference to the file entry representing this ContentCache. |
| 72 | /// This reference does not own the FileEntry object. |
| 73 | /// It is possible for this to be NULL if |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 74 | /// the ContentCache encapsulates an imaginary text buffer. |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 75 | const FileEntry *OrigEntry; |
| 76 | |
| 77 | /// \brief References the file which the contents were actually loaded from. |
| 78 | /// Can be different from 'Entry' if we overridden the contents of one file |
| 79 | /// with the contents of another file. |
| 80 | const FileEntry *ContentsEntry; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 81 | |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 82 | /// SourceLineCache - A bump pointer allocated array of offsets for each |
| 83 | /// source line. This is lazily computed. This is owned by the |
| 84 | /// SourceManager BumpPointerAllocator object. |
Chris Lattner | 0581659 | 2009-01-17 03:54:16 +0000 | [diff] [blame] | 85 | unsigned *SourceLineCache; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 86 | |
Ted Kremenek | b6427f8 | 2007-12-04 18:59:28 +0000 | [diff] [blame] | 87 | /// NumLines - The number of lines in this ContentCache. This is only valid |
| 88 | /// if SourceLineCache is non-null. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 89 | unsigned NumLines; |
Argyrios Kyrtzidis | 10b46d2 | 2009-06-20 08:09:57 +0000 | [diff] [blame] | 90 | |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 91 | /// \brief Lazily computed map of macro argument chunks to their expanded |
| 92 | /// source location. |
| 93 | typedef std::map<unsigned, SourceLocation> MacroArgsMap; |
| 94 | MacroArgsMap *MacroArgsCache; |
| 95 | |
Douglas Gregor | 36c35ba | 2010-03-16 00:35:39 +0000 | [diff] [blame] | 96 | /// getBuffer - Returns the memory buffer for the associated content. |
| 97 | /// |
Jonathan D. Turner | a92d7e7 | 2011-06-16 20:47:21 +0000 | [diff] [blame] | 98 | /// \param Diag Object through which diagnostics will be emitted if the |
Douglas Gregor | 36c35ba | 2010-03-16 00:35:39 +0000 | [diff] [blame] | 99 | /// buffer cannot be retrieved. |
| 100 | /// |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 101 | /// \param Loc If specified, is the location that invalid file diagnostics |
| 102 | /// will be emitted at. |
| 103 | /// |
Douglas Gregor | 36c35ba | 2010-03-16 00:35:39 +0000 | [diff] [blame] | 104 | /// \param Invalid If non-NULL, will be set \c true if an error occurred. |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 105 | const llvm::MemoryBuffer *getBuffer(Diagnostic &Diag, |
| 106 | const SourceManager &SM, |
| 107 | SourceLocation Loc = SourceLocation(), |
Douglas Gregor | 36c35ba | 2010-03-16 00:35:39 +0000 | [diff] [blame] | 108 | bool *Invalid = 0) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 109 | |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 110 | /// getSize - Returns the size of the content encapsulated by this |
| 111 | /// ContentCache. This can be the size of the source file or the size of an |
| 112 | /// arbitrary scratch buffer. If the ContentCache encapsulates a source |
| 113 | /// file this size is retrieved from the file's FileEntry. |
| 114 | unsigned getSize() const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 115 | |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 116 | /// getSizeBytesMapped - Returns the number of bytes actually mapped for |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 117 | /// this ContentCache. This can be 0 if the MemBuffer was not actually |
| 118 | /// expanded. |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 119 | unsigned getSizeBytesMapped() const; |
Ted Kremenek | f61b831 | 2011-04-28 20:36:42 +0000 | [diff] [blame] | 120 | |
| 121 | /// Returns the kind of memory used to back the memory buffer for |
| 122 | /// this content cache. This is used for performance analysis. |
| 123 | llvm::MemoryBuffer::BufferKind getMemoryBufferKind() const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 124 | |
Chris Lattner | 0581659 | 2009-01-17 03:54:16 +0000 | [diff] [blame] | 125 | void setBuffer(const llvm::MemoryBuffer *B) { |
Douglas Gregor | c815108 | 2010-03-16 22:53:51 +0000 | [diff] [blame] | 126 | assert(!Buffer.getPointer() && "MemoryBuffer already set."); |
| 127 | Buffer.setPointer(B); |
| 128 | Buffer.setInt(false); |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 129 | } |
Douglas Gregor | cc5888d | 2010-07-31 00:40:00 +0000 | [diff] [blame] | 130 | |
| 131 | /// \brief Get the underlying buffer, returning NULL if the buffer is not |
| 132 | /// yet available. |
| 133 | const llvm::MemoryBuffer *getRawBuffer() const { |
| 134 | return Buffer.getPointer(); |
| 135 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 136 | |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 137 | /// \brief Replace the existing buffer (which will be deleted) |
| 138 | /// with the given buffer. |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 139 | void replaceBuffer(const llvm::MemoryBuffer *B, bool DoNotFree = false); |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 140 | |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 141 | /// \brief Determine whether the buffer itself is invalid. |
| 142 | bool isBufferInvalid() const { |
| 143 | return Buffer.getInt() & InvalidFlag; |
| 144 | } |
| 145 | |
| 146 | /// \brief Determine whether the buffer should be freed. |
| 147 | bool shouldFreeBuffer() const { |
| 148 | return (Buffer.getInt() & DoNotFreeFlag) == 0; |
| 149 | } |
| 150 | |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 151 | ContentCache(const FileEntry *Ent = 0) |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 152 | : Buffer(0, false), OrigEntry(Ent), ContentsEntry(Ent), |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 153 | SourceLineCache(0), NumLines(0), MacroArgsCache(0) {} |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 154 | |
| 155 | ContentCache(const FileEntry *Ent, const FileEntry *contentEnt) |
| 156 | : Buffer(0, false), OrigEntry(Ent), ContentsEntry(contentEnt), |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 157 | SourceLineCache(0), NumLines(0), MacroArgsCache(0) {} |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 158 | |
| 159 | ~ContentCache(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 160 | |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 161 | /// The copy ctor does not allow copies where source object has either |
| 162 | /// a non-NULL Buffer or SourceLineCache. Ownership of allocated memory |
Chris Lattner | fc8f0e1 | 2011-04-15 05:22:18 +0000 | [diff] [blame] | 163 | /// is not transferred, so this is a logical error. |
Douglas Gregor | c815108 | 2010-03-16 22:53:51 +0000 | [diff] [blame] | 164 | ContentCache(const ContentCache &RHS) |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 165 | : Buffer(0, false), SourceLineCache(0), MacroArgsCache(0) |
Douglas Gregor | c815108 | 2010-03-16 22:53:51 +0000 | [diff] [blame] | 166 | { |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 167 | OrigEntry = RHS.OrigEntry; |
| 168 | ContentsEntry = RHS.ContentsEntry; |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 169 | |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 170 | assert (RHS.Buffer.getPointer() == 0 && RHS.SourceLineCache == 0 && |
| 171 | RHS.MacroArgsCache == 0 |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 172 | && "Passed ContentCache object cannot own a buffer."); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 173 | |
| 174 | NumLines = RHS.NumLines; |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 175 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 176 | |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 177 | private: |
| 178 | // Disable assignments. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 179 | ContentCache &operator=(const ContentCache& RHS); |
| 180 | }; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 181 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 182 | /// FileInfo - Information about a FileID, basically just the logical file |
| 183 | /// that it represents and include stack information. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 184 | /// |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 185 | /// Each FileInfo has include stack information, indicating where it came |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 186 | /// from. This information encodes the #include chain that a token was |
| 187 | /// expanded from. The main include file has an invalid IncludeLoc. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 188 | /// |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 189 | /// FileInfos contain a "ContentCache *", with the contents of the file. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 190 | /// |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 191 | class FileInfo { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 192 | /// IncludeLoc - The location of the #include that brought in this file. |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 193 | /// This is an invalid SLOC for the main file (top of the #include chain). |
| 194 | unsigned IncludeLoc; // Really a SourceLocation |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 195 | |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 196 | /// \brief Number of FileIDs (files and macros) that were created during |
| 197 | /// preprocessing of this #include, including this SLocEntry. |
| 198 | /// Zero means the preprocessor didn't provide such info for this SLocEntry. |
| 199 | unsigned NumCreatedFIDs; |
| 200 | |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 201 | /// Data - This contains the ContentCache* and the bits indicating the |
| 202 | /// characteristic of the file and whether it has #line info, all bitmangled |
| 203 | /// together. |
| 204 | uintptr_t Data; |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 205 | |
Argyrios Kyrtzidis | 21032df | 2011-08-21 23:49:52 +0000 | [diff] [blame] | 206 | friend class clang::SourceManager; |
| 207 | friend class clang::ASTWriter; |
| 208 | friend class clang::ASTReader; |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 209 | public: |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 210 | /// get - Return a FileInfo object. |
| 211 | static FileInfo get(SourceLocation IL, const ContentCache *Con, |
| 212 | CharacteristicKind FileCharacter) { |
| 213 | FileInfo X; |
| 214 | X.IncludeLoc = IL.getRawEncoding(); |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 215 | X.NumCreatedFIDs = 0; |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 216 | X.Data = (uintptr_t)Con; |
Chris Lattner | 00282d6 | 2009-02-03 07:41:46 +0000 | [diff] [blame] | 217 | assert((X.Data & 7) == 0 &&"ContentCache pointer insufficiently aligned"); |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 218 | assert((unsigned)FileCharacter < 4 && "invalid file character"); |
| 219 | X.Data |= (unsigned)FileCharacter; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 220 | return X; |
| 221 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 222 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 223 | SourceLocation getIncludeLoc() const { |
| 224 | return SourceLocation::getFromRawEncoding(IncludeLoc); |
| 225 | } |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 226 | const ContentCache* getContentCache() const { |
Chris Lattner | 00282d6 | 2009-02-03 07:41:46 +0000 | [diff] [blame] | 227 | return reinterpret_cast<const ContentCache*>(Data & ~7UL); |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 228 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 229 | |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 230 | /// getCharacteristic - Return whether this is a system header or not. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 231 | CharacteristicKind getFileCharacteristic() const { |
Chris Lattner | 6e1aff2 | 2009-01-26 06:49:09 +0000 | [diff] [blame] | 232 | return (CharacteristicKind)(Data & 3); |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 233 | } |
Chris Lattner | ac50e34 | 2009-02-03 22:13:05 +0000 | [diff] [blame] | 234 | |
| 235 | /// hasLineDirectives - Return true if this FileID has #line directives in |
| 236 | /// it. |
| 237 | bool hasLineDirectives() const { return (Data & 4) != 0; } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 238 | |
Chris Lattner | ac50e34 | 2009-02-03 22:13:05 +0000 | [diff] [blame] | 239 | /// setHasLineDirectives - Set the flag that indicates that this FileID has |
| 240 | /// line table entries associated with it. |
| 241 | void setHasLineDirectives() { |
| 242 | Data |= 4; |
| 243 | } |
Chris Lattner | 9dc1f53 | 2007-07-20 16:37:10 +0000 | [diff] [blame] | 244 | }; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 245 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 246 | /// ExpansionInfo - Each ExpansionInfo encodes the expansion location - where |
| 247 | /// the token was ultimately expanded, and the SpellingLoc - where the actual |
| 248 | /// character data for the token came from. |
| 249 | class ExpansionInfo { |
| 250 | // Really these are all SourceLocations. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 251 | |
Chris Lattner | e7fb484 | 2009-02-15 20:52:18 +0000 | [diff] [blame] | 252 | /// SpellingLoc - Where the spelling for the token can be found. |
| 253 | unsigned SpellingLoc; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 254 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 255 | /// ExpansionLocStart/ExpansionLocEnd - In a macro expansion, these |
| 256 | /// indicate the start and end of the expansion. In object-like macros, |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 257 | /// these will be the same. In a function-like macro expansion, the start |
| 258 | /// will be the identifier and the end will be the ')'. Finally, in |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 259 | /// macro-argument instantitions, the end will be 'SourceLocation()', an |
| 260 | /// invalid location. |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 261 | unsigned ExpansionLocStart, ExpansionLocEnd; |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 262 | |
Chris Lattner | 9dc1f53 | 2007-07-20 16:37:10 +0000 | [diff] [blame] | 263 | public: |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 264 | SourceLocation getSpellingLoc() const { |
| 265 | return SourceLocation::getFromRawEncoding(SpellingLoc); |
| 266 | } |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 267 | SourceLocation getExpansionLocStart() const { |
| 268 | return SourceLocation::getFromRawEncoding(ExpansionLocStart); |
Chris Lattner | e7fb484 | 2009-02-15 20:52:18 +0000 | [diff] [blame] | 269 | } |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 270 | SourceLocation getExpansionLocEnd() const { |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 271 | SourceLocation EndLoc = |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 272 | SourceLocation::getFromRawEncoding(ExpansionLocEnd); |
| 273 | return EndLoc.isInvalid() ? getExpansionLocStart() : EndLoc; |
Chris Lattner | e7fb484 | 2009-02-15 20:52:18 +0000 | [diff] [blame] | 274 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 275 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 276 | std::pair<SourceLocation,SourceLocation> getExpansionLocRange() const { |
| 277 | return std::make_pair(getExpansionLocStart(), getExpansionLocEnd()); |
Chris Lattner | e7fb484 | 2009-02-15 20:52:18 +0000 | [diff] [blame] | 278 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 279 | |
Chandler Carruth | 96d3589 | 2011-07-26 03:03:00 +0000 | [diff] [blame] | 280 | bool isMacroArgExpansion() const { |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 281 | // Note that this needs to return false for default constructed objects. |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 282 | return getExpansionLocStart().isValid() && |
| 283 | SourceLocation::getFromRawEncoding(ExpansionLocEnd).isInvalid(); |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 284 | } |
| 285 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 286 | /// create - Return a ExpansionInfo for an expansion. Start and End specify |
| 287 | /// the expansion range (where the macro is expanded), and SpellingLoc |
| 288 | /// specifies the spelling location (where the characters from the token |
| 289 | /// come from). All three can refer to normal File SLocs or expansion |
| 290 | /// locations. |
| 291 | static ExpansionInfo create(SourceLocation SpellingLoc, |
| 292 | SourceLocation Start, SourceLocation End) { |
| 293 | ExpansionInfo X; |
| 294 | X.SpellingLoc = SpellingLoc.getRawEncoding(); |
| 295 | X.ExpansionLocStart = Start.getRawEncoding(); |
| 296 | X.ExpansionLocEnd = End.getRawEncoding(); |
Chris Lattner | 9dc1f53 | 2007-07-20 16:37:10 +0000 | [diff] [blame] | 297 | return X; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 298 | } |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 299 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 300 | /// createForMacroArg - Return a special ExpansionInfo for the expansion of |
| 301 | /// a macro argument into a function-like macro's body. ExpansionLoc |
| 302 | /// specifies the expansion location (where the macro is expanded). This |
| 303 | /// doesn't need to be a range because a macro is always expanded at |
| 304 | /// a macro parameter reference, and macro parameters are always exactly |
| 305 | /// one token. SpellingLoc specifies the spelling location (where the |
| 306 | /// characters from the token come from). ExpansionLoc and SpellingLoc can |
| 307 | /// both refer to normal File SLocs or expansion locations. |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 308 | /// |
| 309 | /// Given the code: |
| 310 | /// \code |
| 311 | /// #define F(x) f(x) |
| 312 | /// F(42); |
| 313 | /// \endcode |
| 314 | /// |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 315 | /// When expanding '\c F(42)', the '\c x' would call this with an |
| 316 | /// SpellingLoc pointing at '\c 42' anad an ExpansionLoc pointing at its |
| 317 | /// location in the definition of '\c F'. |
| 318 | static ExpansionInfo createForMacroArg(SourceLocation SpellingLoc, |
| 319 | SourceLocation ExpansionLoc) { |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 320 | // We store an intentionally invalid source location for the end of the |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 321 | // expansion range to mark that this is a macro argument ion rather than |
| 322 | // a normal one. |
| 323 | return create(SpellingLoc, ExpansionLoc, SourceLocation()); |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 324 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 325 | }; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 326 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 327 | /// SLocEntry - This is a discriminated union of FileInfo and |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 328 | /// ExpansionInfo. SourceManager keeps an array of these objects, and |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 329 | /// they are uniquely identified by the FileID datatype. |
| 330 | class SLocEntry { |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 331 | unsigned Offset; // low bit is set for expansion info. |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 332 | union { |
| 333 | FileInfo File; |
Chandler Carruth | 1728762 | 2011-07-26 04:56:51 +0000 | [diff] [blame] | 334 | ExpansionInfo Expansion; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 335 | }; |
| 336 | public: |
| 337 | unsigned getOffset() const { return Offset >> 1; } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 338 | |
Chandler Carruth | 1728762 | 2011-07-26 04:56:51 +0000 | [diff] [blame] | 339 | bool isExpansion() const { return Offset & 1; } |
| 340 | bool isFile() const { return !isExpansion(); } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 341 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 342 | const FileInfo &getFile() const { |
| 343 | assert(isFile() && "Not a file SLocEntry!"); |
| 344 | return File; |
| 345 | } |
| 346 | |
Chandler Carruth | 1728762 | 2011-07-26 04:56:51 +0000 | [diff] [blame] | 347 | const ExpansionInfo &getExpansion() const { |
| 348 | assert(isExpansion() && "Not a macro expansion SLocEntry!"); |
| 349 | return Expansion; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 350 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 351 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 352 | static SLocEntry get(unsigned Offset, const FileInfo &FI) { |
| 353 | SLocEntry E; |
| 354 | E.Offset = Offset << 1; |
| 355 | E.File = FI; |
| 356 | return E; |
| 357 | } |
| 358 | |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 359 | static SLocEntry get(unsigned Offset, const ExpansionInfo &Expansion) { |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 360 | SLocEntry E; |
| 361 | E.Offset = (Offset << 1) | 1; |
Chandler Carruth | 1728762 | 2011-07-26 04:56:51 +0000 | [diff] [blame] | 362 | E.Expansion = Expansion; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 363 | return E; |
| 364 | } |
| 365 | }; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 366 | } // end SrcMgr namespace. |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 367 | |
| 368 | /// \brief External source of source location entries. |
| 369 | class ExternalSLocEntrySource { |
| 370 | public: |
| 371 | virtual ~ExternalSLocEntrySource(); |
| 372 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 373 | /// \brief Read the source location entry with index ID, which will always be |
| 374 | /// less than -1. |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 375 | /// |
| 376 | /// \returns true if an error occurred that prevented the source-location |
| 377 | /// entry from being loaded. |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 378 | virtual bool ReadSLocEntry(int ID) = 0; |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 379 | }; |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 380 | |
| 381 | |
| 382 | /// IsBeforeInTranslationUnitCache - This class holds the cache used by |
| 383 | /// isBeforeInTranslationUnit. The cache structure is complex enough to be |
| 384 | /// worth breaking out of SourceManager. |
| 385 | class IsBeforeInTranslationUnitCache { |
| 386 | /// L/R QueryFID - These are the FID's of the cached query. If these match up |
| 387 | /// with a subsequent query, the result can be reused. |
| 388 | FileID LQueryFID, RQueryFID; |
Argyrios Kyrtzidis | 37e59a1 | 2011-08-17 00:31:18 +0000 | [diff] [blame] | 389 | |
| 390 | /// \brief True if LQueryFID was created before RQueryFID. This is used |
| 391 | /// to compare macro expansion locations. |
| 392 | bool IsLQFIDBeforeRQFID; |
| 393 | |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 394 | /// CommonFID - This is the file found in common between the two #include |
| 395 | /// traces. It is the nearest common ancestor of the #include tree. |
| 396 | FileID CommonFID; |
| 397 | |
| 398 | /// L/R CommonOffset - This is the offset of the previous query in CommonFID. |
| 399 | /// Usually, this represents the location of the #include for QueryFID, but if |
| 400 | /// LQueryFID is a parent of RQueryFID (or vise versa) then these can be a |
| 401 | /// random token in the parent. |
| 402 | unsigned LCommonOffset, RCommonOffset; |
| 403 | public: |
| 404 | |
| 405 | /// isCacheValid - Return true if the currently cached values match up with |
| 406 | /// the specified LHS/RHS query. If not, we can't use the cache. |
| 407 | bool isCacheValid(FileID LHS, FileID RHS) const { |
| 408 | return LQueryFID == LHS && RQueryFID == RHS; |
| 409 | } |
| 410 | |
| 411 | /// getCachedResult - If the cache is valid, compute the result given the |
| 412 | /// specified offsets in the LHS/RHS FID's. |
| 413 | bool getCachedResult(unsigned LOffset, unsigned ROffset) const { |
| 414 | // If one of the query files is the common file, use the offset. Otherwise, |
| 415 | // use the #include loc in the common file. |
| 416 | if (LQueryFID != CommonFID) LOffset = LCommonOffset; |
| 417 | if (RQueryFID != CommonFID) ROffset = RCommonOffset; |
Argyrios Kyrtzidis | 37e59a1 | 2011-08-17 00:31:18 +0000 | [diff] [blame] | 418 | |
| 419 | // It is common for multiple macro expansions to be "included" from the same |
| 420 | // location (expansion location), in which case use the order of the FileIDs |
| 421 | // to determine which came first. |
| 422 | if (LOffset == ROffset && LQueryFID != CommonFID && RQueryFID != CommonFID) |
| 423 | return IsLQFIDBeforeRQFID; |
| 424 | |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 425 | return LOffset < ROffset; |
| 426 | } |
| 427 | |
| 428 | // Set up a new query. |
Argyrios Kyrtzidis | 37e59a1 | 2011-08-17 00:31:18 +0000 | [diff] [blame] | 429 | void setQueryFIDs(FileID LHS, FileID RHS, bool isLFIDBeforeRFID) { |
| 430 | assert(LHS != RHS); |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 431 | LQueryFID = LHS; |
| 432 | RQueryFID = RHS; |
Argyrios Kyrtzidis | 37e59a1 | 2011-08-17 00:31:18 +0000 | [diff] [blame] | 433 | IsLQFIDBeforeRQFID = isLFIDBeforeRFID; |
| 434 | } |
| 435 | |
| 436 | void clear() { |
| 437 | LQueryFID = RQueryFID = FileID(); |
| 438 | IsLQFIDBeforeRQFID = false; |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 439 | } |
| 440 | |
| 441 | void setCommonLoc(FileID commonFID, unsigned lCommonOffset, |
| 442 | unsigned rCommonOffset) { |
| 443 | CommonFID = commonFID; |
| 444 | LCommonOffset = lCommonOffset; |
| 445 | RCommonOffset = rCommonOffset; |
| 446 | } |
| 447 | |
| 448 | }; |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 449 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 450 | /// \brief This class handles loading and caching of source files into memory. |
| 451 | /// |
| 452 | /// This object owns the MemoryBuffer objects for all of the loaded |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 453 | /// files and assigns unique FileID's for each unique #include chain. |
| 454 | /// |
| 455 | /// The SourceManager can be queried for information about SourceLocation |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 456 | /// objects, turning them into either spelling or expansion locations. Spelling |
| 457 | /// locations represent where the bytes corresponding to a token came from and |
| 458 | /// expansion locations represent where the location is in the user's view. In |
| 459 | /// the case of a macro expansion, for example, the spelling location indicates |
| 460 | /// where the expanded token came from and the expansion location specifies |
| 461 | /// where it was expanded. |
Ted Kremenek | 4f32786 | 2011-03-21 18:40:17 +0000 | [diff] [blame] | 462 | class SourceManager : public llvm::RefCountedBase<SourceManager> { |
Douglas Gregor | f715ca1 | 2010-03-16 00:06:06 +0000 | [diff] [blame] | 463 | /// \brief Diagnostic object. |
| 464 | Diagnostic &Diag; |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 465 | |
| 466 | FileManager &FileMgr; |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 467 | |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 468 | mutable llvm::BumpPtrAllocator ContentCacheAlloc; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 469 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 470 | /// FileInfos - Memoized information about all of the files tracked by this |
Ted Kremenek | 0d892d8 | 2007-10-30 22:57:35 +0000 | [diff] [blame] | 471 | /// SourceManager. This set allows us to merge ContentCache entries based |
| 472 | /// on their FileEntry*. All ContentCache objects will thus have unique, |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 473 | /// non-null, FileEntry pointers. |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 474 | llvm::DenseMap<const FileEntry*, SrcMgr::ContentCache*> FileInfos; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 475 | |
Argyrios Kyrtzidis | 299a4a9 | 2011-03-08 23:35:24 +0000 | [diff] [blame] | 476 | /// \brief True if the ContentCache for files that are overriden by other |
| 477 | /// files, should report the original file name. Defaults to true. |
| 478 | bool OverridenFilesKeepOriginalName; |
| 479 | |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 480 | /// \brief Files that have been overriden with the contents from another file. |
| 481 | llvm::DenseMap<const FileEntry *, const FileEntry *> OverriddenFiles; |
| 482 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 483 | /// MemBufferInfos - Information about various memory buffers that we have |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 484 | /// read in. All FileEntry* within the stored ContentCache objects are NULL, |
| 485 | /// as they do not refer to a file. |
| 486 | std::vector<SrcMgr::ContentCache*> MemBufferInfos; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 487 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 488 | /// \brief The table of SLocEntries that are local to this module. |
| 489 | /// |
| 490 | /// Positive FileIDs are indexes into this table. Entry 0 indicates an invalid |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 491 | /// expansion. |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 492 | std::vector<SrcMgr::SLocEntry> LocalSLocEntryTable; |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 493 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 494 | /// \brief The table of SLocEntries that are loaded from other modules. |
| 495 | /// |
| 496 | /// Negative FileIDs are indexes into this table. To get from ID to an index, |
| 497 | /// use (-ID - 2). |
| 498 | std::vector<SrcMgr::SLocEntry> LoadedSLocEntryTable; |
| 499 | |
| 500 | /// \brief The starting offset of the next local SLocEntry. |
| 501 | /// |
| 502 | /// This is LocalSLocEntryTable.back().Offset + the size of that entry. |
| 503 | unsigned NextLocalOffset; |
| 504 | |
| 505 | /// \brief The starting offset of the latest batch of loaded SLocEntries. |
| 506 | /// |
| 507 | /// This is LoadedSLocEntryTable.back().Offset, except that that entry might |
| 508 | /// not have been loaded, so that value would be unknown. |
| 509 | unsigned CurrentLoadedOffset; |
| 510 | |
Argyrios Kyrtzidis | ac836e4 | 2011-08-17 00:31:20 +0000 | [diff] [blame] | 511 | /// \brief The highest possible offset is 2^31-1, so CurrentLoadedOffset |
| 512 | /// starts at 2^31. |
| 513 | static const unsigned MaxLoadedOffset = 1U << 31U; |
| 514 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 515 | /// \brief A bitmap that indicates whether the entries of LoadedSLocEntryTable |
| 516 | /// have already been loaded from the external source. |
| 517 | /// |
| 518 | /// Same indexing as LoadedSLocEntryTable. |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 519 | std::vector<bool> SLocEntryLoaded; |
| 520 | |
| 521 | /// \brief An external source for source location entries. |
| 522 | ExternalSLocEntrySource *ExternalSLocEntries; |
| 523 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 524 | /// LastFileIDLookup - This is a one-entry cache to speed up getFileID. |
| 525 | /// LastFileIDLookup records the last FileID looked up or created, because it |
| 526 | /// is very common to look up many tokens from the same file. |
| 527 | mutable FileID LastFileIDLookup; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 528 | |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 529 | /// LineTable - This holds information for #line directives. It is referenced |
| 530 | /// by indices from SLocEntryTable. |
| 531 | LineTableInfo *LineTable; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 532 | |
Chris Lattner | 5e36a7a | 2007-07-24 05:57:19 +0000 | [diff] [blame] | 533 | /// LastLineNo - These ivars serve as a cache used in the getLineNumber |
| 534 | /// method which is used to speedup getLineNumber calls to nearby locations. |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 535 | mutable FileID LastLineNoFileIDQuery; |
Chris Lattner | f812a45 | 2008-11-18 06:51:15 +0000 | [diff] [blame] | 536 | mutable SrcMgr::ContentCache *LastLineNoContentCache; |
| 537 | mutable unsigned LastLineNoFilePos; |
| 538 | mutable unsigned LastLineNoResult; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 539 | |
Ted Kremenek | 76edd0e | 2007-12-19 22:29:55 +0000 | [diff] [blame] | 540 | /// MainFileID - The file ID for the main source file of the translation unit. |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 541 | FileID MainFileID; |
Steve Naroff | 49c1f4a | 2008-02-02 00:10:46 +0000 | [diff] [blame] | 542 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 543 | // Statistics for -print-stats. |
| 544 | mutable unsigned NumLinearScans, NumBinaryProbes; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 545 | |
Argyrios Kyrtzidis | 2aa03d5 | 2009-06-23 22:01:48 +0000 | [diff] [blame] | 546 | // Cache results for the isBeforeInTranslationUnit method. |
Chris Lattner | dcb1d68 | 2010-05-07 01:17:07 +0000 | [diff] [blame] | 547 | mutable IsBeforeInTranslationUnitCache IsBeforeInTUCache; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 548 | |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 549 | // Cache for the "fake" buffer used for error-recovery purposes. |
| 550 | mutable llvm::MemoryBuffer *FakeBufferForRecovery; |
| 551 | |
Steve Naroff | 49c1f4a | 2008-02-02 00:10:46 +0000 | [diff] [blame] | 552 | // SourceManager doesn't support copy construction. |
| 553 | explicit SourceManager(const SourceManager&); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 554 | void operator=(const SourceManager&); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 555 | public: |
Chris Lattner | 39b49bc | 2010-11-23 08:35:12 +0000 | [diff] [blame] | 556 | SourceManager(Diagnostic &Diag, FileManager &FileMgr); |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 557 | ~SourceManager(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 558 | |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 559 | void clearIDTables(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 560 | |
Argyrios Kyrtzidis | 78a916e | 2010-09-22 14:32:24 +0000 | [diff] [blame] | 561 | Diagnostic &getDiagnostics() const { return Diag; } |
| 562 | |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 563 | FileManager &getFileManager() const { return FileMgr; } |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 564 | |
Argyrios Kyrtzidis | 299a4a9 | 2011-03-08 23:35:24 +0000 | [diff] [blame] | 565 | /// \brief Set true if the SourceManager should report the original file name |
| 566 | /// for contents of files that were overriden by other files.Defaults to true. |
| 567 | void setOverridenFilesKeepOriginalName(bool value) { |
| 568 | OverridenFilesKeepOriginalName = value; |
| 569 | } |
| 570 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 571 | /// createMainFileIDForMembuffer - Create the FileID for a memory buffer |
| 572 | /// that will represent the FileID for the main source. One example |
| 573 | /// of when this would be used is when the main source is read from STDIN. |
| 574 | FileID createMainFileIDForMemBuffer(const llvm::MemoryBuffer *Buffer) { |
| 575 | assert(MainFileID.isInvalid() && "MainFileID already set!"); |
| 576 | MainFileID = createFileIDForMemBuffer(Buffer); |
| 577 | return MainFileID; |
| 578 | } |
| 579 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 580 | //===--------------------------------------------------------------------===// |
| 581 | // MainFileID creation and querying methods. |
| 582 | //===--------------------------------------------------------------------===// |
| 583 | |
Ted Kremenek | 76edd0e | 2007-12-19 22:29:55 +0000 | [diff] [blame] | 584 | /// getMainFileID - Returns the FileID of the main source file. |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 585 | FileID getMainFileID() const { return MainFileID; } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 586 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 587 | /// createMainFileID - Create the FileID for the main source file. |
Dan Gohman | f155dfa | 2010-08-27 15:44:11 +0000 | [diff] [blame] | 588 | FileID createMainFileID(const FileEntry *SourceFile) { |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 589 | assert(MainFileID.isInvalid() && "MainFileID already set!"); |
Dan Gohman | f155dfa | 2010-08-27 15:44:11 +0000 | [diff] [blame] | 590 | MainFileID = createFileID(SourceFile, SourceLocation(), SrcMgr::C_User); |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 591 | return MainFileID; |
| 592 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 593 | |
Douglas Gregor | 414cb64 | 2010-11-30 05:23:00 +0000 | [diff] [blame] | 594 | /// \brief Set the file ID for the precompiled preamble, which is also the |
| 595 | /// main file. |
| 596 | void SetPreambleFileID(FileID Preamble) { |
| 597 | assert(MainFileID.isInvalid() && "MainFileID already set!"); |
| 598 | MainFileID = Preamble; |
| 599 | } |
| 600 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 601 | //===--------------------------------------------------------------------===// |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 602 | // Methods to create new FileID's and macro expansions. |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 603 | //===--------------------------------------------------------------------===// |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 604 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 605 | /// createFileID - Create a new FileID that represents the specified file |
Peter Collingbourne | d57b7ff | 2011-06-30 16:41:03 +0000 | [diff] [blame] | 606 | /// being #included from the specified IncludePosition. This translates NULL |
| 607 | /// into standard input. |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 608 | FileID createFileID(const FileEntry *SourceFile, SourceLocation IncludePos, |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 609 | SrcMgr::CharacteristicKind FileCharacter, |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 610 | int LoadedID = 0, unsigned LoadedOffset = 0) { |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 611 | const SrcMgr::ContentCache *IR = getOrCreateContentCache(SourceFile); |
Dan Gohman | 0d06e99 | 2010-10-26 20:47:28 +0000 | [diff] [blame] | 612 | assert(IR && "getOrCreateContentCache() cannot return NULL"); |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 613 | return createFileID(IR, IncludePos, FileCharacter, LoadedID, LoadedOffset); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 614 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 615 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 616 | /// createFileIDForMemBuffer - Create a new FileID that represents the |
| 617 | /// specified memory buffer. This does no caching of the buffer and takes |
| 618 | /// ownership of the MemoryBuffer, so only pass a MemoryBuffer to this once. |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 619 | FileID createFileIDForMemBuffer(const llvm::MemoryBuffer *Buffer, |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 620 | int LoadedID = 0, unsigned LoadedOffset = 0) { |
Nico Weber | 7bfaaae | 2008-08-10 19:59:06 +0000 | [diff] [blame] | 621 | return createFileID(createMemBufferContentCache(Buffer), SourceLocation(), |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 622 | SrcMgr::C_User, LoadedID, LoadedOffset); |
Ted Kremenek | 1036b68 | 2007-12-19 23:48:45 +0000 | [diff] [blame] | 623 | } |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 624 | |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 625 | /// createMacroArgExpansionLoc - Return a new SourceLocation that encodes the |
| 626 | /// fact that a token from SpellingLoc should actually be referenced from |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 627 | /// ExpansionLoc, and that it represents the expansion of a macro argument |
| 628 | /// into the function-like macro body. |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 629 | SourceLocation createMacroArgExpansionLoc(SourceLocation Loc, |
| 630 | SourceLocation ExpansionLoc, |
| 631 | unsigned TokLength); |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 632 | |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 633 | /// createExpansionLoc - Return a new SourceLocation that encodes the fact |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 634 | /// that a token from SpellingLoc should actually be referenced from |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 635 | /// ExpansionLoc. |
| 636 | SourceLocation createExpansionLoc(SourceLocation Loc, |
| 637 | SourceLocation ExpansionLocStart, |
| 638 | SourceLocation ExpansionLocEnd, |
| 639 | unsigned TokLength, |
| 640 | int LoadedID = 0, |
| 641 | unsigned LoadedOffset = 0); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 642 | |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 643 | /// \brief Retrieve the memory buffer associated with the given file. |
Douglas Gregor | 50f6af7 | 2010-03-16 05:20:39 +0000 | [diff] [blame] | 644 | /// |
| 645 | /// \param Invalid If non-NULL, will be set \c true if an error |
| 646 | /// occurs while retrieving the memory buffer. |
| 647 | const llvm::MemoryBuffer *getMemoryBufferForFile(const FileEntry *File, |
| 648 | bool *Invalid = 0); |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 649 | |
| 650 | /// \brief Override the contents of the given source file by providing an |
| 651 | /// already-allocated buffer. |
| 652 | /// |
Dan Gohman | afbf5f8 | 2010-08-26 02:27:03 +0000 | [diff] [blame] | 653 | /// \param SourceFile the source file whose contents will be overriden. |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 654 | /// |
| 655 | /// \param Buffer the memory buffer whose contents will be used as the |
| 656 | /// data in the given source file. |
| 657 | /// |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 658 | /// \param DoNotFree If true, then the buffer will not be freed when the |
| 659 | /// source manager is destroyed. |
Dan Gohman | 0d06e99 | 2010-10-26 20:47:28 +0000 | [diff] [blame] | 660 | void overrideFileContents(const FileEntry *SourceFile, |
Douglas Gregor | f4f6c9d | 2010-07-26 21:36:20 +0000 | [diff] [blame] | 661 | const llvm::MemoryBuffer *Buffer, |
| 662 | bool DoNotFree = false); |
Douglas Gregor | 2968442 | 2009-12-02 06:49:09 +0000 | [diff] [blame] | 663 | |
Argyrios Kyrtzidis | b1c8649 | 2011-03-05 01:03:53 +0000 | [diff] [blame] | 664 | /// \brief Override the the given source file with another one. |
| 665 | /// |
| 666 | /// \param SourceFile the source file which will be overriden. |
| 667 | /// |
| 668 | /// \param NewFile the file whose contents will be used as the |
| 669 | /// data instead of the contents of the given source file. |
| 670 | void overrideFileContents(const FileEntry *SourceFile, |
| 671 | const FileEntry *NewFile); |
| 672 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 673 | //===--------------------------------------------------------------------===// |
| 674 | // FileID manipulation methods. |
| 675 | //===--------------------------------------------------------------------===// |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 676 | |
Daniel Dunbar | 2ffb14f | 2009-12-06 09:19:25 +0000 | [diff] [blame] | 677 | /// getBuffer - Return the buffer for the specified FileID. If there is an |
| 678 | /// error opening this buffer the first time, this manufactures a temporary |
| 679 | /// buffer and returns a non-empty error string. |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 680 | const llvm::MemoryBuffer *getBuffer(FileID FID, SourceLocation Loc, |
| 681 | bool *Invalid = 0) const { |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 682 | bool MyInvalid = false; |
| 683 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &MyInvalid); |
| 684 | if (MyInvalid || !Entry.isFile()) { |
| 685 | if (Invalid) |
| 686 | *Invalid = true; |
| 687 | |
| 688 | return getFakeBufferForRecovery(); |
| 689 | } |
| 690 | |
| 691 | return Entry.getFile().getContentCache()->getBuffer(Diag, *this, Loc, |
| 692 | Invalid); |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 693 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 694 | |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 695 | const llvm::MemoryBuffer *getBuffer(FileID FID, bool *Invalid = 0) const { |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 696 | bool MyInvalid = false; |
| 697 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &MyInvalid); |
| 698 | if (MyInvalid || !Entry.isFile()) { |
| 699 | if (Invalid) |
| 700 | *Invalid = true; |
| 701 | |
| 702 | return getFakeBufferForRecovery(); |
| 703 | } |
| 704 | |
| 705 | return Entry.getFile().getContentCache()->getBuffer(Diag, *this, |
| 706 | SourceLocation(), |
| 707 | Invalid); |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 708 | } |
| 709 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 710 | /// getFileEntryForID - Returns the FileEntry record for the provided FileID. |
| 711 | const FileEntry *getFileEntryForID(FileID FID) const { |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 712 | bool MyInvalid = false; |
| 713 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &MyInvalid); |
| 714 | if (MyInvalid || !Entry.isFile()) |
| 715 | return 0; |
| 716 | |
| 717 | return Entry.getFile().getContentCache()->OrigEntry; |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 718 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 719 | |
Ted Kremenek | 9d5a165 | 2011-03-23 02:16:44 +0000 | [diff] [blame] | 720 | /// Returns the FileEntry record for the provided SLocEntry. |
| 721 | const FileEntry *getFileEntryForSLocEntry(const SrcMgr::SLocEntry &sloc) const |
| 722 | { |
| 723 | return sloc.getFile().getContentCache()->OrigEntry; |
| 724 | } |
| 725 | |
Benjamin Kramer | ceafc4b | 2010-03-16 14:48:07 +0000 | [diff] [blame] | 726 | /// getBufferData - Return a StringRef to the source buffer data for the |
| 727 | /// specified FileID. |
| 728 | /// |
Douglas Gregor | f715ca1 | 2010-03-16 00:06:06 +0000 | [diff] [blame] | 729 | /// \param FID The file ID whose contents will be returned. |
| 730 | /// \param Invalid If non-NULL, will be set true if an error occurred. |
Chris Lattner | 686775d | 2011-07-20 06:58:45 +0000 | [diff] [blame] | 731 | StringRef getBufferData(FileID FID, bool *Invalid = 0) const; |
Benjamin Kramer | f6ac97b | 2010-03-16 14:14:31 +0000 | [diff] [blame] | 732 | |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 733 | /// \brief Get the number of FileIDs (files and macros) that were created |
| 734 | /// during preprocessing of \arg FID, including it. |
| 735 | unsigned getNumCreatedFIDsForFileID(FileID FID) const { |
| 736 | bool Invalid = false; |
| 737 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &Invalid); |
| 738 | if (Invalid || !Entry.isFile()) |
| 739 | return 0; |
| 740 | |
| 741 | return Entry.getFile().NumCreatedFIDs; |
| 742 | } |
| 743 | |
| 744 | /// \brief Set the number of FileIDs (files and macros) that were created |
| 745 | /// during preprocessing of \arg FID, including it. |
| 746 | void setNumCreatedFIDsForFileID(FileID FID, unsigned NumFIDs) const { |
| 747 | bool Invalid = false; |
| 748 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &Invalid); |
| 749 | if (Invalid || !Entry.isFile()) |
| 750 | return; |
| 751 | |
| 752 | assert(Entry.getFile().NumCreatedFIDs == 0 && "Already set!"); |
| 753 | const_cast<SrcMgr::FileInfo &>(Entry.getFile()).NumCreatedFIDs = NumFIDs; |
| 754 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 755 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 756 | //===--------------------------------------------------------------------===// |
| 757 | // SourceLocation manipulation methods. |
| 758 | //===--------------------------------------------------------------------===// |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 759 | |
Chris Lattner | 668ab1a | 2009-03-13 01:05:57 +0000 | [diff] [blame] | 760 | /// getFileID - Return the FileID for a SourceLocation. This is a very |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 761 | /// hot method that is used for all SourceManager queries that start with a |
| 762 | /// SourceLocation object. It is responsible for finding the entry in |
| 763 | /// SLocEntryTable which contains the specified location. |
| 764 | /// |
| 765 | FileID getFileID(SourceLocation SpellingLoc) const { |
| 766 | unsigned SLocOffset = SpellingLoc.getOffset(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 767 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 768 | // If our one-entry cache covers this offset, just return it. |
| 769 | if (isOffsetInFileID(LastFileIDLookup, SLocOffset)) |
| 770 | return LastFileIDLookup; |
| 771 | |
| 772 | return getFileIDSlow(SLocOffset); |
| 773 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 774 | |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 775 | /// getLocForStartOfFile - Return the source location corresponding to the |
| 776 | /// first byte of the specified file. |
| 777 | SourceLocation getLocForStartOfFile(FileID FID) const { |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 778 | bool Invalid = false; |
| 779 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &Invalid); |
| 780 | if (Invalid || !Entry.isFile()) |
| 781 | return SourceLocation(); |
| 782 | |
| 783 | unsigned FileOffset = Entry.getOffset(); |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 784 | return SourceLocation::getFileLoc(FileOffset); |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 785 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 786 | |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 787 | /// \brief Returns the include location if \arg FID is a #include'd file |
| 788 | /// otherwise it returns an invalid location. |
| 789 | SourceLocation getIncludeLoc(FileID FID) const { |
| 790 | bool Invalid = false; |
| 791 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID, &Invalid); |
| 792 | if (Invalid || !Entry.isFile()) |
| 793 | return SourceLocation(); |
| 794 | |
| 795 | return Entry.getFile().getIncludeLoc(); |
| 796 | } |
| 797 | |
Chandler Carruth | 4027853 | 2011-07-25 16:49:02 +0000 | [diff] [blame] | 798 | /// getExpansionLoc - Given a SourceLocation object, return the expansion |
| 799 | /// location referenced by the ID. |
| 800 | SourceLocation getExpansionLoc(SourceLocation Loc) const { |
Chris Lattner | addb797 | 2009-01-26 20:04:19 +0000 | [diff] [blame] | 801 | // Handle the non-mapped case inline, defer to out of line code to handle |
Chandler Carruth | 4027853 | 2011-07-25 16:49:02 +0000 | [diff] [blame] | 802 | // expansions. |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 803 | if (Loc.isFileID()) return Loc; |
Chandler Carruth | f84ef95 | 2011-07-25 20:52:26 +0000 | [diff] [blame] | 804 | return getExpansionLocSlowCase(Loc); |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 805 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 806 | |
Chandler Carruth | 999f739 | 2011-07-25 20:52:21 +0000 | [diff] [blame] | 807 | /// getImmediateExpansionRange - Loc is required to be an expansion location. |
| 808 | /// Return the start/end of the expansion information. |
Chris Lattner | e7fb484 | 2009-02-15 20:52:18 +0000 | [diff] [blame] | 809 | std::pair<SourceLocation,SourceLocation> |
Chandler Carruth | 999f739 | 2011-07-25 20:52:21 +0000 | [diff] [blame] | 810 | getImmediateExpansionRange(SourceLocation Loc) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 811 | |
Chandler Carruth | edc3dcc | 2011-07-25 16:56:02 +0000 | [diff] [blame] | 812 | /// getExpansionRange - Given a SourceLocation object, return the range of |
| 813 | /// tokens covered by the expansion the ultimate file. |
Chris Lattner | 6678133 | 2009-02-15 21:26:50 +0000 | [diff] [blame] | 814 | std::pair<SourceLocation,SourceLocation> |
Chandler Carruth | edc3dcc | 2011-07-25 16:56:02 +0000 | [diff] [blame] | 815 | getExpansionRange(SourceLocation Loc) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 816 | |
| 817 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 818 | /// getSpellingLoc - Given a SourceLocation object, return the spelling |
| 819 | /// location referenced by the ID. This is the place where the characters |
| 820 | /// that make up the lexed token can be found. |
| 821 | SourceLocation getSpellingLoc(SourceLocation Loc) const { |
Chris Lattner | addb797 | 2009-01-26 20:04:19 +0000 | [diff] [blame] | 822 | // Handle the non-mapped case inline, defer to out of line code to handle |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 823 | // expansions. |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 824 | if (Loc.isFileID()) return Loc; |
Chris Lattner | addb797 | 2009-01-26 20:04:19 +0000 | [diff] [blame] | 825 | return getSpellingLocSlowCase(Loc); |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 826 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 827 | |
Chris Lattner | 387616e | 2009-02-17 08:04:48 +0000 | [diff] [blame] | 828 | /// getImmediateSpellingLoc - Given a SourceLocation object, return the |
| 829 | /// spelling location referenced by the ID. This is the first level down |
| 830 | /// towards the place where the characters that make up the lexed token can be |
| 831 | /// found. This should not generally be used by clients. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 832 | SourceLocation getImmediateSpellingLoc(SourceLocation Loc) const; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 833 | |
| 834 | /// getDecomposedLoc - Decompose the specified location into a raw FileID + |
| 835 | /// Offset pair. The first element is the FileID, the second is the |
| 836 | /// offset from the start of the buffer of the location. |
| 837 | std::pair<FileID, unsigned> getDecomposedLoc(SourceLocation Loc) const { |
| 838 | FileID FID = getFileID(Loc); |
| 839 | return std::make_pair(FID, Loc.getOffset()-getSLocEntry(FID).getOffset()); |
| 840 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 841 | |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 842 | /// getDecomposedExpansionLoc - Decompose the specified location into a raw |
| 843 | /// FileID + Offset pair. If the location is an expansion record, walk |
| 844 | /// through it until we find the final location expanded. |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 845 | std::pair<FileID, unsigned> |
Chandler Carruth | e7b2b6e | 2011-07-25 20:52:32 +0000 | [diff] [blame] | 846 | getDecomposedExpansionLoc(SourceLocation Loc) const { |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 847 | FileID FID = getFileID(Loc); |
| 848 | const SrcMgr::SLocEntry *E = &getSLocEntry(FID); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 849 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 850 | unsigned Offset = Loc.getOffset()-E->getOffset(); |
| 851 | if (Loc.isFileID()) |
| 852 | return std::make_pair(FID, Offset); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 853 | |
Chandler Carruth | e7b2b6e | 2011-07-25 20:52:32 +0000 | [diff] [blame] | 854 | return getDecomposedExpansionLocSlowCase(E); |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 855 | } |
| 856 | |
| 857 | /// getDecomposedSpellingLoc - Decompose the specified location into a raw |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 858 | /// FileID + Offset pair. If the location is an expansion record, walk |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 859 | /// through it until we find its spelling record. |
| 860 | std::pair<FileID, unsigned> |
| 861 | getDecomposedSpellingLoc(SourceLocation Loc) const { |
| 862 | FileID FID = getFileID(Loc); |
| 863 | const SrcMgr::SLocEntry *E = &getSLocEntry(FID); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 864 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 865 | unsigned Offset = Loc.getOffset()-E->getOffset(); |
| 866 | if (Loc.isFileID()) |
| 867 | return std::make_pair(FID, Offset); |
| 868 | return getDecomposedSpellingLocSlowCase(E, Offset); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 869 | } |
| 870 | |
Chris Lattner | 52c2908 | 2009-01-27 06:27:13 +0000 | [diff] [blame] | 871 | /// getFileOffset - This method returns the offset from the start |
| 872 | /// of the file that the specified SourceLocation represents. This is not very |
| 873 | /// meaningful for a macro ID. |
| 874 | unsigned getFileOffset(SourceLocation SpellingLoc) const { |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 875 | return getDecomposedLoc(SpellingLoc).second; |
| 876 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 877 | |
Chandler Carruth | 96d3589 | 2011-07-26 03:03:00 +0000 | [diff] [blame] | 878 | /// isMacroArgExpansion - This method tests whether the given source location |
| 879 | /// represents a macro argument's expansion into the function-like macro |
| 880 | /// definition. Such source locations only appear inside of the expansion |
| 881 | /// locations representing where a particular function-like macro was |
| 882 | /// expanded. |
| 883 | bool isMacroArgExpansion(SourceLocation Loc) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 884 | |
Argyrios Kyrtzidis | 499ea55 | 2011-08-23 21:02:38 +0000 | [diff] [blame] | 885 | /// \brief Returns true if \arg Loc is inside the [\arg Start, +\arg Length) |
| 886 | /// chunk of the source location address space. |
| 887 | /// If it's true and \arg RelativeOffset is non-null, it will be set to the |
| 888 | /// relative offset of \arg Loc inside the chunk. |
| 889 | bool isInSLocAddrSpace(SourceLocation Loc, |
| 890 | SourceLocation Start, unsigned Length, |
| 891 | unsigned *RelativeOffset = 0) const { |
| 892 | assert(((Start.getOffset() < NextLocalOffset && |
| 893 | Start.getOffset()+Length <= NextLocalOffset) || |
| 894 | (Start.getOffset() >= CurrentLoadedOffset && |
| 895 | Start.getOffset()+Length < MaxLoadedOffset)) && |
| 896 | "Chunk is not valid SLoc address space"); |
| 897 | unsigned LocOffs = Loc.getOffset(); |
| 898 | unsigned BeginOffs = Start.getOffset(); |
| 899 | unsigned EndOffs = BeginOffs + Length; |
| 900 | if (LocOffs >= BeginOffs && LocOffs < EndOffs) { |
| 901 | if (RelativeOffset) |
| 902 | *RelativeOffset = LocOffs - BeginOffs; |
| 903 | return true; |
| 904 | } |
| 905 | |
| 906 | return false; |
| 907 | } |
| 908 | |
Argyrios Kyrtzidis | b6c465e | 2011-08-23 21:02:41 +0000 | [diff] [blame^] | 909 | /// \brief Return true if both \arg LHS and \arg RHS are in the local source |
| 910 | /// location address space or the loaded one. If it's true and |
| 911 | /// \arg RelativeOffset is non-null, it will be set to the offset of \arg RHS |
| 912 | /// relative to \arg LHS. |
| 913 | bool isInSameSLocAddrSpace(SourceLocation LHS, SourceLocation RHS, |
| 914 | int *RelativeOffset) const { |
| 915 | unsigned LHSOffs = LHS.getOffset(), RHSOffs = RHS.getOffset(); |
| 916 | bool LHSLoaded = LHSOffs >= CurrentLoadedOffset; |
| 917 | bool RHSLoaded = RHSOffs >= CurrentLoadedOffset; |
| 918 | |
| 919 | if (LHSLoaded == RHSLoaded) { |
| 920 | if (RelativeOffset) |
| 921 | *RelativeOffset = RHSOffs - LHSOffs; |
| 922 | return true; |
| 923 | } |
| 924 | |
| 925 | return false; |
| 926 | } |
| 927 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 928 | //===--------------------------------------------------------------------===// |
| 929 | // Queries about the code at a SourceLocation. |
| 930 | //===--------------------------------------------------------------------===// |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 931 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 932 | /// getCharacterData - Return a pointer to the start of the specified location |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 933 | /// in the appropriate spelling MemoryBuffer. |
Douglas Gregor | 50f6af7 | 2010-03-16 05:20:39 +0000 | [diff] [blame] | 934 | /// |
| 935 | /// \param Invalid If non-NULL, will be set \c true if an error occurs. |
| 936 | const char *getCharacterData(SourceLocation SL, bool *Invalid = 0) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 937 | |
Chris Lattner | 9dc1f53 | 2007-07-20 16:37:10 +0000 | [diff] [blame] | 938 | /// getColumnNumber - Return the column # for the specified file position. |
| 939 | /// This is significantly cheaper to compute than the line number. This |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 940 | /// returns zero if the column number isn't known. This may only be called |
| 941 | /// on a file sloc, so you must choose a spelling or expansion location |
Chris Lattner | f7cf85b | 2009-01-16 07:36:28 +0000 | [diff] [blame] | 942 | /// before calling this method. |
Douglas Gregor | 50f6af7 | 2010-03-16 05:20:39 +0000 | [diff] [blame] | 943 | unsigned getColumnNumber(FileID FID, unsigned FilePos, |
| 944 | bool *Invalid = 0) const; |
Chandler Carruth | 5ef04ee | 2011-02-23 00:47:48 +0000 | [diff] [blame] | 945 | unsigned getSpellingColumnNumber(SourceLocation Loc, bool *Invalid = 0) const; |
Chandler Carruth | a77c031 | 2011-07-25 20:57:57 +0000 | [diff] [blame] | 946 | unsigned getExpansionColumnNumber(SourceLocation Loc, |
Chandler Carruth | b49dcd2 | 2011-07-25 20:59:15 +0000 | [diff] [blame] | 947 | bool *Invalid = 0) const; |
Chandler Carruth | 5ef04ee | 2011-02-23 00:47:48 +0000 | [diff] [blame] | 948 | unsigned getPresumedColumnNumber(SourceLocation Loc, bool *Invalid = 0) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 949 | |
| 950 | |
Chris Lattner | df7c17a | 2009-01-16 07:00:02 +0000 | [diff] [blame] | 951 | /// getLineNumber - Given a SourceLocation, return the spelling line number |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 952 | /// for the position indicated. This requires building and caching a table of |
| 953 | /// line offsets for the MemoryBuffer, so this is not cheap: use only when |
| 954 | /// about to emit a diagnostic. |
Douglas Gregor | 50f6af7 | 2010-03-16 05:20:39 +0000 | [diff] [blame] | 955 | unsigned getLineNumber(FileID FID, unsigned FilePos, bool *Invalid = 0) const; |
Chandler Carruth | 5ef04ee | 2011-02-23 00:47:48 +0000 | [diff] [blame] | 956 | unsigned getSpellingLineNumber(SourceLocation Loc, bool *Invalid = 0) const; |
Chandler Carruth | 6421162 | 2011-07-25 21:09:52 +0000 | [diff] [blame] | 957 | unsigned getExpansionLineNumber(SourceLocation Loc, bool *Invalid = 0) const; |
Chandler Carruth | 5ef04ee | 2011-02-23 00:47:48 +0000 | [diff] [blame] | 958 | unsigned getPresumedLineNumber(SourceLocation Loc, bool *Invalid = 0) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 959 | |
Chris Lattner | bff5c51 | 2009-02-17 08:39:06 +0000 | [diff] [blame] | 960 | /// Return the filename or buffer identifier of the buffer the location is in. |
| 961 | /// Note that this name does not respect #line directives. Use getPresumedLoc |
| 962 | /// for normal clients. |
Douglas Gregor | 50f6af7 | 2010-03-16 05:20:39 +0000 | [diff] [blame] | 963 | const char *getBufferName(SourceLocation Loc, bool *Invalid = 0) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 964 | |
Chris Lattner | 6b30667 | 2009-02-04 05:33:01 +0000 | [diff] [blame] | 965 | /// getFileCharacteristic - return the file characteristic of the specified |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 966 | /// source location, indicating whether this is a normal file, a system |
Chris Lattner | 6b30667 | 2009-02-04 05:33:01 +0000 | [diff] [blame] | 967 | /// header, or an "implicit extern C" system header. |
| 968 | /// |
| 969 | /// This state can be modified with flags on GNU linemarker directives like: |
| 970 | /// # 4 "foo.h" 3 |
| 971 | /// which changes all source locations in the current file after that to be |
| 972 | /// considered to be from a system header. |
| 973 | SrcMgr::CharacteristicKind getFileCharacteristic(SourceLocation Loc) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 974 | |
Chris Lattner | b9c3f96 | 2009-01-27 07:57:44 +0000 | [diff] [blame] | 975 | /// getPresumedLoc - This method returns the "presumed" location of a |
| 976 | /// SourceLocation specifies. A "presumed location" can be modified by #line |
| 977 | /// or GNU line marker directives. This provides a view on the data that a |
| 978 | /// user should see in diagnostics, for example. |
| 979 | /// |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 980 | /// Note that a presumed location is always given as the expansion point of |
| 981 | /// an expansion location, not at the spelling location. |
Douglas Gregor | cb7b1e1 | 2010-11-12 07:15:47 +0000 | [diff] [blame] | 982 | /// |
| 983 | /// \returns The presumed location of the specified SourceLocation. If the |
| 984 | /// presumed location cannot be calculate (e.g., because \p Loc is invalid |
| 985 | /// or the file containing \p Loc has changed on disk), returns an invalid |
| 986 | /// presumed location. |
Chris Lattner | b9c3f96 | 2009-01-27 07:57:44 +0000 | [diff] [blame] | 987 | PresumedLoc getPresumedLoc(SourceLocation Loc) const; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 988 | |
Ted Kremenek | 9fd87b1 | 2008-04-14 21:04:18 +0000 | [diff] [blame] | 989 | /// isFromSameFile - Returns true if both SourceLocations correspond to |
| 990 | /// the same file. |
| 991 | bool isFromSameFile(SourceLocation Loc1, SourceLocation Loc2) const { |
Chris Lattner | a11d617 | 2009-01-19 07:46:45 +0000 | [diff] [blame] | 992 | return getFileID(Loc1) == getFileID(Loc2); |
Ted Kremenek | 9fd87b1 | 2008-04-14 21:04:18 +0000 | [diff] [blame] | 993 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 994 | |
Ted Kremenek | 9fd87b1 | 2008-04-14 21:04:18 +0000 | [diff] [blame] | 995 | /// isFromMainFile - Returns true if the file of provided SourceLocation is |
| 996 | /// the main file. |
| 997 | bool isFromMainFile(SourceLocation Loc) const { |
Chris Lattner | a11d617 | 2009-01-19 07:46:45 +0000 | [diff] [blame] | 998 | return getFileID(Loc) == getMainFileID(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 999 | } |
| 1000 | |
Nico Weber | 7bfaaae | 2008-08-10 19:59:06 +0000 | [diff] [blame] | 1001 | /// isInSystemHeader - Returns if a SourceLocation is in a system header. |
| 1002 | bool isInSystemHeader(SourceLocation Loc) const { |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 1003 | return getFileCharacteristic(Loc) != SrcMgr::C_User; |
Nico Weber | 7bfaaae | 2008-08-10 19:59:06 +0000 | [diff] [blame] | 1004 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1005 | |
Chris Lattner | 0d45658 | 2009-06-13 23:31:51 +0000 | [diff] [blame] | 1006 | /// isInExternCSystemHeader - Returns if a SourceLocation is in an "extern C" |
| 1007 | /// system header. |
| 1008 | bool isInExternCSystemHeader(SourceLocation Loc) const { |
| 1009 | return getFileCharacteristic(Loc) == SrcMgr::C_ExternCSystem; |
| 1010 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1011 | |
Argyrios Kyrtzidis | 54232ad | 2011-08-19 22:34:01 +0000 | [diff] [blame] | 1012 | /// \brief The size of the SLocEnty that \arg FID represents. |
Argyrios Kyrtzidis | 984e42c | 2011-08-23 21:02:28 +0000 | [diff] [blame] | 1013 | unsigned getFileIDSize(FileID FID) const; |
Argyrios Kyrtzidis | 54232ad | 2011-08-19 22:34:01 +0000 | [diff] [blame] | 1014 | |
Argyrios Kyrtzidis | d60a34a | 2011-08-19 22:34:17 +0000 | [diff] [blame] | 1015 | /// \brief Given a specific FileID, returns true if \arg Loc is inside that |
| 1016 | /// FileID chunk and sets relative offset (offset of \arg Loc from beginning |
| 1017 | /// of FileID) to \arg relativeOffset. |
| 1018 | bool isInFileID(SourceLocation Loc, FileID FID, |
| 1019 | unsigned *RelativeOffset = 0) const { |
| 1020 | return isInFileID(Loc, FID, 0, getFileIDSize(FID), RelativeOffset); |
| 1021 | } |
| 1022 | |
Argyrios Kyrtzidis | b73377e | 2011-07-07 03:40:34 +0000 | [diff] [blame] | 1023 | /// \brief Given a specific chunk of a FileID (FileID with offset+length), |
| 1024 | /// returns true if \arg Loc is inside that chunk and sets relative offset |
| 1025 | /// (offset of \arg Loc from beginning of chunk) to \arg relativeOffset. |
| 1026 | bool isInFileID(SourceLocation Loc, |
| 1027 | FileID FID, unsigned offset, unsigned length, |
Argyrios Kyrtzidis | 984e42c | 2011-08-23 21:02:28 +0000 | [diff] [blame] | 1028 | unsigned *relativeOffset = 0) const; |
Argyrios Kyrtzidis | 469244a | 2011-05-28 03:56:11 +0000 | [diff] [blame] | 1029 | |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 1030 | //===--------------------------------------------------------------------===// |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 1031 | // Line Table Manipulation Routines |
| 1032 | //===--------------------------------------------------------------------===// |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1033 | |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 1034 | /// getLineTableFilenameID - Return the uniqued ID for the specified filename. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1035 | /// |
Chris Lattner | 686775d | 2011-07-20 06:58:45 +0000 | [diff] [blame] | 1036 | unsigned getLineTableFilenameID(StringRef Str); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1037 | |
Chris Lattner | 4c4ea17 | 2009-02-03 21:52:55 +0000 | [diff] [blame] | 1038 | /// AddLineNote - Add a line note to the line table for the FileID and offset |
| 1039 | /// specified by Loc. If FilenameID is -1, it is considered to be |
| 1040 | /// unspecified. |
| 1041 | void AddLineNote(SourceLocation Loc, unsigned LineNo, int FilenameID); |
Chris Lattner | 9d79eba | 2009-02-04 05:21:58 +0000 | [diff] [blame] | 1042 | void AddLineNote(SourceLocation Loc, unsigned LineNo, int FilenameID, |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1043 | bool IsFileEntry, bool IsFileExit, |
Chris Lattner | 9d79eba | 2009-02-04 05:21:58 +0000 | [diff] [blame] | 1044 | bool IsSystemHeader, bool IsExternCHeader); |
Douglas Gregor | bd94500 | 2009-04-13 16:31:14 +0000 | [diff] [blame] | 1045 | |
| 1046 | /// \brief Determine if the source manager has a line table. |
| 1047 | bool hasLineTable() const { return LineTable != 0; } |
| 1048 | |
| 1049 | /// \brief Retrieve the stored line table. |
| 1050 | LineTableInfo &getLineTable(); |
| 1051 | |
Chris Lattner | 5b9a504 | 2009-01-26 07:57:50 +0000 | [diff] [blame] | 1052 | //===--------------------------------------------------------------------===// |
Ted Kremenek | 457aaf0 | 2011-04-28 04:10:31 +0000 | [diff] [blame] | 1053 | // Queries for performance analysis. |
| 1054 | //===--------------------------------------------------------------------===// |
| 1055 | |
| 1056 | /// Return the total amount of physical memory allocated by the |
| 1057 | /// ContentCache allocator. |
| 1058 | size_t getContentCacheSize() const { |
| 1059 | return ContentCacheAlloc.getTotalMemory(); |
| 1060 | } |
Ted Kremenek | f61b831 | 2011-04-28 20:36:42 +0000 | [diff] [blame] | 1061 | |
| 1062 | struct MemoryBufferSizes { |
| 1063 | const size_t malloc_bytes; |
| 1064 | const size_t mmap_bytes; |
| 1065 | |
| 1066 | MemoryBufferSizes(size_t malloc_bytes, size_t mmap_bytes) |
| 1067 | : malloc_bytes(malloc_bytes), mmap_bytes(mmap_bytes) {} |
| 1068 | }; |
| 1069 | |
| 1070 | /// Return the amount of memory used by memory buffers, breaking down |
| 1071 | /// by heap-backed versus mmap'ed memory. |
| 1072 | MemoryBufferSizes getMemoryBufferSizes() const; |
Ted Kremenek | ca7dc2b | 2011-07-26 23:46:06 +0000 | [diff] [blame] | 1073 | |
| 1074 | // Return the amount of memory used for various side tables and |
| 1075 | // data structures in the SourceManager. |
| 1076 | size_t getDataStructureSizes() const; |
Ted Kremenek | 457aaf0 | 2011-04-28 04:10:31 +0000 | [diff] [blame] | 1077 | |
| 1078 | //===--------------------------------------------------------------------===// |
Chris Lattner | 06a062d | 2009-01-19 08:02:45 +0000 | [diff] [blame] | 1079 | // Other miscellaneous methods. |
| 1080 | //===--------------------------------------------------------------------===// |
Argyrios Kyrtzidis | 10b46d2 | 2009-06-20 08:09:57 +0000 | [diff] [blame] | 1081 | |
| 1082 | /// \brief Get the source location for the given file:line:col triplet. |
| 1083 | /// |
| 1084 | /// If the source file is included multiple times, the source location will |
| 1085 | /// be based upon the first inclusion. |
Argyrios Kyrtzidis | ac836e4 | 2011-08-17 00:31:20 +0000 | [diff] [blame] | 1086 | /// |
| 1087 | /// If the location points inside a function macro argument, the returned |
| 1088 | /// location will be the macro location in which the argument was expanded. |
| 1089 | /// \sa getMacroArgExpandedLocation |
Argyrios Kyrtzidis | 10b46d2 | 2009-06-20 08:09:57 +0000 | [diff] [blame] | 1090 | SourceLocation getLocation(const FileEntry *SourceFile, |
Argyrios Kyrtzidis | ac836e4 | 2011-08-17 00:31:20 +0000 | [diff] [blame] | 1091 | unsigned Line, unsigned Col) { |
| 1092 | SourceLocation Loc = translateFileLineCol(SourceFile, Line, Col); |
| 1093 | return getMacroArgExpandedLocation(Loc); |
| 1094 | } |
| 1095 | |
| 1096 | /// \brief Get the source location for the given file:line:col triplet. |
| 1097 | /// |
| 1098 | /// If the source file is included multiple times, the source location will |
| 1099 | /// be based upon the first inclusion. |
| 1100 | SourceLocation translateFileLineCol(const FileEntry *SourceFile, |
| 1101 | unsigned Line, unsigned Col); |
| 1102 | |
| 1103 | /// \brief If \arg Loc points inside a function macro argument, the returned |
| 1104 | /// location will be the macro location in which the argument was expanded. |
| 1105 | /// If a macro argument is used multiple times, the expanded location will |
| 1106 | /// be at the first expansion of the argument. |
| 1107 | /// e.g. |
| 1108 | /// MY_MACRO(foo); |
| 1109 | /// ^ |
| 1110 | /// Passing a file location pointing at 'foo', will yield a macro location |
| 1111 | /// where 'foo' was expanded into. |
| 1112 | SourceLocation getMacroArgExpandedLocation(SourceLocation Loc); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1113 | |
Argyrios Kyrtzidis | 2aa03d5 | 2009-06-23 22:01:48 +0000 | [diff] [blame] | 1114 | /// \brief Determines the order of 2 source locations in the translation unit. |
| 1115 | /// |
| 1116 | /// \returns true if LHS source location comes before RHS, false otherwise. |
| 1117 | bool isBeforeInTranslationUnit(SourceLocation LHS, SourceLocation RHS) const; |
| 1118 | |
Argyrios Kyrtzidis | b73377e | 2011-07-07 03:40:34 +0000 | [diff] [blame] | 1119 | /// \brief Determines the order of 2 source locations in the "source location |
| 1120 | /// address space". |
Argyrios Kyrtzidis | 5d579e7 | 2011-08-23 21:02:35 +0000 | [diff] [blame] | 1121 | bool isBeforeInSLocAddrSpace(SourceLocation LHS, SourceLocation RHS) const { |
| 1122 | return isBeforeInSLocAddrSpace(LHS, RHS.getOffset()); |
Argyrios Kyrtzidis | b73377e | 2011-07-07 03:40:34 +0000 | [diff] [blame] | 1123 | } |
| 1124 | |
| 1125 | /// \brief Determines the order of a source location and a source location |
| 1126 | /// offset in the "source location address space". |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1127 | /// |
| 1128 | /// Note that we always consider source locations loaded from |
Argyrios Kyrtzidis | 5d579e7 | 2011-08-23 21:02:35 +0000 | [diff] [blame] | 1129 | bool isBeforeInSLocAddrSpace(SourceLocation LHS, unsigned RHS) const { |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1130 | unsigned LHSOffset = LHS.getOffset(); |
| 1131 | bool LHSLoaded = LHSOffset >= CurrentLoadedOffset; |
| 1132 | bool RHSLoaded = RHS >= CurrentLoadedOffset; |
| 1133 | if (LHSLoaded == RHSLoaded) |
Argyrios Kyrtzidis | 5d579e7 | 2011-08-23 21:02:35 +0000 | [diff] [blame] | 1134 | return LHSOffset < RHS; |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1135 | |
| 1136 | return LHSLoaded; |
Argyrios Kyrtzidis | b73377e | 2011-07-07 03:40:34 +0000 | [diff] [blame] | 1137 | } |
| 1138 | |
Chris Lattner | c6fe32a | 2009-01-17 03:48:08 +0000 | [diff] [blame] | 1139 | // Iterators over FileInfos. |
Chris Lattner | 0d0bf8c | 2009-02-03 07:30:45 +0000 | [diff] [blame] | 1140 | typedef llvm::DenseMap<const FileEntry*, SrcMgr::ContentCache*> |
| 1141 | ::const_iterator fileinfo_iterator; |
Chris Lattner | c6fe32a | 2009-01-17 03:48:08 +0000 | [diff] [blame] | 1142 | fileinfo_iterator fileinfo_begin() const { return FileInfos.begin(); } |
| 1143 | fileinfo_iterator fileinfo_end() const { return FileInfos.end(); } |
Douglas Gregor | d93256e | 2010-01-28 06:00:51 +0000 | [diff] [blame] | 1144 | bool hasFileInfo(const FileEntry *File) const { |
| 1145 | return FileInfos.find(File) != FileInfos.end(); |
| 1146 | } |
Chris Lattner | c6fe32a | 2009-01-17 03:48:08 +0000 | [diff] [blame] | 1147 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1148 | /// PrintStats - Print statistics to stderr. |
| 1149 | /// |
| 1150 | void PrintStats() const; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1151 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1152 | /// \brief Get the number of local SLocEntries we have. |
| 1153 | unsigned local_sloc_entry_size() const { return LocalSLocEntryTable.size(); } |
| 1154 | |
| 1155 | /// \brief Get a local SLocEntry. This is exposed for indexing. |
| 1156 | const SrcMgr::SLocEntry &getLocalSLocEntry(unsigned Index, |
| 1157 | bool *Invalid = 0) const { |
| 1158 | assert(Index < LocalSLocEntryTable.size() && "Invalid index"); |
| 1159 | return LocalSLocEntryTable[Index]; |
Douglas Gregor | bdfe48a | 2009-10-16 22:46:09 +0000 | [diff] [blame] | 1160 | } |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1161 | |
| 1162 | /// \brief Get the number of loaded SLocEntries we have. |
| 1163 | unsigned loaded_sloc_entry_size() const { return LoadedSLocEntryTable.size();} |
| 1164 | |
| 1165 | /// \brief Get a loaded SLocEntry. This is exposed for indexing. |
| 1166 | const SrcMgr::SLocEntry &getLoadedSLocEntry(unsigned Index, bool *Invalid=0) const { |
| 1167 | assert(Index < LoadedSLocEntryTable.size() && "Invalid index"); |
| 1168 | if (!SLocEntryLoaded[Index]) |
| 1169 | ExternalSLocEntries->ReadSLocEntry(-(static_cast<int>(Index) + 2)); |
| 1170 | return LoadedSLocEntryTable[Index]; |
| 1171 | } |
| 1172 | |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 1173 | const SrcMgr::SLocEntry &getSLocEntry(FileID FID, bool *Invalid = 0) const { |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1174 | return getSLocEntryByID(FID.ID); |
Douglas Gregor | bd94500 | 2009-04-13 16:31:14 +0000 | [diff] [blame] | 1175 | } |
| 1176 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1177 | unsigned getNextLocalOffset() const { return NextLocalOffset; } |
| 1178 | |
| 1179 | void setExternalSLocEntrySource(ExternalSLocEntrySource *Source) { |
| 1180 | assert(LoadedSLocEntryTable.empty() && |
| 1181 | "Invalidating existing loaded entries"); |
| 1182 | ExternalSLocEntries = Source; |
| 1183 | } |
| 1184 | |
| 1185 | /// \brief Allocate a number of loaded SLocEntries, which will be actually |
| 1186 | /// loaded on demand from the external source. |
| 1187 | /// |
| 1188 | /// NumSLocEntries will be allocated, which occupy a total of TotalSize space |
| 1189 | /// in the global source view. The lowest ID and the base offset of the |
| 1190 | /// entries will be returned. |
| 1191 | std::pair<int, unsigned> |
| 1192 | AllocateLoadedSLocEntries(unsigned NumSLocEntries, unsigned TotalSize); |
| 1193 | |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 1194 | private: |
Douglas Gregor | e23ac65 | 2011-04-20 00:21:03 +0000 | [diff] [blame] | 1195 | const llvm::MemoryBuffer *getFakeBufferForRecovery() const; |
| 1196 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1197 | /// \brief Get the entry with the given unwrapped FileID. |
| 1198 | const SrcMgr::SLocEntry &getSLocEntryByID(int ID) const { |
| 1199 | assert(ID != -1 && "Using FileID sentinel value"); |
| 1200 | if (ID < 0) |
| 1201 | return getLoadedSLocEntryByID(ID); |
| 1202 | return getLocalSLocEntry(static_cast<unsigned>(ID)); |
| 1203 | } |
| 1204 | |
| 1205 | const SrcMgr::SLocEntry &getLoadedSLocEntryByID(int ID) const { |
| 1206 | return getLoadedSLocEntry(static_cast<unsigned>(-ID - 2)); |
| 1207 | } |
| 1208 | |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 1209 | /// createExpansionLoc - Implements the common elements of storing an |
Chandler Carruth | 3201f38 | 2011-07-26 05:17:23 +0000 | [diff] [blame] | 1210 | /// expansion info struct into the SLocEntry table and producing a source |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 1211 | /// location that refers to it. |
Chandler Carruth | 78df836 | 2011-07-26 04:41:47 +0000 | [diff] [blame] | 1212 | SourceLocation createExpansionLocImpl(const SrcMgr::ExpansionInfo &Expansion, |
Chandler Carruth | bf340e4 | 2011-07-26 03:03:05 +0000 | [diff] [blame] | 1213 | unsigned TokLength, |
| 1214 | int LoadedID = 0, |
| 1215 | unsigned LoadedOffset = 0); |
Chandler Carruth | c8d1ecc | 2011-07-07 23:56:36 +0000 | [diff] [blame] | 1216 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1217 | /// isOffsetInFileID - Return true if the specified FileID contains the |
| 1218 | /// specified SourceLocation offset. This is a very hot method. |
| 1219 | inline bool isOffsetInFileID(FileID FID, unsigned SLocOffset) const { |
| 1220 | const SrcMgr::SLocEntry &Entry = getSLocEntry(FID); |
| 1221 | // If the entry is after the offset, it can't contain it. |
| 1222 | if (SLocOffset < Entry.getOffset()) return false; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1223 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1224 | // If this is the very last entry then it does. |
| 1225 | if (FID.ID == -2) |
| 1226 | return true; |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 1227 | |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1228 | // If it is the last local entry, then it does if the location is local. |
| 1229 | if (static_cast<unsigned>(FID.ID+1) == LocalSLocEntryTable.size()) { |
| 1230 | return SLocOffset < NextLocalOffset; |
| 1231 | } |
| 1232 | |
| 1233 | // Otherwise, the entry after it has to not include it. This works for both |
| 1234 | // local and loaded entries. |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 1235 | return SLocOffset < getSLocEntry(FileID::get(FID.ID+1)).getOffset(); |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1236 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1237 | |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 1238 | /// createFileID - Create a new fileID for the specified ContentCache and |
| 1239 | /// include position. This works regardless of whether the ContentCache |
| 1240 | /// corresponds to a file or some other input source. |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 1241 | FileID createFileID(const SrcMgr::ContentCache* File, |
| 1242 | SourceLocation IncludePos, |
Douglas Gregor | 7f94b0b | 2009-04-27 06:38:32 +0000 | [diff] [blame] | 1243 | SrcMgr::CharacteristicKind DirCharacter, |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1244 | int LoadedID, unsigned LoadedOffset); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1245 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1246 | const SrcMgr::ContentCache * |
| 1247 | getOrCreateContentCache(const FileEntry *SourceFile); |
Ted Kremenek | c16c208 | 2009-01-06 01:55:26 +0000 | [diff] [blame] | 1248 | |
Ted Kremenek | 78d85f5 | 2007-10-30 21:08:08 +0000 | [diff] [blame] | 1249 | /// createMemBufferContentCache - Create a new ContentCache for the specified |
| 1250 | /// memory buffer. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1251 | const SrcMgr::ContentCache* |
Chris Lattner | 2b2453a | 2009-01-17 06:22:33 +0000 | [diff] [blame] | 1252 | createMemBufferContentCache(const llvm::MemoryBuffer *Buf); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 1253 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1254 | FileID getFileIDSlow(unsigned SLocOffset) const; |
Douglas Gregor | f62d43d | 2011-07-19 16:10:42 +0000 | [diff] [blame] | 1255 | FileID getFileIDLocal(unsigned SLocOffset) const; |
| 1256 | FileID getFileIDLoaded(unsigned SLocOffset) const; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1257 | |
Chandler Carruth | f84ef95 | 2011-07-25 20:52:26 +0000 | [diff] [blame] | 1258 | SourceLocation getExpansionLocSlowCase(SourceLocation Loc) const; |
Chris Lattner | addb797 | 2009-01-26 20:04:19 +0000 | [diff] [blame] | 1259 | SourceLocation getSpellingLocSlowCase(SourceLocation Loc) const; |
| 1260 | |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1261 | std::pair<FileID, unsigned> |
Chandler Carruth | e7b2b6e | 2011-07-25 20:52:32 +0000 | [diff] [blame] | 1262 | getDecomposedExpansionLocSlowCase(const SrcMgr::SLocEntry *E) const; |
Chris Lattner | de7aeef | 2009-01-26 00:43:02 +0000 | [diff] [blame] | 1263 | std::pair<FileID, unsigned> |
| 1264 | getDecomposedSpellingLocSlowCase(const SrcMgr::SLocEntry *E, |
| 1265 | unsigned Offset) const; |
Argyrios Kyrtzidis | d9d2b67 | 2011-08-21 23:33:04 +0000 | [diff] [blame] | 1266 | void computeMacroArgsCache(SrcMgr::ContentCache *Content, FileID FID); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1267 | }; |
| 1268 | |
| 1269 | |
| 1270 | } // end namespace clang |
| 1271 | |
| 1272 | #endif |