diff --git a/include/llvm/Metadata.h b/include/llvm/Metadata.h index f490bad2cc1..194de47ffe3 100644 --- a/include/llvm/Metadata.h +++ b/include/llvm/Metadata.h @@ -194,10 +194,10 @@ class NamedMDNode : public Value, public ilist_node { void setParent(Module *M) { Parent = M; } protected: - explicit NamedMDNode(LLVMContext &C, StringRef N, MDNode*const *Vals, + explicit NamedMDNode(LLVMContext &C, const Twine &N, MDNode*const *Vals, unsigned NumVals, Module *M = 0); public: - static NamedMDNode *Create(LLVMContext &C, StringRef N, + static NamedMDNode *Create(LLVMContext &C, const Twine &N, MDNode *const *MDs, unsigned NumMDs, Module *M = 0) { return new NamedMDNode(C, N, MDs, NumMDs, M); @@ -229,7 +229,7 @@ public: void addOperand(MDNode *M); /// setName - Set the name of this named metadata. - void setName(StringRef Name); + void setName(const Twine &NewName); /// getName - Return a constant reference to this named metadata's name. StringRef getName() const; diff --git a/lib/VMCore/Metadata.cpp b/lib/VMCore/Metadata.cpp index 5ea7bd45e57..e6a962bbca0 100644 --- a/lib/VMCore/Metadata.cpp +++ b/lib/VMCore/Metadata.cpp @@ -18,6 +18,7 @@ #include "llvm/Instruction.h" #include "llvm/ADT/DenseMap.h" #include "llvm/ADT/StringMap.h" +#include "llvm/ADT/SmallString.h" #include "SymbolTableListTraitsImpl.h" #include "llvm/Support/ValueHandle.h" using namespace llvm; @@ -263,7 +264,7 @@ static SmallVector &getNMDOps(void *Operands) { return *(SmallVector*)Operands; } -NamedMDNode::NamedMDNode(LLVMContext &C, StringRef N, +NamedMDNode::NamedMDNode(LLVMContext &C, const Twine &N, MDNode *const *MDs, unsigned NumMDs, Module *ParentModule) : Value(Type::getMetadataTy(C), Value::NamedMDNodeVal), Parent(0) { @@ -322,11 +323,23 @@ void NamedMDNode::dropAllReferences() { } /// setName - Set the name of this named metadata. -void NamedMDNode::setName(StringRef N) { - assert (!N.empty() && "Invalid named metadata name!"); - Name = N.str(); +void NamedMDNode::setName(const Twine &NewName) { + assert (!NewName.isTriviallyEmpty() && "Invalid named metadata name!"); + + SmallString<256> NameData; + NewName.toVector(NameData); + + const char *NameStr = NameData.data(); + unsigned NameLen = NameData.size(); + + StringRef NameRef = StringRef(NameStr, NameLen); + // Name isn't changing? + if (getName() == NameRef) + return; + + Name = NameRef.str(); if (Parent) - Parent->getMDSymbolTable().insert(N, this); + Parent->getMDSymbolTable().insert(NameRef, this); } /// getName - Return a constant reference to this named metadata's name.