contrib/llvm/lib/Target/WebAssembly/MCTargetDesc/WebAssemblyTargetStreamer.h

   1 //==-- WebAssemblyTargetStreamer.h - WebAssembly Target Streamer -*- 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 /// \file
  11 /// \brief This file declares WebAssembly-specific target streamer classes.
  12 /// These are for implementing support for target-specific assembly directives.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_LIB_TARGET_WEBASSEMBLY_MCTARGETDESC_WEBASSEMBLYTARGETSTREAMER_H
  17 #define LLVM_LIB_TARGET_WEBASSEMBLY_MCTARGETDESC_WEBASSEMBLYTARGETSTREAMER_H
  18
  19 #include "llvm/BinaryFormat/Wasm.h"
  20 #include "llvm/CodeGen/MachineValueType.h"
  21 #include "llvm/MC/MCStreamer.h"
  22
  23 namespace llvm {
  24
  25 class MCELFStreamer;
  26 class MCWasmStreamer;
  27
  28 /// WebAssembly-specific streamer interface, to implement support
  29 /// WebAssembly-specific assembly directives.
  30 class WebAssemblyTargetStreamer : public MCTargetStreamer {
  31 public:
  32   explicit WebAssemblyTargetStreamer(MCStreamer &S);
  33
  34   /// .param
  35   virtual void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) = 0;
  36   /// .result
  37   virtual void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) = 0;
  38   /// .local
  39   virtual void emitLocal(ArrayRef<MVT> Types) = 0;
  40   /// .globalvar
  41   virtual void emitGlobal(ArrayRef<wasm::Global> Globals) = 0;
  42   /// .endfunc
  43   virtual void emitEndFunc() = 0;
  44   /// .functype
  45   virtual void emitIndirectFunctionType(MCSymbol *Symbol,
  46                                         SmallVectorImpl<MVT> &Params,
  47                                         SmallVectorImpl<MVT> &Results) = 0;
  48   /// .indidx
  49   virtual void emitIndIdx(const MCExpr *Value) = 0;
  50   /// .import_global
  51   virtual void emitGlobalImport(StringRef name) = 0;
  52
  53 protected:
  54   void emitValueType(wasm::ValType Type);
  55 };
  56
  57 /// This part is for ascii assembly output
  58 class WebAssemblyTargetAsmStreamer final : public WebAssemblyTargetStreamer {
  59   formatted_raw_ostream &OS;
  60
  61 public:
  62   WebAssemblyTargetAsmStreamer(MCStreamer &S, formatted_raw_ostream &OS);
  63
  64   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
  65   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
  66   void emitLocal(ArrayRef<MVT> Types) override;
  67   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
  68   void emitEndFunc() override;
  69   void emitIndirectFunctionType(MCSymbol *Symbol,
  70                                 SmallVectorImpl<MVT> &Params,
  71                                 SmallVectorImpl<MVT> &Results) override;
  72   void emitIndIdx(const MCExpr *Value) override;
  73   void emitGlobalImport(StringRef name) override;
  74 };
  75
  76 /// This part is for ELF object output
  77 class WebAssemblyTargetELFStreamer final : public WebAssemblyTargetStreamer {
  78 public:
  79   explicit WebAssemblyTargetELFStreamer(MCStreamer &S);
  80
  81   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
  82   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
  83   void emitLocal(ArrayRef<MVT> Types) override;
  84   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
  85   void emitEndFunc() override;
  86   void emitIndirectFunctionType(MCSymbol *Symbol,
  87                                 SmallVectorImpl<MVT> &Params,
  88                                 SmallVectorImpl<MVT> &Results) override;
  89   void emitIndIdx(const MCExpr *Value) override;
  90   void emitGlobalImport(StringRef name) override;
  91 };
  92
  93 /// This part is for Wasm object output
  94 class WebAssemblyTargetWasmStreamer final : public WebAssemblyTargetStreamer {
  95 public:
  96   explicit WebAssemblyTargetWasmStreamer(MCStreamer &S);
  97
  98   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
  99   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
 100   void emitLocal(ArrayRef<MVT> Types) override;
 101   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
 102   void emitEndFunc() override;
 103   void emitIndirectFunctionType(MCSymbol *Symbol,
 104                                 SmallVectorImpl<MVT> &Params,
 105                                 SmallVectorImpl<MVT> &Results) override;
 106   void emitIndIdx(const MCExpr *Value) override;
 107   void emitGlobalImport(StringRef name) override;
 108 };
 109
 110 } // end namespace llvm
 111
 112 #endif