window-basic-main.cpp 95.4 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>
J
jp9000 已提交
22
#include <QMessageBox>
23
#include <QShowEvent>
24
#include <QDesktopServices>
J
jp9000 已提交
25
#include <QFileDialog>
26

J
jp9000 已提交
27
#include <util/dstr.h>
28 29
#include <util/util.hpp>
#include <util/platform.h>
P
Palana 已提交
30
#include <util/profiler.hpp>
J
jp9000 已提交
31
#include <graphics/math-defs.h>
32

33
#include "obs-app.hpp"
34
#include "platform.hpp"
35
#include "visibility-item-widget.hpp"
36
#include "item-widget-helpers.hpp"
37
#include "window-basic-settings.hpp"
38
#include "window-namedialog.hpp"
J
jp9000 已提交
39
#include "window-basic-source-select.hpp"
J
jp9000 已提交
40
#include "window-basic-main.hpp"
J
jp9000 已提交
41
#include "window-basic-main-outputs.hpp"
42
#include "window-basic-properties.hpp"
J
jp9000 已提交
43
#include "window-log-reply.hpp"
J
jp9000 已提交
44
#include "window-projector.hpp"
P
Palana 已提交
45
#include "window-remux.hpp"
J
jp9000 已提交
46
#include "qt-wrappers.hpp"
47
#include "display-helpers.hpp"
48
#include "volume-control.hpp"
49
#include "remote-text.hpp"
50

J
jp9000 已提交
51
#include "ui_OBSBasic.h"
52

J
jp9000 已提交
53
#include <fstream>
54 55
#include <sstream>

56 57 58
#include <QScreen>
#include <QWindow>

59 60
#define PREVIEW_EDGE_SIZE 10

61
using namespace std;
J
jp9000 已提交
62

J
jp9000 已提交
63 64
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
65
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
66
Q_DECLARE_METATYPE(obs_order_movement);
67
Q_DECLARE_METATYPE(std::vector<std::shared_ptr<OBSSignal>>);
J
jp9000 已提交
68

P
Palana 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81
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));
}

82 83
static void AddExtraModulePaths()
{
84
	char base_module_dir[512];
85
	int ret = GetConfigPath(base_module_dir, sizeof(base_module_dir),
86
			"obs-studio/plugins/%module%");
B
BtbN 已提交
87

88
	if (ret <= 0)
89 90 91
		return;

	string path = (char*)base_module_dir;
92
#if defined(__APPLE__)
93
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
94 95 96 97 98 99 100
#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
101 102
}

103 104
static QList<QKeySequence> DeleteKeys;

105
OBSBasic::OBSBasic(QWidget *parent)
106
	: OBSMainWindow  (parent),
J
jp9000 已提交
107
	  ui             (new Ui::OBSBasic)
108 109
{
	ui->setupUi(this);
J
jp9000 已提交
110 111
	ui->previewDisabledLabel->setVisible(false);

S
Socapex 已提交
112
	copyActionsDynamicProperties();
113

114 115
	ui->sources->setItemDelegate(new VisibilityItemDelegate(ui->sources));

J
jp9000 已提交
116
	int width = config_get_int(App()->GlobalConfig(), "BasicWindow", "cx");
117 118 119

	// Check if no values are saved (new installation).
	if (width != 0) {
J
jp9000 已提交
120 121 122
		int height = config_get_int(App()->GlobalConfig(),
				"BasicWindow", "cy");
		int posx = config_get_int(App()->GlobalConfig(), "BasicWindow",
123
				"posx");
J
jp9000 已提交
124
		int posy = config_get_int(App()->GlobalConfig(), "BasicWindow",
125 126
				"posy");

127
		setGeometry(posx, posy, width, height);
128 129
	}

130
	char styleSheetPath[512];
J
jp9000 已提交
131 132
	int ret = GetProfilePath(styleSheetPath, sizeof(styleSheetPath),
			"stylesheet.qss");
133
	if (ret > 0) {
H
HomeWorld 已提交
134 135 136 137 138
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
139 140
	}

P
Palana 已提交
141 142 143
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
144
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
145

146 147 148
	qRegisterMetaTypeStreamOperators<
		std::vector<std::shared_ptr<OBSSignal>>>(
				"std::vector<std::shared_ptr<OBSSignal>>");
149
	qRegisterMetaTypeStreamOperators<OBSScene>("OBSScene");
150
	qRegisterMetaTypeStreamOperators<OBSSceneItem>("OBSSceneItem");
151

P
Palana 已提交
152 153 154
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

155 156 157 158 159 160
	connect(windowHandle(), &QWindow::screenChanged, [this]() {
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
	});
J
jp9000 已提交
161

P
Palana 已提交
162 163
	installEventFilter(CreateShortcutFilter());

164 165 166 167 168
	stringstream name;
	name << "OBS " << App()->GetVersionString();	
	blog(LOG_INFO, "%s", name.str().c_str());
	blog(LOG_INFO, "---------------------------------");

169
	UpdateTitleBar();
J
jp9000 已提交
170 171 172 173 174 175 176 177 178 179 180 181 182 183

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

	connect(ui->sources->itemDelegate(),
			SIGNAL(closeEditor(QWidget*,
					QAbstractItemDelegate::EndEditHint)),
			this,
			SLOT(SceneItemNameEdited(QWidget*,
					QAbstractItemDelegate::EndEditHint)));
184 185

	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
186 187 188 189
	cpuUsageTimer = new QTimer(this);
	connect(cpuUsageTimer, SIGNAL(timeout()),
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
190

191 192 193 194 195 196
	DeleteKeys =
#ifdef __APPLE__
		QList<QKeySequence>{{Qt::Key_Backspace}} <<
#endif
		QKeySequence::keyBindings(QKeySequence::Delete);

197
#ifdef __APPLE__
198 199
	ui->actionRemoveSource->setShortcuts(DeleteKeys);
	ui->actionRemoveScene->setShortcuts(DeleteKeys);
200 201 202

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
203
#endif
204 205 206 207 208 209 210 211 212 213 214 215 216 217

	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()));
218 219
}

220
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent)
221
{
222
	obs_source_t *source = obs_get_output_source(channel);
223 224 225
	if (!source)
		return;

226
	obs_data_t *data = obs_save_source(source);
227

J
jp9000 已提交
228
	obs_data_set_obj(parent, name, data);
229 230 231 232 233

	obs_data_release(data);
	obs_source_release(source);
}

J
jp9000 已提交
234
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder)
235
{
236 237 238
	obs_data_t       *saveData     = obs_data_create();
	obs_data_array_t *sourcesArray = obs_save_sources();
	obs_source_t     *currentScene = obs_get_output_source(0);
239
	const char       *sceneName   = obs_source_get_name(currentScene);
240

J
jp9000 已提交
241 242 243
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

244 245 246 247 248 249
	SaveAudioDevice(DESKTOP_AUDIO_1, 1, saveData);
	SaveAudioDevice(DESKTOP_AUDIO_2, 2, saveData);
	SaveAudioDevice(AUX_AUDIO_1,     3, saveData);
	SaveAudioDevice(AUX_AUDIO_2,     4, saveData);
	SaveAudioDevice(AUX_AUDIO_3,     5, saveData);

J
jp9000 已提交
250
	obs_data_set_string(saveData, "current_scene", sceneName);
J
jp9000 已提交
251
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
252
	obs_data_set_string(saveData, "name", sceneCollection);
J
jp9000 已提交
253
	obs_data_set_array(saveData, "sources", sourcesArray);
254 255 256 257 258 259
	obs_data_array_release(sourcesArray);
	obs_source_release(currentScene);

	return saveData;
}

S
Socapex 已提交
260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279
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));
		}
	}
}

280 281 282 283 284 285 286 287 288 289 290 291
void OBSBasic::ClearVolumeControls()
{
	VolControl *control;

	for (size_t i = 0; i < volumes.size(); i++) {
		control = volumes[i];
		delete control;
	}

	volumes.clear();
}

J
jp9000 已提交
292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
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;
}

307 308
void OBSBasic::Save(const char *file)
{
J
jp9000 已提交
309 310
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
	obs_data_t *saveData  = GenerateSaveData(sceneOrder);
311 312 313

	if (!obs_data_save_json_safe(saveData, file, "tmp", "bak"))
		blog(LOG_ERROR, "Could not save scene data to %s", file);
314 315

	obs_data_release(saveData);
J
jp9000 已提交
316
	obs_data_array_release(sceneOrder);
317 318
}

319
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
320
{
321
	obs_data_t *data = obs_data_get_obj(parent, name);
322 323 324
	if (!data)
		return;

325
	obs_source_t *source = obs_load_source(data);
326 327 328 329 330 331 332 333
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
	obs_properties_t *props = obs_get_source_properties(
			OBS_SOURCE_TYPE_INPUT, output_id);
	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;
}

353 354
void OBSBasic::CreateDefaultScene()
{
J
jp9000 已提交
355 356
	disableSaving++;

357 358 359
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

360 361
	ClearSceneData();

362 363
	obs_scene_t  *scene  = obs_scene_create(Str("Basic.Scene"));
	obs_source_t *source = obs_scene_get_source(scene);
364 365 366 367 368

	obs_add_source(source);

#ifdef __APPLE__
	source = obs_source_create(OBS_SOURCE_TYPE_INPUT, "display_capture",
369
			Str("Basic.DisplayCapture"), NULL, nullptr);
370 371 372 373 374 375 376 377

	if (source) {
		obs_scene_add(scene, source);
		obs_add_source(source);
		obs_source_release(source);
	}
#endif

378
	obs_set_output_source(0, obs_scene_get_source(scene));
379
	obs_scene_release(scene);
380 381 382 383 384 385 386

	if (hasDesktopAudio)
		ResetAudioDevice(App()->OutputAudioSource(), "default",
				Str("Basic.DesktopDevice1"), 1);
	if (hasInputAudio)
		ResetAudioDevice(App()->InputAudioSource(), "default",
				Str("Basic.AuxDevice1"), 3);
J
jp9000 已提交
387 388

	disableSaving--;
389 390
}

J
jp9000 已提交
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419
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);
	}
}

420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443
void OBSBasic::CleanupUnusedSources()
{
	auto removeUnusedSources = [&](obs_source_t *source)
	{
		obs_scene_t *scene = obs_scene_from_source(source);
		if (scene)
			return;

		if (sourceSceneRefs[source] == 0) {
			sourceSceneRefs.erase(source);
			obs_source_remove(source);
		}
	};
	using func_type = decltype(removeUnusedSources);

	obs_enum_sources(
			[](void *f, obs_source_t *source)
			{
				(*static_cast<func_type*>(f))(source);
				return true;
			},
			static_cast<void*>(&removeUnusedSources));
}

444 445
void OBSBasic::Load(const char *file)
{
446
	if (!file || !os_file_exists(file)) {
447 448 449 450
		blog(LOG_ERROR, "Could not find file %s", file);
		return;
	}

451 452 453 454 455
	disableSaving++;

	obs_data_t *data = obs_data_create_from_json_file_safe(file, "bak");
	if (!data) {
		disableSaving--;
456
		CreateDefaultScene();
J
jp9000 已提交
457
		SaveProject();
458
		return;
459
	}
460

461 462
	ClearSceneData();

J
jp9000 已提交
463
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
464
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
J
jp9000 已提交
465 466
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
J
jp9000 已提交
467 468 469 470 471 472 473

	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");
474
	obs_source_t     *curScene;
475

J
jp9000 已提交
476 477 478
	if (!name || !*name)
		name = curSceneCollection;

479 480 481 482 483 484
	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);

485 486
	obs_load_sources(sources);

J
jp9000 已提交
487 488 489
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

490 491 492 493 494
	curScene = obs_get_source_by_name(sceneName);
	obs_set_output_source(0, curScene);
	obs_source_release(curScene);

	obs_data_array_release(sources);
J
jp9000 已提交
495
	obs_data_array_release(sceneOrder);
J
jp9000 已提交
496 497 498 499 500 501 502 503 504

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

505
	obs_data_release(data);
J
jp9000 已提交
506

507 508
	CleanupUnusedSources();

J
jp9000 已提交
509
	disableSaving--;
510 511
}

J
jp9000 已提交
512
#define SERVICE_PATH "service.json"
513 514 515 516 517 518

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

519
	char serviceJsonPath[512];
J
jp9000 已提交
520
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
521 522
			SERVICE_PATH);
	if (ret <= 0)
523 524
		return;

525 526
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
527

528
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
529
	obs_data_set_obj(data, "settings", settings);
530

531 532
	if (!obs_data_save_json_safe(data, serviceJsonPath, "tmp", "bak"))
		blog(LOG_WARNING, "Failed to save service");
533 534 535 536 537 538 539 540 541

	obs_data_release(settings);
	obs_data_release(data);
}

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

542
	char serviceJsonPath[512];
J
jp9000 已提交
543
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
544 545
			SERVICE_PATH);
	if (ret <= 0)
546 547
		return false;

548 549
	obs_data_t *data = obs_data_create_from_json_file_safe(serviceJsonPath,
			"bak");
550 551

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

554
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
555
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
556

557
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
558
			hotkey_data);
559
	obs_service_release(service);
560

P
Palana 已提交
561
	obs_data_release(hotkey_data);
562 563 564 565 566 567 568 569
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
P
Palana 已提交
570 571
	ProfileScope("OBSBasic::InitService");

572 573 574
	if (LoadService())
		return true;

575 576
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
577 578
	if (!service)
		return false;
579
	obs_service_release(service);
580 581 582 583

	return true;
}

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
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
};

600 601 602 603 604 605 606 607 608 609 610 611 612 613
bool OBSBasic::InitBasicConfigDefaults()
{
	vector<MonitorInfo> monitors;
	GetMonitors(monitors);

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

	uint32_t cx = monitors[0].cx;
	uint32_t cy = monitors[0].cy;

614 615 616 617 618 619 620 621 622 623
	/* ----------------------------------------------------- */
	/* 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");
624
		config_save_safe(basicConfig, "tmp", nullptr);
625 626 627 628
	}

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

629
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
630

631 632
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
633 634
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
635 636
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
637
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
638 639
	config_set_default_bool  (basicConfig, "SimpleOutput", "Reconnect",
			true);
640 641
	config_set_default_uint  (basicConfig, "SimpleOutput", "RetryDelay",
			10);
J
jp9000 已提交
642 643
	config_set_default_uint  (basicConfig, "SimpleOutput", "MaxRetries",
			20);
J
jp9000 已提交
644 645
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
646
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseCBR", true);
647 648 649
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseBufsize",
			false);
	config_set_default_int   (basicConfig, "SimpleOutput", "Bufsize", 2500);
J
jp9000 已提交
650 651
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
652

J
jp9000 已提交
653
	config_set_default_bool  (basicConfig, "AdvOut", "Reconnect", true);
654
	config_set_default_uint  (basicConfig, "AdvOut", "RetryDelay", 10);
J
jp9000 已提交
655
	config_set_default_uint  (basicConfig, "AdvOut", "MaxRetries", 20);
656 657
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
658 659 660 661 662 663 664 665
	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());
666
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
667 668
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
669
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
670 671 672
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

673 674 675 676 677
	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 已提交
678 679 680 681 682 683 684 685 686 687 688
	config_set_default_uint  (basicConfig, "AdvOut", "FFVBitrate", 2500);
	config_set_default_bool  (basicConfig, "AdvOut", "FFUseRescale",
			false);
	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);

689 690 691
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

692 693 694 695
	config_set_default_bool  (basicConfig, "Output", "DelayEnable", false);
	config_set_default_uint  (basicConfig, "Output", "DelaySec", 20);
	config_set_default_bool  (basicConfig, "Output", "DelayPreserve", true);

696 697 698 699 700 701 702 703 704 705 706 707 708 709
	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);
710 711 712 713 714 715

	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);
716
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
717 718 719 720
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
	config_set_default_string(basicConfig, "Video", "ColorSpace", "709");
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
721 722 723 724 725 726 727 728 729 730 731

	config_set_default_uint  (basicConfig, "Audio", "SampleRate", 44100);
	config_set_default_string(basicConfig, "Audio", "ChannelSetup",
			"Stereo");
	config_set_default_uint  (basicConfig, "Audio", "BufferingTime", 1000);

	return true;
}

bool OBSBasic::InitBasicConfig()
{
P
Palana 已提交
732 733
	ProfileScope("OBSBasic::InitBasicConfig");

734
	char configPath[512];
J
jp9000 已提交
735 736 737 738 739 740 741 742 743 744 745 746 747

	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");
748 749 750 751
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
752

753 754 755
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
756 757 758
		return false;
	}

J
jp9000 已提交
759 760 761 762 763
	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);
764
		basicConfig.SaveSafe("tmp");
J
jp9000 已提交
765 766
	}

767 768 769
	return InitBasicConfigDefaults();
}

770 771
void OBSBasic::InitOBSCallbacks()
{
P
Palana 已提交
772 773
	ProfileScope("OBSBasic::InitOBSCallbacks");

P
Palana 已提交
774 775
	signalHandlers.reserve(signalHandlers.size() + 6);
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_add",
776
			OBSBasic::SourceAdded, this);
P
Palana 已提交
777
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_remove",
778
			OBSBasic::SourceRemoved, this);
P
Palana 已提交
779
	signalHandlers.emplace_back(obs_get_signal_handler(), "channel_change",
780
			OBSBasic::ChannelChanged, this);
P
Palana 已提交
781
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_activate",
782
			OBSBasic::SourceActivated, this);
P
Palana 已提交
783
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_deactivate",
784
			OBSBasic::SourceDeactivated, this);
P
Palana 已提交
785
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
786
			OBSBasic::SourceRenamed, this);
787 788
}

J
jp9000 已提交
789 790
void OBSBasic::InitPrimitives()
{
P
Palana 已提交
791 792
	ProfileScope("OBSBasic::InitPrimitives");

J
jp9000 已提交
793
	obs_enter_graphics();
J
jp9000 已提交
794

795
	gs_render_start(true);
J
jp9000 已提交
796 797 798 799 800
	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);
801
	box = gs_render_save();
J
jp9000 已提交
802

803
	gs_render_start(true);
J
jp9000 已提交
804 805 806 807
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
808
	circle = gs_render_save();
J
jp9000 已提交
809

J
jp9000 已提交
810
	obs_leave_graphics();
J
jp9000 已提交
811 812
}

J
jp9000 已提交
813 814
void OBSBasic::ResetOutputs()
{
P
Palana 已提交
815 816
	ProfileScope("OBSBasic::ResetOutputs");

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

J
jp9000 已提交
820 821
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
822 823 824
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
J
jp9000 已提交
825 826 827 828 829
	} else {
		outputHandler->Update();
	}
}

830 831 832
#define MAIN_SEPARATOR \
	"====================================================================="

833 834
void OBSBasic::OBSInit()
{
P
Palana 已提交
835 836
	ProfileScope("OBSBasic::OBSInit");

J
jp9000 已提交
837 838
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
839
	char savePath[512];
J
jp9000 已提交
840 841 842 843 844 845 846 847
	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);
848
	if (ret <= 0)
J
jp9000 已提交
849 850 851 852 853
		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";
854

855 856
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
857
	if (!ResetAudio())
858 859
		throw "Failed to initialize audio";

860
	ret = ResetVideo();
861 862 863 864 865

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
J
jp9000 已提交
866 867 868
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
869 870 871 872 873 874 875
	case OBS_VIDEO_INVALID_PARAM:
		throw "Failed to initialize video:  Invalid parameters";
	default:
		if (ret != OBS_VIDEO_SUCCESS)
			throw "Failed to initialize video:  Unspecified error";
	}

876
	InitOBSCallbacks();
P
Palana 已提交
877
	InitHotkeys();
878

879
	AddExtraModulePaths();
J
jp9000 已提交
880
	obs_load_all_modules();
J
jp9000 已提交
881

882 883
	blog(LOG_INFO, MAIN_SEPARATOR);

J
jp9000 已提交
884
	ResetOutputs();
885
	CreateHotkeys();
J
jp9000 已提交
886

887 888 889
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
890 891
	InitPrimitives();

P
Palana 已提交
892 893 894 895 896 897
	{
		ProfileScope("OBSBasic::Load");
		disableSaving--;
		Load(savePath);
		disableSaving++;
	}
898

J
jp9000 已提交
899
	TimedCheckForUpdates();
900
	loaded = true;
J
jp9000 已提交
901

J
jp9000 已提交
902 903 904 905 906
	bool previewEnabled = config_get_bool(App()->GlobalConfig(),
			"BasicWindow", "PreviewEnabled");
	if (!previewEnabled)
		QMetaObject::invokeMethod(this, "TogglePreview",
				Qt::QueuedConnection);
907 908 909 910 911 912 913 914 915

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

J
jp9000 已提交
917
	RefreshSceneCollections();
J
jp9000 已提交
918
	RefreshProfiles();
J
jp9000 已提交
919
	disableSaving--;
920 921 922 923 924 925 926 927 928 929 930 931 932 933

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

	show();
934 935
}

P
Palana 已提交
936 937
void OBSBasic::InitHotkeys()
{
P
Palana 已提交
938 939
	ProfileScope("OBSBasic::InitHotkeys");

P
Palana 已提交
940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981
	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"),
982
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002

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

1003 1004
void OBSBasic::CreateHotkeys()
{
P
Palana 已提交
1005 1006
	ProfileScope("OBSBasic::CreateHotkeys");

1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
	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 已提交
1023 1024 1025 1026 1027 1028 1029 1030 1031
	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);
	};

1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068
	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);
	};

#define MAKE_CALLBACK(pred, method) \
	[](void *data, obs_hotkey_pair_id, obs_hotkey_t*, bool pressed) \
	{ \
		OBSBasic &basic = *static_cast<OBSBasic*>(data); \
		if (pred && pressed) { \
			method(); \
			return true; \
		} \
		return false; \
	}

	streamingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartStreaming",
			Str("Basic.Hotkeys.StartStreaming"),
			"OBSBasic.StopStreaming",
			Str("Basic.Hotkeys.StopStreaming"),
			MAKE_CALLBACK(!basic.outputHandler->StreamingActive(),
				basic.StartStreaming),
			MAKE_CALLBACK(basic.outputHandler->StreamingActive(),
				basic.StopStreaming),
			this, this);
	LoadHotkeyPair(streamingHotkeys,
			"OBSBasic.StartStreaming", "OBSBasic.StopStreaming");

J
jp9000 已提交
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
	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");

1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
	recordingHotkeys = obs_hotkey_pair_register_frontend(
			"OBSBasic.StartRecording",
			Str("Basic.Hotkeys.StartRecording"),
			"OBSBasic.StopRecording",
			Str("Basic.Hotkeys.StopRecording"),
			MAKE_CALLBACK(!basic.outputHandler->RecordingActive(),
				basic.StartRecording),
			MAKE_CALLBACK(basic.outputHandler->RecordingActive(),
				basic.StopRecording),
			this, this);
	LoadHotkeyPair(recordingHotkeys,
			"OBSBasic.StartRecording", "OBSBasic.StopRecording");

#undef MAKE_CALLBACK
}

J
jp9000 已提交
1100 1101 1102 1103
void OBSBasic::ClearHotkeys()
{
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
J
jp9000 已提交
1104
	obs_hotkey_unregister(forceStreamingStopHotkey);
J
jp9000 已提交
1105 1106
}

1107 1108
OBSBasic::~OBSBasic()
{
1109
	bool previewEnabled = obs_display_enabled(ui->preview->GetDisplay());
J
jp9000 已提交
1110

J
jp9000 已提交
1111 1112 1113 1114 1115 1116
	/* 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 已提交
1117
	delete cpuUsageTimer;
1118 1119
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
1120
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
J
jp9000 已提交
1121
	ClearHotkeys();
P
Palana 已提交
1122

1123
	service = nullptr;
J
jp9000 已提交
1124 1125
	outputHandler.reset();

J
John Bradley 已提交
1126 1127 1128
	if (interaction)
		delete interaction;

1129 1130 1131
	if (properties)
		delete properties;

J
jp9000 已提交
1132 1133 1134
	if (filters)
		delete filters;

1135 1136
	if (transformWindow)
		delete transformWindow;
1137

J
jp9000 已提交
1138 1139 1140
	if (advAudioWindow)
		delete advAudioWindow;

1141 1142 1143
	obs_display_remove_draw_callback(ui->preview->GetDisplay(),
			OBSBasic::RenderMain, this);

J
jp9000 已提交
1144
	obs_enter_graphics();
1145 1146
	gs_vertexbuffer_destroy(box);
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
1147
	obs_leave_graphics();
J
jp9000 已提交
1148

1149 1150 1151 1152 1153 1154 1155 1156 1157
	/* 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 已提交
1158 1159
	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
J
jp9000 已提交
1160 1161 1162

	QRect lastGeom = normalGeometry();

J
jp9000 已提交
1163
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cx",
J
jp9000 已提交
1164
			lastGeom.width());
J
jp9000 已提交
1165
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cy",
J
jp9000 已提交
1166
			lastGeom.height());
J
jp9000 已提交
1167
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posx",
J
jp9000 已提交
1168
			lastGeom.x());
J
jp9000 已提交
1169
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posy",
J
jp9000 已提交
1170
			lastGeom.y());
J
jp9000 已提交
1171 1172
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
1173
	config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184

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

J
jp9000 已提交
1187 1188 1189 1190 1191 1192 1193 1194 1195
void OBSBasic::SaveProjectNow()
{
	if (disableSaving)
		return;

	projectChanged = true;
	SaveProjectDeferred();
}

J
jp9000 已提交
1196 1197
void OBSBasic::SaveProject()
{
J
jp9000 已提交
1198 1199 1200
	if (disableSaving)
		return;

J
jp9000 已提交
1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215
	projectChanged = true;
	QMetaObject::invokeMethod(this, "SaveProjectDeferred",
			Qt::QueuedConnection);
}

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

	if (!projectChanged)
		return;

	projectChanged = false;

J
jp9000 已提交
1216 1217
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
1218
	char savePath[512];
J
jp9000 已提交
1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
	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);
1231 1232 1233
	if (ret <= 0)
		return;

J
jp9000 已提交
1234 1235 1236
	Save(savePath);
}

J
jp9000 已提交
1237
OBSScene OBSBasic::GetCurrentScene()
1238
{
J
jp9000 已提交
1239
	QListWidgetItem *item = ui->scenes->currentItem();
P
Palana 已提交
1240
	return item ? GetOBSRef<OBSScene>(item) : nullptr;
1241 1242
}

1243
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
1244
{
P
Palana 已提交
1245
	return item ? GetOBSRef<OBSSceneItem>(item) : nullptr;
J
jp9000 已提交
1246 1247
}

1248 1249 1250 1251 1252
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
	return GetSceneItem(ui->sources->currentItem());
}

1253 1254
void OBSBasic::UpdateSources(OBSScene scene)
{
1255
	ClearListItems(ui->sources);
1256 1257

	obs_scene_enum_items(scene,
1258
			[] (obs_scene_t *scene, obs_sceneitem_t *item, void *p)
1259 1260
			{
				OBSBasic *window = static_cast<OBSBasic*>(p);
1261
				window->InsertSceneItem(item);
J
jp9000 已提交
1262 1263

				UNUSED_PARAMETER(scene);
1264 1265 1266 1267
				return true;
			}, this);
}

1268
void OBSBasic::InsertSceneItem(obs_sceneitem_t *item)
1269
{
1270
	obs_source_t *source = obs_sceneitem_get_source(item);
1271

1272
	QListWidgetItem *listItem = new QListWidgetItem();
P
Palana 已提交
1273
	SetOBSRef(listItem, OBSSceneItem(item));
1274 1275

	ui->sources->insertItem(0, listItem);
1276
	ui->sources->setCurrentRow(0, QItemSelectionModel::ClearAndSelect);
1277

1278 1279
	SetupVisibilityItem(ui->sources, listItem, item);

1280
	/* if the source was just created, open properties dialog */
1281 1282 1283 1284
	if (sourceSceneRefs[source] == 0 && loaded)
		CreatePropertiesWindow(source);
}

1285
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
1286 1287 1288 1289 1290 1291 1292 1293 1294
{
	if (interaction)
		interaction->close();

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

1295
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
1296 1297 1298 1299 1300 1301 1302
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

1315 1316 1317
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
1318
{
1319
	const char *name  = obs_source_get_name(source);
1320
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
1321 1322

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

P
Palana 已提交
1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337
	obs_hotkey_register_source(source, "OBSBasic.SelectScene",
			Str("Basic.Hotkeys.SelectScene"),
			[](void *data,
				obs_hotkey_id, obs_hotkey_t*, bool pressed)
	{
		auto potential_source = static_cast<obs_source_t*>(data);
		auto source = obs_source_get_ref(potential_source);
		if (source && pressed)
			obs_set_output_source(0, source);
		obs_source_release(source);
	}, static_cast<obs_source_t*>(source));

1338
	signal_handler_t *handler = obs_source_get_signal_handler(source);
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354

	std::vector<std::shared_ptr<OBSSignal>> handlers{
		std::make_shared<OBSSignal>(handler, "item_add",
					OBSBasic::SceneItemAdded, this),
		std::make_shared<OBSSignal>(handler, "item_remove",
					OBSBasic::SceneItemRemoved, 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),
	};

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

1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372
	/* 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 已提交
1373
	SaveProject();
1374 1375
}

1376
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
1377
{
1378
	const char *name = obs_source_get_name(source);
J
jp9000 已提交
1379

J
jp9000 已提交
1380 1381 1382
	QListWidgetItem *sel = ui->scenes->currentItem();
	QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
			Qt::MatchExactly);
J
jp9000 已提交
1383

J
jp9000 已提交
1384 1385
	if (sel != nullptr) {
		if (items.contains(sel))
1386
			ClearListItems(ui->sources);
J
jp9000 已提交
1387
		delete sel;
J
jp9000 已提交
1388
	}
J
jp9000 已提交
1389

1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408
	auto DeleteSceneRefs = [&](obs_sceneitem_t *si)
	{
		obs_source_t *source = obs_sceneitem_get_source(si);
		sourceSceneRefs[source] -= 1;

		if (!sourceSceneRefs[source]) {
			obs_source_remove(source);
			sourceSceneRefs.erase(source);
		}
	};
	using DeleteSceneRefs_t = decltype(DeleteSceneRefs);

	obs_scene_enum_items(obs_scene_from_source(source),
			[](obs_scene_t *, obs_sceneitem_t *si, void *data)
	{
		(*static_cast<DeleteSceneRefs_t*>(data))(si);
		return true;
	}, static_cast<void*>(&DeleteSceneRefs));

J
jp9000 已提交
1409
	SaveProject();
1410 1411
}

1412
void OBSBasic::AddSceneItem(OBSSceneItem item)
1413
{
1414 1415
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
1416

1417 1418
	if (GetCurrentScene() == scene)
		InsertSceneItem(item);
J
jp9000 已提交
1419 1420

	sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
J
jp9000 已提交
1421
	SaveProject();
1422 1423
}

1424
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
1425
{
1426
	obs_scene_t *scene = obs_sceneitem_get_scene(item);
1427

J
jp9000 已提交
1428
	if (GetCurrentScene() == scene) {
B
BtbN 已提交
1429
		for (int i = 0; i < ui->sources->count(); i++) {
J
jp9000 已提交
1430
			QListWidgetItem *listItem = ui->sources->item(i);
J
jp9000 已提交
1431

P
Palana 已提交
1432
			if (GetOBSRef<OBSSceneItem>(listItem) == item) {
1433
				DeleteListItem(ui->sources, listItem);
J
jp9000 已提交
1434 1435
				break;
			}
1436 1437
		}
	}
J
jp9000 已提交
1438

1439
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
1440 1441

	int scenes = sourceSceneRefs[source] - 1;
1442 1443
	sourceSceneRefs[source] = scenes;

J
jp9000 已提交
1444 1445 1446 1447
	if (scenes == 0) {
		obs_source_remove(source);
		sourceSceneRefs.erase(source);
	}
J
jp9000 已提交
1448 1449

	SaveProject();
1450 1451
}

1452
void OBSBasic::UpdateSceneSelection(OBSSource source)
1453 1454
{
	if (source) {
1455
		obs_scene_t *scene = obs_scene_from_source(source);
1456
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
1457

1458 1459 1460
		if (!scene)
			return;

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

1464 1465 1466 1467 1468
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

J
jp9000 已提交
1469
			UpdateSources(scene);
1470
		}
J
jp9000 已提交
1471
	}
1472 1473
}

J
jp9000 已提交
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
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);
}

void OBSBasic::RenameSources(QString newName, QString prevName)
{
	RenameListValues(ui->scenes,  newName, prevName);
1487 1488 1489 1490 1491

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

	SaveProject();
J
jp9000 已提交
1494 1495
}

1496 1497
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
1498
	if (scene != GetCurrentScene() || ignoreSelectionUpdate)
1499 1500 1501 1502
		return;

	for (int i = 0; i < ui->sources->count(); i++) {
		QListWidgetItem *witem = ui->sources->item(i);
P
Palana 已提交
1503 1504
		QVariant data =
			witem->data(static_cast<int>(QtDataRole::OBSRef));
1505 1506 1507 1508 1509 1510
		if (!data.canConvert<OBSSceneItem>())
			continue;

		if (item != data.value<OBSSceneItem>())
			continue;

1511 1512 1513
		if (select)
			ui->sources->setCurrentItem(witem,
					QItemSelectionModel::ClearAndSelect);
1514

1515 1516 1517 1518
		break;
	}
}

1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561
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);
}

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

	QAction filtersAction(QTStr("Filters"), this);
	QAction propertiesAction(QTStr("Properties"), this);

	connect(&filtersAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceFilters,
			Qt::DirectConnection);
	connect(&propertiesAction, &QAction::triggered,
			this, &OBSBasic::GetAudioSourceProperties,
			Qt::DirectConnection);

	filtersAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));
	propertiesAction.setProperty("volControl",
			QVariant::fromValue<VolControl*>(vol));

	QMenu popup(this);
	popup.addAction(&filtersAction);
	popup.addAction(&propertiesAction);
	popup.exec(QCursor::pos());
}

1562 1563
void OBSBasic::ActivateAudioSource(OBSSource source)
{
1564 1565 1566 1567
	VolControl *vol = new VolControl(source, true);

	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583

	volumes.push_back(vol);
	ui->volumeWidgets->layout()->addWidget(vol);
}

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

1584
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
1585
{
1586
	const char *name  = obs_source_get_name(source);
1587 1588 1589

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

1591
	QMessageBox remove_source(this);
1592 1593 1594
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
1595 1596 1597 1598 1599 1600
	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();
1601
}
J
jp9000 已提交
1602

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

P
Palana 已提交
1605 1606 1607 1608 1609
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
1610 1611
void OBSBasic::TimedCheckForUpdates()
{
P
Palana 已提交
1612 1613 1614 1615
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
#else
J
jp9000 已提交
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631
	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)
		CheckForUpdates();
P
Palana 已提交
1632
#endif
J
jp9000 已提交
1633 1634 1635 1636
}

void OBSBasic::CheckForUpdates()
{
P
Palana 已提交
1637 1638 1639
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
#else
J
jp9000 已提交
1640 1641
	ui->actionCheckForUpdates->setEnabled(false);

1642 1643 1644 1645
	if (updateCheckThread) {
		updateCheckThread->wait();
		delete updateCheckThread;
	}
1646

1647 1648 1649 1650 1651 1652
	RemoteTextThread *thread = new RemoteTextThread(
			"https://obsproject.com/obs2_update/basic.json");
	updateCheckThread = thread;
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::updateFileFinished);
	updateCheckThread->start();
P
Palana 已提交
1653
#endif
J
jp9000 已提交
1654 1655
}

J
jp9000 已提交
1656 1657
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
1658 1659
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
1660 1661 1662 1663
#else
#define VERSION_ENTRY "other"
#endif

1664
void OBSBasic::updateFileFinished(const QString &text, const QString &error)
J
jp9000 已提交
1665 1666 1667
{
	ui->actionCheckForUpdates->setEnabled(true);

1668 1669
	if (text.isEmpty()) {
		blog(LOG_WARNING, "Update check failed: %s", QT_TO_UTF8(error));
J
jp9000 已提交
1670 1671 1672
		return;
	}

1673
	obs_data_t *returnData  = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
1674
	obs_data_t *versionData = obs_data_get_obj(returnData, VERSION_ENTRY);
J
jp9000 已提交
1675 1676 1677
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
1678 1679

	if (returnData && versionData && description && download) {
J
jp9000 已提交
1680 1681 1682
		long major   = obs_data_get_int(versionData, "major");
		long minor   = obs_data_get_int(versionData, "minor");
		long patch   = obs_data_get_int(versionData, "patch");
J
jp9000 已提交
1683 1684
		long version = MAKE_SEMANTIC_VERSION(major, minor, patch);

1685 1686
		blog(LOG_INFO, "Update check: last known remote version "
				"is %ld.%ld.%ld",
J
jp9000 已提交
1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706
				major, minor, patch);

		if (version > LIBOBS_API_VER) {
			QString     str = QTStr("UpdateAvailable.Text");
			QMessageBox messageBox(this);

			str = str.arg(QString::number(major),
			              QString::number(minor),
			              QString::number(patch),
			              download);

			messageBox.setWindowTitle(QTStr("UpdateAvailable"));
			messageBox.setTextFormat(Qt::RichText);
			messageBox.setText(str);
			messageBox.setInformativeText(QT_UTF8(description));
			messageBox.exec();

			long long t = (long long)time(nullptr);
			config_set_int(App()->GlobalConfig(), "General",
					"LastUpdateCheck", t);
1707
			config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
J
jp9000 已提交
1708 1709 1710 1711 1712 1713 1714 1715 1716
		}
	} else {
		blog(LOG_WARNING, "Bad JSON file received from server");
	}

	obs_data_release(versionData);
	obs_data_release(returnData);
}

1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773
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()) {
			QMessageBox::information(this,
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
			continue;
		}

		obs_source_t *source = obs_get_source_by_name(name.c_str());
		if (source) {
			QMessageBox::information(this,
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));

			obs_source_release(source);
			continue;
		}

		obs_scene_t *scene = obs_scene_duplicate(curScene,
				name.c_str());
		source = obs_scene_get_source(scene);
		obs_add_source(source);
		obs_scene_release(scene);

		obs_set_output_source(0, source);
		return;
	}
}

1774 1775 1776 1777
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1778
		obs_source_t *source = obs_scene_get_source(scene);
1779 1780 1781 1782 1783 1784 1785 1786 1787
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
1788
		obs_source_t *source = obs_sceneitem_get_source(item);
1789
		if (QueryRemoveSource(source))
J
jp9000 已提交
1790 1791 1792 1793
			obs_sceneitem_remove(item);
	}
}

1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807
struct ReorderInfo {
	int idx = 0;
	OBSBasic *window;

	inline ReorderInfo(OBSBasic *window_) : window(window_) {}
};

void OBSBasic::ReorderSceneItem(obs_sceneitem_t *item, size_t idx)
{
	int count = ui->sources->count();
	int idx_inv = count - (int)idx - 1;

	for (int i = 0; i < count; i++) {
		QListWidgetItem *listItem = ui->sources->item(i);
P
Palana 已提交
1808
		OBSSceneItem sceneItem = GetOBSRef<OBSSceneItem>(listItem);
1809 1810 1811 1812 1813

		if (sceneItem == item) {
			if ((int)idx_inv != i) {
				bool sel = (ui->sources->currentRow() == i);

1814
				listItem = TakeListItem(ui->sources, i);
1815 1816 1817
				if (listItem)  {
					ui->sources->insertItem(idx_inv,
							listItem);
1818 1819 1820
					SetupVisibilityItem(ui->sources,
							listItem, item);

1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835
					if (sel)
						ui->sources->setCurrentRow(
								idx_inv);
				}
			}

			break;
		}
	}
}

void OBSBasic::ReorderSources(OBSScene scene)
{
	ReorderInfo info(this);

1836
	if (scene != GetCurrentScene() || ui->sources->IgnoreReorder())
1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848
		return;

	obs_scene_enum_items(scene,
			[] (obs_scene_t*, obs_sceneitem_t *item, void *p)
			{
				ReorderInfo *info =
					reinterpret_cast<ReorderInfo*>(p);

				info->window->ReorderSceneItem(item,
					info->idx++);
				return true;
			}, &info);
J
jp9000 已提交
1849 1850

	SaveProject();
1851 1852
}

1853 1854
/* OBS Callbacks */

1855 1856 1857 1858 1859 1860 1861 1862 1863 1864
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)));
}

