@@ -19,22 +19,34 @@ class HornSoundDetector: NSObject, ObservableObject {
1919 private var inputNode : AVAudioInputNode !
2020 private var soundClassifier : HornSoundClassifier_V11 ?
2121 private var streamAnalyzer : SNAudioStreamAnalyzer ?
22+ //fix 시작
23+ // @Published var isRecording = false
24+ // @Published var classificationResult = "녹음 시작 전"
25+ // @Published var detectedHornSound = false
26+ // @Published var topClassification: SNClassification? // 가장 높은 분류 저장
27+ // @Published var mlConfidences: [Double] = Array(repeating: 0.0, count: 4) // 각 채널의 신뢰도 배열
28+ // private var cancellables = Set<AnyCancellable>()
2229
30+ // override init() {
31+ //fix 끝
32+ private var appRootManager : AppRootManager // appRootManager 속성 추가
33+
2334 @Published var isRecording = false
2435 @Published var classificationResult = " 녹음 시작 전 "
25- @Published var detectedHornSound = false
26- @Published var topClassification : SNClassification ?
27- @Published var mlConfidences : [ Double ] = Array ( repeating: 0.0 , count: 4 ) // 각 채널의 신뢰도 배열
2836 private var cancellables = Set < AnyCancellable > ( )
2937
30- override init ( ) {
38+ init ( appRootManager: AppRootManager ) {
39+ self . appRootManager = appRootManager
40+
3141 super. init ( )
3242 setupAudioSession ( )
3343 setupAudioEngine ( )
3444 setupSoundClassifier ( )
35- checkNotificationPermission ( )
45+ //fix
46+ // checkNotificationPermission()
3647
3748 // 앱이 백그라운드로 전환될 때 녹음을 중지하도록 옵저버 설정
49+ //fix
3850 NotificationCenter . default. addObserver ( self , selector: #selector( stopRecording) , name: UIApplication . didEnterBackgroundNotification, object: nil )
3951 }
4052
@@ -47,11 +59,22 @@ class HornSoundDetector: NSObject, ObservableObject {
4759 print ( " 오디오 세션 설정 실패: \( error) " )
4860 }
4961 }
62+ //fix
63+ // private func setupAudioEngine() {
64+ // audioEngine = AVAudioEngine()
65+ // inputNode = audioEngine.inputNode
66+ // }
67+
68+ //fix
69+
5070 private func setupAudioEngine( ) {
5171 audioEngine = AVAudioEngine ( )
5272 inputNode = audioEngine. inputNode
73+ let format = inputNode. outputFormat ( forBus: 0 )
74+ let monoFormat = AVAudioFormat ( commonFormat: format. commonFormat, sampleRate: format. sampleRate, channels: 1 , interleaved: format. isInterleaved)
75+ streamAnalyzer = SNAudioStreamAnalyzer ( format: monoFormat ?? format)
5376 }
54-
77+
5578 private func setupSoundClassifier( ) {
5679 do {
5780 let config = MLModelConfiguration ( )
@@ -62,59 +85,79 @@ class HornSoundDetector: NSObject, ObservableObject {
6285 print ( " 소리 분류기 생성 실패: \( error) " )
6386 }
6487 }
88+ //fix
6589
66- private func checkNotificationPermission( ) {
67- UNUserNotificationCenter . current ( ) . getNotificationSettings { settings in
68- switch settings. authorizationStatus {
69- case . notDetermined:
70- self . requestNotificationPermission ( )
71- case . denied:
72- print ( " 알림 권한이 거부되었습니다. 설정에서 권한을 변경해주세요. " )
73- case . authorized, . provisional, . ephemeral:
74- print ( " 알림 권한이 허용되었습니다. " )
75- @unknown default :
76- break
77- }
78- }
79- }
90+ // private func checkNotificationPermission() {
91+ // UNUserNotificationCenter.current().getNotificationSettings { settings in
92+ // switch settings.authorizationStatus {
93+ // case .notDetermined:
94+ // self.requestNotificationPermission()
95+ // case .denied:
96+ // print("알림 권한이 거부되었습니다. 설정에서 권한을 변경해주세요.")
97+ // case .authorized, .provisional, .ephemeral:
98+ // print("알림 권한이 허용되었습니다.")
99+ // @unknown default:
100+ // break
101+ // }
102+ // }
103+ // }
80104
81- private func requestNotificationPermission( ) {
82- UNUserNotificationCenter . current ( ) . requestAuthorization ( options: [ . alert, . sound] ) { granted, error in
83- if granted {
84- print ( " 알림 권한이 허용되었습니다. " )
85- } else if let error = error {
86- print ( " 알림 권한 요청 중 오류 발생: \( error. localizedDescription) " )
87- }
88- }
89- }
105+ // private func requestNotificationPermission() {
106+ // UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound]) { granted, error in
107+ // if granted {
108+ // print("알림 권한이 허용되었습니다.")
109+ // } else if let error = error {
110+ // print("알림 권한 요청 중 오류 발생: \(error.localizedDescription)")
111+ // }
112+ // }
113+ // }
90114
115+ //fix
116+
117+
91118 func startRecording( ) {
92119 guard !isRecording else {
93120 print ( " 녹음이 이미 시작된 상태입니다. " )
94121 return
95122 }
123+ //fix
96124
97- let format = inputNode. outputFormat ( forBus: 0 )
98- streamAnalyzer = SNAudioStreamAnalyzer ( format: format)
125+ // let format = inputNode.outputFormat(forBus: 0)
126+ // streamAnalyzer = SNAudioStreamAnalyzer(format: format)
99127
100- guard let streamAnalyzer = streamAnalyzer,
101- let soundClassifier = soundClassifier else {
128+ // guard let streamAnalyzer = streamAnalyzer,
129+ // let soundClassifier = soundClassifier else {
130+ // print("스트림 분석기 또는 소리 분류기 생성 실패")
131+ // return
132+ // }
133+ //fix
134+ guard let streamAnalyzer = streamAnalyzer, let soundClassifier = soundClassifier else {
102135 print ( " 스트림 분석기 또는 소리 분류기 생성 실패 " )
103136 return
104137 }
105-
138+
139+
106140 do {
107141 let request = try SNClassifySoundRequest ( mlModel: soundClassifier. model)
108142 try . add ( request, withObserver: self )
109143 } catch {
110144 print ( " 분류 요청 생성 실패: \( error) " )
111145 return
112146 }
147+ // //fix
148+
149+ // inputNode.installTap(onBus: 0, bufferSize: 8192, format: format) { [weak self] buffer, time in
150+ // self?.streamAnalyzer?.analyze(buffer, atAudioFramePosition: time.sampleTime)
151+ // }
113152
153+ //fix
154+
155+ let format = inputNode. outputFormat ( forBus: 0 )
114156 inputNode. installTap ( onBus: 0 , bufferSize: 8192 , format: format) { [ weak self] buffer, time in
115157 self ? . streamAnalyzer? . analyze ( buffer, atAudioFramePosition: time. sampleTime)
116158 }
117-
159+
160+
118161 audioEngine. prepare ( )
119162 do {
120163 try . start ( )
@@ -125,7 +168,7 @@ class HornSoundDetector: NSObject, ObservableObject {
125168 print ( " 오디오 엔진 시작 실패: \( error) " )
126169 }
127170 }
128-
171+
129172 @objc func stopRecording( ) {
130173 guard isRecording else {
131174 print ( " 녹음이 이미 중지된 상태입니다. " )
@@ -138,9 +181,11 @@ class HornSoundDetector: NSObject, ObservableObject {
138181 isRecording = false
139182 print ( " 오디오 엔진 중지됨 " )
140183 }
184+
141185
142186 func sendNotification( title: String , body: String ) {
143187 print ( " 알림 발송 시도 " )
188+
144189 let content = UNMutableNotificationContent ( )
145190 content. title = title
146191 content. body = body
@@ -161,25 +206,39 @@ extension HornSoundDetector: SNResultsObserving {
161206 func request( _ request: SNRequest , didProduce result: SNResult ) {
162207 guard let result = result as? SNClassificationResult else { return }
163208
164- let topClassifications = result. classifications. prefix ( 3 )
209+ //fix
210+ // let topClassifications = result.classifications.prefix(3)
165211
166- DispatchQueue . main. async {
167- // 첫 번째 분류를 가장 신뢰도 높은 것으로 설정
168- if let topClassification = topClassifications. first ( where: { classification in
169- return classification. identifier == " Bicyclebell " || classification. identifier == " Carhorn " || classification. identifier == " Siren "
170- } ) {
171- self . topClassification = topClassification // 가장 높은 분류 저장
172- self . classificationResult = topClassification. identifier // 소리 종류만 저장
173- }
212+ // DispatchQueue.main.async {
213+ // // 첫 번째 분류를 가장 신뢰도 높은 것으로 설정
214+ // if let topClassification = topClassifications.first(where: { classification in
215+ // return classification.identifier == "Bicyclebell" || classification.identifier == "Carhorn" || classification.identifier == "Siren"
216+ // }) {
217+ // self.topClassification = topClassification // 가장 높은 분류 저장
218+ // self.classificationResult = topClassification.identifier // 소리 종류만 저장
219+ // }
174220
175- for (index, classification) in topClassifications. enumerated ( ) {
176- if classification. identifier == " Bicyclebell " || classification. identifier == " Carhorn " || classification. identifier == " Siren " {
177- // 경적 및 사이렌 소리 감지
178- if classification. confidence >= 1.0 {
179- self . mlConfidences [ index] = classification. confidence
180- // 원하는 로직을 추가하세요
181- }
221+ // for (index, classification) in topClassifications.enumerated() {
222+ // if classification.identifier == "Bicyclebell" || classification.identifier == "Carhorn" || classification.identifier == "Siren" {
223+ // // 경적 및 사이렌 소리 감지
224+ // if classification.confidence >= 1.0 {
225+ // self.mlConfidences[index] = classification.confidence
226+ // // 원하는 로직을 추가하세요
227+ // }
228+ // }
229+ //fix
230+ DispatchQueue . main. async {
231+ if let topClassification = result. classifications. first, topClassification. confidence >= 1.0 {
232+ let isRelevantSound = [ " Bicyclebell " , " Carhorn " , " Siren " ] . contains ( topClassification. identifier)
233+ if isRelevantSound {
234+ self . classificationResult = topClassification. identifier
235+ self . appRootManager. detectedSound = topClassification. identifier
236+ self . appRootManager. currentRoot = . warning // 루트 변경
237+ print ( " 감지된 소리: \( topClassification. identifier) - 신뢰도: \( topClassification. confidence) " )
182238 }
239+ } else {
240+ print ( " 신뢰도 부족 또는 관련 없는 소리 감지됨 " )
241+
183242 }
184243 }
185244 }
0 commit comments