]> git.sesse.net Git - casparcg/blobdiff - common/concurrency/executor.h
2.1.0: Refactoring.
[casparcg] / common / concurrency / executor.h
index a422b11fd5431f71690d36b8ca0d3be64470cc97..8d1ed1a1b846360fc80889346173cf71b9fa2efa 100644 (file)
 \r
 #include "../except.h"\r
 #include "../log.h"\r
-#include "../utility/move_on_copy.h"\r
-\r
-#include <Windows.h>\r
+#include "../enum_class.h"\r
+#include "../os/windows/windows.h"\r
 \r
 #include <tbb/atomic.h>\r
 #include <tbb/concurrent_queue.h>\r
 \r
 #include <boost/thread.hpp>\r
-#include <boost/optional.hpp>\r
-#include <boost/noncopyable.hpp>\r
 \r
 #include <functional>\r
 \r
 namespace caspar {\r
 \r
 namespace detail {\r
+       \r
+template<typename T>\r
+struct move_on_copy\r
+{\r
+       move_on_copy(const move_on_copy<T>& other) : value(std::move(other.value)){}\r
+       move_on_copy(T&& value) : value(std::move(value)){}\r
+       mutable T value;\r
+};\r
 \r
-//typedef struct tagTHREADNAME_INFO\r
-//{\r
-//     DWORD dwType; // must be 0x1000\r
-//     LPCSTR szName; // pointer to name (in user addr space)\r
-//     DWORD dwThreadID; // thread ID (-1=caller thread)\r
-//     DWORD dwFlags; // reserved for future use, must be zero\r
-//} THREADNAME_INFO;\r
-//\r
-//inline void SetThreadName(DWORD dwThreadID, LPCSTR szThreadName)\r
-//{\r
-//     THREADNAME_INFO info;\r
-//     {\r
-//             info.dwType = 0x1000;\r
-//             info.szName = szThreadName;\r
-//             info.dwThreadID = dwThreadID;\r
-//             info.dwFlags = 0;\r
-//     }\r
-//     __try\r
-//     {\r
-//             RaiseException( 0x406D1388, 0, sizeof(info)/sizeof(DWORD), (DWORD*)&info );\r
-//     }\r
-//     __except (EXCEPTION_CONTINUE_EXECUTION){}       \r
-//}\r
-//\r
-//}\r
+template<typename T>\r
+move_on_copy<T> make_move_on_copy(T&& value)\r
+{\r
+       return move_on_copy<T>(std::move(value));\r
 }\r
 \r
-enum task_priority\r
+}\r
+       \r
+struct task_priority_def\r
 {\r
-       high_priority,\r
-       normal_priority,\r
-       priority_count\r
+       enum type\r
+       {\r
+               high_priority,\r
+               normal_priority,\r
+               priority_count\r
+       };\r
 };\r
+typedef enum_class<task_priority_def> task_priority;\r
 \r
-enum thread_priority\r
+struct thread_priority_def\r
 {\r
-       high_priority_class,\r
-       above_normal_priority_class,\r
-       normal_priority_class,\r
-       below_normal_priority_class\r
+       enum type\r
+       {\r
+               high_priority_class,\r
+               above_normal_priority_class,\r
+               normal_priority_class,\r
+               below_normal_priority_class\r
+       };\r
 };\r
+typedef enum_class<thread_priority_def> thread_priority;\r
 \r
-class executor : boost::noncopyable\r
+class executor\r
 {\r
+       executor(const executor&);\r
+       executor& operator=(const executor&);\r
+\r
        const std::string name_;\r
        boost::thread thread_;\r
        tbb::atomic<bool> is_running_;\r
        \r
        typedef tbb::concurrent_bounded_queue<std::function<void()>> function_queue;\r
-       function_queue execution_queue_[priority_count];\r
+       function_queue execution_queue_[task_priority::priority_count];\r
                \r
        template<typename Func>\r
        auto create_task(Func&& func) -> boost::packaged_task<decltype(func())> // noexcept\r
@@ -127,20 +124,20 @@ public:
 \r
        void set_capacity(size_t capacity) // noexcept\r
        {\r
-               execution_queue_[normal_priority].set_capacity(capacity);\r
+               execution_queue_[task_priority::normal_priority].set_capacity(capacity);\r
        }\r
 \r
        void set_priority_class(thread_priority p)\r
        {\r
                begin_invoke([=]\r
                {\r
-                       if(p == high_priority_class)\r
+                       if(p == thread_priority::high_priority_class)\r
                                SetThreadPriority(GetCurrentThread(), HIGH_PRIORITY_CLASS);\r
-                       else if(p == above_normal_priority_class)\r
+                       else if(p == thread_priority::above_normal_priority_class)\r
                                SetThreadPriority(GetCurrentThread(), ABOVE_NORMAL_PRIORITY_CLASS);\r
-                       else if(p == normal_priority_class)\r
+                       else if(p == thread_priority::normal_priority_class)\r
                                SetThreadPriority(GetCurrentThread(), NORMAL_PRIORITY_CLASS);\r
-                       else if(p == below_normal_priority_class)\r
+                       else if(p == thread_priority::below_normal_priority_class)\r
                                SetThreadPriority(GetCurrentThread(), BELOW_NORMAL_PRIORITY_CLASS);\r
                });\r
        }\r
@@ -148,14 +145,14 @@ public:
        void clear()\r
        {               \r
                std::function<void()> func;\r
-               while(execution_queue_[normal_priority].try_pop(func));\r
-               while(execution_queue_[high_priority].try_pop(func));\r
+               while(execution_queue_[task_priority::normal_priority].try_pop(func));\r
+               while(execution_queue_[task_priority::high_priority].try_pop(func));\r
        }\r
                                \r
        void stop() // noexcept\r
        {\r
                is_running_ = false;    \r
-               execution_queue_[normal_priority].try_push([]{}); // Wake the execution thread.\r
+               execution_queue_[task_priority::normal_priority].try_push([]{}); // Wake the execution thread.\r
        }\r
 \r
        void wait() // noexcept\r
@@ -165,22 +162,24 @@ public:
 \r
        void join()\r
        {\r
-               if(boost::this_thread::get_id() != thread_.get_id())\r
-                       thread_.join();\r
+               if(boost::this_thread::get_id() == thread_.get_id())\r
+                       BOOST_THROW_EXCEPTION(invalid_operation());\r
+\r
+               thread_.join();\r
        }\r
                                \r
        template<typename Func>\r
-       auto begin_invoke(Func&& func, task_priority priority = normal_priority) -> boost::unique_future<decltype(func())> // noexcept\r
+       auto begin_invoke(Func&& func, task_priority priority = task_priority::normal_priority) -> boost::unique_future<decltype(func())> // noexcept\r
        {       \r
                if(!is_running_)\r
                        BOOST_THROW_EXCEPTION(invalid_operation() << msg_info("executor not running."));\r
 \r
                // Create a move on copy adaptor to avoid copying the functor into the queue, tbb::concurrent_queue does not support move semantics.\r
-               auto task_adaptor = make_move_on_copy(create_task(func));\r
+               auto task_adaptor = detail::make_move_on_copy(create_task(func));\r
 \r
                auto future = task_adaptor.value.get_future();\r
 \r
-               execution_queue_[priority].push([=]\r
+               execution_queue_[priority.value()].push([=]\r
                {\r
                        try\r
                        {\r
@@ -195,14 +194,14 @@ public:
                        }\r
                });\r
 \r
-               if(priority != normal_priority)\r
-                       execution_queue_[normal_priority].push(nullptr);\r
+               if(priority != task_priority::normal_priority)\r
+                       execution_queue_[task_priority::normal_priority].push(nullptr);\r
                                        \r
                return std::move(future);               \r
        }\r
        \r
        template<typename Func>\r
-       auto invoke(Func&& func, task_priority prioriy = normal_priority) -> decltype(func()) // noexcept\r
+       auto invoke(Func&& func, task_priority prioriy = task_priority::normal_priority) -> decltype(func()) // noexcept\r
        {\r
                if(boost::this_thread::get_id() == thread_.get_id())  // Avoids potential deadlock.\r
                        return func();\r
@@ -216,10 +215,10 @@ public:
                        BOOST_THROW_EXCEPTION(invalid_operation() << msg_info("Executor can only yield inside of thread context."));\r
 \r
                std::function<void()> func;\r
-               execution_queue_[normal_priority].pop(func);    \r
+               execution_queue_[task_priority::normal_priority].pop(func);     \r
                \r
                std::function<void()> func2;\r
-               while(execution_queue_[high_priority].try_pop(func2))\r
+               while(execution_queue_[task_priority::high_priority].try_pop(func2))\r
                {\r
                        if(func2)\r
                                func2();\r
@@ -229,13 +228,14 @@ public:
                        func();\r
        }\r
                \r
-       function_queue::size_type capacity() const /*noexcept*/ { return execution_queue_[normal_priority].capacity();  }\r
-       function_queue::size_type size() const /*noexcept*/ { return execution_queue_[normal_priority].size();  }\r
-       bool empty() const /*noexcept*/ { return execution_queue_[normal_priority].empty();     }\r
+       function_queue::size_type size() const /*noexcept*/\r
+       {\r
+               return execution_queue_[task_priority::normal_priority].size() + execution_queue_[task_priority::high_priority].size(); \r
+       }\r
+               \r
        bool is_running() const /*noexcept*/ { return is_running_; }    \r
                \r
-private:\r
-       \r
+private:       \r
 \r
        void run() // noexcept\r
        {\r