1865
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
1866 1867 1868
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

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

1871 1872
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1873 1874
}

1875
void OBSBasic::SceneItemRemoved(void *data, calldata_t *params)
1876
{
1877
	OBSBasic *window = static_cast<OBSBasic*>(data);
1878

1879
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
1880

1881 1882
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1883 1884
}

1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908
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));
}

1909
void OBSBasic::SourceAdded(void *data, calldata_t *params)
1910
{
J
jp9000 已提交
1911
	OBSBasic *window = static_cast<OBSBasic*>(data);
1912
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1913

1914
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1915
		QMetaObject::invokeMethod(window,
1916 1917
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1918 1919
}

1920
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
1921
{
1922
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1923

1924
	if (obs_scene_from_source(source) != NULL)
1925 1926 1927
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1928 1929
}

1930
void OBSBasic::SourceActivated(void *data, calldata_t *params)
1931
{
1932
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1933 1934 1935 1936 1937 1938 1939 1940
	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)));
}

1941
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
1942
{
1943
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1944 1945 1946 1947 1948 1949 1950 1951
	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)));
}

1952
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
1953 1954 1955 1956 1957 1958 1959 1960 1961 1962
{
	const char *newName  = calldata_string(params, "new_name");
	const char *prevName = calldata_string(params, "prev_name");

	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"RenameSources",
			Q_ARG(QString, QT_UTF8(newName)),
			Q_ARG(QString, QT_UTF8(prevName)));
}

1963
void OBSBasic::ChannelChanged(void *data, calldata_t *params)
1964
{
1965
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1966
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1967 1968

	if (channel == 0)
1969 1970 1971
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1972 1973
}

1974 1975 1976 1977 1978
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1979 1980 1981
	gs_effect_t    *solid = obs_get_solid_effect();
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
1982 1983 1984

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1985
	gs_effect_set_vec4(color, &colorVal);
1986

1987 1988
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1989 1990 1991 1992 1993 1994 1995 1996
	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();
1997 1998
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1999 2000 2001 2002

	gs_load_vertexbuffer(nullptr);
}

2003 2004
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
2005
	OBSBasic *window = static_cast<OBSBasic*>(data);
2006 2007 2008 2009
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
2010 2011
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
2012 2013 2014

	gs_viewport_push();
	gs_projection_push();
2015 2016 2017

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

2018 2019
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
2020
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
2021
			window->previewCX, window->previewCY);
2022

2023 2024
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
2025
	obs_render_main_view();
2026
	gs_load_vertexbuffer(nullptr);
2027

2028 2029
	/* --------------------------------------- */

2030 2031 2032
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
2033 2034 2035 2036

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
2037
	gs_reset_viewport();
J
jp9000 已提交
2038 2039 2040

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

2041 2042
	/* --------------------------------------- */

2043 2044
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
2045 2046 2047

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
2048 2049
}

2050 2051
/* Main class functions */

2052
obs_service_t *OBSBasic::GetService()
2053
{
2054
	if (!service) {
2055 2056
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
2057 2058
		obs_service_release(service);
	}
2059 2060 2061
	return service;
}

2062
void OBSBasic::SetService(obs_service_t *newService)
2063
{
2064
	if (newService)
2065 2066 2067
		service = newService;
}

2068 2069 2070 2071 2072 2073 2074
bool OBSBasic::StreamingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

2075 2076 2077 2078 2079 2080
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

2081 2082
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
2083
	return obs_reset_video(ovi);
2084 2085
}

2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098
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;
}

2099 2100 2101 2102 2103 2104
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 已提交
2105 2106
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
2107 2108 2109 2110 2111 2112 2113 2114 2115
#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 已提交
2116
		return VIDEO_FORMAT_RGBA;
2117 2118
}

2119
int OBSBasic::ResetVideo()
J
jp9000 已提交
2120
{
P
Palana 已提交
2121 2122
	ProfileScope("OBSBasic::ResetVideo");

J
jp9000 已提交
2123
	struct obs_video_info ovi;
2124
	int ret;
J
jp9000 已提交
2125

2126
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
2127

2128 2129 2130 2131 2132 2133 2134
	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 已提交
2135
	ovi.graphics_module = App()->GetRenderModule();
2136
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2137
			"Video", "BaseCX");
2138
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2139
			"Video", "BaseCY");
2140
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2141
			"Video", "OutputCX");
2142
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2143
			"Video", "OutputCY");
2144 2145 2146 2147 2148
	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;
J
jp9000 已提交
2149 2150
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
2151
	ovi.scale_type     = GetScaleType(basicConfig);
2152

2153
	ret = AttemptToResetVideo(&ovi);
2154 2155
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
		/* Try OpenGL if DirectX fails on windows */
2156
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
2157 2158 2159 2160
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
2161 2162
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
2163 2164
			ret = AttemptToResetVideo(&ovi);
		}
2165 2166
	} else if (ret == OBS_VIDEO_SUCCESS) {
		ResizePreview(ovi.base_width, ovi.base_height);
2167 2168
	}

2169
	return ret;
J
jp9000 已提交
2170
}
J
jp9000 已提交
2171

2172
bool OBSBasic::ResetAudio()
J
jp9000 已提交
2173
{
P
Palana 已提交
2174 2175
	ProfileScope("OBSBasic::ResetAudio");

2176
	struct obs_audio_info ai;
2177
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
2178 2179
			"SampleRate");

2180
	const char *channelSetupStr = config_get_string(basicConfig,
2181 2182 2183 2184 2185 2186 2187
			"Audio", "ChannelSetup");

	if (strcmp(channelSetupStr, "Mono") == 0)
		ai.speakers = SPEAKERS_MONO;
	else
		ai.speakers = SPEAKERS_STEREO;

2188
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
2189 2190

	return obs_reset_audio(&ai);
J
jp9000 已提交
2191 2192
}

2193
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceId,
2194
		const char *deviceDesc, int channel)
J
jp9000 已提交
2195
{
2196 2197
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
2198 2199 2200 2201
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
2202
		settings = obs_source_get_settings(source);
J
jp9000 已提交
2203
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214

		same = (strcmp(curId, deviceId) == 0);

		obs_data_release(settings);
		obs_source_release(source);
	}

	if (!same)
		obs_set_output_source(channel, nullptr);

	if (!same && strcmp(deviceId, "disabled") != 0) {
2215
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
2216
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
2217
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
2218
				sourceId, deviceDesc, settings, nullptr);
J
jp9000 已提交
2219 2220 2221 2222 2223 2224 2225
		obs_data_release(settings);

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

J
jp9000 已提交
2226
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
2227
{
2228
	QSize  targetSize;
J
jp9000 已提交
2229

2230
	/* resize preview panel to fix to the top section of the window */
2231
	targetSize = GetPixelSize(ui->preview);
2232
	GetScaleAndCenterPos(int(cx), int(cy),
2233 2234
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
2235
			previewX, previewY, previewScale);
J
jp9000 已提交
2236

2237 2238
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);
J
jp9000 已提交
2239 2240
}

2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257
void OBSBasic::CloseDialogs()
{
	QList<QDialog*> childDialogs = this->findChildren<QDialog *>();
	if (!childDialogs.isEmpty()) {
		for (int i = 0; i < childDialogs.size(); ++i) {
			childDialogs.at(i)->close();
		}
	}

	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}
}

void OBSBasic::ClearSceneData()
{
J
jp9000 已提交
2258 2259
	disableSaving++;

2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282
	CloseDialogs();

	ClearVolumeControls();
	ClearListItems(ui->scenes);
	ClearListItems(ui->sources);

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

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

	obs_enum_sources(cb, nullptr);

	sourceSceneRefs.clear();
J
jp9000 已提交
2283 2284

	disableSaving--;
2285 2286 2287

	blog(LOG_INFO, "All scene data cleared");
	blog(LOG_INFO, "------------------------------------------------");
2288 2289
}

J
jp9000 已提交
2290
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
2291
{
2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302
	if (outputHandler && outputHandler->Active()) {
		QMessageBox::StandardButton button = QMessageBox::question(
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

2303 2304 2305 2306
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

2307 2308 2309 2310 2311
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
2312 2313
	signalHandlers.clear();

J
jp9000 已提交
2314
	SaveProjectNow();
J
jp9000 已提交
2315
	disableSaving++;
J
jp9000 已提交
2316

2317 2318 2319
	/* Clear all scene data (dialogs, widgets, widget sub-items, scenes,
	 * sources, etc) so that all references are released before shutdown */
	ClearSceneData();
2320 2321
}

J
jp9000 已提交
2322
void OBSBasic::changeEvent(QEvent *event)
2323
{
J
jp9000 已提交
2324 2325
	/* TODO */
	UNUSED_PARAMETER(event);
2326 2327
}

J
jp9000 已提交
2328
void OBSBasic::resizeEvent(QResizeEvent *event)
2329
{
J
jp9000 已提交
2330 2331 2332 2333
	struct obs_video_info ovi;

	if (obs_get_video_info(&ovi))
		ResizePreview(ovi.base_width, ovi.base_height);
J
jp9000 已提交
2334

J
jp9000 已提交
2335
	OBSMainWindow::resizeEvent(event);
J
jp9000 已提交
2336 2337
}

2338 2339
void OBSBasic::on_actionShow_Recordings_triggered()
{
2340 2341 2342 2343
	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");
2344 2345 2346
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
2347 2348
void OBSBasic::on_actionRemux_triggered()
{
2349 2350 2351 2352
	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");
P
Palana 已提交
2353 2354 2355 2356
	OBSRemux remux(path, this);
	remux.exec();
}

P
Palana 已提交
2357 2358 2359 2360 2361 2362
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

J
jp9000 已提交
2363 2364
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
2365 2366 2367 2368 2369
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
2370 2371 2372
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
2373 2374 2375

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

2378 2379 2380 2381 2382
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

2383 2384 2385 2386 2387
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

2388 2389
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
2390
{
2391
	obs_source_t *source = NULL;
J
jp9000 已提交
2392

2393 2394 2395 2396
	if (sceneChanging)
		return;

	if (current) {
2397
		obs_scene_t *scene;
J
jp9000 已提交
2398

P
Palana 已提交
2399
		scene = GetOBSRef<OBSScene>(current);
2400
		source = obs_scene_get_source(scene);
2401 2402
	}

2403
	/* TODO: allow transitions */
2404
	obs_set_output_source(0, source);
2405 2406

	UNUSED_PARAMETER(prev);
2407 2408
}

J
jp9000 已提交
2409 2410
void OBSBasic::EditSceneName()
{
2411 2412 2413 2414 2415 2416
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

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

J
jp9000 已提交
2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441
static void AddProjectorMenuMonitors(QMenu *parent, QObject *target,
		const char *slot)
{
	QAction *action;
	std::vector<MonitorInfo> monitors;
	GetMonitors(monitors);

	for (int i = 0; (size_t)i < monitors.size(); i++) {
		const MonitorInfo &monitor = monitors[i];

		QString str = QString("%1 %2: %3x%4 @ %5,%6").
			arg(QTStr("Display"),
			    QString::number(i),
			    QString::number((int)monitor.cx),
			    QString::number((int)monitor.cy),
			    QString::number((int)monitor.x),
			    QString::number((int)monitor.y));

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

J
jp9000 已提交
2442
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
2443
{
J
jp9000 已提交
2444
	QListWidgetItem *item = ui->scenes->itemAt(pos);
J
jp9000 已提交
2445
	QPointer<QMenu> sceneProjectorMenu;
J
jp9000 已提交
2446

2447
	QMenu popup(this);
J
jp9000 已提交
2448
	QMenu order(QTStr("Basic.MainMenu.Edit.Order"), this);
J
jp9000 已提交
2449 2450 2451
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
2452 2453
	if (item) {
		popup.addSeparator();
2454 2455
		popup.addAction(QTStr("Duplicate"),
				this, SLOT(DuplicateSelectedScene()));
P
Palana 已提交
2456 2457
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
2458
		popup.addAction(QTStr("Remove"),
2459 2460
				this, SLOT(RemoveSelectedScene()),
				DeleteKeys.front());
J
jp9000 已提交
2461
		popup.addSeparator();
J
jp9000 已提交
2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474

		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();
J
jp9000 已提交
2475 2476 2477 2478 2479
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
		popup.addSeparator();
J
jp9000 已提交
2480 2481
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
P
Palana 已提交
2482
	}
J
jp9000 已提交
2483 2484

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

J
jp9000 已提交
2487
void OBSBasic::on_actionAddScene_triggered()
2488
{
2489
	string name;
S
Socapex 已提交
2490
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
2491 2492 2493

	int i = 1;
	QString placeHolderText = format.arg(i);
2494
	obs_source_t *source = nullptr;
P
Palana 已提交
2495 2496
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
2497
		placeHolderText = format.arg(++i);
P
Palana 已提交
2498
	}
S
Socapex 已提交
2499

J
jp9000 已提交
2500
	bool accepted = NameDialog::AskForName(this,
2501 2502
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
2503 2504
			name,
			placeHolderText);
2505

J
jp9000 已提交
2506
	if (accepted) {
J
jp9000 已提交
2507 2508
		if (name.empty()) {
			QMessageBox::information(this,
2509 2510
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
2511 2512 2513 2514
			on_actionAddScene_triggered();
			return;
		}

2515
		obs_source_t *source = obs_get_source_by_name(name.c_str());
2516
		if (source) {
J
jp9000 已提交
2517
			QMessageBox::information(this,
2518 2519
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
2520 2521

			obs_source_release(source);
J
jp9000 已提交
2522
			on_actionAddScene_triggered();
2523 2524 2525
			return;
		}

2526
		obs_scene_t *scene = obs_scene_create(name.c_str());
2527
		source = obs_scene_get_source(scene);
2528
		obs_add_source(source);
2529
		obs_scene_release(scene);
2530 2531

		obs_set_output_source(0, source);
2532
	}
2533 2534
}

J
jp9000 已提交
2535
void OBSBasic::on_actionRemoveScene_triggered()
2536
{
2537
	OBSScene     scene  = GetCurrentScene();
2538
	obs_source_t *source = obs_scene_get_source(scene);
2539 2540 2541

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
2542 2543
}

J
jp9000 已提交
2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563
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;
}

J
jp9000 已提交
2564
void OBSBasic::on_actionSceneUp_triggered()
2565
{
J
jp9000 已提交
2566
	ChangeSceneIndex(true, -1, 0);
2567 2568
}

J
jp9000 已提交
2569
void OBSBasic::on_actionSceneDown_triggered()
2570
{
J
jp9000 已提交
2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582
	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);
2583 2584
}

2585 2586
void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
2587
{
2588
	auto select_one = [] (obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2589 2590
			void *param)
	{
2591
		obs_sceneitem_t *selectedItem =
J
jp9000 已提交
2592 2593 2594 2595 2596 2597 2598 2599 2600 2601
			*reinterpret_cast<OBSSceneItem*>(param);
		obs_sceneitem_select(item, (selectedItem == item));

		UNUSED_PARAMETER(scene);
		return true;
	};

	if (!current)
		return;

P
Palana 已提交
2602
	OBSSceneItem item = GetOBSRef<OBSSceneItem>(current);
2603
	obs_source_t *source = obs_sceneitem_get_source(item);
2604 2605 2606
	if ((obs_source_get_output_flags(source) & OBS_SOURCE_VIDEO) == 0)
		return;

2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619
	auto IgnoreSelfUpdate = [&](obs_scene_t *scene)
	{
		ignoreSelectionUpdate = true;
		obs_scene_enum_items(scene, select_one, &item);
		ignoreSelectionUpdate = false;
	};
	using IgnoreSelfUpdate_t = decltype(IgnoreSelfUpdate);

	obs_scene_atomic_update(GetCurrentScene(),
			[](void *data, obs_scene_t *scene)
	{
		(*static_cast<IgnoreSelfUpdate_t*>(data))(scene);
	}, static_cast<void*>(&IgnoreSelfUpdate));
J
jp9000 已提交
2620

2621
	UNUSED_PARAMETER(prev);
2622 2623
}

J
jp9000 已提交
2624 2625
void OBSBasic::EditSceneItemName()
{
2626 2627
	QListWidgetItem *item = ui->sources->currentItem();
	Qt::ItemFlags flags   = item->flags();
P
Palana 已提交
2628
	OBSSceneItem sceneItem= GetOBSRef<OBSSceneItem>(item);
2629 2630
	obs_source_t *source  = obs_sceneitem_get_source(sceneItem);
	const char *name      = obs_source_get_name(source);
2631

2632
	item->setText(QT_UTF8(name));
2633
	item->setFlags(flags | Qt::ItemIsEditable);
2634
	ui->sources->removeItemWidget(item);
2635 2636
	ui->sources->editItem(item);
	item->setFlags(flags);
J
jp9000 已提交
2637 2638
}

J
jp9000 已提交
2639
void OBSBasic::CreateSourcePopupMenu(QListWidgetItem *item, bool preview)
2640
{
2641
	QMenu popup(this);
J
jp9000 已提交
2642 2643
	QPointer<QMenu> previewProjector;
	QPointer<QMenu> sourceProjector;
J
jp9000 已提交
2644 2645 2646 2647 2648 2649

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

J
jp9000 已提交
2653 2654 2655 2656 2657 2658
		previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
				SLOT(OpenPreviewProjector()));

		popup.addMenu(previewProjector);

J
jp9000 已提交
2659 2660 2661
		popup.addSeparator();
	}

J
jp9000 已提交
2662 2663 2664 2665 2666 2667 2668 2669
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

	if (item) {
		if (addSourceMenu)
			popup.addSeparator();

J
John Bradley 已提交
2670
		OBSSceneItem sceneItem = GetSceneItem(item);
2671
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
John Bradley 已提交
2672 2673
		QAction *action;

J
jp9000 已提交
2674 2675
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
2676 2677
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
2678
				DeleteKeys.front());
J
jp9000 已提交
2679 2680
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
2681
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
2682 2683 2684 2685 2686 2687 2688

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

		popup.addSeparator();
		popup.addMenu(sourceProjector);
J
jp9000 已提交
2689
		popup.addSeparator();
J
John Bradley 已提交
2690 2691 2692 2693 2694 2695 2696

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

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

J
jp9000 已提交
2697 2698
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
2699 2700 2701 2702 2703
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

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

J
jp9000 已提交
2706 2707 2708 2709 2710
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->itemAt(pos), false);
}

P
Palana 已提交
2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722
void OBSBasic::on_sources_itemDoubleClicked(QListWidgetItem *witem)
{
	if (!witem)
		return;

	OBSSceneItem item = GetSceneItem(witem);
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreatePropertiesWindow(source);
}

J
jp9000 已提交
2723
void OBSBasic::AddSource(const char *id)
2724
{
2725 2726 2727 2728
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
	}
2729 2730
}

2731
QMenu *OBSBasic::CreateAddSourcePopupMenu()
2732
{
2733
	const char *type;
J
jp9000 已提交
2734 2735
	bool foundValues = false;
	size_t idx = 0;
2736

2737
	QMenu *popup = new QMenu(QTStr("Add"), this);
J
jp9000 已提交
2738
	while (obs_enum_input_types(idx++, &type)) {
2739
		const char *name = obs_source_get_display_name(
2740
				OBS_SOURCE_TYPE_INPUT, type);
2741

2742 2743 2744
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
2745 2746
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
2747 2748 2749
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
2750

J
jp9000 已提交
2751
		foundValues = true;
2752 2753
	}

2754 2755 2756
	if (!foundValues) {
		delete popup;
		popup = nullptr;
2757
	}
2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783

	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).
		QMessageBox::information(this,
				QTStr("Basic.Main.AddSourceHelp.Title"),
				QTStr("Basic.Main.AddSourceHelp.Text"));
		return;
	}

	QPointer<QMenu> popup = CreateAddSourcePopupMenu();
	if (popup)
		popup->exec(pos);
2784 2785
}

J
jp9000 已提交
2786
void OBSBasic::on_actionAddSource_triggered()
2787
{
J
jp9000 已提交
2788
	AddSourcePopupMenu(QCursor::pos());
2789 2790
}

J
jp9000 已提交
2791
void OBSBasic::on_actionRemoveSource_triggered()
2792
{
2793
	OBSSceneItem item   = GetCurrentSceneItem();
2794
	obs_source_t *source = obs_sceneitem_get_source(item);
2795 2796

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
2797
		obs_sceneitem_remove(item);
2798 2799
}

J
John Bradley 已提交
2800 2801 2802 2803 2804 2805 2806 2807 2808
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
2809
void OBSBasic::on_actionSourceProperties_triggered()
2810
{
2811
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2812
	OBSSource source = obs_sceneitem_get_source(item);
2813

2814 2815
	if (source)
		CreatePropertiesWindow(source);
2816 2817
}

J
jp9000 已提交
2818
void OBSBasic::on_actionSourceUp_triggered()
2819
{
J
jp9000 已提交
2820
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2821
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
2822
}
J
jp9000 已提交
2823

J
jp9000 已提交
2824
void OBSBasic::on_actionSourceDown_triggered()
2825
{
J
jp9000 已提交
2826
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2827
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
2828 2829
}

J
jp9000 已提交
2830 2831 2832
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2833
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
2834 2835 2836 2837 2838
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2839
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
2840 2841 2842 2843 2844
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2845
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
2846 2847 2848 2849 2850
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2851
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
2852 2853
}

2854
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
2855
{
2856
	char logDir[512];
2857
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2858
		return nullptr;
J
jp9000 已提交
2859 2860 2861 2862 2863

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

2864
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
2865 2866 2867 2868 2869 2870 2871 2872
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

void OBSBasic::UploadLog(const char *file)
{
2873
	BPtr<char> fileString{ReadLogFile(file)};
J
jp9000 已提交
2874

2875
	if (!fileString)
J
jp9000 已提交
2876 2877
		return;

2878
	if (!*fileString)
J
jp9000 已提交
2879 2880 2881 2882
		return;

	ui->menuLogFiles->setEnabled(false);

2883
	auto data_deleter = [](obs_data_t *d) { obs_data_release(d); };
2884
	using data_t = unique_ptr<struct obs_data, decltype(data_deleter)>;
J
jp9000 已提交
2885

2886 2887 2888
	data_t content{obs_data_create(), data_deleter};
	data_t files{obs_data_create(), data_deleter};
	data_t request{obs_data_create(), data_deleter};
J
jp9000 已提交
2889

2890
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
2891

2892 2893 2894 2895 2896 2897 2898 2899 2900 2901
	obs_data_set_obj(files.get(), file, content.get());

	stringstream ss;
	ss << "OBS " << App()->GetVersionString()
	   << " log file uploaded at " << CurrentDateTimeString();
	obs_data_set_string(request.get(), "description", ss.str().c_str());
	obs_data_set_bool(request.get(), "public", false);
	obs_data_set_obj(request.get(), "files", files.get());

	const char *json = obs_data_get_json(request.get());
F
fryshorts 已提交
2902
	if (!json) {
2903 2904 2905 2906
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

F
fryshorts 已提交
2907 2908 2909
	QBuffer *postData = new QBuffer();
	postData->setData(json, (int) strlen(json));

2910 2911 2912 2913
	if (logUploadThread) {
		logUploadThread->wait();
		delete logUploadThread;
	}
F
fryshorts 已提交
2914

2915 2916 2917 2918 2919 2920 2921
	RemoteTextThread *thread = new RemoteTextThread(
			"https://api.github.com/gists",
			"application/json", json);
	logUploadThread = thread;
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::logUploadFinished);
	logUploadThread->start();
J
jp9000 已提交
2922 2923
}

P
Palana 已提交
2924 2925
void OBSBasic::on_actionShowLogs_triggered()
{
2926
	char logDir[512];
2927
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2928 2929
		return;

P
Palana 已提交
2930 2931 2932 2933
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
2934 2935 2936 2937 2938 2939 2940 2941 2942 2943
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

void OBSBasic::on_actionUploadLastLog_triggered()
{
	UploadLog(App()->GetLastLog());
}

2944 2945 2946
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
2947
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959
		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 已提交
2960 2961 2962 2963 2964
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

2965
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
2966 2967 2968
{
	ui->menuLogFiles->setEnabled(true);

2969
	if (text.isEmpty()) {
J
jp9000 已提交
2970 2971
		QMessageBox::information(this,
				QTStr("LogReturnDialog.ErrorUploadingLog"),
2972
				error);
J
jp9000 已提交
2973 2974 2975
		return;
	}

2976
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
2977
	QString logURL         = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
2978 2979 2980 2981 2982 2983
	obs_data_release(returnData);

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

J
jp9000 已提交
2984
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
2985
		obs_source_t *source, const string &name)
2986
{
2987 2988 2989 2990
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

2991
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
2992
	QListWidgetItem *listItem    = listWidget->currentItem();
2993

2994
	if (foundSource || name.empty()) {
2995
		listItem->setText(QT_UTF8(prevName));
2996

2997
		if (foundSource) {
2998 2999 3000 3001 3002 3003 3004 3005 3006
			QMessageBox::information(parent,
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
			QMessageBox::information(parent,
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

3007 3008 3009
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
3010
		obs_source_set_name(source, name.c_str());
3011 3012 3013
	}
}

J
jp9000 已提交
3014 3015 3016 3017 3018
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
3019
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
3020 3021 3022 3023

	if (!scene)
		return;

3024
	obs_source_t *source = obs_scene_get_source(scene);
3025
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
3026 3027 3028 3029 3030 3031 3032 3033 3034

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
3035
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
3036 3037 3038 3039

	if (!item)
		return;

3040
	obs_source_t *source = obs_sceneitem_get_source(item);
3041
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
3042

3043 3044 3045 3046
	QListWidgetItem *listItem = ui->sources->currentItem();
	listItem->setText(QString());
	SetupVisibilityItem(ui->sources, listItem, item);

J
jp9000 已提交
3047 3048 3049
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
3050 3051 3052 3053 3054 3055 3056 3057
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
3058 3059 3060 3061 3062 3063 3064 3065
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
3066 3067 3068 3069 3070 3071 3072 3073 3074
#define RECORDING_START \
	"==== Recording Start ==============================================="
#define RECORDING_STOP \
	"==== Recording Stop ================================================"
#define STREAMING_START \
	"==== Streaming Start ==============================================="
#define STREAMING_STOP \
	"==== Streaming Stop ================================================"

3075 3076 3077 3078
void OBSBasic::StartStreaming()
{
	SaveProject();

J
jp9000 已提交
3079 3080
	ui->streamButton->setEnabled(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
3081

J
jp9000 已提交
3082 3083 3084
	if (!outputHandler->StartStreaming(service)) {
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
3085 3086 3087 3088 3089 3090 3091 3092 3093
	}
}

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

	if (outputHandler->StreamingActive())
		outputHandler->StopStreaming();
J
jp9000 已提交
3094 3095

	if (!outputHandler->Active()) {
J
jp9000 已提交
3096
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
3097
	}
3098 3099
}

J
jp9000 已提交
3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147
void OBSBasic::ForceStopStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
		outputHandler->ForceStopStreaming();

	if (!outputHandler->Active()) {
		ui->profileMenu->setEnabled(true);
	}
}

void OBSBasic::StreamDelayStarting(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
	ui->streamButton->setEnabled(true);

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

void OBSBasic::StreamDelayStopping(int sec)
{
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
	ui->streamButton->setEnabled(true);

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

3148
void OBSBasic::StreamingStart()
3149
{
3150
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
3151
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
3152
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
J
jp9000 已提交
3153
	ui->profileMenu->setEnabled(false);
3154
	blog(LOG_INFO, STREAMING_START);
3155 3156
}

3157
void OBSBasic::StreamingStop(int code)
3158
{
3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173
	const char *errorMessage;

	switch (code) {
	case OBS_OUTPUT_BAD_PATH:
		errorMessage = Str("Output.ConnectFail.BadPath");
		break;

	case OBS_OUTPUT_CONNECT_FAILED:
		errorMessage = Str("Output.ConnectFail.ConnectFailed");
		break;

	case OBS_OUTPUT_INVALID_STREAM:
		errorMessage = Str("Output.ConnectFail.InvalidStream");
		break;

3174
	default:
3175 3176 3177 3178 3179 3180 3181 3182 3183 3184
	case OBS_OUTPUT_ERROR:
		errorMessage = Str("Output.ConnectFail.Error");
		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 */
		errorMessage = Str("Output.ConnectFail.Disconnected");
	}

J
jp9000 已提交
3185
	ui->statusbar->StreamStopped();
3186

3187
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
3188
	ui->streamButton->setEnabled(true);
3189

3190
	if (!outputHandler->Active())
J
jp9000 已提交
3191
		ui->profileMenu->setEnabled(true);
3192 3193

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

3195 3196 3197 3198
	if (code != OBS_OUTPUT_SUCCESS)
		QMessageBox::information(this,
				QTStr("Output.ConnectFail.Title"),
				QT_UTF8(errorMessage));
J
jp9000 已提交
3199 3200 3201 3202 3203 3204

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

3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220
void OBSBasic::StartRecording()
{
	SaveProject();

	if (!outputHandler->RecordingActive())
		outputHandler->StartRecording();
}

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

	if (outputHandler->RecordingActive())
		outputHandler->StopRecording();
J
jp9000 已提交
3221 3222

	if (!outputHandler->Active()) {
J
jp9000 已提交
3223
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
3224
	}
3225 3226
}

P
Palana 已提交
3227 3228
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
3229
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
3230
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
J
jp9000 已提交
3231
	ui->profileMenu->setEnabled(false);
3232
	blog(LOG_INFO, RECORDING_START);
P
Palana 已提交
3233 3234
}

3235
void OBSBasic::RecordingStop(int code)
3236
{
P
Palana 已提交
3237
	ui->statusbar->RecordingStopped();
3238
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
3239
	blog(LOG_INFO, RECORDING_STOP);
3240 3241 3242 3243 3244

	if (code == OBS_OUTPUT_UNSUPPORTED)
		QMessageBox::information(this,
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));
J
jp9000 已提交
3245 3246

	if (!outputHandler->Active()) {
J
jp9000 已提交
3247
		ui->profileMenu->setEnabled(true);
J
jp9000 已提交
3248
	}
3249
}
3250

3251 3252
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
3253
	if (outputHandler->StreamingActive()) {
3254
		StopStreaming();
3255
	} else {
3256
		StartStreaming();
3257 3258 3259 3260 3261
	}
}

void OBSBasic::on_recordButton_clicked()
{
3262 3263 3264 3265
	if (outputHandler->RecordingActive())
		StopRecording();
	else
		StartRecording();
J
jp9000 已提交
3266 3267
}

J
jp9000 已提交
3268
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
3269
{
3270 3271
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
3272
}
3273

3274 3275 3276 3277 3278 3279
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299
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 已提交
3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->currentItem(), true);

	UNUSED_PARAMETER(pos);
}

void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
J
jp9000 已提交
3311
	QPointer<QMenu> previewProjector;
J
jp9000 已提交
3312 3313 3314 3315 3316

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

J
jp9000 已提交
3319 3320 3321 3322 3323
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));

	popup.addMenu(previewProjector);
