#ifndef TGSOURCEMGR_H
#define TGSOURCEMGR_H
#include <string>
#include <vector>
#include <cassert>
namespace llvm {
class MemoryBuffer;
class TGSourceMgr;
class TGLoc {
const char *Ptr;
public:
TGLoc() : Ptr(0) {}
TGLoc(const TGLoc &RHS) : Ptr(RHS.Ptr) {}
bool operator==(const TGLoc &RHS) const { return RHS.Ptr == Ptr; }
bool operator!=(const TGLoc &RHS) const { return RHS.Ptr != Ptr; }
const char *getPointer() const { return Ptr; }
static TGLoc getFromPointer(const char *Ptr) {
TGLoc L;
L.Ptr = Ptr;
return L;
}
};
class TGSourceMgr {
struct SrcBuffer {
MemoryBuffer *Buffer;
TGLoc IncludeLoc;
};
std::vector<SrcBuffer> Buffers;
TGSourceMgr(const TGSourceMgr&); void operator=(const TGSourceMgr&); public:
TGSourceMgr() {}
~TGSourceMgr();
const SrcBuffer &getBufferInfo(unsigned i) const {
assert(i < Buffers.size() && "Invalid Buffer ID!");
return Buffers[i];
}
const MemoryBuffer *getMemoryBuffer(unsigned i) const {
assert(i < Buffers.size() && "Invalid Buffer ID!");
return Buffers[i].Buffer;
}
TGLoc getParentIncludeLoc(unsigned i) const {
assert(i < Buffers.size() && "Invalid Buffer ID!");
return Buffers[i].IncludeLoc;
}
unsigned AddNewSourceBuffer(MemoryBuffer *F, TGLoc IncludeLoc) {
SrcBuffer NB;
NB.Buffer = F;
NB.IncludeLoc = IncludeLoc;
Buffers.push_back(NB);
return Buffers.size()-1;
}
int FindBufferContainingLoc(TGLoc Loc) const;
unsigned FindLineNumber(TGLoc Loc, int BufferID = -1) const;
void PrintError(TGLoc ErrorLoc, const std::string &Msg) const;
private:
void PrintIncludeStack(TGLoc IncludeLoc) const;
};
}
#endif