dynamic_task_graph.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org. It
// supplements something to oneTBB which has been removed and which I need:
// dynamic tasking. Therefore I add it to the tbb namespace rather than
// Peano's tarch.
#pragma once
 
 
 
#include <functional>
#include <set>
#include <tbb/task_arena.h>
#include <oneapi/tbb/task_group.h>
#include <oneapi/tbb/spin_mutex.h>
#include <oneapi/tbb/spin_rw_mutex.h>
#include <oneapi/tbb/concurrent_unordered_set.h>
 
 
namespace tbb {
  class dynamic_task_graph_node;
  class dynamic_task_graph_spawned_node;
  class dynamic_task_graph;
  class dynamic_task_graph_node_repository;
}
 
 
namespace tbb {
  class dynamic_task_graph {
      public:
        friend class dynamic_task_graph_spawned_node;
 
        dynamic_task_graph(
          ::oneapi::tbb::task_group&      arena
        );
 
        void put(dynamic_task_graph_node& node );
 
        void wait( dynamic_task_graph_node node );
 
        void wait( const std::set<dynamic_task_graph_node>& node );
 
        void wait_for_all();
 
        std::size_t size() const;
      private:
        ::oneapi::tbb::task_group&      _task_group;
 
        using SpawnedTaskMutex = ::oneapi::tbb::spin_rw_mutex;
        SpawnedTaskMutex       _spawned_task_mutex;
 
        using SpawnedTasksContainer = std::set< std::shared_ptr<dynamic_task_graph_spawned_node> >;
 
        SpawnedTasksContainer _spawned_tasks;
 
        void clean_up_completed_tasks();
  };
}