window-basic-main.cpp 181.6 KB
Newer Older
1
/******************************************************************************
2
    Copyright (C) 2013-2015 by Hugh Bailey <obs.jim@gmail.com>
J
jp9000 已提交
3
                               Zachary Lund <admin@computerquip.com>
4
                               Philippe Groarke <philippe.groarke@gmail.com>
5 6 7

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
8
    the Free Software Foundation, either version 2 of the License, or
9 10 11
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
12
    but WITHOUT ANY WARRANTY; without even the implied warranty of
13 14 15 16 17 18 19
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/

S
Shaolin 已提交
20
#include <ctime>
J
jp9000 已提交
21
#include <obs.hpp>
22
#include <QGuiApplication>
J
jp9000 已提交
23
#include <QMessageBox>
24
#include <QShowEvent>
25
#include <QDesktopServices>
J
jp9000 已提交
26
#include <QFileDialog>
27
#include <QDesktopWidget>
28
#include <QScreen>
29 30
#include <QColorDialog>
#include <QSizePolicy>
31

J
jp9000 已提交
32
#include <util/dstr.h>
33 34
#include <util/util.hpp>
#include <util/platform.h>
P
Palana 已提交
35
#include <util/profiler.hpp>
36
#include <util/dstr.hpp>
37

38
#include "obs-app.hpp"
39
#include "platform.hpp"
40
#include "visibility-item-widget.hpp"
41
#include "item-widget-helpers.hpp"
42
#include "window-basic-settings.hpp"
43
#include "window-namedialog.hpp"
J
jp9000 已提交
44
#include "window-basic-auto-config.hpp"
J
jp9000 已提交
45
#include "window-basic-source-select.hpp"
J
jp9000 已提交
46
#include "window-basic-main.hpp"
J
jp9000 已提交
47
#include "window-basic-stats.hpp"
J
jp9000 已提交
48
#include "window-basic-main-outputs.hpp"
J
jp9000 已提交
49
#include "window-log-reply.hpp"
J
jp9000 已提交
50
#include "window-projector.hpp"
P
Palana 已提交
51
#include "window-remux.hpp"
J
jp9000 已提交
52
#include "qt-wrappers.hpp"
53
#include "display-helpers.hpp"
54
#include "volume-control.hpp"
55
#include "remote-text.hpp"
S
Shaolin 已提交
56 57
#include <fstream>
#include <sstream>
58

59
#ifdef _WIN32
J
jp9000 已提交
60 61 62
#include "win-update/win-update.hpp"
#endif

J
jp9000 已提交
63
#include "ui_OBSBasic.h"
64
#include "ui_ColorSelect.h"
65

J
jp9000 已提交
66
#include <fstream>
67 68
#include <sstream>

69 70 71
#include <QScreen>
#include <QWindow>

J
jp9000 已提交
72 73 74
#include <json11.hpp>

using namespace json11;
75
using namespace std;
J
jp9000 已提交
76

77
#ifdef BROWSER_AVAILABLE
78
#include <browser-panel.hpp>
J
jp9000 已提交
79 80 81
static CREATE_BROWSER_WIDGET_PROC create_browser_widget = nullptr;
#endif

J
jp9000 已提交
82 83 84 85 86 87 88 89 90 91
namespace {

template <typename OBSRef>
struct SignalContainer {
	OBSRef ref;
	vector<shared_ptr<OBSSignal>> handlers;
};

}

J
jp9000 已提交
92 93
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
94
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
95
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
96
Q_DECLARE_METATYPE(SignalContainer<OBSScene>);
J
jp9000 已提交
97

P
Palana 已提交
98 99 100 101 102 103 104 105 106 107 108 109 110
template <typename T>
static T GetOBSRef(QListWidgetItem *item)
{
	return item->data(static_cast<int>(QtDataRole::OBSRef)).value<T>();
}

template <typename T>
static void SetOBSRef(QListWidgetItem *item, T &&val)
{
	item->setData(static_cast<int>(QtDataRole::OBSRef),
			QVariant::fromValue(val));
}

111 112
static void AddExtraModulePaths()
{
113
	char base_module_dir[512];
114 115 116 117
#if defined(_WIN32) || defined(__APPLE__)
	int ret = GetProgramDataPath(base_module_dir, sizeof(base_module_dir),
			"obs-studio/plugins/%module%");
#else
118
	int ret = GetConfigPath(base_module_dir, sizeof(base_module_dir),
119
			"obs-studio/plugins/%module%");
120
#endif
B
BtbN 已提交
121

122
	if (ret <= 0)
123 124 125
		return;

	string path = (char*)base_module_dir;
126
#if defined(__APPLE__)
127
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
128 129 130 131 132

	BPtr<char> config_bin = os_get_config_path_ptr("obs-studio/plugins/%module%/bin");
	BPtr<char> config_data = os_get_config_path_ptr("obs-studio/plugins/%module%/data");
	obs_add_module_path(config_bin, config_data);

133 134 135 136 137 138 139
#elif ARCH_BITS == 64
	obs_add_module_path((path + "/bin/64bit").c_str(),
			(path + "/data").c_str());
#else
	obs_add_module_path((path + "/bin/32bit").c_str(),
			(path + "/data").c_str());
#endif
140 141
}

142 143
extern obs_frontend_callbacks *InitializeAPIInterface(OBSBasic *main);

144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
static int CountVideoSources()
{
	int count = 0;

	auto countSources = [] (void *param, obs_source_t *source)
	{
		if (!source)
			return true;

		uint32_t flags = obs_source_get_output_flags(source);
		if ((flags & OBS_SOURCE_VIDEO) != 0)
			(*reinterpret_cast<int*>(param))++;

		return true;
	};

	obs_enum_sources(countSources, &count);
	return count;
}

164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
void assignDockToggle(QDockWidget *dock, QAction *action)
{
	auto handleWindowToggle = [action] (bool vis)
	{
		action->blockSignals(true);
		action->setChecked(vis);
		action->blockSignals(false);
	};
	auto handleMenuToggle = [dock] (bool check)
	{
		dock->blockSignals(true);
		dock->setVisible(check);
		dock->blockSignals(false);
	};

	dock->connect(dock->toggleViewAction(), &QAction::toggled,
			handleWindowToggle);
	dock->connect(action, &QAction::toggled,
			handleMenuToggle);
}

185
OBSBasic::OBSBasic(QWidget *parent)
186
	: OBSMainWindow  (parent),
J
jp9000 已提交
187
	  ui             (new Ui::OBSBasic)
188
{
189 190
	setAttribute(Qt::WA_NativeWindow);

J
jp9000 已提交
191 192
	setAcceptDrops(true);

193 194
	api = InitializeAPIInterface(this);

195
	ui->setupUi(this);
J
jp9000 已提交
196 197
	ui->previewDisabledLabel->setVisible(false);

J
jp9000 已提交
198 199
	startingDockLayout = saveState();

200 201 202 203 204 205 206 207 208
	statsDock = new QDockWidget();
	statsDock->setObjectName(QStringLiteral("statsDock"));
	statsDock->setFeatures(QDockWidget::AllDockWidgetFeatures);
	statsDock->setWindowTitle(QTStr("Basic.Stats"));
	addDockWidget(Qt::BottomDockWidgetArea, statsDock);
	statsDock->setVisible(false);
	statsDock->setFloating(true);
	statsDock->resize(700, 200);

S
Socapex 已提交
209
	copyActionsDynamicProperties();
210

211
	char styleSheetPath[512];
J
jp9000 已提交
212 213
	int ret = GetProfilePath(styleSheetPath, sizeof(styleSheetPath),
			"stylesheet.qss");
214
	if (ret > 0) {
H
HomeWorld 已提交
215 216 217 218 219
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
220 221
	}

P
Palana 已提交
222 223 224
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
225
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
226

227 228 229
	qRegisterMetaTypeStreamOperators<
		std::vector<std::shared_ptr<OBSSignal>>>(
				"std::vector<std::shared_ptr<OBSSignal>>");
230
	qRegisterMetaTypeStreamOperators<OBSScene>("OBSScene");
231
	qRegisterMetaTypeStreamOperators<OBSSceneItem>("OBSSceneItem");
232

P
Palana 已提交
233 234 235
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

236
	auto displayResize = [this]() {
237 238 239 240
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
241 242 243 244
	};

	connect(windowHandle(), &QWindow::screenChanged, displayResize);
	connect(ui->preview, &OBSQTDisplay::DisplayResized, displayResize);
J
jp9000 已提交
245

P
pkv 已提交
246 247 248
	delete shortcutFilter;
	shortcutFilter = CreateShortcutFilter();
	installEventFilter(shortcutFilter);
P
Palana 已提交
249

250
	stringstream name;
J
jp9000 已提交
251
	name << "OBS " << App()->GetVersionString();
252 253 254
	blog(LOG_INFO, "%s", name.str().c_str());
	blog(LOG_INFO, "---------------------------------");

255
	UpdateTitleBar();
J
jp9000 已提交
256 257 258 259 260 261 262 263

	connect(ui->scenes->itemDelegate(),
			SIGNAL(closeEditor(QWidget*,
					QAbstractItemDelegate::EndEditHint)),
			this,
			SLOT(SceneNameEdited(QWidget*,
					QAbstractItemDelegate::EndEditHint)));

264
	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
265
	cpuUsageTimer = new QTimer(this);
266
	connect(cpuUsageTimer.data(), SIGNAL(timeout()),
J
jp9000 已提交
267 268
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
269

S
Shaolin 已提交
270 271 272 273 274 275 276 277 278 279 280
	QAction *renameScene = new QAction(ui->scenesDock);
	renameScene->setShortcutContext(Qt::WidgetWithChildrenShortcut);
	connect(renameScene, SIGNAL(triggered()), this, SLOT(EditSceneName()));
	ui->scenesDock->addAction(renameScene);

	QAction *renameSource = new QAction(ui->sourcesDock);
	renameSource->setShortcutContext(Qt::WidgetWithChildrenShortcut);
	connect(renameSource, SIGNAL(triggered()), this,
			SLOT(EditSceneItemName()));
	ui->sourcesDock->addAction(renameSource);

281
#ifdef __APPLE__
S
Shaolin 已提交
282 283 284
	renameScene->setShortcut({Qt::Key_Return});
	renameSource->setShortcut({Qt::Key_Return});

285 286
	ui->actionRemoveSource->setShortcuts({Qt::Key_Backspace});
	ui->actionRemoveScene->setShortcuts({Qt::Key_Backspace});
287 288 289

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
S
Shaolin 已提交
290 291 292
#else
	renameScene->setShortcut({Qt::Key_F2});
	renameSource->setShortcut({Qt::Key_F2});
293
#endif
294 295 296 297 298 299 300 301 302 303 304 305 306 307

	auto addNudge = [this](const QKeySequence &seq, const char *s)
	{
		QAction *nudge = new QAction(ui->preview);
		nudge->setShortcut(seq);
		nudge->setShortcutContext(Qt::WidgetShortcut);
		ui->preview->addAction(nudge);
		connect(nudge, SIGNAL(triggered()), this, s);
	};

	addNudge(Qt::Key_Up, SLOT(NudgeUp()));
	addNudge(Qt::Key_Down, SLOT(NudgeDown()));
	addNudge(Qt::Key_Left, SLOT(NudgeLeft()));
	addNudge(Qt::Key_Right, SLOT(NudgeRight()));
J
jp9000 已提交
308 309 310 311 312 313

	assignDockToggle(ui->scenesDock, ui->toggleScenes);
	assignDockToggle(ui->sourcesDock, ui->toggleSources);
	assignDockToggle(ui->mixerDock, ui->toggleMixer);
	assignDockToggle(ui->transitionsDock, ui->toggleTransitions);
	assignDockToggle(ui->controlsDock, ui->toggleControls);
314
	assignDockToggle(statsDock, ui->toggleStats);
S
SuslikV 已提交
315 316 317 318 319 320 321

	//hide all docking panes
	ui->toggleScenes->setChecked(false);
	ui->toggleSources->setChecked(false);
	ui->toggleMixer->setChecked(false);
	ui->toggleTransitions->setChecked(false);
	ui->toggleControls->setChecked(false);
A
Alex Anderson 已提交
322
	ui->toggleStats->setChecked(false);
S
SuslikV 已提交
323

324 325
	QPoint curPos;

S
SuslikV 已提交
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
	//restore parent window geometry
	const char *geometry = config_get_string(App()->GlobalConfig(),
			"BasicWindow", "geometry");
	if (geometry != NULL) {
		QByteArray byteArray = QByteArray::fromBase64(
				QByteArray(geometry));
		restoreGeometry(byteArray);

		QRect windowGeometry = normalGeometry();
		if (!WindowPositionValid(windowGeometry)) {
			QRect rect = App()->desktop()->geometry();
			setGeometry(QStyle::alignedRect(
						Qt::LeftToRight,
						Qt::AlignCenter,
						size(), rect));
		}
342 343 344 345 346 347

		curPos = pos();
	} else {
		QRect desktopRect = QGuiApplication::primaryScreen()->geometry();
		QSize adjSize = desktopRect.size() / 2 - size() / 2;
		curPos = QPoint(adjSize.width(), adjSize.height());
S
SuslikV 已提交
348
	}
349 350 351 352 353 354

	QPoint curSize(width(), height());
	QPoint statsDockSize(statsDock->width(), statsDock->height());
	QPoint statsDockPos = curSize / 2 - statsDockSize / 2;
	QPoint newPos = curPos + statsDockPos;
	statsDock->move(newPos);
355 356
}

357 358
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent,
		vector<OBSSource> &audioSources)
359
{
360
	obs_source_t *source = obs_get_output_source(channel);
361 362 363
	if (!source)
		return;

364 365
	audioSources.push_back(source);

366
	obs_data_t *data = obs_save_source(source);
367

J
jp9000 已提交
368
	obs_data_set_obj(parent, name, data);
369 370 371 372 373

	obs_data_release(data);
	obs_source_release(source);
}

374 375
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder,
		obs_data_array_t *quickTransitionData, int transitionDuration,
J
jp9000 已提交
376
		obs_data_array_t *transitions,
C
cg2121 已提交
377
		OBSScene &scene, OBSSource &curProgramScene,
378
		obs_data_array_t *savedProjectorList)
379
{
380 381 382 383 384 385 386 387 388 389
	obs_data_t *saveData = obs_data_create();

	vector<OBSSource> audioSources;
	audioSources.reserve(5);

	SaveAudioDevice(DESKTOP_AUDIO_1, 1, saveData, audioSources);
	SaveAudioDevice(DESKTOP_AUDIO_2, 2, saveData, audioSources);
	SaveAudioDevice(AUX_AUDIO_1,     3, saveData, audioSources);
	SaveAudioDevice(AUX_AUDIO_2,     4, saveData, audioSources);
	SaveAudioDevice(AUX_AUDIO_3,     5, saveData, audioSources);
390
	SaveAudioDevice(AUX_AUDIO_4,     6, saveData, audioSources);
391

392 393 394
	/* -------------------------------- */
	/* save non-group sources           */

395 396
	auto FilterAudioSources = [&](obs_source_t *source)
	{
397 398 399
		if (obs_source_is_group(source))
			return false;

400 401 402 403 404 405 406 407 408 409 410
		return find(begin(audioSources), end(audioSources), source) ==
				end(audioSources);
	};
	using FilterAudioSources_t = decltype(FilterAudioSources);

	obs_data_array_t *sourcesArray = obs_save_sources_filtered(
			[](void *data, obs_source_t *source)
	{
		return (*static_cast<FilterAudioSources_t*>(data))(source);
	}, static_cast<void*>(&FilterAudioSources));

411 412 413 414 415 416 417 418 419 420 421 422
	/* -------------------------------- */
	/* save group sources separately    */

	/* saving separately ensures they won't be loaded in older versions */
	obs_data_array_t *groupsArray = obs_save_sources_filtered(
			[](void*, obs_source_t *source)
	{
		return obs_source_is_group(source);
	}, nullptr);

	/* -------------------------------- */

423 424
	obs_source_t *transition = obs_get_output_source(0);
	obs_source_t *currentScene = obs_scene_get_source(scene);
425
	const char   *sceneName   = obs_source_get_name(currentScene);
426
	const char   *programName = obs_source_get_name(curProgramScene);
427

J
jp9000 已提交
428 429 430
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

J
jp9000 已提交
431
	obs_data_set_string(saveData, "current_scene", sceneName);
432
	obs_data_set_string(saveData, "current_program_scene", programName);
J
jp9000 已提交
433
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
434
	obs_data_set_string(saveData, "name", sceneCollection);
J
jp9000 已提交
435
	obs_data_set_array(saveData, "sources", sourcesArray);
436
	obs_data_set_array(saveData, "groups", groupsArray);
437
	obs_data_set_array(saveData, "quick_transitions", quickTransitionData);
J
jp9000 已提交
438
	obs_data_set_array(saveData, "transitions", transitions);
C
cg2121 已提交
439
	obs_data_set_array(saveData, "saved_projectors", savedProjectorList);
440
	obs_data_array_release(sourcesArray);
441
	obs_data_array_release(groupsArray);
442 443 444 445 446

	obs_data_set_string(saveData, "current_transition",
			obs_source_get_name(transition));
	obs_data_set_int(saveData, "transition_duration", transitionDuration);
	obs_source_release(transition);
447 448 449 450

	return saveData;
}

S
Socapex 已提交
451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470
void OBSBasic::copyActionsDynamicProperties()
{
	// Themes need the QAction dynamic properties
	for (QAction *x : ui->scenesToolbar->actions()) {
		QWidget* temp = ui->scenesToolbar->widgetForAction(x);

		for (QByteArray &y : x->dynamicPropertyNames()) {
			temp->setProperty(y, x->property(y));
		}
	}

	for (QAction *x : ui->sourcesToolbar->actions()) {
		QWidget* temp = ui->sourcesToolbar->widgetForAction(x);

		for (QByteArray &y : x->dynamicPropertyNames()) {
			temp->setProperty(y, x->property(y));
		}
	}
}

S
Shaolin 已提交
471 472 473 474 475 476 477 478 479 480
void OBSBasic::UpdateVolumeControlsDecayRate()
{
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");

	for (size_t i = 0; i < volumes.size(); i++) {
		volumes[i]->SetMeterDecayRate(meterDecayRate);
	}
}

481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503
void OBSBasic::UpdateVolumeControlsPeakMeterType()
{
	uint32_t peakMeterTypeIdx = config_get_uint(basicConfig, "Audio",
			"PeakMeterType");

	enum obs_peak_meter_type peakMeterType;
	switch (peakMeterTypeIdx) {
	case 0:
		peakMeterType = SAMPLE_PEAK_METER;
		break;
	case 1:
		peakMeterType = TRUE_PEAK_METER;
		break;
	default:
		peakMeterType = SAMPLE_PEAK_METER;
		break;
	}

	for (size_t i = 0; i < volumes.size(); i++) {
		volumes[i]->setPeakMeterType(peakMeterType);
	}
}

504 505
void OBSBasic::ClearVolumeControls()
{
C
craftwar 已提交
506 507
	for (VolControl *vol : volumes)
		delete vol;
508 509 510 511

	volumes.clear();
}

J
jp9000 已提交
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526
obs_data_array_t *OBSBasic::SaveSceneListOrder()
{
	obs_data_array_t *sceneOrder = obs_data_array_create();

	for (int i = 0; i < ui->scenes->count(); i++) {
		obs_data_t *data = obs_data_create();
		obs_data_set_string(data, "name",
				QT_TO_UTF8(ui->scenes->item(i)->text()));
		obs_data_array_push_back(sceneOrder, data);
		obs_data_release(data);
	}

	return sceneOrder;
}

C
cg2121 已提交
527 528
obs_data_array_t *OBSBasic::SaveProjectors()
{
529
	obs_data_array_t *savedProjectors = obs_data_array_create();
C
cg2121 已提交
530

531 532 533
	auto saveProjector = [savedProjectors](OBSProjector *projector) {
		if (!projector)
			return;
R
Ryan Foster 已提交
534 535

		obs_data_t *data = obs_data_create();
536
		ProjectorType type = projector->GetProjectorType();
S
Shaolin 已提交
537 538 539 540 541 542 543 544 545 546 547
		switch (type) {
		case ProjectorType::Scene:
		case ProjectorType::Source: {
			obs_source_t *source = projector->GetSource();
			const char *name = obs_source_get_name(source);
			obs_data_set_string(data, "name", name);
			break;
		}
		default:
			break;
		}
548 549
		obs_data_set_int(data, "monitor", projector->GetMonitor());
		obs_data_set_int(data, "type", static_cast<int>(type));
S
Shaolin 已提交
550 551 552
		obs_data_set_string(data, "geometry",
				projector->saveGeometry().toBase64()
						.constData());
553
		obs_data_array_push_back(savedProjectors, data);
R
Ryan Foster 已提交
554
		obs_data_release(data);
555
	};
R
Ryan Foster 已提交
556

557 558
	for (QPointer<QWidget> &proj : projectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
559

S
Shaolin 已提交
560 561
	for (QPointer<QWidget> &proj : windowProjectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
562

563
	return savedProjectors;
S
Shaolin 已提交
564 565
}

566 567
void OBSBasic::Save(const char *file)
{
568 569 570 571 572
	OBSScene scene = GetCurrentScene();
	OBSSource curProgramScene = OBSGetStrongRef(programScene);
	if (!curProgramScene)
		curProgramScene = obs_scene_get_source(scene);

J
jp9000 已提交
573
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
J
jp9000 已提交
574
	obs_data_array_t *transitions = SaveTransitions();
575
	obs_data_array_t *quickTrData = SaveQuickTransitions();
C
cg2121 已提交
576
	obs_data_array_t *savedProjectorList = SaveProjectors();
R
Ryan Foster 已提交
577
	obs_data_t *saveData = GenerateSaveData(sceneOrder, quickTrData,
J
jp9000 已提交
578
			ui->transitionDuration->value(), transitions,
579
			scene, curProgramScene, savedProjectorList);
580

J
jp9000 已提交
581
	obs_data_set_bool(saveData, "preview_locked", ui->preview->Locked());
582 583 584 585 586 587 588 589
	obs_data_set_bool(saveData, "scaling_enabled",
			ui->preview->IsFixedScaling());
	obs_data_set_int(saveData, "scaling_level",
			ui->preview->GetScalingLevel());
	obs_data_set_double(saveData, "scaling_off_x",
			ui->preview->GetScrollX());
	obs_data_set_double(saveData, "scaling_off_y",
			ui->preview->GetScrollY());
J
jp9000 已提交
590

J
jp9000 已提交
591 592 593 594 595 596 597
	if (api) {
		obs_data_t *moduleObj = obs_data_create();
		api->on_save(moduleObj);
		obs_data_set_obj(saveData, "modules", moduleObj);
		obs_data_release(moduleObj);
	}

598 599
	if (!obs_data_save_json_safe(saveData, file, "tmp", "bak"))
		blog(LOG_ERROR, "Could not save scene data to %s", file);
600 601

	obs_data_release(saveData);
J
jp9000 已提交
602
	obs_data_array_release(sceneOrder);
603
	obs_data_array_release(quickTrData);
J
jp9000 已提交
604
	obs_data_array_release(transitions);
C
cg2121 已提交
605
	obs_data_array_release(savedProjectorList);
606 607
}

I
Ilya M 已提交
608 609 610 611 612 613 614 615 616 617 618 619 620
void OBSBasic::DeferSaveBegin()
{
	os_atomic_inc_long(&disableSaving);
}

void OBSBasic::DeferSaveEnd()
{
	long result = os_atomic_dec_long(&disableSaving);
	if (result == 0) {
		SaveProject();
	}
}

621
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
622
{
623
	obs_data_t *data = obs_data_get_obj(parent, name);
624 625 626
	if (!data)
		return;

627
	obs_source_t *source = obs_load_source(data);
628 629 630 631 632 633 634 635
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

636 637 638
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
639
	obs_properties_t *props = obs_get_source_properties(output_id);
640 641 642 643 644 645 646 647 648 649 650 651 652 653
	size_t count = 0;

	if (!props)
		return false;

	obs_property_t *devices = obs_properties_get(props, "device_id");
	if (devices)
		count = obs_property_list_item_count(devices);

	obs_properties_destroy(props);

	return count != 0;
}

654
void OBSBasic::CreateFirstRunSources()
655
{
656 657 658 659 660 661 662 663 664
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

	if (hasDesktopAudio)
		ResetAudioDevice(App()->OutputAudioSource(), "default",
				Str("Basic.DesktopDevice1"), 1);
	if (hasInputAudio)
		ResetAudioDevice(App()->InputAudioSource(), "default",
				Str("Basic.AuxDevice1"), 3);
665 666
}

667
void OBSBasic::CreateDefaultScene(bool firstStart)
668 669 670 671
{
	disableSaving++;

	ClearSceneData();
672 673 674 675
	InitDefaultTransitions();
	CreateDefaultQuickTransitions();
	ui->transitionDuration->setValue(300);
	SetTransition(fadeTransition);
676 677 678

	obs_scene_t  *scene  = obs_scene_create(Str("Basic.Scene"));

679
	if (firstStart)
680
		CreateFirstRunSources();
681

682
	SetCurrentScene(scene, true);
683
	obs_scene_release(scene);
J
jp9000 已提交
684 685

	disableSaving--;
686 687
}

J
jp9000 已提交
688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716
static void ReorderItemByName(QListWidget *lw, const char *name, int newIndex)
{
	for (int i = 0; i < lw->count(); i++) {
		QListWidgetItem *item = lw->item(i);

		if (strcmp(name, QT_TO_UTF8(item->text())) == 0) {
			if (newIndex != i) {
				item = lw->takeItem(i);
				lw->insertItem(newIndex, item);
			}
			break;
		}
	}
}

void OBSBasic::LoadSceneListOrder(obs_data_array_t *array)
{
	size_t num = obs_data_array_count(array);

	for (size_t i = 0; i < num; i++) {
		obs_data_t *data = obs_data_array_item(array, i);
		const char *name = obs_data_get_string(data, "name");

		ReorderItemByName(ui->scenes, name, (int)i);

		obs_data_release(data);
	}
}

C
cg2121 已提交
717 718
void OBSBasic::LoadSavedProjectors(obs_data_array_t *array)
{
719 720 721 722 723
	for (SavedProjectorInfo *info : savedProjectorsArray) {
		delete info;
	}
	savedProjectorsArray.clear();

C
cg2121 已提交
724 725 726 727 728
	size_t num = obs_data_array_count(array);

	for (size_t i = 0; i < num; i++) {
		obs_data_t *data = obs_data_array_item(array, i);

729 730 731 732
		SavedProjectorInfo *info = new SavedProjectorInfo();
		info->monitor = obs_data_get_int(data, "monitor");
		info->type = static_cast<ProjectorType>(obs_data_get_int(data,
				"type"));
S
Shaolin 已提交
733 734 735
		info->geometry = std::string(
				obs_data_get_string(data, "geometry"));
		info->name = std::string(obs_data_get_string(data, "name"));
736
		savedProjectorsArray.emplace_back(info);
S
Shaolin 已提交
737 738 739 740 741

		obs_data_release(data);
	}
}

742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762
static void LogFilter(obs_source_t*, obs_source_t *filter, void *v_val)
{
	const char *name = obs_source_get_name(filter);
	const char *id = obs_source_get_id(filter);
	int val = (int)(intptr_t)v_val;
	string indent;

	for (int i = 0; i < val; i++)
		indent += "    ";

	blog(LOG_INFO, "%s- filter: '%s' (%s)", indent.c_str(), name, id);
}

static bool LogSceneItem(obs_scene_t*, obs_sceneitem_t *item, void*)
{
	obs_source_t *source = obs_sceneitem_get_source(item);
	const char *name = obs_source_get_name(source);
	const char *id = obs_source_get_id(source);

	blog(LOG_INFO, "    - source: '%s' (%s)", name, id);

763 764 765 766 767 768 769 770 771 772 773 774
	obs_monitoring_type monitoring_type =
		obs_source_get_monitoring_type(source);

	if (monitoring_type != OBS_MONITORING_TYPE_NONE) {
		const char *type =
			(monitoring_type == OBS_MONITORING_TYPE_MONITOR_ONLY)
			? "monitor only"
			: "monitor and output";

		blog(LOG_INFO, "        - monitoring: %s", type);
	}

775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798
	obs_source_enum_filters(source, LogFilter, (void*)(intptr_t)2);
	return true;
}

void OBSBasic::LogScenes()
{
	blog(LOG_INFO, "------------------------------------------------");
	blog(LOG_INFO, "Loaded scenes:");

	for (int i = 0; i < ui->scenes->count(); i++) {
		QListWidgetItem *item = ui->scenes->item(i);
		OBSScene scene = GetOBSRef<OBSScene>(item);

		obs_source_t *source = obs_scene_get_source(scene);
		const char *name = obs_source_get_name(source);

		blog(LOG_INFO, "- scene '%s':", name);
		obs_scene_enum_items(scene, LogSceneItem, nullptr);
		obs_source_enum_filters(source, LogFilter, (void*)(intptr_t)1);
	}

	blog(LOG_INFO, "------------------------------------------------");
}

799 800
void OBSBasic::Load(const char *file)
{
801 802 803 804 805
	disableSaving++;

	obs_data_t *data = obs_data_create_from_json_file_safe(file, "bak");
	if (!data) {
		disableSaving--;
806
		blog(LOG_INFO, "No scene file found, creating default scene");
807
		CreateDefaultScene(true);
J
jp9000 已提交
808
		SaveProject();
809
		return;
810
	}
811

812
	ClearSceneData();
813
	InitDefaultTransitions();
814

815 816 817 818
	obs_data_t *modulesObj = obs_data_get_obj(data, "modules");
	if (api)
		api->on_preload(modulesObj);

J
jp9000 已提交
819
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
820
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
821
	obs_data_array_t *groups     = obs_data_get_array(data, "groups");
J
jp9000 已提交
822
	obs_data_array_t *transitions= obs_data_get_array(data, "transitions");
J
jp9000 已提交
823 824
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
825 826 827 828 829
	const char       *programSceneName = obs_data_get_string(data,
			"current_program_scene");
	const char       *transitionName = obs_data_get_string(data,
			"current_transition");

830 831 832 833 834 835
	if (!opt_starting_scene.empty()) {
		programSceneName = opt_starting_scene.c_str();
		if (!IsPreviewProgramMode())
			sceneName = opt_starting_scene.c_str();
	}

836 837 838 839 840 841
	int newDuration = obs_data_get_int(data, "transition_duration");
	if (!newDuration)
		newDuration = 300;

	if (!transitionName)
		transitionName = obs_source_get_name(fadeTransition);
J
jp9000 已提交
842 843 844 845 846 847 848

	const char *curSceneCollection = config_get_string(
			App()->GlobalConfig(), "Basic", "SceneCollection");

	obs_data_set_default_string(data, "name", curSceneCollection);

	const char       *name = obs_data_get_string(data, "name");
849
	obs_source_t     *curScene;
850 851
	obs_source_t     *curProgramScene;
	obs_source_t     *curTransition;
852

J
jp9000 已提交
853 854 855
	if (!name || !*name)
		name = curSceneCollection;

856 857 858 859 860
	LoadAudioDevice(DESKTOP_AUDIO_1, 1, data);
	LoadAudioDevice(DESKTOP_AUDIO_2, 2, data);
	LoadAudioDevice(AUX_AUDIO_1,     3, data);
	LoadAudioDevice(AUX_AUDIO_2,     4, data);
	LoadAudioDevice(AUX_AUDIO_3,     5, data);
861
	LoadAudioDevice(AUX_AUDIO_4,     6, data);
862

863 864 865 866 867 868 869
	if (!sources) {
		sources = groups;
		groups = nullptr;
	} else {
		obs_data_array_push_back_array(sources, groups);
	}

870
	obs_load_sources(sources, nullptr, nullptr);
871

J
jp9000 已提交
872 873
	if (transitions)
		LoadTransitions(transitions);
J
jp9000 已提交
874 875 876
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

J
jp9000 已提交
877 878
	obs_data_array_release(transitions);

879 880 881 882 883 884 885
	curTransition = FindTransition(transitionName);
	if (!curTransition)
		curTransition = fadeTransition;

	ui->transitionDuration->setValue(newDuration);
	SetTransition(curTransition);

886
retryScene:
887
	curScene = obs_get_source_by_name(sceneName);
888
	curProgramScene = obs_get_source_by_name(programSceneName);
889 890 891 892 893 894 895 896 897 898 899 900 901

	/* if the starting scene command line parameter is bad at all,
	 * fall back to original settings */
	if (!opt_starting_scene.empty() && (!curScene || !curProgramScene)) {
		sceneName = obs_data_get_string(data, "current_scene");
		programSceneName = obs_data_get_string(data,
				"current_program_scene");
		obs_source_release(curScene);
		obs_source_release(curProgramScene);
		opt_starting_scene.clear();
		goto retryScene;
	}

902 903 904 905 906 907 908 909
	if (!curProgramScene) {
		curProgramScene = curScene;
		obs_source_addref(curScene);
	}

	SetCurrentScene(curScene, true);
	if (IsPreviewProgramMode())
		TransitionToScene(curProgramScene, true);
910
	obs_source_release(curScene);
911
	obs_source_release(curProgramScene);
912 913

	obs_data_array_release(sources);
914
	obs_data_array_release(groups);
J
jp9000 已提交
915
	obs_data_array_release(sceneOrder);
J
jp9000 已提交
916

917 918 919 920 921 922 923 924 925
	/* ------------------- */

	bool projectorSave = config_get_bool(GetGlobalConfig(), "BasicWindow",
			"SaveProjectors");

	if (projectorSave) {
		obs_data_array_t *savedProjectors = obs_data_get_array(data,
				"saved_projectors");

926
		if (savedProjectors) {
927
			LoadSavedProjectors(savedProjectors);
928 929 930
			OpenSavedProjectors();
			activateWindow();
		}
931 932 933 934 935 936

		obs_data_array_release(savedProjectors);
	}

	/* ------------------- */

J
jp9000 已提交
937 938 939 940 941 942 943 944
	std::string file_base = strrchr(file, '/') + 1;
	file_base.erase(file_base.size() - 5, 5);

	config_set_string(App()->GlobalConfig(), "Basic", "SceneCollection",
			name);
	config_set_string(App()->GlobalConfig(), "Basic", "SceneCollectionFile",
			file_base.c_str());

945 946 947 948 949 950 951
	obs_data_array_t *quickTransitionData = obs_data_get_array(data,
			"quick_transitions");
	LoadQuickTransitions(quickTransitionData);
	obs_data_array_release(quickTransitionData);

	RefreshQuickTransitions();

J
jp9000 已提交
952 953 954 955
	bool previewLocked = obs_data_get_bool(data, "preview_locked");
	ui->preview->SetLocked(previewLocked);
	ui->actionLockPreview->setChecked(previewLocked);

956 957 958 959 960 961 962 963 964 965
	/* ---------------------- */

	bool fixedScaling = obs_data_get_bool(data, "scaling_enabled");
	int scalingLevel = (int)obs_data_get_int(data, "scaling_level");
	float scrollOffX = (float)obs_data_get_double(data, "scaling_off_x");
	float scrollOffY = (float)obs_data_get_double(data, "scaling_off_y");

	if (fixedScaling) {
		ui->preview->SetScalingLevel(scalingLevel);
		ui->preview->SetScrollingOffset(scrollOffX, scrollOffY);
J
Joseph El-Khouri 已提交
966
	}
967
	ui->preview->SetFixedScaling(fixedScaling);
J
Joseph El-Khouri 已提交
968

969
	/* ---------------------- */
J
Joseph El-Khouri 已提交
970

971
	if (api)
J
jp9000 已提交
972 973
		api->on_load(modulesObj);

974
	obs_data_release(modulesObj);
975
	obs_data_release(data);
J
jp9000 已提交
976

977 978 979
	if (!opt_starting_scene.empty())
		opt_starting_scene.clear();

980
	if (opt_start_streaming) {
981
		blog(LOG_INFO, "Starting stream due to command line parameter");
982 983 984 985 986 987
		QMetaObject::invokeMethod(this, "StartStreaming",
				Qt::QueuedConnection);
		opt_start_streaming = false;
	}

	if (opt_start_recording) {
988
		blog(LOG_INFO, "Starting recording due to command line parameter");
989 990 991 992 993
		QMetaObject::invokeMethod(this, "StartRecording",
				Qt::QueuedConnection);
		opt_start_recording = false;
	}

C
cg2121 已提交
994 995 996 997 998 999
	if (opt_start_replaybuffer) {
		QMetaObject::invokeMethod(this, "StartReplayBuffer",
				Qt::QueuedConnection);
		opt_start_replaybuffer = false;
	}

1000 1001
	LogScenes();

J
jp9000 已提交
1002
	disableSaving--;
1003

1004
	if (api) {
1005
		api->on_event(OBS_FRONTEND_EVENT_SCENE_CHANGED);
1006 1007
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
	}
1008 1009
}

J
jp9000 已提交
1010
#define SERVICE_PATH "service.json"
1011 1012 1013 1014 1015 1016

void OBSBasic::SaveService()
{
	if (!service)
		return;

1017
	char serviceJsonPath[512];
J
jp9000 已提交
1018
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
1019 1020
			SERVICE_PATH);
	if (ret <= 0)
1021 1022
		return;

1023 1024
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
1025

1026
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
1027
	obs_data_set_obj(data, "settings", settings);
1028

1029 1030
	if (!obs_data_save_json_safe(data, serviceJsonPath, "tmp", "bak"))
		blog(LOG_WARNING, "Failed to save service");
1031 1032 1033 1034 1035 1036 1037 1038 1039

	obs_data_release(settings);
	obs_data_release(data);
}

bool OBSBasic::LoadService()
{
	const char *type;

1040
	char serviceJsonPath[512];
J
jp9000 已提交
1041
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
1042 1043
			SERVICE_PATH);
	if (ret <= 0)
1044 1045
		return false;

1046 1047
	obs_data_t *data = obs_data_create_from_json_file_safe(serviceJsonPath,
			"bak");
1048 1049

	obs_data_set_default_string(data, "type", "rtmp_common");
J
jp9000 已提交
1050
	type = obs_data_get_string(data, "type");
1051

1052
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
1053
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
1054

1055
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
1056
			hotkey_data);
