window-basic-main.cpp 174.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/>.
******************************************************************************/

J
jp9000 已提交
20
#include <time.h>
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 29
#include <QRect>
#include <QScreen>
30 31 32
#include <QColorDialog>
#include <QWidgetAction>
#include <QSizePolicy>
33

J
jp9000 已提交
34
#include <util/dstr.h>
35 36
#include <util/util.hpp>
#include <util/platform.h>
P
Palana 已提交
37
#include <util/profiler.hpp>
38
#include <util/dstr.hpp>
J
jp9000 已提交
39
#include <graphics/math-defs.h>
40

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

62
#ifdef _WIN32
J
jp9000 已提交
63 64 65
#include "win-update/win-update.hpp"
#endif

J
jp9000 已提交
66
#include "ui_OBSBasic.h"
67
#include "ui_ColorSelect.h"
68

J
jp9000 已提交
69
#include <fstream>
70 71
#include <sstream>

72 73 74
#include <QScreen>
#include <QWindow>

J
jp9000 已提交
75 76 77 78 79 80 81
#ifdef _WIN32
#include <browser-panel.hpp>
#endif

#include <json11.hpp>

using namespace json11;
82
using namespace std;
J
jp9000 已提交
83

J
jp9000 已提交
84 85 86 87
#ifdef _WIN32
static CREATE_BROWSER_WIDGET_PROC create_browser_widget = nullptr;
#endif

J
jp9000 已提交
88 89 90 91 92 93 94 95 96 97
namespace {

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

}

J
jp9000 已提交
98 99
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
100
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
101
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
102
Q_DECLARE_METATYPE(SignalContainer<OBSScene>);
J
jp9000 已提交
103

P
Palana 已提交
104 105 106 107 108 109 110 111 112 113 114 115 116
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));
}

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

128
	if (ret <= 0)
129 130 131
		return;

	string path = (char*)base_module_dir;
132
#if defined(__APPLE__)
133
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
134 135 136 137 138

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

139 140 141 142 143 144 145
#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
146 147
}

148 149
extern obs_frontend_callbacks *InitializeAPIInterface(OBSBasic *main);

150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
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;
}

170
OBSBasic::OBSBasic(QWidget *parent)
171
	: OBSMainWindow  (parent),
J
jp9000 已提交
172
	  ui             (new Ui::OBSBasic)
173
{
174 175
	setAttribute(Qt::WA_NativeWindow);

J
jp9000 已提交
176 177
	setAcceptDrops(true);

178 179
	api = InitializeAPIInterface(this);

180
	ui->setupUi(this);
J
jp9000 已提交
181 182
	ui->previewDisabledLabel->setVisible(false);

J
jp9000 已提交
183 184
	startingDockLayout = saveState();

S
Socapex 已提交
185
	copyActionsDynamicProperties();
186

187
	char styleSheetPath[512];
J
jp9000 已提交
188 189
	int ret = GetProfilePath(styleSheetPath, sizeof(styleSheetPath),
			"stylesheet.qss");
190
	if (ret > 0) {
H
HomeWorld 已提交
191 192 193 194 195
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
196 197
	}

P
Palana 已提交
198 199 200
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
201
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
202

203 204 205
	qRegisterMetaTypeStreamOperators<
		std::vector<std::shared_ptr<OBSSignal>>>(
				"std::vector<std::shared_ptr<OBSSignal>>");
206
	qRegisterMetaTypeStreamOperators<OBSScene>("OBSScene");
207
	qRegisterMetaTypeStreamOperators<OBSSceneItem>("OBSSceneItem");
208

P
Palana 已提交
209 210 211
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

212
	auto displayResize = [this]() {
213 214 215 216
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
217 218 219 220
	};

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

P
Palana 已提交
222 223
	installEventFilter(CreateShortcutFilter());

224
	stringstream name;
J
jp9000 已提交
225
	name << "OBS " << App()->GetVersionString();
226 227 228
	blog(LOG_INFO, "%s", name.str().c_str());
	blog(LOG_INFO, "---------------------------------");

229
	UpdateTitleBar();
J
jp9000 已提交
230 231 232 233 234 235 236 237

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

238
	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
239
	cpuUsageTimer = new QTimer(this);
240
	connect(cpuUsageTimer.data(), SIGNAL(timeout()),
J
jp9000 已提交
241 242
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
243 244

#ifdef __APPLE__
245 246
	ui->actionRemoveSource->setShortcuts({Qt::Key_Backspace});
	ui->actionRemoveScene->setShortcuts({Qt::Key_Backspace});
247 248 249

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
250
#endif
251 252 253 254 255 256 257 258 259 260 261 262 263 264

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

266
	auto assignDockToggle = [] (QDockWidget *dock, QAction *action)
J
jp9000 已提交
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291
	{
		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);
	};

	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);
S
SuslikV 已提交
292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316

	//hide all docking panes
	ui->toggleScenes->setChecked(false);
	ui->toggleSources->setChecked(false);
	ui->toggleMixer->setChecked(false);
	ui->toggleTransitions->setChecked(false);
	ui->toggleControls->setChecked(false);

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

319 320
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent,
		vector<OBSSource> &audioSources)
321
{
322
	obs_source_t *source = obs_get_output_source(channel);
323 324 325
	if (!source)
		return;

326 327
	audioSources.push_back(source);

328
	obs_data_t *data = obs_save_source(source);
329

J
jp9000 已提交
330
	obs_data_set_obj(parent, name, data);
331 332 333 334 335

	obs_data_release(data);
	obs_source_release(source);
}

336 337
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder,
		obs_data_array_t *quickTransitionData, int transitionDuration,
J
jp9000 已提交
338
		obs_data_array_t *transitions,
C
cg2121 已提交
339
		OBSScene &scene, OBSSource &curProgramScene,
340
		obs_data_array_t *savedProjectorList)
341
{
342 343 344 345 346 347 348 349 350 351 352
	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);

353 354 355
	/* -------------------------------- */
	/* save non-group sources           */

356 357
	auto FilterAudioSources = [&](obs_source_t *source)
	{
358 359 360
		if (obs_source_is_group(source))
			return false;

361 362 363 364 365 366 367 368 369 370 371
		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));

372 373 374 375 376 377 378 379 380 381 382 383
	/* -------------------------------- */
	/* 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);

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

384 385
	obs_source_t *transition = obs_get_output_source(0);
	obs_source_t *currentScene = obs_scene_get_source(scene);
386
	const char   *sceneName   = obs_source_get_name(currentScene);
387
	const char   *programName = obs_source_get_name(curProgramScene);
388

J
jp9000 已提交
389 390 391
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

J
jp9000 已提交
392
	obs_data_set_string(saveData, "current_scene", sceneName);
393
	obs_data_set_string(saveData, "current_program_scene", programName);
J
jp9000 已提交
394
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
395
	obs_data_set_string(saveData, "name", sceneCollection);
J
jp9000 已提交
396
	obs_data_set_array(saveData, "sources", sourcesArray);
397
	obs_data_set_array(saveData, "groups", groupsArray);
398
	obs_data_set_array(saveData, "quick_transitions", quickTransitionData);
J
jp9000 已提交
399
	obs_data_set_array(saveData, "transitions", transitions);
C
cg2121 已提交
400
	obs_data_set_array(saveData, "saved_projectors", savedProjectorList);
401
	obs_data_array_release(sourcesArray);
402
	obs_data_array_release(groupsArray);
403 404 405 406 407

	obs_data_set_string(saveData, "current_transition",
			obs_source_get_name(transition));
	obs_data_set_int(saveData, "transition_duration", transitionDuration);
	obs_source_release(transition);
408 409 410 411

	return saveData;
}

S
Socapex 已提交
412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
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 已提交
432 433 434 435 436 437 438 439 440 441
void OBSBasic::UpdateVolumeControlsDecayRate()
{
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");

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

442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464
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);
	}
}

465 466
void OBSBasic::ClearVolumeControls()
{
C
craftwar 已提交
467 468
	for (VolControl *vol : volumes)
		delete vol;
469 470 471 472

	volumes.clear();
}

J
jp9000 已提交
473 474 475 476 477 478 479 480 481 482 483 484 485 486 487
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 已提交
488 489
obs_data_array_t *OBSBasic::SaveProjectors()
{
490
	obs_data_array_t *savedProjectors = obs_data_array_create();
C
cg2121 已提交
491

492 493 494
	auto saveProjector = [savedProjectors](OBSProjector *projector) {
		if (!projector)
			return;
R
Ryan Foster 已提交
495 496

		obs_data_t *data = obs_data_create();
497
		ProjectorType type = projector->GetProjectorType();
S
Shaolin 已提交
498 499 500 501 502 503 504 505 506 507 508
		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;
		}
509 510
		obs_data_set_int(data, "monitor", projector->GetMonitor());
		obs_data_set_int(data, "type", static_cast<int>(type));
S
Shaolin 已提交
511 512 513
		obs_data_set_string(data, "geometry",
				projector->saveGeometry().toBase64()
						.constData());
514
		obs_data_array_push_back(savedProjectors, data);
R
Ryan Foster 已提交
515
		obs_data_release(data);
516
	};
R
Ryan Foster 已提交
517

518 519
	for (QPointer<QWidget> &proj : projectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
520

S
Shaolin 已提交
521 522
	for (QPointer<QWidget> &proj : windowProjectors)
		saveProjector(static_cast<OBSProjector *>(proj.data()));
S
Shaolin 已提交
523

524
	return savedProjectors;
S
Shaolin 已提交
525 526
}

527 528
void OBSBasic::Save(const char *file)
{
529 530 531 532 533
	OBSScene scene = GetCurrentScene();
	OBSSource curProgramScene = OBSGetStrongRef(programScene);
	if (!curProgramScene)
		curProgramScene = obs_scene_get_source(scene);

J
jp9000 已提交
534
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
J
jp9000 已提交
535
	obs_data_array_t *transitions = SaveTransitions();
536
	obs_data_array_t *quickTrData = SaveQuickTransitions();
C
cg2121 已提交
537
	obs_data_array_t *savedProjectorList = SaveProjectors();
R
Ryan Foster 已提交
538
	obs_data_t *saveData = GenerateSaveData(sceneOrder, quickTrData,
J
jp9000 已提交
539
			ui->transitionDuration->value(), transitions,
540
			scene, curProgramScene, savedProjectorList);
541

J
jp9000 已提交
542
	obs_data_set_bool(saveData, "preview_locked", ui->preview->Locked());
543 544 545 546 547 548 549 550
	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 已提交
551

J
jp9000 已提交
552 553 554 555 556 557 558
	if (api) {
		obs_data_t *moduleObj = obs_data_create();
		api->on_save(moduleObj);
		obs_data_set_obj(saveData, "modules", moduleObj);
		obs_data_release(moduleObj);
	}

559 560
	if (!obs_data_save_json_safe(saveData, file, "tmp", "bak"))
		blog(LOG_ERROR, "Could not save scene data to %s", file);
561 562

	obs_data_release(saveData);
J
jp9000 已提交
563
	obs_data_array_release(sceneOrder);
564
	obs_data_array_release(quickTrData);
J
jp9000 已提交
565
	obs_data_array_release(transitions);
C
cg2121 已提交
566
	obs_data_array_release(savedProjectorList);
567 568
}

I
Ilya M 已提交
569 570 571 572 573 574 575 576 577 578 579 580 581
void OBSBasic::DeferSaveBegin()
{
	os_atomic_inc_long(&disableSaving);
}

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

582
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
583
{
584
	obs_data_t *data = obs_data_get_obj(parent, name);
585 586 587
	if (!data)
		return;

588
	obs_source_t *source = obs_load_source(data);
589 590 591 592 593 594 595 596
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

597 598 599
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
600
	obs_properties_t *props = obs_get_source_properties(output_id);
601 602 603 604 605 606 607 608 609 610 611 612 613 614
	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;
}

615
void OBSBasic::CreateFirstRunSources()
616
{
617 618 619 620 621 622 623 624 625
	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);
626 627
}

628
void OBSBasic::CreateDefaultScene(bool firstStart)
629 630 631 632
{
	disableSaving++;

	ClearSceneData();
633 634 635 636
	InitDefaultTransitions();
	CreateDefaultQuickTransitions();
	ui->transitionDuration->setValue(300);
	SetTransition(fadeTransition);
637 638 639

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

640
	if (firstStart)
641
		CreateFirstRunSources();
642

643
	SetCurrentScene(scene, true);
644
	obs_scene_release(scene);
J
jp9000 已提交
645 646

	disableSaving--;
647 648
}

J
jp9000 已提交
649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677
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 已提交
678 679 680 681 682 683 684
void OBSBasic::LoadSavedProjectors(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);

685 686 687 688
		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 已提交
689 690 691
		info->geometry = std::string(
				obs_data_get_string(data, "geometry"));
		info->name = std::string(obs_data_get_string(data, "name"));
692
		savedProjectorsArray.emplace_back(info);
S
Shaolin 已提交
693 694 695 696 697

		obs_data_release(data);
	}
}

698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718
static void LogFilter(obs_source_t*, obs_source_t *filter, void *v_val)
{
	const char *name = obs_source_get_name(filter);
	const char *id = obs_source_get_id(filter);
	int val = (int)(intptr_t)v_val;
	string indent;

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

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

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

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

719 720 721 722 723 724 725 726 727 728 729 730
	obs_monitoring_type monitoring_type =
		obs_source_get_monitoring_type(source);

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

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

731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754
	obs_source_enum_filters(source, LogFilter, (void*)(intptr_t)2);
	return true;
}

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

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

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

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

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

755 756
void OBSBasic::Load(const char *file)
{
757 758 759 760 761
	disableSaving++;

	obs_data_t *data = obs_data_create_from_json_file_safe(file, "bak");
	if (!data) {
		disableSaving--;
762
		blog(LOG_INFO, "No scene file found, creating default scene");
763
		CreateDefaultScene(true);
J
jp9000 已提交
764
		SaveProject();
765
		return;
766
	}
767

768
	ClearSceneData();
769
	InitDefaultTransitions();
770

771 772 773 774
	obs_data_t *modulesObj = obs_data_get_obj(data, "modules");
	if (api)
		api->on_preload(modulesObj);

J
jp9000 已提交
775
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
776
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
777
	obs_data_array_t *groups     = obs_data_get_array(data, "groups");
J
jp9000 已提交
778
	obs_data_array_t *transitions= obs_data_get_array(data, "transitions");
J
jp9000 已提交
779 780
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
781 782 783 784 785
	const char       *programSceneName = obs_data_get_string(data,
			"current_program_scene");
	const char       *transitionName = obs_data_get_string(data,
			"current_transition");

786 787 788 789 790 791
	if (!opt_starting_scene.empty()) {
		programSceneName = opt_starting_scene.c_str();
		if (!IsPreviewProgramMode())
			sceneName = opt_starting_scene.c_str();
	}

792 793 794 795 796 797
	int newDuration = obs_data_get_int(data, "transition_duration");
	if (!newDuration)
		newDuration = 300;

	if (!transitionName)
		transitionName = obs_source_get_name(fadeTransition);
J
jp9000 已提交
798 799 800 801 802 803 804

	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");
805
	obs_source_t     *curScene;
806 807
	obs_source_t     *curProgramScene;
	obs_source_t     *curTransition;
808

J
jp9000 已提交
809 810 811
	if (!name || !*name)
		name = curSceneCollection;

812 813 814 815 816 817
	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);

818 819 820 821 822 823 824
	if (!sources) {
		sources = groups;
		groups = nullptr;
	} else {
		obs_data_array_push_back_array(sources, groups);
	}

825
	obs_load_sources(sources, nullptr, nullptr);
826

J
jp9000 已提交
827 828
	if (transitions)
		LoadTransitions(transitions);
J
jp9000 已提交
829 830 831
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

J
jp9000 已提交
832 833
	obs_data_array_release(transitions);

834 835 836 837 838 839 840
	curTransition = FindTransition(transitionName);
	if (!curTransition)
		curTransition = fadeTransition;

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

841
retryScene:
842
	curScene = obs_get_source_by_name(sceneName);
843
	curProgramScene = obs_get_source_by_name(programSceneName);
844 845 846 847 848 849 850 851 852 853 854 855 856

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

857 858 859 860 861 862 863 864
	if (!curProgramScene) {
		curProgramScene = curScene;
		obs_source_addref(curScene);
	}

	SetCurrentScene(curScene, true);
	if (IsPreviewProgramMode())
		TransitionToScene(curProgramScene, true);
865
	obs_source_release(curScene);
866
	obs_source_release(curProgramScene);
867 868

	obs_data_array_release(sources);
869
	obs_data_array_release(groups);
J
jp9000 已提交
870
	obs_data_array_release(sceneOrder);
J
jp9000 已提交
871

872 873 874 875 876 877 878 879 880
	/* ------------------- */

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

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

881
		if (savedProjectors) {
882
			LoadSavedProjectors(savedProjectors);
883 884 885
			OpenSavedProjectors();
			activateWindow();
		}
886 887 888 889 890 891

		obs_data_array_release(savedProjectors);
	}

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

J
jp9000 已提交
892 893 894 895 896 897 898 899
	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());

900 901 902 903 904 905 906
	obs_data_array_t *quickTransitionData = obs_data_get_array(data,
			"quick_transitions");
	LoadQuickTransitions(quickTransitionData);
	obs_data_array_release(quickTransitionData);

	RefreshQuickTransitions();

J
jp9000 已提交
907 908 909 910
	bool previewLocked = obs_data_get_bool(data, "preview_locked");
	ui->preview->SetLocked(previewLocked);
	ui->actionLockPreview->setChecked(previewLocked);

911 912 913 914 915 916 917 918 919 920
	/* ---------------------- */

	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 已提交
921
	}
922
	ui->preview->SetFixedScaling(fixedScaling);
J
Joseph El-Khouri 已提交
923

924
	/* ---------------------- */
J
Joseph El-Khouri 已提交
925

926
	if (api)
J
jp9000 已提交
927 928
		api->on_load(modulesObj);

929
	obs_data_release(modulesObj);
