context.h

#pragma once
 
#include <cstdint>
#include <cstring>
#include <memory>
#include <string>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/component.h>
#include <ucxx/config.h>
#include <ucxx/context_builder.h>
#include <ucxx/memory_handle_builder.h>
#include <ucxx/worker_builder.h>
 
namespace ucxx {
 
class ContextBuilder;
 
class MemoryHandle;
class Worker;

class Context : public Component {
 private:
  ucp_context_h _handle{nullptr};  
  Config _config{{}};              
  uint64_t _featureFlags{0};       
  bool _cudaSupport{false};        

  Context(const ConfigMap ucxConfig, const uint64_t featureFlags);
 
 public:
  static constexpr uint64_t defaultFeatureFlags =
    UCP_FEATURE_TAG | UCP_FEATURE_WAKEUP | UCP_FEATURE_STREAM | UCP_FEATURE_AM |
    UCP_FEATURE_RMA;  
 
  Context()                          = delete;
  Context(const Context&)            = delete;
  Context& operator=(Context const&) = delete;
  Context(Context&& o)               = delete;
  Context& operator=(Context&& o)    = delete;
 
  // Allow internal construction without exposing factory functions.
  friend class detail::ConstructorFactory;

  friend class ContextBuilder;

  ~Context();

  [[nodiscard]] ConfigMap getConfig();

  [[nodiscard]] ucp_context_h getHandle();

  [[nodiscard]] std::string getInfo();

  [[nodiscard]] uint64_t getFeatureFlags() const;

  [[nodiscard]] bool hasCudaSupport() const;

  [[nodiscard]] WorkerBuilder workerBuilder();

  [[nodiscard]] MemoryHandleBuilder memoryHandleBuilder(size_t size);
};
 
}  // namespace ucxx