Add analysisMode to AmplitudeView

AudioKit · Jan 16, 2024 · 7e8d048 · 7e8d048
1 parent a5b6c27
commit 7e8d048
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/Sources/AudioKitUI/Visualizations/AmplitudeView.swift b/Sources/AudioKitUI/Visualizations/AmplitudeView.swift
@@ -8,6 +8,7 @@ class AmplitudeModel: ObservableObject {
     var nodeTap: AmplitudeTap!
     var node: Node?
     var stereoMode: StereoMode = .center
+    var analysisMode: AnalysisMode = .peak
     @Environment(\.isPreview) var isPreview
 
     init() {
@@ -19,7 +20,7 @@ class AmplitudeModel: ObservableObject {
     func updateNode(_ node: Node) {
         if node !== self.node {
             self.node = node
-            nodeTap = AmplitudeTap(node, stereoMode: stereoMode, analysisMode: .peak, callbackQueue: .main) { amp in
+            nodeTap = AmplitudeTap(node, stereoMode: stereoMode, analysisMode: analysisMode, callbackQueue: .main) { amp in
                 self.pushData(amp)
             }
             nodeTap.start()
@@ -43,26 +44,30 @@ public struct AmplitudeView: View {
     @StateObject var amplitudeModel = AmplitudeModel()
     let node: Node
     let stereoMode: StereoMode
+    let analysisMode: AnalysisMode
     let numberOfSegments: Int
     let fillType: FillType
 
-    public init(_ node: Node, stereoMode: StereoMode = .center, numberOfSegments: Int = 20) {
+    public init(_ node: Node, stereoMode: StereoMode = .center, analysisMode: AnalysisMode = .peak, numberOfSegments: Int = 20) {
         self.node = node
         self.stereoMode = stereoMode
+        self.analysisMode = analysisMode
         self.fillType = .gradient(gradient: Gradient(colors: [.red, .yellow, .green]))
         self.numberOfSegments = numberOfSegments
     }
 
-    public init(_ node: Node, color: Color, stereoMode: StereoMode = .center, numberOfSegments: Int = 20) {
+    public init(_ node: Node, color: Color, stereoMode: StereoMode = .center, analysisMode: AnalysisMode = .peak, numberOfSegments: Int = 20) {
         self.node = node
         self.stereoMode = stereoMode
+        self.analysisMode = analysisMode
         self.fillType = .solid(color: color)
         self.numberOfSegments = numberOfSegments
     }
 
-    public init(_ node: Node, colors: Gradient, stereoMode: StereoMode = .center, numberOfSegments: Int = 20) {
+    public init(_ node: Node, colors: Gradient, stereoMode: StereoMode = .center, analysisMode: AnalysisMode = .peak, numberOfSegments: Int = 20) {
         self.node = node
         self.stereoMode = stereoMode
+        self.analysisMode = analysisMode
         self.fillType = .gradient(gradient: colors)
         self.numberOfSegments = numberOfSegments
     }