-//===- llvm/Transforms/Utils/Linker.h - Module Linker Interface -*- C++ -*-===//
-//
-// The LLVM Compiler Infrastructure
+//===- llvm/Linker.h - Module Linker Interface ------------------*- C++ -*-===//
//
-// This file was developed by the LLVM research group and is distributed under
-// the University of Illinois Open Source License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
+// The LLVM Compiler Infrastructure
//
-// This file defines the interface to the module linker.
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
-#ifndef LLVM_TRANSFORMATIONS_UTILS_LINKER_H
-#define LLVM_TRANSFORMATIONS_UTILS_LINKER_H
+#ifndef LLVM_LINKER_H
+#define LLVM_LINKER_H
+#include "llvm/ADT/SmallPtrSet.h"
#include <string>
namespace llvm {
class Module;
-
-// LinkModules - This function links two modules together, with the resulting
-// left module modified to be the composite of the two input modules. If an
-// error occurs, true is returned and ErrorMsg (if not null) is set to indicate
-// the problem.
-//
-bool LinkModules(Module *Dest, const Module *Src, std::string *ErrorMsg = 0);
+class StringRef;
+class StructType;
+
+/// This class provides the core functionality of linking in LLVM. It keeps a
+/// pointer to the merged module so far. It doesn't take ownership of the
+/// module since it is assumed that the user of this class will want to do
+/// something with it after the linking.
+class Linker {
+ public:
+ enum LinkerMode {
+ DestroySource = 0, // Allow source module to be destroyed.
+ PreserveSource = 1 // Preserve the source module.
+ };
+
+ Linker(Module *M);
+ ~Linker();
+ Module *getModule() const { return Composite; }
+
+ /// \brief Link \p Src into the composite. The source is destroyed if
+ /// \p Mode is DestroySource and preserved if it is PreserveSource.
+ /// If \p ErrorMsg is not null, information about any error is written
+ /// to it.
+ /// Returns true on error.
+ bool linkInModule(Module *Src, unsigned Mode, std::string *ErrorMsg);
+ bool linkInModule(Module *Src, std::string *ErrorMsg) {
+ return linkInModule(Src, Linker::DestroySource, ErrorMsg);
+ }
+
+ static bool LinkModules(Module *Dest, Module *Src, unsigned Mode,
+ std::string *ErrorMsg);
+
+ private:
+ Module *Composite;
+ SmallPtrSet<StructType*, 32> IdentifiedStructTypes;
+};
} // End llvm namespace
#endif
-