930
	obs_data_release(data);
J
jp9000 已提交
931

932 933 934
	if (!opt_starting_scene.empty())
		opt_starting_scene.clear();

935
	if (opt_start_streaming) {
936
		blog(LOG_INFO, "Starting stream due to command line parameter");
937 938 939 940 941 942
		QMetaObject::invokeMethod(this, "StartStreaming",
				Qt::QueuedConnection);
		opt_start_streaming = false;
	}

	if (opt_start_recording) {
943
		blog(LOG_INFO, "Starting recording due to command line parameter");
944 945 946 947 948
		QMetaObject::invokeMethod(this, "StartRecording",
				Qt::QueuedConnection);
		opt_start_recording = false;
	}

C
cg2121 已提交
949 950 951 952 953 954
	if (opt_start_replaybuffer) {
		QMetaObject::invokeMethod(this, "StartReplayBuffer",
				Qt::QueuedConnection);
		opt_start_replaybuffer = false;
	}

955 956
	LogScenes();

J
jp9000 已提交
957
	disableSaving--;
958

959
	if (api) {
960
		api->on_event(OBS_FRONTEND_EVENT_SCENE_CHANGED);
961 962
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
	}
963 964
}

J
jp9000 已提交
965
#define SERVICE_PATH "service.json"
966 967 968 969 970 971

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

972
	char serviceJsonPath[512];
J
jp9000 已提交
973
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
974 975
			SERVICE_PATH);
	if (ret <= 0)
976 977
		return;

978 979
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
980

981
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
982
	obs_data_set_obj(data, "settings", settings);
983

984 985
	if (!obs_data_save_json_safe(data, serviceJsonPath, "tmp", "bak"))
		blog(LOG_WARNING, "Failed to save service");
986 987 988 989 990 991 992 993 994

	obs_data_release(settings);
	obs_data_release(data);
}

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

995
	char serviceJsonPath[512];
J
jp9000 已提交
996
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
997 998
			SERVICE_PATH);
	if (ret <= 0)
999 1000
		return false;

1001 1002
	obs_data_t *data = obs_data_create_from_json_file_safe(serviceJsonPath,
			"bak");
1003 1004

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

1007
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
1008
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
1009

1010
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
1011
			hotkey_data);
1012
	obs_service_release(service);
1013

P
Palana 已提交
1014
	obs_data_release(hotkey_data);
1015 1016 1017 1018 1019 1020 1021 1022
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
P
Palana 已提交
1023 1024
	ProfileScope("OBSBasic::InitService");

1025 1026 1027
	if (LoadService())
		return true;

1028 1029
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
1030 1031
	if (!service)
		return false;
1032
	obs_service_release(service);
1033 1034 1035 1036

	return true;
}

1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
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
};

1053 1054
bool OBSBasic::InitBasicConfigDefaults()
{
1055
	QList<QScreen*> screens = QGuiApplication::screens();
1056

1057
	if (!screens.size()) {
1058 1059 1060 1061 1062
		OBSErrorBox(NULL, "There appears to be no monitors.  Er, this "
		                  "technically shouldn't be possible.");
		return false;
	}

1063 1064 1065 1066
	QScreen *primaryScreen = QGuiApplication::primaryScreen();

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

1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
	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;
	}

1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
	/* ----------------------------------------------------- */
	/* 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");
1089
		config_save_safe(basicConfig, "tmp", nullptr);
1090 1091 1092 1093
	}

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

1094
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
1095

1096 1097
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
1098 1099
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
1100 1101
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
1102 1103
	config_set_default_string(basicConfig, "SimpleOutput", "StreamEncoder",
			SIMPLE_ENCODER_X264);
1104
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
1105 1106
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
1107 1108
	config_set_default_bool  (basicConfig, "SimpleOutput", "EnforceBitrate",
			true);
J
jp9000 已提交
1109 1110
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
1111 1112 1113 1114
	config_set_default_string(basicConfig, "SimpleOutput", "RecQuality",
			"Stream");
	config_set_default_string(basicConfig, "SimpleOutput", "RecEncoder",
			SIMPLE_ENCODER_X264);
1115 1116 1117
	config_set_default_bool(basicConfig, "SimpleOutput", "RecRB", false);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBTime", 20);
	config_set_default_int(basicConfig, "SimpleOutput", "RecRBSize", 512);
1118 1119
	config_set_default_string(basicConfig, "SimpleOutput", "RecRBPrefix",
			"Replay");
1120

1121 1122
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
1123 1124 1125 1126 1127 1128 1129 1130
	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());
1131
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
1132 1133
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
1134
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
1135 1136 1137
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

1138 1139 1140 1141 1142
	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 已提交
1143
	config_set_default_uint  (basicConfig, "AdvOut", "FFVBitrate", 2500);
1144
	config_set_default_uint  (basicConfig, "AdvOut", "FFVGOPSize", 250);
J
jp9000 已提交
1145 1146
	config_set_default_bool  (basicConfig, "AdvOut", "FFUseRescale",
			false);
1147 1148
	config_set_default_bool  (basicConfig, "AdvOut", "FFIgnoreCompat",
			false);
J
jp9000 已提交
1149 1150 1151 1152 1153 1154 1155
	config_set_default_uint  (basicConfig, "AdvOut", "FFABitrate", 160);
	config_set_default_uint  (basicConfig, "AdvOut", "FFAudioTrack", 1);

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

1159 1160 1161 1162
	config_set_default_bool  (basicConfig, "AdvOut", "RecRB", false);
	config_set_default_uint  (basicConfig, "AdvOut", "RecRBTime", 20);
	config_set_default_int   (basicConfig, "AdvOut", "RecRBSize", 512);

1163 1164 1165
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

1166 1167 1168 1169 1170 1171 1172 1173
	/* 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);
	}

1174
	config_set_default_string(basicConfig, "Output", "FilenameFormatting",
B
bl 已提交
1175
			"%CCYY-%MM-%DD %hh-%mm-%ss");
1176

1177 1178 1179 1180
	config_set_default_bool  (basicConfig, "Output", "DelayEnable", false);
	config_set_default_uint  (basicConfig, "Output", "DelaySec", 20);
	config_set_default_bool  (basicConfig, "Output", "DelayPreserve", true);

1181 1182 1183 1184
	config_set_default_bool  (basicConfig, "Output", "Reconnect", true);
	config_set_default_uint  (basicConfig, "Output", "RetryDelay", 10);
	config_set_default_uint  (basicConfig, "Output", "MaxRetries", 20);

1185
	config_set_default_string(basicConfig, "Output", "BindIP", "default");
D
derrod 已提交
1186 1187 1188 1189
	config_set_default_bool  (basicConfig, "Output", "NewSocketLoopEnable",
			false);
	config_set_default_bool  (basicConfig, "Output", "LowLatencyEnable",
			false);
1190

1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204
	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);
1205

1206 1207 1208 1209 1210 1211 1212 1213 1214
	/* 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);
	}

1215 1216 1217 1218 1219
	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);
1220
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
1221
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
1222
	config_set_default_string(basicConfig, "Video", "ColorSpace", "601");
1223 1224
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
1225

1226 1227 1228 1229 1230
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceId",
			"default");
	config_set_default_string(basicConfig, "Audio", "MonitoringDeviceName",
			Str("Basic.Settings.Advanced.Audio.MonitoringDevice"
				".Default"));
1231 1232 1233
	config_set_default_uint  (basicConfig, "Audio", "SampleRate", 44100);
	config_set_default_string(basicConfig, "Audio", "ChannelSetup",
			"Stereo");
S
Shaolin 已提交
1234 1235
	config_set_default_double(basicConfig, "Audio", "MeterDecayRate",
			VOLUME_METER_DECAY_FAST);
1236
	config_set_default_uint  (basicConfig, "Audio", "PeakMeterType", 0);
1237 1238 1239 1240 1241 1242

	return true;
}

bool OBSBasic::InitBasicConfig()
{
P
Palana 已提交
1243 1244
	ProfileScope("OBSBasic::InitBasicConfig");

1245
	char configPath[512];
J
jp9000 已提交
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258

	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");
1259 1260 1261 1262
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
1263

1264 1265 1266
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
1267 1268 1269
		return false;
	}

J
jp9000 已提交
1270 1271 1272 1273 1274
	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);
1275
		basicConfig.SaveSafe("tmp");
J
jp9000 已提交
1276 1277
	}

1278 1279 1280
	return InitBasicConfigDefaults();
}

1281 1282
void OBSBasic::InitOBSCallbacks()
{
P
Palana 已提交
1283 1284
	ProfileScope("OBSBasic::InitOBSCallbacks");

P
Palana 已提交
1285
	signalHandlers.reserve(signalHandlers.size() + 6);
1286 1287
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_create",
			OBSBasic::SourceCreated, this);
P
Palana 已提交
1288
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_remove",
1289
			OBSBasic::SourceRemoved, this);
P
Palana 已提交
1290
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_activate",
1291
			OBSBasic::SourceActivated, this);
P
Palana 已提交
1292
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_deactivate",
1293
			OBSBasic::SourceDeactivated, this);
P
Palana 已提交
1294
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
1295
			OBSBasic::SourceRenamed, this);
1296 1297
}

J
jp9000 已提交
1298 1299
void OBSBasic::InitPrimitives()
{
P
Palana 已提交
1300 1301
	ProfileScope("OBSBasic::InitPrimitives");

J
jp9000 已提交
1302
	obs_enter_graphics();
J
jp9000 已提交
1303

1304
	gs_render_start(true);
J
jp9000 已提交
1305 1306 1307 1308 1309
	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);
1310
	box = gs_render_save();
J
jp9000 已提交
1311

1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
	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();

1332
	gs_render_start(true);
J
jp9000 已提交
1333 1334 1335 1336
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
1337
	circle = gs_render_save();
J
jp9000 已提交
1338

J
jp9000 已提交
1339
	obs_leave_graphics();
J
jp9000 已提交
1340 1341
}

J
jp9000 已提交
1342 1343 1344 1345 1346 1347 1348 1349
void OBSBasic::ReplayBufferClicked()
{
	if (outputHandler->ReplayBufferActive())
		StopReplayBuffer();
	else
		StartReplayBuffer();
};

J
jp9000 已提交
1350 1351
void OBSBasic::ResetOutputs()
{
P
Palana 已提交
1352 1353
	ProfileScope("OBSBasic::ResetOutputs");

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

J
jp9000 已提交
1357 1358
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
1359 1360 1361
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
1362

J
jp9000 已提交
1363 1364 1365 1366 1367 1368
		delete replayBufferButton;

		if (outputHandler->replayBuffer) {
			replayBufferButton = new QPushButton(
					QTStr("Basic.Main.StartReplayBuffer"),
					this);
1369
			replayBufferButton->setCheckable(true);
1370
			connect(replayBufferButton.data(),
J
jp9000 已提交
1371 1372 1373 1374
					&QPushButton::clicked,
					this,
					&OBSBasic::ReplayBufferClicked);

1375
			replayBufferButton->setProperty("themeID", "replayBufferButton");
J
jp9000 已提交
1376 1377
			ui->buttonsVLayout->insertWidget(2, replayBufferButton);
		}
1378

J
jp9000 已提交
1379 1380 1381
		if (sysTrayReplayBuffer)
			sysTrayReplayBuffer->setEnabled(
					!!outputHandler->replayBuffer);
J
jp9000 已提交
1382 1383 1384 1385 1386
	} else {
		outputHandler->Update();
	}
}

J
jp9000 已提交
1387 1388 1389
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot);

1390 1391 1392 1393
#define STARTUP_SEPARATOR \
	"==== Startup complete ==============================================="
#define SHUTDOWN_SEPARATOR \
	"==== Shutting down =================================================="
1394

1395 1396 1397 1398 1399
#define UNSUPPORTED_ERROR \
	"Failed to initialize video:\n\nRequired graphics API functionality " \
	"not found.  Your GPU may not be supported."

#define UNKNOWN_ERROR \
1400 1401
	"Failed to initialize video.  Your GPU may not be supported, " \
	"or your graphics drivers may need to be updated."
1402

1403 1404
void OBSBasic::OBSInit()
{
P
Palana 已提交
1405 1406
	ProfileScope("OBSBasic::OBSInit");

J
jp9000 已提交
1407 1408
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
1409
	char savePath[512];
J
jp9000 已提交
1410 1411 1412 1413 1414 1415 1416 1417
	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);
1418
	if (ret <= 0)
J
jp9000 已提交
1419 1420 1421 1422 1423
		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";
1424

1425 1426
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
1427
	if (!ResetAudio())
1428 1429
		throw "Failed to initialize audio";

1430
	ret = ResetVideo();
1431 1432 1433 1434 1435

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
1436
		throw UNSUPPORTED_ERROR;
1437 1438 1439 1440
	case OBS_VIDEO_INVALID_PARAM:
		throw "Failed to initialize video:  Invalid parameters";
	default:
		if (ret != OBS_VIDEO_SUCCESS)
1441
			throw UNKNOWN_ERROR;
1442 1443
	}

1444
	/* load audio monitoring */
1445
#if defined(_WIN32) || defined(__APPLE__) || HAVE_PULSEAUDIO
1446 1447 1448 1449 1450 1451
	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);
1452 1453 1454

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

1457
	InitOBSCallbacks();
P
Palana 已提交
1458
	InitHotkeys();
1459

1460
	AddExtraModulePaths();
1461
	blog(LOG_INFO, "---------------------------------");
J
jp9000 已提交
1462
	obs_load_all_modules();
1463 1464
	blog(LOG_INFO, "---------------------------------");
	obs_log_loaded_modules();
1465 1466
	blog(LOG_INFO, "---------------------------------");
	obs_post_load_modules();
J
jp9000 已提交
1467

J
jp9000 已提交
1468 1469 1470 1471
#ifdef _WIN32
	create_browser_widget = obs_browser_init_panel();
#endif

1472 1473
	CheckForSimpleModeX264Fallback();

1474
	blog(LOG_INFO, STARTUP_SEPARATOR);
1475

J
jp9000 已提交
1476
	ResetOutputs();
1477
	CreateHotkeys();
J
jp9000 已提交
1478

1479 1480 1481
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
1482 1483
	InitPrimitives();

1484 1485 1486 1487 1488 1489
	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 已提交
1490 1491 1492 1493 1494 1495 1496 1497

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

1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
#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

1513 1514 1515 1516 1517 1518 1519 1520 1521
#ifndef __APPLE__
	{
		ProfileScope("OBSBasic::Load");
		disableSaving--;
		Load(savePath);
		disableSaving++;
	}
#endif

J
jp9000 已提交
1522
	TimedCheckForUpdates();
1523
	loaded = true;
J
jp9000 已提交
1524

1525
	previewEnabled = config_get_bool(App()->GlobalConfig(),
J
jp9000 已提交
1526
			"BasicWindow", "PreviewEnabled");
1527 1528 1529 1530 1531

	if (!previewEnabled && !IsPreviewProgramMode())
		QMetaObject::invokeMethod(this, "EnablePreviewDisplay",
				Qt::QueuedConnection,
				Q_ARG(bool, previewEnabled));
1532 1533 1534 1535 1536 1537 1538 1539 1540

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

1542 1543 1544
#ifndef __APPLE__
	RefreshSceneCollections();
#endif
J
jp9000 已提交
1545
	RefreshProfiles();
J
jp9000 已提交
1546
	disableSaving--;
1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559

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

1560
#ifdef _WIN32
J
jp9000 已提交
1561
	SetWin32DropStyle(this);
1562 1563 1564 1565 1566
	show();
#endif

	bool alwaysOnTop = config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"AlwaysOnTop");
1567
	if (alwaysOnTop || opt_always_on_top) {
1568 1569 1570 1571 1572
		SetAlwaysOnTop(this, true);
		ui->actionAlwaysOnTop->setChecked(true);
	}

#ifndef _WIN32
1573
	show();
1574
#endif
J
jp9000 已提交
1575

J
jp9000 已提交
1576 1577 1578 1579
	const char *dockStateStr = config_get_string(App()->GlobalConfig(),
			"BasicWindow", "DockState");
	if (!dockStateStr) {
		on_resetUI_triggered();
J
jp9000 已提交
1580
	} else {
J
jp9000 已提交
1581 1582 1583 1584
		QByteArray dockState =
			QByteArray::fromBase64(QByteArray(dockStateStr));
		if (!restoreState(dockState))
			on_resetUI_triggered();
J
jp9000 已提交
1585 1586
	}

J
jp9000 已提交
1587 1588 1589 1590 1591 1592 1593 1594 1595
	config_set_default_bool(App()->GlobalConfig(), "BasicWindow",
			"DocksLocked", true);

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

1597
#ifndef __APPLE__
C
cg2121 已提交
1598
	SystemTray(true);
1599
#endif
C
cg2121 已提交
1600

1601 1602 1603
	if (windowState().testFlag(Qt::WindowFullScreen))
		fullscreenInterface = true;

J
jp9000 已提交
1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
	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 =
1620
			OBSMessageBox::question(this, QTStr("Basic.AutoConfig"),
J
jp9000 已提交
1621 1622 1623 1624 1625 1626
					msg);

		if (button == QMessageBox::Yes) {
			on_autoConfigure_triggered();
		} else {
			msg = QTStr("Basic.FirstStartup.RunWizard.NoClicked");
1627
			OBSMessageBox::information(this,
J
jp9000 已提交
1628 1629 1630
					QTStr("Basic.AutoConfig"), msg);
		}
	}
1631

1632 1633
	ToggleMixerLayout(config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"VerticalVolControl"));
S
Shaolin 已提交
1634

1635 1636
	if (config_get_bool(basicConfig, "General", "OpenStatsOnStartup"))
		on_stats_triggered();
1637 1638

	OBSBasicStats::InitializeValues();
J
jp9000 已提交
1639 1640 1641 1642 1643 1644

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

	ui->viewMenu->addSeparator();

1645
	multiviewProjectorMenu = new QMenu(QTStr("MultiviewProjector"));
