1 files changed, 105 insertions, 0 deletions
diff --git a/experimental/sksg/paint/SkSGGradient.h b/experimental/sksg/paint/SkSGGradient.h
new file mode 100644
index 0000000000..d69cb1495c
--- /dev/null
+++ b/experimental/sksg/paint/SkSGGradient.h
@@ -0,0 +1,105 @@
+/*
+ * Copyright 2018 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#ifndef SkSGGradient_DEFINED
+#define SkSGGradient_DEFINED
+
+#include "SkSGPaintNode.h"
+
+#include "SkColor.h"
+#include "SkPoint.h"
+#include "SkScalar.h"
+#include "SkShader.h"
+
+#include <vector>
+
+namespace sksg {
+
+/**
+ * Gradient base class.
+ */
+class Gradient : public PaintNode {
+public:
+    struct ColorStop {
+        SkScalar fPosition;
+        SkColor  fColor;
+
+        bool operator==(const ColorStop& other) const {
+            return fPosition == other.fPosition && fColor == other.fColor;
+        }
+    };
+
+    SG_ATTRIBUTE(ColorStops, std::vector<ColorStop>, fColorStops)
+    SG_ATTRIBUTE(TileMode  , SkShader::TileMode    , fTileMode  )
+
+protected:
+    void onApplyToPaint(SkPaint*) const final;
+
+    virtual sk_sp<SkShader> onMakeShader(const std::vector<SkColor>& colors,
+                                         const std::vector<SkScalar>& positions) const = 0;
+
+protected:
+    Gradient() = default;
+
+private:
+    std::vector<ColorStop> fColorStops;
+    SkShader::TileMode     fTileMode = SkShader::kClamp_TileMode;
+
+    using INHERITED = PaintNode;
+};
+
+class LinearGradient final : public Gradient {
+public:
+    static sk_sp<LinearGradient> Make() {
+        return sk_sp<LinearGradient>(new LinearGradient());
+    }
+
+    SG_ATTRIBUTE(StartPoint, SkPoint, fStartPoint)
+    SG_ATTRIBUTE(EndPoint  , SkPoint, fEndPoint  )
+
+protected:
+    sk_sp<SkShader> onMakeShader(const std::vector<SkColor>& colors,
+                                 const std::vector<SkScalar>& positions) const override;
+
+private:
+    LinearGradient() = default;
+
+    SkPoint fStartPoint = SkPoint::Make(0, 0),
+            fEndPoint   = SkPoint::Make(0, 0);
+
+    using INHERITED = Gradient;
+};
+
+class RadialGradient final : public Gradient {
+public:
+    static sk_sp<RadialGradient> Make() {
+        return sk_sp<RadialGradient>(new RadialGradient());
+    }
+
+    SG_ATTRIBUTE(StartCenter, SkPoint , fStartCenter)
+    SG_ATTRIBUTE(EndCenter  , SkPoint , fEndCenter  )
+    SG_ATTRIBUTE(StartRadius, SkScalar, fStartRadius)
+    SG_ATTRIBUTE(EndRadius  , SkScalar, fEndRadius  )
+
+protected:
+    sk_sp<SkShader> onMakeShader(const std::vector<SkColor>& colors,
+                                 const std::vector<SkScalar>& positions) const override;
+
+private:
+    RadialGradient() = default;
+
+    SkPoint  fStartCenter = SkPoint::Make(0, 0),
+             fEndCenter   = SkPoint::Make(0, 0);
+    SkScalar fStartRadius = 0,
+             fEndRadius   = 0;
+
+    using INHERITED = Gradient;
+};
+
+} // namespace sksg
+
+#endif // SkSGGradient_DEFINED