Avoid redundant allocations in the linker optimisation hint

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@210747 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Alp Toker 2014-06-12 04:02:46 +00:00
parent 729fc8facc
commit 09d875f8ab

View File

@ -96,6 +96,19 @@ static inline int MCLOHIdToNbArgs(MCLOHType Kind) {
return -1;
}
namespace {
class raw_counting_ostream : public raw_ostream {
uint64_t Count = 0;
void write_impl(const char *, size_t size) override { Count += size; }
uint64_t current_pos() const override { return Count; }
public:
~raw_counting_ostream() { flush(); }
};
}
/// Store Linker Optimization Hint information (LOH).
class MCLOHDirective {
MCLOHType Kind;
@ -132,8 +145,7 @@ public:
/// the given @p Layout.
uint64_t getEmitSize(const MachObjectWriter &ObjWriter,
const MCAsmLayout &Layout) const {
std::string Buffer;
raw_string_ostream OutStream(Buffer);
raw_counting_ostream OutStream;
Emit_impl(OutStream, ObjWriter, Layout);
return OutStream.tell();
}