contrib/llvm/include/llvm/Transforms/Utils/ValueMapper.h

   1 //===- ValueMapper.h - Remapping for constants and metadata -----*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file defines the MapValue interface which is used by various parts of
  11 // the Transforms/Utils library to implement cloning and linking facilities.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_TRANSFORMS_UTILS_VALUEMAPPER_H
  16 #define LLVM_TRANSFORMS_UTILS_VALUEMAPPER_H
  17
  18 #include "llvm/IR/ValueMap.h"
  19
  20 namespace llvm {
  21   class Value;
  22   class Instruction;
  23   typedef ValueMap<const Value *, WeakVH> ValueToValueMapTy;
  24
  25   /// ValueMapTypeRemapper - This is a class that can be implemented by clients
  26   /// to remap types when cloning constants and instructions.
  27   class ValueMapTypeRemapper {
  28     virtual void anchor();  // Out of line method.
  29   public:
  30     virtual ~ValueMapTypeRemapper() {}
  31
  32     /// remapType - The client should implement this method if they want to
  33     /// remap types while mapping values.
  34     virtual Type *remapType(Type *SrcTy) = 0;
  35   };
  36
  37   /// ValueMaterializer - This is a class that can be implemented by clients
  38   /// to materialize Values on demand.
  39   class ValueMaterializer {
  40     virtual void anchor(); // Out of line method.
  41
  42   protected:
  43     ~ValueMaterializer() = default;
  44     ValueMaterializer() = default;
  45     ValueMaterializer(const ValueMaterializer&) = default;
  46     ValueMaterializer &operator=(const ValueMaterializer&) = default;
  47
  48   public:
  49     /// The client should implement this method if they want to generate a
  50     /// mapped Value on demand. For example, if linking lazily.
  51     virtual Value *materializeDeclFor(Value *V) = 0;
  52
  53     /// If the data being mapped is recursive, the above function can map
  54     /// just the declaration and this is called to compute the initializer.
  55     /// It is called after the mapping is recorded, so it doesn't need to worry
  56     /// about recursion.
  57     virtual void materializeInitFor(GlobalValue *New, GlobalValue *Old);
  58
  59     /// If the client needs to handle temporary metadata it must implement
  60     /// these methods.
  61     virtual Metadata *mapTemporaryMetadata(Metadata *MD) { return nullptr; }
  62     virtual void replaceTemporaryMetadata(const Metadata *OrigMD,
  63                                           Metadata *NewMD) {}
  64
  65     /// The client should implement this method if some metadata need
  66     /// not be mapped, for example DISubprogram metadata for functions not
  67     /// linked into the destination module.
  68     virtual bool isMetadataNeeded(Metadata *MD) { return true; }
  69   };
  70
  71   /// RemapFlags - These are flags that the value mapping APIs allow.
  72   enum RemapFlags {
  73     RF_None = 0,
  74
  75     /// RF_NoModuleLevelChanges - If this flag is set, the remapper knows that
  76     /// only local values within a function (such as an instruction or argument)
  77     /// are mapped, not global values like functions and global metadata.
  78     RF_NoModuleLevelChanges = 1,
  79
  80     /// RF_IgnoreMissingEntries - If this flag is set, the remapper ignores
  81     /// entries that are not in the value map.  If it is unset, it aborts if an
  82     /// operand is asked to be remapped which doesn't exist in the mapping.
  83     RF_IgnoreMissingEntries = 2,
  84
  85     /// Instruct the remapper to move distinct metadata instead of duplicating
  86     /// it when there are module-level changes.
  87     RF_MoveDistinctMDs = 4,
  88
  89     /// Any global values not in value map are mapped to null instead of
  90     /// mapping to self. Illegal if RF_IgnoreMissingEntries is also set.
  91     RF_NullMapMissingGlobalValues = 8,
  92
  93     /// Set when there is still temporary metadata that must be handled,
  94     /// such as when we are doing function importing and will materialize
  95     /// and link metadata as a postpass.
  96     RF_HaveUnmaterializedMetadata = 16,
  97   };
  98
  99   static inline RemapFlags operator|(RemapFlags LHS, RemapFlags RHS) {
 100     return RemapFlags(unsigned(LHS)|unsigned(RHS));
 101   }
 102
 103   Value *MapValue(const Value *V, ValueToValueMapTy &VM,
 104                   RemapFlags Flags = RF_None,
 105                   ValueMapTypeRemapper *TypeMapper = nullptr,
 106                   ValueMaterializer *Materializer = nullptr);
 107
 108   Metadata *MapMetadata(const Metadata *MD, ValueToValueMapTy &VM,
 109                         RemapFlags Flags = RF_None,
 110                         ValueMapTypeRemapper *TypeMapper = nullptr,
 111                         ValueMaterializer *Materializer = nullptr);
 112
 113   /// MapMetadata - provide versions that preserve type safety for MDNodes.
 114   MDNode *MapMetadata(const MDNode *MD, ValueToValueMapTy &VM,
 115                       RemapFlags Flags = RF_None,
 116                       ValueMapTypeRemapper *TypeMapper = nullptr,
 117                       ValueMaterializer *Materializer = nullptr);
 118
 119   void RemapInstruction(Instruction *I, ValueToValueMapTy &VM,
 120                         RemapFlags Flags = RF_None,
 121                         ValueMapTypeRemapper *TypeMapper = nullptr,
 122                         ValueMaterializer *Materializer = nullptr);
 123
 124   /// MapValue - provide versions that preserve type safety for Constants.
 125   inline Constant *MapValue(const Constant *V, ValueToValueMapTy &VM,
 126                             RemapFlags Flags = RF_None,
 127                             ValueMapTypeRemapper *TypeMapper = nullptr,
 128                             ValueMaterializer *Materializer = nullptr) {
 129     return cast<Constant>(MapValue((const Value*)V, VM, Flags, TypeMapper,
 130                                    Materializer));
 131   }
 132
 133 } // End llvm namespace
 134
 135 #endif