Wrap SherpaOnnxOfflineTtsModelConfig.

Author: csukuangfj

wasm/sherpa-onnx-wasm-main.cc

@@ -13,7 +13,8 @@ extern "C" {
 
 static_assert(sizeof(SherpaOnnxOfflineTtsVitsModelConfig) == 7 * 4, "");
 
-void MyPrint(SherpaOnnxOfflineTtsVitsModelConfig *vits_model_config) {
+void MyPrint(SherpaOnnxOfflineTtsModelConfig *tts_model_config) {
+  auto vits_model_config = &tts_model_config->vits;
   fprintf(stdout, "----------vits model config----------\n");
   fprintf(stdout, "model: %s\n", vits_model_config->model);
   fprintf(stdout, "lexicon: %s\n", vits_model_config->lexicon);
@@ -22,6 +23,11 @@ void MyPrint(SherpaOnnxOfflineTtsVitsModelConfig *vits_model_config) {
   fprintf(stdout, "noise scale: %.3f\n", vits_model_config->noise_scale);
   fprintf(stdout, "noise scale w: %.3f\n", vits_model_config->noise_scale_w);
   fprintf(stdout, "length scale: %.3f\n", vits_model_config->length_scale);
+
+  fprintf(stdout, "----------tts model config----------\n");
+  fprintf(stdout, "num threads: %d\n", tts_model_config->num_threads);
+  fprintf(stdout, "debug: %d\n", tts_model_config->debug);
+  fprintf(stdout, "provider: %s\n", tts_model_config->provider);
 }
 
 void CopyHeap(const char *src, int32_t num_bytes, char *dst) {

wasm/sherpa-onnx.js

@@ -3,6 +3,11 @@ function freeConfig(config) {
   if ('buffer' in config) {
     _free(config.buffer);
   }
+
+  if ('config' in config) {
+    freeConfig(config.config)
+  }
+
   _free(config.ptr);
 }
 
@@ -18,7 +23,7 @@ function initSherpaOnnxOfflineTtsVitsModelConfig(config) {
   let buffer = _malloc(n);
 
   let len = 7 * 4;
-  let ptr = _malloc(7 * 4);
+  let ptr = _malloc(len);
 
   let offset = 0;
   stringToUTF8(config.model, buffer + offset, modelLen);
@@ -55,16 +60,50 @@ function initSherpaOnnxOfflineTtsVitsModelConfig(config) {
   }
 }
 
+function initSherpaOnnxOfflineTtsModelConfig(config) {
+  let vitsModelConfig =
+      initSherpaOnnxOfflineTtsVitsModelConfig(config.offlineTtsVitsModelConfig);
+
+  let len = vitsModelConfig.len + 3 * 4;
+  let ptr = _malloc(len);
+
+  let offset = 0;
+  Module._CopyHeap(vitsModelConfig.ptr, vitsModelConfig.len, ptr + offset);
+  offset += vitsModelConfig.len;
+
+  Module.setValue(ptr + offset, config.numThreads, 'i32');
+  offset += 4;
+
+  Module.setValue(ptr + offset, config.debug, 'i32');
+  offset += 4;
+
+  let providerLen = lengthBytesUTF8(config.provider) + 1;
+  let buffer = _malloc(providerLen);
+  stringToUTF8(config.provider, buffer, providerLen);
+  Module.setValue(ptr + offset, buffer, 'i8*');
+
+  return {
+    buffer: buffer, ptr: ptr, len: len, config: vitsModelConfig,
+  }
+}
+
 function initSherpaOnnxOfflineTts() {
-  let offlineTtsVitsModelConfig = initSherpaOnnxOfflineTtsVitsModelConfig({
+  let offlineTtsVitsModelConfig = {
     model: './model.onnx',
     lexicon: './lexicon.txt',
     tokens: './tokens.txt',
     dataDir: './espeak-ng-data',
     noiseScale: 0.667,
     noiseScaleW: 0.8,
     lengthScale: 1.0,
+  };
+  let offlineTtsModelConfig = initSherpaOnnxOfflineTtsModelConfig({
+    offlineTtsVitsModelConfig: offlineTtsVitsModelConfig,
+    numThreads: 1,
+    debug: 1,
+    provider: 'cpu',
   })
   console.log(offlineTtsVitsModelConfig)
-  Module._MyPrint(offlineTtsVitsModelConfig.ptr);
+  console.log(offlineTtsModelConfig)
+  Module._MyPrint(offlineTtsModelConfig.ptr);
 }