scheduler.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_SCHEDULER_H
#define PXR_EXEC_VDF_SCHEDULER_H
 
 
#include "pxr/pxr.h"
 
#include "pxr/exec/vdf/api.h"
#include "pxr/exec/vdf/poolChainIndex.h"
 
PXR_NAMESPACE_OPEN_SCOPE
 
class VdfMaskedOutput;
using VdfMaskedOutputVector = std::vector<VdfMaskedOutput>;
class VdfNetwork;
class VdfNode;
class VdfRequest;
class VdfSchedule;
class VdfScheduleNode;
class VdfScheduleOutput;
 
class VdfScheduler 
{
public:
    // Vector of priorities for pool outputs.
    typedef std::vector<
        std::pair<VdfPoolChainIndex, const VdfOutput *>
    > PoolPriorityVector;
 
    typedef
        TfHashMap<const VdfNode *, VdfMaskedOutputVector, TfHash> NodeToRequestMap;
 
    VDF_API
    static void Schedule(const VdfRequest &request, VdfSchedule *schedule,
        bool topologicallySort);
 
    VDF_API
    static bool UpdateAffectsMaskForOutput(
        VdfSchedule *schedule,
        const VdfOutput &output);
 
protected:
    VDF_API
    static void _SetScheduleValid(
        VdfSchedule *schedule, 
        const VdfNetwork *network);
 
    VDF_API
    static void _InitializeRequestMasks(
        const VdfRequest &request,
        VdfSchedule *schedule,
        PoolPriorityVector *poolOutputs);
 
    VDF_API
    static void _MarkSmallSchedule(VdfSchedule *schedule);
 
    VDF_API
    static void _ScheduleBufferPasses(
        const VdfRequest &request,
        VdfSchedule *schedule);
 
    VDF_API
    static void _ScheduleForPassThroughs(
        const VdfRequest &request,
        VdfSchedule *schedule,
        const PoolPriorityVector &sortedPoolOutputs);
 
    VDF_API
    static void _GenerateTasks(
        VdfSchedule *schedule,
        const PoolPriorityVector &sortedPoolOutputs);
 
    VDF_API
    static void _ScheduleTaskGraph(
        VdfSchedule *schedule,
        const PoolPriorityVector &sortedPoolOutputs);
 
    VDF_API
    static void _ComputeLockMasks(
        const VdfRequest &request, 
        VdfSchedule *schedule,
        const PoolPriorityVector &sortedPoolOutputs);
 
    VDF_API
    static void _ApplyAffectsMasks(VdfSchedule *schedule);
 
    VDF_API
    static bool _ApplyAffectsMasksForNode(VdfScheduleNode *node);
 
    VDF_API
    static void _UpdateAffectsMaskForInvocation(
        VdfSchedule *schedule,
        VdfScheduleNode *node);
 
    VDF_API
    static bool _UpdateAffectsMasksForNode(
        VdfSchedule *schedule,
        VdfScheduleNode *node);
 
    VDF_API
    static void _UpdateLockMaskForNode(
        VdfSchedule *schedule,
        VdfScheduleNode *node);
 
};
 
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif