executor.h

Go to the documentation of this file.

//
// Copyright 2025 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_EXEC_VDF_EXECUTOR_H
#define PXR_EXEC_VDF_EXECUTOR_H
 
 
#include "pxr/pxr.h"
 
#include "pxr/base/tf/mallocTag.h"
#include "pxr/base/trace/trace.h"
#include "pxr/exec/vdf/dataManagerBasedExecutor.h"
#include "pxr/exec/vdf/executorFactory.h"
#include "pxr/exec/vdf/executorInterface.h"
#include "pxr/exec/vdf/request.h"
#include "pxr/exec/vdf/speculationExecutor.h"
#include "pxr/exec/vdf/subExecutor.h"
 
PXR_NAMESPACE_OPEN_SCOPE
 
class VdfExecutorErrorLogger;
class VdfSchedule;
 
template <
    template <typename> class EngineType,
    typename DataManagerType>
class VdfExecutor : 
    public VdfDataManagerBasedExecutor<DataManagerType, VdfExecutorInterface>
{
    // Base class type.
    typedef
        VdfDataManagerBasedExecutor<DataManagerType, VdfExecutorInterface>
        Base;
 
    // The speculation executor engine alias declaration, to be bound as a
    // template template parameter.
    template <typename T>
    using SpeculationEngineType =
        typename EngineType<T>::SpeculationExecutorEngine;
 
    // Executor factory.
    typedef 
        VdfExecutorFactory<
            VdfSubExecutor<EngineType, DataManagerType>,
            VdfSpeculationExecutor<SpeculationEngineType, DataManagerType>>
        _Factory;
 
public:
 
    VdfExecutor() : _engine(*this, &this->_dataManager) {}
 
    virtual ~VdfExecutor() {}
 
    virtual const VdfExecutorFactoryBase &GetFactory() const override final {
        return _factory;
    }
 
private:
 
    // Run this executor with the given \p schedule and \p request.
    //
    virtual void _Run(
        const VdfSchedule &schedule,
        const VdfRequest &computeRequest,
        VdfExecutorErrorLogger *errorLogger) override;
 
    // Causes the DataManager to clear its temporary execution buffers.
    //
    virtual void _ClearData() override;
 
private:
 
    // The factory shared amongst executors of this type.
    //
    static const _Factory _factory;
 
    // This is the engine that will do most of our hard work for us.
    //
    EngineType<DataManagerType> _engine;
};
 
 
template <template <typename> class EngineType, typename DataManagerType>
const typename VdfExecutor<EngineType, DataManagerType>::_Factory
    VdfExecutor<EngineType, DataManagerType>::_factory;
 
template <template <typename> class EngineType, typename DataManagerType>
void
VdfExecutor<EngineType, DataManagerType>::_Run(
    const VdfSchedule &schedule,
    const VdfRequest &computeRequest,
    VdfExecutorErrorLogger *errorLogger)
{    
    // If we have an empty request, bail.
    if (computeRequest.IsEmpty()) {
        return;
    }
 
    TRACE_FUNCTION();
    TfAutoMallocTag2 tag("Ef", "VdfExecutor::Run");
 
    _engine.RunSchedule(schedule, computeRequest, errorLogger);
}
 
template <template <typename> class EngineType, typename DataManagerType>
void
VdfExecutor<EngineType, DataManagerType>::_ClearData()
{
    Base::_dataManager.Clear();
}
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif