Expose Repeat Option

Lakr233 · Aug 13, 2024 · 03cb783 · 03cb783
1 parent 7df6a12
commit 03cb783
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/Sources/ColorfulX/AnimatedMulticolorGradientView+SwiftUI.swift b/Sources/ColorfulX/AnimatedMulticolorGradientView+SwiftUI.swift
@@ -16,14 +16,16 @@ public struct AnimatedMulticolorGradientViewRepresentable {
     @Binding var transitionSpeed: Double
 
     let frameLimit: Int
+    let repeatToFillColorSlots: Bool
 
     public init(
         color: Binding<[ColorVector]>,
         speed: Binding<Double> = .constant(1),
         bias: Binding<Double> = .constant(0.01),
         noise: Binding<Double> = .constant(0),
         transitionSpeed: Binding<Double> = .constant(3.25),
-        frameLimit: Int = 0
+        frameLimit: Int = 0,
+        repeatToFillColorSlots: Bool = true
     ) {
         _color = color
         _speed = speed
@@ -32,10 +34,15 @@ public struct AnimatedMulticolorGradientViewRepresentable {
         _transitionSpeed = transitionSpeed
 
         self.frameLimit = frameLimit
+        self.repeatToFillColorSlots = repeatToFillColorSlots
     }
 
     public func updatePropertyToView(_ view: AnimatedMulticolorGradientView) {
-        view.setColors(color, interpolationEnabled: transitionSpeed > 0)
+        view.setColors(
+            color,
+            interpolationEnabled: transitionSpeed > 0,
+            repeatToFillColorSlots: repeatToFillColorSlots
+        )
         view.speed = speed
         view.bias = bias
         view.noise = noise

diff --git a/Sources/ColorfulX/AnimatedMulticolorGradientView.swift b/Sources/ColorfulX/AnimatedMulticolorGradientView.swift
@@ -51,12 +51,12 @@ open class AnimatedMulticolorGradientView: MulticolorGradientView {
         initializeRenderParameters()
     }
 
-    public func setColors(_ colors: [ColorVector], interpolationEnabled: Bool = true, fillColorsIfNeeded: Bool = true) {
+    public func setColors(_ colors: [ColorVector], interpolationEnabled: Bool = true, repeatToFillColorSlots: Bool = true) {
         var colors = colors
         if colors.isEmpty { colors.append(.init(v: .zero, space: .rgb)) }
         colors = colors.map { $0.color(in: .lab) }
 
-        let endingIndex = fillColorsIfNeeded ? Uniforms.COLOR_SLOT : min(colors.count, Uniforms.COLOR_SLOT)
+        let endingIndex = repeatToFillColorSlots ? Uniforms.COLOR_SLOT : min(colors.count, Uniforms.COLOR_SLOT)
         guard endingIndex > 0 else { return }
         for idx in 0 ..< endingIndex {
             var read = colorElements[idx]