forked from OSchip/llvm-project
48 lines
1.5 KiB
C++
48 lines
1.5 KiB
C++
//===- MemoryBufferCache.cpp - Cache for loaded memory buffers ------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Basic/MemoryBufferCache.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
using namespace clang;
|
|
|
|
llvm::MemoryBuffer &
|
|
MemoryBufferCache::addBuffer(llvm::StringRef Filename,
|
|
std::unique_ptr<llvm::MemoryBuffer> Buffer) {
|
|
auto Insertion =
|
|
Buffers.insert({Filename, BufferEntry{std::move(Buffer), NextIndex++}});
|
|
assert(Insertion.second && "Already has a buffer");
|
|
return *Insertion.first->second.Buffer;
|
|
}
|
|
|
|
llvm::MemoryBuffer *MemoryBufferCache::lookupBuffer(llvm::StringRef Filename) {
|
|
auto I = Buffers.find(Filename);
|
|
if (I == Buffers.end())
|
|
return nullptr;
|
|
return I->second.Buffer.get();
|
|
}
|
|
|
|
bool MemoryBufferCache::isBufferFinal(llvm::StringRef Filename) {
|
|
auto I = Buffers.find(Filename);
|
|
if (I == Buffers.end())
|
|
return false;
|
|
return I->second.Index < FirstRemovableIndex;
|
|
}
|
|
|
|
bool MemoryBufferCache::tryToRemoveBuffer(llvm::StringRef Filename) {
|
|
auto I = Buffers.find(Filename);
|
|
assert(I != Buffers.end() && "No buffer to remove...");
|
|
if (I->second.Index < FirstRemovableIndex)
|
|
return true;
|
|
|
|
Buffers.erase(I);
|
|
return false;
|
|
}
|
|
|
|
void MemoryBufferCache::finalizeCurrentBuffers() { FirstRemovableIndex = NextIndex; }
|