J
jp9000 已提交
1646
	ui->viewMenu->addMenu(multiviewProjectorMenu);
1647 1648
	connect(ui->viewMenu->menuAction(), &QAction::hovered, this,
			&OBSBasic::UpdateMultiviewProjectorMenu);
J
jp9000 已提交
1649 1650
	ui->viewMenu->addAction(QTStr("MultiviewWindowed"),
			this, SLOT(OpenMultiviewWindow()));
C
cg2121 已提交
1651 1652

#if !defined(_WIN32) && !defined(__APPLE__)
J
jp9000 已提交
1653 1654 1655
	delete ui->actionShowCrashLogs;
	delete ui->actionUploadLastCrashLog;
	delete ui->menuCrashLogs;
C
cg2121 已提交
1656
	delete ui->actionCheckForUpdates;
J
jp9000 已提交
1657 1658 1659
	ui->actionShowCrashLogs = nullptr;
	ui->actionUploadLastCrashLog = nullptr;
	ui->menuCrashLogs = nullptr;
C
cg2121 已提交
1660 1661
	ui->actionCheckForUpdates = nullptr;
#endif
1662

1663
#ifdef __APPLE__
1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682
	/* 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));
1683 1684
#else
	OnFirstLoad();
1685
#endif
1686 1687
}

1688 1689 1690 1691
void OBSBasic::OnFirstLoad()
{
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_FINISHED_LOADING);
J
jp9000 已提交
1692 1693 1694 1695 1696 1697 1698 1699 1700 1701

#ifdef _WIN32
	/* Attempt to load init screen if available */
	if (create_browser_widget) {
		WhatsNewInfoThread *wnit = new WhatsNewInfoThread();
		if (wnit) {
			connect(wnit, &WhatsNewInfoThread::Result,
					this, &OBSBasic::ReceivedIntroJson);
		}
		if (wnit) {
1702
			introCheckThread.reset(wnit);
J
jp9000 已提交
1703 1704 1705 1706
			introCheckThread->start();
		}
	}
#endif
1707 1708
}

1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719
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));
1720
	RefreshSceneCollections();
1721
	OnFirstLoad();
1722 1723 1724 1725

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

J
jp9000 已提交
1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745
/* shows a "what's new" page on startup of new versions using CEF */
void OBSBasic::ReceivedIntroJson(const QString &text)
{
#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();
1746
		int rc = item["RC"].int_value();
J
jp9000 已提交
1747 1748 1749 1750 1751

		int major = 0;
		int minor = 0;

		sscanf(version.c_str(), "%d.%d", &major, &minor);
1752 1753 1754 1755 1756
#if OBS_RELEASE_CANDIDATE > 0
		if (major == OBS_RELEASE_CANDIDATE_MAJOR &&
		    minor == OBS_RELEASE_CANDIDATE_MINOR &&
		    rc == OBS_RELEASE_CANDIDATE) {
#else
J
jp9000 已提交
1757
		if (major == LIBOBS_API_MAJOR_VER &&
1758 1759 1760
		    minor == LIBOBS_API_MINOR_VER &&
		    rc == 0) {
#endif
J
jp9000 已提交
1761 1762 1763 1764 1765 1766 1767 1768 1769 1770
			info_url = url;
			info_increment = increment;
		}
	}

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

1771 1772 1773 1774
#if OBS_RELEASE_CANDIDATE > 0
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastRCVersion");
#else
J
jp9000 已提交
1775 1776
	uint32_t lastVersion = config_get_int(App()->GlobalConfig(), "General",
			"LastVersion");
1777
#endif
J
jp9000 已提交
1778 1779 1780

	int current_version_increment = -1;

1781 1782 1783
#if OBS_RELEASE_CANDIDATE > 0
	if (lastVersion < OBS_RELEASE_CANDIDATE_VER) {
#else
J
jp9000 已提交
1784
	if (lastVersion < LIBOBS_API_VER) {
1785
#endif
J
jp9000 已提交
1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802
		config_set_int(App()->GlobalConfig(), "General",
				"InfoIncrement", -1);
	} else {
		current_version_increment = config_get_int(
				App()->GlobalConfig(), "General",
				"InfoIncrement");
	}

	if (info_increment <= current_version_increment) {
		return;
	}

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

	QDialog dlg(this);
	dlg.setWindowTitle("What's New");
1803
	dlg.resize(700, 600);
J
jp9000 已提交
1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831

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

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

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

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

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

	dlg.exec();
#else
	UNUSED_PARAMETER(text);
#endif
}

1832 1833 1834 1835 1836 1837 1838
void OBSBasic::UpdateMultiviewProjectorMenu()
{
	multiviewProjectorMenu->clear();
	AddProjectorMenuMonitors(multiviewProjectorMenu, this,
			SLOT(OpenMultiviewProjector()));
}

P
Palana 已提交
1839 1840
void OBSBasic::InitHotkeys()
{
P
Palana 已提交
1841 1842
	ProfileScope("OBSBasic::InitHotkeys");

P
Palana 已提交
1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884
	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"),
1885
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905

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

1906 1907
void OBSBasic::CreateHotkeys()
{
P
Palana 已提交
1908 1909
	ProfileScope("OBSBasic::CreateHotkeys");

1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925
	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 已提交
1926 1927 1928 1929 1930 1931 1932 1933 1934
	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);
	};

1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947
	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);
	};

1948
#define MAKE_CALLBACK(pred, method, log_action) \
1949 1950 1951
	[](void *data, obs_hotkey_pair_id, obs_hotkey_t*, bool pressed) \
	{ \
		OBSBasic &basic = *static_cast<OBSBasic*>(data); \
1952
		if ((pred) && pressed) { \
1953
			blog(LOG_INFO, log_action " due to hotkey"); \
1954 1955 1956 1957 1958 1959 1960 1961
			method(); \
			return true; \
		} \
		return false; \
	}

	streamingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartStreaming",
J
jp9000 已提交
1962
			Str("Basic.Main.StartStreaming"),
1963
			"OBSBasic.StopStreaming",
J
jp9000 已提交
1964
			Str("Basic.Main.StopStreaming"),
1965 1966
			MAKE_CALLBACK(!basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
1967
				basic.StartStreaming, "Starting stream"),
1968 1969
			MAKE_CALLBACK(basic.outputHandler->StreamingActive() &&
				basic.ui->streamButton->isEnabled(),
1970
				basic.StopStreaming, "Stopping stream"),
1971 1972 1973 1974
			this, this);
	LoadHotkeyPair(streamingHotkeys,
			"OBSBasic.StartStreaming", "OBSBasic.StopStreaming");

J
jp9000 已提交
1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989
	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");

1990 1991
	recordingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartRecording",
J
jp9000 已提交
1992
			Str("Basic.Main.StartRecording"),
1993
			"OBSBasic.StopRecording",
J
jp9000 已提交
1994
			Str("Basic.Main.StopRecording"),
1995 1996
			MAKE_CALLBACK(!basic.outputHandler->RecordingActive() &&
				!basic.ui->recordButton->isChecked(),
1997
				basic.StartRecording, "Starting recording"),
1998 1999
			MAKE_CALLBACK(basic.outputHandler->RecordingActive() &&
				basic.ui->recordButton->isChecked(),
2000
				basic.StopRecording, "Stopping recording"),
2001 2002 2003
			this, this);
	LoadHotkeyPair(recordingHotkeys,
			"OBSBasic.StartRecording", "OBSBasic.StopRecording");
J
jp9000 已提交
2004 2005 2006 2007 2008 2009 2010

	replayBufHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartReplayBuffer",
			Str("Basic.Main.StartReplayBuffer"),
			"OBSBasic.StopReplayBuffer",
			Str("Basic.Main.StopReplayBuffer"),
			MAKE_CALLBACK(!basic.outputHandler->ReplayBufferActive(),
2011
				basic.StartReplayBuffer, "Starting replay buffer"),
J
jp9000 已提交
2012
			MAKE_CALLBACK(basic.outputHandler->ReplayBufferActive(),
2013
				basic.StopReplayBuffer, "Stopping replay buffer"),
J
jp9000 已提交
2014 2015 2016
			this, this);
	LoadHotkeyPair(replayBufHotkeys,
			"OBSBasic.StartReplayBuffer", "OBSBasic.StopReplayBuffer");
2017
#undef MAKE_CALLBACK
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

	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");
2047 2048
}

J
jp9000 已提交
2049 2050 2051 2052
void OBSBasic::ClearHotkeys()
{
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
J
jp9000 已提交
2053
	obs_hotkey_pair_unregister(replayBufHotkeys);
J
jp9000 已提交
2054
	obs_hotkey_unregister(forceStreamingStopHotkey);
2055 2056
	obs_hotkey_unregister(togglePreviewProgramHotkey);
	obs_hotkey_unregister(transitionHotkey);
J
jp9000 已提交
2057 2058
}

2059 2060
OBSBasic::~OBSBasic()
{
J
jp9000 已提交
2061 2062 2063
	if (updateCheckThread && updateCheckThread->isRunning())
		updateCheckThread->wait();

2064
	delete multiviewProjectorMenu;
P
pkviet 已提交
2065
	delete trayMenu;
2066 2067
	delete programOptions;
	delete program;
J
jp9000 已提交
2068

J
jp9000 已提交
2069 2070 2071 2072 2073 2074
	/* 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 已提交
2075
	delete cpuUsageTimer;
2076 2077
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
2078
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
J
jp9000 已提交
2079
	ClearHotkeys();
P
Palana 已提交
2080

2081
	service = nullptr;
J
jp9000 已提交
2082 2083
	outputHandler.reset();

J
John Bradley 已提交
2084 2085 2086
	if (interaction)
		delete interaction;

2087 2088 2089
	if (properties)
		delete properties;

J
jp9000 已提交
2090 2091 2092
	if (filters)
		delete filters;

2093 2094
	if (transformWindow)
		delete transformWindow;
2095

J
jp9000 已提交
2096 2097 2098
	if (advAudioWindow)
		delete advAudioWindow;

2099 2100 2101
	obs_display_remove_draw_callback(ui->preview->GetDisplay(),
			OBSBasic::RenderMain, this);

J
jp9000 已提交
2102
	obs_enter_graphics();
2103
	gs_vertexbuffer_destroy(box);
2104 2105 2106 2107
	gs_vertexbuffer_destroy(boxLeft);
	gs_vertexbuffer_destroy(boxTop);
	gs_vertexbuffer_destroy(boxRight);
	gs_vertexbuffer_destroy(boxBottom);
2108
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
2109
	obs_leave_graphics();
J
jp9000 已提交
2110

2111 2112 2113 2114 2115 2116 2117 2118 2119
	/* 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 已提交
2120 2121
	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
2122 2123 2124 2125
#if OBS_RELEASE_CANDIDATE > 0
	config_set_int(App()->GlobalConfig(), "General", "LastRCVersion",
			OBS_RELEASE_CANDIDATE_VER);
#endif
J
jp9000 已提交
2126

2127
	bool alwaysOnTop = IsAlwaysOnTop(this);
J
jp9000 已提交
2128

J
jp9000 已提交
2129 2130
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
2131 2132
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "AlwaysOnTop",
			alwaysOnTop);
2133 2134 2135 2136 2137 2138 2139 2140
	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 已提交
2141 2142
	config_set_bool(App()->GlobalConfig(), "BasicWindow",
			"DocksLocked", ui->lockUI->isChecked());
2143
	config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154

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

J
jp9000 已提交
2157 2158 2159 2160 2161 2162 2163 2164 2165
void OBSBasic::SaveProjectNow()
{
	if (disableSaving)
		return;

	projectChanged = true;
	SaveProjectDeferred();
}

J
jp9000 已提交
2166 2167
void OBSBasic::SaveProject()
{
J
jp9000 已提交
2168 2169 2170
	if (disableSaving)
		return;

J
jp9000 已提交
2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185
	projectChanged = true;
	QMetaObject::invokeMethod(this, "SaveProjectDeferred",
			Qt::QueuedConnection);
}

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

	if (!projectChanged)
		return;

	projectChanged = false;

J
jp9000 已提交
2186 2187
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
2188
	char savePath[512];
J
jp9000 已提交
2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200
	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);
2201 2202 2203
	if (ret <= 0)
		return;

J
jp9000 已提交
2204 2205 2206
	Save(savePath);
}

S
Shaolin 已提交
2207 2208 2209 2210 2211
OBSSource OBSBasic::GetProgramSource()
{
	return OBSGetStrongRef(programScene);
}

J
jp9000 已提交
2212
OBSScene OBSBasic::GetCurrentScene()
2213
{
J
jp9000 已提交
2214
	QListWidgetItem *item = ui->scenes->currentItem();
P
Palana 已提交
2215
	return item ? GetOBSRef<OBSScene>(item) : nullptr;
2216 2217
}

2218
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
2219
{
P
Palana 已提交
2220
	return item ? GetOBSRef<OBSSceneItem>(item) : nullptr;
J
jp9000 已提交
2221 2222
}

2223 2224
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
J
jp9000 已提交
2225
	return ui->sources->Get(GetTopSelectedSourceItem());
2226 2227
}

J
Joseph El-Khouri 已提交
2228 2229
void OBSBasic::UpdatePreviewScalingMenu()
{
2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
	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 已提交
2244
	ui->actionScaleOutput->setChecked(
2245
			scalingAmount == float(ovi.output_width) / float(ovi.base_width));
J
Joseph El-Khouri 已提交
2246 2247
}

2248
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
2249 2250 2251 2252 2253 2254 2255 2256 2257
{
	if (interaction)
		interaction->close();

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

2258
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
2259 2260 2261 2262 2263 2264 2265
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
2268 2269 2270 2271 2272 2273 2274 2275 2276 2277
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

2278 2279 2280
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
2281
{
2282
	const char *name  = obs_source_get_name(source);
2283
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
2284 2285

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

P
Palana 已提交
2289 2290 2291 2292 2293
	obs_hotkey_register_source(source, "OBSBasic.SelectScene",
			Str("Basic.Hotkeys.SelectScene"),
			[](void *data,
				obs_hotkey_id, obs_hotkey_t*, bool pressed)
	{
2294 2295 2296
		OBSBasic *main =
			reinterpret_cast<OBSBasic*>(App()->GetMainWindow());

P
Palana 已提交
2297 2298 2299
		auto potential_source = static_cast<obs_source_t*>(data);
		auto source = obs_source_get_ref(potential_source);
		if (source && pressed)
2300
			main->SetCurrentScene(source);
P
Palana 已提交
2301 2302 2303
		obs_source_release(source);
	}, static_cast<obs_source_t*>(source));

2304
	signal_handler_t *handler = obs_source_get_signal_handler(source);
2305

J
jp9000 已提交
2306 2307 2308
	SignalContainer<OBSScene> container;
	container.ref = scene;
	container.handlers.assign({
2309 2310 2311 2312 2313 2314 2315 2316
		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 已提交
2317
	});
2318 2319

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

2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338
	/* 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 已提交
2339
	SaveProject();
2340 2341 2342 2343 2344

	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 已提交
2345 2346

		OBSProjector::UpdateMultiviewProjectors();
2347
	}
2348 2349 2350

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2351 2352
}

2353
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
2354
{
P
Palana 已提交
2355 2356 2357 2358
	obs_scene_t *scene = obs_scene_from_source(source);

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

P
Palana 已提交
2360 2361 2362 2363 2364
	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 已提交
2365

P
Palana 已提交
2366 2367 2368
		sel = item;
		break;
	}
J
jp9000 已提交
2369

J
jp9000 已提交
2370
	if (sel != nullptr) {
P
Palana 已提交
2371
		if (sel == ui->scenes->currentItem())
J
jp9000 已提交
2372
			ui->sources->Clear();
J
jp9000 已提交
2373
		delete sel;
J
jp9000 已提交
2374
	}
J
jp9000 已提交
2375 2376

	SaveProject();
2377 2378 2379 2380

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

		OBSProjector::UpdateMultiviewProjectors();
2383
	}
2384 2385 2386

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
2387 2388
}

J
jp9000 已提交
2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401
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;
}

2402
void OBSBasic::AddSceneItem(OBSSceneItem item)
2403
{
2404
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
J
jp9000 已提交
2405

2406
	if (GetCurrentScene() == scene)
J
jp9000 已提交
2407
		ui->sources->Add(item);
J
jp9000 已提交
2408

J
jp9000 已提交
2409
	SaveProject();
2410 2411 2412 2413 2414 2415 2416 2417

	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 已提交
2418 2419
		
		obs_scene_enum_items(scene, select_one, (obs_sceneitem_t*)item);
2420
	}
2421 2422
}

2423
void OBSBasic::UpdateSceneSelection(OBSSource source)
2424 2425
{
	if (source) {
2426
		obs_scene_t *scene = obs_scene_from_source(source);
2427
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
2428

2429 2430 2431
		if (!scene)
			return;

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

2435 2436 2437 2438 2439
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

2440 2441 2442 2443
			OBSScene curScene =
				GetOBSRef<OBSScene>(ui->scenes->currentItem());
			if (api && scene != curScene)
				api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);
2444
		}
J
jp9000 已提交
2445
	}
2446 2447
}

J
jp9000 已提交
2448 2449 2450 2451 2452 2453 2454 2455 2456 2457
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 已提交
2458 2459
void OBSBasic::RenameSources(OBSSource source, QString newName,
		QString prevName)
J
jp9000 已提交
2460 2461
{
	RenameListValues(ui->scenes,  newName, prevName);
2462 2463 2464 2465 2466

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

2468
	OBSProjector::RenameProjector(prevName, newName);
C
cg2121 已提交
2469

J
jp9000 已提交
2470
	SaveProject();
S
Shaolin 已提交
2471 2472 2473 2474

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

2477 2478
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
J
jp9000 已提交
2479 2480
	SignalBlocker sourcesSignalBlocker(ui->sources);

2481
	if (scene != GetCurrentScene() || ignoreSelectionUpdate)
2482 2483
		return;

J
jp9000 已提交
2484
	ui->sources->SelectItem(item, select);
2485 2486
}

2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502
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);
}

2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520
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);
}

2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570
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);
	}
}

2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595
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;
		}

2596 2597
		OBSSource sourceTest = obs_get_source_by_name(name.c_str());
		obs_source_release(sourceTest);
2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610

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

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

2611 2612 2613 2614
void OBSBasic::VolControlContextMenu()
{
	VolControl *vol = reinterpret_cast<VolControl*>(sender());

2615 2616 2617 2618
	/* ------------------- */

	QAction hideAction(QTStr("Hide"), this);
	QAction unhideAllAction(QTStr("UnhideAll"), this);
2619
	QAction mixerRenameAction(QTStr("Rename"), this);
2620

2621 2622
	QAction filtersAction(QTStr("Filters"), this);
	QAction propertiesAction(QTStr("Properties"), this);
2623
	QAction advPropAction(QTStr("Basic.MainMenu.Edit.AdvAudio"), this);
2624

S
Shaolin 已提交
2625 2626 2627 2628 2629
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

2630 2631 2632 2633 2634 2635 2636 2637
	/* ------------------- */

	connect(&hideAction, &QAction::triggered,
			this, &OBSBasic::HideAudioControl,
			Qt::DirectConnection);
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);
2638 2639 2640
	connect(&mixerRenameAction, &QAction::triggered,
			this, &OBSBasic::MixerRenameSource,
			Qt::DirectConnection);
2641

2642 2643 2644 2645 2646 2647
	connect(&filtersAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceFilters,
			Qt::DirectConnection);
	connect(&propertiesAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceProperties,
			Qt::DirectConnection);
2648 2649 2650
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);
2651

2652 2653
	/* ------------------- */

S
Shaolin 已提交
2654 2655 2656 2657 2658 2659
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

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

2660 2661
	hideAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2662 2663
	mixerRenameAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
2664

2665 2666 2667 2668 2669
	filtersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
	propertiesAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));

2670 2671
	/* ------------------- */

2672
	QMenu popup;
2673 2674
	popup.addAction(&unhideAllAction);
	popup.addAction(&hideAction);
2675
	popup.addAction(&mixerRenameAction);
2676
	popup.addSeparator();
S
Shaolin 已提交
2677 2678
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
2679 2680
	popup.addAction(&filtersAction);
	popup.addAction(&propertiesAction);
2681
	popup.addAction(&advPropAction);
2682 2683 2684
	popup.exec(QCursor::pos());
}

S
Shaolin 已提交
2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695
void OBSBasic::on_hMixerScrollArea_customContextMenuRequested()
{
	StackedMixerAreaContextMenuRequested();
}

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

void OBSBasic::StackedMixerAreaContextMenuRequested()
2696 2697
{
	QAction unhideAllAction(QTStr("UnhideAll"), this);
S
SuslikV 已提交
2698 2699 2700

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

S
Shaolin 已提交
2701 2702 2703 2704 2705
	QAction toggleControlLayoutAction(QTStr("VerticalLayout"), this);
	toggleControlLayoutAction.setCheckable(true);
	toggleControlLayoutAction.setChecked(config_get_bool(GetGlobalConfig(),
			"BasicWindow", "VerticalVolControl"));

S
SuslikV 已提交
2706 2707
	/* ------------------- */

2708 2709 2710 2711
	connect(&unhideAllAction, &QAction::triggered,
			this, &OBSBasic::UnhideAllAudioControls,
			Qt::DirectConnection);

S
SuslikV 已提交
2712 2713 2714 2715 2716 2717
	connect(&advPropAction, &QAction::triggered,
			this, &OBSBasic::on_actionAdvAudioProperties_triggered,
			Qt::DirectConnection);

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

S
Shaolin 已提交
2718 2719 2720 2721 2722 2723
	connect(&toggleControlLayoutAction, &QAction::changed, this,
			&OBSBasic::ToggleVolControlLayout,
			Qt::DirectConnection);

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

2724
	QMenu popup;
2725
	popup.addAction(&unhideAllAction);
S
SuslikV 已提交
2726
	popup.addSeparator();
S
Shaolin 已提交
2727 2728
	popup.addAction(&toggleControlLayoutAction);
	popup.addSeparator();
S
SuslikV 已提交
2729
	popup.addAction(&advPropAction);
2730 2731 2732
	popup.exec(QCursor::pos());
}

2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743
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 已提交
2744 2745 2746 2747 2748 2749
void OBSBasic::ToggleVolControlLayout()
{
	bool vertical = !config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	config_set_bool(GetGlobalConfig(), "BasicWindow", "VerticalVolControl",
			vertical);
2750
	ToggleMixerLayout(vertical);
S
Shaolin 已提交
2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763

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

2764 2765
void OBSBasic::ActivateAudioSource(OBSSource source)
{
2766 2767 2768
	if (SourceMixerHidden(source))
		return;

S
Shaolin 已提交
2769 2770 2771
	bool vertical = config_get_bool(GetGlobalConfig(), "BasicWindow",
			"VerticalVolControl");
	VolControl *vol = new VolControl(source, true, vertical);
2772

S
Shaolin 已提交
2773 2774 2775
	double meterDecayRate = config_get_double(basicConfig, "Audio",
			"MeterDecayRate");
	vol->SetMeterDecayRate(meterDecayRate);
2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794

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

2795 2796 2797 2798
	vol->setContextMenuPolicy(Qt::CustomContextMenu);

	connect(vol, &QWidget::customContextMenuRequested,
			this, &OBSBasic::VolControlContextMenu);
2799 2800
	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
2801

2802 2803 2804
	InsertQObjectByName(volumes, vol);

	for (auto volume : volumes) {
S
Shaolin 已提交
2805 2806 2807 2808
		if (vertical)
			ui->vVolControlLayout->addWidget(volume);
		else
			ui->hVolControlLayout->addWidget(volume);
2809
	}
2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822
}

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

2823
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
2824
{
J
jp9000 已提交
2825
	if (obs_source_get_type(source) == OBS_SOURCE_TYPE_SCENE &&
2826
	    !obs_source_is_group(source)) {
2827 2828 2829
		int count = ui->scenes->count();

		if (count == 1) {
2830
			OBSMessageBox::information(this,
2831 2832 2833 2834
						QTStr("FinalScene.Title"),
						QTStr("FinalScene.Text"));
			return false;
		}
2835 2836
	}

2837
	const char *name  = obs_source_get_name(source);
2838 2839 2840

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

2842
	QMessageBox remove_source(this);
2843 2844 2845
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
2846 2847 2848 2849 2850 2851
	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();
2852
}
J
jp9000 已提交
2853

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

P
Palana 已提交
2856 2857 2858 2859 2860
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
2861 2862
void OBSBasic::TimedCheckForUpdates()
{
J
jp9000 已提交
2863 2864 2865 2866
	if (!config_get_bool(App()->GlobalConfig(), "General",
				"EnableAutoUpdates"))
		return;

P
Palana 已提交
2867 2868 2869
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
2870
#elif _WIN32
J
jp9000 已提交
2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885
	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 已提交
2886
		CheckForUpdates(false);
P
Palana 已提交
2887
#endif
J
jp9000 已提交
2888 2889
}

J
jp9000 已提交
2890
void OBSBasic::CheckForUpdates(bool manualUpdate)
J
jp9000 已提交
2891
{
P
Palana 已提交
2892 2893
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
2894
#elif _WIN32
J
jp9000 已提交
2895 2896
	ui->actionCheckForUpdates->setEnabled(false);

J
jp9000 已提交
2897 2898
	if (updateCheckThread && updateCheckThread->isRunning())
		return;
2899

2900
	updateCheckThread.reset(new AutoUpdateThread(manualUpdate));
2901
	updateCheckThread->start();
P
Palana 已提交
2902
#endif
2903 2904

	UNUSED_PARAMETER(manualUpdate);
J
jp9000 已提交
2905 2906
}

J
jp9000 已提交
2907
void OBSBasic::updateCheckFinished()
J
jp9000 已提交
2908 2909 2910 2911
{
	ui->actionCheckForUpdates->setEnabled(true);
}

2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941
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()) {
2942
			OBSMessageBox::information(this,
2943 2944 2945 2946 2947 2948 2949
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
			continue;
		}

		obs_source_t *source = obs_get_source_by_name(name.c_str());
		if (source) {
2950
			OBSMessageBox::information(this,
2951 2952 2953 2954 2955 2956 2957 2958
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));

			obs_source_release(source);
			continue;
		}

		obs_scene_t *scene = obs_scene_duplicate(curScene,
2959
				name.c_str(), OBS_SCENE_DUP_REFS);
2960
		source = obs_scene_get_source(scene);
2961
		SetCurrentScene(source, true);
2962
		obs_scene_release(scene);
J
jp9000 已提交
2963

2964
		break;
2965 2966 2967
	}
}

2968 2969 2970 2971
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
2972
		obs_source_t *source = obs_scene_get_source(scene);
J
jp9000 已提交
2973
		if (QueryRemoveSource(source)) {
2974
			obs_source_remove(source);
J
jp9000 已提交
2975 2976 2977 2978

			if (api)
				api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);
		}
2979 2980 2981 2982 2983 2984 2985
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
2986
		obs_source_t *source = obs_sceneitem_get_source(item);
2987
		if (QueryRemoveSource(source))
J
jp9000 已提交
2988 2989 2990 2991
			obs_sceneitem_remove(item);
	}
}

2992 2993
void OBSBasic::ReorderSources(OBSScene scene)
{
2994
	if (scene != GetCurrentScene() || ui->sources->IgnoreReorder())
2995 2996
		return;

J
jp9000 已提交
2997
	ui->sources->ReorderItems();
J
jp9000 已提交
2998
	SaveProject();
2999 3000
}

3001 3002
/* OBS Callbacks */

3003 3004 3005 3006 3007 3008 3009 3010 3011 3012
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)));
}

3013
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
3014 3015 3016
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

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

3019 3020
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
3021 3022
}

3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044
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));
3045

3046 3047
}

3048
void OBSBasic::SourceCreated(void *data, calldata_t *params)
3049
{
3050
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3051

3052
	if (obs_scene_from_source(source) != NULL)
3053
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
3054
				"AddScene", WaitConnection(),
3055
				Q_ARG(OBSSource, OBSSource(source)));
3056 3057
}

3058
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
3059
{
3060
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3061

3062
	if (obs_scene_from_source(source) != NULL)
3063 3064 3065
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
3066 3067
}

3068
void OBSBasic::SourceActivated(void *data, calldata_t *params)
3069
{
3070
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3071 3072 3073 3074 3075 3076 3077 3078
	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)));
}

3079
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
3080
{
3081
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
3082 3083 3084 3085 3086 3087 3088 3089
	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)));
}

3090
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
3091
{
S
Shaolin 已提交
3092
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
J
jp9000 已提交
3093 3094 3095 3096 3097
	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 已提交
3098
			Q_ARG(OBSSource, source),
J
jp9000 已提交
3099 3100
			Q_ARG(QString, QT_UTF8(newName)),
			Q_ARG(QString, QT_UTF8(prevName)));
3101 3102

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

3105 3106 3107 3108 3109
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

3110
	gs_effect_t    *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
3111 3112
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
3113 3114 3115

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
3116
	gs_effect_set_vec4(color, &colorVal);
3117

3118 3119
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
3120 3121 3122 3123 3124 3125 3126 3127
	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();
3128 3129
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
3130 3131 3132 3133

	gs_load_vertexbuffer(nullptr);
}

3134 3135
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
3136
	OBSBasic *window = static_cast<OBSBasic*>(data);
3137 3138 3139 3140
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
3141 3142
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
3143 3144 3145

	gs_viewport_push();
	gs_projection_push();
3146 3147 3148

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

3149 3150
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
3151
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
3152
			window->previewCX, window->previewCY);
3153

3154 3155
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

3156 3157 3158 3159 3160 3161
	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 已提交
3162
		obs_render_main_texture();
3163
	}
3164
	gs_load_vertexbuffer(nullptr);
3165

3166 3167
	/* --------------------------------------- */

3168 3169 3170
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
3171 3172 3173 3174

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
3175
	gs_reset_viewport();
J
jp9000 已提交
3176 3177 3178

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

3179 3180
	/* --------------------------------------- */

3181 3182
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
3183 3184 3185

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
3186 3187
}

3188 3189
/* Main class functions */

3190
obs_service_t *OBSBasic::GetService()
3191
{
3192
	if (!service) {
3193 3194
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
3195 3196
		obs_service_release(service);
	}
3197 3198 3199
	return service;
}

3200
void OBSBasic::SetService(obs_service_t *newService)
3201
{
3202
	if (newService)
3203 3204 3205
		service = newService;
}

3206
bool OBSBasic::StreamingActive() const
3207 3208 3209 3210 3211 3212
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

3213 3214 3215 3216 3217 3218 3219
bool OBSBasic::Active() const
{
	if (!outputHandler)
		return false;
	return outputHandler->Active();
}

3220 3221 3222 3223 3224 3225
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

3226 3227
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
3228
	return obs_reset_video(ovi);
3229 3230
}

3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243
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;
}

3244 3245 3246 3247 3248 3249
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 已提交
3250 3251
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
3252 3253 3254 3255 3256 3257 3258 3259 3260
#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 已提交
3261
		return VIDEO_FORMAT_RGBA;
3262 3263
}

3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274
void OBSBasic::ResetUI()
{
	bool studioPortraitLayout = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "StudioPortraitLayout");

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

3275
int OBSBasic::ResetVideo()
J
jp9000 已提交
3276
{
3277 3278 3279
	if (outputHandler && outputHandler->Active())
		return OBS_VIDEO_CURRENTLY_ACTIVE;

P
Palana 已提交
3280 3281
	ProfileScope("OBSBasic::ResetVideo");

J
jp9000 已提交
3282
	struct obs_video_info ovi;
3283
	int ret;
J
jp9000 已提交
3284

3285
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
3286

3287 3288 3289 3290 3291 3292 3293
	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 已提交
3294
	ovi.graphics_module = App()->GetRenderModule();
3295
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3296
			"Video", "BaseCX");
3297
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3298
			"Video", "BaseCY");
3299
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3300
			"Video", "OutputCX");
3301
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
3302
			"Video", "OutputCY");
3303 3304 3305 3306 3307
	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;
3308 3309
	ovi.adapter        = config_get_uint(App()->GlobalConfig(),
			"Video", "AdapterIdx");
J
jp9000 已提交
3310
	ovi.gpu_conversion = true;
3311
	ovi.scale_type     = GetScaleType(basicConfig);
3312

3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328
	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);
	}

3329
	ret = AttemptToResetVideo(&ovi);
3330
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
3331 3332 3333 3334 3335 3336
		if (ret == OBS_VIDEO_CURRENTLY_ACTIVE) {
			blog(LOG_WARNING, "Tried to reset when "
			                  "already active");
			return ret;
		}

3337
		/* Try OpenGL if DirectX fails on windows */
3338
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
3339 3340 3341 3342
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
3343 3344
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
3345 3346
			ret = AttemptToResetVideo(&ovi);
		}
3347 3348
	} else if (ret == OBS_VIDEO_SUCCESS) {
		ResizePreview(ovi.base_width, ovi.base_height);
3349 3350
		if (program)
			ResizeProgram(ovi.base_width, ovi.base_height);
3351 3352
	}

3353
	if (ret == OBS_VIDEO_SUCCESS) {
3354
		OBSBasicStats::InitializeValues();
3355 3356
		OBSProjector::UpdateMultiviewProjectors();
	}
3357

3358
	return ret;
J
jp9000 已提交
3359
}
J
jp9000 已提交
3360

3361
bool OBSBasic::ResetAudio()
J
jp9000 已提交
3362
{
P
Palana 已提交
3363 3364
	ProfileScope("OBSBasic::ResetAudio");

3365
	struct obs_audio_info ai;
3366
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
3367 3368
			"SampleRate");

3369
	const char *channelSetupStr = config_get_string(basicConfig,
3370 3371 3372 3373
			"Audio", "ChannelSetup");

	if (strcmp(channelSetupStr, "Mono") == 0)
		ai.speakers = SPEAKERS_MONO;
P
pkviet 已提交
3374 3375
	else if (strcmp(channelSetupStr, "2.1") == 0)
		ai.speakers = SPEAKERS_2POINT1;
P
pkviet 已提交
3376 3377
	else if (strcmp(channelSetupStr, "4.0") == 0)
		ai.speakers = SPEAKERS_4POINT0;
P
pkviet 已提交
3378 3379 3380 3381 3382 3383
	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;
3384 3385 3386
	else
		ai.speakers = SPEAKERS_STEREO;

J
jp9000 已提交
3387
	return obs_reset_audio(&ai);
J
jp9000 已提交
3388 3389
}

3390
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceId,
3391
		const char *deviceDesc, int channel)
J
jp9000 已提交
3392
{
3393
	bool disable = deviceId && strcmp(deviceId, "disabled") == 0;
3394 3395
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
3396 3397 3398

	source = obs_get_output_source(channel);
	if (source) {
3399 3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411
		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 已提交
3412 3413 3414

		obs_source_release(source);

3415 3416
	} else if (!disable) {
		settings = obs_data_create();
J
jp9000 已提交
3417
		obs_data_set_string(settings, "device_id", deviceId);
3418 3419
		source = obs_source_create(sourceId, deviceDesc, settings,
				nullptr);
J
jp9000 已提交
3420 3421 3422 3423 3424 3425 3426
		obs_data_release(settings);

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

J
jp9000 已提交
3427
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
3428
{
3429
	QSize  targetSize;
3430
	bool isFixedScaling;
J
Joseph El-Khouri 已提交
3431
	obs_video_info ovi;
J
jp9000 已提交
3432

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

3436
	isFixedScaling = ui->preview->IsFixedScaling();
J
Joseph El-Khouri 已提交
3437 3438
	obs_get_video_info(&ovi);

3439 3440
	if (isFixedScaling) {
		previewScale = ui->preview->GetScalingAmount();
J
Joseph El-Khouri 已提交
3441 3442 3443 3444
		GetCenterPosFromFixedScale(int(cx), int(cy),
				targetSize.width() - PREVIEW_EDGE_SIZE * 2,
				targetSize.height() - PREVIEW_EDGE_SIZE * 2,
				previewX, previewY, previewScale);
3445 3446
		previewX += ui->preview->GetScrollX();
		previewY += ui->preview->GetScrollY();
J
Joseph El-Khouri 已提交
3447 3448 3449 3450 3451 3452 3453

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

3455 3456
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);
J
jp9000 已提交
3457 3458
}

3459 3460 3461 3462 3463 3464 3465 3466 3467
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 已提交
3468 3469 3470 3471
	for (QPointer<QWidget> &projector : windowProjectors) {
		delete projector;
		projector.clear();
	}
3472 3473 3474 3475
	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}
J
jp9000 已提交
3476

3477
	if (!stats.isNull()) stats->close(); //call close to save Stats geometry
3478
	if (!remux.isNull()) remux->close();
3479 3480
}

3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503
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);
	}
}

3504 3505
void OBSBasic::ClearSceneData()
{
J
jp9000 已提交
3506 3507
	disableSaving++;

3508 3509 3510 3511
	CloseDialogs();

	ClearVolumeControls();
	ClearListItems(ui->scenes);
J
jp9000 已提交
3512
	ui->sources->Clear();
3513 3514
	ClearQuickTransitions();
	ui->transitions->clear();
3515 3516 3517 3518 3519 3520 3521

	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);
3522 3523 3524
	lastScene = nullptr;
	swapScene = nullptr;
	programScene = nullptr;
3525 3526 3527 3528 3529 3530 3531 3532 3533 3534

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

	obs_enum_sources(cb, nullptr);

3535 3536 3537
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CLEANUP);

J
jp9000 已提交
3538
	disableSaving--;
3539 3540 3541

	blog(LOG_INFO, "All scene data cleared");
	blog(LOG_INFO, "------------------------------------------------");
3542 3543
}

J
jp9000 已提交
3544
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
3545
{
3546 3547 3548 3549 3550
	if (isVisible())
		config_set_string(App()->GlobalConfig(),
				"BasicWindow", "geometry",
				saveGeometry().toBase64().constData());

J
jp9000 已提交
3551 3552 3553 3554
	config_set_string(App()->GlobalConfig(),
			"BasicWindow", "DockState",
			saveState().toBase64().constData());

3555
	if (outputHandler && outputHandler->Active()) {
C
cg2121 已提交
3556 3557
		SetShowing(true);

3558
		QMessageBox::StandardButton button = OBSMessageBox::question(
3559 3560 3561 3562 3563 3564 3565 3566 3567
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

3568 3569 3570 3571
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

3572 3573
	blog(LOG_INFO, SHUTDOWN_SEPARATOR);

J
jp9000 已提交
3574 3575
	if (introCheckThread)
		introCheckThread->wait();
3576 3577 3578 3579 3580
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
3581 3582
	signalHandlers.clear();

J
jp9000 已提交
3583
	SaveProjectNow();
J
jp9000 已提交
3584 3585 3586 3587

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

J
jp9000 已提交
3588
	disableSaving++;
J
jp9000 已提交
3589

3590 3591 3592
	/* Clear all scene data (dialogs, widgets, widget sub-items, scenes,
	 * sources, etc) so that all references are released before shutdown */
	ClearSceneData();
3593 3594

	App()->quit();
3595 3596
}

J
jp9000 已提交
3597
void OBSBasic::changeEvent(QEvent *event)
3598
{
3599 3600
	if (event->type() == QEvent::WindowStateChange &&
	    isMinimized() &&
3601
	    trayIcon &&
3602 3603 3604 3605 3606
	    trayIcon->isVisible() &&
	    sysTrayMinimizeToTray()) {

		ToggleShowHide();
	}
3607 3608
}

3609 3610
void OBSBasic::on_actionShow_Recordings_triggered()
{
3611
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
P
pkviet 已提交
3612 3613 3614 3615
	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");
3616
	const char *path = strcmp(mode, "Advanced") ?
P
pkviet 已提交
3617 3618
			config_get_string(basicConfig, "SimpleOutput", "FilePath") :
			adv_path;
3619 3620 3621
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
3622 3623
void OBSBasic::on_actionRemux_triggered()
{
3624 3625 3626 3627 3628 3629
	if (!remux.isNull()) {
		remux->show();
		remux->raise();
		return;
	}

3630 3631 3632 3633
	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");
3634 3635 3636 3637 3638

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

P
Palana 已提交
3641 3642 3643 3644
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
C
cg2121 已提交
3645
	SystemTray(false);
P
Palana 已提交
3646 3647
}

J
jp9000 已提交
3648 3649
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
3650 3651 3652 3653 3654
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
3655 3656 3657
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
3658 3659 3660

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

3663 3664 3665 3666 3667
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

3668 3669 3670 3671 3672
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

3673 3674
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
3675
{
3676
	obs_source_t *source = NULL;
J
jp9000 已提交
3677

3678 3679 3680 3681
	if (sceneChanging)
		return;

	if (current) {
3682
		obs_scene_t *scene;
J
jp9000 已提交
3683

P
Palana 已提交
3684
		scene = GetOBSRef<OBSScene>(current);
3685
		source = obs_scene_get_source(scene);
3686 3687
	}

3688
	SetCurrentScene(source);
3689

3690 3691 3692
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_PREVIEW_SCENE_CHANGED);

3693
	UNUSED_PARAMETER(prev);
3694 3695
}

J
jp9000 已提交
3696 3697
void OBSBasic::EditSceneName()
{
3698 3699 3700 3701 3702 3703
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

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

J
jp9000 已提交
3706 3707 3708 3709
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot)
{
	QAction *action;
3710 3711
	QList<QScreen*> screens = QGuiApplication::screens();
	for (int i = 0; i < screens.size(); i++) {
3712
		QRect screenGeometry = screens[i]->geometry();
J
jp9000 已提交
3713 3714 3715
		QString str = QString("%1 %2: %3x%4 @ %5,%6").
			arg(QTStr("Display"),
			    QString::number(i),
S
Shaolin 已提交
3716 3717 3718 3719
			    QString::number(screenGeometry.width()),
			    QString::number(screenGeometry.height()),
			    QString::number(screenGeometry.x()),
			    QString::number(screenGeometry.y()));
J
jp9000 已提交
3720 3721 3722 3723 3724 3725

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

J
jp9000 已提交
3726
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
3727
{
J
jp9000 已提交
3728
	QListWidgetItem *item = ui->scenes->itemAt(pos);
J
jp9000 已提交
3729
	QPointer<QMenu> sceneProjectorMenu;
J
jp9000 已提交
3730

3731
	QMenu popup(this);
J
jp9000 已提交
3732
	QMenu order(QTStr("Basic.MainMenu.Edit.Order"), this);
J
jp9000 已提交
3733 3734 3735
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
3736 3737
	if (item) {
		popup.addSeparator();
3738 3739
		popup.addAction(QTStr("Duplicate"),
				this, SLOT(DuplicateSelectedScene()));
P
Palana 已提交
3740 3741
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
3742
		popup.addAction(QTStr("Remove"),
3743
				this, SLOT(RemoveSelectedScene()));
J
jp9000 已提交
3744
		popup.addSeparator();
J
jp9000 已提交
3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757

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

J
jp9000 已提交
3759 3760 3761 3762
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
C
cg2121 已提交
3763 3764 3765 3766 3767 3768

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

		popup.addAction(sceneWindow);
J
jp9000 已提交
3769
		popup.addSeparator();
J
jp9000 已提交
3770 3771
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
3772 3773 3774 3775 3776

		popup.addSeparator();

		QMenu *transitionMenu = CreatePerSceneTransitionMenu();
		popup.addMenu(transitionMenu);
S
Shaolin 已提交
3777 3778 3779

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

J
jp9000 已提交
3780 3781
		QAction *multiviewAction = popup.addAction(
				QTStr("ShowInMultiview"));
S
Shaolin 已提交
3782

J
jp9000 已提交
3783 3784 3785
		OBSSource source = GetCurrentSceneSource();
		OBSData data = obs_source_get_private_settings(source);
		obs_data_release(data);
S
Shaolin 已提交
3786

J
jp9000 已提交
3787 3788 3789
		obs_data_set_default_bool(data, "show_in_multiview",
				true);
		bool show = obs_data_get_bool(data, "show_in_multiview");
S
Shaolin 已提交
3790

J
jp9000 已提交
3791 3792
		multiviewAction->setCheckable(true);
		multiviewAction->setChecked(show);
S
Shaolin 已提交
3793

3794
		auto showInMultiview = [] (OBSData data)
J
jp9000 已提交
3795 3796 3797 3798 3799 3800 3801 3802 3803 3804
		{
			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 已提交
3805
	}
J
jp9000 已提交
3806 3807

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

J
jp9000 已提交
3810
void OBSBasic::on_actionAddScene_triggered()
3811
{
3812
	string name;
S
Socapex 已提交
3813
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
3814

3815
	int i = 2;
P
Palana 已提交
3816
	QString placeHolderText = format.arg(i);
3817
	obs_source_t *source = nullptr;
P
Palana 已提交
3818 3819
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
3820
		placeHolderText = format.arg(++i);
P
Palana 已提交
3821
	}
S
Socapex 已提交
3822

J
jp9000 已提交
3823
	bool accepted = NameDialog::AskForName(this,
3824 3825
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
3826 3827
			name,
			placeHolderText);
3828

J
jp9000 已提交
3829
	if (accepted) {
J
jp9000 已提交
3830
		if (name.empty()) {
3831
			OBSMessageBox::information(this,
3832 3833
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
3834 3835 3836 3837
			on_actionAddScene_triggered();
			return;
		}

3838
		obs_source_t *source = obs_get_source_by_name(name.c_str());
3839
		if (source) {
3840
			OBSMessageBox::information(this,
3841 3842
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
3843 3844

			obs_source_release(source);
J
jp9000 已提交
3845
			on_actionAddScene_triggered();
3846 3847 3848
			return;
		}

3849
		obs_scene_t *scene = obs_scene_create(name.c_str());
3850
		source = obs_scene_get_source(scene);
3851
		SetCurrentScene(source);
3852
		obs_scene_release(scene);
3853
	}
3854 3855
}

J
jp9000 已提交
3856
void OBSBasic::on_actionRemoveScene_triggered()
3857
{
3858
	OBSScene     scene  = GetCurrentScene();
3859
	obs_source_t *source = obs_scene_get_source(scene);
3860 3861 3862

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
3863 3864
}

J
jp9000 已提交
3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882
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;
3883 3884

	OBSProjector::UpdateMultiviewProjectors();
J
jp9000 已提交
3885 3886
}

J
jp9000 已提交
3887
void OBSBasic::on_actionSceneUp_triggered()
3888
{
J
jp9000 已提交
3889
	ChangeSceneIndex(true, -1, 0);
3890 3891
}

J
jp9000 已提交
3892
void OBSBasic::on_actionSceneDown_triggered()
3893
{
J
jp9000 已提交
3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905
	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);
3906 3907
}

J
jp9000 已提交
3908 3909
void OBSBasic::EditSceneItemName()
{
J
jp9000 已提交
3910 3911
	int idx = GetTopSelectedSourceItem();
	ui->sources->Edit(idx);
J
jp9000 已提交
3912 3913
}

J
jp9000 已提交
3914 3915 3916 3917 3918 3919 3920 3921 3922 3923 3924 3925 3926 3927 3928 3929 3930 3931 3932 3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978
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);
}

QMenu *OBSBasic::AddDeinterlacingMenu(obs_source_t *source)
{
	QMenu *menu = new QMenu(QTStr("Deinterlacing"));
	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;
}

3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010
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);
}

QMenu *OBSBasic::AddScaleFilteringMenu(obs_sceneitem_t *item)
{
	QMenu *menu = new QMenu(QTStr("ScaleFiltering"));
	obs_scale_type scaleFilter = obs_sceneitem_get_scale_filter(item);
	QAction *action;

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

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

	return menu;
}

4011 4012 4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067
QMenu *OBSBasic::AddBackgroundColorMenu(obs_sceneitem_t *item)
{
	QMenu *menu = new QMenu(QTStr("ChangeBG"));
	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();

	QWidgetAction *widgetAction = new QWidgetAction(menu);
	ColorSelect *select = new ColorSelect(menu);
	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;
}

J
jp9000 已提交
4068
void OBSBasic::CreateSourcePopupMenu(int idx, bool preview)
4069
{
4070
	QMenu popup(this);
J
jp9000 已提交
4071 4072
	QPointer<QMenu> previewProjector;
	QPointer<QMenu> sourceProjector;
J
jp9000 已提交
4073 4074 4075 4076 4077 4078

	if (preview) {
		QAction *action = popup.addAction(
				QTStr("Basic.Main.PreviewConextMenu.Enable"),
				this, SLOT(TogglePreview()));
		action->setCheckable(true);
4079 4080
		action->setChecked(
				obs_display_enabled(ui->preview->GetDisplay()));
4081 4082
		if (IsPreviewProgramMode())
			action->setEnabled(false);
J
jp9000 已提交
4083

J
Joseph El-Khouri 已提交
4084 4085
		popup.addAction(ui->actionLockPreview);
		popup.addMenu(ui->scalingMenu);
J
jp9000 已提交
4086

J
jp9000 已提交
4087 4088
		previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
4089
				SLOT(OpenPreviewProjector()));
J
jp9000 已提交
4090 4091 4092

		popup.addMenu(previewProjector);

C
cg2121 已提交
4093 4094
		QAction *previewWindow = popup.addAction(
				QTStr("PreviewWindow"),
4095
				this, SLOT(OpenPreviewWindow()));
C
cg2121 已提交
4096 4097 4098

		popup.addAction(previewWindow);

J
jp9000 已提交
4099 4100 4101
		popup.addSeparator();
	}

J
jp9000 已提交
4102 4103 4104 4105
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

4106
	ui->actionCopyFilters->setEnabled(false);
4107
	ui->actionCopySource->setEnabled(false);
4108

J
jp9000 已提交
4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119
	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()));
	}

4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130
	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 已提交
4131
	if (idx != -1) {
J
jp9000 已提交
4132 4133 4134
		if (addSourceMenu)
			popup.addSeparator();

J
jp9000 已提交
4135
		OBSSceneItem sceneItem = ui->sources->Get(idx);
4136
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
jp9000 已提交
4137 4138 4139
		uint32_t flags = obs_source_get_output_flags(source);
		bool isAsyncVideo = (flags & OBS_SOURCE_ASYNC_VIDEO) ==
			OBS_SOURCE_ASYNC_VIDEO;
4140 4141
		bool hasAudio = (flags & OBS_SOURCE_AUDIO) ==
			OBS_SOURCE_AUDIO;
J
John Bradley 已提交
4142 4143
		QAction *action;

4144
		popup.addMenu(AddBackgroundColorMenu(sceneItem));
J
jp9000 已提交
4145 4146
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
4147
		popup.addAction(QTStr("Remove"), this,
4148
				SLOT(on_actionRemoveSource_triggered()));
J
jp9000 已提交
4149 4150
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
4151
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
4152 4153 4154 4155 4156

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

C
cg2121 已提交
4157 4158 4159 4160 4161 4162
		QAction *sourceWindow = popup.addAction(
				QTStr("SourceWindow"),
				this, SLOT(OpenSourceWindow()));

		popup.addAction(sourceWindow);

J
jp9000 已提交
4163
		popup.addSeparator();
4164 4165 4166 4167 4168 4169 4170 4171 4172

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

J
jp9000 已提交
4173 4174 4175 4176
		if (isAsyncVideo) {
			popup.addMenu(AddDeinterlacingMenu(source));
			popup.addSeparator();
		}
4177 4178 4179 4180

		popup.addMenu(AddScaleFilteringMenu(sceneItem));
		popup.addSeparator();

J
jp9000 已提交
4181
		popup.addMenu(sourceProjector);
C
cg2121 已提交
4182
		popup.addAction(sourceWindow);
J
jp9000 已提交
4183
		popup.addSeparator();
J
John Bradley 已提交
4184 4185 4186 4187 4188 4189 4190

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

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

J
jp9000 已提交
4191 4192
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
4193 4194
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
4195 4196

		ui->actionCopyFilters->setEnabled(true);
4197
		ui->actionCopySource->setEnabled(true);
4198 4199
	} else {
		ui->actionPasteFilters->setEnabled(false);
J
jp9000 已提交
4200 4201 4202
	}

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

J
jp9000 已提交
4205 4206
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
J
jp9000 已提交
4207 4208 4209 4210
	if (ui->scenes->count()) {
		QModelIndex idx = ui->sources->indexAt(pos);
		CreateSourcePopupMenu(idx.row(), false);
	}
P
Palana 已提交
4211 4212
}

4213 4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230
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 已提交
4231
void OBSBasic::AddSource(const char *id)
4232
{
4233 4234 4235
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
4236
		if (sourceSelect.newSource && strcmp(id, "group") != 0)
4237
			CreatePropertiesWindow(sourceSelect.newSource);
4238
	}
4239 4240
}

