Serialise globals.

Author: ptersilie

llvm/lib/YkIR/YkIRWriter.cpp

@@ -7,6 +7,7 @@
 #include "llvm/BinaryFormat/ELF.h"
 #include "llvm/IR/Constant.h"
 #include "llvm/IR/Constants.h"
+#include "llvm/IR/GlobalValue.h"
 #include "llvm/IR/InstrTypes.h"
 #include "llvm/IR/Instructions.h"
 #include "llvm/IR/Module.h"
@@ -56,6 +57,7 @@ enum OperandKind {
   Function,
   Block,
   Arg,
+  Global,
   UnimplementedOperand = 255,
 };
 
@@ -128,6 +130,7 @@ class YkIRWriter {
 
   vector<llvm::Type *> Types;
   vector<llvm::Constant *> Constants;
+  vector<llvm::GlobalVariable *> Globals;
 
   // Return the index of the LLVM type `Ty`, inserting a new entry if
   // necessary.
@@ -164,6 +167,19 @@ class YkIRWriter {
     return Idx;
   }
 
+  // Return the index of the LLVM global `G`, inserting a new entry if
+  // necessary.
+  size_t globalIndex(class GlobalVariable *G) {
+    vector<class GlobalVariable *>::iterator Found =
+        std::find(Globals.begin(), Globals.end(), G);
+    if (Found != Globals.end()) {
+      return std::distance(Globals.begin(), Found);
+    }
+    size_t Idx = Globals.size();
+    Globals.push_back(G);
+    return Idx;
+  }
+
   size_t functionIndex(llvm::Function *F) {
     // FIXME: For now we assume that function indicies in LLVM IR and our IR
     // are the same.
@@ -221,9 +237,16 @@ class YkIRWriter {
     OutStreamer.emitSizeT(A->getArgNo());
   }
 
+  void serialiseGlobalOperand(GlobalVariable *G) {
+    OutStreamer.emitInt8(OperandKind::Global);
+    OutStreamer.emitSizeT(globalIndex(G));
+  }
+
   void serialiseOperand(Instruction *Parent, ValueLoweringMap &VLMap,
                         Value *V) {
-    if (llvm::Function *F = dyn_cast<llvm::Function>(V)) {
+    if (llvm::GlobalVariable *G = dyn_cast<llvm::GlobalVariable>(V)) {
+      serialiseGlobalOperand(G);
+    } else if (llvm::Function *F = dyn_cast<llvm::Function>(V)) {
       serialiseFunctionOperand(F);
     } else if (llvm::Constant *C = dyn_cast<llvm::Constant>(V)) {
       serialiseConstantOperand(Parent, C);
@@ -511,6 +534,11 @@ class YkIRWriter {
     }
   }
 
+  void serialiseGlobal(class GlobalVariable *G) {
+    OutStreamer.emitInt8(G->isThreadLocal());
+    serialiseString(G->getName());
+  }
+
 public:
   YkIRWriter(Module &M, MCStreamer &OutStreamer)
       : M(M), OutStreamer(OutStreamer), DL(&M) {}
@@ -542,6 +570,13 @@ class YkIRWriter {
       serialiseConstant(C);
     }
 
+    // num_globals:
+    OutStreamer.emitSizeT(Globals.size());
+    // globals:
+    for (class GlobalVariable *&G : Globals) {
+      serialiseGlobal(G);
+    }
+
     // num_types:
     OutStreamer.emitSizeT(Types.size());
     // types: