Open3D (C++ API)  0.18.0+6e910d9
StdAllocator.h
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // - Open3D: www.open3d.org -
3 // ----------------------------------------------------------------------------
4 // Copyright (c) 2018-2024 www.open3d.org
5 // SPDX-License-Identifier: MIT
6 // ----------------------------------------------------------------------------
7 
8 #pragma once
9 
10 #include "open3d/core/Device.h"
12 
13 namespace open3d {
14 namespace core {
15 
22 template <typename T>
23 class StdAllocator {
24 public:
26  using value_type = T;
27 
29  StdAllocator() = default;
30 
32  explicit StdAllocator(int device_id) : device_id_(device_id) {}
33 
35  StdAllocator(const StdAllocator&) = default;
36 
38  StdAllocator& operator=(const StdAllocator&) = default;
39 
41  StdAllocator(StdAllocator&&) = default;
42 
45 
47  template <typename U>
48  StdAllocator(const StdAllocator<U>& other) : device_id_(other.device_id_) {}
49 
52  void* ptr = MemoryManager::Malloc(
53  n * sizeof(T), Device(Device::DeviceType::CUDA, device_id_));
54  return static_cast<T*>(ptr);
55  }
56 
58  void deallocate(T* p, std::size_t n) {
60  }
61 
63  bool operator==(const StdAllocator& other) const {
64  return device_id_ == other.device_id_;
65  }
66 
68  bool operator!=(const StdAllocator& other) const {
69  return !operator==(other);
70  }
71 
73  int GetDeviceID() const { return device_id_; }
74 
75 private:
76  // Allow access in rebind constructor.
77  template <typename T2>
78  friend class StdAllocator;
79 
80  int device_id_;
81 };
82 
83 } // namespace core
84 } // namespace open3d
Definition: Device.h:18
static void * Malloc(size_t byte_size, const Device &device)
Definition: MemoryManager.cpp:22
static void Free(void *ptr, const Device &device)
Frees previously allocated memory at address ptr on device device.
Definition: MemoryManager.cpp:28
Definition: StdAllocator.h:23
int GetDeviceID() const
Returns the device on which memory is allocated.
Definition: StdAllocator.h:73
void deallocate(T *p, std::size_t n)
Deallocates memory from pointer p of size n .
Definition: StdAllocator.h:58
StdAllocator(const StdAllocator &)=default
Default copy constructor.
bool operator==(const StdAllocator &other) const
Returns true if the instances are equal, false otherwise.
Definition: StdAllocator.h:63
T value_type
T.
Definition: StdAllocator.h:26
bool operator!=(const StdAllocator &other) const
Returns true if the instances are not equal, false otherwise.
Definition: StdAllocator.h:68
StdAllocator(const StdAllocator< U > &other)
Rebind copy constructor.
Definition: StdAllocator.h:48
T * allocate(std::size_t n)
Allocates memory of size n.
Definition: StdAllocator.h:51
StdAllocator & operator=(const StdAllocator &)=default
Default copy assignment operator.
StdAllocator(int device_id)
Constructor from device.
Definition: StdAllocator.h:32
StdAllocator()=default
Default constructor.
StdAllocator(StdAllocator &&)=default
Default move constructor.
StdAllocator & operator=(StdAllocator &&)=default
Default move assignment operator.
const char const char value recording_handle imu_sample recording_handle uint8_t size_t data_size k4a_record_configuration_t config target_format k4a_capture_t capture_handle k4a_imu_sample_t imu_sample playback_handle k4a_logging_message_cb_t void min_level device_handle k4a_imu_sample_t timeout_in_ms capture_handle capture_handle capture_handle image_handle temperature_c k4a_image_t image_handle uint8_t image_handle image_handle image_handle image_handle image_handle timestamp_usec white_balance image_handle k4a_device_configuration_t config device_handle char size_t serial_number_size bool int32_t int32_t int32_t int32_t k4a_color_control_mode_t default_mode value const const k4a_calibration_t calibration char size_t
Definition: K4aPlugin.cpp:719
Definition: PinholeCameraIntrinsic.cpp:16