WaitingTaskList.h

Go to the documentation of this file.

#ifndef FWCore_Concurrency_WaitingTaskList_h
#define FWCore_Concurrency_WaitingTaskList_h
// -*- C++ -*-
//
// Package:     Concurrency
// Class  :     WaitingTaskList
//
//
// Original Author:  Chris Jones
//         Created:  Thu Feb 21 13:46:31 CST 2013
// $Id$
//
 
// system include files
#include <atomic>
 
// user include files
#include "FWCore/Concurrency/interface/WaitingTask.h"
#include "FWCore/Concurrency/interface/WaitingTaskHolder.h"
#include "FWCore/Utilities/interface/thread_safety_macros.h"
 
// forward declarations
 
namespace edm {
  class WaitingTaskList {
  public:
 
    explicit WaitingTaskList(unsigned int iInitialSize = 2);
    WaitingTaskList(const WaitingTaskList&) = delete;                   // stop default
    const WaitingTaskList& operator=(const WaitingTaskList&) = delete;  // stop default
    ~WaitingTaskList() = default;
 
    // ---------- member functions ---------------------------
 
    void presetTaskAsFailed(std::exception_ptr iExcept);
 
 
    void add(tbb::task_group*, WaitingTask*);
 
 
    void add(WaitingTaskHolder);
 
 
    void doneWaiting(std::exception_ptr iPtr);
 
 
    void reset();
 
  private:
    void announce();
 
    struct WaitNode {
      WaitingTask* m_task;
      tbb::task_group* m_group;
      std::atomic<WaitNode*> m_next;
      bool m_fromCache;
 
      void setNextNode(WaitNode* iNext) { m_next = iNext; }
 
      WaitNode* nextNode() const { return m_next; }
    };
 
    WaitNode* createNode(tbb::task_group* iGroup, WaitingTask* iTask);
 
    // ---------- member data --------------------------------
    std::atomic<WaitNode*> m_head;
    std::unique_ptr<WaitNode[]> m_nodeCache;
    CMS_THREAD_GUARD(m_waiting) std::exception_ptr m_exceptionPtr;
    unsigned int m_nodeCacheSize;
    std::atomic<unsigned int> m_lastAssignedCacheIndex;
    std::atomic<bool> m_waiting;
  };
}  // namespace edm
 
#endif