1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// SPDX-FileCopyrightText: 2023 Arjen Hiemstra <ahiemstra@heimr.nl>
//
// SPDX-License-Identifier: LGPL-2.1-only OR LGPL-3.0-only OR LicenseRef-KDE-Accepted-LGPL
#include <csignal>
#include <QCommandLineParser>
#include <QDebug>
#include <QFile>
#include <QGuiApplication>
#include <QTimer>
#include <QUrl>
#include "PortalSession.h"
#include "VideoStream.h"
using namespace Qt::StringLiterals;
int main(int argc, char **argv)
{
QGuiApplication application{argc, argv};
QCommandLineParser parser;
parser.addHelpOption();
parser.addOptions({
{u"quit-after"_s, u"Quit after running for this amount of seconds"_s, u"seconds"_s},
{u"monitor"_s, u"Index of the monitor to display."_s, u"monitor"_s, u"-1"_s},
{u"quality"_s, u"Encoding quality of the stream, from 0 (lowest) to 100 (highest)"_s, u"quality"_s},
});
parser.process(application);
KRdp::PortalSession session;
session.requestStreamingEnable(&application);
session.setActiveStream(parser.value(u"monitor"_s).toInt());
if (parser.isSet(u"quality"_s)) {
session.setVideoQuality(parser.value(u"quality"_s).toUShort());
}
signal(SIGINT, [](int) {
QCoreApplication::exit(0);
});
signal(SIGUSR1, [](int) {
QCoreApplication::exit(0);
});
QFile file{u"stream.raw"_s};
if (!file.open(QFile::WriteOnly)) {
qDebug() << "Failed opening stream.raw";
return -1;
}
QObject::connect(&session, &KRdp::PortalSession::frameReceived, &session, [&file](const KRdp::VideoFrame &frame) {
file.write(frame.data);
});
QTimer timer;
QObject::connect(&timer, &QTimer::timeout, &application, &QCoreApplication::quit);
if (parser.isSet(u"quit-after"_s)) {
QObject::connect(&session, &KRdp::PortalSession::started, &timer, qOverload<>(&QTimer::start));
timer.setInterval(parser.value(u"quit-after"_s).toInt() * 1000);
}
session.start();
auto result = application.exec();
file.close();
return result;
}
|
