window-basic-main.cpp 188.2 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
#endif

J
jp9000 已提交
81 82
#include "ui-config.h"

J
jp9000 已提交
83
struct QCef;
84 85 86 87 88 89
struct QCefCookieManager;

QCef              *cef           = nullptr;
QCefCookieManager *panel_cookies = nullptr;

void DestroyPanelCookieManager();
J
jp9000 已提交
90

J
jp9000 已提交
91 92 93 94 95 96 97 98 99 100
namespace {

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

}

101 102
extern volatile long insideEventLoop;

J
jp9000 已提交
103 104
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
105
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
106
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
107
Q_DECLARE_METATYPE(SignalContainer<OBSScene>);
J
jp9000 已提交
108

P
Palana 已提交
109 110 111 112 113 114 115 116 117 118 119 120 121
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));
}

122 123
static void AddExtraModulePaths()
{
124
	char base_module_dir[512];
125 126 127 128
#if defined(_WIN32) || defined(__APPLE__)
	int ret = GetProgramDataPath(base_module_dir, sizeof(base_module_dir),
			"obs-studio/plugins/%module%");
#else
129
	int ret = GetConfigPath(base_module_dir, sizeof(base_module_dir),
130
			"obs-studio/plugins/%module%");
131
#endif
B
BtbN 已提交
132

133
	if (ret <= 0)
134 135 136
		return;

	string path = (char*)base_module_dir;
137
#if defined(__APPLE__)
138
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
139 140 141 142 143

	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);

144 145 146 147 148 149 150
#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
151 152
}

153 154
extern obs_frontend_callbacks *InitializeAPIInterface(OBSBasic *main);

155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
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;
}

175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
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);
}

J
jp9000 已提交
196
extern void RegisterTwitchAuth();
J
jp9000 已提交
197
extern void RegisterMixerAuth();
S
SoftArch 已提交
198
extern void RegisterRestreamAuth();
J
jp9000 已提交
199

200
OBSBasic::OBSBasic(QWidget *parent)
201
	: OBSMainWindow  (parent),
J
jp9000 已提交
202
	  ui             (new Ui::OBSBasic)
203
{
204 205
	setAttribute(Qt::WA_NativeWindow);

J
jp9000 已提交
206 207 208
#if TWITCH_ENABLED
	RegisterTwitchAuth();
#endif
J
jp9000 已提交
209 210 211
#if MIXER_ENABLED
	RegisterMixerAuth();
#endif
S
SoftArch 已提交
212 213 214
#if RESTREAM_ENABLED
	RegisterRestreamAuth();
#endif
J
jp9000 已提交
215

J
jp9000 已提交
216 217
	setAcceptDrops(true);

218 219
	api = InitializeAPIInterface(this);

220
	ui->setupUi(this);
J
jp9000 已提交
221 222
	ui->previewDisabledLabel->setVisible(false);

J
jp9000 已提交
223 224
	startingDockLayout = saveState();

225
	statsDock = new OBSDock();
226 227 228 229 230 231 232 233
	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 已提交
234
	copyActionsDynamicProperties();
235

236
	char styleSheetPath[512];
J
jp9000 已提交
237 238
	int ret = GetProfilePath(styleSheetPath, sizeof(styleSheetPath),
			"stylesheet.qss");
239
	if (ret > 0) {
H
HomeWorld 已提交
240 241 242 243 244
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
245 246
	}

P
Palana 已提交
247 248 249
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
250
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
251

252 253 254
	qRegisterMetaTypeStreamOperators<
		std::vector<std::shared_ptr<OBSSignal>>>(
				"std::vector<std::shared_ptr<OBSSignal>>");
255
	qRegisterMetaTypeStreamOperators<OBSScene>("OBSScene");
256
	qRegisterMetaTypeStreamOperators<OBSSceneItem>("OBSSceneItem");
257

P
Palana 已提交
258 259 260
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

261
	auto displayResize = [this]() {
262 263 264 265
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
266 267 268 269
	};

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

P
pkv 已提交
271 272 273
	delete shortcutFilter;
	shortcutFilter = CreateShortcutFilter();
	installEventFilter(shortcutFilter);
P
Palana 已提交
274

275
	stringstream name;
J
jp9000 已提交
276
	name << "OBS " << App()->GetVersionString();
277 278 279
	blog(LOG_INFO, "%s", name.str().c_str());
	blog(LOG_INFO, "---------------------------------");

280
	UpdateTitleBar();
J
jp9000 已提交
281 282 283 284 285 286 287 288

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

289
	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
290
	cpuUsageTimer = new QTimer(this);
291
	connect(cpuUsageTimer.data(), SIGNAL(timeout()),
J
jp9000 已提交
292 293
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
294

S
Shaolin 已提交
295 296 297 298 299 300 301 302 303 304 305
	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);

306
#ifdef __APPLE__
S
Shaolin 已提交
307 308 309
	renameScene->setShortcut({Qt::Key_Return});
	renameSource->setShortcut({Qt::Key_Return});

310 311
	ui->actionRemoveSource->setShortcuts({Qt::Key_Backspace});
	ui->actionRemoveScene->setShortcuts({Qt::Key_Backspace});
312 313 314

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
S
Shaolin 已提交
315 316 317
#else
	renameScene->setShortcut({Qt::Key_F2});
	renameSource->setShortcut({Qt::Key_F2});
318
#endif
319 320 321 322 323 324 325 326 327 328 329 330 331 332

	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 已提交
333 334 335 336 337 338

	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);
339
	assignDockToggle(statsDock, ui->toggleStats);
S
SuslikV 已提交
340 341 342 343 344 345 346

	//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 已提交
347
	ui->toggleStats->setChecked(false);
S
SuslikV 已提交
348

349 350
	QPoint curPos;

S
SuslikV 已提交
351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
	//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));
		}
367 368 369 370 371 372

		curPos = pos();
	} else {
		QRect desktopRect = QGuiApplication::primaryScreen()->geometry();
		QSize adjSize = desktopRect.size() / 2 - size() / 2;
		curPos = QPoint(adjSize.width(), adjSize.height());
S
SuslikV 已提交
373
	}
374 375 376 377 378 379

	QPoint curSize(width(), height());
	QPoint statsDockSize(statsDock->width(), statsDock->height());
	QPoint statsDockPos = curSize / 2 - statsDockSize / 2;
	QPoint newPos = curPos + statsDockPos;
	statsDock->move(newPos);
380 381 382 383 384 385 386 387 388 389 390

	ui->previewLabel->setAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
	ui->previewLabel->setProperty("themeID", "previewProgramLabels");

	bool labels = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "StudioModeLabels");

	if (!previewProgramMode)
		ui->previewLabel->setHidden(true);
	else
		ui->previewLabel->setHidden(!labels);
391 392
}

393 394
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent,
		vector<OBSSource> &audioSources)
395
{
396
	obs_source_t *source = obs_get_output_source(channel);
397 398 399
	if (!source)
		return;

400 401
	audioSources.push_back(source);

402
	obs_data_t *data = obs_save_source(source);
403

J
jp9000 已提交
404
	obs_data_set_obj(parent, name, data);
405 406 407 408 409

	obs_data_release(data);
	obs_source_release(source);
}

410 411
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder,
		obs_data_array_t *quickTransitionData, int transitionDuration,
J
jp9000 已提交
412
		obs_data_array_t *transitions,
C
cg2121 已提交
413
		OBSScene &scene, OBSSource &curProgramScene,
414
		obs_data_array_t *savedProjectorList)
415
{
416 417 418 419 420 421 422 423 424 425
	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);
426
	SaveAudioDevice(AUX_AUDIO_4,     6, saveData, audioSources);
427

428 429 430
	/* -------------------------------- */
	/* save non-group sources           */

431 432
	auto FilterAudioSources = [&](obs_source_t *source)
	{
433 434 435
		if (obs_source_is_group(source))
			return false;

436 437 438 439 440 441 442 443 444 445 446
		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));

447 448 449 450 451 452 453 454 455 456 457 458
	/* -------------------------------- */
	/* 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);

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

459 460
	obs_source_t *transition = obs_get_output_source(0);
	obs_source_t *currentScene = obs_scene_get_source(scene);
461
	const char   *sceneName   = obs_source_get_name(currentScene);
462
	const char   *programName = obs_source_get_name(curProgramScene);
463

J
jp9000 已提交
464 465 466
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

J
jp9000 已提交
467
	obs_data_set_string(saveData, "current_scene", sceneName);
468
	obs_data_set_string(saveData, "current_program_scene", programName);
J
jp9000 已提交
469
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
470
	obs_data_set_string(saveData, "name", sceneCollection);
J
jp9000 已提交
471
	obs_data_set_array(saveData, "sources", sourcesArray);
472
	obs_data_set_array(saveData, "groups", groupsArray);
473
	obs_data_set_array(saveData, "quick_transitions", quickTransitionData);
J
jp9000 已提交
474
	obs_data_set_array(saveData, "transitions", transitions);
C
cg2121 已提交
475
	obs_data_set_array(saveData, "saved_projectors", savedProjectorList);
476
	obs_data_array_release(sourcesArray);
477
	obs_data_array_release(groupsArray);
478 479 480 481 482

	obs_data_set_string(saveData, "current_transition",
			obs_source_get_name(transition));
	obs_data_set_int(saveData, "transition_duration", transitionDuration);
	obs_source_release(transition);
483 484 485 486

	return saveData;
}

S
Socapex 已提交
487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506
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 已提交
507 508 509 510 511 512 513 514 515 516
void OBSBasic::UpdateVolumeControlsDecayRate()
{
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");

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

517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
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);
	}
}

540 541
void OBSBasic::ClearVolumeControls()
{
C
craftwar 已提交
542 543
	for (VolControl *vol : volumes)
		delete vol;
544 545 546 547

	volumes.clear();
}

J
jp9000 已提交
548 549 550 551 552 553 554 555 556 557 558 559 560 561 562
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 已提交
563 564
obs_data_array_t *OBSBasic::SaveProjectors()
{
565
	obs_data_array_t *savedProjectors = obs_data_array_create();
C
cg2121 已提交
566

567 568 569
	auto saveProjector = [savedProjectors](OBSProjector *projector) {
		if (!projector)
			return;
R
Ryan Foster 已提交
570 571

		obs_data_t *data = obs_data_create();
572
		ProjectorType type = projector->GetProjectorType();
S
Shaolin 已提交
573 574 575 576 577 578 579 580 581 582 583
		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;
		}
584 585
		obs_data_set_int(data, "monitor", projector->GetMonitor());
		obs_data_set_int(data, "type", static_cast<int>(type));
S
Shaolin 已提交
586 587 588
		obs_data_set_string(data, "geometry",
				projector->saveGeometry().toBase64()
						.constData());
589
		obs_data_array_push_back(savedProjectors, data);
R
Ryan Foster 已提交
590
		obs_data_release(data);
591
	};
R
Ryan Foster 已提交
592

593 594
	for (QPointer<QWidget> &proj : projectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
595

S
Shaolin 已提交
596 597
	for (QPointer<QWidget> &proj : windowProjectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
598

599
	return savedProjectors;
S
Shaolin 已提交
600 601
}

602 603
void OBSBasic::Save(const char *file)
{
604 605 606 607 608
	OBSScene scene = GetCurrentScene();
	OBSSource curProgramScene = OBSGetStrongRef(programScene);
	if (!curProgramScene)
		curProgramScene = obs_scene_get_source(scene);

J
jp9000 已提交
609
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
J
jp9000 已提交
610
	obs_data_array_t *transitions = SaveTransitions();
611
	obs_data_array_t *quickTrData = SaveQuickTransitions();
C
cg2121 已提交
612
	obs_data_array_t *savedProjectorList = SaveProjectors();
R
Ryan Foster 已提交
613
	obs_data_t *saveData = GenerateSaveData(sceneOrder, quickTrData,
J
jp9000 已提交
614
			ui->transitionDuration->value(), transitions,
615
			scene, curProgramScene, savedProjectorList);
616

J
jp9000 已提交
617
	obs_data_set_bool(saveData, "preview_locked", ui->preview->Locked());
618 619 620 621 622 623 624 625
	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 已提交
626

J
jp9000 已提交
627 628 629 630 631 632 633
	if (api) {
		obs_data_t *moduleObj = obs_data_create();
		api->on_save(moduleObj);
		obs_data_set_obj(saveData, "modules", moduleObj);
		obs_data_release(moduleObj);
	}

634 635
	if (!obs_data_save_json_safe(saveData, file, "tmp", "bak"))
		blog(LOG_ERROR, "Could not save scene data to %s", file);
636 637

	obs_data_release(saveData);
J
jp9000 已提交
638
	obs_data_array_release(sceneOrder);
639
	obs_data_array_release(quickTrData);
J
jp9000 已提交
640
	obs_data_array_release(transitions);
C
cg2121 已提交
641
	obs_data_array_release(savedProjectorList);
642 643
}

I
Ilya M 已提交
644 645 646 647 648 649 650 651 652 653 654 655 656
void OBSBasic::DeferSaveBegin()
{
	os_atomic_inc_long(&disableSaving);
}

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

657
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
658
{
659
	obs_data_t *data = obs_data_get_obj(parent, name);
660 661 662
	if (!data)
		return;

663
	obs_source_t *source = obs_load_source(data);
664 665 666 667 668 669 670 671
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

672 673 674
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
675
	obs_properties_t *props = obs_get_source_properties(output_id);
676 677 678 679 680 681 682 683 684 685 686 687 688 689
	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;
}

690
void OBSBasic::CreateFirstRunSources()
691
{
692 693 694 695 696 697 698 699 700
	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);
701 702
}

703
void OBSBasic::CreateDefaultScene(bool firstStart)
704 705 706 707
{
	disableSaving++;

	ClearSceneData();
708 709 710 711
	InitDefaultTransitions();
	CreateDefaultQuickTransitions();
	ui->transitionDuration->setValue(300);
	SetTransition(fadeTransition);
712 713 714

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

715
	if (firstStart)
716
		CreateFirstRunSources();
717

718
	SetCurrentScene(scene, true);
719
	obs_scene_release(scene);
J
jp9000 已提交
720 721

	disableSaving--;
722 723
}

J
jp9000 已提交
724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752
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 已提交
753 754
void OBSBasic::LoadSavedProjectors(obs_data_array_t *array)
{
755 756 757 758 759
	for (SavedProjectorInfo *info : savedProjectorsArray) {
		delete info;
	}
	savedProjectorsArray.clear();

C
cg2121 已提交
760 761 762 763 764
	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);

765 766 767 768
		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 已提交
769 770 771
		info->geometry = std::string(
				obs_data_get_string(data, "geometry"));
		info->name = std::string(obs_data_get_string(data, "name"));
772
		savedProjectorsArray.emplace_back(info);
S
Shaolin 已提交
773 774 775 776 777

		obs_data_release(data);
	}
}

778 779 780 781 782 783 784 785 786 787 788 789 790
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);
}

M
Matt Gajownik 已提交
791
static bool LogSceneItem(obs_scene_t*, obs_sceneitem_t *item, void *v_val)
792 793 794 795
{
	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);
M
Matt Gajownik 已提交
796 797 798 799 800
	int indent_count = (int)(intptr_t)v_val;
	string indent;

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

M
Matt Gajownik 已提交
802
	blog(LOG_INFO, "%s- source: '%s' (%s)", indent.c_str(), name, id);
803

804 805 806 807 808 809 810 811 812
	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";

M
Matt Gajownik 已提交
813
		blog(LOG_INFO, "    %s- monitoring: %s", indent.c_str(), type);
814
	}
M
Matt Gajownik 已提交
815 816 817 818
	int child_indent = 1 + indent_count;
	obs_source_enum_filters(source, LogFilter, (void*)(intptr_t)child_indent);
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, LogSceneItem, (void*)(intptr_t)child_indent);
819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834
	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);
M
Matt Gajownik 已提交
835
		obs_scene_enum_items(scene, LogSceneItem, (void*)(intptr_t)1);
836 837 838 839 840 841
		obs_source_enum_filters(source, LogFilter, (void*)(intptr_t)1);
	}

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

842 843
void OBSBasic::Load(const char *file)
{
844 845 846 847 848
	disableSaving++;

	obs_data_t *data = obs_data_create_from_json_file_safe(file, "bak");
	if (!data) {
		disableSaving--;
849
		blog(LOG_INFO, "No scene file found, creating default scene");
850
		CreateDefaultScene(true);
J
jp9000 已提交
851
		SaveProject();
852
		return;
853
	}
854

855
	ClearSceneData();
856
	InitDefaultTransitions();
857

858 859 860 861
	obs_data_t *modulesObj = obs_data_get_obj(data, "modules");
	if (api)
		api->on_preload(modulesObj);

J
jp9000 已提交
862
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
863
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
864
	obs_data_array_t *groups     = obs_data_get_array(data, "groups");
J
jp9000 已提交
865
	obs_data_array_t *transitions= obs_data_get_array(data, "transitions");
J
jp9000 已提交
866 867
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
868 869 870 871 872
	const char       *programSceneName = obs_data_get_string(data,
			"current_program_scene");
	const char       *transitionName = obs_data_get_string(data,
			"current_transition");

873 874 875 876 877 878
	if (!opt_starting_scene.empty()) {
		programSceneName = opt_starting_scene.c_str();
		if (!IsPreviewProgramMode())
			sceneName = opt_starting_scene.c_str();
	}

879 880 881 882 883 884
	int newDuration = obs_data_get_int(data, "transition_duration");
	if (!newDuration)
		newDuration = 300;

	if (!transitionName)
		transitionName = obs_source_get_name(fadeTransition);
J
jp9000 已提交
885 886 887 888 889 890 891

	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");
892
	obs_source_t     *curScene;
893 894
	obs_source_t     *curProgramScene;
	obs_source_t     *curTransition;
895

J
jp9000 已提交
896 897 898
	if (!name || !*name)
		name = curSceneCollection;

899 900 901 902 903
	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);
904
	LoadAudioDevice(AUX_AUDIO_4,     6, data);
905

906 907 908 909 910 911 912
	if (!sources) {
		sources = groups;
		groups = nullptr;
	} else {
		obs_data_array_push_back_array(sources, groups);
	}

913
	obs_load_sources(sources, nullptr, nullptr);
914

J
jp9000 已提交
915 916
	if (transitions)
		LoadTransitions(transitions);
J
jp9000 已提交
917 918 919
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

J
jp9000 已提交
920 921
	obs_data_array_release(transitions);

922 923 924 925 926 927 928
	curTransition = FindTransition(transitionName);
	if (!curTransition)
		curTransition = fadeTransition;

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

929
retryScene:
930
	curScene = obs_get_source_by_name(sceneName);
931
	curProgramScene = obs_get_source_by_name(programSceneName);
932 933 934 935 936 937 938 939 940 941 942 943 944

	/* 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;
	}

945 946 947 948 949 950 951 952
	if (!curProgramScene) {
		curProgramScene = curScene;
		obs_source_addref(curScene);
	}

	SetCurrentScene(curScene, true);
	if (IsPreviewProgramMode())
		TransitionToScene(curProgramScene, true);
953
	obs_source_release(curScene);
954
	obs_source_release(curProgramScene);
955 956

	obs_data_array_release(sources);
957
	obs_data_array_release(groups);
J
jp9000 已提交
958
	obs_data_array_release(sceneOrder);
J
jp9000 已提交
959

960 961 962 963 964 965 966 967 968
	/* ------------------- */

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

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

969
		if (savedProjectors) {
970
			LoadSavedProjectors(savedProjectors);
971 972 973
			OpenSavedProjectors();
			activateWindow();
		}
974 975 976 977 978 979

		obs_data_array_release(savedProjectors);
	}

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

J
jp9000 已提交
980 981 982 983 984 985 986 987
	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());

988 989 990 991 992 993 994
	obs_data_array_t *quickTransitionData = obs_data_get_array(data,
			"quick_transitions");
	LoadQuickTransitions(quickTransitionData);
	obs_data_array_release(quickTransitionData);

	RefreshQuickTransitions();

J
jp9000 已提交
995 996 997 998
	bool previewLocked = obs_data_get_bool(data, "preview_locked");
	ui->preview->SetLocked(previewLocked);
	ui->actionLockPreview->setChecked(previewLocked);

999 1000 1001 1002 1003 1004 1005 1006 1007 1008
	/* ---------------------- */

	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 已提交
1009
	}
1010
	ui->preview->SetFixedScaling(fixedScaling);
J
Joseph El-Khouri 已提交
1011

1012
	/* ---------------------- */
J
Joseph El-Khouri 已提交
1013

1014
	if (api)
J
jp9000 已提交
1015 1016
		api->on_load(modulesObj);

1017
	obs_data_release(modulesObj);
1018
	obs_data_release(data);
J
jp9000 已提交
1019

1020 1021 1022
	if (!opt_starting_scene.empty())
		opt_starting_scene.clear();

1023
	if (opt_start_streaming) {
1024
		blog(LOG_INFO, "Starting stream due to command line parameter");
1025 1026 1027 1028 1029 1030
		QMetaObject::invokeMethod(this, "StartStreaming",
				Qt::QueuedConnection);
		opt_start_streaming = false;
	}

	if (opt_start_recording) {
1031
		blog(LOG_INFO, "Starting recording due to command line parameter");
1032 1033 1034 1035 1036
		QMetaObject::invokeMethod(this, "StartRecording",
				Qt::QueuedConnection);
		opt_start_recording = false;
	}

C
cg2121 已提交
1037 1038 1039 1040 1041 1042
	if (opt_start_replaybuffer) {
		QMetaObject::invokeMethod(this, "StartReplayBuffer",
				Qt::QueuedConnection);
		opt_start_replaybuffer = false;
	}

1043 1044
	LogScenes();

J
jp9000 已提交
1045
	disableSaving--;
1046

1047
	if (api) {
1048
		api->on_event(OBS_FRONTEND_EVENT_SCENE_CHANGED);
1049 1050
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
	}
1051 1052
}

J
jp9000 已提交
1053
#define SERVICE_PATH "service.json"
1054 1055 1056 1057 1058 1059

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

1060
	char serviceJsonPath[512];
J
jp9000 已提交
1061
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
1062 1063
			SERVICE_PATH);
	if (ret <= 0)
1064 1065
		return;

1066 1067
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
1068

1069
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
1070
	obs_data_set_obj(data, "settings", settings);
1071

1072 1073
	if (!obs_data_save_json_safe(data, serviceJsonPath, "tmp", "bak"))
		blog(LOG_WARNING, "Failed to save service");
1074 1075 1076 1077 1078 1079 1080 1081 1082

	obs_data_release(settings);
	obs_data_release(data);
}

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

1083
	char serviceJsonPath[512];
J
jp9000 已提交
1084
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
1085 1086
			SERVICE_PATH);
	if (ret <= 0)
1087 1088
		return false;

1089 1090
	obs_data_t *data = obs_data_create_from_json_file_safe(serviceJsonPath,
			"bak");
1091 1092

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

1095
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
1096
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
1097

1098
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
1099
			hotkey_data);
1100
	obs_service_release(service);
1101

P
Palana 已提交
1102
	obs_data_release(hotkey_data);
1103 1104 1105 1106 1107 1108 1109 1110
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
P
Palana 已提交
1111 1112
	ProfileScope("OBSBasic::InitService");

1113 1114 1115
	if (LoadService())
		return true;

1116 1117
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
1118 1119
	if (!service)
		return false;
1120
	obs_service_release(service);
1121 1122 1123 1124

	return true;
}

1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140
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
};

1141 1142
extern void CheckExistingCookieId();

1143 1144
bool OBSBasic::InitBasicConfigDefaults()
{
1145
	QList<QScreen*> screens = QGuiApplication::screens();
1146

1147
	if (!screens.size()) {
1148 1149 1150 1151 1152
		OBSErrorBox(NULL, "There appears to be no monitors.  Er, this "
		                  "technically shouldn't be possible.");
		return false;
	}

1153 1154 1155 1156
	QScreen *primaryScreen = QGuiApplication::primaryScreen();

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

1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168
	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 已提交
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183
	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;
	}

1184 1185 1186 1187 1188 1189 1190 1191 1192 1193
	/* ----------------------------------------------------- */
	/* 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 已提交
1194
		changed = true;
1195 1196 1197 1198
	}

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

P
pkviet 已提交
1199 1200 1201 1202 1203
	if (changed)
		config_save_safe(basicConfig, "tmp", nullptr);

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

1204
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
1205

1206 1207
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
1208 1209
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
1210 1211
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
1212
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
1213 1214
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
1215 1216
	config_set_default_bool  (basicConfig, "SimpleOutput", "EnforceBitrate",
			true);
J
jp9000 已提交
1217 1218
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
1219 1220
	config_set_default_string(basicConfig, "SimpleOutput", "NVENCPreset",
			"hq");
1221 1222
	config_set_default_string(basicConfig, "SimpleOutput", "RecQuality",
			"Stream");
1223 1224 1225
	config_set_default_bool(basicConfig, "SimpleOutput", "RecRB", false);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBTime", 20);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBSize", 512);
1226 1227
	config_set_default_string(basicConfig, "SimpleOutput", "RecRBPrefix",
			"Replay");
1228

1229 1230
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
1231 1232 1233 1234 1235 1236 1237 1238
	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());
1239
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
1240 1241
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
1242
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
1243 1244 1245
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

1246 1247 1248 1249 1250
	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 已提交
1251
	config_set_default_uint  (basicConfig, "AdvOut", "FFVBitrate", 2500);
1252
	config_set_default_uint  (basicConfig, "AdvOut", "FFVGOPSize", 250);
J
jp9000 已提交
1253 1254
	config_set_default_bool  (basicConfig, "AdvOut", "FFUseRescale",
			false);
1255 1256
	config_set_default_bool  (basicConfig, "AdvOut", "FFIgnoreCompat",
			false);
J
jp9000 已提交
1257
	config_set_default_uint  (basicConfig, "AdvOut", "FFABitrate", 160);
P
pkviet 已提交
1258
	config_set_default_uint  (basicConfig, "AdvOut", "FFAudioMixes", 1);
J
jp9000 已提交
1259 1260 1261 1262 1263

	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 已提交
1264 1265
	config_set_default_uint  (basicConfig, "AdvOut", "Track5Bitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "Track6Bitrate", 160);
J
jp9000 已提交
1266

1267 1268 1269 1270
	config_set_default_bool  (basicConfig, "AdvOut", "RecRB", false);
	config_set_default_uint  (basicConfig, "AdvOut", "RecRBTime", 20);
	config_set_default_int   (basicConfig, "AdvOut", "RecRBSize", 512);

1271 1272 1273
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

1274 1275 1276 1277 1278 1279 1280 1281
	/* 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);
	}

1282
	config_set_default_string(basicConfig, "Output", "FilenameFormatting",
B
bl 已提交
1283
			"%CCYY-%MM-%DD %hh-%mm-%ss");
1284

1285 1286 1287 1288
	config_set_default_bool  (basicConfig, "Output", "DelayEnable", false);
	config_set_default_uint  (basicConfig, "Output", "DelaySec", 20);
	config_set_default_bool  (basicConfig, "Output", "DelayPreserve", true);

1289 1290 1291 1292
	config_set_default_bool  (basicConfig, "Output", "Reconnect", true);
	config_set_default_uint  (basicConfig, "Output", "RetryDelay", 10);
	config_set_default_uint  (basicConfig, "Output", "MaxRetries", 20);

1293
	config_set_default_string(basicConfig, "Output", "BindIP", "default");
D
derrod 已提交
1294 1295 1296 1297
	config_set_default_bool  (basicConfig, "Output", "NewSocketLoopEnable",
			false);
	config_set_default_bool  (basicConfig, "Output", "LowLatencyEnable",
			false);
1298

1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
	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);
1313

1314 1315 1316 1317 1318 1319 1320 1321 1322
	/* 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);
	}

1323 1324 1325 1326 1327
	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);
1328
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
1329
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
1330
	config_set_default_string(basicConfig, "Video", "ColorSpace", "601");
1331 1332
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
1333

1334 1335 1336 1337 1338
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceId",
			"default");
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceName",
			Str("Basic.Settings.Advanced.Audio.MonitoringDevice"
				".Default"));
1339 1340 1341
	config_set_default_uint  (basicConfig, "Audio", "SampleRate", 44100);
	config_set_default_string(basicConfig, "Audio", "ChannelSetup",
			"Stereo");
S
Shaolin 已提交
1342 1343
	config_set_default_double(basicConfig, "Audio", "MeterDecayRate",
			VOLUME_METER_DECAY_FAST);
1344
	config_set_default_uint  (basicConfig, "Audio", "PeakMeterType", 0);
1345

1346 1347
	CheckExistingCookieId();

1348 1349 1350
	return true;
}

1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
extern bool EncoderAvailable(const char *encoder);

void OBSBasic::InitBasicConfigDefaults2()
{
	bool oldEncDefaults = config_get_bool(App()->GlobalConfig(),
			"General", "Pre23Defaults");
	bool useNV = EncoderAvailable("ffmpeg_nvenc") && !oldEncDefaults;

	config_set_default_string(basicConfig, "SimpleOutput", "StreamEncoder",
			useNV ? SIMPLE_ENCODER_NVENC : SIMPLE_ENCODER_X264);
	config_set_default_string(basicConfig, "SimpleOutput", "RecEncoder",
			useNV ? SIMPLE_ENCODER_NVENC : SIMPLE_ENCODER_X264);
}

1365 1366
bool OBSBasic::InitBasicConfig()
{
P
Palana 已提交
1367 1368
	ProfileScope("OBSBasic::InitBasicConfig");

1369
	char configPath[512];
J
jp9000 已提交
1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382

	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");
1383 1384 1385 1386
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
1387

1388 1389 1390
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
1391 1392 1393
		return false;
	}

J
jp9000 已提交
1394 1395 1396 1397 1398
	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);
1399
		basicConfig.SaveSafe("tmp");
J
jp9000 已提交
1400 1401
	}

1402 1403 1404
	return InitBasicConfigDefaults();
}

1405 1406
void OBSBasic::InitOBSCallbacks()
{
P
Palana 已提交
1407 1408
	ProfileScope("OBSBasic::InitOBSCallbacks");

P
Palana 已提交
1409
	signalHandlers.reserve(signalHandlers.size() + 6);
1410 1411
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_create",
			OBSBasic::SourceCreated, this);
P
Palana 已提交
1412
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_remove",
1413
			OBSBasic::SourceRemoved, this);
P
Palana 已提交
1414
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_activate",
1415
			OBSBasic::SourceActivated, this);
P
Palana 已提交
1416
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_deactivate",
1417
			OBSBasic::SourceDeactivated, this);
P
Palana 已提交
1418
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
1419
			OBSBasic::SourceRenamed, this);
1420 1421
}

J
jp9000 已提交
1422 1423
void OBSBasic::InitPrimitives()
{
P
Palana 已提交
1424 1425
	ProfileScope("OBSBasic::InitPrimitives");

J
jp9000 已提交
1426
	obs_enter_graphics();
J
jp9000 已提交
1427

1428
	gs_render_start(true);
J
jp9000 已提交
1429 1430 1431 1432 1433
	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);
1434
	box = gs_render_save();
J
jp9000 已提交
1435

1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455
	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();

1456
	gs_render_start(true);
J
jp9000 已提交
1457 1458 1459 1460
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
1461
	circle = gs_render_save();
J
jp9000 已提交
1462

J
jp9000 已提交
1463
	obs_leave_graphics();
J
jp9000 已提交
1464 1465
}

J
jp9000 已提交
1466 1467 1468 1469 1470 1471 1472 1473
void OBSBasic::ReplayBufferClicked()
{
	if (outputHandler->ReplayBufferActive())
		StopReplayBuffer();
	else
		StartReplayBuffer();
};

J
jp9000 已提交
1474 1475
void OBSBasic::ResetOutputs()
{
P
Palana 已提交
1476 1477
	ProfileScope("OBSBasic::ResetOutputs");

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

J
jp9000 已提交
1481 1482
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
1483 1484 1485
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
1486

J
jp9000 已提交
1487 1488 1489 1490 1491 1492
		delete replayBufferButton;

		if (outputHandler->replayBuffer) {
			replayBufferButton = new QPushButton(
					QTStr("Basic.Main.StartReplayBuffer"),
					this);
1493
			replayBufferButton->setCheckable(true);
1494
			connect(replayBufferButton.data(),
J
jp9000 已提交
1495 1496 1497 1498
					&QPushButton::clicked,
					this,
					&OBSBasic::ReplayBufferClicked);

1499
			replayBufferButton->setProperty("themeID", "replayBufferButton");
J
jp9000 已提交
1500 1501
			ui->buttonsVLayout->insertWidget(2, replayBufferButton);
		}
1502

J
jp9000 已提交
1503 1504 1505
		if (sysTrayReplayBuffer)
			sysTrayReplayBuffer->setEnabled(
					!!outputHandler->replayBuffer);
J
jp9000 已提交
1506 1507 1508 1509 1510
	} else {
		outputHandler->Update();
	}
}

J
jp9000 已提交
1511 1512 1513
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot);

1514 1515 1516 1517
#define STARTUP_SEPARATOR \
	"==== Startup complete ==============================================="
#define SHUTDOWN_SEPARATOR \
	"==== Shutting down =================================================="
1518

1519 1520 1521 1522 1523
#define UNSUPPORTED_ERROR \
	"Failed to initialize video:\n\nRequired graphics API functionality " \
	"not found.  Your GPU may not be supported."

#define UNKNOWN_ERROR \
1524 1525
	"Failed to initialize video.  Your GPU may not be supported, " \
	"or your graphics drivers may need to be updated."
1526

1527 1528
void OBSBasic::OBSInit()
{
P
Palana 已提交
1529 1530
	ProfileScope("OBSBasic::OBSInit");

J
jp9000 已提交
1531 1532
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
1533
	char savePath[512];
J
jp9000 已提交
1534 1535 1536 1537 1538 1539 1540 1541
	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);
1542
	if (ret <= 0)
J
jp9000 已提交
1543 1544 1545 1546 1547
		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";
1548

1549 1550
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
1551
	if (!ResetAudio())
1552 1553
		throw "Failed to initialize audio";

1554
	ret = ResetVideo();
1555 1556 1557 1558 1559

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
1560
		throw UNSUPPORTED_ERROR;
1561 1562 1563 1564
	case OBS_VIDEO_INVALID_PARAM:
		throw "Failed to initialize video:  Invalid parameters";
	default:
		if (ret != OBS_VIDEO_SUCCESS)
1565
			throw UNKNOWN_ERROR;
1566 1567
	}

1568
	/* load audio monitoring */
1569
#if defined(_WIN32) || defined(__APPLE__) || HAVE_PULSEAUDIO
1570 1571 1572 1573 1574 1575
	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);
1576 1577 1578

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

1581
	InitOBSCallbacks();
P
Palana 已提交
1582
	InitHotkeys();
1583

1584
	AddExtraModulePaths();
1585
	blog(LOG_INFO, "---------------------------------");
J
jp9000 已提交
1586
	obs_load_all_modules();
1587 1588
	blog(LOG_INFO, "---------------------------------");
	obs_log_loaded_modules();
1589 1590
	blog(LOG_INFO, "---------------------------------");
	obs_post_load_modules();
J
jp9000 已提交
1591

1592
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1593
	cef = obs_browser_init_panel();
J
jp9000 已提交
1594 1595
#endif

1596 1597
	InitBasicConfigDefaults2();

1598 1599
	CheckForSimpleModeX264Fallback();

1600
	blog(LOG_INFO, STARTUP_SEPARATOR);
1601

J
jp9000 已提交
1602
	ResetOutputs();
1603
	CreateHotkeys();
J
jp9000 已提交
1604

1605 1606 1607
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
1608 1609
	InitPrimitives();

1610 1611 1612 1613 1614 1615
	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 已提交
1616 1617 1618 1619 1620 1621 1622 1623

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

1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638
#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

1639 1640 1641 1642 1643 1644 1645 1646 1647
#ifndef __APPLE__
	{
		ProfileScope("OBSBasic::Load");
		disableSaving--;
		Load(savePath);
		disableSaving++;
	}
#endif

J
jp9000 已提交
1648
	TimedCheckForUpdates();
1649
	loaded = true;
J
jp9000 已提交
1650

1651
	previewEnabled = config_get_bool(App()->GlobalConfig(),
J
jp9000 已提交
1652
			"BasicWindow", "PreviewEnabled");
1653 1654 1655 1656 1657

	if (!previewEnabled && !IsPreviewProgramMode())
		QMetaObject::invokeMethod(this, "EnablePreviewDisplay",
				Qt::QueuedConnection,
				Q_ARG(bool, previewEnabled));
1658 1659 1660 1661 1662 1663 1664 1665 1666

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

1668 1669 1670
#ifndef __APPLE__
	RefreshSceneCollections();
#endif
J
jp9000 已提交
1671
	RefreshProfiles();
J
jp9000 已提交
1672
	disableSaving--;
1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685

	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);

1686
#ifdef _WIN32
J
jp9000 已提交
1687
	SetWin32DropStyle(this);
1688 1689 1690 1691 1692
	show();
#endif

	bool alwaysOnTop = config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"AlwaysOnTop");
1693
	if (alwaysOnTop || opt_always_on_top) {
1694 1695 1696 1697 1698
		SetAlwaysOnTop(this, true);
		ui->actionAlwaysOnTop->setChecked(true);
	}

#ifndef _WIN32
1699
	show();
1700
#endif
J
jp9000 已提交
1701

A
Alex Anderson 已提交
1702
	/* setup stats dock */
1703 1704
	OBSBasicStats *statsDlg = new OBSBasicStats(statsDock, false);
	statsDock->setWidget(statsDlg);
A
Alex Anderson 已提交
1705

J
jp9000 已提交
1706 1707 1708 1709
	const char *dockStateStr = config_get_string(App()->GlobalConfig(),
			"BasicWindow", "DockState");
	if (!dockStateStr) {
		on_resetUI_triggered();
J
jp9000 已提交
1710
	} else {
J
jp9000 已提交
1711 1712 1713 1714
		QByteArray dockState =
			QByteArray::fromBase64(QByteArray(dockStateStr));
		if (!restoreState(dockState))
			on_resetUI_triggered();
J
jp9000 已提交
1715 1716
	}

1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729
	bool pre23Defaults = config_get_bool(App()->GlobalConfig(),
			"General", "Pre23Defaults");
	if (pre23Defaults) {
		bool resetDockLock23 = config_get_bool(App()->GlobalConfig(),
				"General", "ResetDockLock23");
		if (!resetDockLock23) {
			config_set_bool(App()->GlobalConfig(),
					"General", "ResetDockLock23", true);
			config_remove_value(App()->GlobalConfig(),
					"BasicWindow", "DocksLocked");
			config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
		}
	}
J
jp9000 已提交
1730 1731 1732 1733 1734 1735 1736

	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 已提交
1737

1738
#ifndef __APPLE__
C
cg2121 已提交
1739
	SystemTray(true);
1740
#endif
C
cg2121 已提交
1741

1742 1743 1744
	if (windowState().testFlag(Qt::WindowFullScreen))
		fullscreenInterface = true;

J
jp9000 已提交
1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760
	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 =
1761
			OBSMessageBox::question(this, QTStr("Basic.AutoConfig"),
J
jp9000 已提交
1762 1763 1764
					msg);

		if (button == QMessageBox::Yes) {
J
jp9000 已提交
1765 1766 1767
			QMetaObject::invokeMethod(this,
					"on_autoConfigure_triggered",
					Qt::QueuedConnection);
J
jp9000 已提交
1768 1769
		} else {
			msg = QTStr("Basic.FirstStartup.RunWizard.NoClicked");
1770
			OBSMessageBox::information(this,
J
jp9000 已提交
1771 1772 1773
					QTStr("Basic.AutoConfig"), msg);
		}
	}
1774

1775 1776
	ToggleMixerLayout(config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"VerticalVolControl"));
S
Shaolin 已提交
1777

1778 1779
	if (config_get_bool(basicConfig, "General", "OpenStatsOnStartup"))
		on_stats_triggered();
1780 1781

	OBSBasicStats::InitializeValues();
J
jp9000 已提交
1782 1783 1784 1785 1786 1787

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

	ui->viewMenu->addSeparator();

1788
	multiviewProjectorMenu = new QMenu(QTStr("MultiviewProjector"));
J
jp9000 已提交
1789
	ui->viewMenu->addMenu(multiviewProjectorMenu);
1790 1791
	AddProjectorMenuMonitors(multiviewProjectorMenu, this,
			SLOT(OpenMultiviewProjector()));
1792 1793
	connect(ui->viewMenu->menuAction(), &QAction::hovered, this,
			&OBSBasic::UpdateMultiviewProjectorMenu);
J
jp9000 已提交
1794 1795
	ui->viewMenu->addAction(QTStr("MultiviewWindowed"),
			this, SLOT(OpenMultiviewWindow()));
C
cg2121 已提交
1796 1797

#if !defined(_WIN32) && !defined(__APPLE__)
J
jp9000 已提交
1798 1799 1800
	delete ui->actionShowCrashLogs;
	delete ui->actionUploadLastCrashLog;
	delete ui->menuCrashLogs;
C
cg2121 已提交
1801
	delete ui->actionCheckForUpdates;
J
jp9000 已提交
1802 1803 1804
	ui->actionShowCrashLogs = nullptr;
	ui->actionUploadLastCrashLog = nullptr;
	ui->menuCrashLogs = nullptr;
C
cg2121 已提交
1805 1806
	ui->actionCheckForUpdates = nullptr;
#endif
1807

1808
#ifdef __APPLE__
1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827
	/* 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));
1828 1829
#else
	OnFirstLoad();
1830
#endif
1831 1832
}

1833 1834 1835 1836
void OBSBasic::OnFirstLoad()
{
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_FINISHED_LOADING);
J
jp9000 已提交
1837

1838
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1839
	/* Attempt to load init screen if available */
J
jp9000 已提交
1840
	if (cef) {
J
jp9000 已提交
1841 1842 1843 1844 1845 1846
		WhatsNewInfoThread *wnit = new WhatsNewInfoThread();
		if (wnit) {
			connect(wnit, &WhatsNewInfoThread::Result,
					this, &OBSBasic::ReceivedIntroJson);
		}
		if (wnit) {
1847
			introCheckThread.reset(wnit);
J
jp9000 已提交
1848 1849 1850 1851
			introCheckThread->start();
		}
	}
#endif
J
jp9000 已提交
1852 1853

	Auth::Load();
1854 1855
}

1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866
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));
1867
	RefreshSceneCollections();
1868
	OnFirstLoad();
1869 1870 1871 1872

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

J
jp9000 已提交
1875 1876 1877
/* shows a "what's new" page on startup of new versions using CEF */
void OBSBasic::ReceivedIntroJson(const QString &text)
{
1878
#ifdef BROWSER_AVAILABLE
J
jp9000 已提交
1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893
#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();
1894
		int rc = item["RC"].int_value();
J
jp9000 已提交
1895 1896 1897 1898 1899

		int major = 0;
		int minor = 0;

		sscanf(version.c_str(), "%d.%d", &major, &minor);
1900 1901 1902 1903 1904
#if OBS_RELEASE_CANDIDATE > 0
		if (major == OBS_RELEASE_CANDIDATE_MAJOR &&
		    minor == OBS_RELEASE_CANDIDATE_MINOR &&
		    rc == OBS_RELEASE_CANDIDATE) {
#else
J
jp9000 已提交
1905
		if (major == LIBOBS_API_MAJOR_VER &&
1906 1907 1908
		    minor == LIBOBS_API_MINOR_VER &&
		    rc == 0) {
#endif
J
jp9000 已提交
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918
			info_url = url;
			info_increment = increment;
		}
	}

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

1919 1920 1921 1922
#if OBS_RELEASE_CANDIDATE > 0
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastRCVersion");
#else
J
jp9000 已提交
1923 1924
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastVersion");
1925
#endif
J
jp9000 已提交
1926 1927 1928

	int current_version_increment = -1;

1929 1930 1931
#if OBS_RELEASE_CANDIDATE > 0
	if (lastVersion < OBS_RELEASE_CANDIDATE_VER) {
#else
1932
	if ((lastVersion & ~0xFFFF) < (LIBOBS_API_VER & ~0xFFFF)) {
1933
#endif
J
jp9000 已提交
1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948
		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);

1949 1950 1951 1952 1953 1954 1955 1956 1957
	/* Don't show What's New dialog for new users */
#if !defined(OBS_RELEASE_CANDIDATE) || OBS_RELEASE_CANDIDATE == 0
	if (!lastVersion) {
		return;
	}
#endif
	cef->init_browser();
	ExecuteFuncSafeBlock([] {cef->wait_for_browser_init();});

J
jp9000 已提交
1958 1959 1960 1961
	QDialog *dlg = new QDialog(this);
	dlg->setAttribute(Qt::WA_DeleteOnClose, true);
	dlg->setWindowTitle("What's New");
	dlg->resize(700, 600);
J
jp9000 已提交
1962

1963 1964 1965 1966
	Qt::WindowFlags flags = dlg->windowFlags();
	Qt::WindowFlags helpFlag = Qt::WindowContextHelpButtonHint;
	dlg->setWindowFlags(flags & (~helpFlag));

J
jp9000 已提交
1967
	QCefWidget *cefWidget = cef->create_widget(nullptr, info_url);
J
jp9000 已提交
1968 1969 1970 1971 1972
	if (!cefWidget) {
		return;
	}

	connect(cefWidget, SIGNAL(titleChanged(const QString &)),
J
jp9000 已提交
1973
			dlg, SLOT(setWindowTitle(const QString &)));
J
jp9000 已提交
1974 1975 1976

	QPushButton *close = new QPushButton(QTStr("Close"));
	connect(close, &QAbstractButton::clicked,
J
jp9000 已提交
1977
			dlg, &QDialog::accept);
J
jp9000 已提交
1978 1979 1980 1981 1982 1983

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

J
jp9000 已提交
1984
	QVBoxLayout *topLayout = new QVBoxLayout(dlg);
J
jp9000 已提交
1985 1986 1987
	topLayout->addWidget(cefWidget);
	topLayout->addLayout(bottomLayout);

J
jp9000 已提交
1988
	dlg->show();
J
jp9000 已提交
1989 1990 1991
#else
	UNUSED_PARAMETER(text);
#endif
1992 1993 1994
#else
	UNUSED_PARAMETER(text);
#endif
J
jp9000 已提交
1995 1996
}

1997 1998 1999 2000 2001 2002 2003
void OBSBasic::UpdateMultiviewProjectorMenu()
{
	multiviewProjectorMenu->clear();
	AddProjectorMenuMonitors(multiviewProjectorMenu, this,
			SLOT(OpenMultiviewProjector()));
}

P
Palana 已提交
2004 2005
void OBSBasic::InitHotkeys()
{
P
Palana 已提交
2006 2007
	ProfileScope("OBSBasic::InitHotkeys");

P
Palana 已提交
2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049
	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"),
2050
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070

	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));
}

2071 2072
void OBSBasic::CreateHotkeys()
{
P
Palana 已提交
2073 2074
	ProfileScope("OBSBasic::CreateHotkeys");

2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090
	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 已提交
2091 2092 2093 2094 2095 2096 2097 2098 2099
	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);
	};

2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112
	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);
	};

2113
#define MAKE_CALLBACK(pred, method, log_action) \
2114 2115 2116
	[](void *data, obs_hotkey_pair_id, obs_hotkey_t*, bool pressed) \
	{ \
		OBSBasic &basic = *static_cast<OBSBasic*>(data); \
2117
		if ((pred) && pressed) { \
2118
			blog(LOG_INFO, log_action " due to hotkey"); \
2119 2120 2121 2122 2123 2124 2125 2126
			method(); \
			return true; \
		} \
		return false; \
	}

	streamingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartStreaming",
J
jp9000 已提交
2127
			Str("Basic.Main.StartStreaming"),
2128
			"OBSBasic.StopStreaming",
J
jp9000 已提交
2129
			Str("Basic.Main.StopStreaming"),
2130 2131
			MAKE_CALLBACK(!basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
2132
				basic.StartStreaming, "Starting stream"),
2133 2134
			MAKE_CALLBACK(basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
2135
				basic.StopStreaming, "Stopping stream"),
2136 2137 2138 2139
			this, this);
	LoadHotkeyPair(streamingHotkeys,
			"OBSBasic.StartStreaming", "OBSBasic.StopStreaming");

J
jp9000 已提交
2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154
	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");

2155 2156
	recordingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartRecording",
J
jp9000 已提交
2157
			Str("Basic.Main.StartRecording"),
2158
			"OBSBasic.StopRecording",
J
jp9000 已提交
2159
			Str("Basic.Main.StopRecording"),
2160 2161
			MAKE_CALLBACK(!basic.outputHandler->RecordingActive() &&
				!basic.ui->recordButton->isChecked(),
2162
				basic.StartRecording, "Starting recording"),
2163 2164
			MAKE_CALLBACK(basic.outputHandler->RecordingActive() &&
				basic.ui->recordButton->isChecked(),
2165
				basic.StopRecording, "Stopping recording"),
2166 2167 2168
			this, this);
	LoadHotkeyPair(recordingHotkeys,
			"OBSBasic.StartRecording", "OBSBasic.StopRecording");
J
jp9000 已提交
2169 2170 2171 2172 2173 2174 2175

	replayBufHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartReplayBuffer",
			Str("Basic.Main.StartReplayBuffer"),
			"OBSBasic.StopReplayBuffer",
			Str("Basic.Main.StopReplayBuffer"),
			MAKE_CALLBACK(!basic.outputHandler->ReplayBufferActive(),
2176
				basic.StartReplayBuffer, "Starting replay buffer"),
J
jp9000 已提交
2177
			MAKE_CALLBACK(basic.outputHandler->ReplayBufferActive(),
2178
				basic.StopReplayBuffer, "Stopping replay buffer"),
J
jp9000 已提交
2179 2180 2181
			this, this);
	LoadHotkeyPair(replayBufHotkeys,
			"OBSBasic.StartReplayBuffer", "OBSBasic.StopReplayBuffer");
2182
#undef MAKE_CALLBACK
2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211

	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");
2212 2213
}

J
jp9000 已提交
2214 2215 2216 2217
void OBSBasic::ClearHotkeys()
{
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
J
jp9000 已提交
2218
	obs_hotkey_pair_unregister(replayBufHotkeys);
J
jp9000 已提交
2219
	obs_hotkey_unregister(forceStreamingStopHotkey);
2220 2221
	obs_hotkey_unregister(togglePreviewProgramHotkey);
	obs_hotkey_unregister(transitionHotkey);
J
jp9000 已提交
2222 2223
}

2224 2225
OBSBasic::~OBSBasic()
{
J
jp9000 已提交
2226 2227 2228
	if (updateCheckThread && updateCheckThread->isRunning())
		updateCheckThread->wait();

2229
	delete multiviewProjectorMenu;
P
pkv 已提交
2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242
	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 已提交
2243
	delete trayMenu;
2244 2245
	delete programOptions;
	delete program;
J
jp9000 已提交
2246

J
jp9000 已提交
2247 2248 2249 2250 2251 2252
	/* 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 已提交
2253
	delete cpuUsageTimer;
2254 2255
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
2256
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
J
jp9000 已提交
2257
	ClearHotkeys();
P
Palana 已提交
2258

2259
	service = nullptr;
J
jp9000 已提交
2260 2261
	outputHandler.reset();

J
John Bradley 已提交
2262 2263 2264
	if (interaction)
		delete interaction;

2265 2266 2267
	if (properties)
		delete properties;

J
jp9000 已提交
2268 2269 2270
	if (filters)
		delete filters;

2271 2272
	if (transformWindow)
		delete transformWindow;
2273

J
jp9000 已提交
2274 2275 2276
	if (advAudioWindow)
		delete advAudioWindow;

C
cg2121 已提交
2277 2278 2279
	if (about)
		delete about;

2280 2281 2282
	obs_display_remove_draw_callback(ui->preview->GetDisplay(),
			OBSBasic::RenderMain, this);

J
jp9000 已提交
2283
	obs_enter_graphics();
2284
	gs_vertexbuffer_destroy(box);
2285 2286 2287 2288
	gs_vertexbuffer_destroy(boxLeft);
	gs_vertexbuffer_destroy(boxTop);
	gs_vertexbuffer_destroy(boxRight);
	gs_vertexbuffer_destroy(boxBottom);
2289
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
2290
	obs_leave_graphics();
J
jp9000 已提交
2291

2292 2293 2294 2295 2296 2297 2298 2299 2300
	/* 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 已提交
2301 2302
	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
2303 2304 2305 2306
#if OBS_RELEASE_CANDIDATE > 0
	config_set_int(App()->GlobalConfig(), "General", "LastRCVersion",
			OBS_RELEASE_CANDIDATE_VER);
#endif
J
jp9000 已提交
2307

2308
	bool alwaysOnTop = IsAlwaysOnTop(this);
J
jp9000 已提交
2309

J
jp9000 已提交
2310 2311
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
2312 2313
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "AlwaysOnTop",
			alwaysOnTop);
2314 2315 2316 2317 2318 2319 2320 2321
	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 已提交
2322 2323
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"DocksLocked", ui->lockUI->isChecked());
2324
	config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335

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

#ifdef BROWSER_AVAILABLE
2338
	DestroyPanelCookieManager();
J
jp9000 已提交
2339 2340 2341
	delete cef;
	cef = nullptr;
#endif
2342 2343
}

J
jp9000 已提交
2344 2345 2346 2347 2348 2349 2350 2351 2352
void OBSBasic::SaveProjectNow()
{
	if (disableSaving)
		return;

	projectChanged = true;
	SaveProjectDeferred();
}

J
jp9000 已提交
2353 2354
void OBSBasic::SaveProject()
{
J
jp9000 已提交
2355 2356 2357
	if (disableSaving)
		return;

J
jp9000 已提交
2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372
	projectChanged = true;
	QMetaObject::invokeMethod(this, "SaveProjectDeferred",
			Qt::QueuedConnection);
}

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

	if (!projectChanged)
		return;

	projectChanged = false;

J
jp9000 已提交
2373 2374
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
2375
	char savePath[512];
J
jp9000 已提交
2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387
	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);
2388 2389 2390
	if (ret <= 0)
		return;

J
jp9000 已提交
2391 2392 2393
	Save(savePath);
}

S
Shaolin 已提交
2394 2395 2396 2397 2398
OBSSource OBSBasic::GetProgramSource()
{
	return OBSGetStrongRef(programScene);
}

J
jp9000 已提交
2399
OBSScene OBSBasic::GetCurrentScene()
2400
{
J
jp9000 已提交
2401
	QListWidgetItem *item = ui->scenes->currentItem();
P
Palana 已提交
2402
	return item ? GetOBSRef<OBSScene>(item) : nullptr;
2403 2404
}

2405
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
2406
{
P
Palana 已提交
2407
	return item ? GetOBSRef<OBSSceneItem>(item) : nullptr;
J
jp9000 已提交
2408 2409
}

2410 2411
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
J
jp9000 已提交
2412
	return ui->sources->Get(GetTopSelectedSourceItem());
2413 2414
}

J
Joseph El-Khouri 已提交
2415 2416
void OBSBasic::UpdatePreviewScalingMenu()
{
2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430
	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 已提交
2431
	ui->actionScaleOutput->setChecked(
2432
			scalingAmount == float(ovi.output_width) / float(ovi.base_width));
J
Joseph El-Khouri 已提交
2433 2434
}

2435
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
2436 2437 2438 2439 2440 2441 2442 2443 2444
{
	if (interaction)
		interaction->close();

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

2445
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
2446 2447 2448 2449 2450 2451 2452
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
2455 2456 2457 2458 2459 2460 2461 2462 2463 2464
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

2465 2466 2467
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
2468
{
2469
	const char *name  = obs_source_get_name(source);
2470
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
2471 2472

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

P
Palana 已提交
2476 2477 2478 2479 2480
	obs_hotkey_register_source(source, "OBSBasic.SelectScene",
			Str("Basic.Hotkeys.SelectScene"),
			[](void *data,
				obs_hotkey_id, obs_hotkey_t*, bool pressed)
	{
2481 2482 2483
		OBSBasic *main =
			reinterpret_cast<OBSBasic*>(App()->GetMainWindow());

P
Palana 已提交
2484 2485 2486
		auto potential_source = static_cast<obs_source_t*>(data);
		auto source = obs_source_get_ref(potential_source);
		if (source && pressed)
2487
			main->SetCurrentScene(source);
P
Palana 已提交
2488 2489 2490
		obs_source_release(source);
	}, static_cast<obs_source_t*>(source));

2491
	signal_handler_t *handler = obs_source_get_signal_handler(source);
2492

J
jp9000 已提交
2493 2494 2495
	SignalContainer<OBSScene> container;
	container.ref = scene;
	container.handlers.assign({
2496 2497 2498 2499 2500 2501 2502 2503
		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 已提交
2504
	});
2505 2506

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

2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525
	/* 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 已提交
2526
	SaveProject();
2527 2528 2529 2530 2531

	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 已提交
2532 2533

		OBSProjector::UpdateMultiviewProjectors();
2534
	}
2535 2536 2537

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2538 2539
}

2540
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
2541
{
P
Palana 已提交
2542 2543 2544 2545
	obs_scene_t *scene = obs_scene_from_source(source);

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

P
Palana 已提交
2547 2548 2549 2550 2551
	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 已提交
2552

P
Palana 已提交
2553 2554 2555
		sel = item;
		break;
	}
J
jp9000 已提交
2556

J
jp9000 已提交
2557
	if (sel != nullptr) {
P
Palana 已提交
2558
		if (sel == ui->scenes->currentItem())
J
jp9000 已提交
2559
			ui->sources->Clear();
J
jp9000 已提交
2560
		delete sel;
J
jp9000 已提交
2561
	}
J
jp9000 已提交
2562 2563

	SaveProject();
2564 2565 2566 2567

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

		OBSProjector::UpdateMultiviewProjectors();
2570
	}
2571 2572 2573

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2574 2575
}

J
jp9000 已提交
2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588
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;
}

2589
void OBSBasic::AddSceneItem(OBSSceneItem item)
2590
{
2591
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
J
jp9000 已提交
2592

2593
	if (GetCurrentScene() == scene)
J
jp9000 已提交
2594
		ui->sources->Add(item);
J
jp9000 已提交
2595

J
jp9000 已提交
2596
	SaveProject();
2597 2598 2599 2600 2601 2602 2603 2604

	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 已提交
2605 2606
		
		obs_scene_enum_items(scene, select_one, (obs_sceneitem_t*)item);
2607
	}
2608 2609
}

2610
void OBSBasic::UpdateSceneSelection(OBSSource source)
2611 2612
{
	if (source) {
2613
		obs_scene_t *scene = obs_scene_from_source(source);
2614
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
2615

2616 2617 2618
		if (!scene)
			return;

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

2622 2623 2624 2625 2626
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

2627 2628 2629 2630
			OBSScene curScene =
				GetOBSRef<OBSScene>(ui->scenes->currentItem());
			if (api && scene != curScene)
				api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
2631
		}
J
jp9000 已提交
2632
	}
2633 2634
}

J
jp9000 已提交
2635 2636 2637 2638 2639 2640 2641 2642 2643 2644
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 已提交
2645 2646
void OBSBasic::RenameSources(OBSSource source, QString newName,
		QString prevName)
J
jp9000 已提交
2647 2648
{
	RenameListValues(ui->scenes,  newName, prevName);
2649 2650 2651 2652 2653

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

2655
	OBSProjector::RenameProjector(prevName, newName);
C
cg2121 已提交
2656

J
jp9000 已提交
2657
	SaveProject();
S
Shaolin 已提交
2658 2659 2660 2661

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

2664 2665
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
J
jp9000 已提交
2666 2667
	SignalBlocker sourcesSignalBlocker(ui->sources);

2668
	if (scene != GetCurrentScene() || ignoreSelectionUpdate)
2669 2670
		return;

J
jp9000 已提交
2671
	ui->sources->SelectItem(item, select);
2672 2673
}

2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689
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);
}

2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707
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);
}

2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757
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);
	}
}

2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782
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;
		}

2783 2784
		OBSSource sourceTest = obs_get_source_by_name(name.c_str());
		obs_source_release(sourceTest);
2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797

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

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

2798 2799 2800 2801
void OBSBasic::VolControlContextMenu()
{
	VolControl *vol = reinterpret_cast<VolControl*>(sender());

2802 2803 2804 2805
	/* ------------------- */

	QAction hideAction(QTStr("Hide"), this);
	QAction unhideAllAction(QTStr("UnhideAll"), this);
2806
	QAction mixerRenameAction(QTStr("Rename"), this);
2807

2808 2809 2810
	QAction copyFiltersAction(QTStr("Copy.Filters"), this);
	QAction pasteFiltersAction(QTStr("Paste.Filters"), this);

2811 2812
	QAction filtersAction(QTStr("Filters"), this);
	QAction propertiesAction(QTStr("Properties"), this);
2813
	QAction advPropAction(QTStr("Basic.MainMenu.Edit.AdvAudio"), this);
2814

S
Shaolin 已提交
2815 2816 2817 2818 2819
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

2820 2821 2822 2823 2824 2825 2826 2827
	/* ------------------- */

	connect(&hideAction, &QAction::triggered,
			this, &OBSBasic::HideAudioControl,
			Qt::DirectConnection);
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);
2828 2829 2830
	connect(&mixerRenameAction, &QAction::triggered,
			this, &OBSBasic::MixerRenameSource,
			Qt::DirectConnection);
2831

2832 2833 2834 2835 2836 2837 2838
	connect(&copyFiltersAction, &QAction::triggered,
			this, &OBSBasic::AudioMixerCopyFilters,
			Qt::DirectConnection);
	connect(&pasteFiltersAction, &QAction::triggered,
			this, &OBSBasic::AudioMixerPasteFilters,
			Qt::DirectConnection);

2839 2840 2841 2842 2843 2844
	connect(&filtersAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceFilters,
			Qt::DirectConnection);
	connect(&propertiesAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceProperties,
			Qt::DirectConnection);
2845 2846 2847
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);
2848

2849 2850
	/* ------------------- */

S
Shaolin 已提交
2851 2852 2853 2854 2855 2856
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

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

2857 2858
	hideAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2859 2860
	mixerRenameAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2861

2862 2863 2864 2865 2866
	copyFiltersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
	pasteFiltersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));

2867 2868 2869 2870 2871
	filtersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
	propertiesAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));

2872 2873
	/* ------------------- */

2874 2875 2876 2877 2878
	if (copyFiltersString == nullptr)
		pasteFiltersAction.setEnabled(false);
	else
		pasteFiltersAction.setEnabled(true);

2879
	QMenu popup;
2880 2881
	popup.addAction(&unhideAllAction);
	popup.addAction(&hideAction);
2882
	popup.addAction(&mixerRenameAction);
2883
	popup.addSeparator();
2884 2885 2886
	popup.addAction(&copyFiltersAction);
	popup.addAction(&pasteFiltersAction);
	popup.addSeparator();
S
Shaolin 已提交
2887 2888
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
2889 2890
	popup.addAction(&filtersAction);
	popup.addAction(&propertiesAction);
2891
	popup.addAction(&advPropAction);
2892 2893 2894
	popup.exec(QCursor::pos());
}

S
Shaolin 已提交
2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905
void OBSBasic::on_hMixerScrollArea_customContextMenuRequested()
{
	StackedMixerAreaContextMenuRequested();
}

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

void OBSBasic::StackedMixerAreaContextMenuRequested()
2906 2907
{
	QAction unhideAllAction(QTStr("UnhideAll"), this);
S
SuslikV 已提交
2908 2909 2910

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

S
Shaolin 已提交
2911 2912 2913 2914 2915
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

S
SuslikV 已提交
2916 2917
	/* ------------------- */

2918 2919 2920 2921
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);

S
SuslikV 已提交
2922 2923 2924 2925 2926 2927
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);

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

S
Shaolin 已提交
2928 2929 2930 2931 2932 2933
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

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

2934
	QMenu popup;
2935
	popup.addAction(&unhideAllAction);
S
SuslikV 已提交
2936
	popup.addSeparator();
S
Shaolin 已提交
2937 2938
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
S
SuslikV 已提交
2939
	popup.addAction(&advPropAction);
2940 2941 2942
	popup.exec(QCursor::pos());
}

2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953
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 已提交
2954 2955 2956 2957 2958 2959
void OBSBasic::ToggleVolControlLayout()
{
	bool vertical = !config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	config_set_bool(GetGlobalConfig(), "BasicWindow", "VerticalVolControl",
			vertical);
2960
	ToggleMixerLayout(vertical);
S
Shaolin 已提交
2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973

	// 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);
}

2974 2975
void OBSBasic::ActivateAudioSource(OBSSource source)
{
2976 2977 2978
	if (SourceMixerHidden(source))
		return;

S
Shaolin 已提交
2979 2980 2981
	bool vertical = config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	VolControl *vol = new VolControl(source, true, vertical);
2982

S
Shaolin 已提交
2983 2984 2985
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");
	vol->SetMeterDecayRate(meterDecayRate);
2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004

	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);

3005 3006 3007 3008
	vol->setContextMenuPolicy(Qt::CustomContextMenu);

	connect(vol, &QWidget::customContextMenuRequested,
			this, &OBSBasic::VolControlContextMenu);
3009 3010
	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
3011

3012 3013 3014
	InsertQObjectByName(volumes, vol);

	for (auto volume : volumes) {
S
Shaolin 已提交
3015 3016 3017 3018
		if (vertical)
			ui->vVolControlLayout->addWidget(volume);
		else
			ui->hVolControlLayout->addWidget(volume);
3019
	}
3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032
}

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;
		}
	}
}

3033
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
3034
{
J
jp9000 已提交
3035
	if (obs_source_get_type(source) == OBS_SOURCE_TYPE_SCENE &&
3036
	    !obs_source_is_group(source)) {
3037 3038 3039
		int count = ui->scenes->count();

		if (count == 1) {
3040
			OBSMessageBox::information(this,
3041 3042 3043 3044
						QTStr("FinalScene.Title"),
						QTStr("FinalScene.Text"));
			return false;
		}
3045 3046
	}

3047
	const char *name  = obs_source_get_name(source);
3048 3049 3050

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

3052
	QMessageBox remove_source(this);
3053 3054 3055
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
3056 3057 3058 3059 3060 3061
	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();
3062
}
J
jp9000 已提交
3063

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

P
Palana 已提交
3066 3067 3068 3069 3070
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
3071 3072
void OBSBasic::TimedCheckForUpdates()
{
J
jp9000 已提交
3073 3074 3075 3076
	if (!config_get_bool(App()->GlobalConfig(), "General",
				"EnableAutoUpdates"))
		return;

P
Palana 已提交
3077 3078 3079
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
3080
#elif _WIN32
J
jp9000 已提交
3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095
	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 已提交
3096
		CheckForUpdates(false);
P
Palana 已提交
3097
#endif
J
jp9000 已提交
3098 3099
}

J
jp9000 已提交
3100
void OBSBasic::CheckForUpdates(bool manualUpdate)
J
jp9000 已提交
3101
{
P
Palana 已提交
3102 3103
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
3104
#elif _WIN32
J
jp9000 已提交
3105 3106
	ui->actionCheckForUpdates->setEnabled(false);

J
jp9000 已提交
3107 3108
	if (updateCheckThread && updateCheckThread->isRunning())
		return;
3109

3110
	updateCheckThread.reset(new AutoUpdateThread(manualUpdate));
3111
	updateCheckThread->start();
P
Palana 已提交
3112
#endif
3113 3114

	UNUSED_PARAMETER(manualUpdate);
J
jp9000 已提交
3115 3116
}

J
jp9000 已提交
3117
void OBSBasic::updateCheckFinished()
J
jp9000 已提交
3118 3119 3120 3121
{
	ui->actionCheckForUpdates->setEnabled(true);
}

3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151
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()) {
3152
			OBSMessageBox::information(this,
3153 3154 3155 3156 3157 3158 3159
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
			continue;
		}

		obs_source_t *source = obs_get_source_by_name(name.c_str());
		if (source) {
3160
			OBSMessageBox::information(this,
3161 3162 3163 3164 3165 3166 3167 3168
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));

			obs_source_release(source);
			continue;
		}

		obs_scene_t *scene = obs_scene_duplicate(curScene,
3169
				name.c_str(), OBS_SCENE_DUP_REFS);
3170
		source = obs_scene_get_source(scene);
3171
		SetCurrentScene(source, true);
3172
		obs_scene_release(scene);
J
jp9000 已提交
3173

3174
		break;
3175 3176 3177
	}
}

3178 3179 3180 3181
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
3182
		obs_source_t *source = obs_scene_get_source(scene);
J
jp9000 已提交
3183
		if (QueryRemoveSource(source)) {
3184
			obs_source_remove(source);
J
jp9000 已提交
3185 3186 3187 3188

			if (api)
				api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
		}
3189 3190 3191 3192 3193 3194 3195
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
3196
		obs_source_t *source = obs_sceneitem_get_source(item);
3197
		if (QueryRemoveSource(source))
J
jp9000 已提交
3198 3199 3200 3201
			obs_sceneitem_remove(item);
	}
}

3202 3203
void OBSBasic::ReorderSources(OBSScene scene)
{
3204
	if (scene != GetCurrentScene() || ui->sources->IgnoreReorder())
3205 3206
		return;

J
jp9000 已提交
3207
	ui->sources->ReorderItems();
J
jp9000 已提交
3208
	SaveProject();
3209 3210
}

3211 3212
/* OBS Callbacks */

3213 3214 3215 3216 3217 3218 3219 3220 3221 3222
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)));
}

3223
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
3224 3225 3226
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

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

3229 3230
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
3231 3232
}

3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254
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));
3255

3256 3257
}

3258
void OBSBasic::SourceCreated(void *data, calldata_t *params)
3259
{
3260
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3261

3262
	if (obs_scene_from_source(source) != NULL)
3263
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
3264
				"AddScene", WaitConnection(),
3265
				Q_ARG(OBSSource, OBSSource(source)));
3266 3267
}

3268
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
3269
{
3270
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3271

3272
	if (obs_scene_from_source(source) != NULL)
3273 3274 3275
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
3276 3277
}

3278
void OBSBasic::SourceActivated(void *data, calldata_t *params)
3279
{
3280
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3281 3282 3283 3284 3285 3286 3287 3288
	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)));
}

3289
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
3290
{
3291
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3292 3293 3294 3295 3296 3297 3298 3299
	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)));
}

3300
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
3301
{
S
Shaolin 已提交
3302
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
J
jp9000 已提交
3303 3304 3305 3306 3307
	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 已提交
3308
			Q_ARG(OBSSource, source),
J
jp9000 已提交
3309 3310
			Q_ARG(QString, QT_UTF8(newName)),
			Q_ARG(QString, QT_UTF8(prevName)));
3311 3312

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

3315 3316 3317 3318 3319
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

3320
	gs_effect_t    *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
3321 3322
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
3323 3324 3325

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
3326
	gs_effect_set_vec4(color, &colorVal);
3327

3328 3329
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
3330 3331 3332 3333 3334 3335 3336 3337
	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();
3338 3339
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
3340 3341 3342 3343

	gs_load_vertexbuffer(nullptr);
}

3344 3345
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
3346
	OBSBasic *window = static_cast<OBSBasic*>(data);
3347 3348 3349 3350
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
3351 3352
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
3353 3354 3355

	gs_viewport_push();
	gs_projection_push();
3356

3357 3358 3359 3360 3361 3362 3363 3364 3365 3366
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;

	gs_ortho(-window->previewX, right,
		-window->previewY, bottom,
		-100.0f, 100.0f);

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

3367 3368
	/* --------------------------------------- */

3369 3370
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
3371
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
3372
			window->previewCX, window->previewCY);
3373

3374 3375
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

3376

3377 3378 3379 3380 3381 3382
	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 已提交
3383
		obs_render_main_texture();
3384
	}
3385
	gs_load_vertexbuffer(nullptr);
3386

3387 3388 3389 3390 3391 3392
	/* --------------------------------------- */


	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
3393
	gs_reset_viewport();
J
jp9000 已提交
3394 3395 3396

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

3397 3398
	/* --------------------------------------- */

3399 3400
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
3401 3402 3403

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
3404 3405
}

3406 3407
/* Main class functions */

3408
obs_service_t *OBSBasic::GetService()
3409
{
3410
	if (!service) {
3411 3412
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
3413 3414
		obs_service_release(service);
	}
3415 3416 3417
	return service;
}

3418
void OBSBasic::SetService(obs_service_t *newService)
3419
{
3420
	if (newService)
3421 3422 3423
		service = newService;
}

3424
bool OBSBasic::StreamingActive() const
3425 3426 3427 3428 3429 3430
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

3431 3432 3433 3434 3435 3436 3437
bool OBSBasic::Active() const
{
	if (!outputHandler)
		return false;
	return outputHandler->Active();
}

3438 3439 3440 3441 3442 3443
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

3444 3445
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
3446
	return obs_reset_video(ovi);
3447 3448
}

3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461
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;
}

3462 3463 3464 3465 3466 3467
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 已提交
3468 3469
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
3470 3471 3472 3473 3474 3475 3476 3477 3478
#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 已提交
3479
		return VIDEO_FORMAT_RGBA;
3480 3481
}

3482 3483 3484 3485 3486
void OBSBasic::ResetUI()
{
	bool studioPortraitLayout = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "StudioPortraitLayout");

3487 3488 3489
	bool labels = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "StudioModeLabels");

3490 3491 3492 3493
	if (studioPortraitLayout)
		ui->previewLayout->setDirection(QBoxLayout::TopToBottom);
	else
		ui->previewLayout->setDirection(QBoxLayout::LeftToRight);
3494 3495 3496 3497 3498 3499

	if (previewProgramMode)
		ui->previewLabel->setHidden(!labels);

	if (programLabel)
		programLabel->setHidden(!labels);
3500 3501
}

3502
int OBSBasic::ResetVideo()
J
jp9000 已提交
3503
{
3504 3505 3506
	if (outputHandler && outputHandler->Active())
		return OBS_VIDEO_CURRENTLY_ACTIVE;

P
Palana 已提交
3507 3508
	ProfileScope("OBSBasic::ResetVideo");

J
jp9000 已提交
3509
	struct obs_video_info ovi;
3510
	int ret;
J
jp9000 已提交
3511

3512
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
3513

3514 3515 3516 3517 3518 3519 3520
	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 已提交
3521
	ovi.graphics_module = App()->GetRenderModule();
3522
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3523
			"Video", "BaseCX");
3524
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3525
			"Video", "BaseCY");
3526
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3527
			"Video", "OutputCX");
3528
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3529
			"Video", "OutputCY");
3530 3531 3532 3533 3534
	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;
3535 3536
	ovi.adapter        = config_get_uint(App()->GlobalConfig(),
			"Video", "AdapterIdx");
J
jp9000 已提交
3537
	ovi.gpu_conversion = true;
3538
	ovi.scale_type     = GetScaleType(basicConfig);
3539

3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555
	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);
	}

3556
	ret = AttemptToResetVideo(&ovi);
3557
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
3558 3559 3560 3561 3562 3563
		if (ret == OBS_VIDEO_CURRENTLY_ACTIVE) {
			blog(LOG_WARNING, "Tried to reset when "
			                  "already active");
			return ret;
		}

3564
		/* Try OpenGL if DirectX fails on windows */
3565
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
3566 3567 3568 3569
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
3570 3571
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
3572 3573
			ret = AttemptToResetVideo(&ovi);
		}
3574 3575
	} else if (ret == OBS_VIDEO_SUCCESS) {
		ResizePreview(ovi.base_width, ovi.base_height);
3576 3577
		if (program)
			ResizeProgram(ovi.base_width, ovi.base_height);
3578 3579
	}

3580
	if (ret == OBS_VIDEO_SUCCESS) {
3581
		OBSBasicStats::InitializeValues();
3582 3583
		OBSProjector::UpdateMultiviewProjectors();
	}
3584

3585
	return ret;
J
jp9000 已提交
3586
}
J
jp9000 已提交
3587

3588
bool OBSBasic::ResetAudio()
J
jp9000 已提交
3589
{
P
Palana 已提交
3590 3591
	ProfileScope("OBSBasic::ResetAudio");

3592
	struct obs_audio_info ai;
3593
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
3594 3595
			"SampleRate");

3596
	const char *channelSetupStr = config_get_string(basicConfig,
3597 3598 3599 3600
			"Audio", "ChannelSetup");

	if (strcmp(channelSetupStr, "Mono") == 0)
		ai.speakers = SPEAKERS_MONO;
P
pkviet 已提交
3601 3602
	else if (strcmp(channelSetupStr, "2.1") == 0)
		ai.speakers = SPEAKERS_2POINT1;
P
pkviet 已提交
3603 3604
	else if (strcmp(channelSetupStr, "4.0") == 0)
		ai.speakers = SPEAKERS_4POINT0;
P
pkviet 已提交
3605 3606 3607 3608 3609 3610
	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;
3611 3612 3613
	else
		ai.speakers = SPEAKERS_STEREO;

J
jp9000 已提交
3614
	return obs_reset_audio(&ai);
J
jp9000 已提交
3615 3616
}

3617
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceId,
3618
		const char *deviceDesc, int channel)
J
jp9000 已提交
3619
{
3620
	bool disable = deviceId && strcmp(deviceId, "disabled") == 0;
3621 3622
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
3623 3624 3625

	source = obs_get_output_source(channel);
	if (source) {
3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638
		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 已提交
3639 3640 3641

		obs_source_release(source);

3642 3643
	} else if (!disable) {
		settings = obs_data_create();
J
jp9000 已提交
3644
		obs_data_set_string(settings, "device_id", deviceId);
3645 3646
		source = obs_source_create(sourceId, deviceDesc, settings,
				nullptr);
J
jp9000 已提交
3647 3648 3649 3650 3651 3652 3653
		obs_data_release(settings);

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

J
jp9000 已提交
3654
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
3655
{
3656
	QSize  targetSize;
3657
	bool isFixedScaling;
J
Joseph El-Khouri 已提交
3658
	obs_video_info ovi;
J
jp9000 已提交
3659

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

3663
	isFixedScaling = ui->preview->IsFixedScaling();
J
Joseph El-Khouri 已提交
3664 3665
	obs_get_video_info(&ovi);

3666 3667
	if (isFixedScaling) {
		previewScale = ui->preview->GetScalingAmount();
J
Joseph El-Khouri 已提交
3668 3669 3670 3671
		GetCenterPosFromFixedScale(int(cx), int(cy),
				targetSize.width() - PREVIEW_EDGE_SIZE * 2,
				targetSize.height() - PREVIEW_EDGE_SIZE * 2,
				previewX, previewY, previewScale);
3672 3673
		previewX += ui->preview->GetScrollX();
		previewY += ui->preview->GetScrollY();
J
Joseph El-Khouri 已提交
3674 3675 3676 3677 3678 3679 3680

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

3682 3683
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);
J
jp9000 已提交
3684 3685
}

3686 3687 3688 3689 3690 3691 3692 3693 3694
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 已提交
3695 3696 3697 3698
	for (QPointer<QWidget> &projector : windowProjectors) {
		delete projector;
		projector.clear();
	}
3699 3700 3701 3702
	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}
J
jp9000 已提交
3703

3704
	if (!stats.isNull()) stats->close(); //call close to save Stats geometry
3705
	if (!remux.isNull()) remux->close();
3706 3707
}

3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730
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);
	}
}

3731 3732
void OBSBasic::ClearSceneData()
{
J
jp9000 已提交
3733 3734
	disableSaving++;

3735 3736 3737 3738
	CloseDialogs();

	ClearVolumeControls();
	ClearListItems(ui->scenes);
J
jp9000 已提交
3739
	ui->sources->Clear();
3740 3741
	ClearQuickTransitions();
	ui->transitions->clear();
3742 3743 3744 3745 3746 3747 3748

	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);
3749 3750 3751
	lastScene = nullptr;
	swapScene = nullptr;
	programScene = nullptr;
3752 3753 3754 3755 3756 3757 3758 3759 3760 3761

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

	obs_enum_sources(cb, nullptr);

3762 3763 3764
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CLEANUP);

J
jp9000 已提交
3765
	disableSaving--;
3766 3767 3768

	blog(LOG_INFO, "All scene data cleared");
	blog(LOG_INFO, "------------------------------------------------");
3769 3770
}

J
jp9000 已提交
3771
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
3772
{
3773 3774 3775 3776 3777 3778 3779 3780 3781
	/* Do not close window if inside of a temporary event loop because we
	 * could be inside of an Auth::LoadUI call.  Keep trying once per
	 * second until we've exit any known sub-loops. */
	if (os_atomic_load_long(&insideEventLoop) != 0) {
		QTimer::singleShot(1000, this, SLOT(close()));
		event->ignore();
		return;
	}

3782 3783 3784 3785 3786
	if (isVisible())
		config_set_string(App()->GlobalConfig(),
				"BasicWindow", "geometry",
				saveGeometry().toBase64().constData());

3787
	if (outputHandler && outputHandler->Active()) {
C
cg2121 已提交
3788 3789
		SetShowing(true);

3790
		QMessageBox::StandardButton button = OBSMessageBox::question(
3791 3792 3793 3794 3795 3796 3797 3798 3799
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

3800 3801 3802 3803
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

3804 3805
	blog(LOG_INFO, SHUTDOWN_SEPARATOR);

J
jp9000 已提交
3806 3807
	if (introCheckThread)
		introCheckThread->wait();
3808 3809 3810 3811 3812
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
3813 3814
	signalHandlers.clear();

J
jp9000 已提交
3815
	Auth::Save();
J
jp9000 已提交
3816
	SaveProjectNow();
J
jp9000 已提交
3817 3818 3819 3820 3821
	auth.reset();

	config_set_string(App()->GlobalConfig(),
			"BasicWindow", "DockState",
			saveState().toBase64().constData());
J
jp9000 已提交
3822 3823 3824 3825

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

J
jp9000 已提交
3826
	disableSaving++;
J
jp9000 已提交
3827

3828 3829 3830
	/* Clear all scene data (dialogs, widgets, widget sub-items, scenes,
	 * sources, etc) so that all references are released before shutdown */
	ClearSceneData();
3831 3832

	App()->quit();
3833 3834
}

J
jp9000 已提交
3835
void OBSBasic::changeEvent(QEvent *event)
3836
{
3837 3838
	if (event->type() == QEvent::WindowStateChange &&
	    isMinimized() &&
3839
	    trayIcon &&
3840 3841 3842 3843 3844
	    trayIcon->isVisible() &&
	    sysTrayMinimizeToTray()) {

		ToggleShowHide();
	}
3845 3846
}

3847 3848
void OBSBasic::on_actionShow_Recordings_triggered()
{
3849
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
P
pkviet 已提交
3850 3851 3852 3853
	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");
3854
	const char *path = strcmp(mode, "Advanced") ?
P
pkviet 已提交
3855 3856
			config_get_string(basicConfig, "SimpleOutput", "FilePath") :
			adv_path;
3857 3858 3859
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
3860 3861
void OBSBasic::on_actionRemux_triggered()
{
3862 3863 3864 3865 3866 3867
	if (!remux.isNull()) {
		remux->show();
		remux->raise();
		return;
	}

3868 3869 3870 3871
	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");
3872 3873 3874 3875 3876

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

P
Palana 已提交
3879 3880
void OBSBasic::on_action_Settings_triggered()
{
3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897
	static bool settings_already_executing = false;

	/* Do not load settings window if inside of a temporary event loop
	 * because we could be inside of an Auth::LoadUI call.  Keep trying
	 * once per second until we've exit any known sub-loops. */
	if (os_atomic_load_long(&insideEventLoop) != 0) {
		QTimer::singleShot(1000, this,
				SLOT(on_action_Settings_triggered()));
		return;
	}

	if (settings_already_executing) {
		return;
	}

	settings_already_executing = true;

P
Palana 已提交
3898 3899
	OBSBasicSettings settings(this);
	settings.exec();
C
cg2121 已提交
3900
	SystemTray(false);
3901 3902

	settings_already_executing = false;
P
Palana 已提交
3903 3904
}

J
jp9000 已提交
3905 3906
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
3907 3908 3909 3910 3911
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
3912 3913 3914
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
3915 3916 3917

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

3920 3921 3922 3923 3924
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

3925 3926 3927 3928 3929
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

3930 3931
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
3932
{
3933
	obs_source_t *source = NULL;
J
jp9000 已提交
3934

3935 3936 3937 3938
	if (sceneChanging)
		return;

	if (current) {
3939
		obs_scene_t *scene;
J
jp9000 已提交
3940

P
Palana 已提交
3941
		scene = GetOBSRef<OBSScene>(current);
3942
		source = obs_scene_get_source(scene);
3943 3944
	}

3945
	SetCurrentScene(source);
3946

3947 3948 3949
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);

3950
	UNUSED_PARAMETER(prev);
3951 3952
}

J
jp9000 已提交
3953 3954
void OBSBasic::EditSceneName()
{
3955 3956 3957 3958 3959 3960
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

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

J
jp9000 已提交
3963 3964 3965 3966
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot)
{
	QAction *action;
3967 3968
	QList<QScreen*> screens = QGuiApplication::screens();
	for (int i = 0; i < screens.size(); i++) {
3969
		QRect screenGeometry = screens[i]->geometry();
J
jp9000 已提交
3970 3971
		QString str = QString("%1 %2: %3x%4 @ %5,%6").
			arg(QTStr("Display"),
3972
			    QString::number(i + 1),
S
Shaolin 已提交
3973 3974 3975 3976
			    QString::number(screenGeometry.width()),
			    QString::number(screenGeometry.height()),
			    QString::number(screenGeometry.x()),
			    QString::number(screenGeometry.y()));
J
jp9000 已提交
3977 3978 3979 3980 3981 3982

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

J
jp9000 已提交
3983
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
3984
{
J
jp9000 已提交
3985 3986
	QListWidgetItem *item = ui->scenes->itemAt(pos);

3987
	QMenu popup(this);
J
jp9000 已提交
3988
	QMenu order(QTStr("Basic.MainMenu.Edit.Order"), this);
J
jp9000 已提交
3989 3990 3991
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
3992 3993
	if (item) {
		popup.addSeparator();
3994 3995
		popup.addAction(QTStr("Duplicate"),
				this, SLOT(DuplicateSelectedScene()));
P
Palana 已提交
3996 3997
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
3998
		popup.addAction(QTStr("Remove"),
3999
				this, SLOT(RemoveSelectedScene()));
J
jp9000 已提交
4000
		popup.addSeparator();
J
jp9000 已提交
4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013

		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 已提交
4014

P
pkv 已提交
4015
		delete sceneProjectorMenu;
J
jp9000 已提交
4016 4017 4018 4019
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
C
cg2121 已提交
4020 4021 4022 4023 4024 4025

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

		popup.addAction(sceneWindow);
J
jp9000 已提交
4026
		popup.addSeparator();
J
jp9000 已提交
4027 4028
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
4029 4030 4031

		popup.addSeparator();

P
pkv 已提交
4032 4033 4034
		delete perSceneTransitionMenu;
		perSceneTransitionMenu = CreatePerSceneTransitionMenu();
		popup.addMenu(perSceneTransitionMenu);
S
Shaolin 已提交
4035 4036 4037

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

J
jp9000 已提交
4038 4039
		QAction *multiviewAction = popup.addAction(
				QTStr("ShowInMultiview"));
S
Shaolin 已提交
4040

J
jp9000 已提交
4041 4042 4043
		OBSSource source = GetCurrentSceneSource();
		OBSData data = obs_source_get_private_settings(source);
		obs_data_release(data);
S
Shaolin 已提交
4044

J
jp9000 已提交
4045 4046 4047
		obs_data_set_default_bool(data, "show_in_multiview",
				true);
		bool show = obs_data_get_bool(data, "show_in_multiview");
S
Shaolin 已提交
4048

J
jp9000 已提交
4049 4050
		multiviewAction->setCheckable(true);
		multiviewAction->setChecked(show);
S
Shaolin 已提交
4051

4052
		auto showInMultiview = [] (OBSData data)
J
jp9000 已提交
4053 4054 4055 4056 4057 4058 4059 4060 4061 4062
		{
			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 已提交
4063
	}
J
jp9000 已提交
4064 4065

	popup.exec(QCursor::pos());
4066 4067
}

J
jp9000 已提交
4068
void OBSBasic::on_actionAddScene_triggered()
4069
{
4070
	string name;
S
Socapex 已提交
4071
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
4072

4073
	int i = 2;
P
Palana 已提交
4074
	QString placeHolderText = format.arg(i);
4075
	obs_source_t *source = nullptr;
P
Palana 已提交
4076 4077
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
4078
		placeHolderText = format.arg(++i);
P
Palana 已提交
4079
	}
S
Socapex 已提交
4080

J
jp9000 已提交
4081
	bool accepted = NameDialog::AskForName(this,
4082 4083
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
4084 4085
			name,
			placeHolderText);
4086

J
jp9000 已提交
4087
	if (accepted) {
J
jp9000 已提交
4088
		if (name.empty()) {
4089
			OBSMessageBox::information(this,
4090 4091
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
4092 4093 4094 4095
			on_actionAddScene_triggered();
			return;
		}

4096
		obs_source_t *source = obs_get_source_by_name(name.c_str());
4097
		if (source) {
4098
			OBSMessageBox::information(this,
4099 4100
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
4101 4102

			obs_source_release(source);
J
jp9000 已提交
4103
			on_actionAddScene_triggered();
4104 4105 4106
			return;
		}

4107
		obs_scene_t *scene = obs_scene_create(name.c_str());
4108
		source = obs_scene_get_source(scene);
4109
		SetCurrentScene(source);
4110
		obs_scene_release(scene);
4111
	}
4112 4113
}

J
jp9000 已提交
4114
void OBSBasic::on_actionRemoveScene_triggered()
4115
{
4116
	OBSScene     scene  = GetCurrentScene();
4117
	obs_source_t *source = obs_scene_get_source(scene);
4118 4119 4120

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
4121 4122
}

J
jp9000 已提交
4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140
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;
4141 4142

	OBSProjector::UpdateMultiviewProjectors();
J
jp9000 已提交
4143 4144
}

J
jp9000 已提交
4145
void OBSBasic::on_actionSceneUp_triggered()
4146
{
J
jp9000 已提交
4147
	ChangeSceneIndex(true, -1, 0);
4148 4149
}

J
jp9000 已提交
4150
void OBSBasic::on_actionSceneDown_triggered()
4151
{
J
jp9000 已提交
4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163
	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);
4164 4165
}

J
jp9000 已提交
4166 4167
void OBSBasic::EditSceneItemName()
{
J
jp9000 已提交
4168 4169
	int idx = GetTopSelectedSourceItem();
	ui->sources->Edit(idx);
J
jp9000 已提交
4170 4171
}

J
jp9000 已提交
4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193
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 已提交
4194
QMenu *OBSBasic::AddDeinterlacingMenu(QMenu *menu, obs_source_t *source)
J
jp9000 已提交
4195 4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230 4231 4232 4233 4234 4235
{
	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;
}

4236 4237 4238 4239 4240 4241 4242 4243 4244
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 已提交
4245
QMenu *OBSBasic::AddScaleFilteringMenu(QMenu *menu, obs_sceneitem_t *item)
4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261
{
	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);
J
James Park 已提交
4262
	ADD_MODE("ScaleFiltering.Area",     OBS_SCALE_AREA);
4263 4264 4265 4266 4267
#undef ADD_MODE

	return menu;
}

P
pkv 已提交
4268 4269
QMenu *OBSBasic::AddBackgroundColorMenu(QMenu *menu, QWidgetAction *widgetAction,
		ColorSelect *select, obs_sceneitem_t *item)
4270 4271 4272 4273 4274 4275 4276 4277 4278 4279 4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322
{
	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 已提交
4323 4324 4325 4326 4327 4328 4329
ColorSelect::ColorSelect(QWidget *parent)
	: QWidget(parent),
	ui(new Ui::ColorSelect)
{
	ui->setupUi(this);
}

J
jp9000 已提交
4330
void OBSBasic::CreateSourcePopupMenu(int idx, bool preview)
4331
{
4332
	QMenu popup(this);
P
pkv 已提交
4333 4334 4335 4336 4337 4338 4339
	delete previewProjectorSource;
	delete sourceProjector;
	delete scaleFilteringMenu;
	delete colorMenu;
	delete colorWidgetAction;
	delete colorSelect;
	delete deinterlaceMenu;
J
jp9000 已提交
4340 4341 4342 4343 4344 4345

	if (preview) {
		QAction *action = popup.addAction(
				QTStr("Basic.Main.PreviewConextMenu.Enable"),
				this, SLOT(TogglePreview()));
		action->setCheckable(true);
4346 4347
		action->setChecked(
				obs_display_enabled(ui->preview->GetDisplay()));
4348 4349
		if (IsPreviewProgramMode())
			action->setEnabled(false);
J
jp9000 已提交
4350

J
Joseph El-Khouri 已提交
4351 4352
		popup.addAction(ui->actionLockPreview);
		popup.addMenu(ui->scalingMenu);
J
jp9000 已提交
4353

P
pkv 已提交
4354 4355
		previewProjectorSource = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjectorSource, this,
4356
				SLOT(OpenPreviewProjector()));
J
jp9000 已提交
4357

P
pkv 已提交
4358
		popup.addMenu(previewProjectorSource);
J
jp9000 已提交
4359

C
cg2121 已提交
4360 4361
		QAction *previewWindow = popup.addAction(
				QTStr("PreviewWindow"),
4362
				this, SLOT(OpenPreviewWindow()));
C
cg2121 已提交
4363 4364 4365

		popup.addAction(previewWindow);

J
jp9000 已提交
4366 4367 4368
		popup.addSeparator();
	}

J
jp9000 已提交
4369 4370 4371 4372
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

4373
	ui->actionCopyFilters->setEnabled(false);
4374
	ui->actionCopySource->setEnabled(false);
4375

J
jp9000 已提交
4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386
	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()));
	}

4387 4388 4389 4390 4391 4392 4393 4394 4395 4396 4397
	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 已提交
4398
	if (idx != -1) {
J
jp9000 已提交
4399 4400 4401
		if (addSourceMenu)
			popup.addSeparator();

J
jp9000 已提交
4402
		OBSSceneItem sceneItem = ui->sources->Get(idx);
4403
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
jp9000 已提交
4404 4405 4406
		uint32_t flags = obs_source_get_output_flags(source);
		bool isAsyncVideo = (flags & OBS_SOURCE_ASYNC_VIDEO) ==
			OBS_SOURCE_ASYNC_VIDEO;
4407 4408
		bool hasAudio = (flags & OBS_SOURCE_AUDIO) ==
			OBS_SOURCE_AUDIO;
J
John Bradley 已提交
4409 4410
		QAction *action;

P
pkv 已提交
4411 4412 4413 4414 4415
		colorMenu = new QMenu(QTStr("ChangeBG"));
		colorWidgetAction = new QWidgetAction(colorMenu);
		colorSelect = new ColorSelect(colorMenu);
		popup.addMenu(AddBackgroundColorMenu(colorMenu,
				colorWidgetAction, colorSelect, sceneItem));
J
jp9000 已提交
4416 4417
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
4418
		popup.addAction(QTStr("Remove"), this,
4419
				SLOT(on_actionRemoveSource_triggered()));
J
jp9000 已提交
4420 4421
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
4422
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
4423 4424 4425 4426 4427

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

C
cg2121 已提交
4428 4429 4430 4431 4432 4433
		QAction *sourceWindow = popup.addAction(
				QTStr("SourceWindow"),
				this, SLOT(OpenSourceWindow()));

		popup.addAction(sourceWindow);

J
jp9000 已提交
4434
		popup.addSeparator();
4435 4436 4437 4438 4439 4440 4441 4442 4443

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

J
jp9000 已提交
4444
		if (isAsyncVideo) {
P
pkv 已提交
4445 4446
			deinterlaceMenu = new QMenu(QTStr("Deinterlacing"));
			popup.addMenu(AddDeinterlacingMenu(deinterlaceMenu, source));
J
jp9000 已提交
4447 4448
			popup.addSeparator();
		}
4449

4450 4451 4452 4453 4454 4455 4456
		QAction *resizeOutput = popup.addAction(
				QTStr("ResizeOutputSizeOfSource"), this,
				SLOT(ResizeOutputSizeOfSource()));

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

4457
		resizeOutput->setEnabled(!obs_video_active());
4458 4459 4460 4461

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

P
pkv 已提交
4462 4463
		scaleFilteringMenu = new QMenu(QTStr("ScaleFiltering"));
		popup.addMenu(AddScaleFilteringMenu(scaleFilteringMenu, sceneItem));
4464 4465
		popup.addSeparator();

J
jp9000 已提交
4466
		popup.addMenu(sourceProjector);
C
cg2121 已提交
4467
		popup.addAction(sourceWindow);
J
jp9000 已提交
4468
		popup.addSeparator();
J
John Bradley 已提交
4469 4470 4471 4472 4473 4474 4475

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

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

J
jp9000 已提交
4476 4477
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
4478 4479
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
4480 4481

		ui->actionCopyFilters->setEnabled(true);
4482
		ui->actionCopySource->setEnabled(true);
4483 4484
	} else {
		ui->actionPasteFilters->setEnabled(false);
J
jp9000 已提交
4485 4486 4487
	}

	popup.exec(QCursor::pos());
4488 4489
}

J
jp9000 已提交
4490 4491
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
J
jp9000 已提交
4492 4493 4494 4495
	if (ui->scenes->count()) {
		QModelIndex idx = ui->sources->indexAt(pos);
		CreateSourcePopupMenu(idx.row(), false);
	}
P
Palana 已提交
4496 4497
}

4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515
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 已提交
4516
void OBSBasic::AddSource(const char *id)
4517
{
4518 4519 4520
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
4521
		if (sourceSelect.newSource && strcmp(id, "group") != 0)
4522
			CreatePropertiesWindow(sourceSelect.newSource);
4523
	}
4524 4525
}

4526
QMenu *OBSBasic::CreateAddSourcePopupMenu()
4527
{
4528
	const char *type;
J
jp9000 已提交
4529
	bool foundValues = false;
4530
	bool foundDeprecated = false;
J
jp9000 已提交
4531
	size_t idx = 0;
4532

4533
	QMenu *popup = new QMenu(QTStr("Add"), this);
4534
	QMenu *deprecated = new QMenu(QTStr("Deprecated"), popup);
4535

J
jp9000 已提交
4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552
	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 已提交
4553
		popupItem->setData(QT_UTF8(type));
4554 4555
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4556 4557 4558

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

J
jp9000 已提交
4561 4562
	while (obs_enum_input_types(idx++, &type)) {
		const char *name = obs_source_get_display_name(type);
4563
		uint32_t caps = obs_get_source_output_flags(type);
J
jp9000 已提交
4564

4565 4566 4567
		if ((caps & OBS_SOURCE_CAP_DISABLED) != 0)
			continue;

4568 4569
		if ((caps & OBS_SOURCE_DEPRECATED) == 0) {
			addSource(popup, type, name);
4570 4571 4572
		} else {
			addSource(deprecated, type, name);
			foundDeprecated = true;
4573
		}
4574
		foundValues = true;
4575 4576
	}

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

J
jp9000 已提交
4579 4580
	popup->addSeparator();
	QAction *addGroup = new QAction(QTStr("Group"), this);
4581
	addGroup->setData(QT_UTF8("group"));
J
jp9000 已提交
4582
	connect(addGroup, SIGNAL(triggered(bool)),
4583
			this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4584 4585
	popup->addAction(addGroup);

4586 4587 4588 4589 4590
	if (!foundDeprecated) {
		delete deprecated;
		deprecated = nullptr;
	}

4591 4592 4593
	if (!foundValues) {
		delete popup;
		popup = nullptr;
4594 4595

	} else if (foundDeprecated) {
J
jp9000 已提交
4596
		popup->addSeparator();
4597
		popup->addMenu(deprecated);
4598
	}
4599 4600 4601 4602 4603 4604 4605 4606 4607 4608 4609 4610 4611 4612 4613 4614 4615

	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).
4616
		OBSMessageBox::information(this,
4617 4618 4619 4620 4621
				QTStr("Basic.Main.AddSourceHelp.Title"),
				QTStr("Basic.Main.AddSourceHelp.Text"));
		return;
	}

4622
	QScopedPointer<QMenu> popup(CreateAddSourcePopupMenu());
4623 4624
	if (popup)
		popup->exec(pos);
4625 4626
}

J
jp9000 已提交
4627
void OBSBasic::on_actionAddSource_triggered()
4628
{
J
jp9000 已提交
4629
	AddSourcePopupMenu(QCursor::pos());
4630 4631
}

J
jp9000 已提交
4632 4633 4634 4635 4636 4637 4638 4639 4640 4641 4642 4643 4644
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 已提交
4645
void OBSBasic::on_actionRemoveSource_triggered()
4646
{
4647
	vector<OBSSceneItem> items;
4648

J
jp9000 已提交
4649
	obs_scene_enum_items(GetCurrentScene(), remove_items, &items);
4650 4651 4652 4653 4654 4655 4656 4657 4658 4659 4660 4661 4662 4663 4664 4665 4666 4667 4668 4669 4670 4671 4672 4673 4674 4675 4676 4677 4678 4679 4680 4681 4682

	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);
		}
	}
4683 4684
}

J
John Bradley 已提交
4685 4686 4687 4688 4689 4690 4691 4692 4693
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
4694
void OBSBasic::on_actionSourceProperties_triggered()
4695
{
4696
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4697
	OBSSource source = obs_sceneitem_get_source(item);
4698

4699 4700
	if (source)
		CreatePropertiesWindow(source);
4701 4702
}

J
jp9000 已提交
4703
void OBSBasic::on_actionSourceUp_triggered()
4704
{
J
jp9000 已提交
4705
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4706
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
4707
}
J
jp9000 已提交
4708

J
jp9000 已提交
4709
void OBSBasic::on_actionSourceDown_triggered()
4710
{
J
jp9000 已提交
4711
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4712
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
4713 4714
}

J
jp9000 已提交
4715 4716 4717
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4718
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
4719 4720 4721 4722 4723
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4724
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
4725 4726 4727 4728 4729
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4730
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
4731 4732 4733 4734 4735
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4736
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
4737 4738
}

4739
static BPtr<char> ReadLogFile(const char *subdir, const char *log)
J
jp9000 已提交
4740
{
4741
	char logDir[512];
4742
	if (GetConfigPath(logDir, sizeof(logDir), subdir) <= 0)
4743
		return nullptr;
J
jp9000 已提交
4744 4745 4746 4747 4748

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

4749
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
4750 4751 4752 4753 4754 4755
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

4756
void OBSBasic::UploadLog(const char *subdir, const char *file)
J
jp9000 已提交
4757
{
4758
	BPtr<char> fileString{ReadLogFile(subdir, file)};
J
jp9000 已提交
4759

4760
	if (!fileString)
J
jp9000 已提交
4761 4762
		return;

4763
	if (!*fileString)
J
jp9000 已提交
4764 4765 4766 4767
		return;

	ui->menuLogFiles->setEnabled(false);

4768 4769
	stringstream ss;
	ss << "OBS " << App()->GetVersionString()
4770 4771
	   << " log file uploaded at " << CurrentDateTimeString()
	   << "\n\n" << fileString;
4772

F
fryshorts 已提交
4773

4774 4775 4776
	if (logUploadThread) {
		logUploadThread->wait();
	}
F
fryshorts 已提交
4777

4778
	RemoteTextThread *thread = new RemoteTextThread(
4779
			"https://obsproject.com/logs/upload",
4780 4781
			"text/plain", ss.str().c_str());

4782
	logUploadThread.reset(thread);
4783 4784 4785
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::logUploadFinished);
	logUploadThread->start();
J
jp9000 已提交
4786 4787
}

P
Palana 已提交
4788 4789
void OBSBasic::on_actionShowLogs_triggered()
{
4790
	char logDir[512];
4791
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4792 4793
		return;

P
Palana 已提交
4794 4795 4796 4797
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
4798 4799
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
4800
	UploadLog("obs-studio/logs", App()->GetCurrentLog());
J
jp9000 已提交
4801 4802 4803 4804
}

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

4808 4809 4810
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
4811
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4812 4813 4814 4815 4816 4817 4818 4819 4820 4821 4822 4823
		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 已提交
4824 4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838
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 已提交
4839 4840
void OBSBasic::on_actionCheckForUpdates_triggered()
{
J
jp9000 已提交
4841
	CheckForUpdates(true);
J
jp9000 已提交
4842 4843
}

4844
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
4845 4846 4847
{
	ui->menuLogFiles->setEnabled(true);

4848
	if (text.isEmpty()) {
4849
		OBSMessageBox::information(this,
J
jp9000 已提交
4850
				QTStr("LogReturnDialog.ErrorUploadingLog"),
4851
				error);
J
jp9000 已提交
4852 4853 4854
		return;
	}

4855
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
4856
	string resURL = obs_data_get_string(returnData, "url");
4857
	QString logURL = resURL.c_str();
J
jp9000 已提交
4858 4859 4860 4861 4862 4863
	obs_data_release(returnData);

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

J
jp9000 已提交
4864
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
4865
		obs_source_t *source, const string &name)
4866
{
4867 4868 4869 4870
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

4871
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
4872
	QListWidgetItem *listItem    = listWidget->currentItem();
4873

4874
	if (foundSource || name.empty()) {
4875
		listItem->setText(QT_UTF8(prevName));
4876

4877
		if (foundSource) {
4878
			OBSMessageBox::information(parent,
4879 4880 4881
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
4882
			OBSMessageBox::information(parent,
4883 4884 4885 4886
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

4887 4888 4889
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
4890
		obs_source_set_name(source, name.c_str());
4891 4892 4893
	}
}

J
jp9000 已提交
4894 4895 4896 4897 4898
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
4899
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
4900 4901 4902 4903

	if (!scene)
		return;

4904
	obs_source_t *source = obs_scene_get_source(scene);
4905
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
4906

J
jp9000 已提交
4907 4908 4909
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);

J
jp9000 已提交
4910 4911 4912
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
4913 4914 4915 4916 4917 4918 4919 4920
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4921 4922 4923 4924 4925 4926 4927 4928
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4929 4930 4931 4932
#define RECORDING_START \
	"==== Recording Start ==============================================="
#define RECORDING_STOP \
	"==== Recording Stop ================================================"
J
jp9000 已提交
4933 4934 4935 4936
#define REPLAY_BUFFER_START \
	"==== Replay Buffer Start ==========================================="
#define REPLAY_BUFFER_STOP \
	"==== Replay Buffer Stop ============================================"
J
jp9000 已提交
4937 4938 4939 4940 4941
#define STREAMING_START \
	"==== Streaming Start ==============================================="
#define STREAMING_STOP \
	"==== Streaming Stop ================================================"

4942 4943
void OBSBasic::StartStreaming()
{
J
jp9000 已提交
4944 4945
	if (outputHandler->StreamingActive())
		return;
4946
	if (disableOutputsRef)
4947
		return;
J
jp9000 已提交
4948 4949 4950 4951

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

4952 4953
	SaveProject();

J
jp9000 已提交
4954 4955
	ui->streamButton->setEnabled(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
4956 4957 4958 4959 4960

	if (sysTrayStream) {
		sysTrayStream->setEnabled(false);
		sysTrayStream->setText(ui->streamButton->text());
	}
4961

J
jp9000 已提交
4962 4963 4964
	if (!outputHandler->StartStreaming(service)) {
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
4965
		ui->streamButton->setChecked(false);
4966 4967 4968 4969 4970

		if (sysTrayStream) {
			sysTrayStream->setText(ui->streamButton->text());
			sysTrayStream->setEnabled(true);
		}
4971 4972 4973 4974 4975

		QMessageBox::critical(this,
				QTStr("Output.StartStreamFailed"),
				QTStr("Output.StartFailedGeneric"));
		return;
4976
	}
4977 4978 4979 4980 4981

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	if (recordWhenStreaming)
		StartRecording();
4982 4983 4984 4985 4986

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	if (replayBufferWhileStreaming)
		StartReplayBuffer();
4987 4988
}

4989 4990 4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009
#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

5010
inline void OBSBasic::OnActivate()
5011
{
5012 5013
	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
J
jp9000 已提交
5014
		ui->autoConfigure->setEnabled(false);
5015 5016
		App()->IncrementSleepInhibition();
		UpdateProcessPriority();
C
cg2121 已提交
5017

5018
		if (trayIcon)
5019 5020
			trayIcon->setIcon(QIcon::fromTheme("obs-tray-active",
					QIcon(":/res/images/tray_active.png")));
5021 5022
	}
}
J
jp9000 已提交
5023

5024 5025
inline void OBSBasic::OnDeactivate()
{
5026
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
5027
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
5028
		ui->autoConfigure->setEnabled(true);
5029
		App()->DecrementSleepInhibition();
5030
		ClearProcessPriority();
C
cg2121 已提交
5031

5032
		if (trayIcon)
5033 5034
			trayIcon->setIcon(QIcon::fromTheme("obs-tray",
					QIcon(":/res/images/obs.png")));
J
jp9000 已提交
5035
	}
5036 5037 5038 5039 5040 5041 5042
}

void OBSBasic::StopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
5043
		outputHandler->StopStreaming(streamingStopping);
5044 5045

	OnDeactivate();
5046 5047 5048 5049 5050 5051 5052

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
5053 5054 5055 5056 5057 5058 5059

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	bool keepReplayBufferStreamStops = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "KeepReplayBufferStreamStops");
	if (replayBufferWhileStreaming && !keepReplayBufferStreamStops)
		StopReplayBuffer();
5060 5061
}

J
jp9000 已提交
5062 5063 5064 5065 5066
void OBSBasic::ForceStopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
5067
		outputHandler->StopStreaming(true);
J
jp9000 已提交
5068

5069
	OnDeactivate();
5070 5071 5072 5073 5074 5075 5076

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
5077 5078 5079 5080 5081 5082 5083

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	bool keepReplayBufferStreamStops = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "KeepReplayBufferStreamStops");
	if (replayBufferWhileStreaming && !keepReplayBufferStreamStops)
		StopReplayBuffer();
J
jp9000 已提交
5084 5085 5086 5087 5088 5089
}

void OBSBasic::StreamDelayStarting(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
	ui->streamButton->setEnabled(true);
5090
	ui->streamButton->setChecked(true);
5091 5092 5093 5094 5095

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107

	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);
5108

5109
	OnActivate();
J
jp9000 已提交
5110 5111 5112 5113 5114 5115
}

void OBSBasic::StreamDelayStopping(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
	ui->streamButton->setEnabled(true);
5116
	ui->streamButton->setChecked(false);
5117 5118 5119 5120 5121

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
5122 5123 5124 5125 5126 5127 5128 5129 5130 5131 5132 5133

	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);
5134 5135 5136

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPING);
J
jp9000 已提交
5137 5138
}

5139
void OBSBasic::StreamingStart()
5140
{
5141
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
5142
	ui->streamButton->setEnabled(true);
5143
	ui->streamButton->setChecked(true);
J
jp9000 已提交
5144
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
5145 5146 5147 5148 5149

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
5150

J
jp9000 已提交
5151 5152 5153
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTED);

5154
	OnActivate();
5155

5156
	blog(LOG_INFO, STREAMING_START);
5157 5158
}

5159 5160 5161
void OBSBasic::StreamStopping()
{
	ui->streamButton->setText(QTStr("Basic.Main.StoppingStreaming"));
5162 5163 5164

	if (sysTrayStream)
		sysTrayStream->setText(ui->streamButton->text());
J
jp9000 已提交
5165

5166
	streamingStopping = true;
J
jp9000 已提交
5167 5168
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPING);
5169 5170
}

5171
void OBSBasic::StreamingStop(int code, QString last_error)
5172
{
5173 5174 5175
	const char *errorDescription;
	DStr errorMessage;
	bool use_last_error = false;
5176 5177 5178

	switch (code) {
	case OBS_OUTPUT_BAD_PATH:
5179
		errorDescription = Str("Output.ConnectFail.BadPath");
5180 5181 5182
		break;

	case OBS_OUTPUT_CONNECT_FAILED:
5183 5184
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.ConnectFailed");
5185 5186 5187
		break;

	case OBS_OUTPUT_INVALID_STREAM:
5188
		errorDescription = Str("Output.ConnectFail.InvalidStream");
5189 5190
		break;

5191
	default:
5192
	case OBS_OUTPUT_ERROR:
5193 5194
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Error");
5195 5196 5197 5198 5199
		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 */
5200 5201
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Disconnected");
5202 5203
	}

5204 5205 5206 5207 5208 5209
	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 已提交
5210
	ui->statusbar->StreamStopped();
5211

5212
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
5213
	ui->streamButton->setEnabled(true);
5214
	ui->streamButton->setChecked(false);
5215 5216 5217 5218 5219

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
5220

5221
	streamingStopping = false;
J
jp9000 已提交
5222 5223 5224
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPED);

5225
	OnDeactivate();
5226 5227

	blog(LOG_INFO, STREAMING_STOP);
J
jp9000 已提交
5228

C
cg2121 已提交
5229
	if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5230
		OBSMessageBox::information(this,
5231 5232
				QTStr("Output.ConnectFail.Title"),
				QT_UTF8(errorMessage));
C
cg2121 已提交
5233
	} else if (code != OBS_OUTPUT_SUCCESS && !isVisible()) {
5234
		SysTrayNotify(QT_UTF8(errorDescription), QSystemTrayIcon::Warning);
C
cg2121 已提交
5235
	}
J
jp9000 已提交
5236 5237 5238 5239 5240 5241

	if (!startStreamMenu.isNull()) {
		ui->streamButton->setMenu(nullptr);
		startStreamMenu->deleteLater();
		startStreamMenu = nullptr;
	}
J
jp9000 已提交
5242 5243
}

C
cg2121 已提交
5244 5245 5246
void OBSBasic::AutoRemux()
{
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
5247 5248 5249
	bool advanced = astrcmpi(mode, "Advanced") == 0;

	const char *path = !advanced
5250 5251 5252
		? config_get_string(basicConfig, "SimpleOutput", "FilePath")
		: config_get_string(basicConfig, "AdvOut", "RecFilePath");

5253 5254 5255 5256 5257 5258 5259 5260 5261
	/* do not save if using FFmpeg output in advanced output mode */
	if (advanced) {
		const char *type = config_get_string(basicConfig, "AdvOut",
				"RecType");
		if (astrcmpi(type, "FFmpeg") == 0) {
			return;
		}
	}

J
jp9000 已提交
5262 5263 5264 5265 5266 5267 5268
	QString input;
	input += path;
	input += "/";
	input += remuxFilename.c_str();

	QFileInfo fi(remuxFilename.c_str());

5269 5270 5271 5272 5273
	/* do not remux if lossless */
	if (fi.suffix().compare("avi", Qt::CaseInsensitive) == 0) {
		return;
	}

J
jp9000 已提交
5274 5275 5276 5277 5278
	QString output;
	output += path;
	output += "/";
	output += fi.completeBaseName();
	output += ".mp4";
C
cg2121 已提交
5279 5280 5281

	OBSRemux *remux = new OBSRemux(path, this, true);
	remux->show();
J
jp9000 已提交
5282
	remux->AutoRemux(input, output);
C
cg2121 已提交
5283 5284
}

5285 5286
void OBSBasic::StartRecording()
{
5287 5288
	if (outputHandler->RecordingActive())
		return;
5289
	if (disableOutputsRef)
5290
		return;
5291

J
jp9000 已提交
5292 5293 5294
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTING);

5295
	SaveProject();
5296 5297 5298

	if (!outputHandler->StartRecording())
		ui->recordButton->setChecked(false);
5299 5300
}

5301 5302
void OBSBasic::RecordStopping()
{
J
jp9000 已提交
5303
	ui->recordButton->setText(QTStr("Basic.Main.StoppingRecording"));
5304 5305 5306

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
J
jp9000 已提交
5307

5308
	recordingStopping = true;
J
jp9000 已提交
5309 5310
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPING);
5311 5312
}

5313 5314 5315 5316 5317
void OBSBasic::StopRecording()
{
	SaveProject();

	if (outputHandler->RecordingActive())
5318
		outputHandler->StopRecording(recordingStopping);
J
jp9000 已提交
5319

5320
	OnDeactivate();
5321 5322
}

P
Palana 已提交
5323 5324
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
5325
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
J
jp9000 已提交
5326
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
5327
	ui->recordButton->setChecked(true);
5328 5329 5330

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
5331

5332
	recordingStopping = false;
J
jp9000 已提交
5333 5334 5335
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTED);

5336
	OnActivate();
5337

5338
	blog(LOG_INFO, RECORDING_START);
P
Palana 已提交
5339 5340
}

5341
void OBSBasic::RecordingStop(int code)
5342
{
P
Palana 已提交
5343
	ui->statusbar->RecordingStopped();
J
jp9000 已提交
5344
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
5345
	ui->recordButton->setChecked(false);
5346 5347 5348

	if (sysTrayRecord)
		sysTrayRecord->setText(ui->recordButton->text());
J
jp9000 已提交
5349

5350
	blog(LOG_INFO, RECORDING_STOP);
5351

C
cg2121 已提交
5352
	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5353
		OBSMessageBox::information(this,
5354 5355
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));
J
jp9000 已提交
5356

C
cg2121 已提交
5357
	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5358
		OBSMessageBox::information(this,
J
jp9000 已提交
5359 5360 5361
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

C
cg2121 已提交
5362
	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5363
		OBSMessageBox::information(this,
J
jp9000 已提交
5364 5365
				QTStr("Output.RecordError.Title"),
				QTStr("Output.RecordError.Msg"));
C
cg2121 已提交
5366 5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377

	} 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 已提交
5378 5379
	}

J
jp9000 已提交
5380 5381 5382
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPED);

C
cg2121 已提交
5383 5384 5385
	if (remuxAfterRecord)
		AutoRemux();

5386
	OnDeactivate();
5387
}
5388

J
jp9000 已提交
5389 5390 5391 5392 5393 5394 5395 5396 5397
#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;
5398
	if (disableOutputsRef)
5399
		return;
J
jp9000 已提交
5400

5401 5402 5403 5404 5405
	if (!NoSourcesConfirmation()) {
		replayBufferButton->setChecked(false);
		return;
	}

J
jp9000 已提交
5406 5407 5408 5409 5410 5411 5412 5413 5414
	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) {
5415
		OBSMessageBox::information(this,
J
jp9000 已提交
5416 5417
				RP_NO_HOTKEY_TITLE,
				RP_NO_HOTKEY_TEXT);
5418
		replayBufferButton->setChecked(false);
J
jp9000 已提交
5419 5420 5421 5422 5423 5424 5425
		return;
	}

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

	SaveProject();
5426 5427
	if (!outputHandler->StartReplayBuffer())
		replayBufferButton->setChecked(false);
J
jp9000 已提交
5428 5429 5430 5431 5432 5433 5434 5435 5436 5437 5438 5439 5440 5441 5442 5443 5444 5445 5446 5447 5448 5449 5450 5451 5452 5453 5454 5455 5456 5457 5458 5459 5460 5461 5462 5463
}

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"));
5464
	replayBufferButton->setChecked(true);
J
jp9000 已提交
5465 5466 5467 5468 5469 5470 5471 5472 5473 5474 5475 5476 5477

	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);
}

5478 5479 5480 5481 5482 5483 5484 5485 5486 5487 5488 5489 5490 5491
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 已提交
5492 5493 5494 5495 5496 5497
void OBSBasic::ReplayBufferStop(int code)
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StartReplayBuffer"));
5498
	replayBufferButton->setChecked(false);
J
jp9000 已提交
5499 5500 5501 5502 5503 5504 5505

	if (sysTrayReplayBuffer)
		sysTrayReplayBuffer->setText(replayBufferButton->text());

	blog(LOG_INFO, REPLAY_BUFFER_STOP);

	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5506
		OBSMessageBox::information(this,
J
jp9000 已提交
5507 5508 5509 5510
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));

	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5511
		OBSMessageBox::information(this,
J
jp9000 已提交
5512 5513 5514 5515
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5516
		OBSMessageBox::information(this,
J
jp9000 已提交
5517 5518 5519 5520 5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532 5533 5534 5535 5536 5537 5538
				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();
}

5539 5540 5541 5542 5543 5544 5545 5546
bool OBSBasic::NoSourcesConfirmation()
{
	if (CountVideoSources() == 0 && isVisible()) {
		QString msg;
		msg = QTStr("NoSources.Text");
		msg += "\n\n";
		msg += QTStr("NoSources.Text.AddSource");

5547 5548 5549 5550 5551 5552 5553 5554 5555 5556
		QMessageBox messageBox(this);
		messageBox.setWindowTitle(QTStr("NoSources.Title"));
		messageBox.setText(msg);
		QAbstractButton *Yes = messageBox.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
		messageBox.addButton(QTStr("No"), QMessageBox::NoRole);
		messageBox.setIcon(QMessageBox::Question);
		messageBox.exec();

		if (messageBox.clickedButton() != Yes)
5557 5558 5559 5560 5561 5562
			return false;
	}

	return true;
}

5563 5564
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
5565
	if (outputHandler->StreamingActive()) {
5566 5567 5568
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStoppingStream");

C
cg2121 已提交
5569
		if (confirm && isVisible()) {
5570
			QMessageBox::StandardButton button =
5571
				OBSMessageBox::question(this,
5572 5573 5574
						QTStr("ConfirmStop.Title"),
						QTStr("ConfirmStop.Text"));

C
cg2121 已提交
5575 5576
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(true);
5577
				return;
C
cg2121 已提交
5578
			}
5579 5580
		}

5581
		StopStreaming();
5582
	} else {
5583 5584 5585 5586 5587
		if (!NoSourcesConfirmation()) {
			ui->streamButton->setChecked(false);
			return;
		}

5588 5589 5590
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStartingStream");

5591 5592 5593 5594 5595 5596 5597 5598 5599 5600 5601 5602 5603 5604 5605
		obs_data_t *settings = obs_service_get_settings(service);
		bool bwtest = obs_data_get_bool(settings, "bwtest");
		obs_data_release(settings);

		if (bwtest && isVisible()) {
			QMessageBox::StandardButton button =
				OBSMessageBox::question(this,
						QTStr("ConfirmBWTest.Title"),
						QTStr("ConfirmBWTest.Text"));

			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(false);
				return;
			}
		} else if (confirm && isVisible()) {
5606
			QMessageBox::StandardButton button =
5607
				OBSMessageBox::question(this,
5608 5609 5610
						QTStr("ConfirmStart.Title"),
						QTStr("ConfirmStart.Text"));

C
cg2121 已提交
5611 5612
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(false);
5613
				return;
C
cg2121 已提交
5614
			}
5615 5616
		}

5617
		StartStreaming();
5618 5619 5620 5621 5622
	}
}

void OBSBasic::on_recordButton_clicked()
{
5623
	if (outputHandler->RecordingActive()) {
5624
		StopRecording();
5625 5626 5627 5628 5629 5630
	} else {
		if (!NoSourcesConfirmation()) {
			ui->recordButton->setChecked(false);
			return;
		}

5631
		StartRecording();
5632
	}
J
jp9000 已提交
5633 5634
}

J
jp9000 已提交
5635
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
5636
{
J
jp9000 已提交
5637
	on_action_Settings_triggered();
J
jp9000 已提交
5638
}
5639

5640 5641 5642 5643 5644 5645
void OBSBasic::on_actionHelpPortal_triggered()
{
	QUrl url = QUrl("https://obsproject.com/help", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5646 5647 5648 5649 5650 5651
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5652 5653
void OBSBasic::on_actionDiscord_triggered()
{
5654
	QUrl url = QUrl("https://obsproject.com/discord", QUrl::TolerantMode);
5655 5656 5657
	QDesktopServices::openUrl(url);
}

5658 5659 5660 5661 5662 5663 5664 5665 5666 5667 5668 5669 5670 5671 5672 5673 5674 5675 5676 5677
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 已提交
5678
int OBSBasic::GetTopSelectedSourceItem()
5679
{
J
jp9000 已提交
5680 5681 5682
	QModelIndexList selectedItems =
		ui->sources->selectionModel()->selectedIndexes();
	return selectedItems.count() ? selectedItems[0].row() : -1;
5683 5684
}

J
jp9000 已提交
5685 5686
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
5687
	CreateSourcePopupMenu(GetTopSelectedSourceItem(), true);
J
jp9000 已提交
5688 5689 5690 5691

	UNUSED_PARAMETER(pos);
}

R
Ryan Foster 已提交
5692 5693 5694 5695 5696 5697 5698 5699
void OBSBasic::on_program_customContextMenuRequested(const QPoint&)
{
	QMenu popup(this);
	QPointer<QMenu> studioProgramProjector;

	studioProgramProjector = new QMenu(
			QTStr("StudioProgramProjector"));
	AddProjectorMenuMonitors(studioProgramProjector, this,
5700
			SLOT(OpenStudioProgramProjector()));
R
Ryan Foster 已提交
5701 5702 5703 5704 5705

	popup.addMenu(studioProgramProjector);

	QAction *studioProgramWindow = popup.addAction(
			QTStr("StudioProgramWindow"),
5706
			this, SLOT(OpenStudioProgramWindow()));
R
Ryan Foster 已提交
5707 5708 5709 5710 5711 5712

	popup.addAction(studioProgramWindow);

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

J
jp9000 已提交
5713 5714 5715 5716
void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
P
pkv 已提交
5717
	delete previewProjectorMain;
J
jp9000 已提交
5718 5719 5720 5721 5722

	QAction *action = popup.addAction(
			QTStr("Basic.Main.PreviewConextMenu.Enable"),
			this, SLOT(TogglePreview()));
	action->setCheckable(true);
5723
	action->setChecked(obs_display_enabled(ui->preview->GetDisplay()));
J
jp9000 已提交
5724

P
pkv 已提交
5725 5726
	previewProjectorMain = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjectorMain, this,
J
jp9000 已提交
5727 5728
			SLOT(OpenPreviewProjector()));

C
cg2121 已提交
5729 5730 5731 5732
	QAction *previewWindow = popup.addAction(
			QTStr("PreviewWindow"),
			this, SLOT(OpenPreviewWindow()));

P
pkv 已提交
5733
	popup.addMenu(previewProjectorMain);
C
cg2121 已提交
5734
	popup.addAction(previewWindow);
J
jp9000 已提交
5735 5736 5737 5738 5739
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

5740 5741
void OBSBasic::on_actionAlwaysOnTop_triggered()
{
5742
#ifndef _WIN32
5743 5744 5745 5746
	/* 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 */
5747 5748 5749
	CloseDialogs();
#endif

5750 5751 5752 5753 5754 5755 5756 5757 5758 5759 5760 5761 5762 5763
	QMetaObject::invokeMethod(this, "ToggleAlwaysOnTop",
			Qt::QueuedConnection);
}

void OBSBasic::ToggleAlwaysOnTop()
{
	bool isAlwaysOnTop = IsAlwaysOnTop(this);

	ui->actionAlwaysOnTop->setChecked(!isAlwaysOnTop);
	SetAlwaysOnTop(this, !isAlwaysOnTop);

	show();
}

5764 5765 5766 5767 5768 5769 5770 5771 5772 5773
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 已提交
5774 5775 5776
	} else if (strcmp(val, "24 NTSC") == 0) {
		num = 24000;
		den = 1001;
5777
	} else if (strcmp(val, "25 PAL") == 0) {
5778 5779 5780 5781 5782 5783 5784 5785
		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;
5786 5787 5788
	} else if (strcmp(val, "50 PAL") == 0) {
		num = 50;
		den = 1;
5789 5790 5791 5792 5793 5794 5795 5796 5797 5798 5799 5800 5801 5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812 5813 5814 5815 5816 5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827 5828 5829 5830 5831 5832
	} 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);
}

5833
config_t *OBSBasic::Config() const
5834 5835 5836
{
	return basicConfig;
}
J
jp9000 已提交
5837 5838 5839

void OBSBasic::on_actionEditTransform_triggered()
{
5840 5841 5842
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
5843 5844
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
5845
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
5846 5847
}

5848 5849 5850 5851 5852 5853 5854 5855 5856 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 5887 5888 5889 5890 5891
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);
}

5892
static bool reset_tr(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
5893
{
5894 5895 5896 5897
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, reset_tr, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
J
jp9000 已提交
5898

5899
	obs_sceneitem_defer_update_begin(item);
J
jp9000 已提交
5900

5901 5902 5903 5904 5905 5906 5907 5908 5909
	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 已提交
5910

5911 5912
	obs_sceneitem_crop crop = {};
	obs_sceneitem_set_crop(item, &crop);
J
jp9000 已提交
5913

5914
	obs_sceneitem_defer_update_end(item);
J
jp9000 已提交
5915

5916 5917 5918 5919
	UNUSED_PARAMETER(scene);
	UNUSED_PARAMETER(param);
	return true;
}
J
jp9000 已提交
5920

5921 5922 5923
void OBSBasic::on_actionResetTransform_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), reset_tr, nullptr);
J
jp9000 已提交
5924 5925
}

5926
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
5927 5928 5929 5930 5931
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

	vec3_set(&tl, M_INFINITE, M_INFINITE, 0.0f);
5932
	vec3_set(&br, -M_INFINITE, -M_INFINITE, 0.0f);
J
jp9000 已提交
5933

5934
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
5935 5936 5937 5938
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
5939 5940
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
5941 5942
	};

5943 5944 5945 5946 5947 5948
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

5949
static vec3 GetItemTL(obs_sceneitem_t *item)
5950 5951 5952
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
5953 5954 5955
	return tl;
}

5956
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
5957 5958 5959 5960
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
5961
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
5962 5963 5964
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
5965
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
5966 5967
}

5968
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5969 5970
		void *param)
{
5971 5972 5973
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, RotateSelectedSources,
				param);
J
jp9000 已提交
5974 5975 5976 5977 5978 5979 5980
	if (!obs_sceneitem_selected(item))
		return true;

	float rot = *reinterpret_cast<float*>(param);

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
5981
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
5982 5983
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
5984
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
5985

5986 5987
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
5988 5989 5990 5991 5992 5993 5994 5995 5996 5997 5998 5999 6000 6001 6002 6003 6004 6005 6006 6007 6008 6009 6010 6011
	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);
}

6012
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
6013 6014 6015 6016
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

6017 6018 6019
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, MultiplySelectedItemScale,
				param);
J
jp9000 已提交
6020 6021 6022 6023 6024 6025
	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
6026
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
6027
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
6028
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
6029

6030 6031
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
6032
	SetItemTL(item, tl);
J
jp9000 已提交
6033 6034

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
6035 6036 6037 6038 6039
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
6040 6041
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
6042 6043 6044 6045 6046 6047
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
6048 6049
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
6050 6051 6052 6053
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

6054
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
6055 6056 6057 6058
		void *param)
{
	obs_bounds_type boundsType = *reinterpret_cast<obs_bounds_type*>(param);

6059 6060 6061
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, CenterAlignSelectedItems,
				param);
J
jp9000 已提交
6062 6063 6064 6065 6066 6067
	if (!obs_sceneitem_selected(item))
		return true;

	obs_video_info ovi;
	obs_get_video_info(&ovi);

6068
	obs_transform_info itemInfo;
J
jp9000 已提交
6069 6070 6071 6072 6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098
	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);
}

6099
static bool center_to_scene(obs_scene_t *, obs_sceneitem_t *item, void *)
J
jp9000 已提交
6100
{
6101 6102
	vec3 tl, br, itemCenter, screenCenter, offset;
	obs_video_info ovi;
6103

6104 6105 6106 6107
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, center_to_scene, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
6108

6109
	obs_get_video_info(&ovi);
6110

6111 6112 6113
	vec3_set(&screenCenter, float(ovi.base_width),
			float(ovi.base_height), 0.0f);
	vec3_mulf(&screenCenter, &screenCenter, 0.5f);
6114

6115
	GetItemBox(item, tl, br);
6116

6117 6118 6119
	vec3_sub(&itemCenter, &br, &tl);
	vec3_mulf(&itemCenter, &itemCenter, 0.5f);
	vec3_add(&itemCenter, &itemCenter, &tl);
6120

6121 6122
	vec3_sub(&offset, &screenCenter, &itemCenter);
	vec3_add(&tl, &tl, &offset);
6123

6124 6125 6126
	SetItemTL(item, tl);
	return true;
};
6127

6128 6129 6130
void OBSBasic::on_actionCenterToScreen_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), center_to_scene, nullptr);
J
jp9000 已提交
6131
}
J
jp9000 已提交
6132

6133 6134 6135 6136 6137 6138 6139
void OBSBasic::EnablePreviewDisplay(bool enable)
{
	obs_display_set_enabled(ui->preview->GetDisplay(), enable);
	ui->preview->setVisible(enable);
	ui->previewDisabledLabel->setVisible(!enable);
}

J
jp9000 已提交
6140 6141
void OBSBasic::TogglePreview()
{
6142 6143
	previewEnabled = !previewEnabled;
	EnablePreviewDisplay(previewEnabled);
J
jp9000 已提交
6144
}
6145

J
jp9000 已提交
6146
static bool nudge_callback(obs_scene_t*, obs_sceneitem_t *item, void *param)
6147
{
J
jp9000 已提交
6148 6149
	if (obs_sceneitem_locked(item))
		return true;
6150

J
jp9000 已提交
6151 6152
	struct vec2 &offset = *reinterpret_cast<struct vec2*>(param);
	struct vec2 pos;
6153

J
jp9000 已提交
6154
	if (!obs_sceneitem_selected(item)) {
J
jp9000 已提交
6155 6156 6157 6158 6159 6160 6161 6162 6163 6164 6165 6166 6167 6168 6169 6170
		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 已提交
6171 6172
		return true;
	}
6173

J
jp9000 已提交
6174 6175 6176 6177 6178
	obs_sceneitem_get_pos(item, &pos);
	vec2_add(&pos, &pos, &offset);
	obs_sceneitem_set_pos(item, &pos);
	return true;
}
J
jp9000 已提交
6179

J
jp9000 已提交
6180 6181 6182 6183
void OBSBasic::Nudge(int dist, MoveDir dir)
{
	if (ui->preview->Locked())
		return;
6184

J
jp9000 已提交
6185 6186
	struct vec2 offset;
	vec2_set(&offset, 0.0f, 0.0f);
6187

J
jp9000 已提交
6188 6189 6190 6191 6192 6193
	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;
	}
6194

J
jp9000 已提交
6195
	obs_scene_enum_items(GetCurrentScene(), nudge_callback, &offset);
6196 6197 6198 6199 6200 6201
}

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 已提交
6202

S
Shaolin 已提交
6203
OBSProjector *OBSBasic::OpenProjector(obs_source_t *source, int monitor,
S
Shaolin 已提交
6204
		QString title, ProjectorType type)
J
jp9000 已提交
6205 6206
{
	/* seriously?  10 monitors? */
C
cg2121 已提交
6207
	if (monitor > 9 || monitor > QGuiApplication::screens().size() - 1)
S
Shaolin 已提交
6208
		return nullptr;
C
cg2121 已提交
6209

S
Shaolin 已提交
6210 6211
	OBSProjector *projector = new OBSProjector(nullptr, source, monitor,
			title, type);
C
cg2121 已提交
6212

S
Shaolin 已提交
6213
	if (monitor < 0) {
C
cg2121 已提交
6214 6215 6216 6217 6218 6219 6220 6221 6222
		for (auto &projPtr : windowProjectors) {
			if (!projPtr) {
				projPtr = projector;
				projector = nullptr;
			}
		}

		if (projector)
			windowProjectors.push_back(projector);
S
Shaolin 已提交
6223
	} else {
C
cg2121 已提交
6224 6225
		delete projectors[monitor];
		projectors[monitor].clear();
C
cg2121 已提交
6226

C
cg2121 已提交
6227 6228
		projectors[monitor] = projector;
	}
S
Shaolin 已提交
6229 6230

	return projector;
J
jp9000 已提交
6231 6232
}