1057
	obs_service_release(service);
1058

P
Palana 已提交
1059
	obs_data_release(hotkey_data);
1060 1061 1062 1063 1064 1065 1066 1067
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
P
Palana 已提交
1068 1069
	ProfileScope("OBSBasic::InitService");

1070 1071 1072
	if (LoadService())
		return true;

1073 1074
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
1075 1076
	if (!service)
		return false;
1077
	obs_service_release(service);
1078 1079 1080 1081

	return true;
}

1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097
static const double scaled_vals[] =
{
	1.0,
	1.25,
	(1.0/0.75),
	1.5,
	(1.0/0.6),
	1.75,
	2.0,
	2.25,
	2.5,
	2.75,
	3.0,
	0.0
};

1098 1099
bool OBSBasic::InitBasicConfigDefaults()
{
1100
	QList<QScreen*> screens = QGuiApplication::screens();
1101

1102
	if (!screens.size()) {
1103 1104 1105 1106 1107
		OBSErrorBox(NULL, "There appears to be no monitors.  Er, this "
		                  "technically shouldn't be possible.");
		return false;
	}

1108 1109 1110 1111
	QScreen *primaryScreen = QGuiApplication::primaryScreen();

	uint32_t cx = primaryScreen->size().width();
	uint32_t cy = primaryScreen->size().height();
1112

1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123
	bool oldResolutionDefaults = config_get_bool(App()->GlobalConfig(),
			"General", "Pre19Defaults");

	/* use 1920x1080 for new default base res if main monitor is above
	 * 1920x1080, but don't apply for people from older builds -- only to
	 * new users */
	if (!oldResolutionDefaults && (cx * cy) > (1920 * 1080)) {
		cx = 1920;
		cy = 1080;
	}

P
pkviet 已提交
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
	bool changed = false;

	/* ----------------------------------------------------- */
	/* move over old FFmpeg track settings                   */
	if (config_has_user_value(basicConfig, "AdvOut", "FFAudioTrack") &&
	    !config_has_user_value(basicConfig, "AdvOut", "Pre22.1Settings")) {

		int track = (int)config_get_int(basicConfig, "AdvOut",
				"FFAudioTrack");
		config_set_int(basicConfig, "AdvOut", "FFAudioMixes",
				1LL << (track - 1));
		config_set_bool(basicConfig, "AdvOut", "Pre22.1Settings", true);
		changed = true;
	}

1139 1140 1141 1142 1143 1144 1145 1146 1147 1148
	/* ----------------------------------------------------- */
	/* move over mixer values in advanced if older config */
	if (config_has_user_value(basicConfig, "AdvOut", "RecTrackIndex") &&
	    !config_has_user_value(basicConfig, "AdvOut", "RecTracks")) {

		uint64_t track = config_get_uint(basicConfig, "AdvOut",
				"RecTrackIndex");
		track = 1ULL << (track - 1);
		config_set_uint(basicConfig, "AdvOut", "RecTracks", track);
		config_remove_value(basicConfig, "AdvOut", "RecTrackIndex");
P
pkviet 已提交
1149
		changed = true;
1150 1151 1152 1153
	}

	/* ----------------------------------------------------- */

P
pkviet 已提交
1154 1155 1156 1157 1158
	if (changed)
		config_save_safe(basicConfig, "tmp", nullptr);

	/* ----------------------------------------------------- */

1159
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
1160

1161 1162
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
1163 1164
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
1165 1166
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
1167 1168
	config_set_default_string(basicConfig, "SimpleOutput", "StreamEncoder",
			SIMPLE_ENCODER_X264);
1169
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
1170 1171
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
1172 1173
	config_set_default_bool  (basicConfig, "SimpleOutput", "EnforceBitrate",
			true);
J
jp9000 已提交
1174 1175
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
1176 1177 1178 1179
	config_set_default_string(basicConfig, "SimpleOutput", "RecQuality",
			"Stream");
	config_set_default_string(basicConfig, "SimpleOutput", "RecEncoder",
			SIMPLE_ENCODER_X264);
1180 1181 1182
	config_set_default_bool(basicConfig, "SimpleOutput", "RecRB", false);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBTime", 20);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBSize", 512);
1183 1184
	config_set_default_string(basicConfig, "SimpleOutput", "RecRBPrefix",
			"Replay");
1185

1186 1187
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
1188 1189 1190 1191 1192 1193 1194 1195
	config_set_default_bool  (basicConfig, "AdvOut", "UseRescale", false);
	config_set_default_uint  (basicConfig, "AdvOut", "TrackIndex", 1);
	config_set_default_string(basicConfig, "AdvOut", "Encoder", "obs_x264");

	config_set_default_string(basicConfig, "AdvOut", "RecType", "Standard");

	config_set_default_string(basicConfig, "AdvOut", "RecFilePath",
			GetDefaultVideoSavePath().c_str());
1196
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
1197 1198
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
1199
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
1200 1201 1202
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

1203 1204 1205 1206 1207
	config_set_default_bool  (basicConfig, "AdvOut", "FFOutputToFile",
			true);
	config_set_default_string(basicConfig, "AdvOut", "FFFilePath",
			GetDefaultVideoSavePath().c_str());
	config_set_default_string(basicConfig, "AdvOut", "FFExtension", "mp4");
J
jp9000 已提交
1208
	config_set_default_uint  (basicConfig, "AdvOut", "FFVBitrate", 2500);
1209
	config_set_default_uint  (basicConfig, "AdvOut", "FFVGOPSize", 250);
J
jp9000 已提交
1210 1211
	config_set_default_bool  (basicConfig, "AdvOut", "FFUseRescale",
			false);
1212 1213
	config_set_default_bool  (basicConfig, "AdvOut", "FFIgnoreCompat",
			false);
J
jp9000 已提交
1214
	config_set_default_uint  (basicConfig, "AdvOut", "FFABitrate", 160);
P
pkviet 已提交
1215
	config_set_default_uint  (basicConfig, "AdvOut", "FFAudioMixes", 1);
J
jp9000 已提交
1216 1217 1218 1219 1220

	config_set_default_uint  (basicConfig, "AdvOut", "Track1Bitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "Track2Bitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "Track3Bitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "Track4Bitrate", 160);
J
jp9000 已提交
1221 1222
	config_set_default_uint  (basicConfig, "AdvOut", "Track5Bitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "Track6Bitrate", 160);
J
jp9000 已提交
1223

1224 1225 1226 1227
	config_set_default_bool  (basicConfig, "AdvOut", "RecRB", false);
	config_set_default_uint  (basicConfig, "AdvOut", "RecRBTime", 20);
	config_set_default_int   (basicConfig, "AdvOut", "RecRBSize", 512);

1228 1229 1230
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

1231 1232 1233 1234 1235 1236 1237 1238
	/* don't allow BaseCX/BaseCY to be susceptible to defaults changing */
	if (!config_has_user_value(basicConfig, "Video", "BaseCX") ||
	    !config_has_user_value(basicConfig, "Video", "BaseCY")) {
		config_set_uint(basicConfig, "Video", "BaseCX", cx);
		config_set_uint(basicConfig, "Video", "BaseCY", cy);
		config_save_safe(basicConfig, "tmp", nullptr);
	}

1239
	config_set_default_string(basicConfig, "Output", "FilenameFormatting",
B
bl 已提交
1240
			"%CCYY-%MM-%DD %hh-%mm-%ss");
1241

1242 1243 1244 1245
	config_set_default_bool  (basicConfig, "Output", "DelayEnable", false);
	config_set_default_uint  (basicConfig, "Output", "DelaySec", 20);
	config_set_default_bool  (basicConfig, "Output", "DelayPreserve", true);

1246 1247 1248 1249
	config_set_default_bool  (basicConfig, "Output", "Reconnect", true);
	config_set_default_uint  (basicConfig, "Output", "RetryDelay", 10);
	config_set_default_uint  (basicConfig, "Output", "MaxRetries", 20);

1250
	config_set_default_string(basicConfig, "Output", "BindIP", "default");
D
derrod 已提交
1251 1252 1253 1254
	config_set_default_bool  (basicConfig, "Output", "NewSocketLoopEnable",
			false);
	config_set_default_bool  (basicConfig, "Output", "LowLatencyEnable",
			false);
1255

1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269
	int i = 0;
	uint32_t scale_cx = cx;
	uint32_t scale_cy = cy;

	/* use a default scaled resolution that has a pixel count no higher
	 * than 1280x720 */
	while (((scale_cx * scale_cy) > (1280 * 720)) && scaled_vals[i] > 0.0) {
		double scale = scaled_vals[i++];
		scale_cx = uint32_t(double(cx) / scale);
		scale_cy = uint32_t(double(cy) / scale);
	}

	config_set_default_uint  (basicConfig, "Video", "OutputCX", scale_cx);
	config_set_default_uint  (basicConfig, "Video", "OutputCY", scale_cy);
1270

1271 1272 1273 1274 1275 1276 1277 1278 1279
	/* don't allow OutputCX/OutputCY to be susceptible to defaults
	 * changing */
	if (!config_has_user_value(basicConfig, "Video", "OutputCX") ||
	    !config_has_user_value(basicConfig, "Video", "OutputCY")) {
		config_set_uint(basicConfig, "Video", "OutputCX", scale_cx);
		config_set_uint(basicConfig, "Video", "OutputCY", scale_cy);
		config_save_safe(basicConfig, "tmp", nullptr);
	}

1280 1281 1282 1283 1284
	config_set_default_uint  (basicConfig, "Video", "FPSType", 0);
	config_set_default_string(basicConfig, "Video", "FPSCommon", "30");
	config_set_default_uint  (basicConfig, "Video", "FPSInt", 30);
	config_set_default_uint  (basicConfig, "Video", "FPSNum", 30);
	config_set_default_uint  (basicConfig, "Video", "FPSDen", 1);
1285
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
1286
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
1287
	config_set_default_string(basicConfig, "Video", "ColorSpace", "601");
1288 1289
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
1290

1291 1292 1293 1294 1295
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceId",
			"default");
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceName",
			Str("Basic.Settings.Advanced.Audio.MonitoringDevice"
				".Default"));
1296 1297 1298
	config_set_default_uint  (basicConfig, "Audio", "SampleRate", 44100);
	config_set_default_string(basicConfig, "Audio", "ChannelSetup",
			"Stereo");
S
Shaolin 已提交
1299 1300
	config_set_default_double(basicConfig, "Audio", "MeterDecayRate",
			VOLUME_METER_DECAY_FAST);
1301
	config_set_default_uint  (basicConfig, "Audio", "PeakMeterType", 0);
1302 1303 1304 1305 1306 1307

	return true;
}

bool OBSBasic::InitBasicConfig()
{
P
Palana 已提交
1308 1309
	ProfileScope("OBSBasic::InitBasicConfig");

1310
	char configPath[512];
J
jp9000 已提交
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323

	int ret = GetProfilePath(configPath, sizeof(configPath), "");
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get profile path");
		return false;
	}

	if (os_mkdir(configPath) == MKDIR_ERROR) {
		OBSErrorBox(nullptr, "Failed to create profile path");
		return false;
	}

	ret = GetProfilePath(configPath, sizeof(configPath), "basic.ini");
1324 1325 1326 1327
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
1328

1329 1330 1331
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
1332 1333 1334
		return false;
	}

J
jp9000 已提交
1335 1336 1337 1338 1339
	if (config_get_string(basicConfig, "General", "Name") == nullptr) {
		const char *curName = config_get_string(App()->GlobalConfig(),
				"Basic", "Profile");

		config_set_string(basicConfig, "General", "Name", curName);
1340
		basicConfig.SaveSafe("tmp");
J
jp9000 已提交
1341 1342
	}

1343 1344 1345
	return InitBasicConfigDefaults();
}

1346 1347
void OBSBasic::InitOBSCallbacks()
{
P
Palana 已提交
1348 1349
	ProfileScope("OBSBasic::InitOBSCallbacks");

P
Palana 已提交
1350
	signalHandlers.reserve(signalHandlers.size() + 6);
1351 1352
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_create",
			OBSBasic::SourceCreated, this);
P
Palana 已提交
1353
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_remove",
1354
			OBSBasic::SourceRemoved, this);
P
Palana 已提交
1355
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_activate",
1356
			OBSBasic::SourceActivated, this);
P
Palana 已提交
1357
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_deactivate",
1358
			OBSBasic::SourceDeactivated, this);
P
Palana 已提交
1359
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
1360
			OBSBasic::SourceRenamed, this);
1361 1362
}

J
jp9000 已提交
1363 1364
void OBSBasic::InitPrimitives()
{
P
Palana 已提交
1365 1366
	ProfileScope("OBSBasic::InitPrimitives");

J
jp9000 已提交
1367
	obs_enter_graphics();
J
jp9000 已提交
1368

1369
	gs_render_start(true);
J
jp9000 已提交
1370 1371 1372 1373 1374
	gs_vertex2f(0.0f, 0.0f);
	gs_vertex2f(0.0f, 1.0f);
	gs_vertex2f(1.0f, 1.0f);
	gs_vertex2f(1.0f, 0.0f);
	gs_vertex2f(0.0f, 0.0f);
1375
	box = gs_render_save();
J
jp9000 已提交
1376

1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396
	gs_render_start(true);
	gs_vertex2f(0.0f, 0.0f);
	gs_vertex2f(0.0f, 1.0f);
	boxLeft = gs_render_save();

	gs_render_start(true);
	gs_vertex2f(0.0f, 0.0f);
	gs_vertex2f(1.0f, 0.0f);
	boxTop = gs_render_save();

	gs_render_start(true);
	gs_vertex2f(1.0f, 0.0f);
	gs_vertex2f(1.0f, 1.0f);
	boxRight = gs_render_save();

	gs_render_start(true);
	gs_vertex2f(0.0f, 1.0f);
	gs_vertex2f(1.0f, 1.0f);
	boxBottom = gs_render_save();

1397
	gs_render_start(true);
J
jp9000 已提交
1398 1399 1400 1401
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
1402
	circle = gs_render_save();
J
jp9000 已提交
1403

J
jp9000 已提交
1404
	obs_leave_graphics();
J
jp9000 已提交
1405 1406
}

J
jp9000 已提交
1407 1408 1409 1410 1411 1412 1413 1414
void OBSBasic::ReplayBufferClicked()
{
	if (outputHandler->ReplayBufferActive())
		StopReplayBuffer();
	else
		StartReplayBuffer();
};

J
jp9000 已提交
1415 1416
void OBSBasic::ResetOutputs()
{
P
Palana 已提交
1417 1418
	ProfileScope("OBSBasic::ResetOutputs");

J
jp9000 已提交
1419 1420 1421
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	bool advOut = astrcmpi(mode, "Advanced") == 0;

J
jp9000 已提交
1422 1423
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
1424 1425 1426
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
1427

J
jp9000 已提交
1428 1429 1430 1431 1432 1433
		delete replayBufferButton;

		if (outputHandler->replayBuffer) {
			replayBufferButton = new QPushButton(
					QTStr("Basic.Main.StartReplayBuffer"),
					this);
1434
			replayBufferButton->setCheckable(true);
1435
			connect(replayBufferButton.data(),
J
jp9000 已提交
1436 1437 1438 1439
					&QPushButton::clicked,
					this,
					&OBSBasic::ReplayBufferClicked);

1440
			replayBufferButton->setProperty("themeID", "replayBufferButton");
J
jp9000 已提交
1441 1442
			ui->buttonsVLayout->insertWidget(2, replayBufferButton);
		}
1443

J
jp9000 已提交
1444 1445 1446
		if (sysTrayReplayBuffer)
			sysTrayReplayBuffer->setEnabled(
					!!outputHandler->replayBuffer);
J
jp9000 已提交
1447 1448 1449 1450 1451
	} else {
		outputHandler->Update();
	}
}

J
jp9000 已提交
1452 1453 1454
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot);

1455 1456 1457 1458
#define STARTUP_SEPARATOR \
	"==== Startup complete ==============================================="
#define SHUTDOWN_SEPARATOR \
	"==== Shutting down =================================================="
1459

1460 1461 1462 1463 1464
#define UNSUPPORTED_ERROR \
	"Failed to initialize video:\n\nRequired graphics API functionality " \
	"not found.  Your GPU may not be supported."

#define UNKNOWN_ERROR \
1465 1466
	"Failed to initialize video.  Your GPU may not be supported, " \
	"or your graphics drivers may need to be updated."
1467

1468 1469
void OBSBasic::OBSInit()
{
P
Palana 已提交
1470 1471
	ProfileScope("OBSBasic::OBSInit");

J
jp9000 已提交
1472 1473
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
1474
	char savePath[512];
J
jp9000 已提交
1475 1476 1477 1478 1479 1480 1481 1482
	char fileName[512];
	int ret;

	if (!sceneCollection)
		throw "Failed to get scene collection name";

	ret = snprintf(fileName, 512, "obs-studio/basic/scenes/%s.json",
			sceneCollection);
1483
	if (ret <= 0)
J
jp9000 已提交
1484 1485 1486 1487 1488
		throw "Failed to create scene collection file name";

	ret = GetConfigPath(savePath, sizeof(savePath), fileName);
	if (ret <= 0)
		throw "Failed to get scene collection json file path";
1489

1490 1491
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
1492
	if (!ResetAudio())
1493 1494
		throw "Failed to initialize audio";

1495
	ret = ResetVideo();
1496 1497 1498 1499 1500

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
1501
		throw UNSUPPORTED_ERROR;
1502 1503 1504 1505
	case OBS_VIDEO_INVALID_PARAM:
		throw "Failed to initialize video:  Invalid parameters";
	default:
		if (ret != OBS_VIDEO_SUCCESS)
1506
			throw UNKNOWN_ERROR;
1507 1508
	}

1509
	/* load audio monitoring */
1510
#if defined(_WIN32) || defined(__APPLE__) || HAVE_PULSEAUDIO
1511 1512 1513 1514 1515 1516
	const char *device_name = config_get_string(basicConfig, "Audio",
			"MonitoringDeviceName");
	const char *device_id = config_get_string(basicConfig, "Audio",
			"MonitoringDeviceId");

	obs_set_audio_monitoring_device(device_name, device_id);
1517 1518 1519

	blog(LOG_INFO, "Audio monitoring device:\n\tname: %s\n\tid: %s",
			device_name, device_id);
1520 1521
#endif

1522
	InitOBSCallbacks();
P
Palana 已提交
1523
	InitHotkeys();
1524

1525
	AddExtraModulePaths();
1526
	blog(LOG_INFO, "---------------------------------");
J
jp9000 已提交
1527
	obs_load_all_modules();
1528 1529
	blog(LOG_INFO, "---------------------------------");
	obs_log_loaded_modules();
1530 1531
	blog(LOG_INFO, "---------------------------------");
	obs_post_load_modules();
J
jp9000 已提交
1532

1533
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1534 1535 1536
	create_browser_widget = obs_browser_init_panel();
#endif

1537 1538
	CheckForSimpleModeX264Fallback();

1539
	blog(LOG_INFO, STARTUP_SEPARATOR);
1540

J
jp9000 已提交
1541
	ResetOutputs();
1542
	CreateHotkeys();
J
jp9000 已提交
1543

1544 1545 1546
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
1547 1548
	InitPrimitives();

1549 1550 1551 1552 1553 1554
	sceneDuplicationMode = config_get_bool(App()->GlobalConfig(),
				"BasicWindow", "SceneDuplicationMode");
	swapScenesMode = config_get_bool(App()->GlobalConfig(),
				"BasicWindow", "SwapScenesMode");
	editPropertiesMode = config_get_bool(App()->GlobalConfig(),
				"BasicWindow", "EditPropertiesMode");
C
cg2121 已提交
1555 1556 1557 1558 1559 1560 1561 1562

	if (!opt_studio_mode) {
		SetPreviewProgramMode(config_get_bool(App()->GlobalConfig(),
					"BasicWindow", "PreviewProgramMode"));
	} else {
		SetPreviewProgramMode(true);
		opt_studio_mode = false;
	}
1563

1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
#define SET_VISIBILITY(name, control) \
	do { \
		if (config_has_user_value(App()->GlobalConfig(), \
					"BasicWindow", name)) { \
			bool visible = config_get_bool(App()->GlobalConfig(), \
					"BasicWindow", name); \
			ui->control->setChecked(visible); \
		} \
	} while (false)

	SET_VISIBILITY("ShowListboxToolbars", toggleListboxToolbars);
	SET_VISIBILITY("ShowStatusBar", toggleStatusBar);
#undef SET_VISIBILITY

1578 1579 1580 1581 1582 1583 1584 1585 1586
#ifndef __APPLE__
	{
		ProfileScope("OBSBasic::Load");
		disableSaving--;
		Load(savePath);
		disableSaving++;
	}
#endif

J
jp9000 已提交
1587
	TimedCheckForUpdates();
1588
	loaded = true;
J
jp9000 已提交
1589

1590
	previewEnabled = config_get_bool(App()->GlobalConfig(),
J
jp9000 已提交
1591
			"BasicWindow", "PreviewEnabled");
1592 1593 1594 1595 1596

	if (!previewEnabled && !IsPreviewProgramMode())
		QMetaObject::invokeMethod(this, "EnablePreviewDisplay",
				Qt::QueuedConnection,
				Q_ARG(bool, previewEnabled));
1597 1598 1599 1600 1601 1602 1603 1604 1605

#ifdef _WIN32
	uint32_t winVer = GetWindowsVersion();
	if (winVer > 0 && winVer < 0x602) {
		bool disableAero = config_get_bool(basicConfig, "Video",
				"DisableAero");
		SetAeroEnabled(!disableAero);
	}
#endif
J
jp9000 已提交
1606

1607 1608 1609
#ifndef __APPLE__
	RefreshSceneCollections();
#endif
J
jp9000 已提交
1610
	RefreshProfiles();
J
jp9000 已提交
1611
	disableSaving--;
1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624

	auto addDisplay = [this] (OBSQTDisplay *window)
	{
		obs_display_add_draw_callback(window->GetDisplay(),
				OBSBasic::RenderMain, this);

		struct obs_video_info ovi;
		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
	};

	connect(ui->preview, &OBSQTDisplay::DisplayCreated, addDisplay);

1625
#ifdef _WIN32
J
jp9000 已提交
1626
	SetWin32DropStyle(this);
1627 1628 1629 1630 1631
	show();
#endif

	bool alwaysOnTop = config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"AlwaysOnTop");
1632
	if (alwaysOnTop || opt_always_on_top) {
1633 1634 1635 1636 1637
		SetAlwaysOnTop(this, true);
		ui->actionAlwaysOnTop->setChecked(true);
	}

#ifndef _WIN32
1638
	show();
1639
#endif
J
jp9000 已提交
1640

A
Alex Anderson 已提交
1641
	/* setup stats dock */
1642 1643
	OBSBasicStats *statsDlg = new OBSBasicStats(statsDock, false);
	statsDock->setWidget(statsDlg);
A
Alex Anderson 已提交
1644

J
jp9000 已提交
1645 1646 1647 1648
	const char *dockStateStr = config_get_string(App()->GlobalConfig(),
			"BasicWindow", "DockState");
	if (!dockStateStr) {
		on_resetUI_triggered();
J
jp9000 已提交
1649
	} else {
J
jp9000 已提交
1650 1651 1652 1653
		QByteArray dockState =
			QByteArray::fromBase64(QByteArray(dockStateStr));
		if (!restoreState(dockState))
			on_resetUI_triggered();
J
jp9000 已提交
1654 1655
	}

J
jp9000 已提交
1656
	config_set_default_bool(App()->GlobalConfig(), "BasicWindow",
J
jp9000 已提交
1657
			"DocksLocked", false);
J
jp9000 已提交
1658 1659 1660 1661 1662 1663 1664

	bool docksLocked = config_get_bool(App()->GlobalConfig(),
			"BasicWindow", "DocksLocked");
	on_lockUI_toggled(docksLocked);
	ui->lockUI->blockSignals(true);
	ui->lockUI->setChecked(docksLocked);
	ui->lockUI->blockSignals(false);
C
cg2121 已提交
1665

1666
#ifndef __APPLE__
C
cg2121 已提交
1667
	SystemTray(true);
1668
#endif
C
cg2121 已提交
1669

1670 1671 1672
	if (windowState().testFlag(Qt::WindowFullScreen))
		fullscreenInterface = true;

J
jp9000 已提交
1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688
	bool has_last_version = config_has_user_value(App()->GlobalConfig(),
			"General", "LastVersion");
	bool first_run = config_get_bool(App()->GlobalConfig(), "General",
			"FirstRun");

	if (!first_run) {
		config_set_bool(App()->GlobalConfig(), "General", "FirstRun",
				true);
		config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
	}

	if (!first_run && !has_last_version && !Active()) {
		QString msg;
		msg = QTStr("Basic.FirstStartup.RunWizard");

		QMessageBox::StandardButton button =
1689
			OBSMessageBox::question(this, QTStr("Basic.AutoConfig"),
J
jp9000 已提交
1690 1691 1692 1693 1694 1695
					msg);

		if (button == QMessageBox::Yes) {
			on_autoConfigure_triggered();
		} else {
			msg = QTStr("Basic.FirstStartup.RunWizard.NoClicked");
1696
			OBSMessageBox::information(this,
J
jp9000 已提交
1697 1698 1699
					QTStr("Basic.AutoConfig"), msg);
		}
	}
1700

1701 1702
	ToggleMixerLayout(config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"VerticalVolControl"));
S
Shaolin 已提交
1703

1704 1705
	if (config_get_bool(basicConfig, "General", "OpenStatsOnStartup"))
		on_stats_triggered();
1706 1707

	OBSBasicStats::InitializeValues();
J
jp9000 已提交
1708 1709 1710 1711 1712 1713

	/* ----------------------- */
	/* Add multiview menu      */

	ui->viewMenu->addSeparator();

1714
	multiviewProjectorMenu = new QMenu(QTStr("MultiviewProjector"));
J
jp9000 已提交
1715
	ui->viewMenu->addMenu(multiviewProjectorMenu);
1716 1717
	AddProjectorMenuMonitors(multiviewProjectorMenu, this,
			SLOT(OpenMultiviewProjector()));
1718 1719
	connect(ui->viewMenu->menuAction(), &QAction::hovered, this,
			&OBSBasic::UpdateMultiviewProjectorMenu);
J
jp9000 已提交
1720 1721
	ui->viewMenu->addAction(QTStr("MultiviewWindowed"),
			this, SLOT(OpenMultiviewWindow()));
C
cg2121 已提交
1722 1723

#if !defined(_WIN32) && !defined(__APPLE__)
J
jp9000 已提交
1724 1725 1726
	delete ui->actionShowCrashLogs;
	delete ui->actionUploadLastCrashLog;
	delete ui->menuCrashLogs;
C
cg2121 已提交
1727
	delete ui->actionCheckForUpdates;
J
jp9000 已提交
1728 1729 1730
	ui->actionShowCrashLogs = nullptr;
	ui->actionUploadLastCrashLog = nullptr;
	ui->menuCrashLogs = nullptr;
C
cg2121 已提交
1731 1732
	ui->actionCheckForUpdates = nullptr;
#endif
1733

1734
#ifdef __APPLE__
1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753
	/* This is an incredibly unpleasant hack for macOS to isolate CEF
	 * initialization until after all tasks related to Qt startup and main
	 * window initialization have completed.  There is a macOS-specific bug
	 * within either CEF and/or Qt that can cause a crash if both Qt and
	 * CEF are loading at the same time.
	 *
	 * CEF will typically load fine after about two iterations from this
	 * point, and all Qt tasks are typically fully completed after about
	 * four or five iterations, but to be "ultra" safe, an arbitrarily
	 * large number such as 10 is used.  This hack is extremely unpleasant,
	 * but is worth doing instead of being forced to isolate the entire
	 * browser plugin in to a separate process as before.
	 *
	 * Again, this hack is specific to macOS only.  Fortunately, on other
	 * operating systems, such issues do not occur. */
	QMetaObject::invokeMethod(this, "DeferredLoad",
			Qt::QueuedConnection,
			Q_ARG(QString, QT_UTF8(savePath)),
			Q_ARG(int, 10));
1754 1755
#else
	OnFirstLoad();
1756
#endif
1757 1758
}

1759 1760 1761 1762
void OBSBasic::OnFirstLoad()
{
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_FINISHED_LOADING);
J
jp9000 已提交
1763

1764
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1765 1766 1767 1768 1769 1770 1771 1772
	/* Attempt to load init screen if available */
	if (create_browser_widget) {
		WhatsNewInfoThread *wnit = new WhatsNewInfoThread();
		if (wnit) {
			connect(wnit, &WhatsNewInfoThread::Result,
					this, &OBSBasic::ReceivedIntroJson);
		}
		if (wnit) {
1773
			introCheckThread.reset(wnit);
J
jp9000 已提交
1774 1775 1776 1777
			introCheckThread->start();
		}
	}
#endif
1778 1779
}

1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
void OBSBasic::DeferredLoad(const QString &file, int requeueCount)
{
	if (--requeueCount > 0) {
		QMetaObject::invokeMethod(this, "DeferredLoad",
				Qt::QueuedConnection,
				Q_ARG(QString, file),
				Q_ARG(int, requeueCount));
		return;
	}

	Load(QT_TO_UTF8(file));
1791
	RefreshSceneCollections();
1792
	OnFirstLoad();
1793 1794 1795 1796

	/* Minimizng to tray on initial startup does not work on mac
	 * unless it is done in the deferred load */
	SystemTray(true);
1797 1798
}

J
jp9000 已提交
1799 1800 1801
/* shows a "what's new" page on startup of new versions using CEF */
void OBSBasic::ReceivedIntroJson(const QString &text)
{
1802
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817
#ifdef _WIN32
	std::string err;
	Json json = Json::parse(QT_TO_UTF8(text), err);
	if (!err.empty())
		return;

	std::string info_url;
	int info_increment = -1;

	/* check to see if there's an info page for this version */
	const Json::array &items = json.array_items();
	for (const Json &item : items) {
		const std::string &version = item["version"].string_value();
		const std::string &url = item["url"].string_value();
		int increment = item["increment"].int_value();
1818
		int rc = item["RC"].int_value();
J
jp9000 已提交
1819 1820 1821 1822 1823

		int major = 0;
		int minor = 0;

		sscanf(version.c_str(), "%d.%d", &major, &minor);
1824 1825 1826 1827 1828
#if OBS_RELEASE_CANDIDATE > 0
		if (major == OBS_RELEASE_CANDIDATE_MAJOR &&
		    minor == OBS_RELEASE_CANDIDATE_MINOR &&
		    rc == OBS_RELEASE_CANDIDATE) {
#else
J
jp9000 已提交
1829
		if (major == LIBOBS_API_MAJOR_VER &&
1830 1831 1832
		    minor == LIBOBS_API_MINOR_VER &&
		    rc == 0) {
#endif
J
jp9000 已提交
1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
			info_url = url;
			info_increment = increment;
		}
	}

	/* this version was not found, or no info for this version */
	if (info_increment == -1) {
		return;
	}

1843 1844 1845 1846
#if OBS_RELEASE_CANDIDATE > 0
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastRCVersion");
#else
J
jp9000 已提交
1847 1848
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastVersion");
1849
#endif
J
jp9000 已提交
1850 1851 1852

	int current_version_increment = -1;

1853 1854 1855
#if OBS_RELEASE_CANDIDATE > 0
	if (lastVersion < OBS_RELEASE_CANDIDATE_VER) {
#else
J
jp9000 已提交
1856
	if (lastVersion < LIBOBS_API_VER) {
1857
#endif
J
jp9000 已提交
1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874
		config_set_int(App()->GlobalConfig(), "General",
				"InfoIncrement", -1);
	} else {
		current_version_increment = config_get_int(
				App()->GlobalConfig(), "General",
				"InfoIncrement");
	}

	if (info_increment <= current_version_increment) {
		return;
	}

	config_set_int(App()->GlobalConfig(), "General",
			"InfoIncrement", info_increment);

	QDialog dlg(this);
	dlg.setWindowTitle("What's New");
1875
	dlg.resize(700, 600);
J
jp9000 已提交
1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901

	QCefWidget *cefWidget = create_browser_widget(nullptr, info_url);
	if (!cefWidget) {
		return;
	}

	connect(cefWidget, SIGNAL(titleChanged(const QString &)),
			&dlg, SLOT(setWindowTitle(const QString &)));

	QPushButton *close = new QPushButton(QTStr("Close"));
	connect(close, &QAbstractButton::clicked,
			&dlg, &QDialog::accept);

	QHBoxLayout *bottomLayout = new QHBoxLayout();
	bottomLayout->addStretch();
	bottomLayout->addWidget(close);
	bottomLayout->addStretch();

	QVBoxLayout *topLayout = new QVBoxLayout(&dlg);
	topLayout->addWidget(cefWidget);
	topLayout->addLayout(bottomLayout);

	dlg.exec();
#else
	UNUSED_PARAMETER(text);
#endif
1902 1903 1904
#else
	UNUSED_PARAMETER(text);
#endif
J
jp9000 已提交
1905 1906
}

1907 1908 1909 1910 1911 1912 1913
void OBSBasic::UpdateMultiviewProjectorMenu()
{
	multiviewProjectorMenu->clear();
	AddProjectorMenuMonitors(multiviewProjectorMenu, this,
			SLOT(OpenMultiviewProjector()));
}

P
Palana 已提交
1914 1915
void OBSBasic::InitHotkeys()
{
P
Palana 已提交
1916 1917
	ProfileScope("OBSBasic::InitHotkeys");

P
Palana 已提交
1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959
	struct obs_hotkeys_translations t = {};
	t.insert                       = Str("Hotkeys.Insert");
	t.del                          = Str("Hotkeys.Delete");
	t.home                         = Str("Hotkeys.Home");
	t.end                          = Str("Hotkeys.End");
	t.page_up                      = Str("Hotkeys.PageUp");
	t.page_down                    = Str("Hotkeys.PageDown");
	t.num_lock                     = Str("Hotkeys.NumLock");
	t.scroll_lock                  = Str("Hotkeys.ScrollLock");
	t.caps_lock                    = Str("Hotkeys.CapsLock");
	t.backspace                    = Str("Hotkeys.Backspace");
	t.tab                          = Str("Hotkeys.Tab");
	t.print                        = Str("Hotkeys.Print");
	t.pause                        = Str("Hotkeys.Pause");
	t.left                         = Str("Hotkeys.Left");
	t.right                        = Str("Hotkeys.Right");
	t.up                           = Str("Hotkeys.Up");
	t.down                         = Str("Hotkeys.Down");
#ifdef _WIN32
	t.meta                         = Str("Hotkeys.Windows");
#else
	t.meta                         = Str("Hotkeys.Super");
#endif
	t.menu                         = Str("Hotkeys.Menu");
	t.space                        = Str("Hotkeys.Space");
	t.numpad_num                   = Str("Hotkeys.NumpadNum");
	t.numpad_multiply              = Str("Hotkeys.NumpadMultiply");
	t.numpad_divide                = Str("Hotkeys.NumpadDivide");
	t.numpad_plus                  = Str("Hotkeys.NumpadAdd");
	t.numpad_minus                 = Str("Hotkeys.NumpadSubtract");
	t.numpad_decimal               = Str("Hotkeys.NumpadDecimal");
	t.apple_keypad_num             = Str("Hotkeys.AppleKeypadNum");
	t.apple_keypad_multiply        = Str("Hotkeys.AppleKeypadMultiply");
	t.apple_keypad_divide          = Str("Hotkeys.AppleKeypadDivide");
	t.apple_keypad_plus            = Str("Hotkeys.AppleKeypadAdd");
	t.apple_keypad_minus           = Str("Hotkeys.AppleKeypadSubtract");
	t.apple_keypad_decimal         = Str("Hotkeys.AppleKeypadDecimal");
	t.apple_keypad_equal           = Str("Hotkeys.AppleKeypadEqual");
	t.mouse_num                    = Str("Hotkeys.MouseButton");
	obs_hotkeys_set_translations(&t);

	obs_hotkeys_set_audio_hotkeys_translations(Str("Mute"), Str("Unmute"),
1960
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980

	obs_hotkeys_set_sceneitem_hotkeys_translations(
			Str("SceneItemShow"), Str("SceneItemHide"));

	obs_hotkey_enable_callback_rerouting(true);
	obs_hotkey_set_callback_routing_func(OBSBasic::HotkeyTriggered, this);
}

void OBSBasic::ProcessHotkey(obs_hotkey_id id, bool pressed)
{
	obs_hotkey_trigger_routed_callback(id, pressed);
}

void OBSBasic::HotkeyTriggered(void *data, obs_hotkey_id id, bool pressed)
{
	OBSBasic &basic = *static_cast<OBSBasic*>(data);
	QMetaObject::invokeMethod(&basic, "ProcessHotkey",
			Q_ARG(obs_hotkey_id, id), Q_ARG(bool, pressed));
}

1981 1982
void OBSBasic::CreateHotkeys()
{
P
Palana 已提交
1983 1984
	ProfileScope("OBSBasic::CreateHotkeys");

1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000
	auto LoadHotkeyData = [&](const char *name) -> OBSData
	{
		const char *info = config_get_string(basicConfig,
				"Hotkeys", name);
		if (!info)
			return {};

		obs_data_t *data = obs_data_create_from_json(info);
		if (!data)
			return {};

		OBSData res = data;
		obs_data_release(data);
		return res;
	};

J
jp9000 已提交
2001 2002 2003 2004 2005 2006 2007 2008 2009
	auto LoadHotkey = [&](obs_hotkey_id id, const char *name)
	{
		obs_data_array_t *array =
			obs_data_get_array(LoadHotkeyData(name), "bindings");

		obs_hotkey_load(id, array);
		obs_data_array_release(array);
	};

2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022
	auto LoadHotkeyPair = [&](obs_hotkey_pair_id id, const char *name0,
			const char *name1)
	{
		obs_data_array_t *array0 =
			obs_data_get_array(LoadHotkeyData(name0), "bindings");
		obs_data_array_t *array1 =
			obs_data_get_array(LoadHotkeyData(name1), "bindings");

		obs_hotkey_pair_load(id, array0, array1);
		obs_data_array_release(array0);
		obs_data_array_release(array1);
	};

2023
#define MAKE_CALLBACK(pred, method, log_action) \
2024 2025 2026
	[](void *data, obs_hotkey_pair_id, obs_hotkey_t*, bool pressed) \
	{ \
		OBSBasic &basic = *static_cast<OBSBasic*>(data); \
2027
		if ((pred) && pressed) { \
2028
			blog(LOG_INFO, log_action " due to hotkey"); \
2029 2030 2031 2032 2033 2034 2035 2036
			method(); \
			return true; \
		} \
		return false; \
	}

	streamingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartStreaming",
J
jp9000 已提交
2037
			Str("Basic.Main.StartStreaming"),
2038
			"OBSBasic.StopStreaming",
J
jp9000 已提交
2039
			Str("Basic.Main.StopStreaming"),
2040 2041
			MAKE_CALLBACK(!basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
2042
				basic.StartStreaming, "Starting stream"),
2043 2044
			MAKE_CALLBACK(basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
2045
				basic.StopStreaming, "Stopping stream"),
2046 2047 2048 2049
			this, this);
	LoadHotkeyPair(streamingHotkeys,
			"OBSBasic.StartStreaming", "OBSBasic.StopStreaming");

J
jp9000 已提交
2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064
	auto cb = [] (void *data, obs_hotkey_id, obs_hotkey_t*, bool pressed)
	{
		OBSBasic &basic = *static_cast<OBSBasic*>(data);
		if (basic.outputHandler->StreamingActive() && pressed) {
			basic.ForceStopStreaming();
		}
	};

	forceStreamingStopHotkey = obs_hotkey_register_frontend(
			"OBSBasic.ForceStopStreaming",
			Str("Basic.Main.ForceStopStreaming"),
			cb, this);
	LoadHotkey(forceStreamingStopHotkey,
			"OBSBasic.ForceStopStreaming");

2065 2066
	recordingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartRecording",
J
jp9000 已提交
2067
			Str("Basic.Main.StartRecording"),
2068
			"OBSBasic.StopRecording",
J
jp9000 已提交
2069
			Str("Basic.Main.StopRecording"),
2070 2071
			MAKE_CALLBACK(!basic.outputHandler->RecordingActive() &&
				!basic.ui->recordButton->isChecked(),
2072
				basic.StartRecording, "Starting recording"),
2073 2074
			MAKE_CALLBACK(basic.outputHandler->RecordingActive() &&
				basic.ui->recordButton->isChecked(),
2075
				basic.StopRecording, "Stopping recording"),
2076 2077 2078
			this, this);
	LoadHotkeyPair(recordingHotkeys,
			"OBSBasic.StartRecording", "OBSBasic.StopRecording");
J
jp9000 已提交
2079 2080 2081 2082 2083 2084 2085

	replayBufHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartReplayBuffer",
			Str("Basic.Main.StartReplayBuffer"),
			"OBSBasic.StopReplayBuffer",
			Str("Basic.Main.StopReplayBuffer"),
			MAKE_CALLBACK(!basic.outputHandler->ReplayBufferActive(),
2086
				basic.StartReplayBuffer, "Starting replay buffer"),
J
jp9000 已提交
2087
			MAKE_CALLBACK(basic.outputHandler->ReplayBufferActive(),
2088
				basic.StopReplayBuffer, "Stopping replay buffer"),
J
jp9000 已提交
2089 2090 2091
			this, this);
	LoadHotkeyPair(replayBufHotkeys,
			"OBSBasic.StartReplayBuffer", "OBSBasic.StopReplayBuffer");
2092
#undef MAKE_CALLBACK
2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121

	auto togglePreviewProgram = [] (void *data, obs_hotkey_id,
			obs_hotkey_t*, bool pressed)
	{
		if (pressed)
			QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
					"on_modeSwitch_clicked",
					Qt::QueuedConnection);
	};

	togglePreviewProgramHotkey = obs_hotkey_register_frontend(
			"OBSBasic.TogglePreviewProgram",
			Str("Basic.TogglePreviewProgramMode"),
			togglePreviewProgram, this);
	LoadHotkey(togglePreviewProgramHotkey, "OBSBasic.TogglePreviewProgram");

	auto transition = [] (void *data, obs_hotkey_id, obs_hotkey_t*,
			bool pressed)
	{
		if (pressed)
			QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
					"TransitionClicked",
					Qt::QueuedConnection);
	};

	transitionHotkey = obs_hotkey_register_frontend(
			"OBSBasic.Transition",
			Str("Transition"), transition, this);
	LoadHotkey(transitionHotkey, "OBSBasic.Transition");
2122 2123
}

J
jp9000 已提交
2124 2125 2126 2127
void OBSBasic::ClearHotkeys()
{
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
J
jp9000 已提交
2128
	obs_hotkey_pair_unregister(replayBufHotkeys);
J
jp9000 已提交
2129
	obs_hotkey_unregister(forceStreamingStopHotkey);
2130 2131
	obs_hotkey_unregister(togglePreviewProgramHotkey);
	obs_hotkey_unregister(transitionHotkey);
J
jp9000 已提交
2132 2133
}

2134 2135
OBSBasic::~OBSBasic()
{
J
jp9000 已提交
2136 2137 2138
	if (updateCheckThread && updateCheckThread->isRunning())
		updateCheckThread->wait();

2139
	delete multiviewProjectorMenu;
P
pkv 已提交
2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152
	delete previewProjector;
	delete studioProgramProjector;
	delete previewProjectorSource;
	delete previewProjectorMain;
	delete sourceProjector;
	delete sceneProjectorMenu;
	delete scaleFilteringMenu;
	delete colorMenu;
	delete colorWidgetAction;
	delete colorSelect;
	delete deinterlaceMenu;
	delete perSceneTransitionMenu;
	delete shortcutFilter;
P
pkviet 已提交
2153
	delete trayMenu;
2154 2155
	delete programOptions;
	delete program;
J
jp9000 已提交
2156

J
jp9000 已提交
2157 2158 2159 2160 2161 2162
	/* XXX: any obs data must be released before calling obs_shutdown.
	 * currently, we can't automate this with C++ RAII because of the
	 * delicate nature of obs_shutdown needing to be freed before the UI
	 * can be freed, and we have no control over the destruction order of
	 * the Qt UI stuff, so we have to manually clear any references to
	 * libobs. */
J
jp9000 已提交
2163
	delete cpuUsageTimer;
2164 2165
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
2166
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
J
jp9000 已提交
2167
	ClearHotkeys();
P
Palana 已提交
2168

2169
	service = nullptr;
J
jp9000 已提交
2170 2171
	outputHandler.reset();

J
John Bradley 已提交
2172 2173 2174
	if (interaction)
		delete interaction;

2175 2176 2177
	if (properties)
		delete properties;

J
jp9000 已提交
2178 2179 2180
	if (filters)
		delete filters;

2181 2182
	if (transformWindow)
		delete transformWindow;
2183

J
jp9000 已提交
2184 2185 2186
	if (advAudioWindow)
		delete advAudioWindow;

C
cg2121 已提交
2187 2188 2189
	if (about)
		delete about;

2190 2191 2192
	obs_display_remove_draw_callback(ui->preview->GetDisplay(),
			OBSBasic::RenderMain, this);

J
jp9000 已提交
2193
	obs_enter_graphics();
2194
	gs_vertexbuffer_destroy(box);
2195 2196 2197 2198
	gs_vertexbuffer_destroy(boxLeft);
	gs_vertexbuffer_destroy(boxTop);
	gs_vertexbuffer_destroy(boxRight);
	gs_vertexbuffer_destroy(boxBottom);
2199
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
2200
	obs_leave_graphics();
J
jp9000 已提交
2201

2202 2203 2204 2205 2206 2207 2208 2209 2210
	/* When shutting down, sometimes source references can get in to the
	 * event queue, and if we don't forcibly process those events they
	 * won't get processed until after obs_shutdown has been called.  I
	 * really wish there were a more elegant way to deal with this via C++,
	 * but Qt doesn't use C++ in a normal way, so you can't really rely on
	 * normal C++ behavior for your data to be freed in the order that you
	 * expect or want it to. */
	QApplication::sendPostedEvents(this);

J
jp9000 已提交
2211 2212
	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
2213 2214 2215 2216
#if OBS_RELEASE_CANDIDATE > 0
	config_set_int(App()->GlobalConfig(), "General", "LastRCVersion",
			OBS_RELEASE_CANDIDATE_VER);
#endif
J
jp9000 已提交
2217

2218
	bool alwaysOnTop = IsAlwaysOnTop(this);
J
jp9000 已提交
2219

J
jp9000 已提交
2220 2221
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
2222 2223
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "AlwaysOnTop",
			alwaysOnTop);
2224 2225 2226 2227 2228 2229 2230 2231
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"SceneDuplicationMode", sceneDuplicationMode);
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"SwapScenesMode", swapScenesMode);
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"EditPropertiesMode", editPropertiesMode);
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"PreviewProgramMode", IsPreviewProgramMode());
J
jp9000 已提交
2232 2233
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"DocksLocked", ui->lockUI->isChecked());
2234
	config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245

#ifdef _WIN32
	uint32_t winVer = GetWindowsVersion();
	if (winVer > 0 && winVer < 0x602) {
		bool disableAero = config_get_bool(basicConfig, "Video",
				"DisableAero");
		if (disableAero) {
			SetAeroEnabled(true);
		}
	}
#endif
2246 2247
}

J
jp9000 已提交
2248 2249 2250 2251 2252 2253 2254 2255 2256
void OBSBasic::SaveProjectNow()
{
	if (disableSaving)
		return;

	projectChanged = true;
	SaveProjectDeferred();
}

J
jp9000 已提交
2257 2258
void OBSBasic::SaveProject()
{
J
jp9000 已提交
2259 2260 2261
	if (disableSaving)
		return;

J
jp9000 已提交
2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276
	projectChanged = true;
	QMetaObject::invokeMethod(this, "SaveProjectDeferred",
			Qt::QueuedConnection);
}

void OBSBasic::SaveProjectDeferred()
{
	if (disableSaving)
		return;

	if (!projectChanged)
		return;

	projectChanged = false;

J
jp9000 已提交
2277 2278
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
2279
	char savePath[512];
J
jp9000 已提交
2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291
	char fileName[512];
	int ret;

	if (!sceneCollection)
		return;

	ret = snprintf(fileName, 512, "obs-studio/basic/scenes/%s.json",
			sceneCollection);
	if (ret <= 0)
		return;

	ret = GetConfigPath(savePath, sizeof(savePath), fileName);
2292 2293 2294
	if (ret <= 0)
		return;

J
jp9000 已提交
2295 2296 2297
	Save(savePath);
}

S
Shaolin 已提交
2298 2299 2300 2301 2302
OBSSource OBSBasic::GetProgramSource()
{
	return OBSGetStrongRef(programScene);
}

J
jp9000 已提交
2303
OBSScene OBSBasic::GetCurrentScene()
2304
{
J
jp9000 已提交
2305
	QListWidgetItem *item = ui->scenes->currentItem();
P
Palana 已提交
2306
	return item ? GetOBSRef<OBSScene>(item) : nullptr;
2307 2308
}

2309
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
2310
{
P
Palana 已提交
2311
	return item ? GetOBSRef<OBSSceneItem>(item) : nullptr;
J
jp9000 已提交
2312 2313
}

2314 2315
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
J
jp9000 已提交
2316
	return ui->sources->Get(GetTopSelectedSourceItem());
2317 2318
}

J
Joseph El-Khouri 已提交
2319 2320
void OBSBasic::UpdatePreviewScalingMenu()
{
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334
	bool fixedScaling = ui->preview->IsFixedScaling();
	float scalingAmount = ui->preview->GetScalingAmount();
	if (!fixedScaling) {
		ui->actionScaleWindow->setChecked(true);
		ui->actionScaleCanvas->setChecked(false);
		ui->actionScaleOutput->setChecked(false);
		return;
	}

	obs_video_info ovi;
	obs_get_video_info(&ovi);

	ui->actionScaleWindow->setChecked(false);
	ui->actionScaleCanvas->setChecked(scalingAmount == 1.0f);
J
Joseph El-Khouri 已提交
2335
	ui->actionScaleOutput->setChecked(
2336
			scalingAmount == float(ovi.output_width) / float(ovi.base_width));
J
Joseph El-Khouri 已提交
2337 2338
}

2339
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
2340 2341 2342 2343 2344 2345 2346 2347 2348
{
	if (interaction)
		interaction->close();

	interaction = new OBSBasicInteraction(this, source);
	interaction->Init();
	interaction->setAttribute(Qt::WA_DeleteOnClose, true);
}

2349
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
2350 2351 2352 2353 2354 2355 2356
{
	if (properties)
		properties->close();

	properties = new OBSBasicProperties(this, source);
	properties->Init();
	properties->setAttribute(Qt::WA_DeleteOnClose, true);
2357 2358
}

J
jp9000 已提交
2359 2360 2361 2362 2363 2364 2365 2366 2367 2368
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

	filters = new OBSBasicFilters(this, source);
	filters->Init();
	filters->setAttribute(Qt::WA_DeleteOnClose, true);
}

