include/rive/shapes/paint/linear_gradient.hpp

#ifndef _RIVE_LINEAR_GRADIENT_HPP_
#define _RIVE_LINEAR_GRADIENT_HPP_
#include "rive/generated/shapes/paint/linear_gradient_base.hpp"
#include "rive/math/vec2d.hpp"
#include "rive/shapes/paint/color.hpp"
#include "rive/shapes/paint/shape_paint_mutator.hpp"
#include <vector>

namespace rive
{
class Node;
class GradientStop;
class PointDeformer;

class LinearGradient : public LinearGradientBase, public ShapePaintMutator
{
public:
    StatusCode onAddedDirty(CoreContext* context) override;
#ifdef WITH_RIVE_EDITOR
    // Editor: same parent-attachment pattern as `SolidColor`. Body
    // in `component_parent_editor.cpp`.
    void editorParentChanged(ContainerComponent* from,
                             ContainerComponent* to) override;
#endif
    void addStop(GradientStop* stop);
#ifdef WITH_RIVE_EDITOR
    // Idempotent variant for `EditorFile::finalizeBatch` retry. The
    // runtime `addStop` is called from `GradientStop::onAddedDirty`
    // exactly once per stop; coop's intra-batch out-of-order means
    // the stop's `parent()` may be null when its onAddedDirty fires
    // (so it skips addStop). finalizeBatch walks every stop later
    // and calls this — which checks for an existing entry first so
    // a `.riv`-loaded file's importer-added stops don't get
    // double-listed when coop overlays edits on top.
    void addStopForEditor(GradientStop* stop);
    void removeStopForEditor(GradientStop* stop);
#endif
    void update(ComponentDirt value) override;
    void markGradientDirty();
    void markStopsDirty();
    void applyTo(RenderPaint* renderPaint, float opacityModifier) override;

protected:
    void buildDependencies() override;
    void startXChanged() override;
    void startYChanged() override;
    void endXChanged() override;
    void endYChanged() override;
    void opacityChanged() override;
    void renderOpacityChanged() override;

    virtual void makeGradient(RenderPaint* renderPaint,
                              Vec2D start,
                              Vec2D end,
                              const ColorInt[],
                              const float[],
                              size_t count) const;

private:
    // Set m_deformer from the shape paint container
    void updateDeformer();

    std::vector<GradientStop*> m_stops;
    Node* m_shapePaintContainer = nullptr;
    PointDeformer* m_deformer = nullptr;
    std::vector<ColorInt> m_colorStorage;
};
} // namespace rive

#endif