J
jp9000 已提交
3324 3325 3326 3327 3328
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391
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;
	} 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);
}

3392
config_t *OBSBasic::Config() const
3393 3394 3395
{
	return basicConfig;
}
J
jp9000 已提交
3396 3397 3398

void OBSBasic::on_actionEditTransform_triggered()
{
3399 3400 3401
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
3402 3403
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
3404
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
3405 3406 3407 3408
}

void OBSBasic::on_actionResetTransform_triggered()
{
3409
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
3410 3411 3412 3413
	{
		if (!obs_sceneitem_selected(item))
			return true;

3414
		obs_transform_info info;
J
jp9000 已提交
3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431
		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);

		UNUSED_PARAMETER(scene);
		UNUSED_PARAMETER(param);
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
}

3432
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
3433 3434 3435 3436 3437
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

3440
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
3441 3442 3443 3444
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
3445 3446
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
3447 3448
	};

3449 3450 3451 3452 3453 3454
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

3455
static vec3 GetItemTL(obs_sceneitem_t *item)
3456 3457 3458
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
3459 3460 3461
	return tl;
}

3462
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
3463 3464 3465 3466
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
3467
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
3468 3469 3470
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
3471
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
3472 3473
}

3474
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3475 3476 3477 3478 3479 3480 3481 3482 3483
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
3484
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
3485 3486
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
3487
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513

	SetItemTL(item, tl);

	UNUSED_PARAMETER(scene);
	UNUSED_PARAMETER(param);
	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);
}

3514
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3515 3516 3517 3518 3519 3520 3521 3522 3523 3524
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
3525
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
3526
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
3527
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
3528 3529

	SetItemTL(item, tl);
J
jp9000 已提交
3530 3531

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
3532 3533 3534 3535 3536
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
3537 3538
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
3539 3540 3541 3542 3543 3544
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
3545 3546
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
3547 3548 3549 3550
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

3551
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3552 3553 3554 3555 3556 3557 3558 3559 3560 3561
		void *param)
{
	obs_bounds_type boundsType = *reinterpret_cast<obs_bounds_type*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	obs_video_info ovi;
	obs_get_video_info(&ovi);

3562
	obs_transform_info itemInfo;
J
jp9000 已提交
3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594
	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);
}

void OBSBasic::on_actionCenterToScreen_triggered()
{
3595
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620
	{
		vec3 tl, br, itemCenter, screenCenter, offset;
		obs_video_info ovi;

		if (!obs_sceneitem_selected(item))
			return true;

		obs_get_video_info(&ovi);

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

		GetItemBox(item, tl, br);

		vec3_sub(&itemCenter, &br, &tl);
		vec3_mulf(&itemCenter, &itemCenter, 0.5f);
		vec3_add(&itemCenter, &itemCenter, &tl);

		vec3_sub(&offset, &screenCenter, &itemCenter);
		vec3_add(&tl, &tl, &offset);

		SetItemTL(item, tl);

		UNUSED_PARAMETER(scene);
J
jp9000 已提交
3621
		UNUSED_PARAMETER(param);
3622 3623 3624 3625
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
J
jp9000 已提交
3626
}
J
jp9000 已提交
3627 3628 3629

void OBSBasic::TogglePreview()
{
3630 3631
	bool enabled = !obs_display_enabled(ui->preview->GetDisplay());
	obs_display_set_enabled(ui->preview->GetDisplay(), enabled);
J
jp9000 已提交
3632 3633 3634
	ui->preview->setVisible(enabled);
	ui->previewDisabledLabel->setVisible(!enabled);
}
3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645

void OBSBasic::Nudge(int dist, MoveDir dir)
{
	struct MoveInfo {
		float dist;
		MoveDir dir;
	} info = {(float)dist, dir};

	auto func = [] (obs_scene_t*, obs_sceneitem_t *item, void *param)
	{
		MoveInfo *info = reinterpret_cast<MoveInfo*>(param);
J
jp9000 已提交
3646
		struct vec2 dir;
3647 3648
		struct vec2 pos;

J
jp9000 已提交
3649 3650
		vec2_set(&dir, 0.0f, 0.0f);

3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673
		if (!obs_sceneitem_selected(item))
			return true;

		switch (info->dir) {
		case MoveDir::Up:    dir.y = -info->dist; break;
		case MoveDir::Down:  dir.y =  info->dist; break;
		case MoveDir::Left:  dir.x = -info->dist; break;
		case MoveDir::Right: dir.x =  info->dist; break;
		}

		obs_sceneitem_get_pos(item, &pos);
		vec2_add(&pos, &pos, &dir);
		obs_sceneitem_set_pos(item, &pos);
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, &info);
}

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 已提交
3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700 3701 3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712 3713 3714

void OBSBasic::OpenProjector(obs_source_t *source, int monitor)
{
	/* seriously?  10 monitors? */
	if (monitor > 9)
		return;

	delete projectors[monitor];
	projectors[monitor].clear();

	OBSProjector *projector = new OBSProjector(this, source);
	projector->Init(monitor);

	projectors[monitor] = projector;
}

void OBSBasic::OpenPreviewProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OpenProjector(nullptr, monitor);
}

void OBSBasic::OpenSourceProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSSceneItem item = GetCurrentSceneItem();
	if (!item)
		return;

	OpenProjector(obs_sceneitem_get_source(item), monitor);
}

void OBSBasic::OpenSceneProjector()
{
	int monitor = sender()->property("monitor").toInt();
	OBSScene scene = GetCurrentScene();
	if (!scene)
		return;

	OpenProjector(obs_scene_get_source(scene), monitor);
}
3715 3716 3717 3718 3719

void OBSBasic::UpdateTitleBar()
{
	stringstream name;

J
jp9000 已提交
3720 3721
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "Profile");
J
jp9000 已提交
3722 3723 3724
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

3725
	name << "OBS " << App()->GetVersionString();
J
jp9000 已提交
3726
	name << " - " << Str("TitleBar.Profile") << ": " << profile;
J
jp9000 已提交
3727
	name << " - " << Str("TitleBar.Scenes") << ": " << sceneCollection;
3728 3729 3730

	setWindowTitle(QT_UTF8(name.str().c_str()));
}
J
jp9000 已提交
3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754

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