2369 2370 2371
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
2372
{
2373
	const char *name  = obs_source_get_name(source);
2374
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
2375 2376

	QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
P
Palana 已提交
2377
	SetOBSRef(item, OBSScene(scene));
J
jp9000 已提交
2378
	ui->scenes->addItem(item);
2379

P
Palana 已提交
2380 2381 2382 2383 2384
	obs_hotkey_register_source(source, "OBSBasic.SelectScene",
			Str("Basic.Hotkeys.SelectScene"),
			[](void *data,
				obs_hotkey_id, obs_hotkey_t*, bool pressed)
	{
2385 2386 2387
		OBSBasic *main =
			reinterpret_cast<OBSBasic*>(App()->GetMainWindow());

P
Palana 已提交
2388 2389 2390
		auto potential_source = static_cast<obs_source_t*>(data);
		auto source = obs_source_get_ref(potential_source);
		if (source && pressed)
2391
			main->SetCurrentScene(source);
P
Palana 已提交
2392 2393 2394
		obs_source_release(source);
	}, static_cast<obs_source_t*>(source));

2395
	signal_handler_t *handler = obs_source_get_signal_handler(source);
2396

J
jp9000 已提交
2397 2398 2399
	SignalContainer<OBSScene> container;
	container.ref = scene;
	container.handlers.assign({
2400 2401 2402 2403 2404 2405 2406 2407
		std::make_shared<OBSSignal>(handler, "item_add",
					OBSBasic::SceneItemAdded, this),
		std::make_shared<OBSSignal>(handler, "item_select",
					OBSBasic::SceneItemSelected, this),
		std::make_shared<OBSSignal>(handler, "item_deselect",
					OBSBasic::SceneItemDeselected, this),
		std::make_shared<OBSSignal>(handler, "reorder",
					OBSBasic::SceneReordered, this),
J
jp9000 已提交
2408
	});
2409 2410

	item->setData(static_cast<int>(QtDataRole::OBSSignals),
J
jp9000 已提交
2411
			QVariant::fromValue(container));
J
jp9000 已提交
2412

2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429
	/* if the scene already has items (a duplicated scene) add them */
	auto addSceneItem = [this] (obs_sceneitem_t *item)
	{
		AddSceneItem(item);
	};

	using addSceneItem_t = decltype(addSceneItem);

	obs_scene_enum_items(scene,
			[] (obs_scene_t*, obs_sceneitem_t *item, void *param)
			{
				addSceneItem_t *func;
				func = reinterpret_cast<addSceneItem_t*>(param);
				(*func)(item);
				return true;
			}, &addSceneItem);

J
jp9000 已提交
2430
	SaveProject();
2431 2432 2433 2434 2435

	if (!disableSaving) {
		obs_source_t *source = obs_scene_get_source(scene);
		blog(LOG_INFO, "User added scene '%s'",
				obs_source_get_name(source));
S
Shaolin 已提交
2436 2437

		OBSProjector::UpdateMultiviewProjectors();
2438
	}
2439 2440 2441

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2442 2443
}

2444
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
2445
{
P
Palana 已提交
2446 2447 2448 2449
	obs_scene_t *scene = obs_scene_from_source(source);

	QListWidgetItem *sel = nullptr;
	int count = ui->scenes->count();
2450

P
Palana 已提交
2451 2452 2453 2454 2455
	for (int i = 0; i < count; i++) {
		auto item = ui->scenes->item(i);
		auto cur_scene = GetOBSRef<OBSScene>(item);
		if (cur_scene != scene)
			continue;
J
jp9000 已提交
2456

P
Palana 已提交
2457 2458 2459
		sel = item;
		break;
	}
J
jp9000 已提交
2460

J
jp9000 已提交
2461
	if (sel != nullptr) {
P
Palana 已提交
2462
		if (sel == ui->scenes->currentItem())
J
jp9000 已提交
2463
			ui->sources->Clear();
J
jp9000 已提交
2464
		delete sel;
J
jp9000 已提交
2465
	}
J
jp9000 已提交
2466 2467

	SaveProject();
2468 2469 2470 2471

	if (!disableSaving) {
		blog(LOG_INFO, "User Removed scene '%s'",
				obs_source_get_name(source));
S
Shaolin 已提交
2472 2473

		OBSProjector::UpdateMultiviewProjectors();
2474
	}
2475 2476 2477

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2478 2479
}

J
jp9000 已提交
2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492
static bool select_one(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
{
	obs_sceneitem_t *selectedItem =
		reinterpret_cast<obs_sceneitem_t*>(param);
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, select_one, param);

	obs_sceneitem_select(item, (selectedItem == item));

	UNUSED_PARAMETER(scene);
	return true;
}

2493
void OBSBasic::AddSceneItem(OBSSceneItem item)
2494
{
2495
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
J
jp9000 已提交
2496

2497
	if (GetCurrentScene() == scene)
J
jp9000 已提交
2498
		ui->sources->Add(item);
J
jp9000 已提交
2499

J
jp9000 已提交
2500
	SaveProject();
2501 2502 2503 2504 2505 2506 2507 2508

	if (!disableSaving) {
		obs_source_t *sceneSource = obs_scene_get_source(scene);
		obs_source_t *itemSource = obs_sceneitem_get_source(item);
		blog(LOG_INFO, "User added source '%s' (%s) to scene '%s'",
				obs_source_get_name(itemSource),
				obs_source_get_id(itemSource),
				obs_source_get_name(sceneSource));
J
jp9000 已提交
2509 2510
		
		obs_scene_enum_items(scene, select_one, (obs_sceneitem_t*)item);
2511
	}
2512 2513
}

2514
void OBSBasic::UpdateSceneSelection(OBSSource source)
2515 2516
{
	if (source) {
2517
		obs_scene_t *scene = obs_scene_from_source(source);
2518
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
2519

2520 2521 2522
		if (!scene)
			return;

J
jp9000 已提交
2523 2524 2525
		QList<QListWidgetItem*> items =
			ui->scenes->findItems(QT_UTF8(name), Qt::MatchExactly);

2526 2527 2528 2529 2530
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

2531 2532 2533 2534
			OBSScene curScene =
				GetOBSRef<OBSScene>(ui->scenes->currentItem());
			if (api && scene != curScene)
				api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
2535
		}
J
jp9000 已提交
2536
	}
2537 2538
}

J
jp9000 已提交
2539 2540 2541 2542 2543 2544 2545 2546 2547 2548
static void RenameListValues(QListWidget *listWidget, const QString &newName,
		const QString &prevName)
{
	QList<QListWidgetItem*> items =
		listWidget->findItems(prevName, Qt::MatchExactly);

	for (int i = 0; i < items.count(); i++)
		items[i]->setText(newName);
}

S
Shaolin 已提交
2549 2550
void OBSBasic::RenameSources(OBSSource source, QString newName,
		QString prevName)
J
jp9000 已提交
2551 2552
{
	RenameListValues(ui->scenes,  newName, prevName);
2553 2554 2555 2556 2557

	for (size_t i = 0; i < volumes.size(); i++) {
		if (volumes[i]->GetName().compare(prevName) == 0)
			volumes[i]->SetName(newName);
	}
J
jp9000 已提交
2558

2559
	OBSProjector::RenameProjector(prevName, newName);
C
cg2121 已提交
2560

J
jp9000 已提交
2561
	SaveProject();
S
Shaolin 已提交
2562 2563 2564 2565

	obs_scene_t *scene = obs_scene_from_source(source);
	if (scene)
		OBSProjector::UpdateMultiviewProjectors();
J
jp9000 已提交
2566 2567
}

2568 2569
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
J
jp9000 已提交
2570 2571
	SignalBlocker sourcesSignalBlocker(ui->sources);

2572
	if (scene != GetCurrentScene() || ignoreSelectionUpdate)
2573 2574
		return;

J
jp9000 已提交
2575
	ui->sources->SelectItem(item, select);
2576 2577
}

2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593
static inline bool SourceMixerHidden(obs_source_t *source)
{
	obs_data_t *priv_settings = obs_source_get_private_settings(source);
	bool hidden = obs_data_get_bool(priv_settings, "mixer_hidden");
	obs_data_release(priv_settings);

	return hidden;
}

static inline void SetSourceMixerHidden(obs_source_t *source, bool hidden)
{
	obs_data_t *priv_settings = obs_source_get_private_settings(source);
	obs_data_set_bool(priv_settings, "mixer_hidden", hidden);
	obs_data_release(priv_settings);
}

2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611
void OBSBasic::GetAudioSourceFilters()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	VolControl *vol = action->property("volControl").value<VolControl*>();
	obs_source_t *source = vol->GetSource();

	CreateFiltersWindow(source);
}

void OBSBasic::GetAudioSourceProperties()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	VolControl *vol = action->property("volControl").value<VolControl*>();
	obs_source_t *source = vol->GetSource();

	CreatePropertiesWindow(source);
}

2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661
void OBSBasic::HideAudioControl()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	VolControl *vol = action->property("volControl").value<VolControl*>();
	obs_source_t *source = vol->GetSource();

	if (!SourceMixerHidden(source)) {
		SetSourceMixerHidden(source, true);
		DeactivateAudioSource(source);
	}
}

void OBSBasic::UnhideAllAudioControls()
{
	auto UnhideAudioMixer = [this] (obs_source_t *source) /* -- */
	{
		if (!obs_source_active(source))
			return true;
		if (!SourceMixerHidden(source))
			return true;

		SetSourceMixerHidden(source, false);
		ActivateAudioSource(source);
		return true;
	};

	using UnhideAudioMixer_t = decltype(UnhideAudioMixer);

	auto PreEnum = [] (void *data, obs_source_t *source) -> bool /* -- */
	{
		return (*reinterpret_cast<UnhideAudioMixer_t *>(data))(source);
	};

	obs_enum_sources(PreEnum, &UnhideAudioMixer);
}

void OBSBasic::ToggleHideMixer()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (!SourceMixerHidden(source)) {
		SetSourceMixerHidden(source, true);
		DeactivateAudioSource(source);
	} else {
		SetSourceMixerHidden(source, false);
		ActivateAudioSource(source);
	}
}

2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
void OBSBasic::MixerRenameSource()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	VolControl *vol = action->property("volControl").value<VolControl*>();
	OBSSource source = vol->GetSource();

	const char *prevName = obs_source_get_name(source);

	for (;;) {
		string name;
		bool accepted = NameDialog::AskForName(this,
				QTStr("Basic.Main.MixerRename.Title"),
				QTStr("Basic.Main.MixerRename.Text"),
				name,
				QT_UTF8(prevName));
		if (!accepted)
			return;

		if (name.empty()) {
			OBSMessageBox::information(this,
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
			continue;
		}

2687 2688
		OBSSource sourceTest = obs_get_source_by_name(name.c_str());
		obs_source_release(sourceTest);
2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701

		if (sourceTest) {
			OBSMessageBox::information(this,
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
			continue;
		}

		obs_source_set_name(source, name.c_str());
		break;
	}
}

2702 2703 2704 2705
void OBSBasic::VolControlContextMenu()
{
	VolControl *vol = reinterpret_cast<VolControl*>(sender());

2706 2707 2708 2709
	/* ------------------- */

	QAction hideAction(QTStr("Hide"), this);
	QAction unhideAllAction(QTStr("UnhideAll"), this);
2710
	QAction mixerRenameAction(QTStr("Rename"), this);
2711

2712 2713
	QAction filtersAction(QTStr("Filters"), this);
	QAction propertiesAction(QTStr("Properties"), this);
2714
	QAction advPropAction(QTStr("Basic.MainMenu.Edit.AdvAudio"), this);
2715

S
Shaolin 已提交
2716 2717 2718 2719 2720
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

2721 2722 2723 2724 2725 2726 2727 2728
	/* ------------------- */

	connect(&hideAction, &QAction::triggered,
			this, &OBSBasic::HideAudioControl,
			Qt::DirectConnection);
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);
2729 2730 2731
	connect(&mixerRenameAction, &QAction::triggered,
			this, &OBSBasic::MixerRenameSource,
			Qt::DirectConnection);
2732

2733 2734 2735 2736 2737 2738
	connect(&filtersAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceFilters,
			Qt::DirectConnection);
	connect(&propertiesAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceProperties,
			Qt::DirectConnection);
2739 2740 2741
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);
2742

2743 2744
	/* ------------------- */

S
Shaolin 已提交
2745 2746 2747 2748 2749 2750
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

	/* ------------------- */

2751 2752
	hideAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2753 2754
	mixerRenameAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2755

2756 2757 2758 2759 2760
	filtersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
	propertiesAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));

2761 2762
	/* ------------------- */

2763
	QMenu popup;
2764 2765
	popup.addAction(&unhideAllAction);
	popup.addAction(&hideAction);
2766
	popup.addAction(&mixerRenameAction);
2767
	popup.addSeparator();
S
Shaolin 已提交
2768 2769
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
2770 2771
	popup.addAction(&filtersAction);
	popup.addAction(&propertiesAction);
2772
	popup.addAction(&advPropAction);
2773 2774 2775
	popup.exec(QCursor::pos());
}

S
Shaolin 已提交
2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786
void OBSBasic::on_hMixerScrollArea_customContextMenuRequested()
{
	StackedMixerAreaContextMenuRequested();
}

void OBSBasic::on_vMixerScrollArea_customContextMenuRequested()
{
	StackedMixerAreaContextMenuRequested();
}

void OBSBasic::StackedMixerAreaContextMenuRequested()
2787 2788
{
	QAction unhideAllAction(QTStr("UnhideAll"), this);
S
SuslikV 已提交
2789 2790 2791

	QAction advPropAction(QTStr("Basic.MainMenu.Edit.AdvAudio"), this);

S
Shaolin 已提交
2792 2793 2794 2795 2796
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

S
SuslikV 已提交
2797 2798
	/* ------------------- */

2799 2800 2801 2802
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);

S
SuslikV 已提交
2803 2804 2805 2806 2807 2808
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);

	/* ------------------- */

S
Shaolin 已提交
2809 2810 2811 2812 2813 2814
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

	/* ------------------- */

2815
	QMenu popup;
2816
	popup.addAction(&unhideAllAction);
S
SuslikV 已提交
2817
	popup.addSeparator();
S
Shaolin 已提交
2818 2819
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
S
SuslikV 已提交
2820
	popup.addAction(&advPropAction);
2821 2822 2823
	popup.exec(QCursor::pos());
}

2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
void OBSBasic::ToggleMixerLayout(bool vertical)
{
	if (vertical) {
		ui->stackedMixerArea->setMinimumSize(180, 220);
		ui->stackedMixerArea->setCurrentIndex(1);
	} else {
		ui->stackedMixerArea->setMinimumSize(220, 0);
		ui->stackedMixerArea->setCurrentIndex(0);
	}
}

S
Shaolin 已提交
2835 2836 2837 2838 2839 2840
void OBSBasic::ToggleVolControlLayout()
{
	bool vertical = !config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	config_set_bool(GetGlobalConfig(), "BasicWindow", "VerticalVolControl",
			vertical);
2841
	ToggleMixerLayout(vertical);
S
Shaolin 已提交
2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854

	// We need to store it so we can delete current and then add
	// at the right order
	vector<OBSSource> sources;
	for (size_t i = 0; i != volumes.size(); i++)
		sources.emplace_back(volumes[i]->GetSource());

	ClearVolumeControls();

	for (const auto &source : sources)
		ActivateAudioSource(source);
}

2855 2856
void OBSBasic::ActivateAudioSource(OBSSource source)
{
2857 2858 2859
	if (SourceMixerHidden(source))
		return;

S
Shaolin 已提交
2860 2861 2862
	bool vertical = config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	VolControl *vol = new VolControl(source, true, vertical);
2863

S
Shaolin 已提交
2864 2865 2866
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");
	vol->SetMeterDecayRate(meterDecayRate);
2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885

	uint32_t peakMeterTypeIdx = config_get_uint(basicConfig, "Audio",
			"PeakMeterType");

	enum obs_peak_meter_type peakMeterType;
	switch (peakMeterTypeIdx) {
	case 0:
		peakMeterType = SAMPLE_PEAK_METER;
		break;
	case 1:
		peakMeterType = TRUE_PEAK_METER;
		break;
	default:
		peakMeterType = SAMPLE_PEAK_METER;
		break;
	}

	vol->setPeakMeterType(peakMeterType);

2886 2887 2888 2889
	vol->setContextMenuPolicy(Qt::CustomContextMenu);

	connect(vol, &QWidget::customContextMenuRequested,
			this, &OBSBasic::VolControlContextMenu);
2890 2891
	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
2892

2893 2894 2895
	InsertQObjectByName(volumes, vol);

	for (auto volume : volumes) {
S
Shaolin 已提交
2896 2897 2898 2899
		if (vertical)
			ui->vVolControlLayout->addWidget(volume);
		else
			ui->hVolControlLayout->addWidget(volume);
2900
	}
2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913
}

void OBSBasic::DeactivateAudioSource(OBSSource source)
{
	for (size_t i = 0; i < volumes.size(); i++) {
		if (volumes[i]->GetSource() == source) {
			delete volumes[i];
			volumes.erase(volumes.begin() + i);
			break;
		}
	}
}

2914
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
2915
{
J
jp9000 已提交
2916
	if (obs_source_get_type(source) == OBS_SOURCE_TYPE_SCENE &&
2917
	    !obs_source_is_group(source)) {
2918 2919 2920
		int count = ui->scenes->count();

		if (count == 1) {
2921
			OBSMessageBox::information(this,
2922 2923 2924 2925
						QTStr("FinalScene.Title"),
						QTStr("FinalScene.Text"));
			return false;
		}
2926 2927
	}

2928
	const char *name  = obs_source_get_name(source);
2929 2930 2931

	QString text = QTStr("ConfirmRemove.Text");
	text.replace("$1", QT_UTF8(name));
J
jp9000 已提交
2932

2933
	QMessageBox remove_source(this);
2934 2935 2936
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
2937 2938 2939 2940 2941 2942
	remove_source.addButton(QTStr("No"), QMessageBox::NoRole);
	remove_source.setIcon(QMessageBox::Question);
	remove_source.setWindowTitle(QTStr("ConfirmRemove.Title"));
	remove_source.exec();

	return Yes == remove_source.clickedButton();
2943
}
J
jp9000 已提交
2944

J
jp9000 已提交
2945 2946
#define UPDATE_CHECK_INTERVAL (60*60*24*4) /* 4 days */

P
Palana 已提交
2947 2948 2949 2950 2951
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
2952 2953
void OBSBasic::TimedCheckForUpdates()
{
J
jp9000 已提交
2954 2955 2956 2957
	if (!config_get_bool(App()->GlobalConfig(), "General",
				"EnableAutoUpdates"))
		return;

P
Palana 已提交
2958 2959 2960
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
2961
#elif _WIN32
J
jp9000 已提交
2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976
	long long lastUpdate = config_get_int(App()->GlobalConfig(), "General",
			"LastUpdateCheck");
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastVersion");

	if (lastVersion < LIBOBS_API_VER) {
		lastUpdate = 0;
		config_set_int(App()->GlobalConfig(), "General",
				"LastUpdateCheck", 0);
	}

	long long t    = (long long)time(nullptr);
	long long secs = t - lastUpdate;

	if (secs > UPDATE_CHECK_INTERVAL)
J
jp9000 已提交
2977
		CheckForUpdates(false);
P
Palana 已提交
2978
#endif
J
jp9000 已提交
2979 2980
}

J
jp9000 已提交
2981
void OBSBasic::CheckForUpdates(bool manualUpdate)
J
jp9000 已提交
2982
{
P
Palana 已提交
2983 2984
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
2985
#elif _WIN32
J
jp9000 已提交
2986 2987
	ui->actionCheckForUpdates->setEnabled(false);

J
jp9000 已提交
2988 2989
	if (updateCheckThread && updateCheckThread->isRunning())
		return;
2990

2991
	updateCheckThread.reset(new AutoUpdateThread(manualUpdate));
2992
	updateCheckThread->start();
P
Palana 已提交
2993
#endif
2994 2995

	UNUSED_PARAMETER(manualUpdate);
J
jp9000 已提交
2996 2997
}

J
jp9000 已提交
2998
void OBSBasic::updateCheckFinished()
J
jp9000 已提交
2999 3000 3001 3002
{
	ui->actionCheckForUpdates->setEnabled(true);
}

3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032
void OBSBasic::DuplicateSelectedScene()
{
	OBSScene curScene = GetCurrentScene();

	if (!curScene)
		return;

	OBSSource curSceneSource = obs_scene_get_source(curScene);
	QString format{obs_source_get_name(curSceneSource)};
	format += " %1";

	int i = 2;
	QString placeHolderText = format.arg(i);
	obs_source_t *source = nullptr;
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
		placeHolderText = format.arg(++i);
	}

	for (;;) {
		string name;
		bool accepted = NameDialog::AskForName(this,
				QTStr("Basic.Main.AddSceneDlg.Title"),
				QTStr("Basic.Main.AddSceneDlg.Text"),
				name,
				placeHolderText);
		if (!accepted)
			return;

		if (name.empty()) {
3033
			OBSMessageBox::information(this,
3034 3035 3036 3037 3038 3039 3040
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
			continue;
		}

		obs_source_t *source = obs_get_source_by_name(name.c_str());
		if (source) {
3041
			OBSMessageBox::information(this,
3042 3043 3044 3045 3046 3047 3048 3049
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));

			obs_source_release(source);
			continue;
		}

		obs_scene_t *scene = obs_scene_duplicate(curScene,
3050
				name.c_str(), OBS_SCENE_DUP_REFS);
3051
		source = obs_scene_get_source(scene);
3052
		SetCurrentScene(source, true);
3053
		obs_scene_release(scene);
J
jp9000 已提交
3054

3055
		break;
3056 3057 3058
	}
}

3059 3060 3061 3062
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
3063
		obs_source_t *source = obs_scene_get_source(scene);
J
jp9000 已提交
3064
		if (QueryRemoveSource(source)) {
3065
			obs_source_remove(source);
J
jp9000 已提交
3066 3067 3068 3069

			if (api)
				api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
		}
3070 3071 3072 3073 3074 3075 3076
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
3077
		obs_source_t *source = obs_sceneitem_get_source(item);
3078
		if (QueryRemoveSource(source))
J
jp9000 已提交
3079 3080 3081 3082
			obs_sceneitem_remove(item);
	}
}

3083 3084
void OBSBasic::ReorderSources(OBSScene scene)
{
3085
	if (scene != GetCurrentScene() || ui->sources->IgnoreReorder())
3086 3087
		return;

J
jp9000 已提交
3088
	ui->sources->ReorderItems();
J
jp9000 已提交
3089
	SaveProject();
3090 3091
}

3092 3093
/* OBS Callbacks */

3094 3095 3096 3097 3098 3099 3100 3101 3102 3103
void OBSBasic::SceneReordered(void *data, calldata_t *params)
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

	obs_scene_t *scene = (obs_scene_t*)calldata_ptr(params, "scene");

	QMetaObject::invokeMethod(window, "ReorderSources",
			Q_ARG(OBSScene, OBSScene(scene)));
}

3104
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
3105 3106 3107
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

3108
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
J
jp9000 已提交
3109

3110 3111
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
3112 3113
}

3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135
void OBSBasic::SceneItemSelected(void *data, calldata_t *params)
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

	obs_scene_t     *scene = (obs_scene_t*)calldata_ptr(params, "scene");
	obs_sceneitem_t *item  = (obs_sceneitem_t*)calldata_ptr(params, "item");

	QMetaObject::invokeMethod(window, "SelectSceneItem",
			Q_ARG(OBSScene, scene), Q_ARG(OBSSceneItem, item),
			Q_ARG(bool, true));
}

void OBSBasic::SceneItemDeselected(void *data, calldata_t *params)
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

	obs_scene_t     *scene = (obs_scene_t*)calldata_ptr(params, "scene");
	obs_sceneitem_t *item  = (obs_sceneitem_t*)calldata_ptr(params, "item");

	QMetaObject::invokeMethod(window, "SelectSceneItem",
			Q_ARG(OBSScene, scene), Q_ARG(OBSSceneItem, item),
			Q_ARG(bool, false));
3136

3137 3138
}

3139
void OBSBasic::SourceCreated(void *data, calldata_t *params)
3140
{
3141
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3142

3143
	if (obs_scene_from_source(source) != NULL)
3144
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
3145
				"AddScene", WaitConnection(),
3146
				Q_ARG(OBSSource, OBSSource(source)));
3147 3148
}

3149
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
3150
{
3151
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3152

3153
	if (obs_scene_from_source(source) != NULL)
3154 3155 3156
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
3157 3158
}

3159
void OBSBasic::SourceActivated(void *data, calldata_t *params)
3160
{
3161
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3162 3163 3164 3165 3166 3167 3168 3169
	uint32_t     flags  = obs_source_get_output_flags(source);

	if (flags & OBS_SOURCE_AUDIO)
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"ActivateAudioSource",
				Q_ARG(OBSSource, OBSSource(source)));
}

3170
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
3171
{
3172
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3173 3174 3175 3176 3177 3178 3179 3180
	uint32_t     flags  = obs_source_get_output_flags(source);

	if (flags & OBS_SOURCE_AUDIO)
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"DeactivateAudioSource",
				Q_ARG(OBSSource, OBSSource(source)));
}

3181
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
3182
{
S
Shaolin 已提交
3183
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
J
jp9000 已提交
3184 3185 3186 3187 3188
	const char *newName  = calldata_string(params, "new_name");
	const char *prevName = calldata_string(params, "prev_name");

	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"RenameSources",
S
Shaolin 已提交
3189
			Q_ARG(OBSSource, source),
J
jp9000 已提交
3190 3191
			Q_ARG(QString, QT_UTF8(newName)),
			Q_ARG(QString, QT_UTF8(prevName)));
3192 3193

	blog(LOG_INFO, "Source '%s' renamed to '%s'", prevName, newName);
J
jp9000 已提交
3194 3195
}

3196 3197 3198 3199 3200
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

3201
	gs_effect_t    *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
3202 3203
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
3204 3205 3206

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
3207
	gs_effect_set_vec4(color, &colorVal);
3208

3209 3210
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
3211 3212 3213 3214 3215 3216 3217 3218
	gs_matrix_push();
	gs_matrix_identity();
	gs_matrix_scale3f(float(cx), float(cy), 1.0f);

	gs_load_vertexbuffer(box);
	gs_draw(GS_TRISTRIP, 0, 0);

	gs_matrix_pop();
3219 3220
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
3221 3222 3223 3224

	gs_load_vertexbuffer(nullptr);
}

3225 3226
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
3227
	OBSBasic *window = static_cast<OBSBasic*>(data);
3228 3229 3230 3231
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
3232 3233
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
3234 3235 3236

	gs_viewport_push();
	gs_projection_push();
3237 3238 3239

	/* --------------------------------------- */

3240 3241
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
3242
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
3243
			window->previewCX, window->previewCY);
3244

3245 3246
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

3247 3248 3249 3250 3251 3252
	if (window->IsPreviewProgramMode()) {
		OBSScene scene = window->GetCurrentScene();
		obs_source_t *source = obs_scene_get_source(scene);
		if (source)
			obs_source_video_render(source);
	} else {
J
jp9000 已提交
3253
		obs_render_main_texture();
3254
	}
3255
	gs_load_vertexbuffer(nullptr);
3256

3257 3258
	/* --------------------------------------- */

3259 3260 3261
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
3262 3263 3264 3265

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
3266
	gs_reset_viewport();
J
jp9000 已提交
3267 3268 3269

	window->ui->preview->DrawSceneEditing();

3270 3271
	/* --------------------------------------- */

3272 3273
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
3274 3275 3276

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
3277 3278
}

3279 3280
/* Main class functions */

3281
obs_service_t *OBSBasic::GetService()
3282
{
3283
	if (!service) {
3284 3285
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
3286 3287
		obs_service_release(service);
	}
3288 3289 3290
	return service;
}

3291
void OBSBasic::SetService(obs_service_t *newService)
3292
{
3293
	if (newService)
3294 3295 3296
		service = newService;
}

3297
bool OBSBasic::StreamingActive() const
3298 3299 3300 3301 3302 3303
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

3304 3305 3306 3307 3308 3309 3310
bool OBSBasic::Active() const
{
	if (!outputHandler)
		return false;
	return outputHandler->Active();
}

3311 3312 3313 3314 3315 3316
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

3317 3318
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
3319
	return obs_reset_video(ovi);
3320 3321
}

3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334
static inline enum obs_scale_type GetScaleType(ConfigFile &basicConfig)
{
	const char *scaleTypeStr = config_get_string(basicConfig,
			"Video", "ScaleType");

	if (astrcmpi(scaleTypeStr, "bilinear") == 0)
		return OBS_SCALE_BILINEAR;
	else if (astrcmpi(scaleTypeStr, "lanczos") == 0)
		return OBS_SCALE_LANCZOS;
	else
		return OBS_SCALE_BICUBIC;
}

3335 3336 3337 3338 3339 3340
static inline enum video_format GetVideoFormatFromName(const char *name)
{
	if (astrcmpi(name, "I420") == 0)
		return VIDEO_FORMAT_I420;
	else if (astrcmpi(name, "NV12") == 0)
		return VIDEO_FORMAT_NV12;
J
jp9000 已提交
3341 3342
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
3343 3344 3345 3346 3347 3348 3349 3350 3351
#if 0 //currently unsupported
	else if (astrcmpi(name, "YVYU") == 0)
		return VIDEO_FORMAT_YVYU;
	else if (astrcmpi(name, "YUY2") == 0)
		return VIDEO_FORMAT_YUY2;
	else if (astrcmpi(name, "UYVY") == 0)
		return VIDEO_FORMAT_UYVY;
#endif
	else
J
jp9000 已提交
3352
		return VIDEO_FORMAT_RGBA;
3353 3354
}

3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365
void OBSBasic::ResetUI()
{
	bool studioPortraitLayout = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "StudioPortraitLayout");

	if (studioPortraitLayout)
		ui->previewLayout->setDirection(QBoxLayout::TopToBottom);
	else
		ui->previewLayout->setDirection(QBoxLayout::LeftToRight);
}

3366
int OBSBasic::ResetVideo()
J
jp9000 已提交
3367
{
3368 3369 3370
	if (outputHandler && outputHandler->Active())
		return OBS_VIDEO_CURRENTLY_ACTIVE;

P
Palana 已提交
3371 3372
	ProfileScope("OBSBasic::ResetVideo");

J
jp9000 已提交
3373
	struct obs_video_info ovi;
3374
	int ret;
J
jp9000 已提交
3375

3376
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
3377

3378 3379 3380 3381 3382 3383 3384
	const char *colorFormat = config_get_string(basicConfig, "Video",
			"ColorFormat");
	const char *colorSpace = config_get_string(basicConfig, "Video",
			"ColorSpace");
	const char *colorRange = config_get_string(basicConfig, "Video",
			"ColorRange");

J
jp9000 已提交
3385
	ovi.graphics_module = App()->GetRenderModule();
3386
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3387
			"Video", "BaseCX");
3388
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3389
			"Video", "BaseCY");
3390
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3391
			"Video", "OutputCX");
3392
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3393
			"Video", "OutputCY");
3394 3395 3396 3397 3398
	ovi.output_format  = GetVideoFormatFromName(colorFormat);
	ovi.colorspace     = astrcmpi(colorSpace, "601") == 0 ?
		VIDEO_CS_601 : VIDEO_CS_709;
	ovi.range          = astrcmpi(colorRange, "Full") == 0 ?
		VIDEO_RANGE_FULL : VIDEO_RANGE_PARTIAL;
