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