X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/377d2091c1e0f30ef48fb068f672ae624ad59d0f..a96793ec8c496eec19daf68f3aa0c4eded2ea166:/include/xbt/functional.hpp diff --git a/include/xbt/functional.hpp b/include/xbt/functional.hpp index 10c16bf98a..1fefcd8473 100644 --- a/include/xbt/functional.hpp +++ b/include/xbt/functional.hpp @@ -1,5 +1,4 @@ -/* Copyright (c) 2015-2016. The SimGrid Team. - * All rights reserved. */ +/* Copyright (c) 2015-2019. The SimGrid Team. All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ @@ -7,6 +6,9 @@ #ifndef XBT_FUNCTIONAL_HPP #define XBT_FUNCTIONAL_HPP +#include +#include + #include #include #include @@ -21,43 +23,40 @@ #include #include -#include -#include - namespace simgrid { namespace xbt { -template -class MainFunction { -private: +template class MainFunction { F code_; std::shared_ptr> args_; + public: - MainFunction(F code, std::vector args) : - code_(std::move(code)), - args_(std::make_shared>(std::move(args))) - {} + MainFunction(F code, std::vector&& args) + : code_(std::move(code)), args_(std::make_shared>(std::move(args))) + { + } void operator()() const { - char noarg[] = {'\0'}; - const int argc = args_->size(); + const int argc = args_->size(); std::vector args = *args_; - std::unique_ptr argv(new char*[argc + 1]); - for (int i = 0; i != argc; ++i) - argv[i] = args[i].empty() ? noarg : &args[i].front(); - argv[argc] = nullptr; - code_(argc, argv.get()); + if (not args.empty()) { + char noarg[] = {'\0'}; + std::unique_ptr argv(new char*[argc + 1]); + for (int i = 0; i != argc; ++i) + argv[i] = args[i].empty() ? noarg : &args[i].front(); + argv[argc] = nullptr; + code_(argc, argv.get()); + } else + code_(argc, nullptr); } }; -template inline -std::function wrapMain(F code, std::vector args) +template inline std::function wrap_main(F code, std::vector&& args) { return MainFunction(std::move(code), std::move(args)); } -template inline -std::function wrapMain(F code, int argc, const char*const argv[]) +template inline std::function wrap_main(F code, int argc, const char* const argv[]) { std::vector args(argv, argv + argc); return MainFunction(std::move(code), std::move(args)); @@ -109,29 +108,15 @@ template class Task; */ template class Task { -private: - // Placeholder for some class type: struct whatever {}; // Union used for storage: -#if 0 typedef typename std::aligned_union<0, void*, std::pair, std::pair >::type TaskUnion; -#else - union TaskUnion { - void* ptr; - std::pair funcptr; - std::pair memberptr; - char any1[sizeof(std::pair)]; - char any2[sizeof(std::pair)]; - TaskUnion() {} - ~TaskUnion() {} - }; -#endif // Is F suitable for small buffer optimization? template @@ -168,9 +153,8 @@ private: } public: - - Task() {} - Task(std::nullptr_t) {} + Task() { /* Nothing to do */} + explicit Task(std::nullptr_t) { /* Nothing to do */} ~Task() { this->clear(); @@ -183,35 +167,36 @@ public: if (that.vtable_ && that.vtable_->move) that.vtable_->move(buffer_, that.buffer_); else - std::memcpy(&buffer_, &that.buffer_, sizeof(buffer_)); + std::memcpy(static_cast(&buffer_), static_cast(&that.buffer_), sizeof(buffer_)); + vtable_ = that.vtable_; that.vtable_ = nullptr; } - Task& operator=(Task that) + Task& operator=(Task const& that) = delete; + Task& operator=(Task&& that) { this->clear(); if (that.vtable_ && that.vtable_->move) that.vtable_->move(buffer_, that.buffer_); else - std::memcpy(&buffer_, &that.buffer_, sizeof(buffer_)); + std::memcpy(static_cast(&buffer_), static_cast(&that.buffer_), sizeof(buffer_)); vtable_ = that.vtable_; that.vtable_ = nullptr; return *this; } private: - template typename std::enable_if()>::type init(F code) { const static TaskVtable vtable { // Call: - [](TaskUnion& buffer, Args... args) -> R { + [](TaskUnion& buffer, Args&&... args) { F* src = reinterpret_cast(&buffer); F code = std::move(*src); src->~F(); - code(std::forward(args)...); + return code(std::forward(args)...); }, // Destroy: std::is_trivially_destructible::value ? @@ -232,13 +217,11 @@ private: vtable_ = &vtable; } - template - typename std::enable_if()>::type - init(F code) + template typename std::enable_if()>::type init(F code) { const static TaskVtable vtable { // Call: - [](TaskUnion& buffer, Args... args) -> R { + [](TaskUnion& buffer, Args&&... args) { // Delete F when we go out of scope: std::unique_ptr code(*reinterpret_cast(&buffer)); return (*code)(std::forward(args)...); @@ -256,17 +239,12 @@ private: } public: - - template - Task(F code) - { - this->init(std::move(code)); - } + template explicit Task(F code) { this->init(std::move(code)); } operator bool() const { return vtable_ != nullptr; } bool operator!() const { return vtable_ == nullptr; } - R operator()(Args... args) + R operator()(Args&&... args) { if (vtable_ == nullptr) throw std::bad_function_call(); @@ -278,7 +256,6 @@ public: template class TaskImpl { -private: F code_; std::tuple args_; typedef decltype(simgrid::xbt::apply(std::move(code_), std::move(args_))) result_type; @@ -293,15 +270,12 @@ public: } }; -template -auto makeTask(F code, Args... args) --> Task< decltype(code(std::move(args)...))() > +template auto make_task(F code, Args... args) -> Task { TaskImpl task(std::move(code), std::make_tuple(std::move(args)...)); - return std::move(task); -} - -} + return Task(std::move(task)); } +} // namespace xbt +} // namespace simgrid #endif