host.h

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <cstddef>
 
#include <rex/platform.h>
#include <rex/types.h>
 
namespace rex::memory {

enum class Tag : u8 {
  kCore = 0,       
  kThirdParty = 1, 
};

inline constexpr size_t kMaxTags = 16;
 
struct TagStats {
  u64 live_bytes = 0;  
  u64 live_count = 0;  
  u64 peak_bytes = 0;  
  u64 total_count = 0; 
};
 
struct HostStats {
  std::array<TagStats, kMaxTags>
      tags{}; 
  TagStats total;
};

REX_API void* HostAlloc(size_t size, size_t align = 0) noexcept;

REX_API void HostFree(void* ptr, size_t size = 0, size_t align = 0) noexcept;

REX_API size_t HostSize(const void* ptr) noexcept;

REX_API HostStats GetHostStats() noexcept;

REX_API void SetAllocStackDepth(i32 depth) noexcept;

[[nodiscard]] REX_API i32 GetAllocStackDepth() noexcept;

REX_API Tag GetThreadTag() noexcept;

REX_API const char* GetTagName(Tag tag) noexcept;

REX_API void SetTagName(Tag tag, const char* name) noexcept;

class REX_API TagScope {
 public:
  explicit TagScope(Tag tag) noexcept;

  ~TagScope();
 
  TagScope(const TagScope&) = delete;
  TagScope& operator=(const TagScope&) = delete;
 
 private:
  Tag previous_;
};
 
namespace detail {

enum class BlockError : u8 {
  kForeignPointer, 
  kTypeMismatch, 
  kSizeMismatch, 
};

using BlockErrorHandler = void (*)(BlockError error, const void* ptr);

REX_API BlockErrorHandler SetBlockErrorHandler(BlockErrorHandler handler) noexcept;

REX_API void FailNextAllocations(u32 count) noexcept;

enum class AllocType : u8 {
  kScalar, 
  kArray,  
  kRaw,    
};

REX_API void* NewBlock(size_t size, size_t align, AllocType type, bool can_throw);

REX_API void DeleteBlock(void* ptr, size_t size, AllocType type) noexcept;
 
}  // namespace detail
 
}  // namespace rex::memory