102 lines
3.9 KiB
C++
102 lines
3.9 KiB
C++
|
// 图形界面版：悬浮字幕窗。
|
|||
|
|
#include <cstdio>
|
|||
|
|
|
|||
|
|
#include <QApplication>
|
|||
|
|
#include <QCommandLineOption>
|
|||
|
|
#include <QCommandLineParser>
|
|||
|
|
#include <QGuiApplication>
|
|||
|
|
#include <QTimer>
|
|||
|
|
|
|||
|
|
#include "appcli.h"
|
|||
|
|
#include "captionwindow.h"
|
|||
|
|
#include "livetranscriber.h"
|
|||
|
|
#include "qwen3asr.h"
|
|||
|
|
#include "streamingasr.h"
|
|||
|
|
#include "transcript.h"
|
|||
|
|
|
|||
|
|
namespace {
|
|||
|
|
|
|||
|
|
int runGui(lc::Qwen3Asr & asr, const lc::StreamOptions & sop,
|
|||
|
|
const QString & screenshotPath, int screenshotDelaySec, int opacityPercent)
|
|||
|
|
{
|
|||
|
|
lc::CaptionWindow window;
|
|||
|
|
window.setPanelOpacity(opacityPercent);
|
|||
|
|
lc::LiveTranscriber live(asr);
|
|||
|
|
|
|||
|
|
QObject::connect(&live, &lc::LiveTranscriber::caption, &window,
|
|||
|
|
[&window](const QString & committed, const QString & tail,
|
|||
|
|
const QString & lang) {
|
|||
|
|
window.setLanguage(
|
|||
|
|
QString::fromStdString(lc::languageDisplayName(lang.toStdString())));
|
|||
|
|
window.setCaption(committed, tail);
|
|||
|
|
});
|
|||
|
|
QObject::connect(&live, &lc::LiveTranscriber::failed, &window, [](const QString & m) {
|
|||
|
|
qWarning("采集失败: %s", qUtf8Printable(m));
|
|||
|
|
});
|
|||
|
|
QObject::connect(&window, &lc::CaptionWindow::closed, qApp, &QCoreApplication::quit);
|
|||
|
|
QObject::connect(&window, &lc::CaptionWindow::languageClicked, &window, [] {
|
|||
|
|
qInfo() << "语言标签被点击（设置界面将在 Phase 4 接入）";
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
QString err;
|
|||
|
|
if (!live.start(sop, &err)) {
|
|||
|
|
qCritical("启动采集失败: %s", qUtf8Printable(err));
|
|||
|
|
return 1;
|
|||
|
|
}
|
|||
|
|
qInfo("正在采集: %s", qUtf8Printable(live.deviceName()));
|
|||
|
|
|
|||
|
|
window.show();
|
|||
|
|
|
|||
|
|
if (!screenshotPath.isEmpty()) {
|
|||
|
|
// 自截图：用于验证渲染（不依赖外部截图工具）
|
|||
|
|
QTimer::singleShot(screenshotDelaySec * 1000, &window, [&window, screenshotPath] {
|
|||
|
|
const bool ok1 = window.grab().save(screenshotPath);
|
|||
|
|
|
|||
|
|
window.setCollapsed(false);
|
|||
|
|
QString expandedPath = screenshotPath;
|
|||
|
|
expandedPath.replace(QStringLiteral(".png"), QStringLiteral("-expanded.png"));
|
|||
|
|
const bool ok2 = window.grab().save(expandedPath);
|
|||
|
|
|
|||
|
|
std::fprintf(stderr, "截图: 折叠态 %s / 展开态 %s\n",
|
|||
|
|
ok1 ? "成功" : "失败", ok2 ? "成功" : "失败");
|
|||
|
|
QCoreApplication::quit();
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return QCoreApplication::exec();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} // namespace
|
|||
|
|
|
|||
|
|
int main(int argc, char ** argv)
|
|||
|
|
{
|
|||
|
|
// 说明意图：DPI 取整策略必须在 QGuiApplication 之前定。
|
|||
|
|
// （Deepin 的会话环境已经设了 QT_SCALE_FACTOR_ROUNDING_POLICY，
|
|||
|
|
// 所以本机仍会看到 Qt 的那条提醒，那不是本程序的问题。）
|
|||
|
|
QGuiApplication::setHighDpiScaleFactorRoundingPolicy(
|
|||
|
|
Qt::HighDpiScaleFactorRoundingPolicy::PassThrough);
|
|||
|
|
|
|||
|
|
QApplication app(argc, argv);
|
|||
|
|
QCoreApplication::setApplicationName(QStringLiteral("live-caption"));
|
|||
|
|
|
|||
|
|
QCommandLineParser parser;
|
|||
|
|
parser.setApplicationDescription(QStringLiteral("实时字幕（悬浮窗）"));
|
|||
|
|
parser.addHelpOption();
|
|||
|
|
lc::app::addModelOptions(parser);
|
|||
|
|
lc::app::addStreamOptions(parser);
|
|||
|
|
parser.addOption(QCommandLineOption("opacity", "字幕背景不透明度（20-100，百分比）", "n", "84"));
|
|||
|
|
parser.addOption(QCommandLineOption("screenshot", "启动后自动截图并退出（验证渲染用）", "path"));
|
|||
|
|
parser.addOption(QCommandLineOption("screenshot-delay", "截图前等待秒数", "sec", "6"));
|
|||
|
|
parser.process(app);
|
|||
|
|
|
|||
|
|
const lc::AsrOptions asrOpt = lc::app::asrOptions(parser);
|
|||
|
|
const lc::StreamOptions sop = lc::app::streamOptions(parser);
|
|||
|
|
|
|||
|
|
lc::Qwen3Asr asr;
|
|||
|
|
if (!lc::app::loadModel(asr, asrOpt)) return 1;
|
|||
|
|
|
|||
|
|
return runGui(asr, sop,
|
|||
|
|
parser.value("screenshot"),
|
|||
|
|
parser.value("screenshot-delay").toInt(),
|
|||
|
|
parser.value("opacity").toInt());
|
|||
|
|
}
|