R
Ryan Foster 已提交
6233 6234 6235
void OBSBasic::OpenStudioProgramProjector()
{
	int monitor = sender()->property("monitor").toInt();
6236
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::StudioProgram);
R
Ryan Foster 已提交
6237 6238
}

J
jp9000 已提交
6239 6240 6241
void OBSBasic::OpenPreviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
6242
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Preview);
J
jp9000 已提交
6243 6244 6245 6246 6247 6248 6249 6250 6251
}

void OBSBasic::OpenSourceProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

S
Shaolin 已提交
6252 6253
	OpenProjector(obs_sceneitem_get_source(item), monitor, nullptr,
			ProjectorType::Source);
J
jp9000 已提交
6254 6255
}

S
Shaolin 已提交
6256 6257 6258
void OBSBasic::OpenMultiviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
6259
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Multiview);
S
Shaolin 已提交
6260 6261
}

J
jp9000 已提交
6262 6263 6264 6265 6266 6267 6268
void OBSBasic::OpenSceneProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

6269 6270
	OpenProjector(obs_scene_get_source(scene), monitor, nullptr,
			ProjectorType::Scene);
C
cg2121 已提交
6271 6272
}

R
Ryan Foster 已提交
6273 6274
void OBSBasic::OpenStudioProgramWindow()
{
S
Shaolin 已提交
6275
	OpenProjector(nullptr, -1, QTStr("StudioProgramWindow"),
S
Shaolin 已提交
6276
			ProjectorType::StudioProgram);
R
Ryan Foster 已提交
6277 6278
}

C
cg2121 已提交
6279 6280
void OBSBasic::OpenPreviewWindow()
{
S
Shaolin 已提交
6281 6282
	OpenProjector(nullptr, -1, QTStr("PreviewWindow"),
			ProjectorType::Preview);
C
cg2121 已提交
6283 6284 6285 6286 6287 6288 6289 6290 6291
}

void OBSBasic::OpenSourceWindow()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

	OBSSource source = obs_sceneitem_get_source(item);
6292
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
6293 6294 6295

	OpenProjector(obs_sceneitem_get_source(item), -1, title,
			ProjectorType::Source);
C
cg2121 已提交
6296 6297
}

S
Shaolin 已提交
6298 6299
void OBSBasic::OpenMultiviewWindow()
{
S
Shaolin 已提交
6300
	OpenProjector(nullptr, -1, QTStr("MultiviewWindowed"),
S
Shaolin 已提交
6301 6302 6303
			ProjectorType::Multiview);
}

C
cg2121 已提交
6304 6305 6306 6307 6308 6309 6310
void OBSBasic::OpenSceneWindow()
{
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

	OBSSource source = obs_scene_get_source(scene);
6311
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
6312 6313 6314

	OpenProjector(obs_scene_get_source(scene), -1, title,
			ProjectorType::Scene);
J
jp9000 已提交
6315
}
6316

C
cg2121 已提交
6317 6318
void OBSBasic::OpenSavedProjectors()
{
6319
	for (SavedProjectorInfo *info : savedProjectorsArray) {
S
Shaolin 已提交
6320
		OBSProjector *projector = nullptr;
6321
		switch (info->type) {
S
Shaolin 已提交
6322
		case ProjectorType::Source:
6323 6324 6325 6326 6327
		case ProjectorType::Scene: {
			OBSSource source = obs_get_source_by_name(
					info->name.c_str());
			if (!source)
				continue;
C
cg2121 已提交
6328

S
Shaolin 已提交
6329 6330 6331 6332
			QString title = nullptr;
			if (info->monitor < 0)
				title = QString::fromUtf8(
						obs_source_get_name(source));
R
Ryan Foster 已提交
6333

S
Shaolin 已提交
6334 6335
			projector = OpenProjector(source, info->monitor, title,
					info->type);
6336 6337 6338

			obs_source_release(source);
			break;
S
Shaolin 已提交
6339
		}
6340
		case ProjectorType::Preview: {
S
Shaolin 已提交
6341
			projector = OpenProjector(nullptr, info->monitor,
6342
					QTStr("PreviewWindow"),
6343
					ProjectorType::Preview);
6344
			break;
S
Shaolin 已提交
6345
		}
6346
		case ProjectorType::StudioProgram: {
S
Shaolin 已提交
6347
			projector = OpenProjector(nullptr, info->monitor,
6348 6349 6350 6351 6352
					QTStr("StudioProgramWindow"),
					ProjectorType::StudioProgram);
			break;
		}
		case ProjectorType::Multiview: {
S
Shaolin 已提交
6353
			projector = OpenProjector(nullptr, info->monitor,
6354
					QTStr("MultiviewWindowed"),
6355
					ProjectorType::Multiview);
6356 6357
			break;
		}
C
cg2121 已提交
6358
		}
S
Shaolin 已提交
6359

6360
		if (projector && !info->geometry.empty() && info->monitor < 0) {
S
Shaolin 已提交
6361 6362 6363
			QByteArray byteArray = QByteArray::fromBase64(
					QByteArray(info->geometry.c_str()));
			projector->restoreGeometry(byteArray);
S
Shaolin 已提交
6364

S
Shaolin 已提交
6365 6366 6367 6368 6369
			if (!WindowPositionValid(projector->normalGeometry())) {
				QRect rect = App()->desktop()->geometry();
				projector->setGeometry(QStyle::alignedRect(
						Qt::LeftToRight,
						Qt::AlignCenter, size(), rect));
C
cg2121 已提交
6370 6371 6372 6373 6374
			}
		}
	}
}

6375 6376 6377 6378 6379 6380 6381 6382 6383 6384
void OBSBasic::on_actionFullscreenInterface_triggered()
{
	if (!fullscreenInterface)
		showFullScreen();
	else
		showNormal();

	fullscreenInterface = !fullscreenInterface;
}

6385 6386 6387 6388
void OBSBasic::UpdateTitleBar()
{
	stringstream name;

J
jp9000 已提交
6389 6390
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "Profile");
J
jp9000 已提交
6391 6392 6393
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

6394 6395 6396 6397 6398
	name << "OBS ";
	if (previewProgramMode)
		name << "Studio ";

	name << App()->GetVersionString();
6399 6400 6401
	if (App()->IsPortableMode())
		name << " - Portable Mode";

J
jp9000 已提交
6402
	name << " - " << Str("TitleBar.Profile") << ": " << profile;
J
jp9000 已提交
6403
	name << " - " << Str("TitleBar.Scenes") << ": " << sceneCollection;
6404 6405 6406

	setWindowTitle(QT_UTF8(name.str().c_str()));
}
J
jp9000 已提交
6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430

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);
}
6431

J
jp9000 已提交
6432
void OBSBasic::on_resetUI_triggered()
6433
{
6434 6435 6436 6437 6438 6439 6440 6441 6442 6443 6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 6461 6462 6463
	/* 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 已提交
6464
	restoreState(startingDockLayout);
6465

6466
#if QT_VERSION >= QT_VERSION_CHECK(5, 6, 0)
J
jp9000 已提交
6467 6468 6469 6470 6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481
	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,
6482
		ui->controlsDock
J
jp9000 已提交
6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497
	};

	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);
6498 6499
	statsDock->setVisible(false);
	statsDock->setFloating(true);
J
jp9000 已提交
6500 6501 6502

	resizeDocks(docks, {cy, cy, cy, cy, cy}, Qt::Vertical);
	resizeDocks(docks, sizes, Qt::Horizontal);
6503
#endif
J
jp9000 已提交
6504 6505 6506 6507 6508 6509 6510 6511 6512 6513 6514 6515 6516
}

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);
6517
	statsDock->setFeatures(features);
6518 6519 6520 6521 6522 6523 6524 6525

	for (int i = extraDocks.size() - 1; i >= 0 ; i--) {
		if (!extraDocks[i]) {
			extraDocks.removeAt(i);
		} else {
			extraDocks[i]->setFeatures(features);
		}
	}
6526 6527 6528 6529 6530 6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543
}

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 已提交
6544 6545 6546 6547 6548 6549

void OBSBasic::on_actionLockPreview_triggered()
{
	ui->preview->ToggleLocked();
	ui->actionLockPreview->setChecked(ui->preview->Locked());
}
C
cg2121 已提交
6550

J
Joseph El-Khouri 已提交
6551 6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565 6566
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);
6567 6568
	action->setVisible(!(ovi.output_width == ovi.base_width &&
			ovi.output_height == ovi.base_height));
J
Joseph El-Khouri 已提交
6569 6570 6571 6572 6573 6574

	UpdatePreviewScalingMenu();
}

void OBSBasic::on_actionScaleWindow_triggered()
{
6575
	ui->preview->SetFixedScaling(false);
J
Joseph El-Khouri 已提交
6576 6577 6578 6579 6580 6581
	ui->preview->ResetScrollingOffset();
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleCanvas_triggered()
{
6582 6583
	ui->preview->SetFixedScaling(true);
	ui->preview->SetScalingLevel(0);
J
Joseph El-Khouri 已提交
6584 6585 6586 6587 6588
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleOutput_triggered()
{
6589 6590 6591 6592 6593 6594 6595 6596 6597 6598
	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 已提交
6599 6600 6601
	emit ui->preview->DisplayResized();
}

C
cg2121 已提交
6602 6603 6604 6605 6606 6607 6608
void OBSBasic::SetShowing(bool showing)
{
	if (!showing && isVisible()) {
		config_set_string(App()->GlobalConfig(),
			"BasicWindow", "geometry",
			saveGeometry().toBase64().constData());

6609 6610 6611 6612 6613 6614 6615 6616 6617
		/* hide all visible child dialogs */
		visDlgPositions.clear();
		if (!visDialogs.isEmpty()) {
			for (QDialog *dlg : visDialogs) {
				visDlgPositions.append(dlg->pos());
				dlg->hide();
			}
		}

6618 6619
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Show"));
C
cg2121 已提交
6620 6621 6622 6623 6624 6625 6626
		QTimer::singleShot(250, this, SLOT(hide()));

		if (previewEnabled)
			EnablePreviewDisplay(false);

		setVisible(false);

6627 6628 6629 6630
#ifdef __APPLE__
		EnableOSXDockIcon(false);
#endif

C
cg2121 已提交
6631
	} else if (showing && !isVisible()) {
6632 6633
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Hide"));
C
cg2121 已提交
6634 6635 6636 6637 6638 6639
		QTimer::singleShot(250, this, SLOT(show()));

		if (previewEnabled)
			EnablePreviewDisplay(true);

		setVisible(true);
6640

6641 6642 6643 6644 6645 6646 6647 6648
#ifdef __APPLE__
		EnableOSXDockIcon(true);
#endif

		/* raise and activate window to ensure it is on top */
		raise();
		activateWindow();

6649 6650 6651 6652 6653 6654 6655 6656 6657
		/* 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();
			}
		}

6658 6659 6660 6661 6662 6663 6664 6665
		/* 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 已提交
6666 6667 6668
	}
}

6669 6670 6671 6672 6673 6674 6675 6676 6677 6678 6679 6680
void OBSBasic::ToggleShowHide()
{
	bool showing = isVisible();
	if (showing) {
		/* check for modal dialogs */
		EnumDialogs();
		if (!modalDialogs.isEmpty() || !visMsgBoxes.isEmpty())
			return;
	}
	SetShowing(!showing);
}

6681 6682
void OBSBasic::SystemTrayInit()
{
6683 6684
	trayIcon.reset(new QSystemTrayIcon(QIcon::fromTheme("obs-tray",
			QIcon(":/res/images/obs.png")), this));
C
cg2121 已提交
6685 6686 6687
	trayIcon->setToolTip("OBS Studio");

	showHide = new QAction(QTStr("Basic.SystemTray.Show"),
6688
			trayIcon.data());
C
cg2121 已提交
6689
	sysTrayStream = new QAction(QTStr("Basic.Main.StartStreaming"),
6690
			trayIcon.data());
C
cg2121 已提交
6691
	sysTrayRecord = new QAction(QTStr("Basic.Main.StartRecording"),
6692
			trayIcon.data());
J
jp9000 已提交
6693
	sysTrayReplayBuffer = new QAction(QTStr("Basic.Main.StartReplayBuffer"),
6694
			trayIcon.data());
C
cg2121 已提交
6695
	exit = new QAction(QTStr("Exit"),
6696
			trayIcon.data());
C
cg2121 已提交
6697

6698 6699 6700 6701 6702 6703 6704 6705 6706 6707 6708 6709 6710 6711 6712 6713
	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 已提交
6714 6715
	if (outputHandler && !outputHandler->replayBuffer)
		sysTrayReplayBuffer->setEnabled(false);
6716

6717 6718
	connect(trayIcon.data(),
			SIGNAL(activated(QSystemTrayIcon::ActivationReason)),
C
cg2121 已提交
6719 6720 6721 6722 6723 6724 6725 6726
			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()));
6727
	connect(sysTrayReplayBuffer.data(), &QAction::triggered,
J
jp9000 已提交
6728
			this, &OBSBasic::ReplayBufferClicked);
C
cg2121 已提交
6729 6730 6731 6732 6733 6734
	connect(exit, SIGNAL(triggered()),
			this, SLOT(close()));
}

void OBSBasic::IconActivated(QSystemTrayIcon::ActivationReason reason)
{
6735 6736 6737 6738 6739 6740 6741 6742 6743
	// Refresh projector list
	previewProjector->clear();
	studioProgramProjector->clear();
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));
	AddProjectorMenuMonitors(studioProgramProjector, this,
			SLOT(OpenStudioProgramProjector()));

	if (reason == QSystemTrayIcon::Trigger)
C
cg2121 已提交
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
		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 已提交
6772 6773
	} else if ((sysTrayWhenStarted && sysTrayEnabled)
			|| opt_minimize_tray) {
C
cg2121 已提交
6774 6775 6776 6777 6778
		trayIcon->show();
		if (firstStarted) {
			QTimer::singleShot(50, this, SLOT(hide()));
			EnablePreviewDisplay(false);
			setVisible(false);
6779 6780 6781
#ifdef __APPLE__
			EnableOSXDockIcon(false);
#endif
C
cg2121 已提交
6782
			opt_minimize_tray = false;
C
cg2121 已提交
6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793 6794 6795 6796
		}
	} 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"));
}
6797 6798 6799 6800 6801 6802

bool OBSBasic::sysTrayMinimizeToTray()
{
	return config_get_bool(GetGlobalConfig(),
			"BasicWindow", "SysTrayMinimizeToTray");
}
6803 6804 6805 6806 6807 6808

void OBSBasic::on_actionCopySource_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;
J
jp9000 已提交
6809 6810

	on_actionCopyTransform_triggered();
6811 6812 6813 6814 6815 6816 6817

	OBSSource source = obs_sceneitem_get_source(item);

	copyString = obs_source_get_name(source);
	copyVisible = obs_sceneitem_visible(item);

	ui->actionPasteRef->setEnabled(true);
6818 6819 6820 6821 6822 6823

	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);
6824 6825 6826 6827
}

void OBSBasic::on_actionPasteRef_triggered()
{
J
jp9000 已提交
6828 6829 6830 6831 6832
	/* do not allow duplicate refs of the same group in the same scene */
	OBSScene scene = GetCurrentScene();
	if (!!obs_scene_get_group(scene, copyString))
		return;

6833 6834 6835 6836 6837 6838 6839 6840 6841 6842
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, false);
	on_actionPasteTransform_triggered();
}

void OBSBasic::on_actionPasteDup_triggered()
{
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, true);
	on_actionPasteTransform_triggered();
}

6843 6844 6845 6846 6847 6848 6849 6850 6851 6852 6853 6854 6855 6856 6857 6858 6859 6860 6861 6862 6863 6864 6865 6866
void OBSBasic::AudioMixerCopyFilters()
{
	QAction *action = reinterpret_cast<QAction*>(sender());
	VolControl *vol = action->property("volControl").value<VolControl*>();
	obs_source_t *source = vol->GetSource();

	copyFiltersString = obs_source_get_name(source);
}

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

	OBSSource source = obs_get_source_by_name(copyFiltersString);
	obs_source_release(source);

	if (source == dstSource)
		return;

	obs_source_copy_filters(dstSource, source);
}

6867 6868 6869 6870 6871 6872 6873 6874 6875 6876 6877 6878 6879 6880 6881 6882 6883
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);
6884
	obs_source_release(source);
6885

6886
	OBSSceneItem sceneItem = GetCurrentSceneItem();
6887 6888 6889 6890 6891 6892 6893
	OBSSource dstSource = obs_sceneitem_get_source(sceneItem);

	if (source == dstSource)
		return;

	obs_source_copy_filters(dstSource, source);
}
J
jp9000 已提交
6894

6895 6896 6897 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 6923 6924 6925 6926 6927 6928 6929 6930 6931 6932 6933 6934 6935 6936 6937 6938 6939 6940 6941 6942 6943 6944 6945 6946 6947 6948 6949 6950 6951 6952 6953 6954 6955 6956 6957 6958 6959 6960 6961 6962 6963 6964 6965 6966 6967 6968 6969 6970 6971 6972 6973 6974 6975 6976 6977 6978 6979 6980 6981 6982 6983 6984 6985 6986 6987 6988 6989 6990 6991 6992 6993 6994 6995 6996 6997 6998 6999 7000 7001 7002 7003 7004 7005 7006 7007 7008 7009 7010 7011 7012 7013 7014 7015 7016 7017 7018 7019 7020 7021 7022 7023 7024 7025 7026 7027 7028 7029 7030 7031 7032 7033 7034 7035 7036 7037 7038 7039 7040 7041 7042 7043 7044 7045 7046 7047 7048 7049 7050 7051 7052 7053 7054 7055 7056 7057 7058 7059 7060 7061 7062 7063 7064 7065 7066 7067 7068
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 已提交
7069 7070 7071 7072 7073 7074 7075
void OBSBasic::on_autoConfigure_triggered()
{
	AutoConfig test(this);
	test.setModal(true);
	test.show();
	test.exec();
}
J
jp9000 已提交
7076 7077 7078

void OBSBasic::on_stats_triggered()
{
7079 7080 7081 7082 7083 7084
	if (!stats.isNull()) {
		stats->show();
		stats->raise();
		return;
	}

J
jp9000 已提交
7085 7086 7087 7088 7089
	OBSBasicStats *statsDlg;
	statsDlg = new OBSBasicStats(nullptr);
	statsDlg->show();
	stats = statsDlg;
}
7090

C
cg2121 已提交
7091 7092 7093 7094 7095 7096 7097 7098 7099 7100 7101
void OBSBasic::on_actionShowAbout_triggered()
{
	if (about)
		about->close();

	about = new OBSAbout(this);
	about->show();

	about->setAttribute(Qt::WA_DeleteOnClose, true);
}

7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115 7116 7117 7118 7119 7120 7121 7122 7123 7124 7125 7126 7127 7128 7129 7130 7131 7132 7133
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();
}
7134

7135 7136 7137 7138 7139 7140 7141 7142 7143 7144 7145 7146 7147 7148 7149 7150 7151 7152 7153 7154 7155 7156 7157 7158
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;
}

7159 7160 7161 7162
OBSBasic *OBSBasic::Get()
{
	return reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
}
7163 7164 7165 7166 7167 7168 7169 7170 7171 7172 7173 7174 7175 7176 7177 7178 7179 7180 7181 7182 7183

bool OBSBasic::StreamingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

bool OBSBasic::RecordingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->RecordingActive();
}

bool OBSBasic::ReplayBufferActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->ReplayBufferActive();
}