dispatcher.h

Go to the documentation of this file.

//
// Copyright 2016 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_BASE_WORK_DISPATCHER_H
#define PXR_BASE_WORK_DISPATCHER_H
 
 
#include "pxr/pxr.h"
#include "pxr/base/work/api.h"
#include "pxr/base/work/impl.h"
#include "pxr/base/work/threadLimits.h"
 
#include "pxr/base/tf/diagnosticTrap.h"
#include "pxr/base/tf/diagnosticTransport.h"
#include "pxr/base/tf/mallocTag.h"
 
#include <functional>
#include <type_traits>
#include <utility>
 
PXR_NAMESPACE_OPEN_SCOPE
 
// The Work_Dispatcher interface, specialized with a dispatcher impl template
// argument.
// 
// Clients expected to use the WorkDispatcher type instead.
template <class Impl>
class Work_Dispatcher
{
protected:
    // Prevent construction of the work dispatcher base class.
    WORK_API Work_Dispatcher();
 
public:
    WORK_API ~Work_Dispatcher() noexcept;
 
    Work_Dispatcher(Work_Dispatcher const &) = delete;
    Work_Dispatcher &operator=(Work_Dispatcher const &) = delete;
 
#ifdef doxygen
 
    template <class Callable, class A1, class A2, ... class AN>
    void Run(Callable &&c, A1 &&a1, A2 &&a2, ... AN &&aN);
 
#else // doxygen
 
    template <class Callable>
    inline void Run(Callable &&c) {
        if (TfMallocTag::IsInitialized()) {
            _dispatcher.Run(
                _MallocTagsInvokerTask<
                typename std::remove_reference<Callable>::type>(
                    std::forward<Callable>(c), &_diagnostics));
        }
        else {
            _dispatcher.Run(
                _InvokerTask<typename std::remove_reference<Callable>::type>(
                    std::forward<Callable>(c), &_diagnostics));
        }
    }
 
    template <class Callable, class A0, class ... Args>
    inline void Run(Callable &&c, A0 &&a0, Args&&... args) {
        Run(std::bind(std::forward<Callable>(c),
                      std::forward<A0>(a0),
                      std::forward<Args>(args)...));
    }
    
#endif // doxygen
 
    WORK_API void Wait();
 
    WORK_API void Cancel();
 
    WORK_API bool IsCancelled() const;
 
private:
    typedef tbb::concurrent_vector<TfDiagnosticTransport> _DiagnosticTransports;
 
    // Function invoker helper that wraps the invocation with a DiagnosticTrap
    // so we can transmit diagnostics that occur back to the thread that calls
    // Wait() for tasks to complete.
    template <class Fn>
    struct _InvokerTask {
        explicit _InvokerTask(Fn &&fn, _DiagnosticTransports *diag) 
            : _fn(std::move(fn))
            , _diagnostics(diag) {}
 
        explicit _InvokerTask(Fn const &fn, _DiagnosticTransports *diag)
            : _fn(fn)
            , _diagnostics(diag) {}
 
        // Ensure only moves happen, no copies.
        _InvokerTask(_InvokerTask &&other) = default;
        _InvokerTask(const _InvokerTask &other) = delete;
        _InvokerTask &operator=(const _InvokerTask &other) = delete;
 
        void operator()() const {
            TfDiagnosticTrap trap;
            _fn();
            if (!trap.IsClean()) {
                Work_Dispatcher::_TransportDiagnostics(&trap, _diagnostics);
            }
        }
    private:
        Fn _fn;
        _DiagnosticTransports *_diagnostics;
    };
 
    // Function invoker helper that wraps the invocation with a TfDiagnosticTrap
    // so we can transmit diagnostics that occur back to the thread that Wait()
    // s for tasks to complete.  This version also duplicates the caller's
    // malloc tag stack to the callee's thread.
    template <class Fn>
    struct _MallocTagsInvokerTask {
        explicit
        _MallocTagsInvokerTask(Fn &&fn, _DiagnosticTransports *diag) 
            : _fn(std::move(fn))
            , _diagnostics(diag)
            , _mallocTagStack(TfMallocTag::GetCurrentStackState())
            {}
 
        explicit
        _MallocTagsInvokerTask(Fn const &fn, _DiagnosticTransports *diag) 
            : _fn(fn)
            , _diagnostics(diag)
            , _mallocTagStack(TfMallocTag::GetCurrentStackState()) {}
 
        // Ensure only moves happen, no copies.
        _MallocTagsInvokerTask(_MallocTagsInvokerTask &&other) = default;
        _MallocTagsInvokerTask(const _MallocTagsInvokerTask &other) = delete;
        _MallocTagsInvokerTask &
        operator=(const _MallocTagsInvokerTask &other) = delete;
 
        void operator()() const {
            TfDiagnosticTrap trap;
            TfMallocTag::StackOverride ovr(_mallocTagStack);
            _fn();
            if (!trap.IsClean()) {
                Work_Dispatcher::_TransportDiagnostics(&trap, _diagnostics);
            }
        }
    private:
        Fn _fn;
        _DiagnosticTransports *_diagnostics;
        TfMallocTag::StackState _mallocTagStack;
    };
 
    // Helper function that removes diagnostics from \p trap and stores them in
    // a new entry in \p diagnostics.
    WORK_API static void
    _TransportDiagnostics(TfDiagnosticTrap *trap,
                          _DiagnosticTransports *diagnostics);
 
    // WorkDispatcher implementation
    Impl _dispatcher;
    std::atomic<bool> _isCancelled;
 
    // The diagnostic transports we use to transmit those issued in other
    // threads back to this thread.
    _DiagnosticTransports _diagnostics;
 
    // Concurrent calls to Wait() have to serialize certain cleanup operations.
    std::atomic_flag _waitCleanupFlag;
};
 
class WorkDispatcher 
    : public Work_Dispatcher<PXR_WORK_IMPL_NS::WorkImpl_Dispatcher>
{};
 
// Wrapper class for non-const tasks.
template <class Fn>
struct Work_DeprecatedMutableTask {
    explicit Work_DeprecatedMutableTask(Fn &&fn) 
        : _fn(std::move(fn)) {}
 
    explicit Work_DeprecatedMutableTask(Fn const &fn) 
        : _fn(fn) {}
 
    // Ensure only moves happen, no copies.
    Work_DeprecatedMutableTask
        (Work_DeprecatedMutableTask &&other) = default;
    Work_DeprecatedMutableTask
        (const Work_DeprecatedMutableTask &other) = delete;
    Work_DeprecatedMutableTask
        &operator= (const Work_DeprecatedMutableTask &other) = delete;
 
    void operator()() const {
        _fn();
    }
private:
    mutable Fn _fn;
};
 
// Wrapper function to convert non-const tasks to a Work_DeprecatedMutableTask. 
// When adding new tasks refrain from using this wrapper, instead ensure the 
// call operator of the task is const such that it is compatible with oneTBB.
template <typename Fn>
Work_DeprecatedMutableTask<typename std::remove_reference_t<Fn>> 
WorkMakeDeprecatedMutableTask(Fn &&fn) {
    return Work_DeprecatedMutableTask<typename std::remove_reference_t<Fn>>
            (std::forward<Fn>(fn));
}
 
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif // PXR_BASE_WORK_DISPATCHER_H