12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490 |
- #pragma once
- #ifdef __GNUC__
- #pragma GCC diagnostic push
- #pragma GCC diagnostic ignored "-Wunused-parameter"
- #endif
- //===- llvm/IR/Metadata.h - Metadata definitions ----------------*- C++ -*-===//
- //
- // 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
- //
- //===----------------------------------------------------------------------===//
- //
- /// @file
- /// This file contains the declarations for metadata subclasses.
- /// They represent the different flavors of metadata that live in LLVM.
- //
- //===----------------------------------------------------------------------===//
- #ifndef LLVM_IR_METADATA_H
- #define LLVM_IR_METADATA_H
- #include "llvm/ADT/ArrayRef.h"
- #include "llvm/ADT/DenseMap.h"
- #include "llvm/ADT/DenseMapInfo.h"
- #include "llvm/ADT/None.h"
- #include "llvm/ADT/PointerUnion.h"
- #include "llvm/ADT/STLExtras.h"
- #include "llvm/ADT/SmallVector.h"
- #include "llvm/ADT/StringMap.h"
- #include "llvm/ADT/StringRef.h"
- #include "llvm/ADT/ilist_node.h"
- #include "llvm/ADT/iterator_range.h"
- #include "llvm/IR/Constant.h"
- #include "llvm/IR/LLVMContext.h"
- #include "llvm/IR/Value.h"
- #include "llvm/Support/CBindingWrapping.h"
- #include "llvm/Support/Casting.h"
- #include "llvm/Support/ErrorHandling.h"
- #include <cassert>
- #include <cstddef>
- #include <cstdint>
- #include <iterator>
- #include <memory>
- #include <string>
- #include <type_traits>
- #include <utility>
- namespace llvm {
- class Module;
- class ModuleSlotTracker;
- class raw_ostream;
- class Type;
- enum LLVMConstants : uint32_t {
- DEBUG_METADATA_VERSION = 3 // Current debug info version number.
- };
- /// Root of the metadata hierarchy.
- ///
- /// This is a root class for typeless data in the IR.
- class Metadata {
- friend class ReplaceableMetadataImpl;
- /// RTTI.
- const unsigned char SubclassID;
- protected:
- /// Active type of storage.
- enum StorageType { Uniqued, Distinct, Temporary };
- /// Storage flag for non-uniqued, otherwise unowned, metadata.
- unsigned char Storage : 7;
- // TODO: expose remaining bits to subclasses.
- unsigned char ImplicitCode : 1;
- unsigned short SubclassData16 = 0;
- unsigned SubclassData32 = 0;
- public:
- enum MetadataKind {
- #define HANDLE_METADATA_LEAF(CLASS) CLASS##Kind,
- #include "llvm/IR/Metadata.def"
- };
- protected:
- Metadata(unsigned ID, StorageType Storage)
- : SubclassID(ID), Storage(Storage), ImplicitCode(false) {
- static_assert(sizeof(*this) == 8, "Metadata fields poorly packed");
- }
- ~Metadata() = default;
- /// Default handling of a changed operand, which asserts.
- ///
- /// If subclasses pass themselves in as owners to a tracking node reference,
- /// they must provide an implementation of this method.
- void handleChangedOperand(void *, Metadata *) {
- llvm_unreachable("Unimplemented in Metadata subclass");
- }
- public:
- unsigned getMetadataID() const { return SubclassID; }
- /// User-friendly dump.
- ///
- /// If \c M is provided, metadata nodes will be numbered canonically;
- /// otherwise, pointer addresses are substituted.
- ///
- /// Note: this uses an explicit overload instead of default arguments so that
- /// the nullptr version is easy to call from a debugger.
- ///
- /// @{
- void dump() const;
- void dump(const Module *M) const;
- /// @}
- /// Print.
- ///
- /// Prints definition of \c this.
- ///
- /// If \c M is provided, metadata nodes will be numbered canonically;
- /// otherwise, pointer addresses are substituted.
- /// @{
- void print(raw_ostream &OS, const Module *M = nullptr,
- bool IsForDebug = false) const;
- void print(raw_ostream &OS, ModuleSlotTracker &MST, const Module *M = nullptr,
- bool IsForDebug = false) const;
- /// @}
- /// Print as operand.
- ///
- /// Prints reference of \c this.
- ///
- /// If \c M is provided, metadata nodes will be numbered canonically;
- /// otherwise, pointer addresses are substituted.
- /// @{
- void printAsOperand(raw_ostream &OS, const Module *M = nullptr) const;
- void printAsOperand(raw_ostream &OS, ModuleSlotTracker &MST,
- const Module *M = nullptr) const;
- /// @}
- };
- // Create wrappers for C Binding types (see CBindingWrapping.h).
- DEFINE_ISA_CONVERSION_FUNCTIONS(Metadata, LLVMMetadataRef)
- // Specialized opaque metadata conversions.
- inline Metadata **unwrap(LLVMMetadataRef *MDs) {
- return reinterpret_cast<Metadata**>(MDs);
- }
- #define HANDLE_METADATA(CLASS) class CLASS;
- #include "llvm/IR/Metadata.def"
- // Provide specializations of isa so that we don't need definitions of
- // subclasses to see if the metadata is a subclass.
- #define HANDLE_METADATA_LEAF(CLASS) \
- template <> struct isa_impl<CLASS, Metadata> { \
- static inline bool doit(const Metadata &MD) { \
- return MD.getMetadataID() == Metadata::CLASS##Kind; \
- } \
- };
- #include "llvm/IR/Metadata.def"
- inline raw_ostream &operator<<(raw_ostream &OS, const Metadata &MD) {
- MD.print(OS);
- return OS;
- }
- /// Metadata wrapper in the Value hierarchy.
- ///
- /// A member of the \a Value hierarchy to represent a reference to metadata.
- /// This allows, e.g., instrinsics to have metadata as operands.
- ///
- /// Notably, this is the only thing in either hierarchy that is allowed to
- /// reference \a LocalAsMetadata.
- class MetadataAsValue : public Value {
- friend class ReplaceableMetadataImpl;
- friend class LLVMContextImpl;
- Metadata *MD;
- MetadataAsValue(Type *Ty, Metadata *MD);
- /// Drop use of metadata (during teardown).
- void dropUse() { MD = nullptr; }
- public:
- ~MetadataAsValue();
- static MetadataAsValue *get(LLVMContext &Context, Metadata *MD);
- static MetadataAsValue *getIfExists(LLVMContext &Context, Metadata *MD);
- Metadata *getMetadata() const { return MD; }
- static bool classof(const Value *V) {
- return V->getValueID() == MetadataAsValueVal;
- }
- private:
- void handleChangedMetadata(Metadata *MD);
- void track();
- void untrack();
- };
- /// API for tracking metadata references through RAUW and deletion.
- ///
- /// Shared API for updating \a Metadata pointers in subclasses that support
- /// RAUW.
- ///
- /// This API is not meant to be used directly. See \a TrackingMDRef for a
- /// user-friendly tracking reference.
- class MetadataTracking {
- public:
- /// Track the reference to metadata.
- ///
- /// Register \c MD with \c *MD, if the subclass supports tracking. If \c *MD
- /// gets RAUW'ed, \c MD will be updated to the new address. If \c *MD gets
- /// deleted, \c MD will be set to \c nullptr.
- ///
- /// If tracking isn't supported, \c *MD will not change.
- ///
- /// \return true iff tracking is supported by \c MD.
- static bool track(Metadata *&MD) {
- return track(&MD, *MD, static_cast<Metadata *>(nullptr));
- }
- /// Track the reference to metadata for \a Metadata.
- ///
- /// As \a track(Metadata*&), but with support for calling back to \c Owner to
- /// tell it that its operand changed. This could trigger \c Owner being
- /// re-uniqued.
- static bool track(void *Ref, Metadata &MD, Metadata &Owner) {
- return track(Ref, MD, &Owner);
- }
- /// Track the reference to metadata for \a MetadataAsValue.
- ///
- /// As \a track(Metadata*&), but with support for calling back to \c Owner to
- /// tell it that its operand changed. This could trigger \c Owner being
- /// re-uniqued.
- static bool track(void *Ref, Metadata &MD, MetadataAsValue &Owner) {
- return track(Ref, MD, &Owner);
- }
- /// Stop tracking a reference to metadata.
- ///
- /// Stops \c *MD from tracking \c MD.
- static void untrack(Metadata *&MD) { untrack(&MD, *MD); }
- static void untrack(void *Ref, Metadata &MD);
- /// Move tracking from one reference to another.
- ///
- /// Semantically equivalent to \c untrack(MD) followed by \c track(New),
- /// except that ownership callbacks are maintained.
- ///
- /// Note: it is an error if \c *MD does not equal \c New.
- ///
- /// \return true iff tracking is supported by \c MD.
- static bool retrack(Metadata *&MD, Metadata *&New) {
- return retrack(&MD, *MD, &New);
- }
- static bool retrack(void *Ref, Metadata &MD, void *New);
- /// Check whether metadata is replaceable.
- static bool isReplaceable(const Metadata &MD);
- using OwnerTy = PointerUnion<MetadataAsValue *, Metadata *>;
- private:
- /// Track a reference to metadata for an owner.
- ///
- /// Generalized version of tracking.
- static bool track(void *Ref, Metadata &MD, OwnerTy Owner);
- };
- /// Shared implementation of use-lists for replaceable metadata.
- ///
- /// Most metadata cannot be RAUW'ed. This is a shared implementation of
- /// use-lists and associated API for the two that support it (\a ValueAsMetadata
- /// and \a TempMDNode).
- class ReplaceableMetadataImpl {
- friend class MetadataTracking;
- public:
- using OwnerTy = MetadataTracking::OwnerTy;
- private:
- LLVMContext &Context;
- uint64_t NextIndex = 0;
- SmallDenseMap<void *, std::pair<OwnerTy, uint64_t>, 4> UseMap;
- public:
- ReplaceableMetadataImpl(LLVMContext &Context) : Context(Context) {}
- ~ReplaceableMetadataImpl() {
- assert(UseMap.empty() && "Cannot destroy in-use replaceable metadata");
- }
- LLVMContext &getContext() const { return Context; }
- /// Replace all uses of this with MD.
- ///
- /// Replace all uses of this with \c MD, which is allowed to be null.
- void replaceAllUsesWith(Metadata *MD);
- /// Resolve all uses of this.
- ///
- /// Resolve all uses of this, turning off RAUW permanently. If \c
- /// ResolveUsers, call \a MDNode::resolve() on any users whose last operand
- /// is resolved.
- void resolveAllUses(bool ResolveUsers = true);
- private:
- void addRef(void *Ref, OwnerTy Owner);
- void dropRef(void *Ref);
- void moveRef(void *Ref, void *New, const Metadata &MD);
- /// Lazily construct RAUW support on MD.
- ///
- /// If this is an unresolved MDNode, RAUW support will be created on-demand.
- /// ValueAsMetadata always has RAUW support.
- static ReplaceableMetadataImpl *getOrCreate(Metadata &MD);
- /// Get RAUW support on MD, if it exists.
- static ReplaceableMetadataImpl *getIfExists(Metadata &MD);
- /// Check whether this node will support RAUW.
- ///
- /// Returns \c true unless getOrCreate() would return null.
- static bool isReplaceable(const Metadata &MD);
- };
- /// Value wrapper in the Metadata hierarchy.
- ///
- /// This is a custom value handle that allows other metadata to refer to
- /// classes in the Value hierarchy.
- ///
- /// Because of full uniquing support, each value is only wrapped by a single \a
- /// ValueAsMetadata object, so the lookup maps are far more efficient than
- /// those using ValueHandleBase.
- class ValueAsMetadata : public Metadata, ReplaceableMetadataImpl {
- friend class ReplaceableMetadataImpl;
- friend class LLVMContextImpl;
- Value *V;
- /// Drop users without RAUW (during teardown).
- void dropUsers() {
- ReplaceableMetadataImpl::resolveAllUses(/* ResolveUsers */ false);
- }
- protected:
- ValueAsMetadata(unsigned ID, Value *V)
- : Metadata(ID, Uniqued), ReplaceableMetadataImpl(V->getContext()), V(V) {
- assert(V && "Expected valid value");
- }
- ~ValueAsMetadata() = default;
- public:
- static ValueAsMetadata *get(Value *V);
- static ConstantAsMetadata *getConstant(Value *C) {
- return cast<ConstantAsMetadata>(get(C));
- }
- static LocalAsMetadata *getLocal(Value *Local) {
- return cast<LocalAsMetadata>(get(Local));
- }
- static ValueAsMetadata *getIfExists(Value *V);
- static ConstantAsMetadata *getConstantIfExists(Value *C) {
- return cast_or_null<ConstantAsMetadata>(getIfExists(C));
- }
- static LocalAsMetadata *getLocalIfExists(Value *Local) {
- return cast_or_null<LocalAsMetadata>(getIfExists(Local));
- }
- Value *getValue() const { return V; }
- Type *getType() const { return V->getType(); }
- LLVMContext &getContext() const { return V->getContext(); }
- static void handleDeletion(Value *V);
- static void handleRAUW(Value *From, Value *To);
- protected:
- /// Handle collisions after \a Value::replaceAllUsesWith().
- ///
- /// RAUW isn't supported directly for \a ValueAsMetadata, but if the wrapped
- /// \a Value gets RAUW'ed and the target already exists, this is used to
- /// merge the two metadata nodes.
- void replaceAllUsesWith(Metadata *MD) {
- ReplaceableMetadataImpl::replaceAllUsesWith(MD);
- }
- public:
- static bool classof(const Metadata *MD) {
- return MD->getMetadataID() == LocalAsMetadataKind ||
- MD->getMetadataID() == ConstantAsMetadataKind;
- }
- };
- class ConstantAsMetadata : public ValueAsMetadata {
- friend class ValueAsMetadata;
- ConstantAsMetadata(Constant *C)
- : ValueAsMetadata(ConstantAsMetadataKind, C) {}
- public:
- static ConstantAsMetadata *get(Constant *C) {
- return ValueAsMetadata::getConstant(C);
- }
- static ConstantAsMetadata *getIfExists(Constant *C) {
- return ValueAsMetadata::getConstantIfExists(C);
- }
- Constant *getValue() const {
- return cast<Constant>(ValueAsMetadata::getValue());
- }
- static bool classof(const Metadata *MD) {
- return MD->getMetadataID() == ConstantAsMetadataKind;
- }
- };
- class LocalAsMetadata : public ValueAsMetadata {
- friend class ValueAsMetadata;
- LocalAsMetadata(Value *Local)
- : ValueAsMetadata(LocalAsMetadataKind, Local) {
- assert(!isa<Constant>(Local) && "Expected local value");
- }
- public:
- static LocalAsMetadata *get(Value *Local) {
- return ValueAsMetadata::getLocal(Local);
- }
- static LocalAsMetadata *getIfExists(Value *Local) {
- return ValueAsMetadata::getLocalIfExists(Local);
- }
- static bool classof(const Metadata *MD) {
- return MD->getMetadataID() == LocalAsMetadataKind;
- }
- };
- /// Transitional API for extracting constants from Metadata.
- ///
- /// This namespace contains transitional functions for metadata that points to
- /// \a Constants.
- ///
- /// In prehistory -- when metadata was a subclass of \a Value -- \a MDNode
- /// operands could refer to any \a Value. There's was a lot of code like this:
- ///
- /// \code
- /// MDNode *N = ...;
- /// auto *CI = dyn_cast<ConstantInt>(N->getOperand(2));
- /// \endcode
- ///
- /// Now that \a Value and \a Metadata are in separate hierarchies, maintaining
- /// the semantics for \a isa(), \a cast(), \a dyn_cast() (etc.) requires three
- /// steps: cast in the \a Metadata hierarchy, extraction of the \a Value, and
- /// cast in the \a Value hierarchy. Besides creating boiler-plate, this
- /// requires subtle control flow changes.
- ///
- /// The end-goal is to create a new type of metadata, called (e.g.) \a MDInt,
- /// so that metadata can refer to numbers without traversing a bridge to the \a
- /// Value hierarchy. In this final state, the code above would look like this:
- ///
- /// \code
- /// MDNode *N = ...;
- /// auto *MI = dyn_cast<MDInt>(N->getOperand(2));
- /// \endcode
- ///
- /// The API in this namespace supports the transition. \a MDInt doesn't exist
- /// yet, and even once it does, changing each metadata schema to use it is its
- /// own mini-project. In the meantime this API prevents us from introducing
- /// complex and bug-prone control flow that will disappear in the end. In
- /// particular, the above code looks like this:
- ///
- /// \code
- /// MDNode *N = ...;
- /// auto *CI = mdconst::dyn_extract<ConstantInt>(N->getOperand(2));
- /// \endcode
- ///
- /// The full set of provided functions includes:
- ///
- /// mdconst::hasa <=> isa
- /// mdconst::extract <=> cast
- /// mdconst::extract_or_null <=> cast_or_null
- /// mdconst::dyn_extract <=> dyn_cast
- /// mdconst::dyn_extract_or_null <=> dyn_cast_or_null
- ///
- /// The target of the cast must be a subclass of \a Constant.
- namespace mdconst {
- namespace detail {
- template <class T> T &make();
- template <class T, class Result> struct HasDereference {
- using Yes = char[1];
- using No = char[2];
- template <size_t N> struct SFINAE {};
- template <class U, class V>
- static Yes &hasDereference(SFINAE<sizeof(static_cast<V>(*make<U>()))> * = 0);
- template <class U, class V> static No &hasDereference(...);
- static const bool value =
- sizeof(hasDereference<T, Result>(nullptr)) == sizeof(Yes);
- };
- template <class V, class M> struct IsValidPointer {
- static const bool value = std::is_base_of<Constant, V>::value &&
- HasDereference<M, const Metadata &>::value;
- };
- template <class V, class M> struct IsValidReference {
- static const bool value = std::is_base_of<Constant, V>::value &&
- std::is_convertible<M, const Metadata &>::value;
- };
- } // end namespace detail
- /// Check whether Metadata has a Value.
- ///
- /// As an analogue to \a isa(), check whether \c MD has an \a Value inside of
- /// type \c X.
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidPointer<X, Y>::value, bool>
- hasa(Y &&MD) {
- assert(MD && "Null pointer sent into hasa");
- if (auto *V = dyn_cast<ConstantAsMetadata>(MD))
- return isa<X>(V->getValue());
- return false;
- }
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidReference<X, Y &>::value, bool>
- hasa(Y &MD) {
- return hasa(&MD);
- }
- /// Extract a Value from Metadata.
- ///
- /// As an analogue to \a cast(), extract the \a Value subclass \c X from \c MD.
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidPointer<X, Y>::value, X *>
- extract(Y &&MD) {
- return cast<X>(cast<ConstantAsMetadata>(MD)->getValue());
- }
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidReference<X, Y &>::value, X *>
- extract(Y &MD) {
- return extract(&MD);
- }
- /// Extract a Value from Metadata, allowing null.
- ///
- /// As an analogue to \a cast_or_null(), extract the \a Value subclass \c X
- /// from \c MD, allowing \c MD to be null.
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidPointer<X, Y>::value, X *>
- extract_or_null(Y &&MD) {
- if (auto *V = cast_or_null<ConstantAsMetadata>(MD))
- return cast<X>(V->getValue());
- return nullptr;
- }
- /// Extract a Value from Metadata, if any.
- ///
- /// As an analogue to \a dyn_cast_or_null(), extract the \a Value subclass \c X
- /// from \c MD, return null if \c MD doesn't contain a \a Value or if the \a
- /// Value it does contain is of the wrong subclass.
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidPointer<X, Y>::value, X *>
- dyn_extract(Y &&MD) {
- if (auto *V = dyn_cast<ConstantAsMetadata>(MD))
- return dyn_cast<X>(V->getValue());
- return nullptr;
- }
- /// Extract a Value from Metadata, if any, allowing null.
- ///
- /// As an analogue to \a dyn_cast_or_null(), extract the \a Value subclass \c X
- /// from \c MD, return null if \c MD doesn't contain a \a Value or if the \a
- /// Value it does contain is of the wrong subclass, allowing \c MD to be null.
- template <class X, class Y>
- inline std::enable_if_t<detail::IsValidPointer<X, Y>::value, X *>
- dyn_extract_or_null(Y &&MD) {
- if (auto *V = dyn_cast_or_null<ConstantAsMetadata>(MD))
- return dyn_cast<X>(V->getValue());
- return nullptr;
- }
- } // end namespace mdconst
- //===----------------------------------------------------------------------===//
- /// A single uniqued string.
- ///
- /// These are used to efficiently contain a byte sequence for metadata.
- /// MDString is always unnamed.
- class MDString : public Metadata {
- friend class StringMapEntryStorage<MDString>;
- StringMapEntry<MDString> *Entry = nullptr;
- MDString() : Metadata(MDStringKind, Uniqued) {}
- public:
- MDString(const MDString &) = delete;
- MDString &operator=(MDString &&) = delete;
- MDString &operator=(const MDString &) = delete;
- static MDString *get(LLVMContext &Context, StringRef Str);
- static MDString *get(LLVMContext &Context, const char *Str) {
- return get(Context, Str ? StringRef(Str) : StringRef());
- }
- StringRef getString() const;
- unsigned getLength() const { return (unsigned)getString().size(); }
- using iterator = StringRef::iterator;
- /// Pointer to the first byte of the string.
- iterator begin() const { return getString().begin(); }
- /// Pointer to one byte past the end of the string.
- iterator end() const { return getString().end(); }
- const unsigned char *bytes_begin() const { return getString().bytes_begin(); }
- const unsigned char *bytes_end() const { return getString().bytes_end(); }
- /// Methods for support type inquiry through isa, cast, and dyn_cast.
- static bool classof(const Metadata *MD) {
- return MD->getMetadataID() == MDStringKind;
- }
- };
- /// A collection of metadata nodes that might be associated with a
- /// memory access used by the alias-analysis infrastructure.
- struct AAMDNodes {
- explicit AAMDNodes() = default;
- explicit AAMDNodes(MDNode *T, MDNode *TS, MDNode *S, MDNode *N)
- : TBAA(T), TBAAStruct(TS), Scope(S), NoAlias(N) {}
- bool operator==(const AAMDNodes &A) const {
- return TBAA == A.TBAA && TBAAStruct == A.TBAAStruct && Scope == A.Scope &&
- NoAlias == A.NoAlias;
- }
- bool operator!=(const AAMDNodes &A) const { return !(*this == A); }
- explicit operator bool() const {
- return TBAA || TBAAStruct || Scope || NoAlias;
- }
- /// The tag for type-based alias analysis.
- MDNode *TBAA = nullptr;
- /// The tag for type-based alias analysis (tbaa struct).
- MDNode *TBAAStruct = nullptr;
- /// The tag for alias scope specification (used with noalias).
- MDNode *Scope = nullptr;
- /// The tag specifying the noalias scope.
- MDNode *NoAlias = nullptr;
- // Shift tbaa Metadata node to start off bytes later
- static MDNode *ShiftTBAA(MDNode *M, size_t off);
- // Shift tbaa.struct Metadata node to start off bytes later
- static MDNode *ShiftTBAAStruct(MDNode *M, size_t off);
- /// Given two sets of AAMDNodes that apply to the same pointer,
- /// give the best AAMDNodes that are compatible with both (i.e. a set of
- /// nodes whose allowable aliasing conclusions are a subset of those
- /// allowable by both of the inputs). However, for efficiency
- /// reasons, do not create any new MDNodes.
- AAMDNodes intersect(const AAMDNodes &Other) {
- AAMDNodes Result;
- Result.TBAA = Other.TBAA == TBAA ? TBAA : nullptr;
- Result.TBAAStruct = Other.TBAAStruct == TBAAStruct ? TBAAStruct : nullptr;
- Result.Scope = Other.Scope == Scope ? Scope : nullptr;
- Result.NoAlias = Other.NoAlias == NoAlias ? NoAlias : nullptr;
- return Result;
- }
- /// Create a new AAMDNode that describes this AAMDNode after applying a
- /// constant offset to the start of the pointer
- AAMDNodes shift(size_t Offset) {
- AAMDNodes Result;
- Result.TBAA = TBAA ? ShiftTBAA(TBAA, Offset) : nullptr;
- Result.TBAAStruct =
- TBAAStruct ? ShiftTBAAStruct(TBAAStruct, Offset) : nullptr;
- Result.Scope = Scope;
- Result.NoAlias = NoAlias;
- return Result;
- }
- };
- // Specialize DenseMapInfo for AAMDNodes.
- template<>
- struct DenseMapInfo<AAMDNodes> {
- static inline AAMDNodes getEmptyKey() {
- return AAMDNodes(DenseMapInfo<MDNode *>::getEmptyKey(),
- nullptr, nullptr, nullptr);
- }
- static inline AAMDNodes getTombstoneKey() {
- return AAMDNodes(DenseMapInfo<MDNode *>::getTombstoneKey(),
- nullptr, nullptr, nullptr);
- }
- static unsigned getHashValue(const AAMDNodes &Val) {
- return DenseMapInfo<MDNode *>::getHashValue(Val.TBAA) ^
- DenseMapInfo<MDNode *>::getHashValue(Val.TBAAStruct) ^
- DenseMapInfo<MDNode *>::getHashValue(Val.Scope) ^
- DenseMapInfo<MDNode *>::getHashValue(Val.NoAlias);
- }
- static bool isEqual(const AAMDNodes &LHS, const AAMDNodes &RHS) {
- return LHS == RHS;
- }
- };
- /// Tracking metadata reference owned by Metadata.
- ///
- /// Similar to \a TrackingMDRef, but it's expected to be owned by an instance
- /// of \a Metadata, which has the option of registering itself for callbacks to
- /// re-unique itself.
- ///
- /// In particular, this is used by \a MDNode.
- class MDOperand {
- Metadata *MD = nullptr;
- public:
- MDOperand() = default;
- MDOperand(MDOperand &&) = delete;
- MDOperand(const MDOperand &) = delete;
- MDOperand &operator=(MDOperand &&) = delete;
- MDOperand &operator=(const MDOperand &) = delete;
- ~MDOperand() { untrack(); }
- Metadata *get() const { return MD; }
- operator Metadata *() const { return get(); }
- Metadata *operator->() const { return get(); }
- Metadata &operator*() const { return *get(); }
- void reset() {
- untrack();
- MD = nullptr;
- }
- void reset(Metadata *MD, Metadata *Owner) {
- untrack();
- this->MD = MD;
- track(Owner);
- }
- private:
- void track(Metadata *Owner) {
- if (MD) {
- if (Owner)
- MetadataTracking::track(this, *MD, *Owner);
- else
- MetadataTracking::track(MD);
- }
- }
- void untrack() {
- assert(static_cast<void *>(this) == &MD && "Expected same address");
- if (MD)
- MetadataTracking::untrack(MD);
- }
- };
- template <> struct simplify_type<MDOperand> {
- using SimpleType = Metadata *;
- static SimpleType getSimplifiedValue(MDOperand &MD) { return MD.get(); }
- };
- template <> struct simplify_type<const MDOperand> {
- using SimpleType = Metadata *;
- static SimpleType getSimplifiedValue(const MDOperand &MD) { return MD.get(); }
- };
- /// Pointer to the context, with optional RAUW support.
- ///
- /// Either a raw (non-null) pointer to the \a LLVMContext, or an owned pointer
- /// to \a ReplaceableMetadataImpl (which has a reference to \a LLVMContext).
- class ContextAndReplaceableUses {
- PointerUnion<LLVMContext *, ReplaceableMetadataImpl *> Ptr;
- public:
- ContextAndReplaceableUses(LLVMContext &Context) : Ptr(&Context) {}
- ContextAndReplaceableUses(
- std::unique_ptr<ReplaceableMetadataImpl> ReplaceableUses)
- : Ptr(ReplaceableUses.release()) {
- assert(getReplaceableUses() && "Expected non-null replaceable uses");
- }
- ContextAndReplaceableUses() = delete;
- ContextAndReplaceableUses(ContextAndReplaceableUses &&) = delete;
- ContextAndReplaceableUses(const ContextAndReplaceableUses &) = delete;
- ContextAndReplaceableUses &operator=(ContextAndReplaceableUses &&) = delete;
- ContextAndReplaceableUses &
- operator=(const ContextAndReplaceableUses &) = delete;
- ~ContextAndReplaceableUses() { delete getReplaceableUses(); }
- operator LLVMContext &() { return getContext(); }
- /// Whether this contains RAUW support.
- bool hasReplaceableUses() const {
- return Ptr.is<ReplaceableMetadataImpl *>();
- }
- LLVMContext &getContext() const {
- if (hasReplaceableUses())
- return getReplaceableUses()->getContext();
- return *Ptr.get<LLVMContext *>();
- }
- ReplaceableMetadataImpl *getReplaceableUses() const {
- if (hasReplaceableUses())
- return Ptr.get<ReplaceableMetadataImpl *>();
- return nullptr;
- }
- /// Ensure that this has RAUW support, and then return it.
- ReplaceableMetadataImpl *getOrCreateReplaceableUses() {
- if (!hasReplaceableUses())
- makeReplaceable(std::make_unique<ReplaceableMetadataImpl>(getContext()));
- return getReplaceableUses();
- }
- /// Assign RAUW support to this.
- ///
- /// Make this replaceable, taking ownership of \c ReplaceableUses (which must
- /// not be null).
- void
- makeReplaceable(std::unique_ptr<ReplaceableMetadataImpl> ReplaceableUses) {
- assert(ReplaceableUses && "Expected non-null replaceable uses");
- assert(&ReplaceableUses->getContext() == &getContext() &&
- "Expected same context");
- delete getReplaceableUses();
- Ptr = ReplaceableUses.release();
- }
- /// Drop RAUW support.
- ///
- /// Cede ownership of RAUW support, returning it.
- std::unique_ptr<ReplaceableMetadataImpl> takeReplaceableUses() {
- assert(hasReplaceableUses() && "Expected to own replaceable uses");
- std::unique_ptr<ReplaceableMetadataImpl> ReplaceableUses(
- getReplaceableUses());
- Ptr = &ReplaceableUses->getContext();
- return ReplaceableUses;
- }
- };
- struct TempMDNodeDeleter {
- inline void operator()(MDNode *Node) const;
- };
- #define HANDLE_MDNODE_LEAF(CLASS) \
- using Temp##CLASS = std::unique_ptr<CLASS, TempMDNodeDeleter>;
- #define HANDLE_MDNODE_BRANCH(CLASS) HANDLE_MDNODE_LEAF(CLASS)
- #include "llvm/IR/Metadata.def"
- /// Metadata node.
- ///
- /// Metadata nodes can be uniqued, like constants, or distinct. Temporary
- /// metadata nodes (with full support for RAUW) can be used to delay uniquing
- /// until forward references are known. The basic metadata node is an \a
- /// MDTuple.
- ///
- /// There is limited support for RAUW at construction time. At construction
- /// time, if any operand is a temporary node (or an unresolved uniqued node,
- /// which indicates a transitive temporary operand), the node itself will be
- /// unresolved. As soon as all operands become resolved, it will drop RAUW
- /// support permanently.
- ///
- /// If an unresolved node is part of a cycle, \a resolveCycles() needs
- /// to be called on some member of the cycle once all temporary nodes have been
- /// replaced.
- class MDNode : public Metadata {
- friend class ReplaceableMetadataImpl;
- friend class LLVMContextImpl;
- unsigned NumOperands;
- unsigned NumUnresolved;
- ContextAndReplaceableUses Context;
- protected:
- MDNode(LLVMContext &Context, unsigned ID, StorageType Storage,
- ArrayRef<Metadata *> Ops1, ArrayRef<Metadata *> Ops2 = None);
- ~MDNode() = default;
- void *operator new(size_t Size, unsigned NumOps);
- void operator delete(void *Mem);
- /// Required by std, but never called.
- void operator delete(void *, unsigned) {
- llvm_unreachable("Constructor throws?");
- }
- /// Required by std, but never called.
- void operator delete(void *, unsigned, bool) {
- llvm_unreachable("Constructor throws?");
- }
- void dropAllReferences();
- MDOperand *mutable_begin() { return mutable_end() - NumOperands; }
- MDOperand *mutable_end() { return reinterpret_cast<MDOperand *>(this); }
- using mutable_op_range = iterator_range<MDOperand *>;
- mutable_op_range mutable_operands() {
- return mutable_op_range(mutable_begin(), mutable_end());
- }
- public:
- MDNode(const MDNode &) = delete;
- void operator=(const MDNode &) = delete;
- void *operator new(size_t) = delete;
- static inline MDTuple *get(LLVMContext &Context, ArrayRef<Metadata *> MDs);
- static inline MDTuple *getIfExists(LLVMContext &Context,
- ArrayRef<Metadata *> MDs);
- static inline MDTuple *getDistinct(LLVMContext &Context,
- ArrayRef<Metadata *> MDs);
- static inline TempMDTuple getTemporary(LLVMContext &Context,
- ArrayRef<Metadata *> MDs);
- /// Create a (temporary) clone of this.
- TempMDNode clone() const;
- /// Deallocate a node created by getTemporary.
- ///
- /// Calls \c replaceAllUsesWith(nullptr) before deleting, so any remaining
- /// references will be reset.
- static void deleteTemporary(MDNode *N);
- LLVMContext &getContext() const { return Context.getContext(); }
- /// Replace a specific operand.
- void replaceOperandWith(unsigned I, Metadata *New);
- /// Check if node is fully resolved.
- ///
- /// If \a isTemporary(), this always returns \c false; if \a isDistinct(),
- /// this always returns \c true.
- ///
- /// If \a isUniqued(), returns \c true if this has already dropped RAUW
- /// support (because all operands are resolved).
- ///
- /// As forward declarations are resolved, their containers should get
- /// resolved automatically. However, if this (or one of its operands) is
- /// involved in a cycle, \a resolveCycles() needs to be called explicitly.
- bool isResolved() const { return !isTemporary() && !NumUnresolved; }
- bool isUniqued() const { return Storage == Uniqued; }
- bool isDistinct() const { return Storage == Distinct; }
- bool isTemporary() const { return Storage == Temporary; }
- /// RAUW a temporary.
- ///
- /// \pre \a isTemporary() must be \c true.
- void replaceAllUsesWith(Metadata *MD) {
- assert(isTemporary() && "Expected temporary node");
- if (Context.hasReplaceableUses())
- Context.getReplaceableUses()->replaceAllUsesWith(MD);
- }
- /// Resolve cycles.
- ///
- /// Once all forward declarations have been resolved, force cycles to be
- /// resolved.
- ///
- /// \pre No operands (or operands' operands, etc.) have \a isTemporary().
- void resolveCycles();
- /// Resolve a unique, unresolved node.
- void resolve();
- /// Replace a temporary node with a permanent one.
- ///
- /// Try to create a uniqued version of \c N -- in place, if possible -- and
- /// return it. If \c N cannot be uniqued, return a distinct node instead.
- template <class T>
- static std::enable_if_t<std::is_base_of<MDNode, T>::value, T *>
- replaceWithPermanent(std::unique_ptr<T, TempMDNodeDeleter> N) {
- return cast<T>(N.release()->replaceWithPermanentImpl());
- }
- /// Replace a temporary node with a uniqued one.
- ///
- /// Create a uniqued version of \c N -- in place, if possible -- and return
- /// it. Takes ownership of the temporary node.
- ///
- /// \pre N does not self-reference.
- template <class T>
- static std::enable_if_t<std::is_base_of<MDNode, T>::value, T *>
- replaceWithUniqued(std::unique_ptr<T, TempMDNodeDeleter> N) {
- return cast<T>(N.release()->replaceWithUniquedImpl());
- }
- /// Replace a temporary node with a distinct one.
- ///
- /// Create a distinct version of \c N -- in place, if possible -- and return
- /// it. Takes ownership of the temporary node.
- template <class T>
- static std::enable_if_t<std::is_base_of<MDNode, T>::value, T *>
- replaceWithDistinct(std::unique_ptr<T, TempMDNodeDeleter> N) {
- return cast<T>(N.release()->replaceWithDistinctImpl());
- }
- private:
- MDNode *replaceWithPermanentImpl();
- MDNode *replaceWithUniquedImpl();
- MDNode *replaceWithDistinctImpl();
- protected:
- /// Set an operand.
- ///
- /// Sets the operand directly, without worrying about uniquing.
- void setOperand(unsigned I, Metadata *New);
- void storeDistinctInContext();
- template <class T, class StoreT>
- static T *storeImpl(T *N, StorageType Storage, StoreT &Store);
- template <class T> static T *storeImpl(T *N, StorageType Storage);
- private:
- void handleChangedOperand(void *Ref, Metadata *New);
- /// Drop RAUW support, if any.
- void dropReplaceableUses();
- void resolveAfterOperandChange(Metadata *Old, Metadata *New);
- void decrementUnresolvedOperandCount();
- void countUnresolvedOperands();
- /// Mutate this to be "uniqued".
- ///
- /// Mutate this so that \a isUniqued().
- /// \pre \a isTemporary().
- /// \pre already added to uniquing set.
- void makeUniqued();
- /// Mutate this to be "distinct".
- ///
- /// Mutate this so that \a isDistinct().
- /// \pre \a isTemporary().
- void makeDistinct();
- void deleteAsSubclass();
- MDNode *uniquify();
- void eraseFromStore();
- template <class NodeTy> struct HasCachedHash;
- template <class NodeTy>
- static void dispatchRecalculateHash(NodeTy *N, std::true_type) {
- N->recalculateHash();
- }
- template <class NodeTy>
- static void dispatchRecalculateHash(NodeTy *, std::false_type) {}
- template <class NodeTy>
- static void dispatchResetHash(NodeTy *N, std::true_type) {
- N->setHash(0);
- }
- template <class NodeTy>
- static void dispatchResetHash(NodeTy *, std::false_type) {}
- public:
- using op_iterator = const MDOperand *;
- using op_range = iterator_range<op_iterator>;
- op_iterator op_begin() const {
- return const_cast<MDNode *>(this)->mutable_begin();
- }
- op_iterator op_end() const {
- return const_cast<MDNode *>(this)->mutable_end();
- }
- op_range operands() const { return op_range(op_begin(), op_end()); }
- const MDOperand &getOperand(unsigned I) const {
- assert(I < NumOperands && "Out of range");
- return op_begin()[I];
- }
- /// Return number of MDNode operands.
- unsigned getNumOperands() const { return NumOperands; }
- /// Methods for support type inquiry through isa, cast, and dyn_cast:
- static bool classof(const Metadata *MD) {
- switch (MD->getMetadataID()) {
- default:
- return false;
- #define HANDLE_MDNODE_LEAF(CLASS) \
- case CLASS##Kind: \
- return true;
- #include "llvm/IR/Metadata.def"
- }
- }
- /// Check whether MDNode is a vtable access.
- bool isTBAAVtableAccess() const;
- /// Methods for metadata merging.
- static MDNode *concatenate(MDNode *A, MDNode *B);
- static MDNode *intersect(MDNode *A, MDNode *B);
- static MDNode *getMostGenericTBAA(MDNode *A, MDNode *B);
- static MDNode *getMostGenericFPMath(MDNode *A, MDNode *B);
- static MDNode *getMostGenericRange(MDNode *A, MDNode *B);
- static MDNode *getMostGenericAliasScope(MDNode *A, MDNode *B);
- static MDNode *getMostGenericAlignmentOrDereferenceable(MDNode *A, MDNode *B);
- };
- /// Tuple of metadata.
- ///
- /// This is the simple \a MDNode arbitrary tuple. Nodes are uniqued by
- /// default based on their operands.
- class MDTuple : public MDNode {
- friend class LLVMContextImpl;
- friend class MDNode;
- MDTuple(LLVMContext &C, StorageType Storage, unsigned Hash,
- ArrayRef<Metadata *> Vals)
- : MDNode(C, MDTupleKind, Storage, Vals) {
- setHash(Hash);
- }
- ~MDTuple() { dropAllReferences(); }
- void setHash(unsigned Hash) { SubclassData32 = Hash; }
- void recalculateHash();
- static MDTuple *getImpl(LLVMContext &Context, ArrayRef<Metadata *> MDs,
- StorageType Storage, bool ShouldCreate = true);
- TempMDTuple cloneImpl() const {
- return getTemporary(getContext(), SmallVector<Metadata *, 4>(operands()));
- }
- public:
- /// Get the hash, if any.
- unsigned getHash() const { return SubclassData32; }
- static MDTuple *get(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return getImpl(Context, MDs, Uniqued);
- }
- static MDTuple *getIfExists(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return getImpl(Context, MDs, Uniqued, /* ShouldCreate */ false);
- }
- /// Return a distinct node.
- ///
- /// Return a distinct node -- i.e., a node that is not uniqued.
- static MDTuple *getDistinct(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return getImpl(Context, MDs, Distinct);
- }
- /// Return a temporary node.
- ///
- /// For use in constructing cyclic MDNode structures. A temporary MDNode is
- /// not uniqued, may be RAUW'd, and must be manually deleted with
- /// deleteTemporary.
- static TempMDTuple getTemporary(LLVMContext &Context,
- ArrayRef<Metadata *> MDs) {
- return TempMDTuple(getImpl(Context, MDs, Temporary));
- }
- /// Return a (temporary) clone of this.
- TempMDTuple clone() const { return cloneImpl(); }
- static bool classof(const Metadata *MD) {
- return MD->getMetadataID() == MDTupleKind;
- }
- };
- MDTuple *MDNode::get(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return MDTuple::get(Context, MDs);
- }
- MDTuple *MDNode::getIfExists(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return MDTuple::getIfExists(Context, MDs);
- }
- MDTuple *MDNode::getDistinct(LLVMContext &Context, ArrayRef<Metadata *> MDs) {
- return MDTuple::getDistinct(Context, MDs);
- }
- TempMDTuple MDNode::getTemporary(LLVMContext &Context,
- ArrayRef<Metadata *> MDs) {
- return MDTuple::getTemporary(Context, MDs);
- }
- void TempMDNodeDeleter::operator()(MDNode *Node) const {
- MDNode::deleteTemporary(Node);
- }
- /// This is a simple wrapper around an MDNode which provides a higher-level
- /// interface by hiding the details of how alias analysis information is encoded
- /// in its operands.
- class AliasScopeNode {
- const MDNode *Node = nullptr;
- public:
- AliasScopeNode() = default;
- explicit AliasScopeNode(const MDNode *N) : Node(N) {}
- /// Get the MDNode for this AliasScopeNode.
- const MDNode *getNode() const { return Node; }
- /// Get the MDNode for this AliasScopeNode's domain.
- const MDNode *getDomain() const {
- if (Node->getNumOperands() < 2)
- return nullptr;
- return dyn_cast_or_null<MDNode>(Node->getOperand(1));
- }
- StringRef getName() const {
- if (Node->getNumOperands() > 2)
- if (MDString *N = dyn_cast_or_null<MDString>(Node->getOperand(2)))
- return N->getString();
- return StringRef();
- }
- };
- /// Typed iterator through MDNode operands.
- ///
- /// An iterator that transforms an \a MDNode::iterator into an iterator over a
- /// particular Metadata subclass.
- template <class T>
- class TypedMDOperandIterator
- : public std::iterator<std::input_iterator_tag, T *, std::ptrdiff_t, void,
- T *> {
- MDNode::op_iterator I = nullptr;
- public:
- TypedMDOperandIterator() = default;
- explicit TypedMDOperandIterator(MDNode::op_iterator I) : I(I) {}
- T *operator*() const { return cast_or_null<T>(*I); }
- TypedMDOperandIterator &operator++() {
- ++I;
- return *this;
- }
- TypedMDOperandIterator operator++(int) {
- TypedMDOperandIterator Temp(*this);
- ++I;
- return Temp;
- }
- bool operator==(const TypedMDOperandIterator &X) const { return I == X.I; }
- bool operator!=(const TypedMDOperandIterator &X) const { return I != X.I; }
- };
- /// Typed, array-like tuple of metadata.
- ///
- /// This is a wrapper for \a MDTuple that makes it act like an array holding a
- /// particular type of metadata.
- template <class T> class MDTupleTypedArrayWrapper {
- const MDTuple *N = nullptr;
- public:
- MDTupleTypedArrayWrapper() = default;
- MDTupleTypedArrayWrapper(const MDTuple *N) : N(N) {}
- template <class U>
- MDTupleTypedArrayWrapper(
- const MDTupleTypedArrayWrapper<U> &Other,
- std::enable_if_t<std::is_convertible<U *, T *>::value> * = nullptr)
- : N(Other.get()) {}
- template <class U>
- explicit MDTupleTypedArrayWrapper(
- const MDTupleTypedArrayWrapper<U> &Other,
- std::enable_if_t<!std::is_convertible<U *, T *>::value> * = nullptr)
- : N(Other.get()) {}
- explicit operator bool() const { return get(); }
- explicit operator MDTuple *() const { return get(); }
- MDTuple *get() const { return const_cast<MDTuple *>(N); }
- MDTuple *operator->() const { return get(); }
- MDTuple &operator*() const { return *get(); }
- // FIXME: Fix callers and remove condition on N.
- unsigned size() const { return N ? N->getNumOperands() : 0u; }
- bool empty() const { return N ? N->getNumOperands() == 0 : true; }
- T *operator[](unsigned I) const { return cast_or_null<T>(N->getOperand(I)); }
- // FIXME: Fix callers and remove condition on N.
- using iterator = TypedMDOperandIterator<T>;
- iterator begin() const { return N ? iterator(N->op_begin()) : iterator(); }
- iterator end() const { return N ? iterator(N->op_end()) : iterator(); }
- };
- #define HANDLE_METADATA(CLASS) \
- using CLASS##Array = MDTupleTypedArrayWrapper<CLASS>;
- #include "llvm/IR/Metadata.def"
- /// Placeholder metadata for operands of distinct MDNodes.
- ///
- /// This is a lightweight placeholder for an operand of a distinct node. It's
- /// purpose is to help track forward references when creating a distinct node.
- /// This allows distinct nodes involved in a cycle to be constructed before
- /// their operands without requiring a heavyweight temporary node with
- /// full-blown RAUW support.
- ///
- /// Each placeholder supports only a single MDNode user. Clients should pass
- /// an ID, retrieved via \a getID(), to indicate the "real" operand that this
- /// should be replaced with.
- ///
- /// While it would be possible to implement move operators, they would be
- /// fairly expensive. Leave them unimplemented to discourage their use
- /// (clients can use std::deque, std::list, BumpPtrAllocator, etc.).
- class DistinctMDOperandPlaceholder : public Metadata {
- friend class MetadataTracking;
- Metadata **Use = nullptr;
- public:
- explicit DistinctMDOperandPlaceholder(unsigned ID)
- : Metadata(DistinctMDOperandPlaceholderKind, Distinct) {
- SubclassData32 = ID;
- }
- DistinctMDOperandPlaceholder() = delete;
- DistinctMDOperandPlaceholder(DistinctMDOperandPlaceholder &&) = delete;
- DistinctMDOperandPlaceholder(const DistinctMDOperandPlaceholder &) = delete;
- ~DistinctMDOperandPlaceholder() {
- if (Use)
- *Use = nullptr;
- }
- unsigned getID() const { return SubclassData32; }
- /// Replace the use of this with MD.
- void replaceUseWith(Metadata *MD) {
- if (!Use)
- return;
- *Use = MD;
- if (*Use)
- MetadataTracking::track(*Use);
- Metadata *T = cast<Metadata>(this);
- MetadataTracking::untrack(T);
- assert(!Use && "Use is still being tracked despite being untracked!");
- }
- };
- //===----------------------------------------------------------------------===//
- /// A tuple of MDNodes.
- ///
- /// Despite its name, a NamedMDNode isn't itself an MDNode.
- ///
- /// NamedMDNodes are named module-level entities that contain lists of MDNodes.
- ///
- /// It is illegal for a NamedMDNode to appear as an operand of an MDNode.
- class NamedMDNode : public ilist_node<NamedMDNode> {
- friend class LLVMContextImpl;
- friend class Module;
- std::string Name;
- Module *Parent = nullptr;
- void *Operands; // SmallVector<TrackingMDRef, 4>
- void setParent(Module *M) { Parent = M; }
- explicit NamedMDNode(const Twine &N);
- template<class T1, class T2>
- class op_iterator_impl :
- public std::iterator<std::bidirectional_iterator_tag, T2> {
- friend class NamedMDNode;
- const NamedMDNode *Node = nullptr;
- unsigned Idx = 0;
- op_iterator_impl(const NamedMDNode *N, unsigned i) : Node(N), Idx(i) {}
- public:
- op_iterator_impl() = default;
- bool operator==(const op_iterator_impl &o) const { return Idx == o.Idx; }
- bool operator!=(const op_iterator_impl &o) const { return Idx != o.Idx; }
- op_iterator_impl &operator++() {
- ++Idx;
- return *this;
- }
- op_iterator_impl operator++(int) {
- op_iterator_impl tmp(*this);
- operator++();
- return tmp;
- }
- op_iterator_impl &operator--() {
- --Idx;
- return *this;
- }
- op_iterator_impl operator--(int) {
- op_iterator_impl tmp(*this);
- operator--();
- return tmp;
- }
- T1 operator*() const { return Node->getOperand(Idx); }
- };
- public:
- NamedMDNode(const NamedMDNode &) = delete;
- ~NamedMDNode();
- /// Drop all references and remove the node from parent module.
- void eraseFromParent();
- /// Remove all uses and clear node vector.
- void dropAllReferences() { clearOperands(); }
- /// Drop all references to this node's operands.
- void clearOperands();
- /// Get the module that holds this named metadata collection.
- inline Module *getParent() { return Parent; }
- inline const Module *getParent() const { return Parent; }
- MDNode *getOperand(unsigned i) const;
- unsigned getNumOperands() const;
- void addOperand(MDNode *M);
- void setOperand(unsigned I, MDNode *New);
- StringRef getName() const;
- void print(raw_ostream &ROS, bool IsForDebug = false) const;
- void print(raw_ostream &ROS, ModuleSlotTracker &MST,
- bool IsForDebug = false) const;
- void dump() const;
- // ---------------------------------------------------------------------------
- // Operand Iterator interface...
- //
- using op_iterator = op_iterator_impl<MDNode *, MDNode>;
- op_iterator op_begin() { return op_iterator(this, 0); }
- op_iterator op_end() { return op_iterator(this, getNumOperands()); }
- using const_op_iterator = op_iterator_impl<const MDNode *, MDNode>;
- const_op_iterator op_begin() const { return const_op_iterator(this, 0); }
- const_op_iterator op_end() const { return const_op_iterator(this, getNumOperands()); }
- inline iterator_range<op_iterator> operands() {
- return make_range(op_begin(), op_end());
- }
- inline iterator_range<const_op_iterator> operands() const {
- return make_range(op_begin(), op_end());
- }
- };
- // Create wrappers for C Binding types (see CBindingWrapping.h).
- DEFINE_ISA_CONVERSION_FUNCTIONS(NamedMDNode, LLVMNamedMDNodeRef)
- } // end namespace llvm
- #endif // LLVM_IR_METADATA_H
- #ifdef __GNUC__
- #pragma GCC diagnostic pop
- #endif
|