Function compute

plugins/wasi_nn/wasinn_piper.cpp:373–487

}

Expect<WASINN::ErrNo> compute(WASINN::WasiNNEnvironment &Env,
                              uint32_t ContextId) noexcept {
  auto &CxtRef = Env.NNContext[ContextId].get<Context>();
  auto &GraphRef = Env.NNGraph[CxtRef.GraphId].get<Graph>();

  if (!CxtRef.Line) {
    spdlog::error("[WASI-NN] Piper backend: Input is not set."sv);
    return WASINN::ErrNo::InvalidArgument;
  }

  std::string TextToSpeak = CxtRef.Line.value();

  if (GraphRef.Config->JsonInput) {
    simdjson::dom::parser Parser;
    simdjson::dom::element Doc;
    simdjson::padded_string const PaddedInput(TextToSpeak);

    if (Parser.parse(PaddedInput).get(Doc) != simdjson::SUCCESS) {
      spdlog::error("[WASI-NN] Piper backend: Failed to parse JSON input."sv);
      return WASINN::ErrNo::InvalidArgument;
    }

    simdjson::dom::object JsonObj;
    if (Doc.get(JsonObj) != simdjson::SUCCESS) {
      spdlog::error("[WASI-NN] Piper backend: JSON input is not an object."sv);
      return WASINN::ErrNo::InvalidArgument;
    }

    std::string_view TmpText;
    if (JsonObj["text"].get(TmpText) != simdjson::SUCCESS) {
      spdlog::error(
          "[WASI-NN] Piper backend: JSON input must contain 'text' field."sv);
      return WASINN::ErrNo::InvalidArgument;
    }
    TextToSpeak = std::string(TmpText);

    SynthesisConfig NewConfig;
    if (parseSynthesisConfig(NewConfig, JsonObj) == WASINN::ErrNo::Success) {
      CxtRef.JsonInputSynthesisConfig =
          std::make_unique<std::optional<SynthesisConfig>>(NewConfig);
    }
  }

  piper_synthesize_options Options =
      piper_default_synthesize_options(GraphRef.Synth.get());
  updatePiperOptions(GraphRef.Config->DefaultSynthesisConfig, Options);

  auto OutputType = SynthesisConfigOutputType::OUTPUT_WAV;
  if (GraphRef.Config->DefaultSynthesisConfig.OutputType) {
    OutputType = GraphRef.Config->DefaultSynthesisConfig.OutputType.value();
  }

  if (CxtRef.JsonInputSynthesisConfig &&
      CxtRef.JsonInputSynthesisConfig->has_value()) {
    updatePiperOptions(CxtRef.JsonInputSynthesisConfig->value(), Options);
    if (CxtRef.JsonInputSynthesisConfig->value().OutputType) {
      OutputType = CxtRef.JsonInputSynthesisConfig->value().OutputType.value();
    }