ostream.hpp

Go to the documentation of this file.

 
#ifndef SIMPLEKERNEL_OSTREAM_HPP
#define SIMPLEKERNEL_OSTREAM_HPP
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include "efi.h"
#include "efilib.h"
 
#ifdef __cplusplus
}
#endif
 
[[maybe_unused]] auto wait_for_input(EFI_INPUT_KEY *_key) -> EFI_STATUS;
 
class ostream {
public:
  ostream() = default;
 
  ~ostream() = default;
 
  ostream(const ostream &) = delete;
  ostream(ostream &&) = delete;
  auto operator=(const ostream &) -> ostream & = delete;
  auto operator=(ostream &&) -> ostream & = delete;
 
  template <class _t> inline auto operator<<(_t _val) -> ostream &;
 
  auto operator<<(ostream &(*_ostream)(ostream &)) -> ostream &;
 
  static auto hex_x(ostream &_ostream) -> ostream &;
 
  static auto hex_X(ostream &_ostream) -> ostream &;
 
  static auto endl(ostream &_ostream) -> ostream &;
 
private:
  enum mode_t : uint8_t {
    x,
    X,
    d,
  };
  mode_t mode;
};
 
 
template <class _t> auto ostream::operator<<(_t _val) -> ostream & {
  *this << (uint64_t)_val;
  return *this;
}
 
template <> inline auto ostream::operator<<(int32_t _val) -> ostream & {
  if (mode == d) {
    Print(L"%d", _val);
  } else if (mode == x) {
    Print(L"0x%x", _val);
  } else if (mode == X) {
    Print(L"0x%X", _val);
  }
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(uint32_t _val) -> ostream & {
  if (mode == d) {
    Print(L"%d", _val);
  } else if (mode == x) {
    Print(L"0x%x", _val);
  } else if (mode == X) {
    Print(L"0x%X", _val);
  }
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(int64_t _val) -> ostream & {
  if (mode == d) {
    Print(L"%ld", _val);
  } else if (mode == x) {
    Print(L"0x%x", _val);
  } else if (mode == X) {
    Print(L"0x%X", _val);
  }
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(uint64_t _val) -> ostream & {
  if (mode == d) {
    Print(L"%ld", _val);
  } else if (mode == x) {
    Print(L"0x%x", _val);
  } else if (mode == X) {
    Print(L"0x%X", _val);
  }
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(wchar_t _val) -> ostream & {
  Print(L"%c", _val);
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(const wchar_t *_val) -> ostream & {
  Print(L"%s", _val);
  mode = d;
  return *this;
}
 
template <> inline auto ostream::operator<<(void *_val) -> ostream & {
  *this << reinterpret_cast<uint64_t>(_val);
  return *this;
}
 
template <> inline auto ostream::operator<<(const void *_val) -> ostream & {
  *this << reinterpret_cast<uint64_t>(_val);
  return *this;
}
 
 
static ostream debug;
 
#endif /* SIMPLEKERNEL_OSTREAM_HPP */