mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
aacef757ed
It requires direct access to the SourceLoc internals and is generally useful anyway. Swift SVN r7111
117 lines
3.5 KiB
C++
117 lines
3.5 KiB
C++
//===--- SourceLoc.cpp - SourceLoc and SourceRange implementations --------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/Basic/SourceLoc.h"
|
|
#include "swift/Basic/SourceManager.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace swift;
|
|
|
|
SourceLoc SourceManager::getCodeCompletionLoc() const {
|
|
return getLocForBufferStart(CodeCompletionBufferID)
|
|
.getAdvancedLoc(CodeCompletionOffset);
|
|
}
|
|
|
|
SourceLoc SourceManager::getLocForBufferStart(unsigned BufferID) const {
|
|
auto *Buffer = LLVMSourceMgr.getMemoryBuffer(BufferID);
|
|
return SourceLoc(llvm::SMLoc::getFromPointer(Buffer->getBufferStart()));
|
|
}
|
|
|
|
unsigned SourceManager::getLocOffsetInBuffer(SourceLoc Loc,
|
|
unsigned BufferID) const {
|
|
auto *Buffer = LLVMSourceMgr.getMemoryBuffer(BufferID);
|
|
assert(Loc.Value.getPointer() >= Buffer->getBuffer().begin() &&
|
|
Loc.Value.getPointer() <= Buffer->getBuffer().end() &&
|
|
"Location is not from the specified buffer");
|
|
return Loc.Value.getPointer() - Buffer->getBuffer().begin();
|
|
}
|
|
|
|
DecomposedLoc SourceManager::decompose(SourceLoc Loc) const {
|
|
assert(Loc.isValid());
|
|
|
|
unsigned BufferID =
|
|
unsigned(LLVMSourceMgr.FindBufferContainingLoc(Loc.Value));
|
|
assert(BufferID != ~0U);
|
|
|
|
DecomposedLoc Result;
|
|
Result.Buffer = LLVMSourceMgr.getMemoryBuffer(BufferID);
|
|
Result.Line = LLVMSourceMgr.FindLineNumber(Loc.Value, BufferID);
|
|
|
|
const char *BufferStart = Result.Buffer->getBufferStart();
|
|
const char *LineStart = Loc.Value.getPointer();
|
|
while (LineStart != BufferStart &&
|
|
LineStart[-1] != '\n' && LineStart[-1] != '\r')
|
|
--LineStart;
|
|
Result.Column = Loc.Value.getPointer() - LineStart;
|
|
|
|
return Result;
|
|
}
|
|
|
|
void SourceLoc::printLineAndColon(raw_ostream &OS,
|
|
const SourceManager &SM) const {
|
|
int BufferIndex = SM->FindBufferContainingLoc(Value);
|
|
print(OS, SM, BufferIndex);
|
|
}
|
|
|
|
void SourceLoc::print(raw_ostream &OS, const SourceManager &SM,
|
|
int &LastBuffer) const {
|
|
if (isInvalid()) {
|
|
OS << "<invalid loc>";
|
|
return;
|
|
}
|
|
int BufferIndex = SM->FindBufferContainingLoc(Value);
|
|
if (BufferIndex == -1) {
|
|
OS << "<malformed loc>";
|
|
return;
|
|
}
|
|
|
|
if (BufferIndex != LastBuffer) {
|
|
OS << SM->getMemoryBuffer((unsigned)BufferIndex)->getBufferIdentifier();
|
|
LastBuffer = BufferIndex;
|
|
} else {
|
|
OS << "line";
|
|
}
|
|
|
|
auto LineAndCol = SM->getLineAndColumn(Value, BufferIndex);
|
|
|
|
OS << ':' << LineAndCol.first << ':' << LineAndCol.second;
|
|
}
|
|
|
|
void SourceLoc::dump(const SourceManager &SM) const {
|
|
print(llvm::errs(), SM);
|
|
}
|
|
|
|
void SourceRange::print(raw_ostream &OS, const SourceManager &SM,
|
|
int &LastBuffer, bool PrintText) const {
|
|
OS << '[';
|
|
Start.print(OS, SM, LastBuffer);
|
|
OS << " - ";
|
|
End.print(OS, SM, LastBuffer);
|
|
OS << ']';
|
|
|
|
if (Start.isInvalid() || End.isInvalid())
|
|
return;
|
|
|
|
if (PrintText) {
|
|
OS << " RangeText=\""
|
|
<< StringRef(Start.Value.getPointer(),
|
|
End.Value.getPointer() - Start.Value.getPointer()+1)
|
|
<< '"';
|
|
}
|
|
}
|
|
|
|
void SourceRange::dump(const SourceManager &SM) const {
|
|
print(llvm::errs(), SM);
|
|
}
|
|
|