Method HTDemucsSession

src/models/demucs/session.cpp:133–171

}

HTDemucsSession::~HTDemucsSession() = default;

std::string HTDemucsSession::family() const {
    return kFamily;
}

runtime::VoiceTaskKind HTDemucsSession::task_kind() const {
    return task_.task;
}

runtime::RunMode HTDemucsSession::run_mode() const {
    return task_.mode;
}

void HTDemucsSession::prepare(const runtime::SessionPreparationRequest & request) {
    if (!request.audio.has_value()) {
        throw std::runtime_error("HTDemucs prepare() requires an audio contract");
    }
    const auto & config = pipeline_->config();
    if (request.audio->sample_rate != config.sample_rate) {
        throw std::runtime_error(
            "HTDemucs prepare() sample rate mismatch: expected " +
            std::to_string(config.sample_rate) + ", got " +
            std::to_string(request.audio->sample_rate));
    }
    const bool mono_compatible = config.audio_channels == 2 && request.audio->channels == 1;
    if (request.audio->channels != config.audio_channels && !mono_compatible) {
        throw std::runtime_error(
            "HTDemucs prepare() channel mismatch: expected " +
            std::to_string(config.audio_channels) + ", got " +
            std::to_string(request.audio->channels));
    }
    mark_prepared();
}

runtime::TaskResult HTDemucsSession::run(const runtime::TaskRequest & request) {
    require_prepared("HTDemucs run()");
    runtime::validate_spec_backed_request_options(request.options, *contract_, "HTDemucs");
    if (!request.audio_input.has_value()) {
        throw std::runtime_error("HTDemucs run() requires audio_input");
    }