4241
QMenu *OBSBasic::CreateAddSourcePopupMenu()
4242
{
4243
	const char *type;
J
jp9000 已提交
4244
	bool foundValues = false;
4245
	bool foundDeprecated = false;
J
jp9000 已提交
4246
	size_t idx = 0;
4247

4248
	QMenu *popup = new QMenu(QTStr("Add"), this);
4249
	QMenu *deprecated = new QMenu(QTStr("Deprecated"), popup);
4250

J
jp9000 已提交
4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267
	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 已提交
4268
		popupItem->setData(QT_UTF8(type));
4269 4270
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4271 4272 4273

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

J
jp9000 已提交
4276 4277
	while (obs_enum_input_types(idx++, &type)) {
		const char *name = obs_source_get_display_name(type);
4278
		uint32_t caps = obs_get_source_output_flags(type);
J
jp9000 已提交
4279

4280 4281 4282
		if ((caps & OBS_SOURCE_CAP_DISABLED) != 0)
			continue;

4283 4284
		if ((caps & OBS_SOURCE_DEPRECATED) == 0) {
			addSource(popup, type, name);
4285 4286 4287
		} else {
			addSource(deprecated, type, name);
			foundDeprecated = true;
4288
		}
4289
		foundValues = true;
4290 4291
	}

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

J
jp9000 已提交
4294 4295
	popup->addSeparator();
	QAction *addGroup = new QAction(QTStr("Group"), this);
4296
	addGroup->setData(QT_UTF8("group"));
J
jp9000 已提交
4297
	connect(addGroup, SIGNAL(triggered(bool)),
4298
			this, SLOT(AddSourceFromAction()));
J
jp9000 已提交
4299 4300
	popup->addAction(addGroup);

4301 4302 4303 4304 4305
	if (!foundDeprecated) {
		delete deprecated;
		deprecated = nullptr;
	}

4306 4307 4308
	if (!foundValues) {
		delete popup;
		popup = nullptr;
4309 4310

	} else if (foundDeprecated) {
J
jp9000 已提交
4311
		popup->addSeparator();
4312
		popup->addMenu(deprecated);
4313
	}
4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330

	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).
4331
		OBSMessageBox::information(this,
4332 4333 4334 4335 4336
				QTStr("Basic.Main.AddSourceHelp.Title"),
				QTStr("Basic.Main.AddSourceHelp.Text"));
		return;
	}

4337
	QScopedPointer<QMenu> popup(CreateAddSourcePopupMenu());
4338 4339
	if (popup)
		popup->exec(pos);
4340 4341
}

J
jp9000 已提交
4342
void OBSBasic::on_actionAddSource_triggered()
4343
{
J
jp9000 已提交
4344
	AddSourcePopupMenu(QCursor::pos());
4345 4346
}

J
jp9000 已提交
4347 4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358 4359
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 已提交
4360
void OBSBasic::on_actionRemoveSource_triggered()
4361
{
4362
	vector<OBSSceneItem> items;
4363

J
jp9000 已提交
4364
	obs_scene_enum_items(GetCurrentScene(), remove_items, &items);
4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389 4390 4391 4392 4393 4394 4395 4396 4397

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

J
John Bradley 已提交
4400 4401 4402 4403 4404 4405 4406 4407 4408
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
4409
void OBSBasic::on_actionSourceProperties_triggered()
4410
{
4411
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4412
	OBSSource source = obs_sceneitem_get_source(item);
4413

4414 4415
	if (source)
		CreatePropertiesWindow(source);
4416 4417
}

J
jp9000 已提交
4418
void OBSBasic::on_actionSourceUp_triggered()
4419
{
J
jp9000 已提交
4420
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4421
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
4422
}
J
jp9000 已提交
4423

J
jp9000 已提交
4424
void OBSBasic::on_actionSourceDown_triggered()
4425
{
J
jp9000 已提交
4426
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4427
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
4428 4429
}

J
jp9000 已提交
4430 4431 4432
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4433
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
4434 4435 4436 4437 4438
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4439
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
4440 4441 4442 4443 4444
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4445
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
4446 4447 4448 4449 4450
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
4451
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
4452 4453
}

4454
static BPtr<char> ReadLogFile(const char *subdir, const char *log)
J
jp9000 已提交
4455
{
4456
	char logDir[512];
4457
	if (GetConfigPath(logDir, sizeof(logDir), subdir) <= 0)
4458
		return nullptr;
J
jp9000 已提交
4459 4460 4461 4462 4463

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

4464
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
4465 4466 4467 4468 4469 4470
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

4471
void OBSBasic::UploadLog(const char *subdir, const char *file)
J
jp9000 已提交
4472
{
4473
	BPtr<char> fileString{ReadLogFile(subdir, file)};
J
jp9000 已提交
4474

4475
	if (!fileString)
J
jp9000 已提交
4476 4477
		return;

4478
	if (!*fileString)
J
jp9000 已提交
4479 4480 4481 4482
		return;

	ui->menuLogFiles->setEnabled(false);

4483 4484
	stringstream ss;
	ss << "OBS " << App()->GetVersionString()
4485 4486
	   << " log file uploaded at " << CurrentDateTimeString()
	   << "\n\n" << fileString;
4487

F
fryshorts 已提交
4488

4489 4490 4491
	if (logUploadThread) {
		logUploadThread->wait();
	}
F
fryshorts 已提交
4492

4493
	RemoteTextThread *thread = new RemoteTextThread(
4494
			"https://obsproject.com/logs/upload",
4495 4496
			"text/plain", ss.str().c_str());

4497
	logUploadThread.reset(thread);
4498 4499 4500
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::logUploadFinished);
	logUploadThread->start();
J
jp9000 已提交
4501 4502
}

P
Palana 已提交
4503 4504
void OBSBasic::on_actionShowLogs_triggered()
{
4505
	char logDir[512];
4506
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4507 4508
		return;

P
Palana 已提交
4509 4510 4511 4512
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
4513 4514
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
4515
	UploadLog("obs-studio/logs", App()->GetCurrentLog());
J
jp9000 已提交
4516 4517 4518 4519
}

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

4523 4524 4525
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
4526
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
4527 4528 4529 4530 4531 4532 4533 4534 4535 4536 4537 4538
		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 已提交
4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553
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 已提交
4554 4555
void OBSBasic::on_actionCheckForUpdates_triggered()
{
J
jp9000 已提交
4556
	CheckForUpdates(true);
J
jp9000 已提交
4557 4558
}

4559
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
4560 4561 4562
{
	ui->menuLogFiles->setEnabled(true);

4563
	if (text.isEmpty()) {
4564
		OBSMessageBox::information(this,
J
jp9000 已提交
4565
				QTStr("LogReturnDialog.ErrorUploadingLog"),
4566
				error);
J
jp9000 已提交
4567 4568 4569
		return;
	}

4570
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
4571
	string resURL = obs_data_get_string(returnData, "url");
4572
	QString logURL = resURL.c_str();
J
jp9000 已提交
4573 4574 4575 4576 4577 4578
	obs_data_release(returnData);

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

J
jp9000 已提交
4579
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
4580
		obs_source_t *source, const string &name)
4581
{
4582 4583 4584 4585
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

4586
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
4587
	QListWidgetItem *listItem    = listWidget->currentItem();
4588

4589
	if (foundSource || name.empty()) {
4590
		listItem->setText(QT_UTF8(prevName));
4591

4592
		if (foundSource) {
4593
			OBSMessageBox::information(parent,
4594 4595 4596
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
4597
			OBSMessageBox::information(parent,
4598 4599 4600 4601
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

4602 4603 4604
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
4605
		obs_source_set_name(source, name.c_str());
4606 4607 4608
	}
}

J
jp9000 已提交
4609 4610 4611 4612 4613
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
4614
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
4615 4616 4617 4618

	if (!scene)
		return;

4619
	obs_source_t *source = obs_scene_get_source(scene);
4620
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
4621

J
jp9000 已提交
4622 4623 4624
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_SCENE_LIST_CHANGED);

J
jp9000 已提交
4625 4626 4627
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
4628 4629 4630 4631 4632 4633 4634 4635
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4636 4637 4638 4639 4640 4641 4642 4643
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
4644 4645 4646 4647
#define RECORDING_START \
	"==== Recording Start ==============================================="
#define RECORDING_STOP \
	"==== Recording Stop ================================================"
J
jp9000 已提交
4648 4649 4650 4651
#define REPLAY_BUFFER_START \
	"==== Replay Buffer Start ==========================================="
#define REPLAY_BUFFER_STOP \
	"==== Replay Buffer Stop ============================================"
J
jp9000 已提交
4652 4653 4654 4655 4656
#define STREAMING_START \
	"==== Streaming Start ==============================================="
#define STREAMING_STOP \
	"==== Streaming Stop ================================================"

4657 4658
void OBSBasic::StartStreaming()
{
J
jp9000 已提交
4659 4660
	if (outputHandler->StreamingActive())
		return;
4661
	if (disableOutputsRef)
4662
		return;
J
jp9000 已提交
4663 4664 4665 4666

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

4667 4668
	SaveProject();

J
jp9000 已提交
4669 4670
	ui->streamButton->setEnabled(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
4671 4672 4673 4674 4675

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

J
jp9000 已提交
4677 4678 4679
	if (!outputHandler->StartStreaming(service)) {
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
4680
		ui->streamButton->setChecked(false);
4681 4682 4683 4684 4685

		if (sysTrayStream) {
			sysTrayStream->setText(ui->streamButton->text());
			sysTrayStream->setEnabled(true);
		}
4686 4687 4688 4689 4690

		QMessageBox::critical(this,
				QTStr("Output.StartStreamFailed"),
				QTStr("Output.StartFailedGeneric"));
		return;
4691
	}
4692 4693 4694 4695 4696

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	if (recordWhenStreaming)
		StartRecording();
4697 4698 4699 4700 4701

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	if (replayBufferWhileStreaming)
		StartReplayBuffer();
4702 4703
}

4704 4705 4706 4707 4708 4709 4710 4711 4712 4713 4714 4715 4716 4717 4718 4719 4720 4721 4722 4723 4724
#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

4725
inline void OBSBasic::OnActivate()
4726
{
4727 4728
	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
J
jp9000 已提交
4729
		ui->autoConfigure->setEnabled(false);
4730 4731
		App()->IncrementSleepInhibition();
		UpdateProcessPriority();
C
cg2121 已提交
4732

4733 4734
		if (trayIcon)
			trayIcon->setIcon(QIcon(":/res/images/tray_active.png"));
4735 4736
	}
}
J
jp9000 已提交
4737

4738 4739
inline void OBSBasic::OnDeactivate()
{
4740
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
4741
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
4742
		ui->autoConfigure->setEnabled(true);
4743
		App()->DecrementSleepInhibition();
4744
		ClearProcessPriority();
C
cg2121 已提交
4745

4746 4747
		if (trayIcon)
			trayIcon->setIcon(QIcon(":/res/images/obs.png"));
J
jp9000 已提交
4748
	}
4749 4750 4751 4752 4753 4754 4755
}

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

	if (outputHandler->StreamingActive())
4756
		outputHandler->StopStreaming(streamingStopping);
4757 4758

	OnDeactivate();
4759 4760 4761 4762 4763 4764 4765

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
4766 4767 4768 4769 4770 4771 4772

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

J
jp9000 已提交
4775 4776 4777 4778 4779
void OBSBasic::ForceStopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
4780
		outputHandler->StopStreaming(true);
J
jp9000 已提交
4781

4782
	OnDeactivate();
4783 4784 4785 4786 4787 4788 4789

	bool recordWhenStreaming = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "RecordWhenStreaming");
	bool keepRecordingWhenStreamStops = config_get_bool(GetGlobalConfig(),
			"BasicWindow", "KeepRecordingWhenStreamStops");
	if (recordWhenStreaming && !keepRecordingWhenStreamStops)
		StopRecording();
4790 4791 4792 4793 4794 4795 4796

	bool replayBufferWhileStreaming = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "ReplayBufferWhileStreaming");
	bool keepReplayBufferStreamStops = config_get_bool(GetGlobalConfig(),
		"BasicWindow", "KeepReplayBufferStreamStops");
	if (replayBufferWhileStreaming && !keepReplayBufferStreamStops)
		StopReplayBuffer();
J
jp9000 已提交
4797 4798 4799 4800 4801 4802
}

void OBSBasic::StreamDelayStarting(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
	ui->streamButton->setEnabled(true);
4803
	ui->streamButton->setChecked(true);
4804 4805 4806 4807 4808

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
4809 4810 4811 4812 4813 4814 4815 4816 4817 4818 4819 4820

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

4822
	OnActivate();
J
jp9000 已提交
4823 4824 4825 4826 4827 4828
}

void OBSBasic::StreamDelayStopping(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
	ui->streamButton->setEnabled(true);
4829
	ui->streamButton->setChecked(false);
4830 4831 4832 4833 4834

	if (sysTrayStream) {
		sysTrayStream->setText(ui->streamButton->text());
		sysTrayStream->setEnabled(true);
	}
J
jp9000 已提交
4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846 4847 4848

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

4849
void OBSBasic::StreamingStart()
4850
{
4851
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
4852
	ui->streamButton->setEnabled(true);
4853
	ui->streamButton->setChecked(true);
J
jp9000 已提交
4854
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
4855 4856 4857 4858 4859

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

J
jp9000 已提交
4861 4862 4863
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTED);

4864
	OnActivate();
4865

4866
	blog(LOG_INFO, STREAMING_START);
4867 4868
}

4869 4870 4871
void OBSBasic::StreamStopping()
{
	ui->streamButton->setText(QTStr("Basic.Main.StoppingStreaming"));
4872 4873 4874

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

4876
	streamingStopping = true;
J
jp9000 已提交
4877 4878
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPING);
4879 4880
}

4881
void OBSBasic::StreamingStop(int code, QString last_error)
4882
{
4883 4884 4885
	const char *errorDescription;
	DStr errorMessage;
	bool use_last_error = false;
4886 4887 4888

	switch (code) {
	case OBS_OUTPUT_BAD_PATH:
4889
		errorDescription = Str("Output.ConnectFail.BadPath");
4890 4891 4892
		break;

	case OBS_OUTPUT_CONNECT_FAILED:
4893 4894
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.ConnectFailed");
4895 4896 4897
		break;

	case OBS_OUTPUT_INVALID_STREAM:
4898
		errorDescription = Str("Output.ConnectFail.InvalidStream");
4899 4900
		break;

4901
	default:
4902
	case OBS_OUTPUT_ERROR:
4903 4904
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Error");
4905 4906 4907 4908 4909
		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 */
4910 4911
		use_last_error = true;
		errorDescription = Str("Output.ConnectFail.Disconnected");
4912 4913
	}

4914 4915 4916 4917 4918 4919
	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 已提交
4920
	ui->statusbar->StreamStopped();
4921

4922
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
4923
	ui->streamButton->setEnabled(true);
4924
	ui->streamButton->setChecked(false);
4925 4926 4927 4928 4929

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

4931
	streamingStopping = false;
J
jp9000 已提交
4932 4933 4934
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STOPPED);

4935
	OnDeactivate();
4936 4937

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

C
cg2121 已提交
4939
	if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
4940
		OBSMessageBox::information(this,
4941 4942
				QTStr("Output.ConnectFail.Title"),
				QT_UTF8(errorMessage));
C
cg2121 已提交
4943
	} else if (code != OBS_OUTPUT_SUCCESS && !isVisible()) {
4944
		SysTrayNotify(QT_UTF8(errorDescription), QSystemTrayIcon::Warning);
C
cg2121 已提交
4945
	}
J
jp9000 已提交
4946 4947 4948 4949 4950 4951

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

4954 4955
void OBSBasic::StartRecording()
{
4956 4957
	if (outputHandler->RecordingActive())
		return;
4958
	if (disableOutputsRef)
4959
		return;
4960

J
jp9000 已提交
4961 4962 4963
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTING);

4964
	SaveProject();
4965 4966 4967

	if (!outputHandler->StartRecording())
		ui->recordButton->setChecked(false);
4968 4969
}

4970 4971
void OBSBasic::RecordStopping()
{
J
jp9000 已提交
4972
	ui->recordButton->setText(QTStr("Basic.Main.StoppingRecording"));
4973 4974 4975

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

4977
	recordingStopping = true;
J
jp9000 已提交
4978 4979
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPING);
4980 4981
}

4982 4983 4984 4985 4986
void OBSBasic::StopRecording()
{
	SaveProject();

	if (outputHandler->RecordingActive())
4987
		outputHandler->StopRecording(recordingStopping);
J
jp9000 已提交
4988

4989
	OnDeactivate();
4990 4991
}

P
Palana 已提交
4992 4993
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
4994
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
J
jp9000 已提交
4995
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
4996
	ui->recordButton->setChecked(true);
4997 4998 4999

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

5001
	recordingStopping = false;
J
jp9000 已提交
5002 5003 5004
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STARTED);

5005
	OnActivate();
5006

5007
	blog(LOG_INFO, RECORDING_START);
P
Palana 已提交
5008 5009
}

5010
void OBSBasic::RecordingStop(int code)
5011
{
P
Palana 已提交
5012
	ui->statusbar->RecordingStopped();
J
jp9000 已提交
5013
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
5014
	ui->recordButton->setChecked(false);
5015 5016 5017

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

5019
	blog(LOG_INFO, RECORDING_STOP);
5020

C
cg2121 已提交
5021
	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5022
		OBSMessageBox::information(this,
5023 5024
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));
J
jp9000 已提交
5025

C
cg2121 已提交
5026
	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5027
		OBSMessageBox::information(this,
J
jp9000 已提交
5028 5029 5030
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

C
cg2121 已提交
5031
	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5032
		OBSMessageBox::information(this,
J
jp9000 已提交
5033 5034
				QTStr("Output.RecordError.Title"),
				QTStr("Output.RecordError.Msg"));
C
cg2121 已提交
5035 5036 5037 5038 5039 5040 5041 5042 5043 5044 5045 5046

	} 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 已提交
5047 5048
	}

J
jp9000 已提交
5049 5050 5051
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_RECORDING_STOPPED);

5052
	OnDeactivate();
5053
}
5054

J
jp9000 已提交
5055 5056 5057 5058 5059 5060 5061 5062 5063
#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;
5064
	if (disableOutputsRef)
5065
		return;
J
jp9000 已提交
5066

5067 5068 5069 5070 5071
	if (!NoSourcesConfirmation()) {
		replayBufferButton->setChecked(false);
		return;
	}

J
jp9000 已提交
5072 5073 5074 5075 5076 5077 5078 5079 5080
	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) {
5081
		OBSMessageBox::information(this,
J
jp9000 已提交
5082 5083 5084 5085 5086 5087 5088 5089 5090
				RP_NO_HOTKEY_TITLE,
				RP_NO_HOTKEY_TEXT);
		return;
	}

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

	SaveProject();
