Support passing TTS callback in Swift API (#1218)

This commit is contained in:
Fangjun Kuang
2024-08-05 14:06:21 +08:00
committed by GitHub
parent 9caa488019
commit 6422966a7f
2 changed files with 54 additions and 4 deletions

View File

@@ -757,6 +757,14 @@ class SherpaOnnxGeneratedAudioWrapper {
}
}
typealias TtsCallbackWithArg = (
@convention(c) (
UnsafePointer<Float>?, // const float* samples
Int32, // int32_t n
UnsafeMutableRawPointer? // void *arg
) -> Int32
)?
class SherpaOnnxOfflineTtsWrapper {
/// A pointer to the underlying counterpart in C
let tts: OpaquePointer!
@@ -780,6 +788,17 @@ class SherpaOnnxOfflineTtsWrapper {
return SherpaOnnxGeneratedAudioWrapper(audio: audio)
}
func generateWithCallbackWithArg(
text: String, callback: TtsCallbackWithArg, arg: UnsafeMutableRawPointer, sid: Int = 0,
speed: Float = 1.0
) -> SherpaOnnxGeneratedAudioWrapper {
let audio: UnsafePointer<SherpaOnnxGeneratedAudio>? =
SherpaOnnxOfflineTtsGenerateWithCallbackWithArg(
tts, toCPointer(text), Int32(sid), speed, callback, arg)
return SherpaOnnxGeneratedAudioWrapper(audio: audio)
}
}
// spoken language identification