3399 3400
	ovi.adapter        = config_get_uint(App()->GlobalConfig(),
			"Video", "AdapterIdx");
J
jp9000 已提交
3401
	ovi.gpu_conversion = true;
3402
	ovi.scale_type     = GetScaleType(basicConfig);
3403

3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419
	if (ovi.base_width == 0 || ovi.base_height == 0) {
		ovi.base_width = 1920;
		ovi.base_height = 1080;
		config_set_uint(basicConfig, "Video", "BaseCX", 1920);
		config_set_uint(basicConfig, "Video", "BaseCY", 1080);
	}

	if (ovi.output_width == 0 || ovi.output_height == 0) {
		ovi.output_width = ovi.base_width;
		ovi.output_height = ovi.base_height;
		config_set_uint(basicConfig, "Video", "OutputCX",
				ovi.base_width);
		config_set_uint(basicConfig, "Video", "OutputCY",
				ovi.base_height);
	}

3420
	ret = AttemptToResetVideo(&ovi);
3421
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
3422 3423 3424 3425 3426 3427
		if (ret == OBS_VIDEO_CURRENTLY_ACTIVE) {
			blog(LOG_WARNING, "Tried to reset when "
			                  "already active");
			return ret;
		}

3428
		/* Try OpenGL if DirectX fails on windows */
3429
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
3430 3431 3432 3433
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
3434 3435
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
3436 3437
			ret = AttemptToResetVideo(&ovi);
		}
3438 3439
	} else if (ret == OBS_VIDEO_SUCCESS) {
		ResizePreview(ovi.base_width, ovi.base_height);
3440 3441
		if (program)
			ResizeProgram(ovi.base_width, ovi.base_height);
3442 3443
	}

3444
	if (ret == OBS_VIDEO_SUCCESS) {
3445
		OBSBasicStats::InitializeValues();
3446 3447
		OBSProjector::UpdateMultiviewProjectors();
	}
3448

3449
	return ret;
J
jp9000 已提交
3450
}
J
jp9000 已提交
3451

3452
bool OBSBasic::ResetAudio()
J
jp9000 已提交
3453
{
P
Palana 已提交
3454 3455
	ProfileScope("OBSBasic::ResetAudio");

3456
	struct obs_audio_info ai;
3457
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
3458 3459
			"SampleRate");

3460
	const char *channelSetupStr = config_get_string(basicConfig,
3461 3462 3463 3464
			"Audio", "ChannelSetup");

	if (strcmp(channelSetupStr, "Mono") == 0)
		ai.speakers = SPEAKERS_MONO;
P
pkviet 已提交
3465 3466
	else if (strcmp(channelSetupStr, "2.1") == 0)
		ai.speakers = SPEAKERS_2POINT1;
P
pkviet 已提交
3467 3468
	else if (strcmp(channelSetupStr, "4.0") == 0)
		ai.speakers = SPEAKERS_4POINT0;
P
pkviet 已提交
3469 3470 3471 3472 3473 3474
	else if (strcmp(channelSetupStr, "4.1") == 0)
		ai.speakers = SPEAKERS_4POINT1;
	else if (strcmp(channelSetupStr, "5.1") == 0)
		ai.speakers = SPEAKERS_5POINT1;
	else if (strcmp(channelSetupStr, "7.1") == 0)
		ai.speakers = SPEAKERS_7POINT1;
3475 3476 3477
	else
		ai.speakers = SPEAKERS_STEREO;

J
jp9000 已提交
3478
	return obs_reset_audio(&ai);
J
jp9000 已提交
3479 3480
}

3481
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceId,
3482
		const char *deviceDesc, int channel)
J
jp9000 已提交
3483
{
3484
	bool disable = deviceId && strcmp(deviceId, "disabled") == 0;
3485 3486
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
3487 3488 3489

	source = obs_get_output_source(channel);
	if (source) {
3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502
		if (disable) {
			obs_set_output_source(channel, nullptr);
		} else {
			settings = obs_source_get_settings(source);
			const char *oldId = obs_data_get_string(settings,
					"device_id");
			if (strcmp(oldId, deviceId) != 0) {
				obs_data_set_string(settings, "device_id",
						deviceId);
				obs_source_update(source, settings);
			}
			obs_data_release(settings);
		}
J
jp9000 已提交
3503 3504 3505

		obs_source_release(source);

3506 3507
	} else if (!disable) {
		settings = obs_data_create();
J
jp9000 已提交
3508
		obs_data_set_string(settings, "device_id", deviceId);
3509 3510
		source = obs_source_create(sourceId, deviceDesc, settings,
				nullptr);
J
jp9000 已提交
3511 3512 3513 3514 3515 3516 3517
		obs_data_release(settings);

		obs_set_output_source(channel, source);
		obs_source_release(source);
	}
}

J
jp9000 已提交
3518
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
3519
{
3520
	QSize  targetSize;
3521
	bool isFixedScaling;
J
Joseph El-Khouri 已提交
3522
	obs_video_info ovi;
J
jp9000 已提交
3523

3524
	/* resize preview panel to fix to the top section of the window */
3525
	targetSize = GetPixelSize(ui->preview);
J
Joseph El-Khouri 已提交
3526

3527
	isFixedScaling = ui->preview->IsFixedScaling();
J
Joseph El-Khouri 已提交
3528 3529
	obs_get_video_info(&ovi);

3530 3531
	if (isFixedScaling) {
		previewScale = ui->preview->GetScalingAmount();
J
Joseph El-Khouri 已提交
3532 3533 3534 3535
		GetCenterPosFromFixedScale(int(cx), int(cy),
				targetSize.width() - PREVIEW_EDGE_SIZE * 2,
				targetSize.height() - PREVIEW_EDGE_SIZE * 2,
				previewX, previewY, previewScale);
3536 3537
		previewX += ui->preview->GetScrollX();
		previewY += ui->preview->GetScrollY();
J
Joseph El-Khouri 已提交
3538 3539 3540 3541 3542 3543 3544

	} else {
		GetScaleAndCenterPos(int(cx), int(cy),
				targetSize.width() - PREVIEW_EDGE_SIZE * 2,
				targetSize.height() - PREVIEW_EDGE_SIZE * 2,
				previewX, previewY, previewScale);
	}
J
jp9000 已提交
3545

3546 3547
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);
J
jp9000 已提交
3548 3549
}

3550 3551 3552 3553 3554 3555 3556 3557 3558
void OBSBasic::CloseDialogs()
{
	QList<QDialog*> childDialogs = this->findChildren<QDialog *>();
	if (!childDialogs.isEmpty()) {
		for (int i = 0; i < childDialogs.size(); ++i) {
			childDialogs.at(i)->close();
		}
	}

C
cg2121 已提交
3559 3560 3561 3562
	for (QPointer<QWidget> &projector : windowProjectors) {
		delete projector;
		projector.clear();
	}
3563 3564 3565 3566
	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}
J
jp9000 已提交
3567

3568
	if (!stats.isNull()) stats->close(); //call close to save Stats geometry
3569
	if (!remux.isNull()) remux->close();
3570 3571
}

3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594
void OBSBasic::EnumDialogs()
{
	visDialogs.clear();
	modalDialogs.clear();
	visMsgBoxes.clear();

	/* fill list of Visible dialogs and Modal dialogs */
	QList<QDialog*> dialogs = findChildren<QDialog*>();
	for (QDialog *dialog : dialogs) {
		if (dialog->isVisible())
			visDialogs.append(dialog);
		if (dialog->isModal())
			modalDialogs.append(dialog);
	}

	/* fill list of Visible message boxes */
	QList<QMessageBox*> msgBoxes = findChildren<QMessageBox*>();
	for (QMessageBox *msgbox : msgBoxes) {
		if (msgbox->isVisible())
			visMsgBoxes.append(msgbox);
	}
}

3595 3596
void OBSBasic::ClearSceneData()
{
J
jp9000 已提交
3597 3598
	disableSaving++;

3599 3600 3601 3602
	CloseDialogs();

	ClearVolumeControls();
	ClearListItems(ui->scenes);
J
jp9000 已提交
3603
	ui->sources->Clear();
3604 3605
	ClearQuickTransitions();
	ui->transitions->clear();
3606 3607 3608 3609 3610 3611 3612

	obs_set_output_source(0, nullptr);
	obs_set_output_source(1, nullptr);
	obs_set_output_source(2, nullptr);
	obs_set_output_source(3, nullptr);
	obs_set_output_source(4, nullptr);
	obs_set_output_source(5, nullptr);
3613 3614 3615
	lastScene = nullptr;
	swapScene = nullptr;
	programScene = nullptr;
3616 3617 3618 3619 3620 3621 3622 3623 3624 3625

	auto cb = [](void *unused, obs_source_t *source)
	{
		obs_source_remove(source);
		UNUSED_PARAMETER(unused);
		return true;
	};

	obs_enum_sources(cb, nullptr);

3626 3627 3628
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CLEANUP);

J
jp9000 已提交
3629
	disableSaving--;
3630 3631 3632

	blog(LOG_INFO, "All scene data cleared");
	blog(LOG_INFO, "------------------------------------------------");
3633 3634
}

J
jp9000 已提交
3635
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
3636
{
3637 3638 3639 3640 3641
	if (isVisible())
		config_set_string(App()->GlobalConfig(),
				"BasicWindow", "geometry",
				saveGeometry().toBase64().constData());

J
jp9000 已提交
3642 3643 3644 3645
	config_set_string(App()->GlobalConfig(),
			"BasicWindow", "DockState",
			saveState().toBase64().constData());

3646
	if (outputHandler && outputHandler->Active()) {
C
cg2121 已提交
3647 3648
		SetShowing(true);

3649
		QMessageBox::StandardButton button = OBSMessageBox::question(
3650 3651 3652 3653 3654 3655 3656 3657 3658
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

		if (button == QMessageBox::No) {
			event->ignore();
			return;
		}
	}

3659 3660 3661 3662
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

3663 3664
	blog(LOG_INFO, SHUTDOWN_SEPARATOR);

J
jp9000 已提交
3665 3666
	if (introCheckThread)
		introCheckThread->wait();
3667 3668 3669 3670 3671
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
3672 3673
	signalHandlers.clear();

J
jp9000 已提交
3674
	SaveProjectNow();
J
jp9000 已提交
3675 3676 3677 3678

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_EXIT);

J
jp9000 已提交
3679
	disableSaving++;
J
jp9000 已提交
3680

3681 3682 3683
	/* Clear all scene data (dialogs, widgets, widget sub-items, scenes,
	 * sources, etc) so that all references are released before shutdown */
	ClearSceneData();
3684 3685

	App()->quit();
3686 3687
}

J
jp9000 已提交
3688
void OBSBasic::changeEvent(QEvent *event)
3689
{
3690 3691
	if (event->type() == QEvent::WindowStateChange &&
	    isMinimized() &&
3692
	    trayIcon &&
3693 3694 3695 3696 3697
	    trayIcon->isVisible() &&
	    sysTrayMinimizeToTray()) {

		ToggleShowHide();
	}
3698 3699
}

3700 3701
void OBSBasic::on_actionShow_Recordings_triggered()
{
3702
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
P
pkviet 已提交
3703 3704 3705 3706
	const char *type = config_get_string(basicConfig, "AdvOut", "RecType");
	const char *adv_path = strcmp(type, "Standard") ?
			config_get_string(basicConfig, "AdvOut", "FFFilePath") :
			config_get_string(basicConfig, "AdvOut", "RecFilePath");
3707
	const char *path = strcmp(mode, "Advanced") ?
P
pkviet 已提交
3708 3709
			config_get_string(basicConfig, "SimpleOutput", "FilePath") :
			adv_path;
3710 3711 3712
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
3713 3714
void OBSBasic::on_actionRemux_triggered()
{
3715 3716 3717 3718 3719 3720
	if (!remux.isNull()) {
		remux->show();
		remux->raise();
		return;
	}

3721 3722 3723 3724
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	const char *path = strcmp(mode, "Advanced") ?
		config_get_string(basicConfig, "SimpleOutput", "FilePath") :
		config_get_string(basicConfig, "AdvOut", "RecFilePath");
3725 3726 3727 3728 3729

	OBSRemux *remuxDlg;
	remuxDlg = new OBSRemux(path, this);
	remuxDlg->show();
	remux = remuxDlg;
P
Palana 已提交
3730 3731
}

P
Palana 已提交
3732 3733 3734 3735
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
C
cg2121 已提交
3736
	SystemTray(false);
P
Palana 已提交
3737 3738
}

J
jp9000 已提交
3739 3740
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
3741 3742 3743 3744 3745
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
3746 3747 3748
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
3749 3750 3751

	connect(advAudioWindow, SIGNAL(destroyed()),
		this, SLOT(on_advAudioProps_destroyed()));
J
jp9000 已提交
3752 3753
}

3754 3755 3756 3757 3758
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

3759 3760 3761 3762 3763
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

3764 3765
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
3766
{
3767
	obs_source_t *source = NULL;
J
jp9000 已提交
3768

3769 3770 3771 3772
	if (sceneChanging)
		return;

	if (current) {
3773
		obs_scene_t *scene;
J
jp9000 已提交
3774

P
Palana 已提交
3775
		scene = GetOBSRef<OBSScene>(current);
3776
		source = obs_scene_get_source(scene);
3777 3778
	}

3779
	SetCurrentScene(source);
3780

3781 3782 3783
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);

3784
	UNUSED_PARAMETER(prev);
3785 3786
}

J
jp9000 已提交
3787 3788
void OBSBasic::EditSceneName()
{
3789 3790 3791 3792 3793 3794
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

	item->setFlags(flags | Qt::ItemIsEditable);
	ui->scenes->editItem(item);
	item->setFlags(flags);
J
jp9000 已提交
3795 3796
}

J
jp9000 已提交
3797 3798 3799 3800
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot)
{
	QAction *action;
3801 3802
	QList<QScreen*> screens = QGuiApplication::screens();
	for (int i = 0; i < screens.size(); i++) {
3803
		QRect screenGeometry = screens[i]->geometry();
J
jp9000 已提交
3804 3805 3806
		QString str = QString("%1 %2: %3x%4 @ %5,%6").
			arg(QTStr("Display"),
			    QString::number(i),
S
Shaolin 已提交
3807 3808 3809 3810
			    QString::number(screenGeometry.width()),
			    QString::number(screenGeometry.height()),
			    QString::number(screenGeometry.x()),
			    QString::number(screenGeometry.y()));
J
jp9000 已提交
3811 3812 3813 3814 3815 3816

		action = parent->addAction(str, target, slot);
		action->setProperty("monitor", i);
	}
}

J
jp9000 已提交
3817
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
3818
{
J
jp9000 已提交
3819 3820
	QListWidgetItem *item = ui->scenes->itemAt(pos);

3821
	QMenu popup(this);
J
jp9000 已提交
3822
	QMenu order(QTStr("Basic.MainMenu.Edit.Order"), this);
J
jp9000 已提交
3823 3824 3825
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
3826 3827
	if (item) {
		popup.addSeparator();
3828 3829
		popup.addAction(QTStr("Duplicate"),
				this, SLOT(DuplicateSelectedScene()));
P
Palana 已提交
3830 3831
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
3832
		popup.addAction(QTStr("Remove"),
3833
				this, SLOT(RemoveSelectedScene()));
J
jp9000 已提交
3834
		popup.addSeparator();
J
jp9000 已提交
3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847

		order.addAction(QTStr("Basic.MainMenu.Edit.Order.MoveUp"),
				this, SLOT(on_actionSceneUp_triggered()));
		order.addAction(QTStr("Basic.MainMenu.Edit.Order.MoveDown"),
				this, SLOT(on_actionSceneDown_triggered()));
		order.addSeparator();
		order.addAction(QTStr("Basic.MainMenu.Edit.Order.MoveToTop"),
				this, SLOT(MoveSceneToTop()));
		order.addAction(QTStr("Basic.MainMenu.Edit.Order.MoveToBottom"),
				this, SLOT(MoveSceneToBottom()));
		popup.addMenu(&order);

		popup.addSeparator();
S
Shaolin 已提交
3848

P
pkv 已提交
3849
		delete sceneProjectorMenu;
J
jp9000 已提交
3850 3851 3852 3853
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
C
cg2121 已提交
3854 3855 3856 3857 3858 3859

		QAction *sceneWindow = popup.addAction(
				QTStr("SceneWindow"),
				this, SLOT(OpenSceneWindow()));

		popup.addAction(sceneWindow);
J
jp9000 已提交
3860
		popup.addSeparator();
J
jp9000 已提交
3861 3862
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
3863 3864 3865

		popup.addSeparator();

P
pkv 已提交
3866 3867 3868
		delete perSceneTransitionMenu;
		perSceneTransitionMenu = CreatePerSceneTransitionMenu();
		popup.addMenu(perSceneTransitionMenu);
S
Shaolin 已提交
3869 3870 3871

		/* ---------------------- */

J
jp9000 已提交
3872 3873
		QAction *multiviewAction = popup.addAction(
				QTStr("ShowInMultiview"));
S
Shaolin 已提交
3874

J
jp9000 已提交
3875 3876 3877
		OBSSource source = GetCurrentSceneSource();
		OBSData data = obs_source_get_private_settings(source);
		obs_data_release(data);
S
Shaolin 已提交
3878

J
jp9000 已提交
3879 3880 3881
		obs_data_set_default_bool(data, "show_in_multiview",
				true);
		bool show = obs_data_get_bool(data, "show_in_multiview");
S
Shaolin 已提交
3882

J
jp9000 已提交
3883 3884
		multiviewAction->setCheckable(true);
		multiviewAction->setChecked(show);
S
Shaolin 已提交
3885

3886
		auto showInMultiview = [] (OBSData data)
J
jp9000 已提交
3887 3888 3889 3890 3891 3892 3893 3894 3895 3896
		{
			bool show = obs_data_get_bool(data,
					"show_in_multiview");
			obs_data_set_bool(data, "show_in_multiview",
					!show);
			OBSProjector::UpdateMultiviewProjectors();
		};

		connect(multiviewAction, &QAction::triggered,
				std::bind(showInMultiview, data));
P
Palana 已提交
3897
	}
J
jp9000 已提交
3898 3899

	popup.exec(QCursor::pos());
3900 3901
}

J
jp9000 已提交
3902
void OBSBasic::on_actionAddScene_triggered()
3903
{
3904
	string name;
S
Socapex 已提交
3905
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
3906

3907
	int i = 2;
P
Palana 已提交
3908
	QString placeHolderText = format.arg(i);
3909
	obs_source_t *source = nullptr;
P
Palana 已提交
3910 3911
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
3912
		placeHolderText = format.arg(++i);
P
Palana 已提交
3913
	}
S
Socapex 已提交
3914

J
jp9000 已提交
3915
	bool accepted = NameDialog::AskForName(this,
3916 3917
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
3918 3919
			name,
			placeHolderText);
3920

J
jp9000 已提交
3921
	if (accepted) {
J
jp9000 已提交
3922
		if (name.empty()) {
3923
			OBSMessageBox::information(this,
3924 3925
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
3926 3927 3928 3929
			on_actionAddScene_triggered();
			return;
		}

3930
		obs_source_t *source = obs_get_source_by_name(name.c_str());
3931
		if (source) {
3932
			OBSMessageBox::information(this,
3933 3934
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
3935 3936

			obs_source_release(source);
J
jp9000 已提交
3937
			on_actionAddScene_triggered();
3938 3939 3940
			return;
		}

3941
		obs_scene_t *scene = obs_scene_create(name.c_str());
3942
		source = obs_scene_get_source(scene);
3943
		SetCurrentScene(source);
3944
		obs_scene_release(scene);
3945
	}
3946 3947
}

J
jp9000 已提交
3948
void OBSBasic::on_actionRemoveScene_triggered()
3949
{
3950
	OBSScene     scene  = GetCurrentScene();
3951
	obs_source_t *source = obs_scene_get_source(scene);
3952 3953 3954

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
3955 3956
}

J
jp9000 已提交
3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974
void OBSBasic::ChangeSceneIndex(bool relative, int offset, int invalidIdx)
{
	int idx = ui->scenes->currentRow();
	if (idx == -1 || idx == invalidIdx)
		return;

	sceneChanging = true;

	QListWidgetItem *item = ui->scenes->takeItem(idx);

	if (!relative)
		idx = 0;

	ui->scenes->insertItem(idx + offset, item);
	ui->scenes->setCurrentRow(idx + offset);
	item->setSelected(true);

	sceneChanging = false;
3975 3976

	OBSProjector::UpdateMultiviewProjectors();
J
jp9000 已提交
3977 3978
}

J
jp9000 已提交
3979
void OBSBasic::on_actionSceneUp_triggered()
3980
{
J
jp9000 已提交
3981
	ChangeSceneIndex(true, -1, 0);
3982 3983
}

J
jp9000 已提交
3984
void OBSBasic::on_actionSceneDown_triggered()
3985
{
J
jp9000 已提交
3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997
	ChangeSceneIndex(true, 1, ui->scenes->count() - 1);
}

void OBSBasic::MoveSceneToTop()
{
	ChangeSceneIndex(false, 0, 0);
}

void OBSBasic::MoveSceneToBottom()
{
	ChangeSceneIndex(false, ui->scenes->count() - 1,
			ui->scenes->count() - 1);
3998 3999
}

J
jp9000 已提交
4000 4001
void OBSBasic::EditSceneItemName()
{
J
jp9000 已提交
4002 4003
	int idx = GetTopSelectedSourceItem();
	ui->sources->Edit(idx);
J
jp9000 已提交
4004 4005
}

J
jp9000 已提交
4006 4007 4008 4009 4010 4011 4012 4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027
void OBSBasic::SetDeinterlacingMode()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	obs_deinterlace_mode mode =
		(obs_deinterlace_mode)action->property("mode").toInt();
	OBSSceneItem sceneItem = GetCurrentSceneItem();
	obs_source_t *source = obs_sceneitem_get_source(sceneItem);

	obs_source_set_deinterlace_mode(source, mode);
}

void OBSBasic::SetDeinterlacingOrder()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	obs_deinterlace_field_order order =
		(obs_deinterlace_field_order)action->property("order").toInt();
	OBSSceneItem sceneItem = GetCurrentSceneItem();
	obs_source_t *source = obs_sceneitem_get_source(sceneItem);

	obs_source_set_deinterlace_field_order(source, order);
}

P
pkv 已提交
4028
QMenu *OBSBasic::AddDeinterlacingMenu(QMenu *menu, obs_source_t *source)
J
jp9000 已提交
4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069
{
	obs_deinterlace_mode deinterlaceMode =
		obs_source_get_deinterlace_mode(source);
	obs_deinterlace_field_order deinterlaceOrder =
		obs_source_get_deinterlace_field_order(source);
	QAction *action;

#define ADD_MODE(name, mode) \
	action = menu->addAction(QTStr("" name), this, \
				SLOT(SetDeinterlacingMode())); \
	action->setProperty("mode", (int)mode); \
	action->setCheckable(true); \
	action->setChecked(deinterlaceMode == mode);

	ADD_MODE("Disable",                OBS_DEINTERLACE_MODE_DISABLE);
	ADD_MODE("Deinterlacing.Discard",  OBS_DEINTERLACE_MODE_DISCARD);
	ADD_MODE("Deinterlacing.Retro",    OBS_DEINTERLACE_MODE_RETRO);
	ADD_MODE("Deinterlacing.Blend",    OBS_DEINTERLACE_MODE_BLEND);
	ADD_MODE("Deinterlacing.Blend2x",  OBS_DEINTERLACE_MODE_BLEND_2X);
	ADD_MODE("Deinterlacing.Linear",   OBS_DEINTERLACE_MODE_LINEAR);
	ADD_MODE("Deinterlacing.Linear2x", OBS_DEINTERLACE_MODE_LINEAR_2X);
	ADD_MODE("Deinterlacing.Yadif",    OBS_DEINTERLACE_MODE_YADIF);
	ADD_MODE("Deinterlacing.Yadif2x",  OBS_DEINTERLACE_MODE_YADIF_2X);
#undef ADD_MODE

	menu->addSeparator();

#define ADD_ORDER(name, order) \
	action = menu->addAction(QTStr("Deinterlacing." name), this, \
				SLOT(SetDeinterlacingOrder())); \
	action->setProperty("order", (int)order); \
	action->setCheckable(true); \
	action->setChecked(deinterlaceOrder == order);

	ADD_ORDER("TopFieldFirst",    OBS_DEINTERLACE_FIELD_ORDER_TOP);
	ADD_ORDER("BottomFieldFirst", OBS_DEINTERLACE_FIELD_ORDER_BOTTOM);
#undef ADD_ORDER

	return menu;
}

4070 4071 4072 4073 4074 4075 4076 4077 4078
void OBSBasic::SetScaleFilter()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	obs_scale_type mode = (obs_scale_type)action->property("mode").toInt();
	OBSSceneItem sceneItem = GetCurrentSceneItem();

	obs_sceneitem_set_scale_filter(sceneItem, mode);
}

P
pkv 已提交
4079
QMenu *OBSBasic::AddScaleFilteringMenu(QMenu *menu, obs_sceneitem_t *item)
4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100
{
	obs_scale_type scaleFilter = obs_sceneitem_get_scale_filter(item);
	QAction *action;

#define ADD_MODE(name, mode) \
	action = menu->addAction(QTStr("" name), this, \
				SLOT(SetScaleFilter())); \
	action->setProperty("mode", (int)mode); \
	action->setCheckable(true); \
	action->setChecked(scaleFilter == mode);

	ADD_MODE("Disable",                 OBS_SCALE_DISABLE);
	ADD_MODE("ScaleFiltering.Point",    OBS_SCALE_POINT);
	ADD_MODE("ScaleFiltering.Bilinear", OBS_SCALE_BILINEAR);
	ADD_MODE("ScaleFiltering.Bicubic",  OBS_SCALE_BICUBIC);
	ADD_MODE("ScaleFiltering.Lanczos",  OBS_SCALE_LANCZOS);
#undef ADD_MODE

	return menu;
}

P
pkv 已提交
4101 4102
QMenu *OBSBasic::AddBackgroundColorMenu(QMenu *menu, QWidgetAction *widgetAction,
		ColorSelect *select, obs_sceneitem_t *item)
4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155
{
	QAction *action;

	menu->setStyleSheet(QString(
		"*[bgColor=\"1\"]{background-color:rgba(255,68,68,33%);}" \
		"*[bgColor=\"2\"]{background-color:rgba(255,255,68,33%);}" \
		"*[bgColor=\"3\"]{background-color:rgba(68,255,68,33%);}" \
		"*[bgColor=\"4\"]{background-color:rgba(68,255,255,33%);}" \
		"*[bgColor=\"5\"]{background-color:rgba(68,68,255,33%);}" \
		"*[bgColor=\"6\"]{background-color:rgba(255,68,255,33%);}" \
		"*[bgColor=\"7\"]{background-color:rgba(68,68,68,33%);}" \
		"*[bgColor=\"8\"]{background-color:rgba(255,255,255,33%);}"));

	obs_data_t *privData = obs_sceneitem_get_private_settings(item);
	obs_data_release(privData);

	obs_data_set_default_int(privData, "color-preset", 0);
	int preset = obs_data_get_int(privData, "color-preset");

	action = menu->addAction(QTStr("Clear"), this,
		+ SLOT(ColorChange()));
	action->setCheckable(true);
	action->setProperty("bgColor", 0);
	action->setChecked(preset == 0);

	action = menu->addAction(QTStr("CustomColor"), this,
		+ SLOT(ColorChange()));
	action->setCheckable(true);
	action->setProperty("bgColor", 1);
	action->setChecked(preset == 1);

	menu->addSeparator();

	widgetAction->setDefaultWidget(select);

	for (int i = 1; i < 9; i++) {
		stringstream button;
		button << "preset" << i;
		QPushButton *colorButton = select->findChild<QPushButton *>(
				button.str().c_str());
		if (preset == i + 1)
			colorButton->setStyleSheet("border: 2px solid black");

		colorButton->setProperty("bgColor", i);
		select->connect(colorButton, SIGNAL(released()), this,
				SLOT(ColorChange()));
	}

	menu->addAction(widgetAction);

	return menu;
}

P
pkv 已提交
4156 4157 4158 4159 4160 4161 4162
ColorSelect::ColorSelect(QWidget *parent)
	: QWidget(parent),
	ui(new Ui::ColorSelect)
{
	ui->setupUi(this);
}

J
jp9000 已提交
4163
void OBSBasic::CreateSourcePopupMenu(int idx, bool preview)
4164
{
4165
	QMenu popup(this);
P
pkv 已提交
4166 4167 4168 4169 4170 4171 4172
	delete previewProjectorSource;
	delete sourceProjector;
	delete scaleFilteringMenu;
	delete colorMenu;
	delete colorWidgetAction;
	delete colorSelect;
	delete deinterlaceMenu;
J
jp9000 已提交
4173 4174 4175 4176 4177 4178

	if (preview) {
		QAction *action = popup.addAction(
				QTStr("Basic.Main.PreviewConextMenu.Enable"),
				this, SLOT(TogglePreview()));
		action->setCheckable(true);
4179 4180
		action->setChecked(
				obs_display_enabled(ui->preview->GetDisplay()));
4181 4182
		if (IsPreviewProgramMode())
			action->setEnabled(false);
J
jp9000 已提交
4183

J
Joseph El-Khouri 已提交
4184 4185
		popup.addAction(ui->actionLockPreview);
		popup.addMenu(ui->scalingMenu);
J
jp9000 已提交
4186

P
pkv 已提交
4187 4188
		previewProjectorSource = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjectorSource, this,
4189
				SLOT(OpenPreviewProjector()));
J
jp9000 已提交
4190

P
pkv 已提交
4191
		popup.addMenu(previewProjectorSource);
J
jp9000 已提交
4192

C
cg2121 已提交
4193 4194
		QAction *previewWindow = popup.addAction(
				QTStr("PreviewWindow"),
4195
				this, SLOT(OpenPreviewWindow()));
C
cg2121 已提交
4196 4197 4198

		popup.addAction(previewWindow);

J
jp9000 已提交
4199 4200 4201
		popup.addSeparator();
	}