5091 5092
	if (!outputHandler->StartReplayBuffer())
		replayBufferButton->setChecked(false);
J
jp9000 已提交
5093 5094 5095 5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137 5138 5139 5140 5141
}

void OBSBasic::ReplayBufferStopping()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StoppingReplayBuffer"));

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

	replayBufferStopping = true;
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STOPPING);
}

void OBSBasic::StopReplayBuffer()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	SaveProject();

	if (outputHandler->ReplayBufferActive())
		outputHandler->StopReplayBuffer(replayBufferStopping);

	OnDeactivate();
}

void OBSBasic::ReplayBufferStart()
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StopReplayBuffer"));

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

	replayBufferStopping = false;
	if (api)
		api->on_event(OBS_FRONTEND_EVENT_REPLAY_BUFFER_STARTED);

	OnActivate();

	blog(LOG_INFO, REPLAY_BUFFER_START);
}

5142 5143 5144 5145 5146 5147 5148 5149 5150 5151 5152 5153 5154 5155
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 已提交
5156 5157 5158 5159 5160 5161 5162 5163 5164 5165 5166 5167 5168
void OBSBasic::ReplayBufferStop(int code)
{
	if (!outputHandler || !outputHandler->replayBuffer)
		return;

	replayBufferButton->setText(QTStr("Basic.Main.StartReplayBuffer"));

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

	blog(LOG_INFO, REPLAY_BUFFER_STOP);

	if (code == OBS_OUTPUT_UNSUPPORTED && isVisible()) {
5169
		OBSMessageBox::information(this,
J
jp9000 已提交
5170 5171 5172 5173
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));

	} else if (code == OBS_OUTPUT_NO_SPACE && isVisible()) {
5174
		OBSMessageBox::information(this,
J
jp9000 已提交
5175 5176 5177 5178
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

	} else if (code != OBS_OUTPUT_SUCCESS && isVisible()) {
5179
		OBSMessageBox::information(this,
J
jp9000 已提交
5180 5181 5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198 5199 5200 5201
				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();
}

5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216 5217 5218 5219 5220 5221 5222 5223
bool OBSBasic::NoSourcesConfirmation()
{
	if (CountVideoSources() == 0 && isVisible()) {
		QString msg;
		msg = QTStr("NoSources.Text");
		msg += "\n\n";
		msg += QTStr("NoSources.Text.AddSource");

		QMessageBox messageBox(QMessageBox::Question,
				QTStr("NoSources.title"),
				msg,
				QMessageBox::Yes | QMessageBox::No,
				this);
		messageBox.setDefaultButton(QMessageBox::No);

		if (QMessageBox::No == messageBox.exec())
			return false;
	}

	return true;
}

5224 5225
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
5226
	if (outputHandler->StreamingActive()) {
5227 5228 5229
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStoppingStream");

C
cg2121 已提交
5230
		if (confirm && isVisible()) {
5231
			QMessageBox::StandardButton button =
5232
				OBSMessageBox::question(this,
5233 5234 5235
						QTStr("ConfirmStop.Title"),
						QTStr("ConfirmStop.Text"));

C
cg2121 已提交
5236 5237
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(true);
5238
				return;
C
cg2121 已提交
5239
			}
5240 5241
		}

5242
		StopStreaming();
5243
	} else {
5244 5245 5246 5247 5248
		if (!NoSourcesConfirmation()) {
			ui->streamButton->setChecked(false);
			return;
		}

5249 5250 5251
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStartingStream");

C
cg2121 已提交
5252
		if (confirm && isVisible()) {
5253
			QMessageBox::StandardButton button =
5254
				OBSMessageBox::question(this,
5255 5256 5257
						QTStr("ConfirmStart.Title"),
						QTStr("ConfirmStart.Text"));

C
cg2121 已提交
5258 5259
			if (button == QMessageBox::No) {
				ui->streamButton->setChecked(false);
5260
				return;
C
cg2121 已提交
5261
			}
5262 5263
		}

5264
		StartStreaming();
5265 5266 5267 5268 5269
	}
}

void OBSBasic::on_recordButton_clicked()
{
5270
	if (outputHandler->RecordingActive()) {
5271
		StopRecording();
5272 5273 5274 5275 5276 5277
	} else {
		if (!NoSourcesConfirmation()) {
			ui->recordButton->setChecked(false);
			return;
		}

5278
		StartRecording();
5279
	}
J
jp9000 已提交
5280 5281
}

J
jp9000 已提交
5282
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
5283
{
J
jp9000 已提交
5284
	on_action_Settings_triggered();
J
jp9000 已提交
5285
}
5286

5287 5288 5289 5290 5291 5292
void OBSBasic::on_actionHelpPortal_triggered()
{
	QUrl url = QUrl("https://obsproject.com/help", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5293 5294 5295 5296 5297 5298
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

5299 5300
void OBSBasic::on_actionDiscord_triggered()
{
5301
	QUrl url = QUrl("https://obsproject.com/discord", QUrl::TolerantMode);
5302 5303 5304
	QDesktopServices::openUrl(url);
}

5305 5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323 5324
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 已提交
5325
int OBSBasic::GetTopSelectedSourceItem()
5326
{
J
jp9000 已提交
5327 5328 5329
	QModelIndexList selectedItems =
		ui->sources->selectionModel()->selectedIndexes();
	return selectedItems.count() ? selectedItems[0].row() : -1;
5330 5331
}

J
jp9000 已提交
5332 5333
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
5334
	CreateSourcePopupMenu(GetTopSelectedSourceItem(), true);
J
jp9000 已提交
5335 5336 5337 5338

	UNUSED_PARAMETER(pos);
}

R
Ryan Foster 已提交
5339 5340 5341 5342 5343 5344 5345 5346
void OBSBasic::on_program_customContextMenuRequested(const QPoint&)
{
	QMenu popup(this);
	QPointer<QMenu> studioProgramProjector;

	studioProgramProjector = new QMenu(
			QTStr("StudioProgramProjector"));
	AddProjectorMenuMonitors(studioProgramProjector, this,
5347
			SLOT(OpenStudioProgramProjector()));
R
Ryan Foster 已提交
5348 5349 5350 5351 5352

	popup.addMenu(studioProgramProjector);

	QAction *studioProgramWindow = popup.addAction(
			QTStr("StudioProgramWindow"),
5353
			this, SLOT(OpenStudioProgramWindow()));
R
Ryan Foster 已提交
5354 5355 5356 5357 5358 5359

	popup.addAction(studioProgramWindow);

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

J
jp9000 已提交
5360 5361 5362 5363
void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
J
jp9000 已提交
5364
	QPointer<QMenu> previewProjector;
J
jp9000 已提交
5365 5366 5367 5368 5369

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

J
jp9000 已提交
5372 5373 5374 5375
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));

C
cg2121 已提交
5376 5377 5378 5379
	QAction *previewWindow = popup.addAction(
			QTStr("PreviewWindow"),
			this, SLOT(OpenPreviewWindow()));

J
jp9000 已提交
5380
	popup.addMenu(previewProjector);
C
cg2121 已提交
5381
	popup.addAction(previewWindow);
J
jp9000 已提交
5382 5383 5384 5385 5386
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

5387 5388
void OBSBasic::on_actionAlwaysOnTop_triggered()
{
5389
#ifndef _WIN32
5390 5391 5392 5393
	/* 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 */
5394 5395 5396
	CloseDialogs();
#endif

5397 5398 5399 5400 5401 5402 5403 5404 5405 5406 5407 5408 5409 5410
	QMetaObject::invokeMethod(this, "ToggleAlwaysOnTop",
			Qt::QueuedConnection);
}

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

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

	show();
}

5411 5412 5413 5414 5415 5416 5417 5418 5419 5420
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 已提交
5421 5422 5423
	} else if (strcmp(val, "24 NTSC") == 0) {
		num = 24000;
		den = 1001;
5424 5425 5426 5427 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 5464 5465 5466 5467 5468 5469 5470 5471 5472 5473 5474 5475 5476
	} else if (strcmp(val, "25") == 0) {
		num = 25;
		den = 1;
	} else if (strcmp(val, "29.97") == 0) {
		num = 30000;
		den = 1001;
	} else if (strcmp(val, "48") == 0) {
		num = 48;
		den = 1;
	} else if (strcmp(val, "59.94") == 0) {
		num = 60000;
		den = 1001;
	} else if (strcmp(val, "60") == 0) {
		num = 60;
		den = 1;
	} else {
		num = 30;
		den = 1;
	}
}

void OBSBasic::GetFPSInteger(uint32_t &num, uint32_t &den) const
{
	num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSInt");
	den = 1;
}

void OBSBasic::GetFPSFraction(uint32_t &num, uint32_t &den) const
{
	num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNum");
	den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSDen");
}

void OBSBasic::GetFPSNanoseconds(uint32_t &num, uint32_t &den) const
{
	num = 1000000000;
	den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNS");
}

void OBSBasic::GetConfigFPS(uint32_t &num, uint32_t &den) const
{
	uint32_t type = config_get_uint(basicConfig, "Video", "FPSType");

	if (type == 1) //"Integer"
		GetFPSInteger(num, den);
	else if (type == 2) //"Fraction"
		GetFPSFraction(num, den);
	else if (false) //"Nanoseconds", currently not implemented
		GetFPSNanoseconds(num, den);
	else
		GetFPSCommon(num, den);
}

5477
config_t *OBSBasic::Config() const
5478 5479 5480
{
	return basicConfig;
}
J
jp9000 已提交
5481 5482 5483

void OBSBasic::on_actionEditTransform_triggered()
{
5484 5485 5486
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
5487 5488
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
5489
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
5490 5491
}

5492 5493 5494 5495 5496 5497 5498 5499 5500 5501 5502 5503 5504 5505 5506 5507 5508 5509 5510 5511 5512 5513 5514 5515 5516 5517 5518 5519 5520 5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532 5533 5534 5535
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);
}

5536
static bool reset_tr(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
5537
{
5538 5539 5540 5541
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, reset_tr, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
J
jp9000 已提交
5542

5543
	obs_sceneitem_defer_update_begin(item);
J
jp9000 已提交
5544

5545 5546 5547 5548 5549 5550 5551 5552 5553
	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 已提交
5554

5555 5556
	obs_sceneitem_crop crop = {};
	obs_sceneitem_set_crop(item, &crop);
J
jp9000 已提交
5557

5558
	obs_sceneitem_defer_update_end(item);
J
jp9000 已提交
5559

5560 5561 5562 5563
	UNUSED_PARAMETER(scene);
	UNUSED_PARAMETER(param);
	return true;
}
J
jp9000 已提交
5564

5565 5566 5567
void OBSBasic::on_actionResetTransform_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), reset_tr, nullptr);
J
jp9000 已提交
5568 5569
}

5570
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
5571 5572 5573 5574 5575
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

5578
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
5579 5580 5581 5582
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
5583 5584
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
5585 5586
	};

5587 5588 5589 5590 5591 5592
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

5593
static vec3 GetItemTL(obs_sceneitem_t *item)
5594 5595 5596
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
5597 5598 5599
	return tl;
}

5600
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
5601 5602 5603 5604
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
5605
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
5606 5607 5608
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
5609
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
5610 5611
}

5612
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5613 5614
		void *param)
{
5615 5616 5617
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, RotateSelectedSources,
				param);
J
jp9000 已提交
5618 5619 5620 5621 5622 5623 5624
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
5625
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
5626 5627
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
5628
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
5629

5630 5631
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
5632 5633 5634 5635 5636 5637 5638 5639 5640 5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655
	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);
}

5656
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5657 5658 5659 5660
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

5661 5662 5663
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, MultiplySelectedItemScale,
				param);
J
jp9000 已提交
5664 5665 5666 5667 5668 5669
	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
5670
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
5671
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
5672
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
5673

5674 5675
	obs_sceneitem_force_update_transform(item);

J
jp9000 已提交
5676
	SetItemTL(item, tl);
J
jp9000 已提交
5677 5678

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
5679 5680 5681 5682 5683
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
5684 5685
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
5686 5687 5688 5689 5690 5691
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
5692 5693
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
5694 5695 5696 5697
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

5698
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
5699 5700 5701 5702
		void *param)
{
	obs_bounds_type boundsType = *reinterpret_cast<obs_bounds_type*>(param);

5703 5704 5705
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, CenterAlignSelectedItems,
				param);
J
jp9000 已提交
5706 5707 5708 5709 5710 5711
	if (!obs_sceneitem_selected(item))
		return true;

	obs_video_info ovi;
	obs_get_video_info(&ovi);

5712
	obs_transform_info itemInfo;
J
jp9000 已提交
5713 5714 5715 5716 5717 5718 5719 5720 5721 5722 5723 5724 5725 5726 5727 5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742
	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);
}

5743
static bool center_to_scene(obs_scene_t *, obs_sceneitem_t *item, void *)
J
jp9000 已提交
5744
{
5745 5746
	vec3 tl, br, itemCenter, screenCenter, offset;
	obs_video_info ovi;
5747

5748 5749 5750 5751
	if (obs_sceneitem_is_group(item))
		obs_sceneitem_group_enum_items(item, center_to_scene, nullptr);
	if (!obs_sceneitem_selected(item))
		return true;
5752

5753
	obs_get_video_info(&ovi);
5754

5755 5756 5757
	vec3_set(&screenCenter, float(ovi.base_width),
			float(ovi.base_height), 0.0f);
	vec3_mulf(&screenCenter, &screenCenter, 0.5f);
5758

5759
	GetItemBox(item, tl, br);
5760

5761 5762 5763
	vec3_sub(&itemCenter, &br, &tl);
	vec3_mulf(&itemCenter, &itemCenter, 0.5f);
	vec3_add(&itemCenter, &itemCenter, &tl);
5764

5765 5766
	vec3_sub(&offset, &screenCenter, &itemCenter);
	vec3_add(&tl, &tl, &offset);
5767

5768 5769 5770
	SetItemTL(item, tl);
	return true;
};
5771

5772 5773 5774
void OBSBasic::on_actionCenterToScreen_triggered()
{
	obs_scene_enum_items(GetCurrentScene(), center_to_scene, nullptr);
J
jp9000 已提交
5775
}
J
jp9000 已提交
5776

5777 5778 5779 5780 5781 5782 5783
void OBSBasic::EnablePreviewDisplay(bool enable)
{
	obs_display_set_enabled(ui->preview->GetDisplay(), enable);
	ui->preview->setVisible(enable);
	ui->previewDisabledLabel->setVisible(!enable);
}

J
jp9000 已提交
5784 5785
void OBSBasic::TogglePreview()
{
5786 5787
	previewEnabled = !previewEnabled;
	EnablePreviewDisplay(previewEnabled);
J
jp9000 已提交
5788
}
5789

J
jp9000 已提交
5790
static bool nudge_callback(obs_scene_t*, obs_sceneitem_t *item, void *param)
5791
{
J
jp9000 已提交
5792 5793
	if (obs_sceneitem_locked(item))
		return true;
5794

J
jp9000 已提交
5795 5796
	struct vec2 &offset = *reinterpret_cast<struct vec2*>(param);
	struct vec2 pos;
5797

J
jp9000 已提交
5798
	if (!obs_sceneitem_selected(item)) {
J
jp9000 已提交
5799 5800 5801 5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812 5813 5814
		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 已提交
5815 5816
		return true;
	}
5817

J
jp9000 已提交
5818 5819 5820 5821 5822
	obs_sceneitem_get_pos(item, &pos);
	vec2_add(&pos, &pos, &offset);
	obs_sceneitem_set_pos(item, &pos);
	return true;
}
J
jp9000 已提交
5823

J
jp9000 已提交
5824 5825 5826 5827
void OBSBasic::Nudge(int dist, MoveDir dir)
{
	if (ui->preview->Locked())
		return;
5828

J
jp9000 已提交
5829 5830
	struct vec2 offset;
	vec2_set(&offset, 0.0f, 0.0f);
5831

J
jp9000 已提交
5832 5833 5834 5835 5836 5837
	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;
	}
5838

J
jp9000 已提交
5839
	obs_scene_enum_items(GetCurrentScene(), nudge_callback, &offset);
5840 5841 5842 5843 5844 5845
}

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

S
Shaolin 已提交
5847
OBSProjector *OBSBasic::OpenProjector(obs_source_t *source, int monitor,
S
Shaolin 已提交
5848
		QString title, ProjectorType type)
J
jp9000 已提交
5849 5850
{
	/* seriously?  10 monitors? */
C
cg2121 已提交
5851
	if (monitor > 9 || monitor > QGuiApplication::screens().size() - 1)
S
Shaolin 已提交
5852
		return nullptr;
C
cg2121 已提交
5853

S
Shaolin 已提交
5854 5855
	OBSProjector *projector = new OBSProjector(nullptr, source, monitor,
			title, type);
C
cg2121 已提交
5856

S
Shaolin 已提交
5857
	if (monitor < 0) {
C
cg2121 已提交
5858 5859 5860 5861 5862 5863 5864 5865 5866
		for (auto &projPtr : windowProjectors) {
			if (!projPtr) {
				projPtr = projector;
				projector = nullptr;
			}
		}

		if (projector)
			windowProjectors.push_back(projector);
S
Shaolin 已提交
5867
	} else {
C
cg2121 已提交
5868 5869
		delete projectors[monitor];
		projectors[monitor].clear();
C
cg2121 已提交
5870

C
cg2121 已提交
5871 5872
		projectors[monitor] = projector;
	}
S
Shaolin 已提交
5873 5874

	return projector;
J
jp9000 已提交
5875 5876
}

R
Ryan Foster 已提交
5877 5878 5879
void OBSBasic::OpenStudioProgramProjector()
{
	int monitor = sender()->property("monitor").toInt();
5880
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::StudioProgram);
R
Ryan Foster 已提交
5881 5882
}

J
jp9000 已提交
5883 5884 5885
void OBSBasic::OpenPreviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
5886
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Preview);
J
jp9000 已提交
5887 5888 5889 5890 5891 5892 5893 5894 5895
}

void OBSBasic::OpenSourceProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

