Differential D19523 Diff 53151 plugins/dockers/recorder/recorderdocker_dock.cpp

Changeset View

Standalone View

View Options

plugins/dockers/recorder/recorderdocker_dock.cpp

  • This file was added.
1​/*
2​ * Copyright (c) 2009 Cyrille Berger <cberger@cberger.net>
remptUnsubmitted
Not Done

Take care of copyright lines when copying code :-)

rempt: Take care of copyright lines when copying code :-)
3​ *
4​ * This library is free software; you can redistribute it and/or modify
5​ * it under the terms of the GNU Lesser General Public License as published by
6​ * the Free Software Foundation; version 2.1 of the License.
7​ *
8​ * This library is distributed in the hope that it will be useful,
9​ * but WITHOUT ANY WARRANTY; without even the implied warranty of
10​ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11​ * GNU Lesser General Public License for more details.
12​ *
13​ * You should have received a copy of the GNU Lesser General Public License
14​ * along with this program; if not, write to the Free Software
15​ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16​ */
17​
18​#include "recorderdocker_dock.h"
19​
20​#include <QLabel>
21​#include <QVBoxLayout>
22​#include <QStatusBar>
23​#include <klocalizedstring.h>
24​
25​#include "kis_canvas2.h"
26​#include <KisViewManager.h>
27​#include <kis_zoom_manager.h>
28​#include "kis_image.h"
29​#include "kis_paint_device.h"
30​#include "kis_signal_compressor.h"
31​#include <QDir>
32​#include <QStringBuilder>
33​#include <QRegularExpression>
34​#include <QRegularExpressionMatch>
35​#include <QtConcurrent>
36​#include <KoColorSpaceRegistry.h>
37​#include <QFileDialog>
38​#include <kis_icon_utils.h>
39​#include <klocalizedstring.h>
40​#include <kactioncollection.h>
41​#include <QRegExp>
42​#include <QRegExpValidator>
43​#include "encoder.h"
44​
45​RecorderDockerDock::RecorderDockerDock( )
46​ : QDockWidget(i18n("Recorder"))
47​ , m_canvas(0)
48​ , m_imageIdleWatcher(1000)
49​ , m_recordEnabled(false)
50​ , m_recordCounter(0)
51​ , m_encoder(nullptr)
52​{
53​ QWidget *page = new QWidget(this);
54​ m_layout = new QGridLayout(page);
55​ m_recordDirectoryLabel = new QLabel(this);
56​ m_recordDirectoryLabel->setText("Directory:");
yurchorUnsubmitted
Not Done

It might be that i18n() is missing here.

yurchor: It might be that i18n() is missing here.
57​
58​ m_layout->addWidget(m_recordDirectoryLabel, 0, 0,1,2);
59​
60​ m_recordDirectoryLineEdit = new QLineEdit(this);
61​ m_recordDirectoryLineEdit->setText(QDir::homePath());
62​ m_recordDirectoryLineEdit->setReadOnly(true);
63​ m_layout->addWidget(m_recordDirectoryLineEdit, 1, 0);
64​
65​ m_recordDirectoryPushButton = new QPushButton(this);
66​ m_recordDirectoryPushButton->setIcon(KisIconUtils::loadIcon("folder"));
67​ m_recordDirectoryPushButton->setToolTip(i18n("Record Video"));
68​
69​ m_layout->addWidget(m_recordDirectoryPushButton, 1, 1);
70​
71​ m_imageNameLabel = new QLabel(this);
72​ m_imageNameLabel->setText("Video Name:");
yurchorUnsubmitted
Not Done

It might be that i18n() is missing here.

yurchor: It might be that i18n() is missing here.
73​
74​ m_layout->addWidget(m_imageNameLabel, 2,0,1,2);
75​
76​ m_imageNameLineEdit = new QLineEdit(this);
77​ m_imageNameLineEdit->setText("image");
yurchorUnsubmitted
Not Done

It might be that i18n() is missing here

yurchor: It might be that i18n() is missing here
78​
79​ QRegExp rx("[0-9a-zA-z_]+");
80​ QValidator *validator = new QRegExpValidator(rx, this);
81​ m_imageNameLineEdit->setValidator(validator);
82​
83​ m_layout->addWidget(m_imageNameLineEdit, 3, 0);
84​
85​ m_recordToggleButton = new QPushButton(this);
86​ m_recordToggleButton->setCheckable(true);
87​ m_recordToggleButton->setIcon(KisIconUtils::loadIcon("media-record"));
88​ m_recordToggleButton->setToolTip(i18n("Record Video"));
89​ m_layout->addWidget(m_recordToggleButton, 3, 1);
90​
91​ m_logLabel = new QLabel(this);
92​ m_logLabel->setText("Recent Save:");
93​ m_layout->addWidget(m_logLabel, 4,0,1,2);
94​ m_logLineEdit = new QLineEdit(this);
95​ m_logLineEdit->setReadOnly(true);
96​ m_layout->addWidget(m_logLineEdit, 5,0,1,2);
97​
98​ m_spacer = new QSpacerItem(1, 1, QSizePolicy::Minimum,QSizePolicy::Expanding);
99​ m_layout->addItem(m_spacer, 6, 0,1,2);
100​ connect(m_recordDirectoryPushButton, SIGNAL(clicked()), this, SLOT(onSelectRecordFolderButtonClicked()));
101​ connect(m_recordToggleButton, SIGNAL(toggled(bool)), this, SLOT(onRecordButtonToggled(bool)));
102​ setWidget(page);
103​}
104​
105​void RecorderDockerDock::setCanvas(KoCanvasBase * canvas)
106​{
107​ if(m_canvas == canvas)
108​ return;
109​
110​ setEnabled(canvas != 0);
111​
112​ if (m_canvas) {
113​ m_canvas->disconnectCanvasObserver(this);
114​ m_canvas->image()->disconnect(this);
115​ }
116​
117​ m_canvas = dynamic_cast<KisCanvas2*>(canvas);
118​
119​ if (m_canvas) {
120​ m_imageIdleWatcher.setTrackedImage(m_canvas->image());
121​
122​ connect(&m_imageIdleWatcher, &KisIdleWatcher::startedIdleMode, this, &RecorderDockerDock::generateThumbnail);
123​
124​ connect(m_canvas->image(), SIGNAL(sigImageUpdated(QRect)),SLOT(startUpdateCanvasProjection()));
125​ connect(m_canvas->image(), SIGNAL(sigSizeChanged(QPointF, QPointF)),SLOT(startUpdateCanvasProjection()));
126​ }
127​}
128​
129​void RecorderDockerDock::startUpdateCanvasProjection()
130​{
131​ m_imageIdleWatcher.startCountdown();
132​}
133​
134​void RecorderDockerDock::unsetCanvas()
135​{
136​ setEnabled(false);
137​ m_canvas = 0;
138​}
139​
140​void RecorderDockerDock::onRecordButtonToggled(bool enabled)
141​{
142​ bool enabled2 = enabled;
143​ enableRecord(enabled2, m_recordDirectoryLineEdit->text() % "/" % m_imageNameLineEdit->text());
144​
145​ if (enabled && !enabled2)
146​ {
147​ disconnect(m_recordToggleButton, SIGNAL(toggle(bool)), this, SLOT(onRecordButtonToggled(bool)));
148​ m_recordToggleButton->setChecked(false);
149​
150​ connect(m_recordToggleButton, SIGNAL(toggle(bool)), this, SLOT(onRecordButtonToggled(bool)));
151​ }
152​}
153​
154​void RecorderDockerDock::onSelectRecordFolderButtonClicked()
155​{
156​ QFileDialog dialog(this);
157​ dialog.setFileMode(QFileDialog::DirectoryOnly);
158​ QString folder = dialog.getExistingDirectory(this, tr("Select Output Folder"), m_recordDirectoryLineEdit->text(), QFileDialog::ShowDirsOnly);
yurchorUnsubmitted
Not Done

It might be that i18n() should be used instead of tr() here.

yurchor: It might be that i18n() should be used instead of tr() here.
159​ m_recordDirectoryLineEdit->setText(folder);
160​}
161​
162​void RecorderDockerDock::enableRecord(bool &enabled, const QString &path)
163​{
164​ m_recordEnabled = enabled;
165​ if (m_recordEnabled)
166​ {
167​ m_recordPath = path;
168​
169​ QUrl fileUrl(m_recordPath);
170​
171​ QString filename = fileUrl.fileName();
172​ QString dirPath = fileUrl.adjusted(QUrl::RemoveFilename).path();
173​
174​ QDir dir(dirPath);
175​
176​ if (!dir.exists())
177​ {
178​ if (!dir.mkpath(dirPath))
179​ {
180​ enabled = m_recordEnabled = false;
181​ return;
182​ }
183​ }
184​
185​ QFileInfoList images = dir.entryInfoList({filename % "_*.webm"});
186​
187​ QRegularExpression namePattern("^"%filename%"_([0-9]{7}).webm$");
188​ m_recordCounter = -1;
189​ foreach(auto info, images)
190​ {
191​ QRegularExpressionMatch match = namePattern.match(info.fileName());
192​ if (match.hasMatch())
193​ {
194​ QString count = match.captured(1);
195​ int numCount = count.toInt();
196​
197​ if (m_recordCounter < numCount)
198​ {
199​ m_recordCounter = numCount;
200​ }
201​ }
202​ }
203​
204​ if (m_canvas)
205​ {
206​ m_recordingCanvas = m_canvas;
207​
208​ QString finalFileName = QString(m_recordPath % "_%1.webm").arg(++m_recordCounter, 7, 10, QChar('0'));
209​ m_encoder = new Encoder();
210​ m_encoder->init(finalFileName.toStdString().c_str(), m_canvas->image()->width(), m_canvas->image()->height());
211​ startUpdateCanvasProjection();
212​ }
213​ else
214​ {
215​ enabled = m_recordEnabled = false;
216​ return;
217​ }
218​ }
219​ else
220​ {
221​ if (m_encoder)
222​ {
223​ m_encoder->finish();
224​ m_encoder = nullptr;
225​ }
226​ }
227​}
228​
229​void RecorderDockerDock::generateThumbnail()
230​{
231​ if (m_recordEnabled)
232​ {
233​ if (m_canvas && m_recordingCanvas == m_canvas)
234​ {
235​ disconnect(&m_imageIdleWatcher, &KisIdleWatcher::startedIdleMode, this, &RecorderDockerDock::generateThumbnail);
236​ if (m_encoder)
237​ {
238​ KisImageSP image = m_canvas->image();
239​ KisPaintDeviceSP dev = image->projection();
240​
241​ gpointer data;
242​ gsize size = image->width() * image->height() * dev->pixelSize();
243​ data = g_malloc(size);
244​ dev->readBytes((quint8*)data, 0, 0, image->width(), image->height());
245​ m_encoder->pushFrame(data, size);
246​ }
247​
248​ connect(&m_imageIdleWatcher, &KisIdleWatcher::startedIdleMode, this, &RecorderDockerDock::generateThumbnail);
249​ }
250​ }
251​}