J
jp9000 已提交
4202 4203 4204 4205
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

4206
	ui->actionCopyFilters->setEnabled(false);
4207
	ui->actionCopySource->setEnabled(false);
4208

J
jp9000 已提交
4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219
	if (ui->sources->MultipleBaseSelected()) {
		popup.addSeparator();
		popup.addAction(QTStr("Basic.Main.GroupItems"),
				ui->sources, SLOT(GroupSelectedItems()));

	} else if (ui->sources->GroupsSelected()) {
		popup.addSeparator();
		popup.addAction(QTStr("Basic.Main.Ungroup"),
				ui->sources, SLOT(UngroupSelectedGroups()));
	}

4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230
	popup.addSeparator();
	popup.addAction(ui->actionCopySource);
	popup.addAction(ui->actionPasteRef);
	popup.addAction(ui->actionPasteDup);
	popup.addSeparator();

	popup.addSeparator();
	popup.addAction(ui->actionCopyFilters);
	popup.addAction(ui->actionPasteFilters);
	popup.addSeparator();

J
jp9000 已提交
4231
	if (idx != -1) {
J
jp9000 已提交
4232 4233 4234
		if (addSourceMenu)
			popup.addSeparator();

J
jp9000 已提交
4235
		OBSSceneItem sceneItem = ui->sources->Get(idx);
4236
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
jp9000 已提交
4237 4238 4239
		uint32_t flags = obs_source_get_output_flags(source);
		bool isAsyncVideo = (flags & OBS_SOURCE_ASYNC_VIDEO) ==
			OBS_SOURCE_ASYNC_VIDEO;
4240 4241
		bool hasAudio = (flags & OBS_SOURCE_AUDIO) ==
			OBS_SOURCE_AUDIO;
J
John Bradley 已提交
4242 4243
		QAction *action;

P
pkv 已提交
4244 4245 4246 4247 4248
		colorMenu = new QMenu(QTStr("ChangeBG"));
		colorWidgetAction = new QWidgetAction(colorMenu);
		colorSelect = new ColorSelect(colorMenu);
		popup.addMenu(AddBackgroundColorMenu(colorMenu,
				colorWidgetAction, colorSelect, sceneItem));
J
jp9000 已提交
4249 4250
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
4251
		popup.addAction(QTStr("Remove"), this,
4252
				SLOT(on_actionRemoveSource_triggered()));
J
jp9000 已提交
4253 4254
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
4255
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
4256 4257 4258 4259 4260

		sourceProjector = new QMenu(QTStr("SourceProjector"));
		AddProjectorMenuMonitors(sourceProjector, this,
				SLOT(OpenSourceProjector()));

C
cg2121 已提交
4261 4262 4263 4264 4265 4266
		QAction *sourceWindow = popup.addAction(
				QTStr("SourceWindow"),
				this, SLOT(OpenSourceWindow()));

		popup.addAction(sourceWindow);

J
jp9000 已提交
4267
		popup.addSeparator();
4268 4269 4270 4271 4272 4273 4274 4275 4276

		if (hasAudio) {
			QAction *actionHideMixer = popup.addAction(
					QTStr("HideMixer"),
					this, SLOT(ToggleHideMixer()));
			actionHideMixer->setCheckable(true);
			actionHideMixer->setChecked(SourceMixerHidden(source));
		}

J
jp9000 已提交
4277
		if (isAsyncVideo) {
P
pkv 已提交
4278 4279
			deinterlaceMenu = new QMenu(QTStr("Deinterlacing"));
			popup.addMenu(AddDeinterlacingMenu(deinterlaceMenu, source));
J
jp9000 已提交
4280 4281
			popup.addSeparator();
		}
4282

4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297
		QAction *resizeOutput = popup.addAction(
				QTStr("ResizeOutputSizeOfSource"), this,
				SLOT(ResizeOutputSizeOfSource()));

		int width = obs_source_get_width(source);
		int height = obs_source_get_height(source);

		resizeOutput->setEnabled(!(ui->streamButton->isChecked() ||
				ui->recordButton->isChecked() ||
				(replayBufferButton &&
				replayBufferButton->isChecked())));

		if (width == 0 || height == 0)
			resizeOutput->setEnabled(false);

P
pkv 已提交
4298 4299
		scaleFilteringMenu = new QMenu(QTStr("ScaleFiltering"));
		popup.addMenu(AddScaleFilteringMenu(scaleFilteringMenu, sceneItem));
4300 4301
		popup.addSeparator();

J
jp9000 已提交
4302
		popup.addMenu(sourceProjector);
C
cg2121 已提交
4303
		popup.addAction(sourceWindow);
J
jp9000 已提交
4304
		popup.addSeparator();
J
John Bradley 已提交
4305 4306 4307 4308 4309 4310 4311

		action = popup.addAction(QTStr("Interact"), this,
				SLOT(on_actionInteract_triggered()));

		action->setEnabled(obs_source_get_output_flags(source) &
				OBS_SOURCE_INTERACTION);

J
jp9000 已提交
4312 4313
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
4314 4315
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
4316 4317

		ui->actionCopyFilters->setEnabled(true);
4318
		ui->actionCopySource->setEnabled(true);
4319 4320
	} else {
		ui->actionPasteFilters->setEnabled(false);
J
jp9000 已提交
4321 4322 4323
	}

	popup.exec(QCursor::pos());
4324 4325
}

J
jp9000 已提交
4326 4327
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
J
jp9000 已提交
4328 4329 4330 4331
	if (ui->scenes->count()) {
		QModelIndex idx = ui->sources->indexAt(pos);
		CreateSourcePopupMenu(idx.row(), false);
	}
P
Palana 已提交
4332 4333
}

4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351
void OBSBasic::on_scenes_itemDoubleClicked(QListWidgetItem *witem)
{
	if (!witem)
		return;

	if (IsPreviewProgramMode()) {
		bool doubleClickSwitch = config_get_bool(App()->GlobalConfig(),
				"BasicWindow", "TransitionOnDoubleClick");

		if (doubleClickSwitch) {
			OBSScene scene = GetCurrentScene();

			if (scene)
				SetCurrentScene(scene, false, true);
		}
	}
}

J
jp9000 已提交
4352
void OBSBasic::AddSource(const char *id)
4353
{
4354 4355 4356
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
4357
		if (sourceSelect.newSource && strcmp(id, "group") != 0)
4358
			CreatePropertiesWindow(sourceSelect.newSource);
4359
	}
4360 4361
}

4362
QMenu *OBSBasic::CreateAddSourcePopupMenu()
4363
{
4364
	const char *type;
J
jp9000 已提交
4365
	bool foundValues = false;
4366
	bool foundDeprecated = false;
J
jp9000 已提交
4367
	size_t idx = 0;
4368

4369
	QMenu *popup = new QMenu(QTStr("Add"), this);
4370
	QMenu *deprecated = new QMenu(QTStr("Deprecated"), popup);
4371

J
jp9000 已提交
4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388
	auto getActionAfter = [] (QMenu *menu, const QString &name)
	{
		QList<QAction*> actions = menu->actions();

		for (QAction *menuAction : actions) {
			if (menuAction->text().compare(name) >= 0)
				return menuAction;
		}

		return (QAction*)nullptr;
	};

	auto addSource = [this, getActionAfter] (QMenu *popup,
			const char *type, const char *name)
	{
		QString qname = QT_UTF8(name);
		QAction *popupItem = new QAction(qname, this);
J
jp9000 已提交
4389
		popupItem->setData(QT_UTF8(type));
4390 4391
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4392 4393 4394

		QAction *after = getActionAfter(popup, qname);
		popup->insertAction(after, popupItem);
J
jp9000 已提交
4395
	};
4396

J
jp9000 已提交
4397 4398
	while (obs_enum_input_types(idx++, &type)) {
		const char *name = obs_source_get_display_name(type);
4399
		uint32_t caps = obs_get_source_output_flags(type);
J
jp9000 已提交
4400

4401 4402 4403
		if ((caps & OBS_SOURCE_CAP_DISABLED) != 0)
			continue;

4404 4405
		if ((caps & OBS_SOURCE_DEPRECATED) == 0) {
			addSource(popup, type, name);
4406 4407 4408
		} else {
			addSource(deprecated, type, name);
			foundDeprecated = true;
4409
		}
4410
		foundValues = true;
4411 4412
	}

J
jp9000 已提交
4413
	addSource(popup, "scene", Str("Basic.Scene"));
J
jp9000 已提交
4414

J
jp9000 已提交
4415 4416
	popup->addSeparator();
	QAction *addGroup = new QAction(QTStr("Group"), this);
4417
	addGroup->setData(QT_UTF8("group"));
J
jp9000 已提交
4418
	connect(addGroup, SIGNAL(triggered(bool)),
4419
			this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4420 4421
	popup->addAction(addGroup);

4422 4423 4424 4425 4426
	if (!foundDeprecated) {
		delete deprecated;
		deprecated = nullptr;
	}

4427 4428 4429
	if (!foundValues) {
		delete popup;
		popup = nullptr;
4430 4431

	} else if (foundDeprecated) {
J
jp9000 已提交
4432
		popup->addSeparator();
4433
		popup->addMenu(deprecated);
4434
	}
4435 4436 4437 4438 4439 4440 4441 4442 4443 4444 4445 4446 4447 4448 4449 4450 4451

	return popup;
}

void OBSBasic::AddSourceFromAction()
{
	QAction *action = qobject_cast<QAction*>(sender());
	if (!action)
		return;

	AddSource(QT_TO_UTF8(action->data().toString()));
}

void OBSBasic::AddSourcePopupMenu(const QPoint &pos)
{
	if (!GetCurrentScene()) {
		// Tell the user he needs a scene first (help beginners).
4452
		OBSMessageBox::information(this,
4453 4454 4455 4456 4457
				QTStr("Basic.Main.AddSourceHelp.Title"),
				QTStr("Basic.Main.AddSourceHelp.Text"));
		return;
	}

4458
	QScopedPointer<QMenu> popup(CreateAddSourcePopupMenu());
4459 4460
	if (popup)
		popup->exec(pos);
4461 4462
}

J
jp9000 已提交
4463
void OBSBasic::on_actionAddSource_triggered()
4464
{
J
jp9000 已提交
4465
	AddSourcePopupMenu(QCursor::pos());
4466 4467
}

J
jp9000 已提交
4468 4469 4470 4471 4472 4473 4474 4475 4476 4477 4478 4479 4480
static bool remove_items(obs_scene_t *, obs_sceneitem_t *item, void *param)
{
	vector<OBSSceneItem> &items =
		*reinterpret_cast<vector<OBSSceneItem>*>(param);

	if (obs_sceneitem_selected(item)) {
		items.emplace_back(item);
	} else if (obs_sceneitem_is_group(item)) {
		obs_sceneitem_group_enum_items(item, remove_items, &items);
	}
	return true;
};

J
jp9000 已提交
4481
void OBSBasic::on_actionRemoveSource_triggered()
4482
{
4483
	vector<OBSSceneItem> items;
4484

J
jp9000 已提交
4485
	obs_scene_enum_items(GetCurrentScene(), remove_items, &items);
4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515 4516 4517 4518

	if (!items.size())
		return;

	auto removeMultiple = [this] (size_t count)
	{
		QString text = QTStr("ConfirmRemove.TextMultiple")
			.arg(QString::number(count));

		QMessageBox remove_items(this);
		remove_items.setText(text);
		QAbstractButton *Yes = remove_items.addButton(QTStr("Yes"),
				QMessageBox::YesRole);
		remove_items.addButton(QTStr("No"), QMessageBox::NoRole);
		remove_items.setIcon(QMessageBox::Question);
		remove_items.setWindowTitle(QTStr("ConfirmRemove.Title"));
		remove_items.exec();

		return Yes == remove_items.clickedButton();
	};

	if (items.size() == 1) {
		OBSSceneItem &item = items[0];
		obs_source_t *source = obs_sceneitem_get_source(item);

		if (source && QueryRemoveSource(source))
			obs_sceneitem_remove(item);
	} else {
		if (removeMultiple(items.size())) {
			for (auto &item : items)
				obs_sceneitem_remove(item);
		}
	}
4519 4520
}

J
John Bradley 已提交
4521 4522 4523 4524 4525 4526 4527 4528 4529
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
4530
void OBSBasic::on_actionSourceProperties_triggered()
4531
{
4532
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4533
	OBSSource source = obs_sceneitem_get_source(item);
4534

4535 4536
	if (source)
		CreatePropertiesWindow(source);
4537 4538
}

J
jp9000 已提交
4539
void OBSBasic::on_actionSourceUp_triggered()
4540
{
J
jp9000 已提交
4541
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4542
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
4543
}
J
jp9000 已提交
4544

J
jp9000 已提交
4545
void OBSBasic::on_actionSourceDown_triggered()
4546
{
J
jp9000 已提交
4547
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4548
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
4549 4550
}

J
jp9000 已提交
4551 4552 4553
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4554
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
4555 4556 4557 4558 4559
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4560
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
4561 4562 4563 4564 4565
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4566
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
4567 4568 4569 4570 4571
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4572
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
4573 4574
}

4575
static BPtr<char> ReadLogFile(const char *subdir, const char *log)
J
jp9000 已提交
4576
{
4577
	char logDir[512];
4578
	if (GetConfigPath(logDir, sizeof(logDir), subdir) <= 0)
4579
		return nullptr;
J
jp9000 已提交
4580 4581 4582 4583 4584

	string path = (char*)logDir;
	path += "/";
	path += log;

4585
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
4586 4587 4588 4589 4590 4591
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

4592
void OBSBasic::UploadLog(const char *subdir, const char *file)
J
jp9000 已提交
4593
{
4594
	BPtr<char> fileString{ReadLogFile(subdir, file)};
J
jp9000 已提交
4595

4596
	if (!fileString)
J
jp9000 已提交
4597 4598
		return;

4599
	if (!*fileString)
J
jp9000 已提交
4600 4601 4602 4603
		return;

	ui->menuLogFiles->setEnabled(false);

4604 4605
	stringstream ss;
	ss << "OBS " << App()->GetVersionString()
4606 4607
	   << " log file uploaded at " << CurrentDateTimeString()
	   << "\n\n" << fileString;
4608

F
fryshorts 已提交
4609

4610 4611 4612
	if (logUploadThread) {
		logUploadThread->wait();
	}
F
fryshorts 已提交
4613

4614
	RemoteTextThread *thread = new RemoteTextThread(
4615
			"https://obsproject.com/logs/upload",
4616 4617
			"text/plain", ss.str().c_str());

4618
	logUploadThread.reset(thread);
4619 4620 4621
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::logUploadFinished);
	logUploadThread->start();
J
jp9000 已提交
4622 4623
}

P
Palana 已提交
4624 4625
void OBSBasic::on_actionShowLogs_triggered()
{
4626
	char logDir[512];
4627
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4628 4629
		return;

P
Palana 已提交
4630 4631 4632 4633
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
4634 4635
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
4636
	UploadLog("obs-studio/logs", App()->GetCurrentLog());
J
jp9000 已提交
4637 4638 4639 4640
}

void OBSBasic::on_actionUploadLastLog_triggered()
{
4641
	UploadLog("obs-studio/logs", App()->GetLastLog());
J
jp9000 已提交
4642 4643
}

4644 4645 4646
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
4647
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4648 4649 4650 4651 4652 4653 4654 4655 4656 4657 4658 4659
		return;

	const char* log = App()->GetCurrentLog();

	string path = (char*)logDir;
	path += "/";
	path += log;

	QUrl url = QUrl::fromLocalFile(QT_UTF8(path.c_str()));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
4660 4661 4662 4663 4664 4665 4666 4667 4668 4669 4670 4671 4672 4673 4674
void OBSBasic::on_actionShowCrashLogs_triggered()
{
	char logDir[512];
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/crashes") <= 0)
		return;

	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

void OBSBasic::on_actionUploadLastCrashLog_triggered()
{
	UploadLog("obs-studio/crashes", App()->GetLastCrashLog());
}

J
jp9000 已提交
4675 4676
void OBSBasic::on_actionCheckForUpdates_triggered()
{
J
jp9000 已提交
4677
	CheckForUpdates(true);
J
jp9000 已提交
4678 4679
}

4680
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
4681 4682 4683
{
	ui->menuLogFiles->setEnabled(true);

4684
	if (text.isEmpty()) {
4685
		OBSMessageBox::information(this,
J
jp9000 已提交
4686
				QTStr("LogReturnDialog.ErrorUploadingLog"),
4687
				error);
J
jp9000 已提交
4688 4689 4690
		return;
	}

4691
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
4692
	string resURL = obs_data_get_string(returnData, "url");
4693
	QString logURL = resURL.c_str();
J
jp9000 已提交
4694 4695 4696 4697 4698 4699
	obs_data_release(returnData);

	OBSLogReply logDialog(this, logURL);
	logDialog.exec();
}

J
jp9000 已提交
4700
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
4701
		obs_source_t *source, const string &name)
4702
{
4703 4704 4705 4706
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

4707
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
4708
	QListWidgetItem *listItem    = listWidget->currentItem();
4709

4710
	if (foundSource || name.empty()) {
4711
		listItem->setText(QT_UTF8(prevName));
4712

4713
		if (foundSource) {
4714
			OBSMessageBox::information(parent,
4715 4716 4717
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
4718
			OBSMessageBox::information(parent,
4719 4720 4721 4722
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

4723 4724 4725
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
4726
		obs_source_set_name(source, name.c_str());
4727 4728 4729
	}
}

J
jp9000 已提交
4730 4731 4732 4733 4734
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
4735
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
4736 4737 4738 4739

	if (!scene)
		return;

4740
	obs_source_t *source = obs_scene_get_source(scene);
4741
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
4742

J
jp9000 已提交
4743 4744 4745
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);

J
jp9000 已提交
4746 4747 4748
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
4749 4750 4751 4752 4753 4754 4755 4756
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4757 4758 4759 4760 4761 4762 4763 4764
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4765 4766 4767 4768
#define RECORDING_START \
	"==== Recording Start ==============================================="
#define RECORDING_STOP \
	"==== Recording Stop ================================================"
J
jp9000 已提交
4769 4770 4771 4772
#define REPLAY_BUFFER_START \
	"==== Replay Buffer Start ==========================================="
#define REPLAY_BUFFER_STOP \
	"==== Replay Buffer Stop ============================================"
J
jp9000 已提交
4773 4774 4775 4776 4777
#define STREAMING_START \
	"==== Streaming Start ==============================================="
#define STREAMING_STOP \
	"==== Streaming Stop ================================================"

4778 4779
void OBSBasic::StartStreaming()
{
J
jp9000 已提交
4780 4781
	if (outputHandler->StreamingActive())
		return;
4782
	if (disableOutputsRef)
4783
		return;
J
jp9000 已提交
4784 4785 4786 4787

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTING);

4788 4789
	SaveProject();

J
jp9000 已提交
4790 4791
	ui->streamButton->setEnabled(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
4792 4793 4794 4795 4796

	if (sysTrayStream) {
		sysTrayStream->setEnabled(false);
		sysTrayStream->setText(ui->streamButton->text());
	}
4797

J
jp9000 已提交
4798 4799 4800
	if (!outputHandler->StartStreaming(service)) {
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
4801
		ui->streamButton->setChecked(false);
4802 4803 4804 4805 4806

		if (sysTrayStream) {
			sysTrayStream->setText(ui->streamButton->text());
			sysTrayStream->setEnabled(true);
		}
4807 4808 4809 4810 4811

		QMessageBox::critical(this,
				QTStr("Output.StartStreamFailed"),
				QTStr("Output.StartFailedGeneric"));
		return;
4812
	}
4813 4814 4815 4816 4817

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	if (recordWhenStreaming)
		StartRecording();
4818 4819 4820 4821 4822

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	if (replayBufferWhileStreaming)
		StartReplayBuffer();
4823 4824
}

4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845
#ifdef _WIN32
static inline void UpdateProcessPriority()
{
	const char *priority = config_get_string(App()->GlobalConfig(),
			"General", "ProcessPriority");
	if (priority && strcmp(priority, "Normal") != 0)
		SetProcessPriority(priority);
}

static inline void ClearProcessPriority()
{
	const char *priority = config_get_string(App()->GlobalConfig(),
			"General", "ProcessPriority");
	if (priority && strcmp(priority, "Normal") != 0)
		SetProcessPriority("Normal");
}
#else
#define UpdateProcessPriority() do {} while(false)
#define ClearProcessPriority() do {} while(false)
#endif

4846
inline void OBSBasic::OnActivate()
4847
{
4848 4849
	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
J
jp9000 已提交
4850
		ui->autoConfigure->setEnabled(false);
4851 4852
		App()->IncrementSleepInhibition();
		UpdateProcessPriority();
C
cg2121 已提交
4853

4854 4855
		if (trayIcon)
			trayIcon->setIcon(QIcon(":/res/images/tray_active.png"));
4856 4857
	}
}
J
jp9000 已提交
4858

4859 4860
inline void OBSBasic::OnDeactivate()
{
4861
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
4862
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
4863
		ui->autoConfigure->setEnabled(true);
4864
		App()->DecrementSleepInhibition();
4865
		ClearProcessPriority();
C
cg2121 已提交
4866

4867 4868
		if (trayIcon)
			trayIcon->setIcon(QIcon(":/res/images/obs.png"));
J
jp9000 已提交
4869
	}
4870 4871 4872 4873 4874 4875 4876
}

void OBSBasic::StopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
4877
		outputHandler->StopStreaming(streamingStopping);
4878 4879

	OnDeactivate();
4880 4881 4882 4883 4884 4885 4886

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
4887 4888 4889 4890 4891 4892 4893

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	bool keepReplayBufferStreamStops = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "KeepReplayBufferStreamStops");
	if (replayBufferWhileStreaming && !keepReplayBufferStreamStops)
		StopReplayBuffer();
4894 4895
}

J
jp9000 已提交
4896 4897 4898 4899 4900
void OBSBasic::ForceStopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
4901
		outputHandler->StopStreaming(true);
J
jp9000 已提交
4902

4903
	OnDeactivate();
4904 4905 4906 4907 4908 4909 4910

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
4911 4912 4913 4914 4915 4916 4917

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	bool keepReplayBufferStreamStops = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "KeepReplayBufferStreamStops");
	if (replayBufferWhileStreaming && !keepReplayBufferStreamStops)
		StopReplayBuffer();
J
jp9000 已提交
4918 4919 4920 4921 4922 4923
}

void OBSBasic::StreamDelayStarting(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
	ui->streamButton->setEnabled(true);
4924
	ui->streamButton->setChecked(true);
4925 4926 4927 4928 4929

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
4930 4931 4932 4933 4934 4935 4936 4937 4938 4939 4940 4941

	if (!startStreamMenu.isNull())
		startStreamMenu->deleteLater();

	startStreamMenu = new QMenu();
	startStreamMenu->addAction(QTStr("Basic.Main.StopStreaming"),
			this, SLOT(StopStreaming()));
	startStreamMenu->addAction(QTStr("Basic.Main.ForceStopStreaming"),
			this, SLOT(ForceStopStreaming()));
	ui->streamButton->setMenu(startStreamMenu);

	ui->statusbar->StreamDelayStarting(sec);
4942

4943
	OnActivate();
J
jp9000 已提交
4944 4945 4946 4947 4948 4949
}

void OBSBasic::StreamDelayStopping(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
	ui->streamButton->setEnabled(true);
4950
	ui->streamButton->setChecked(false);
4951 4952 4953 4954 4955

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
4956 4957 4958 4959 4960 4961 4962 4963 4964 4965 4966 4967

	if (!startStreamMenu.isNull())
		startStreamMenu->deleteLater();

	startStreamMenu = new QMenu();
	startStreamMenu->addAction(QTStr("Basic.Main.StartStreaming"),
			this, SLOT(StartStreaming()));
	startStreamMenu->addAction(QTStr("Basic.Main.ForceStopStreaming"),
			this, SLOT(ForceStopStreaming()));
	ui->streamButton->setMenu(startStreamMenu);

	ui->statusbar->StreamDelayStopping(sec);
4968 4969 4970

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPING);
J
jp9000 已提交
4971 4972
}

4973
void OBSBasic::StreamingStart()
4974
{
4975
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
4976
	ui->streamButton->setEnabled(true);
4977
	ui->streamButton->setChecked(true);
J
jp9000 已提交
4978
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
4979 4980 4981 4982 4983

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
4984

J
jp9000 已提交
4985 4986 4987
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTED);

4988
	OnActivate();
4989

4990
	blog(LOG_INFO, STREAMING_START);
4991 4992
}

4993 4994 4995
void OBSBasic::StreamStopping()
{
	ui->streamButton->setText(QTStr("Basic.Main.StoppingStreaming"));
4996 4997 4998

	if (sysTrayStream)
		sysTrayStream->setText(ui->streamButton->text());
J
jp9000 已提交
4999

5000
	streamingStopping = true;
J
jp9000 已提交
5001 5002
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPING);
5003 5004
}

5005
void OBSBasic::StreamingStop(int code, QString last_error)
5006
{
5007 5008 5009
	const char *errorDescription;
	DStr errorMessage;
	bool use_last_error = false;
5010 5011 5012

	switch (code) {
	case OBS_OUTPUT_BAD_PATH:
5013
		errorDescription = Str("Output.ConnectFail.BadPath");
5014 5015 5016
		break;

	case OBS_OUTPUT_CONNECT_FAILED:
5017 5018
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.ConnectFailed");
5019 5020 5021
		break;

	case OBS_OUTPUT_INVALID_STREAM:
5022
		errorDescription = Str("Output.ConnectFail.InvalidStream");
5023 5024
		break;

5025
	default:
5026
	case OBS_OUTPUT_ERROR:
5027 5028
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Error");
5029 5030 5031 5032 5033
		break;

	case OBS_OUTPUT_DISCONNECTED:
		/* doesn't happen if output is set to reconnect.  note that
		 * reconnects are handled in the output, not in the UI */
5034 5035
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Disconnected");
5036 5037
	}

5038 5039 5040 5041 5042 5043
	if (use_last_error && !last_error.isEmpty())
		dstr_printf(errorMessage, "%s\n\n%s", errorDescription,
			QT_TO_UTF8(last_error));
	else
		dstr_copy(errorMessage, errorDescription);

J
jp9000 已提交
5044
	ui->statusbar->StreamStopped();
5045

5046
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
5047
	ui->streamButton->setEnabled(true);
5048
	ui->streamButton->setChecked(false);
5049 5050 5051 5052 5053

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
5054

5055
	streamingStopping = false;
J
jp9000 已提交
5056 5057 5058
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPED);

5059
	OnDeactivate();
5060 5061

	blog(LOG_INFO, STREAMING_STOP);
J
jp9000 已提交
5062

C
cg2121 已提交
5063
	if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5064
		OBSMessageBox::information(this,
5065 5066
				QTStr("Output.ConnectFail.Title"),
				QT_UTF8(errorMessage));
C
cg2121 已提交
5067
	} else if (code != OBS_OUTPUT_SUCCESS && !isVisible()) {
5068
		SysTrayNotify(QT_UTF8(errorDescription), QSystemTrayIcon::Warning);
C
cg2121 已提交
5069
	}
J
jp9000 已提交
5070 5071 5072 5073 5074 5075

	if (!startStreamMenu.isNull()) {
		ui->streamButton->setMenu(nullptr);
		startStreamMenu->deleteLater();
		startStreamMenu = nullptr;
	}
J
jp9000 已提交
5076 5077
}

C
cg2121 已提交
5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093 5094
void OBSBasic::AutoRemux()
{
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	const char *path = strcmp(mode, "Advanced") ?
	config_get_string(basicConfig, "SimpleOutput", "FilePath") :
	config_get_string(basicConfig, "AdvOut", "RecFilePath");
	std::string s(path);
	s += "/";
	s += remuxFilename;
	const QString &str = QString::fromStdString(s);
	QString file = str.section(".", 0, 0);

	OBSRemux *remux = new OBSRemux(path, this, true);
	remux->show();
	remux->AutoRemux(str, file + ".mp4");
}

5095 5096
void OBSBasic::StartRecording()
{
5097 5098
	if (outputHandler->RecordingActive())
		return;
5099
	if (disableOutputsRef)
5100
		return;
5101

J
jp9000 已提交
5102 5103 5104
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTING);

5105
	SaveProject();
5106 5107 5108

	if (!outputHandler->StartRecording())
		ui->recordButton->setChecked(false);
5109 5110
}

5111 5112
void OBSBasic::RecordStopping()
{
J
jp9000 已提交
5113
	ui->recordButton->setText(QTStr("Basic.Main.StoppingRecording"));
5114 5115 5116

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
J
jp9000 已提交
5117

5118
	recordingStopping = true;
J
jp9000 已提交
5119 5120
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPING);
5121 5122
}

5123 5124 5125 5126 5127
void OBSBasic::StopRecording()
{
	SaveProject();

	if (outputHandler->RecordingActive())
5128
		outputHandler->StopRecording(recordingStopping);
J
jp9000 已提交
5129

5130
	OnDeactivate();
5131 5132
}

P
Palana 已提交
5133 5134
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
5135
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
J
jp9000 已提交
5136
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
5137
	ui->recordButton->setChecked(true);
5138 5139 5140

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
5141

5142
	recordingStopping = false;
J
jp9000 已提交
5143 5144 5145
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTED);

5146
	OnActivate();
5147

5148
	blog(LOG_INFO, RECORDING_START);
P
Palana 已提交
5149 5150
}

5151
void OBSBasic::RecordingStop(int code)
5152
{
P
Palana 已提交
5153
	ui->statusbar->RecordingStopped();
J
jp9000 已提交
5154
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
5155
	ui->recordButton->setChecked(false);
5156 5157 5158

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
J
jp9000 已提交
5159

5160
	blog(LOG_INFO, RECORDING_STOP);
5161

C
cg2121 已提交
5162
	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5163
		OBSMessageBox::information(this,
5164 5165
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));
J
jp9000 已提交
5166