S
Shaolin 已提交
5896 5897
	OpenProjector(obs_sceneitem_get_source(item), monitor, nullptr,
			ProjectorType::Source);
J
jp9000 已提交
5898 5899
}

S
Shaolin 已提交
5900 5901 5902
void OBSBasic::OpenMultiviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
5903
	OpenProjector(nullptr, monitor, nullptr, ProjectorType::Multiview);
S
Shaolin 已提交
5904 5905
}

J
jp9000 已提交
5906 5907 5908 5909 5910 5911 5912
void OBSBasic::OpenSceneProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

5913 5914
	OpenProjector(obs_scene_get_source(scene), monitor, nullptr,
			ProjectorType::Scene);
C
cg2121 已提交
5915 5916
}

R
Ryan Foster 已提交
5917 5918
void OBSBasic::OpenStudioProgramWindow()
{
S
Shaolin 已提交
5919
	OpenProjector(nullptr, -1, QTStr("StudioProgramWindow"),
S
Shaolin 已提交
5920
			ProjectorType::StudioProgram);
R
Ryan Foster 已提交
5921 5922
}

C
cg2121 已提交
5923 5924
void OBSBasic::OpenPreviewWindow()
{
S
Shaolin 已提交
5925 5926
	OpenProjector(nullptr, -1, QTStr("PreviewWindow"),
			ProjectorType::Preview);
C
cg2121 已提交
5927 5928 5929 5930 5931 5932 5933 5934 5935
}

void OBSBasic::OpenSourceWindow()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

	OBSSource source = obs_sceneitem_get_source(item);
5936
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
5937 5938 5939

	OpenProjector(obs_sceneitem_get_source(item), -1, title,
			ProjectorType::Source);
C
cg2121 已提交
5940 5941
}

S
Shaolin 已提交
5942 5943
void OBSBasic::OpenMultiviewWindow()
{
S
Shaolin 已提交
5944
	OpenProjector(nullptr, -1, QTStr("MultiviewWindowed"),
S
Shaolin 已提交
5945 5946 5947
			ProjectorType::Multiview);
}

C
cg2121 已提交
5948 5949 5950 5951 5952 5953 5954
void OBSBasic::OpenSceneWindow()
{
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

	OBSSource source = obs_scene_get_source(scene);
5955
	QString title = QString::fromUtf8(obs_source_get_name(source));
S
Shaolin 已提交
5956 5957 5958

	OpenProjector(obs_scene_get_source(scene), -1, title,
			ProjectorType::Scene);
J
jp9000 已提交
5959
}
5960

C
cg2121 已提交
5961 5962
void OBSBasic::OpenSavedProjectors()
{
5963
	for (SavedProjectorInfo *info : savedProjectorsArray) {
S
Shaolin 已提交
5964
		OBSProjector *projector = nullptr;
5965
		switch (info->type) {
S
Shaolin 已提交
5966
		case ProjectorType::Source:
5967 5968 5969 5970 5971
		case ProjectorType::Scene: {
			OBSSource source = obs_get_source_by_name(
					info->name.c_str());
			if (!source)
				continue;
C
cg2121 已提交
5972

S
Shaolin 已提交
5973 5974 5975 5976
			QString title = nullptr;
			if (info->monitor < 0)
				title = QString::fromUtf8(
						obs_source_get_name(source));
R
Ryan Foster 已提交
5977

S
Shaolin 已提交
5978 5979
			projector = OpenProjector(source, info->monitor, title,
					info->type);
5980 5981 5982

			obs_source_release(source);
			break;
S
Shaolin 已提交
5983
		}
5984
		case ProjectorType::Preview: {
S
Shaolin 已提交
5985
			projector = OpenProjector(nullptr, info->monitor,
5986
					QTStr("PreviewWindow"),
5987
					ProjectorType::Preview);
5988
			break;
S
Shaolin 已提交
5989
		}
5990
		case ProjectorType::StudioProgram: {
S
Shaolin 已提交
5991
			projector = OpenProjector(nullptr, info->monitor,
5992 5993 5994 5995 5996
					QTStr("StudioProgramWindow"),
					ProjectorType::StudioProgram);
			break;
		}
		case ProjectorType::Multiview: {
S
Shaolin 已提交
5997
			projector = OpenProjector(nullptr, info->monitor,
5998
					QTStr("MultiviewWindowed"),
5999
					ProjectorType::Multiview);
6000 6001
			break;
		}
C
cg2121 已提交
6002
		}
S
Shaolin 已提交
6003

6004
		if (projector && !info->geometry.empty()) {
S
Shaolin 已提交
6005 6006 6007
			QByteArray byteArray = QByteArray::fromBase64(
					QByteArray(info->geometry.c_str()));
			projector->restoreGeometry(byteArray);
S
Shaolin 已提交
6008

S
Shaolin 已提交
6009 6010 6011 6012 6013
			if (!WindowPositionValid(projector->normalGeometry())) {
				QRect rect = App()->desktop()->geometry();
				projector->setGeometry(QStyle::alignedRect(
						Qt::LeftToRight,
						Qt::AlignCenter, size(), rect));
C
cg2121 已提交
6014 6015 6016 6017 6018
			}
		}
	}
}

6019 6020 6021 6022 6023 6024 6025 6026 6027 6028
void OBSBasic::on_actionFullscreenInterface_triggered()
{
	if (!fullscreenInterface)
		showFullScreen();
	else
		showNormal();

	fullscreenInterface = !fullscreenInterface;
}

6029 6030 6031 6032
void OBSBasic::UpdateTitleBar()
{
	stringstream name;

J
jp9000 已提交
6033 6034
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "Profile");
J
jp9000 已提交
6035 6036 6037
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

6038 6039 6040 6041 6042
	name << "OBS ";
	if (previewProgramMode)
		name << "Studio ";

	name << App()->GetVersionString();
6043 6044 6045
	if (App()->IsPortableMode())
		name << " - Portable Mode";

J
jp9000 已提交
6046
	name << " - " << Str("TitleBar.Profile") << ": " << profile;
J
jp9000 已提交
6047
	name << " - " << Str("TitleBar.Scenes") << ": " << sceneCollection;
6048 6049 6050

	setWindowTitle(QT_UTF8(name.str().c_str()));
}
J
jp9000 已提交
6051 6052 6053 6054 6055 6056 6057 6058 6059 6060 6061 6062 6063 6064 6065 6066 6067 6068 6069 6070 6071 6072 6073 6074

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

J
jp9000 已提交
6076
void OBSBasic::on_resetUI_triggered()
6077
{
J
jp9000 已提交
6078
	restoreState(startingDockLayout);
6079

6080
#if QT_VERSION >= QT_VERSION_CHECK(5, 6, 0)
J
jp9000 已提交
6081 6082 6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114
	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,
		ui->controlsDock
	};

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

	resizeDocks(docks, {cy, cy, cy, cy, cy}, Qt::Vertical);
	resizeDocks(docks, sizes, Qt::Horizontal);
6115
#endif
J
jp9000 已提交
6116 6117 6118 6119 6120 6121 6122 6123 6124 6125 6126 6127 6128
}

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);
6129 6130 6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142 6143 6144 6145 6146
}

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 已提交
6147 6148 6149 6150 6151 6152

void OBSBasic::on_actionLockPreview_triggered()
{
	ui->preview->ToggleLocked();
	ui->actionLockPreview->setChecked(ui->preview->Locked());
}
C
cg2121 已提交
6153

J
Joseph El-Khouri 已提交
6154 6155 6156 6157 6158 6159 6160 6161 6162 6163 6164 6165 6166 6167 6168 6169
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);
6170 6171
	action->setVisible(!(ovi.output_width == ovi.base_width &&
			ovi.output_height == ovi.base_height));
J
Joseph El-Khouri 已提交
6172 6173 6174 6175 6176 6177

	UpdatePreviewScalingMenu();
}

void OBSBasic::on_actionScaleWindow_triggered()
{
6178
	ui->preview->SetFixedScaling(false);
J
Joseph El-Khouri 已提交
6179 6180 6181 6182 6183 6184
	ui->preview->ResetScrollingOffset();
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleCanvas_triggered()
{
6185 6186
	ui->preview->SetFixedScaling(true);
	ui->preview->SetScalingLevel(0);
J
Joseph El-Khouri 已提交
6187 6188 6189 6190 6191
	emit ui->preview->DisplayResized();
}

void OBSBasic::on_actionScaleOutput_triggered()
{
6192 6193 6194 6195 6196 6197 6198 6199 6200 6201
	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 已提交
6202 6203 6204
	emit ui->preview->DisplayResized();
}

C
cg2121 已提交
6205 6206 6207 6208 6209 6210 6211
void OBSBasic::SetShowing(bool showing)
{
	if (!showing && isVisible()) {
		config_set_string(App()->GlobalConfig(),
			"BasicWindow", "geometry",
			saveGeometry().toBase64().constData());

6212 6213 6214 6215 6216 6217 6218 6219 6220
		/* hide all visible child dialogs */
		visDlgPositions.clear();
		if (!visDialogs.isEmpty()) {
			for (QDialog *dlg : visDialogs) {
				visDlgPositions.append(dlg->pos());
				dlg->hide();
			}
		}

6221 6222
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Show"));
C
cg2121 已提交
6223 6224 6225 6226 6227 6228 6229
		QTimer::singleShot(250, this, SLOT(hide()));

		if (previewEnabled)
			EnablePreviewDisplay(false);

		setVisible(false);

6230 6231 6232 6233
#ifdef __APPLE__
		EnableOSXDockIcon(false);
#endif

C
cg2121 已提交
6234
	} else if (showing && !isVisible()) {
6235 6236
		if (showHide)
			showHide->setText(QTStr("Basic.SystemTray.Hide"));
C
cg2121 已提交
6237 6238 6239 6240 6241 6242
		QTimer::singleShot(250, this, SLOT(show()));

		if (previewEnabled)
			EnablePreviewDisplay(true);

		setVisible(true);
6243

6244 6245 6246 6247 6248 6249 6250 6251
#ifdef __APPLE__
		EnableOSXDockIcon(true);
#endif

		/* raise and activate window to ensure it is on top */
		raise();
		activateWindow();

6252 6253 6254 6255 6256 6257 6258 6259 6260
		/* 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();
			}
		}

6261 6262 6263 6264 6265 6266 6267 6268
		/* 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 已提交
6269 6270 6271
	}
}

6272 6273 6274 6275 6276 6277 6278 6279 6280 6281 6282 6283
void OBSBasic::ToggleShowHide()
{
	bool showing = isVisible();
	if (showing) {
		/* check for modal dialogs */
		EnumDialogs();
		if (!modalDialogs.isEmpty() || !visMsgBoxes.isEmpty())
			return;
	}
	SetShowing(!showing);
}

6284 6285
void OBSBasic::SystemTrayInit()
{
6286 6287
	trayIcon.reset(new QSystemTrayIcon(QIcon(":/res/images/obs.png"),
			this));
C
cg2121 已提交
6288 6289 6290
	trayIcon->setToolTip("OBS Studio");

	showHide = new QAction(QTStr("Basic.SystemTray.Show"),
6291
			trayIcon.data());
C
cg2121 已提交
6292
	sysTrayStream = new QAction(QTStr("Basic.Main.StartStreaming"),
6293
			trayIcon.data());
C
cg2121 已提交
6294
	sysTrayRecord = new QAction(QTStr("Basic.Main.StartRecording"),
6295
			trayIcon.data());
J
jp9000 已提交
6296
	sysTrayReplayBuffer = new QAction(QTStr("Basic.Main.StartReplayBuffer"),
6297
			trayIcon.data());
C
cg2121 已提交
6298
	exit = new QAction(QTStr("Exit"),
6299
			trayIcon.data());
C
cg2121 已提交
6300

J
jp9000 已提交
6301 6302
	if (outputHandler && !outputHandler->replayBuffer)
		sysTrayReplayBuffer->setEnabled(false);
6303

6304 6305
	connect(trayIcon.data(),
			SIGNAL(activated(QSystemTrayIcon::ActivationReason)),
C
cg2121 已提交
6306 6307 6308 6309 6310 6311 6312 6313
			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()));
6314
	connect(sysTrayReplayBuffer.data(), &QAction::triggered,
J
jp9000 已提交
6315
			this, &OBSBasic::ReplayBufferClicked);
C
cg2121 已提交
6316 6317 6318 6319 6320 6321 6322 6323
	connect(exit, SIGNAL(triggered()),
			this, SLOT(close()));

	trayMenu = new QMenu;
}

void OBSBasic::IconActivated(QSystemTrayIcon::ActivationReason reason)
{
6324
	if (reason == QSystemTrayIcon::Trigger) {
C
cg2121 已提交
6325
		ToggleShowHide();
6326 6327 6328 6329 6330 6331 6332 6333 6334 6335 6336 6337 6338 6339 6340 6341 6342 6343 6344
	} else if (reason == QSystemTrayIcon::Context) {
		QMenu *previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
				SLOT(OpenPreviewProjector()));
		QMenu *studioProgramProjector = new QMenu(
				QTStr("StudioProgramProjector"));
		AddProjectorMenuMonitors(studioProgramProjector, this,
				SLOT(OpenStudioProgramProjector()));

		trayMenu->clear();
		trayMenu->addAction(showHide);
		trayMenu->addMenu(previewProjector);
		trayMenu->addMenu(studioProgramProjector);
		trayMenu->addAction(sysTrayStream);
		trayMenu->addAction(sysTrayRecord);
		trayMenu->addAction(sysTrayReplayBuffer);
		trayMenu->addAction(exit);
		trayMenu->popup(QCursor::pos());
	}
C
cg2121 已提交
6345 6346 6347 6348 6349 6350 6351 6352 6353 6354 6355 6356 6357 6358 6359 6360 6361 6362 6363 6364 6365 6366 6367 6368 6369 6370 6371
}

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 已提交
6372 6373
	} else if ((sysTrayWhenStarted && sysTrayEnabled)
			|| opt_minimize_tray) {
C
cg2121 已提交
6374 6375 6376 6377 6378
		trayIcon->show();
		if (firstStarted) {
			QTimer::singleShot(50, this, SLOT(hide()));
			EnablePreviewDisplay(false);
			setVisible(false);
6379 6380 6381
#ifdef __APPLE__
			EnableOSXDockIcon(false);
#endif
C
cg2121 已提交
6382
			opt_minimize_tray = false;
C
cg2121 已提交
6383 6384 6385 6386 6387 6388 6389 6390 6391 6392 6393 6394 6395 6396
		}
	} 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"));
}
6397 6398 6399 6400 6401 6402

bool OBSBasic::sysTrayMinimizeToTray()
{
	return config_get_bool(GetGlobalConfig(),
			"BasicWindow", "SysTrayMinimizeToTray");
}
6403 6404 6405 6406 6407 6408

void OBSBasic::on_actionCopySource_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;
J
jp9000 已提交
6409 6410

	on_actionCopyTransform_triggered();
6411 6412 6413 6414 6415 6416 6417

	OBSSource source = obs_sceneitem_get_source(item);

	copyString = obs_source_get_name(source);
	copyVisible = obs_sceneitem_visible(item);

	ui->actionPasteRef->setEnabled(true);
6418 6419 6420 6421 6422 6423

	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);
6424 6425 6426 6427
}

void OBSBasic::on_actionPasteRef_triggered()
{
J
jp9000 已提交
6428 6429 6430 6431 6432
	/* do not allow duplicate refs of the same group in the same scene */
	OBSScene scene = GetCurrentScene();
	if (!!obs_scene_get_group(scene, copyString))
		return;

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
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, false);
	on_actionPasteTransform_triggered();
}

void OBSBasic::on_actionPasteDup_triggered()
{
	OBSBasicSourceSelect::SourcePaste(copyString, copyVisible, true);
	on_actionPasteTransform_triggered();
}

void OBSBasic::on_actionCopyFilters_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();

	if (!item)
		return;

	OBSSource source = obs_sceneitem_get_source(item);

	copyFiltersString = obs_source_get_name(source);

	ui->actionPasteFilters->setEnabled(true);
}

void OBSBasic::on_actionPasteFilters_triggered()
{
	OBSSource source = obs_get_source_by_name(copyFiltersString);
6460
	obs_source_release(source);
6461

6462
	OBSSceneItem sceneItem = GetCurrentSceneItem();
6463 6464 6465 6466 6467 6468 6469
	OBSSource dstSource = obs_sceneitem_get_source(sceneItem);

	if (source == dstSource)
		return;

	obs_source_copy_filters(dstSource, source);
}
J
jp9000 已提交
6470

6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501 6502 6503 6504 6505 6506 6507 6508 6509 6510 6511 6512 6513 6514 6515 6516 6517 6518 6519 6520 6521 6522 6523 6524 6525 6526 6527 6528 6529 6530 6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543 6544 6545 6546 6547 6548 6549 6550 6551 6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565 6566 6567 6568 6569 6570 6571 6572 6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584 6585 6586 6587 6588 6589 6590 6591 6592 6593 6594 6595 6596 6597 6598 6599 6600 6601 6602 6603 6604 6605 6606 6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634 6635 6636 6637 6638 6639 6640 6641 6642 6643 6644
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 已提交
6645 6646 6647 6648 6649 6650 6651
void OBSBasic::on_autoConfigure_triggered()
{
	AutoConfig test(this);
	test.setModal(true);
	test.show();
	test.exec();
}
J
jp9000 已提交
6652 6653 6654

void OBSBasic::on_stats_triggered()
{
6655 6656 6657 6658 6659 6660
	if (!stats.isNull()) {
		stats->show();
		stats->raise();
		return;
	}

J
jp9000 已提交
6661 6662 6663 6664 6665
	OBSBasicStats *statsDlg;
	statsDlg = new OBSBasicStats(nullptr);
	statsDlg->show();
	stats = statsDlg;
}
6666 6667 6668 6669 6670 6671 6672

ColorSelect::ColorSelect(QWidget *parent)
	: QWidget(parent),
	  ui(new Ui::ColorSelect)
{
	ui->setupUi(this);
}