add mute/unmute microphone

Author: pedroSG94

.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata

@@ -0,0 +1,27 @@
+// swift-tools-version: 5.10
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+import PackageDescription
+
+let package = Package(
+    name: "RootEncoder",
+    platforms: [.iOS(.v14)],
+    products: [
+        // Products define the executables and libraries a package produces, making them visible to other packages.
+        .library(
+            name: "RootEncoder",
+            targets: ["RootEncoder"]
+        ),
+    ],
+    targets: [
+        // Targets are the basic building blocks of a package, defining a module or a test suite.
+        // Targets can depend on other targets in this package and products from dependencies.
+        .target(
+            name: "RootEncoder"
+        ),
+        .testTarget(
+            name: "RootEncoderTests",
+            dependencies: ["RootEncoder"]
+        ),
+    ]
+)

RootEncoder/Package.swift

@@ -15,6 +15,7 @@ public class MicrophoneManager {
     private let audioEngine = AVAudioEngine()
     private var inputNode: AVAudioInputNode?
     private var inputFormat: AVAudioFormat?
+    private var muted = false
 
     private var callback: GetMicrophoneData?
 
@@ -30,7 +31,7 @@ public class MicrophoneManager {
         }
         inputNode?.installTap(onBus: 0, bufferSize: 2048, format: inputFormat) { buffer, time in
             self.thread.async {
-                self.callback?.getPcmData(buffer: buffer, time: time)
+                self.callback?.getPcmData(buffer: buffer.mute(enabled: !self.muted), time: time)
             }
         }
         audioEngine.prepare()
@@ -52,4 +53,16 @@ public class MicrophoneManager {
         audioEngine.stop()
         audioEngine.inputNode.removeTap(onBus: 0)
     }
+    
+    public func isMuted() -> Bool {
+        return muted
+    }
+    
+    public func mute() {
+        muted = true
+    }
+    
+    public func unmute() {
+        muted = false
+    }
 }

RootEncoder/Sources/RootEncoder/encoder/input/audio/MicrophoneManager.swift

@@ -47,6 +47,40 @@ public extension AVAudioPCMBuffer {
         }
         return audioByteArray
     }
+    
+    func mute(enabled: Bool) -> AVAudioPCMBuffer {
+        if enabled {
+            return self
+        }
+        let numSamples = Int(frameLength)
+        if format.isInterleaved {
+            let channelCount = Int(format.channelCount)
+            switch format.commonFormat {
+            case .pcmFormatInt16:
+                int16ChannelData?[0].update(repeating: 0, count: numSamples * channelCount)
+            case .pcmFormatInt32:
+                int32ChannelData?[0].update(repeating: 0, count: numSamples * channelCount)
+            case .pcmFormatFloat32:
+                floatChannelData?[0].update(repeating: 0, count: numSamples * channelCount)
+            default:
+                break
+            }
+        } else {
+            for i in 0..<Int(format.channelCount) {
+                switch format.commonFormat {
+                case .pcmFormatInt16:
+                    int16ChannelData?[i].update(repeating: 0, count: numSamples)
+                case .pcmFormatInt32:
+                    int32ChannelData?[i].update(repeating: 0, count: numSamples)
+                case .pcmFormatFloat32:
+                    floatChannelData?[i].update(repeating: 0, count: numSamples)
+                default:
+                    break
+                }
+            }
+        }
+        return self
+    }
 }
 
 extension AVAudioTime {

RootEncoder/Sources/RootEncoder/encoder/utils/EncoderUtils.swift

@@ -141,6 +141,18 @@ public class CameraBase: GetMicrophoneData, GetCameraData, GetAacData, GetH264Da
     public func switchCamera() {
         cameraManager.switchCamera()
     }
+    
+    public func isMuted() -> Bool {
+        return microphone.isMuted()
+    }
+    
+    public func mute() {
+        microphone.mute()
+    }
+    
+    public func unmute() {
+        microphone.unmute()
+    }
 
     public func startPreview(resolution: CameraHelper.Resolution, facing: CameraHelper.Facing = .BACK, rotation: Int) {
         if (!isOnPreview()) {

RootEncoder/Sources/RootEncoder/library/base/CameraBase.swift

@@ -116,6 +116,18 @@ public class DisplayBase: GetMicrophoneData, GetCameraData, GetAacData, GetH264D
         streaming
     }
 
+    public func isMuted() -> Bool {
+        return microphone.isMuted()
+    }
+    
+    public func mute() {
+        microphone.mute()
+    }
+    
+    public func unmute() {
+        microphone.unmute()
+    }
+    
     public func setVideoCodec(codec: VideoCodec) {
         setVideoCodecImp(codec: codec)
         recordController.setVideoCodec(codec: codec)

RootEncoder/Sources/RootEncoder/library/base/DisplayBase.swift

@@ -74,7 +74,7 @@ struct RtmpSwiftUIView: View, ConnectChecker {
     }
 
 
-    @State private var endpoint = "rtmp://192.168.0.160:1935/live/pedro"
+    @State private var endpoint = "rtmp://192.168.0.176:1935/live/pedro"
     @State private var bStreamText = "Start stream"
     @State private var bRecordText = "Start record"
     @State private var isShowingToast = false