C
cg2121 已提交
5167
	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5168
		OBSMessageBox::information(this,
J
jp9000 已提交
5169 5170 5171
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

C
cg2121 已提交
5172
	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5173
		OBSMessageBox::information(this,
J
jp9000 已提交
5174 5175
				QTStr("Output.RecordError.Title"),
				QTStr("Output.RecordError.Msg"));
C
cg2121 已提交
5176 5177 5178 5179 5180 5181 5182 5183 5184 5185 5186 5187

	} else if (code == OBS_OUTPUT_UNSUPPORTED && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordFail.Unsupported"),
			QSystemTrayIcon::Warning);

	} else if (code == OBS_OUTPUT_NO_SPACE && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordNoSpace.Msg"),
			QSystemTrayIcon::Warning);

	} else if (code != OBS_OUTPUT_SUCCESS && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordError.Msg"),
			QSystemTrayIcon::Warning);
J
jp9000 已提交
5188 5189
	}

J
jp9000 已提交
5190 5191 5192
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPED);

C
cg2121 已提交
5193 5194 5195
	if (remuxAfterRecord)
		AutoRemux();

5196
	OnDeactivate();
5197
}
5198

J
jp9000 已提交
5199 5200 5201 5202 5203 5204 5205 5206 5207
#define RP_NO_HOTKEY_TITLE QTStr("Output.ReplayBuffer.NoHotkey.Title")
#define RP_NO_HOTKEY_TEXT  QTStr("Output.ReplayBuffer.NoHotkey.Msg")

void OBSBasic::StartReplayBuffer()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;
	if (outputHandler->ReplayBufferActive())
		return;
5208
	if (disableOutputsRef)
5209
		return;
J
jp9000 已提交
5210

5211 5212 5213 5214 5215
	if (!NoSourcesConfirmation()) {
		replayBufferButton->setChecked(false);
		return;
	}

J
jp9000 已提交
5216 5217 5218 5219 5220 5221 5222 5223 5224
	obs_output_t *output = outputHandler->replayBuffer;
	obs_data_t *hotkeys = obs_hotkeys_save_output(output);
	obs_data_array_t *bindings = obs_data_get_array(hotkeys,
			"ReplayBuffer.Save");
	size_t count = obs_data_array_count(bindings);
	obs_data_array_release(bindings);
	obs_data_release(hotkeys);

	if (!count) {
5225
		OBSMessageBox::information(this,
J
jp9000 已提交
5226 5227 5228 5229 5230 5231 5232 5233 5234
				RP_NO_HOTKEY_TITLE,
				RP_NO_HOTKEY_TEXT);
		return;
	}

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STARTING);

	SaveProject();
5235 5236
	if (!outputHandler->StartReplayBuffer())
		replayBufferButton->setChecked(false);
J
jp9000 已提交
5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262 5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285
}

void OBSBasic::ReplayBufferStopping()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StoppingReplayBuffer"));

	if (sysTrayReplayBuffer)
		sysTrayReplayBuffer->setText(replayBufferButton->text());

	replayBufferStopping = true;
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STOPPING);
}

void OBSBasic::StopReplayBuffer()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	SaveProject();

	if (outputHandler->ReplayBufferActive())
		outputHandler->StopReplayBuffer(replayBufferStopping);

	OnDeactivate();
}

void OBSBasic::ReplayBufferStart()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StopReplayBuffer"));

	if (sysTrayReplayBuffer)
		sysTrayReplayBuffer->setText(replayBufferButton->text());

	replayBufferStopping = false;
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STARTED);

	OnActivate();

	blog(LOG_INFO, REPLAY_BUFFER_START);
}

5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299
void OBSBasic::ReplayBufferSave()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;
	if (!outputHandler->ReplayBufferActive())
		return;

	calldata_t cd = {0};
	proc_handler_t *ph = obs_output_get_proc_handler(
			outputHandler->replayBuffer);
	proc_handler_call(ph, "save", &cd);
	calldata_free(&cd);
}

J
jp9000 已提交
5300 5301 5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312
void OBSBasic::ReplayBufferStop(int code)
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StartReplayBuffer"));

	if (sysTrayReplayBuffer)
		sysTrayReplayBuffer->setText(replayBufferButton->text());

	blog(LOG_INFO, REPLAY_BUFFER_STOP);

	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5313
		OBSMessageBox::information(this,
J
jp9000 已提交
5314 5315 5316 5317
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));

	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5318
		OBSMessageBox::information(this,
J
jp9000 已提交
5319 5320 5321 5322
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5323
		OBSMessageBox::information(this,
J
jp9000 已提交
5324 5325 5326 5327 5328 5329 5330 5331 5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345
				QTStr("Output.RecordError.Title"),
				QTStr("Output.RecordError.Msg"));

	} else if (code == OBS_OUTPUT_UNSUPPORTED && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordFail.Unsupported"),
			QSystemTrayIcon::Warning);

	} else if (code == OBS_OUTPUT_NO_SPACE && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordNoSpace.Msg"),
			QSystemTrayIcon::Warning);

	} else if (code != OBS_OUTPUT_SUCCESS && !isVisible()) {
		SysTrayNotify(QTStr("Output.RecordError.Msg"),
			QSystemTrayIcon::Warning);
	}

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STOPPED);

	OnDeactivate();
}

5346 5347 5348 5349 5350 5351 5352 5353 5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364 5365 5366 5367
bool OBSBasic::NoSourcesConfirmation()
{
	if (CountVideoSources() == 0 && isVisible()) {
		QString msg;
		msg = QTStr("NoSources.Text");
		msg += "\n\n";
		msg += QTStr("NoSources.Text.AddSource");

		QMessageBox messageBox(QMessageBox::Question,
				QTStr("NoSources.title"),
				msg,
				QMessageBox::Yes | QMessageBox::No,
				this);
		messageBox.setDefaultButton(QMessageBox::No);

		if (QMessageBox::No == messageBox.exec())
			return false;
	}

	return true;
}

5368 5369
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
5370
	if (outputHandler->StreamingActive()) {
5371 5372 5373
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStoppingStream");

C
cg2121 已提交
5374
		if (confirm && isVisible()) {
5375
			QMessageBox::StandardButton button =
5376
				OBSMessageBox::question(this,
5377 5378 5379
						QTStr("ConfirmStop.Title"),
						QTStr("ConfirmStop.Text"));

C
cg2121 已提交
5380 5381
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(true);
5382
				return;
C
cg2121 已提交
5383
			}
5384 5385
		}

5386
		StopStreaming();
5387
	} else {
5388 5389 5390 5391 5392
		if (!NoSourcesConfirmation()) {
			ui->streamButton->setChecked(false);
			return;
		}

5393 5394 5395
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStartingStream");

C
cg2121 已提交
5396
		if (confirm && isVisible()) {
5397
			QMessageBox::StandardButton button =
5398
				OBSMessageBox::question(this,
5399 5400 5401
						QTStr("ConfirmStart.Title"),
						QTStr("ConfirmStart.Text"));

C
cg2121 已提交
5402 5403
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(false);
5404
				return;
C
cg2121 已提交
5405
			}
5406 5407
		}

5408
		StartStreaming();
5409 5410 5411 5412 5413
	}
}

void OBSBasic::on_recordButton_clicked()
{
5414
	if (outputHandler->RecordingActive()) {
5415
		StopRecording();
5416 5417 5418 5419 5420 5421
	} else {
		if (!NoSourcesConfirmation()) {
			ui->recordButton->setChecked(false);
			return;
		}

5422
		StartRecording();
5423
	}
J
jp9000 已提交
5424 5425
}

J
jp9000 已提交
5426
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
5427
{
J
jp9000 已提交
5428
	on_action_Settings_triggered();
J
jp9000 已提交
5429
}
5430

5431 5432 5433 5434 5435 5436
void OBSBasic::on_actionHelpPortal_triggered()
{
	QUrl url = QUrl("https://obsproject.com/help", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5437 5438 5439 5440 5441 5442
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5443 5444
void OBSBasic::on_actionDiscord_triggered()
{
5445
	QUrl url = QUrl("https://obsproject.com/discord", QUrl::TolerantMode);
5446 5447 5448
	QDesktopServices::openUrl(url);
}

5449 5450 5451 5452 5453 5454 5455 5456 5457 5458 5459 5460 5461 5462 5463 5464 5465 5466 5467 5468
void OBSBasic::on_actionShowSettingsFolder_triggered()
{
	char path[512];
	int ret = GetConfigPath(path, 512, "obs-studio");
	if (ret <= 0)
		return;

	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

void OBSBasic::on_actionShowProfileFolder_triggered()
{
	char path[512];
	int ret = GetProfilePath(path, 512, "");
	if (ret <= 0)
		return;

	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

J
jp9000 已提交
5469
int OBSBasic::GetTopSelectedSourceItem()
5470
{
J
jp9000 已提交
5471 5472 5473
	QModelIndexList selectedItems =
		ui->sources->selectionModel()->selectedIndexes();
	return selectedItems.count() ? selectedItems[0].row() : -1;
5474 5475
}

J
jp9000 已提交
5476 5477
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
5478
	CreateSourcePopupMenu(GetTopSelectedSourceItem(), true);
J
jp9000 已提交
5479 5480 5481 5482

	UNUSED_PARAMETER(pos);
}

R
Ryan Foster 已提交
5483 5484 5485 5486 5487 5488 5489 5490
void OBSBasic::on_program_customContextMenuRequested(const QPoint&)
{
	QMenu popup(this);
	QPointer<QMenu> studioProgramProjector;

	studioProgramProjector = new QMenu(
			QTStr("StudioProgramProjector"));
	AddProjectorMenuMonitors(studioProgramProjector, this,
5491
			SLOT(OpenStudioProgramProjector()));
R
Ryan Foster 已提交
5492 5493 5494 5495 5496

	popup.addMenu(studioProgramProjector);

	QAction *studioProgramWindow = popup.addAction(
			QTStr("StudioProgramWindow"),
5497
			this, SLOT(OpenStudioProgramWindow()));
R
Ryan Foster 已提交
5498 5499 5500 5501 5502 5503

	popup.addAction(studioProgramWindow);

	popup.exec(QCursor::pos());
}

J
jp9000 已提交
5504 5505 5506 5507
void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
P
pkv 已提交
5508
	delete previewProjectorMain;
J
jp9000 已提交
5509 5510 5511 5512 5513

	QAction *action = popup.addAction(
			QTStr("Basic.Main.PreviewConextMenu.Enable"),
			this, SLOT(TogglePreview()));
	action->setCheckable(true);
5514
	action->setChecked(obs_display_enabled(ui->preview->GetDisplay()));
J
jp9000 已提交
5515

P
pkv 已提交
5516 5517
	previewProjectorMain = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjectorMain, this,
J
jp9000 已提交
5518 5519
			SLOT(OpenPreviewProjector()));

C
cg2121 已提交
5520 5521 5522 5523
	QAction *previewWindow = popup.addAction(
			QTStr("PreviewWindow"),
			this, SLOT(OpenPreviewWindow()));

P
pkv 已提交
5524
	popup.addMenu(previewProjectorMain);
C
cg2121 已提交
5525
	popup.addAction(previewWindow);
J
jp9000 已提交
5526 5527 5528 5529 5530
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

5531 5532
void OBSBasic::on_actionAlwaysOnTop_triggered()
{
5533
#ifndef _WIN32
5534 5535 5536 5537
	/* Make sure all dialogs are safely and successfully closed before
	 * switching the always on top mode due to the fact that windows all
	 * have to be recreated, so queue the actual toggle to happen after
	 * all events related to closing the dialogs have finished */
5538 5539 5540
	CloseDialogs();
#endif

5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552 5553 5554
	QMetaObject::invokeMethod(this, "ToggleAlwaysOnTop",
			Qt::QueuedConnection);
}

void OBSBasic::ToggleAlwaysOnTop()
{
	bool isAlwaysOnTop = IsAlwaysOnTop(this);

	ui->actionAlwaysOnTop->setChecked(!isAlwaysOnTop);
	SetAlwaysOnTop(this, !isAlwaysOnTop);

	show();
}

5555 5556 5557 5558 5559 5560 5561 5562 5563 5564
void OBSBasic::GetFPSCommon(uint32_t &num, uint32_t &den) const
{
	const char *val = config_get_string(basicConfig, "Video", "FPSCommon");

	if (strcmp(val, "10") == 0) {
		num = 10;
		den = 1;
	} else if (strcmp(val, "20") == 0) {
		num = 20;
		den = 1;
J
jp9000 已提交
5565 5566 5567
	} else if (strcmp(val, "24 NTSC") == 0) {
		num = 24000;
		den = 1001;
5568 5569 5570 5571 5572 5573 5574 5575 5576 5577 5578 5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594 5595 5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606 5607 5608 5609 5610 5611 5612 5613 5614 5615 5616 5617 5618 5619 5620
	} else if (strcmp(val, "25") == 0) {
		num = 25;
		den = 1;
	} else if (strcmp(val, "29.97") == 0) {
		num = 30000;
		den = 1001;
	} else if (strcmp(val, "48") == 0) {
		num = 48;
		den = 1;
	} else if (strcmp(val, "59.94") == 0) {
		num = 60000;
		den = 1001;
	} else if (strcmp(val, "60") == 0) {
		num = 60;
		den = 1;
	} else {
		num = 30;
		den = 1;
	}
}

void OBSBasic::GetFPSInteger(uint32_t &num, uint32_t &den) const
{
	num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSInt");
	den = 1;
}

void OBSBasic::GetFPSFraction(uint32_t &num, uint32_t &den) const
{
	num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNum");
	den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSDen");
}

void OBSBasic::GetFPSNanoseconds(uint32_t &num, uint32_t &den) const
{
	num = 1000000000;
	den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNS");
}

void OBSBasic::GetConfigFPS(uint32_t &num, uint32_t &den) const
{
	uint32_t type = config_get_uint(basicConfig, "Video", "FPSType");

	if (type == 1) //"Integer"
		GetFPSInteger(num, den);
	else if (type == 2) //"Fraction"
		GetFPSFraction(num, den);
	else if (false) //"Nanoseconds", currently not implemented
		GetFPSNanoseconds(num, den);
	else
		GetFPSCommon(num, den);
}

5621
config_t *OBSBasic::Config() const
5622 5623 5624
{
	return basicConfig;
}
J
jp9000 已提交
5625 5626 5627

void OBSBasic::on_actionEditTransform_triggered()
{
5628 5629 5630
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
5631 5632
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
5633
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
5634 5635
}

5636 5637 5638 5639 5640 5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664 5665 5666 5667 5668 5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679
static obs_transform_info copiedTransformInfo;
static obs_sceneitem_crop copiedCropInfo;

void OBSBasic::on_actionCopyTransform_triggered()
{
	auto func = [](obs_scene_t *scene, obs_sceneitem_t *item, void *param)
	{
		if (!obs_sceneitem_selected(item))
			return true;

		obs_sceneitem_defer_update_begin(item);
		obs_sceneitem_get_info(item, &copiedTransformInfo);
		obs_sceneitem_get_crop(item, &copiedCropInfo);
		obs_sceneitem_defer_update_end(item);

		UNUSED_PARAMETER(scene);
		UNUSED_PARAMETER(param);
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
	ui->actionPasteTransform->setEnabled(true);
}

void OBSBasic::on_actionPasteTransform_triggered()
{
	auto func = [](obs_scene_t *scene, obs_sceneitem_t *item, void *param)
	{
		if (!obs_sceneitem_selected(item))
			return true;

		obs_sceneitem_defer_update_begin(item);
		obs_sceneitem_set_info(item, &copiedTransformInfo);
		obs_sceneitem_set_crop(item, &copiedCropInfo);
		obs_sceneitem_defer_update_end(item);

		UNUSED_PARAMETER(scene);
		UNUSED_PARAMETER(param);
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
}

5680
static bool reset_tr(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
5681
{
5682 5683 5684 5685
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, reset_tr, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
J
jp9000 已提交
5686

5687
	obs_sceneitem_defer_update_begin(item);
J
jp9000 已提交
5688

5689 5690 5691 5692 5693 5694 5695 5696 5697
	obs_transform_info info;
	vec2_set(&info.pos, 0.0f, 0.0f);
	vec2_set(&info.scale, 1.0f, 1.0f);
	info.rot = 0.0f;
	info.alignment = OBS_ALIGN_TOP | OBS_ALIGN_LEFT;
	info.bounds_type = OBS_BOUNDS_NONE;
	info.bounds_alignment = OBS_ALIGN_CENTER;
	vec2_set(&info.bounds, 0.0f, 0.0f);
	obs_sceneitem_set_info(item, &info);
J
jp9000 已提交
5698

5699 5700
	obs_sceneitem_crop crop = {};
	obs_sceneitem_set_crop(item, &crop);
J
jp9000 已提交
5701

5702
	obs_sceneitem_defer_update_end(item);
J
jp9000 已提交
5703

5704 5705 5706 5707
	UNUSED_PARAMETER(scene);
	UNUSED_PARAMETER(param);
	return true;
}
J
jp9000 已提交
5708

5709 5710 5711
void OBSBasic::on_actionResetTransform_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), reset_tr, nullptr);
J
jp9000 已提交
5712 5713
}

5714
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
5715 5716 5717 5718 5719
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

	vec3_set(&tl, M_INFINITE, M_INFINITE, 0.0f);
5720
	vec3_set(&br, -M_INFINITE, -M_INFINITE, 0.0f);
J
jp9000 已提交
5721

5722
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
5723 5724 5725 5726
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
5727 5728
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
5729 5730
	};

5731 5732 5733 5734 5735 5736
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

5737
static vec3 GetItemTL(obs_sceneitem_t *item)
5738 5739 5740
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
5741 5742 5743
	return tl;
}

5744
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
5745 5746 5747 5748
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
5749
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
5750 5751 5752
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
5753
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
5754 5755
}

5756
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5757 5758
		void *param)
{
5759 5760 5761
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, RotateSelectedSources,
				param);
J
jp9000 已提交
5762 5763 5764 5765 5766 5767 5768
	if (!obs_sceneitem_selected(item))
		return true;

	float rot = *reinterpret_cast<float*>(param);

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
5769
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
5770 5771
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
5772
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
5773

5774 5775
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
5776 5777 5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789 5790 5791 5792 5793 5794 5795 5796 5797 5798 5799
	SetItemTL(item, tl);

	UNUSED_PARAMETER(scene);
	return true;
};

void OBSBasic::on_actionRotate90CW_triggered()
{
	float f90CW = 90.0f;
	obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f90CW);
}

void OBSBasic::on_actionRotate90CCW_triggered()
{
	float f90CCW = -90.0f;
	obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f90CCW);
}

void OBSBasic::on_actionRotate180_triggered()
{
	float f180 = 180.0f;
	obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f180);
}

5800
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5801 5802 5803 5804
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

5805 5806 5807
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, MultiplySelectedItemScale,
				param);
J
jp9000 已提交
5808 5809 5810 5811 5812 5813
	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
5814
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
5815
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
5816
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
5817

5818 5819
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
5820
	SetItemTL(item, tl);
J
jp9000 已提交
5821 5822

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
5823 5824 5825 5826 5827
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
5828 5829
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
5830 5831 5832 5833 5834 5835
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
5836 5837
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
5838 5839 5840 5841
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

5842
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5843 5844 5845 5846
		void *param)
{
	obs_bounds_type boundsType = *reinterpret_cast<obs_bounds_type*>(param);

5847 5848 5849
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, CenterAlignSelectedItems,
				param);
J
jp9000 已提交
5850 5851 5852 5853 5854 5855
	if (!obs_sceneitem_selected(item))
		return true;

	obs_video_info ovi;
	obs_get_video_info(&ovi);

5856
	obs_transform_info itemInfo;
J
jp9000 已提交
5857 5858 5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869 5870 5871 5872 5873 5874 5875 5876 5877 5878 5879 5880 5881 5882 5883 5884 5885 5886
	vec2_set(&itemInfo.pos, 0.0f, 0.0f);
	vec2_set(&itemInfo.scale, 1.0f, 1.0f);
	itemInfo.alignment = OBS_ALIGN_LEFT | OBS_ALIGN_TOP;
	itemInfo.rot = 0.0f;

	vec2_set(&itemInfo.bounds,
			float(ovi.base_width), float(ovi.base_height));
	itemInfo.bounds_type = boundsType;
	itemInfo.bounds_alignment = OBS_ALIGN_CENTER;

	obs_sceneitem_set_info(item, &itemInfo);

	UNUSED_PARAMETER(scene);
	return true;
}

void OBSBasic::on_actionFitToScreen_triggered()
{
	obs_bounds_type boundsType = OBS_BOUNDS_SCALE_INNER;
	obs_scene_enum_items(GetCurrentScene(), CenterAlignSelectedItems,
			&boundsType);
}

void OBSBasic::on_actionStretchToScreen_triggered()
{
	obs_bounds_type boundsType = OBS_BOUNDS_STRETCH;
	obs_scene_enum_items(GetCurrentScene(), CenterAlignSelectedItems,
			&boundsType);
}

5887
static bool center_to_scene(obs_scene_t *, obs_sceneitem_t *item, void *)
J
jp9000 已提交
5888
{
5889 5890
	vec3 tl, br, itemCenter, screenCenter, offset;
	obs_video_info ovi;
5891

5892 5893 5894 5895
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, center_to_scene, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
5896

5897
	obs_get_video_info(&ovi);
5898

5899 5900 5901
	vec3_set(&screenCenter, float(ovi.base_width),
			float(ovi.base_height), 0.0f);
	vec3_mulf(&screenCenter, &screenCenter, 0.5f);
5902

5903
	GetItemBox(item, tl, br);
5904

5905 5906 5907
	vec3_sub(&itemCenter, &br, &tl);
	vec3_mulf(&itemCenter, &itemCenter, 0.5f);
	vec3_add(&itemCenter, &itemCenter, &tl);
5908

5909 5910
	vec3_sub(&offset, &screenCenter, &itemCenter);
	vec3_add(&tl, &tl, &offset);
5911

5912 5913 5914
	SetItemTL(item, tl);
	return true;
};
5915

5916 5917 5918
void OBSBasic::on_actionCenterToScreen_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), center_to_scene, nullptr);
J
jp9000 已提交
5919
}
J
jp9000 已提交
5920

5921 5922 5923 5924 5925 5926 5927
void OBSBasic::EnablePreviewDisplay(bool enable)
{
	obs_display_set_enabled(ui->preview->GetDisplay(), enable);
	ui->preview->setVisible(enable);
	ui->previewDisabledLabel->setVisible(!enable);
}

J
jp9000 已提交
5928 5929
void OBSBasic::TogglePreview()
{
5930 5931
	previewEnabled = !previewEnabled;
	EnablePreviewDisplay(previewEnabled);
J
jp9000 已提交
5932
}
5933

J
jp9000 已提交
5934
static bool nudge_callback(obs_scene_t*, obs_sceneitem_t *item, void *param)
5935
{
J
jp9000 已提交
5936 5937
	if (obs_sceneitem_locked(item))
		return true;
5938

J
jp9000 已提交
5939 5940
	struct vec2 &offset = *reinterpret_cast<struct vec2*>(param);
	struct vec2 pos;
5941

J
jp9000 已提交
5942
	if (!obs_sceneitem_selected(item)) {
J
jp9000 已提交
5943 5944 5945 5946 5947 5948 5949 5950 5951 5952 5953 5954 5955 5956 5957 5958
		if (obs_sceneitem_is_group(item)) {
			struct vec3 offset3;
			vec3_set(&offset3, offset.x, offset.y, 0.0f);

			struct matrix4 matrix;
			obs_sceneitem_get_draw_transform(item, &matrix);
			vec4_set(&matrix.t, 0.0f, 0.0f, 0.0f, 1.0f);
			matrix4_inv(&matrix, &matrix);
			vec3_transform(&offset3, &offset3, &matrix);

			struct vec2 new_offset;
			vec2_set(&new_offset, offset3.x, offset3.y);
			obs_sceneitem_group_enum_items(item, nudge_callback,
					&new_offset);
		}

J
jp9000 已提交
5959 5960
		return true;
	}
5961

J
jp9000 已提交
5962 5963 5964 5965 5966
	obs_sceneitem_get_pos(item, &pos);
	vec2_add(&pos, &pos, &offset);
	obs_sceneitem_set_pos(item, &pos);
	return true;
}
J
jp9000 已提交
5967

J
jp9000 已提交
5968 5969 5970 5971
void OBSBasic::Nudge(int dist, MoveDir dir)
{
	if (ui->preview->Locked())
		return;
5972

J
jp9000 已提交
5973 5974
	struct vec2 offset;
	vec2_set(&offset, 0.0f, 0.0f);
5975

J
jp9000 已提交
5976 5977 5978 5979 5980 5981
	switch (dir) {
	case MoveDir::Up:    offset.y = (float)-dist; break;
	case MoveDir::Down:  offset.y = (float) dist; break;
	case MoveDir::Left:  offset.x = (float)-dist; break;
	case MoveDir::Right: offset.x = (float) dist; break;
	}
5982

J
jp9000 已提交
5983
	obs_scene_enum_items(GetCurrentScene(), nudge_callback, &offset);
5984 5985 5986 5987 5988 5989
}

void OBSBasic::NudgeUp()       {Nudge(1,  MoveDir::Up);}
void OBSBasic::NudgeDown()     {Nudge(1,  MoveDir::Down);}
void OBSBasic::NudgeLeft()     {Nudge(1,  MoveDir::Left);}
void OBSBasic::NudgeRight()    {Nudge(1,  MoveDir::Right);}
J
jp9000 已提交
5990

S
Shaolin 已提交
5991
OBSProjector *OBSBasic::OpenProjector(obs_source_t *source, int monitor,
S
Shaolin 已提交
5992
		QString title, ProjectorType type)
J
jp9000 已提交
5993 5994
{
	/* seriously?  10 monitors? */
C
cg2121 已提交
5995
	if (monitor > 9 || monitor > QGuiApplication::screens().size() - 1)
S
Shaolin 已提交
5996
		return nullptr;
C
cg2121 已提交
5997

S
Shaolin 已提交
5998 5999
	OBSProjector *projector = new OBSProjector(nullptr, source, monitor,
			title, type);
C
cg2121 已提交
6000

S
Shaolin 已提交
6001
	if (monitor < 0) {
C
cg2121 已提交
6002 6003 6004 6005 6006 6007 6008 6009 6010
		for (auto &projPtr : windowProjectors) {
			if (!projPtr) {
				projPtr = projector;
				projector = nullptr;
			}
		}

		if (projector)
			windowProjectors.push_back(projector);
S
Shaolin 已提交
6011
	} else {
C
cg2121 已提交
6012 6013
		delete projectors[monitor];
		projectors[monitor].clear();
C
cg2121 已提交
6014

C
cg2121 已提交
6015 6016
		projectors[monitor] = projector;
	}
S
Shaolin 已提交
6017 6018

	return projector;
J
jp9000 已提交
6019 6020
}

R
Ryan Foster 已提交
6021 6022 6023
void OBSBasic::OpenStudioProgramProjector()
{
	int monitor = sender()->property("monitor").toInt();
6024
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::StudioProgram);
R
Ryan Foster 已提交
6025 6026
}

J
jp9000 已提交
6027 6028 6029
void OBSBasic::OpenPreviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
6030
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Preview);
J
jp9000 已提交
6031 6032 6033 6034 6035 6036 6037 6038 6039
}

void OBSBasic::OpenSourceProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

S
Shaolin 已提交
6040 6041
	OpenProjector(obs_sceneitem_get_source(item), monitor, nullptr,
			ProjectorType::Source);
J
jp9000 已提交
6042 6043
}

S
Shaolin 已提交
6044 6045 6046
void OBSBasic::OpenMultiviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
6047
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Multiview);
S
Shaolin 已提交
6048 6049
}

J
jp9000 已提交
6050 6051 6052 6053 6054 6055 6056
void OBSBasic::OpenSceneProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

6057 6058
	OpenProjector(obs_scene_get_source(scene), monitor, nullptr,
			ProjectorType::Scene);
C
cg2121 已提交
6059 6060
}

R
Ryan Foster 已提交
6061 6062
void OBSBasic::OpenStudioProgramWindow()
{
S
Shaolin 已提交
6063
	OpenProjector(nullptr, -1, QTStr("StudioProgramWindow"),
S
Shaolin 已提交
6064
			ProjectorType::StudioProgram);
R
Ryan Foster 已提交
6065 6066
}

C
cg2121 已提交
6067 6068
void OBSBasic::OpenPreviewWindow()
{
S
Shaolin 已提交
6069 6070
	OpenProjector(nullptr, -1, QTStr("PreviewWindow"),
			ProjectorType::Preview);
C
cg2121 已提交
6071 6072 6073 6074 6075 6076 6077 6078 6079
}

void OBSBasic::OpenSourceWindow()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

	OBSSource source = obs_sceneitem_get_source(item);
6080
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
6081 6082 6083

	OpenProjector(obs_sceneitem_get_source(item), -1, title,
			ProjectorType::Source);
C
cg2121 已提交
6084 6085
}

S
Shaolin 已提交
6086 6087
void OBSBasic::OpenMultiviewWindow()
{
S
Shaolin 已提交
6088
	OpenProjector(nullptr, -1, QTStr("MultiviewWindowed"),
S
Shaolin 已提交
6089 6090 6091
			ProjectorType::Multiview);
}

C
cg2121 已提交
6092 6093 6094 6095 6096 6097 6098
void OBSBasic::OpenSceneWindow()
{
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

	OBSSource source = obs_scene_get_source(scene);
6099
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
6100 6101 6102

	OpenProjector(obs_scene_get_source(scene), -1, title,
			ProjectorType::Scene);
J
jp9000 已提交
6103
}
6104

