system.h

//
// Copyright 2025 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_EXEC_EXEC_SYSTEM_H
#define PXR_EXEC_EXEC_SYSTEM_H

#include "pxr/pxr.h"

#include "pxr/exec/exec/api.h"
#include "pxr/exec/exec/valueOverride.h"

#include "pxr/exec/esf/stage.h"

#include <memory>
#include <utility>
#include <vector>

PXR_NAMESPACE_OPEN_SCOPE

class EfLeafNode;
class EfTime;
class Exec_Program;
class Exec_RequestImpl;
class Exec_RequestTracker;
class Exec_Runtime;
class ExecValueKey;
class SdfPath;
class TfToken;
template <typename> class TfFunctionRef;
template <typename> class TfSpan;
class VdfExecutorInterface;
class VdfMaskedOutput;
class VdfRequest;
class VdfSchedule;

class ExecSystem
{
public:
    class Diagnostics;

protected:
    EXEC_API
    explicit ExecSystem(EsfStage &&stage);

    ExecSystem(const ExecSystem &) = delete;
    ExecSystem& operator=(const ExecSystem &) = delete;

    EXEC_API
    ~ExecSystem();

    EXEC_API
    void _ChangeTime(const EfTime &time);

    EXEC_API
    void _Compute(
        const VdfSchedule &schedule,
        const VdfRequest &computeRequest);

    EXEC_API
    std::unique_ptr<VdfExecutorInterface> _ComputeWithOverrides(
        const VdfSchedule &schedule,
        const VdfRequest &computeRequest,
        ExecValueOverrideVector &&valueOverrides);

    EXEC_API
    void _ParallelForEachRequest(
        TfFunctionRef<void(Exec_RequestImpl&)> f) const;

    class _ChangeProcessor;

private:
    // Requires access to _Compute, _Compile, and _HasPendingRecompilation.
    friend class Exec_RequestImpl;

    std::pair<std::vector<VdfMaskedOutput>, std::vector<const EfLeafNode*>>
    _Compile(TfSpan<const ExecValueKey> valueKeys);

    // Returns true if the program has inputs requiring recompilation.
    EXEC_API
    bool _HasPendingRecompilation() const;

    // Discards all internal state, and constructs new internal data structures
    // leaving the system in the same state as if it was newly constructed.
    // 
    EXEC_API
    void _InvalidateAll();

    // Notifies the system of invalidation due to uncompilation.
    EXEC_API
    void _InvalidateDisconnectedInputs();

    // Notifies the system of attribute value invalidation.
    EXEC_API
    void _InvalidateAttributeValues(TfSpan<const SdfPath> invalidAttributes);

    // Notifies the system of metadata value invalidation.
    EXEC_API
    void _InvalidateMetadataValues(
        TfSpan<const std::pair<SdfPath, TfToken>> invalidObjects);

    // Notifies the system to invalidate value keys that don't have a compiled
    // leaf node.
    //
    EXEC_API
    void _InvalidateUnknownValues();

private:
    EsfStage _stage;

    std::unique_ptr<Exec_Program> _program;
    std::unique_ptr<Exec_Runtime> _runtime;

    std::unique_ptr<Exec_RequestTracker> _requestTracker;
};

PXR_NAMESPACE_CLOSE_SCOPE

#endif