mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2025-02-09 13:33:17 +00:00
05b0f7dc93
The ByteStreamer here wasn't taking account of whether the asm streamer was text based and verbose. Only with that combination should we emit comments. This change makes sure that we only actually convert a Twine to a string using Twine::str() if we need the comment. This saves about 10000 small allocations on a test case involving the verify-use_list-order bitcode going through llc with debug info. Note, this is NFC as the comments would ultimately never be emitted unless required. Reviewed by Duncan Exon Smith and David Blaikie. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@237851 91177308-0d34-0410-b5e6-96231b3b80d8
109 lines
3.3 KiB
C++
109 lines
3.3 KiB
C++
//===-- llvm/CodeGen/ByteStreamer.h - ByteStreamer class --------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains a class that can take bytes that would normally be
|
|
// streamed via the AsmPrinter.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_CODEGEN_ASMPRINTER_BYTESTREAMER_H
|
|
#define LLVM_LIB_CODEGEN_ASMPRINTER_BYTESTREAMER_H
|
|
|
|
#include "DIEHash.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/CodeGen/AsmPrinter.h"
|
|
#include "llvm/MC/MCStreamer.h"
|
|
#include "llvm/Support/LEB128.h"
|
|
#include <string>
|
|
|
|
namespace llvm {
|
|
class ByteStreamer {
|
|
public:
|
|
virtual ~ByteStreamer() {}
|
|
|
|
// For now we're just handling the calls we need for dwarf emission/hashing.
|
|
virtual void EmitInt8(uint8_t Byte, const Twine &Comment = "") = 0;
|
|
virtual void EmitSLEB128(uint64_t DWord, const Twine &Comment = "") = 0;
|
|
virtual void EmitULEB128(uint64_t DWord, const Twine &Comment = "") = 0;
|
|
};
|
|
|
|
class APByteStreamer : public ByteStreamer {
|
|
private:
|
|
AsmPrinter &AP;
|
|
|
|
public:
|
|
APByteStreamer(AsmPrinter &Asm) : AP(Asm) {}
|
|
void EmitInt8(uint8_t Byte, const Twine &Comment) override {
|
|
AP.OutStreamer->AddComment(Comment);
|
|
AP.EmitInt8(Byte);
|
|
}
|
|
void EmitSLEB128(uint64_t DWord, const Twine &Comment) override {
|
|
AP.OutStreamer->AddComment(Comment);
|
|
AP.EmitSLEB128(DWord);
|
|
}
|
|
void EmitULEB128(uint64_t DWord, const Twine &Comment) override {
|
|
AP.OutStreamer->AddComment(Comment);
|
|
AP.EmitULEB128(DWord);
|
|
}
|
|
};
|
|
|
|
class HashingByteStreamer : public ByteStreamer {
|
|
private:
|
|
DIEHash &Hash;
|
|
public:
|
|
HashingByteStreamer(DIEHash &H) : Hash(H) {}
|
|
void EmitInt8(uint8_t Byte, const Twine &Comment) override {
|
|
Hash.update(Byte);
|
|
}
|
|
void EmitSLEB128(uint64_t DWord, const Twine &Comment) override {
|
|
Hash.addSLEB128(DWord);
|
|
}
|
|
void EmitULEB128(uint64_t DWord, const Twine &Comment) override {
|
|
Hash.addULEB128(DWord);
|
|
}
|
|
};
|
|
|
|
class BufferByteStreamer : public ByteStreamer {
|
|
private:
|
|
SmallVectorImpl<char> &Buffer;
|
|
SmallVectorImpl<std::string> &Comments;
|
|
|
|
/// \brief Only verbose textual output needs comments. This will be set to
|
|
/// true for that case, and false otherwise. If false, comments passed in to
|
|
/// the emit methods will be ignored.
|
|
bool GenerateComments;
|
|
|
|
public:
|
|
BufferByteStreamer(SmallVectorImpl<char> &Buffer,
|
|
SmallVectorImpl<std::string> &Comments,
|
|
bool GenerateComments)
|
|
: Buffer(Buffer), Comments(Comments), GenerateComments(GenerateComments) {}
|
|
void EmitInt8(uint8_t Byte, const Twine &Comment) override {
|
|
Buffer.push_back(Byte);
|
|
if (GenerateComments)
|
|
Comments.push_back(Comment.str());
|
|
}
|
|
void EmitSLEB128(uint64_t DWord, const Twine &Comment) override {
|
|
raw_svector_ostream OSE(Buffer);
|
|
encodeSLEB128(DWord, OSE);
|
|
if (GenerateComments)
|
|
Comments.push_back(Comment.str());
|
|
}
|
|
void EmitULEB128(uint64_t DWord, const Twine &Comment) override {
|
|
raw_svector_ostream OSE(Buffer);
|
|
encodeULEB128(DWord, OSE);
|
|
if (GenerateComments)
|
|
Comments.push_back(Comment.str());
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|