C
cg2121 已提交
6105 6106
void OBSBasic::OpenSavedProjectors()
{
6107
	for (SavedProjectorInfo *info : savedProjectorsArray) {
S
Shaolin 已提交
6108
		OBSProjector *projector = nullptr;
6109
		switch (info->type) {
S
Shaolin 已提交
6110
		case ProjectorType::Source:
6111 6112 6113 6114 6115
		case ProjectorType::Scene: {
			OBSSource source = obs_get_source_by_name(
					info->name.c_str());
			if (!source)
				continue;
C
cg2121 已提交
6116

S
Shaolin 已提交
6117 6118 6119 6120
			QString title = nullptr;
			if (info->monitor < 0)
				title = QString::fromUtf8(
						obs_source_get_name(source));
R
Ryan Foster 已提交
6121

S
Shaolin 已提交
6122 6123
			projector = OpenProjector(source, info->monitor, title,
					info->type);
6124 6125 6126

			obs_source_release(source);
			break;
S
Shaolin 已提交
6127
		}
6128
		case ProjectorType::Preview: {
S
Shaolin 已提交
6129
			projector = OpenProjector(nullptr, info->monitor,
6130
					QTStr("PreviewWindow"),
6131
					ProjectorType::Preview);
6132
			break;
S
Shaolin 已提交
6133
		}
6134
		case ProjectorType::StudioProgram: {
S
Shaolin 已提交
6135
			projector = OpenProjector(nullptr, info->monitor,
6136 6137 6138 6139 6140
					QTStr("StudioProgramWindow"),
					ProjectorType::StudioProgram);
			break;
		}
		case ProjectorType::Multiview: {
S
Shaolin 已提交
6141
			projector = OpenProjector(nullptr, info->monitor,
6142
					QTStr("MultiviewWindowed"),
6143
					ProjectorType::Multiview);
6144 6145
			break;
		}
C
cg2121 已提交
6146
		}
S
Shaolin 已提交
6147

6148
		if (projector && !info->geometry.empty() && info->monitor < 0) {
S
Shaolin 已提交
6149 6150 6151
			QByteArray byteArray = QByteArray::fromBase64(
					QByteArray(info->geometry.c_str()));
			projector->restoreGeometry(byteArray);
S
Shaolin 已提交
6152

S
Shaolin 已提交
6153 6154 6155 6156 6157
			if (!WindowPositionValid(projector->normalGeometry())) {
				QRect rect = App()->desktop()->geometry();
				projector->setGeometry(QStyle::alignedRect(
						Qt::LeftToRight,
						Qt::AlignCenter, size(), rect));
C
cg2121 已提交
6158 6159 6160 6161 6162
			}
		}
	}
}

6163 6164 6165 6166 6167 6168 6169 6170 6171 6172
void OBSBasic::on_actionFullscreenInterface_triggered()
{
	if (!fullscreenInterface)
		showFullScreen();
	else
		showNormal();

	fullscreenInterface = !fullscreenInterface;
}

6173 6174 6175 6176
void OBSBasic::UpdateTitleBar()
{
	stringstream name;

J
jp9000 已提交
6177 6178
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "Profile");
J
jp9000 已提交
6179 6180 6181
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

6182 6183 6184 6185 6186
	name << "OBS ";
	if (previewProgramMode)
		name << "Studio ";

	name << App()->GetVersionString();
6187 6188 6189
	if (App()->IsPortableMode())
		name << " - Portable Mode";

J
jp9000 已提交
6190
	name << " - " << Str("TitleBar.Profile") << ": " << profile;
J
jp9000 已提交
6191
	name << " - " << Str("TitleBar.Scenes") << ": " << sceneCollection;
6192 6193 6194

	setWindowTitle(QT_UTF8(name.str().c_str()));
}
J
jp9000 已提交
6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218

int OBSBasic::GetProfilePath(char *path, size_t size, const char *file) const
{
	char profiles_path[512];
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "ProfileDir");
	int ret;

	if (!profile)
		return -1;
	if (!path)
		return -1;
	if (!file)
		file = "";

	ret = GetConfigPath(profiles_path, 512, "obs-studio/basic/profiles");
	if (ret <= 0)
		return ret;

	if (!*file)
		return snprintf(path, size, "%s/%s", profiles_path, profile);

	return snprintf(path, size, "%s/%s/%s", profiles_path, profile, file);
}
6219

J
jp9000 已提交
6220
void OBSBasic::on_resetUI_triggered()
6221
{
6222 6223 6224 6225 6226 6227 6228 6229 6230 6231 6232 6233 6234 6235 6236 6237 6238 6239 6240 6241 6242 6243 6244 6245 6246 6247 6248 6249 6250 6251
	/* prune deleted extra docks */
	for (int i = extraDocks.size() - 1; i >= 0 ; i--) {
		if (!extraDocks[i]) {
			extraDocks.removeAt(i);
		}
	}

	if (extraDocks.size()) {
		QMessageBox::StandardButton button = QMessageBox::question(
				this,
				QTStr("ResetUIWarning.Title"),
				QTStr("ResetUIWarning.Text"));

		if (button == QMessageBox::No)
			return;
	}

	/* undock/hide/center extra docks */
	for (int i = extraDocks.size() - 1; i >= 0 ; i--) {
		if (extraDocks[i]) {
			extraDocks[i]->setVisible(true);
			extraDocks[i]->setFloating(true);
			extraDocks[i]->move(
					frameGeometry().topLeft() +
					rect().center() -
					extraDocks[i]->rect().center());
			extraDocks[i]->setVisible(false);
		}
	}

J
jp9000 已提交
6252
	restoreState(startingDockLayout);
6253

6254
#if QT_VERSION >= QT_VERSION_CHECK(5, 6, 0)
J
jp9000 已提交
6255 6256 6257 6258 6259 6260 6261 6262 6263 6264 6265 6266 6267 6268 6269
	int cx = width();
	int cy = height();

	int cx22_5 = cx * 225 / 1000;
	int cx5 = cx * 5 / 100;

	cy = cy * 225 / 1000;

	int mixerSize = cx - (cx22_5 * 2 + cx5 * 2);

	QList<QDockWidget*> docks {
		ui->scenesDock,
		ui->sourcesDock,
		ui->mixerDock,
		ui->transitionsDock,
6270
		ui->controlsDock
J
jp9000 已提交
6271 6272 6273 6274 6275 6276 6277 6278 6279 6280 6281 6282 6283 6284 6285
	};

	QList<int> sizes {
		cx22_5,
		cx22_5,
		mixerSize,
		cx5,
		cx5
	};

	ui->scenesDock->setVisible(true);
	ui->sourcesDock->setVisible(true);
	ui->mixerDock->setVisible(true);
	ui->transitionsDock->setVisible(true);
	ui->controlsDock->setVisible(true);
6286 6287
	statsDock->setVisible(false);
	statsDock->setFloating(true);
J
jp9000 已提交
6288 6289 6290

	resizeDocks(docks, {cy, cy, cy, cy, cy}, Qt::Vertical);
	resizeDocks(docks, sizes, Qt::Horizontal);
6291
#endif
J
jp9000 已提交
6292 6293 6294 6295 6296 6297 6298 6299 6300 6301 6302 6303 6304
}

void OBSBasic::on_lockUI_toggled(bool lock)
{
	QDockWidget::DockWidgetFeatures features = lock
		? QDockWidget::NoDockWidgetFeatures
		: QDockWidget::AllDockWidgetFeatures;

	ui->scenesDock->setFeatures(features);
	ui->sourcesDock->setFeatures(features);
	ui->mixerDock->setFeatures(features);
	ui->transitionsDock->setFeatures(features);
	ui->controlsDock->setFeatures(features);
6305
	statsDock->setFeatures(features);
6306 6307 6308 6309 6310 6311 6312 6313

	for (int i = extraDocks.size() - 1; i >= 0 ; i--) {
		if (!extraDocks[i]) {
			extraDocks.removeAt(i);
		} else {
			extraDocks[i]->setFeatures(features);
		}
	}
6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326 6327 6328 6329 6330 6331
}

void OBSBasic::on_toggleListboxToolbars_toggled(bool visible)
{
	ui->sourcesToolbar->setVisible(visible);
	ui->scenesToolbar->setVisible(visible);

	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"ShowListboxToolbars", visible);
}

void OBSBasic::on_toggleStatusBar_toggled(bool visible)
{
	ui->statusbar->setVisible(visible);

	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"ShowStatusBar", visible);
}
J
jp9000 已提交
6332 6333 6334 6335 6336 6337

void OBSBasic::on_actionLockPreview_triggered()
{
	ui->preview->ToggleLocked();
	ui->actionLockPreview->setChecked(ui->preview->Locked());
}
C
cg2121 已提交
6338

J
Joseph El-Khouri 已提交
6339 6340 6341 6342 6343 6344 6345 6346 6347 6348 6349 6350 6351 6352 6353 6354
void OBSBasic::on_scalingMenu_aboutToShow()
{
	obs_video_info ovi;
	obs_get_video_info(&ovi);

	QAction *action = ui->actionScaleCanvas;
	QString text = QTStr("Basic.MainMenu.Edit.Scale.Canvas");
	text = text.arg(QString::number(ovi.base_width),
			QString::number(ovi.base_height));
	action->setText(text);

	action = ui->actionScaleOutput;
	text = QTStr("Basic.MainMenu.Edit.Scale.Output");
	text = text.arg(QString::number(ovi.output_width),
			QString::number(ovi.output_height));
	action->setText(text);
6355 6356
	action->setVisible(!(ovi.output_width == ovi.base_width &&
			ovi.output_height == ovi.base_height));
J
Joseph El-Khouri 已提交
6357 6358 6359 6360 6361 6362

	UpdatePreviewScalingMenu();
}

void OBSBasic::on_actionScaleWindow_triggered()
{
6363
	ui->preview->SetFixedScaling(false);
J
Joseph El-Khouri 已提交
6364 6365 6366 6367 6368 6369
	ui->preview->ResetScrollingOffset();
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleCanvas_triggered()
{
6370 6371
	ui->preview->SetFixedScaling(true);
	ui->preview->SetScalingLevel(0);
J
Joseph El-Khouri 已提交
6372 6373 6374 6375 6376
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleOutput_triggered()
{
6377 6378 6379 6380 6381 6382 6383 6384 6385 6386
	obs_video_info ovi;
	obs_get_video_info(&ovi);

	ui->preview->SetFixedScaling(true);
	float scalingAmount = float(ovi.output_width) / float(ovi.base_width);
	// log base ZOOM_SENSITIVITY of x = log(x) / log(ZOOM_SENSITIVITY)
	int32_t approxScalingLevel = int32_t(
			round(log(scalingAmount) / log(ZOOM_SENSITIVITY)));
	ui->preview->SetScalingLevel(approxScalingLevel);
	ui->preview->SetScalingAmount(scalingAmount);
J
Joseph El-Khouri 已提交
6387 6388 6389
	emit ui->preview->DisplayResized();
}

C
cg2121 已提交
6390 6391 6392 6393 6394 6395 6396
void OBSBasic::SetShowing(bool showing)
{
	if (!showing && isVisible()) {
		config_set_string(App()->GlobalConfig(),
			"BasicWindow", "geometry",
			saveGeometry().toBase64().constData());

6397 6398 6399 6400 6401 6402 6403 6404 6405
		/* hide all visible child dialogs */
		visDlgPositions.clear();
		if (!visDialogs.isEmpty()) {
			for (QDialog *dlg : visDialogs) {
				visDlgPositions.append(dlg->pos());
				dlg->hide();
			}
		}

6406 6407
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Show"));
C
cg2121 已提交
6408 6409 6410 6411 6412 6413 6414
		QTimer::singleShot(250, this, SLOT(hide()));

		if (previewEnabled)
			EnablePreviewDisplay(false);

		setVisible(false);

6415 6416 6417 6418
#ifdef __APPLE__
		EnableOSXDockIcon(false);
#endif

C
cg2121 已提交
6419
	} else if (showing && !isVisible()) {
6420 6421
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Hide"));
C
cg2121 已提交
6422 6423 6424 6425 6426 6427
		QTimer::singleShot(250, this, SLOT(show()));

		if (previewEnabled)
			EnablePreviewDisplay(true);

		setVisible(true);
6428

6429 6430 6431 6432 6433 6434 6435 6436
#ifdef __APPLE__
		EnableOSXDockIcon(true);
#endif

		/* raise and activate window to ensure it is on top */
		raise();
		activateWindow();

6437 6438 6439 6440 6441 6442 6443 6444 6445
		/* show all child dialogs that was visible earlier */
		if (!visDialogs.isEmpty()) {
			for (int i = 0; i < visDialogs.size(); ++i) {
				QDialog *dlg = visDialogs[i];
				dlg->move(visDlgPositions[i]);
				dlg->show();
			}
		}

6446 6447 6448 6449 6450 6451 6452 6453
		/* Unminimize window if it was hidden to tray instead of task
		 * bar. */
		if (sysTrayMinimizeToTray()) {
			Qt::WindowStates state;
			state  = windowState() & ~Qt::WindowMinimized;
			state |= Qt::WindowActive;
			setWindowState(state);
		}
C
cg2121 已提交
6454 6455 6456
	}
}

6457 6458 6459 6460 6461 6462 6463 6464 6465 6466 6467 6468
void OBSBasic::ToggleShowHide()
{
	bool showing = isVisible();
	if (showing) {
		/* check for modal dialogs */
		EnumDialogs();
		if (!modalDialogs.isEmpty() || !visMsgBoxes.isEmpty())
			return;
	}
	SetShowing(!showing);
}

6469 6470
void OBSBasic::SystemTrayInit()
{
6471 6472
	trayIcon.reset(new QSystemTrayIcon(QIcon(":/res/images/obs.png"),
			this));
C
cg2121 已提交
6473 6474 6475
	trayIcon->setToolTip("OBS Studio");

	showHide = new QAction(QTStr("Basic.SystemTray.Show"),
6476
			trayIcon.data());
C
cg2121 已提交
6477
	sysTrayStream = new QAction(QTStr("Basic.Main.StartStreaming"),
6478
			trayIcon.data());
C
cg2121 已提交
6479
	sysTrayRecord = new QAction(QTStr("Basic.Main.StartRecording"),
6480
			trayIcon.data());
J
jp9000 已提交
6481
	sysTrayReplayBuffer = new QAction(QTStr("Basic.Main.StartReplayBuffer"),
6482
			trayIcon.data());
C
cg2121 已提交
6483
	exit = new QAction(QTStr("Exit"),
6484
			trayIcon.data());
C
cg2121 已提交
6485

6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501
	trayMenu = new QMenu;
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	studioProgramProjector = new QMenu(QTStr("StudioProgramProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));
	AddProjectorMenuMonitors(studioProgramProjector, this,
			SLOT(OpenStudioProgramProjector()));
	trayMenu->addAction(showHide);
	trayMenu->addMenu(previewProjector);
	trayMenu->addMenu(studioProgramProjector);
	trayMenu->addAction(sysTrayStream);
	trayMenu->addAction(sysTrayRecord);
	trayMenu->addAction(sysTrayReplayBuffer);
	trayMenu->addAction(exit);
	trayIcon->setContextMenu(trayMenu);

J
jp9000 已提交
6502 6503
	if (outputHandler && !outputHandler->replayBuffer)
		sysTrayReplayBuffer->setEnabled(false);
6504

6505 6506
	connect(trayIcon.data(),
			SIGNAL(activated(QSystemTrayIcon::ActivationReason)),
C
cg2121 已提交
6507 6508 6509 6510 6511 6512 6513 6514
			this,
			SLOT(IconActivated(QSystemTrayIcon::ActivationReason)));
	connect(showHide, SIGNAL(triggered()),
			this, SLOT(ToggleShowHide()));
	connect(sysTrayStream, SIGNAL(triggered()),
			this, SLOT(on_streamButton_clicked()));
	connect(sysTrayRecord, SIGNAL(triggered()),
			this, SLOT(on_recordButton_clicked()));
6515
	connect(sysTrayReplayBuffer.data(), &QAction::triggered,
J
jp9000 已提交
6516
			this, &OBSBasic::ReplayBufferClicked);
C
cg2121 已提交
6517 6518 6519 6520 6521 6522
	connect(exit, SIGNAL(triggered()),
			this, SLOT(close()));
}

void OBSBasic::IconActivated(QSystemTrayIcon::ActivationReason reason)
{
6523 6524 6525 6526 6527 6528 6529 6530 6531
	// Refresh projector list
	previewProjector->clear();
	studioProgramProjector->clear();
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));
	AddProjectorMenuMonitors(studioProgramProjector, this,
			SLOT(OpenStudioProgramProjector()));

	if (reason == QSystemTrayIcon::Trigger)
C
cg2121 已提交
6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543 6544 6545 6546 6547 6548 6549 6550 6551 6552 6553 6554 6555 6556 6557 6558 6559
		ToggleShowHide();
}

void OBSBasic::SysTrayNotify(const QString &text,
		QSystemTrayIcon::MessageIcon n)
{
	if (QSystemTrayIcon::supportsMessages()) {
		QSystemTrayIcon::MessageIcon icon =
				QSystemTrayIcon::MessageIcon(n);
		trayIcon->showMessage("OBS Studio", text, icon, 10000);
	}
}

void OBSBasic::SystemTray(bool firstStarted)
{
	if (!QSystemTrayIcon::isSystemTrayAvailable())
		return;

	bool sysTrayWhenStarted = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "SysTrayWhenStarted");
	bool sysTrayEnabled = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "SysTrayEnabled");

	if (firstStarted)
		SystemTrayInit();

	if (!sysTrayWhenStarted && !sysTrayEnabled) {
		trayIcon->hide();
C
cg2121 已提交
6560 6561
	} else if ((sysTrayWhenStarted && sysTrayEnabled)
			|| opt_minimize_tray) {
C
cg2121 已提交
6562 6563 6564 6565 6566
		trayIcon->show();
		if (firstStarted) {
			QTimer::singleShot(50, this, SLOT(hide()));
			EnablePreviewDisplay(false);
			setVisible(false);
6567 6568 6569
#ifdef __APPLE__
			EnableOSXDockIcon(false);
#endif
C
cg2121 已提交
6570
			opt_minimize_tray = false;
C
cg2121 已提交
6571 6572 6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584
		}
	} else if (sysTrayEnabled) {
		trayIcon->show();
	} else if (!sysTrayEnabled) {
		trayIcon->hide();
	} else if (!sysTrayWhenStarted && sysTrayEnabled) {
		trayIcon->hide();
	}

	if (isVisible())
		showHide->setText(QTStr("Basic.SystemTray.Hide"));
	else
		showHide->setText(QTStr("Basic.SystemTray.Show"));
}
6585 6586 6587 6588 6589 6590

bool OBSBasic::sysTrayMinimizeToTray()
{
	return config_get_bool(GetGlobalConfig(),
			"BasicWindow", "SysTrayMinimizeToTray");
}
6591 6592 6593 6594 6595 6596

void OBSBasic::on_actionCopySource_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;
J
jp9000 已提交
6597 6598

	on_actionCopyTransform_triggered();
6599 6600 6601 6602 6603 6604 6605

	OBSSource source = obs_sceneitem_get_source(item);

	copyString = obs_source_get_name(source);
	copyVisible = obs_sceneitem_visible(item);

	ui->actionPasteRef->setEnabled(true);
6606 6607 6608 6609 6610 6611

	uint32_t output_flags = obs_source_get_output_flags(source);
	if ((output_flags & OBS_SOURCE_DO_NOT_DUPLICATE) == 0)
		ui->actionPasteDup->setEnabled(true);
	else
		ui->actionPasteDup->setEnabled(false);
6612 6613 6614 6615
}

void OBSBasic::on_actionPasteRef_triggered()
{
J
jp9000 已提交
6616 6617 6618 6619 6620
	/* do not allow duplicate refs of the same group in the same scene */
	OBSScene scene = GetCurrentScene();
	if (!!obs_scene_get_group(scene, copyString))
		return;

6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634 6635 6636 6637 6638 6639 6640 6641 6642 6643 6644 6645 6646 6647
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, false);
	on_actionPasteTransform_triggered();
}

void OBSBasic::on_actionPasteDup_triggered()
{
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, true);
	on_actionPasteTransform_triggered();
}

void OBSBasic::on_actionCopyFilters_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();

	if (!item)
		return;

	OBSSource source = obs_sceneitem_get_source(item);

	copyFiltersString = obs_source_get_name(source);

	ui->actionPasteFilters->setEnabled(true);
}

void OBSBasic::on_actionPasteFilters_triggered()
{
	OBSSource source = obs_get_source_by_name(copyFiltersString);
6648
	obs_source_release(source);
6649

6650
	OBSSceneItem sceneItem = GetCurrentSceneItem();
6651 6652 6653 6654 6655 6656 6657
	OBSSource dstSource = obs_sceneitem_get_source(sceneItem);

	if (source == dstSource)
		return;

	obs_source_copy_filters(dstSource, source);
}
J
jp9000 已提交
6658

6659 6660 6661 6662 6663 6664 6665 6666 6667 6668 6669 6670 6671 6672 6673 6674 6675 6676 6677 6678 6679 6680 6681 6682 6683 6684 6685 6686 6687 6688 6689 6690 6691 6692 6693 6694 6695 6696 6697 6698 6699 6700 6701 6702 6703 6704 6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723 6724 6725 6726 6727 6728 6729 6730 6731 6732 6733 6734 6735 6736 6737 6738 6739 6740 6741 6742 6743 6744 6745 6746 6747 6748 6749 6750 6751 6752 6753 6754 6755 6756 6757 6758 6759 6760 6761 6762 6763 6764 6765 6766 6767 6768 6769 6770 6771 6772 6773 6774 6775 6776 6777 6778 6779 6780 6781 6782 6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793 6794 6795 6796 6797 6798 6799 6800 6801 6802 6803 6804 6805 6806 6807 6808 6809 6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820 6821 6822 6823 6824 6825 6826 6827 6828 6829 6830 6831 6832
static void ConfirmColor(SourceTree *sources, const QColor &color,
		QModelIndexList selectedItems)
{
	for (int x = 0; x < selectedItems.count(); x++) {
		SourceTreeItem *treeItem = sources
			->GetItemWidget(selectedItems[x].row());
		treeItem->setStyleSheet("background: "
			+ color.name(QColor::HexArgb));
		treeItem->style()->unpolish(treeItem);
		treeItem->style()->polish(treeItem);

		OBSSceneItem sceneItem = sources->Get(
			selectedItems[x].row());
		obs_data_t *privData =
			obs_sceneitem_get_private_settings(sceneItem);
		obs_data_set_int(privData, "color-preset", 1);
		obs_data_set_string(privData, "color",
			QT_TO_UTF8(color.name(
				QColor::HexArgb)));
		obs_data_release(privData);
	}
}

void OBSBasic::ColorChange()
{
	QModelIndexList selectedItems =
		ui->sources->selectionModel()->selectedIndexes();
	QAction *action = qobject_cast<QAction*>(sender());
	QPushButton *colorButton = qobject_cast<QPushButton*>(sender());

	if (selectedItems.count() == 0)
		return;

	if (colorButton) {
		int preset = colorButton->property("bgColor").value<int>();

		for (int x = 0; x < selectedItems.count(); x++) {
			SourceTreeItem *treeItem = ui->sources
					->GetItemWidget(selectedItems[x].row());
			treeItem->setStyleSheet("");
			treeItem->setProperty("bgColor", preset);
			treeItem->style()->unpolish(treeItem);
			treeItem->style()->polish(treeItem);

			OBSSceneItem sceneItem = ui->sources->Get(
					selectedItems[x].row());
			obs_data_t *privData =
				obs_sceneitem_get_private_settings(sceneItem);
			obs_data_set_int(privData, "color-preset", preset + 1);
			obs_data_set_string(privData, "color", "");
			obs_data_release(privData);
		}

		for (int i = 1; i < 9; i++) {
			stringstream button;
			button << "preset" << i;
			QPushButton *cButton = colorButton->parentWidget()
				->findChild<QPushButton *>(button.str().c_str());
			cButton->setStyleSheet("border: 1px solid black");
		}

		colorButton->setStyleSheet("border: 2px solid black");
	} else if (action) {
		int preset = action->property("bgColor").value<int>();

		if (preset == 1) {
			OBSSceneItem curSceneItem = GetCurrentSceneItem();
			SourceTreeItem *curTreeItem =
				GetItemWidgetFromSceneItem(curSceneItem);
			obs_data_t *curPrivData =
				obs_sceneitem_get_private_settings(curSceneItem);

			int oldPreset = obs_data_get_int(
				curPrivData, "color-preset");
			const QString oldSheet = curTreeItem->styleSheet();

			auto liveChangeColor = [=](const QColor &color) {
				if (color.isValid()) {
					curTreeItem->setStyleSheet(
						"background: "
						+ color.name(QColor::HexArgb));
				}
			};

			auto changedColor = [=](const QColor &color) {
				if (color.isValid()) {
					ConfirmColor(ui->sources, color,
						selectedItems);
				}
			};

			auto rejected = [=]() {
				if (oldPreset == 1) {
					curTreeItem->setStyleSheet(oldSheet);
					curTreeItem->setProperty("bgColor", 0);
				} else if (oldPreset == 0) {
					curTreeItem->setStyleSheet(
						"background: none");
					curTreeItem->setProperty("bgColor", 0);
				} else {
					curTreeItem->setStyleSheet("");
					curTreeItem->setProperty("bgColor",
						oldPreset - 1);
				}

				curTreeItem->style()->unpolish(curTreeItem);
				curTreeItem->style()->polish(curTreeItem);
			};

			QColorDialog::ColorDialogOptions options =
				QColorDialog::ShowAlphaChannel;

			const char *oldColor = obs_data_get_string(curPrivData,
					"color");
			const char *customColor = *oldColor != 0 ? oldColor
				: "#55FF0000";
#ifdef __APPLE__
			options |= QColorDialog::DontUseNativeDialog;
#endif

			QColorDialog *colorDialog = new QColorDialog(this);
			colorDialog->setOptions(options);
			colorDialog->setCurrentColor(
					QColor(customColor));
			connect(colorDialog, &QColorDialog::currentColorChanged,
					liveChangeColor);
			connect(colorDialog, &QColorDialog::colorSelected,
					changedColor);
			connect(colorDialog, &QColorDialog::rejected,
					rejected);
			colorDialog->open();

			obs_data_release(curPrivData);
		} else {
			for (int x = 0; x < selectedItems.count(); x++) {
				SourceTreeItem *treeItem = ui->sources
					->GetItemWidget(selectedItems[x].row());
				treeItem->setStyleSheet("background: none");
				treeItem->setProperty("bgColor", preset);
				treeItem->style()->unpolish(treeItem);
				treeItem->style()->polish(treeItem);

				OBSSceneItem sceneItem = ui->sources->Get(
					selectedItems[x].row());
				obs_data_t *privData =
					obs_sceneitem_get_private_settings(
						sceneItem);
				obs_data_set_int(privData, "color-preset",
					preset);
				obs_data_set_string(privData, "color", "");
				obs_data_release(privData);
			}
		}
	}
}

SourceTreeItem *OBSBasic::GetItemWidgetFromSceneItem(
		obs_sceneitem_t *sceneItem)
{
	int i = 0;
	SourceTreeItem *treeItem = ui->sources->GetItemWidget(i);
	OBSSceneItem item = ui->sources->Get(i);
	int64_t id = obs_sceneitem_get_id(sceneItem);
	while (treeItem && obs_sceneitem_get_id(item) != id) {
		i++;
		treeItem = ui->sources->GetItemWidget(i);
		item = ui->sources->Get(i);
	}
	if(treeItem)
		return treeItem;

	return nullptr;
}

J
jp9000 已提交
6833 6834 6835 6836 6837 6838 6839
void OBSBasic::on_autoConfigure_triggered()
{
	AutoConfig test(this);
	test.setModal(true);
	test.show();
	test.exec();
}
J
jp9000 已提交
6840 6841 6842

void OBSBasic::on_stats_triggered()
{
6843 6844 6845 6846 6847 6848
	if (!stats.isNull()) {
		stats->show();
		stats->raise();
		return;
	}

J
jp9000 已提交
6849 6850 6851 6852 6853
	OBSBasicStats *statsDlg;
	statsDlg = new OBSBasicStats(nullptr);
	statsDlg->show();
	stats = statsDlg;
}
6854

C
cg2121 已提交
6855 6856 6857 6858 6859 6860 6861 6862 6863 6864 6865
void OBSBasic::on_actionShowAbout_triggered()
{
	if (about)
		about->close();

	about = new OBSAbout(this);
	about->show();

	about->setAttribute(Qt::WA_DeleteOnClose, true);
}

6866 6867 6868 6869 6870 6871 6872 6873 6874 6875 6876 6877 6878 6879 6880 6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894 6895 6896 6897
void OBSBasic::ResizeOutputSizeOfSource()
{
	if (ui->streamButton->isChecked() || ui->recordButton->isChecked() ||
			(replayBufferButton && replayBufferButton->isChecked()))
		return;

	QMessageBox resize_output(this);
	resize_output.setText(QTStr("ResizeOutputSizeOfSource.Text") +
			"\n\n" + QTStr("ResizeOutputSizeOfSource.Continue"));
	QAbstractButton *Yes = resize_output.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
	resize_output.addButton(QTStr("No"), QMessageBox::NoRole);
	resize_output.setIcon(QMessageBox::Warning);
	resize_output.setWindowTitle(QTStr("ResizeOutputSizeOfSource"));
	resize_output.exec();

	if (resize_output.clickedButton() != Yes)
		return;

	OBSSource source = obs_sceneitem_get_source(GetCurrentSceneItem());

	int width = obs_source_get_width(source);
	int height = obs_source_get_height(source);

	config_set_uint(basicConfig, "Video", "BaseCX", width);
	config_set_uint(basicConfig, "Video", "BaseCY", height);
	config_set_uint(basicConfig, "Video", "OutputCX", width);
	config_set_uint(basicConfig, "Video", "OutputCY", height);

	ResetVideo();
	on_actionFitToScreen_triggered();
}
6898

6899 6900 6901 6902 6903 6904 6905 6906 6907 6908 6909 6910 6911 6912 6913 6914 6915 6916 6917 6918 6919 6920 6921 6922
QAction *OBSBasic::AddDockWidget(QDockWidget *dock)
{
	QAction *action = ui->viewMenuDocks->addAction(dock->windowTitle());
	action->setCheckable(true);
	assignDockToggle(dock, action);
	extraDocks.push_back(dock);

	bool lock = ui->lockUI->isChecked();
	QDockWidget::DockWidgetFeatures features = lock
		? QDockWidget::NoDockWidgetFeatures
		: QDockWidget::AllDockWidgetFeatures;

	dock->setFeatures(features);

	/* prune deleted docks */
	for (int i = extraDocks.size() - 1; i >= 0 ; i--) {
		if (!extraDocks[i]) {
			extraDocks.removeAt(i);
		}
	}

	return action;
}

6923 6924 6925 6926
OBSBasic *OBSBasic::Get()
{
	return reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
}