data_printer.h

// This file if part of the llir-opt project.
// Licensing information can be found in the LICENSE file.
// (C) 2018 Nandor Licker. All rights reserved.
 
#pragma once
 
#include <map>
 
#include <llvm/Pass.h>
#include <llvm/IR/DataLayout.h>
#include <llvm/MC/MCObjectFileInfo.h>
#include <llvm/MC/MCStreamer.h>
 
#include "core/visibility.h"
 
class Prog;
class Data;
class Atom;
class Object;
class ISelMapping;
 
 
 
class DataPrinter final : public llvm::ModulePass {
public:
  static char ID;
 
  DataPrinter(
      const Prog &Prog,
      ISelMapping *isel,
      llvm::MCContext *ctx,
      llvm::MCStreamer *os,
      const llvm::MCObjectFileInfo *objInfo,
      const llvm::DataLayout &layout,
      bool shared
  );
 
private:
  bool runOnModule(llvm::Module &M) override;
  llvm::StringRef getPassName() const override;
  void getAnalysisUsage(llvm::AnalysisUsage &AU) const override;
 
private:
  void LowerExtern(const Extern &ext);
  void LowerSection(const Data &data);
  void LowerObject(const Object &object);
  void LowerAtom(const Atom &atom);
  llvm::MCSymbol *LowerSymbol(const std::string_view name);
  void EmitVisibility(llvm::MCSymbol *sym, Visibility visibility);
 
  using XtorMap = std::map<int, std::vector<const Func *>>;
  void LowerXtors(const XtorMap &map, const std::string &name);
 
private:
  llvm::MCSection *GetSection(const Data &data);
 
private:
  const Prog &prog_;
  ISelMapping *isel_;
  llvm::MCContext *ctx_;
  llvm::MCStreamer *os_;
  const llvm::MCObjectFileInfo *objInfo_;
  const llvm::DataLayout &layout_;
  bool shared_;
};