memory_handle_builder.h

#pragma once
 
#include <cstddef>
#include <memory>
#include <utility>
 
#include <ucs/memory/memory_type.h>
 
namespace ucxx {
 
class Context;
class MemoryHandle;

class MemoryHandleBuilder final {
 public:
  MemoryHandleBuilder(std::shared_ptr<Context> context, size_t size);

  ~MemoryHandleBuilder();

  MemoryHandleBuilder(const MemoryHandleBuilder& other);
  MemoryHandleBuilder& operator=(const MemoryHandleBuilder& other);
  MemoryHandleBuilder(MemoryHandleBuilder&&) noexcept;
  MemoryHandleBuilder& operator=(MemoryHandleBuilder&&) noexcept;

  operator std::shared_ptr<MemoryHandle>();

  MemoryHandleBuilder& buffer(void* buffer);

  MemoryHandleBuilder& memoryType(ucs_memory_type_t memoryType);

  [[nodiscard]] std::shared_ptr<MemoryHandle> build();
 
 private:
  struct Impl;
  std::unique_ptr<Impl> _impl;
};
 
}  // namespace ucxx