-/* Copyright (c) 2015-2019. The SimGrid Team.
+/* Copyright (c) 2015-2023. The SimGrid Team.
* All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
#define SIMGRID_XBT_LIB_HPP
#include <cstddef>
+#include <functional>
#include <limits>
+#include <memory>
#include <vector>
-namespace simgrid {
-namespace xbt {
+namespace simgrid::xbt {
template<class T, class U> class Extension;
template<class T> class Extendable;
template<class T, class U>
class Extension {
- static const std::size_t INVALID_ID = std::numeric_limits<std::size_t>::max();
- std::size_t id_;
+ static constexpr std::size_t INVALID_ID = std::numeric_limits<std::size_t>::max();
+ std::size_t id_ = INVALID_ID;
friend class Extendable<T>;
explicit constexpr Extension(std::size_t id) : id_(id) {}
public:
- explicit constexpr Extension() : id_(INVALID_ID) {}
+ explicit constexpr Extension() = default;
std::size_t id() const { return id_; }
bool valid() const { return id_ != INVALID_ID; }
};
template<class T>
class Extendable {
private:
- static std::vector<void(*)(void*)> deleters_;
- std::vector<void*> extensions_;
+ static std::vector<std::function<void(void*)>> deleters_;
+ std::vector<void*> extensions_{deleters_.size(), nullptr};
+
public:
- static size_t extension_create(void (*deleter)(void*))
+ static size_t extension_create(const std::function<void(void*)>& deleter)
{
- if (deleters_.empty()) { // Save space for void* user data
- deleters_.push_back(nullptr);
- }
- deleters_.push_back(deleter);
+ deleters_.emplace_back(deleter);
return deleters_.size() - 1;
}
- template<class U>
- static Extension<T,U> extension_create(void (*deleter)(void*))
+ template <class U> static Extension<T, U> extension_create(const std::function<void(void*)>& deleter)
{
return Extension<T,U>(extension_create(deleter));
}
{
return Extension<T, U>(extension_create([](void* p) { delete static_cast<U*>(p); }));
}
- Extendable() : extensions_((deleters_.size() > 0 ? deleters_.size() : 1), nullptr) {}
+ Extendable() = default;
Extendable(const Extendable&) = delete;
Extendable& operator=(const Extendable&) = delete;
~Extendable()
* an extension of B might need to have the extension of A around when executing
* its cleanup function/destructor. */
for (std::size_t i = extensions_.size(); i > 1; --i) // rank=0 is the spot of user's void*
- if (extensions_[i - 1] != nullptr && deleters_[i - 1] != nullptr)
+ if (extensions_[i - 1] != nullptr && deleters_[i - 1])
deleters_[i - 1](extensions_[i - 1]);
}
void set_data(void* data){
extensions_[0]=data;
}
- void* get_data(){
- return extensions_[0];
- }
- // Convenience extension access when the type has a associated EXTENSION ID:
+ template <typename D> D* get_data() const { return static_cast<D*>(extensions_[0]); }
+ template <typename D> std::unique_ptr<D> get_unique_data() { return std::unique_ptr<D>(get_data<D>()); }
+
+ // Convenience extension access when the type has an associated EXTENSION ID:
template <class U> U* extension() const { return extension<U>(U::EXTENSION_ID); }
template<class U> void extension_set(U* p) { extension_set<U>(U::EXTENSION_ID, p); }
};
-template <class T> std::vector<void (*)(void*)> Extendable<T>::deleters_;
-}
-}
+// Initialized with a first element, to save space for void* user data
+template <class T> std::vector<std::function<void(void*)>> Extendable<T>::deleters_{1};
+} // namespace simgrid::xbt
#endif