window-basic-main.cpp 97.3 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 65 66 67 68 69 70 71 72
namespace {

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

}

J
jp9000 已提交
73 74
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
75
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
76
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
77
Q_DECLARE_METATYPE(SignalContainer<OBSScene>);
J
jp9000 已提交
78

P
Palana 已提交
79 80 81 82 83 84 85 86 87 88 89 90 91
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));
}

92 93
static void AddExtraModulePaths()
{
94
	char base_module_dir[512];
95
	int ret = GetConfigPath(base_module_dir, sizeof(base_module_dir),
96
			"obs-studio/plugins/%module%");
B
BtbN 已提交
97

98
	if (ret <= 0)
99 100 101
		return;

	string path = (char*)base_module_dir;
102
#if defined(__APPLE__)
103
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
104 105 106 107 108 109 110
#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
111 112
}

113 114
static QList<QKeySequence> DeleteKeys;

115
OBSBasic::OBSBasic(QWidget *parent)
116
	: OBSMainWindow  (parent),
J
jp9000 已提交
117
	  ui             (new Ui::OBSBasic)
118 119
{
	ui->setupUi(this);
J
jp9000 已提交
120 121
	ui->previewDisabledLabel->setVisible(false);

S
Socapex 已提交
122
	copyActionsDynamicProperties();
123

124 125
	ui->sources->setItemDelegate(new VisibilityItemDelegate(ui->sources));

J
jp9000 已提交
126
	int width = config_get_int(App()->GlobalConfig(), "BasicWindow", "cx");
127 128 129

	// Check if no values are saved (new installation).
	if (width != 0) {
J
jp9000 已提交
130 131 132
		int height = config_get_int(App()->GlobalConfig(),
				"BasicWindow", "cy");
		int posx = config_get_int(App()->GlobalConfig(), "BasicWindow",
133
				"posx");
J
jp9000 已提交
134
		int posy = config_get_int(App()->GlobalConfig(), "BasicWindow",
135 136
				"posy");

137
		setGeometry(posx, posy, width, height);
138 139
	}

140
	char styleSheetPath[512];
J
jp9000 已提交
141 142
	int ret = GetProfilePath(styleSheetPath, sizeof(styleSheetPath),
			"stylesheet.qss");
143
	if (ret > 0) {
H
HomeWorld 已提交
144 145 146 147 148
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
149 150
	}

P
Palana 已提交
151 152 153
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
154
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
155

156 157 158
	qRegisterMetaTypeStreamOperators<
		std::vector<std::shared_ptr<OBSSignal>>>(
				"std::vector<std::shared_ptr<OBSSignal>>");
159
	qRegisterMetaTypeStreamOperators<OBSScene>("OBSScene");
160
	qRegisterMetaTypeStreamOperators<OBSSceneItem>("OBSSceneItem");
161

P
Palana 已提交
162 163 164
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

165
	auto displayResize = [this]() {
166 167 168 169
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
170 171 172 173
	};

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

P
Palana 已提交
175 176
	installEventFilter(CreateShortcutFilter());

177 178 179 180 181
	stringstream name;
	name << "OBS " << App()->GetVersionString();	
	blog(LOG_INFO, "%s", name.str().c_str());
	blog(LOG_INFO, "---------------------------------");

182
	UpdateTitleBar();
J
jp9000 已提交
183 184 185 186 187 188 189 190 191 192 193 194 195 196

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

	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
199 200 201 202
	cpuUsageTimer = new QTimer(this);
	connect(cpuUsageTimer, SIGNAL(timeout()),
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
203

204 205 206 207 208 209
	DeleteKeys =
#ifdef __APPLE__
		QList<QKeySequence>{{Qt::Key_Backspace}} <<
#endif
		QKeySequence::keyBindings(QKeySequence::Delete);

210
#ifdef __APPLE__
211 212
	ui->actionRemoveSource->setShortcuts(DeleteKeys);
	ui->actionRemoveScene->setShortcuts(DeleteKeys);
213 214 215

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
216
#endif
217 218 219 220 221 222 223 224 225 226 227 228 229 230

	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()));
231 232
}

233 234
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent,
		vector<OBSSource> &audioSources)
235
{
236
	obs_source_t *source = obs_get_output_source(channel);
237 238 239
	if (!source)
		return;

240 241
	audioSources.push_back(source);

242
	obs_data_t *data = obs_save_source(source);
243

J
jp9000 已提交
244
	obs_data_set_obj(parent, name, data);
245 246 247 248 249

	obs_data_release(data);
	obs_source_release(source);
}

J
jp9000 已提交
250
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder)
251
{
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277
	obs_data_t *saveData = obs_data_create();

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

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

	auto FilterAudioSources = [&](obs_source_t *source)
	{
		return find(begin(audioSources), end(audioSources), source) ==
				end(audioSources);
	};
	using FilterAudioSources_t = decltype(FilterAudioSources);

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

	obs_source_t *currentScene = obs_get_output_source(0);
	const char   *sceneName   = obs_source_get_name(currentScene);
278

J
jp9000 已提交
279 280 281
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

J
jp9000 已提交
282
	obs_data_set_string(saveData, "current_scene", sceneName);
J
jp9000 已提交
283
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
284
	obs_data_set_string(saveData, "name", sceneCollection);
J
jp9000 已提交
285
	obs_data_set_array(saveData, "sources", sourcesArray);
286 287 288 289 290 291
	obs_data_array_release(sourcesArray);
	obs_source_release(currentScene);

	return saveData;
}

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

312 313 314 315 316 317 318 319 320 321 322 323
void OBSBasic::ClearVolumeControls()
{
	VolControl *control;

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

	volumes.clear();
}

J
jp9000 已提交
324 325 326 327 328 329 330 331 332 333 334 335 336 337 338
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;
}

339 340
void OBSBasic::Save(const char *file)
{
J
jp9000 已提交
341 342
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
	obs_data_t *saveData  = GenerateSaveData(sceneOrder);
343 344 345

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

	obs_data_release(saveData);
J
jp9000 已提交
348
	obs_data_array_release(sceneOrder);
349 350
}

351
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
352
{
353
	obs_data_t *data = obs_data_get_obj(parent, name);
354 355 356
	if (!data)
		return;

357
	obs_source_t *source = obs_load_source(data);
358 359 360 361 362 363 364 365
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384
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;
}

385
void OBSBasic::CreateFirstRunSources()
386
{
387 388 389 390 391 392 393 394 395
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

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

398
void OBSBasic::CreateDefaultScene(bool firstStart)
399 400 401 402 403 404 405
{
	disableSaving++;

	ClearSceneData();

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

406
	if (firstStart)
407
		CreateFirstRunSources();
408 409

	obs_set_output_source(0, obs_scene_get_source(scene));
410
	AddScene(obs_scene_get_source(scene));
411
	obs_scene_release(scene);
J
jp9000 已提交
412 413

	disableSaving--;
414 415
}

J
jp9000 已提交
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
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);
	}
}

445 446
void OBSBasic::Load(const char *file)
{
447
	if (!file || !os_file_exists(file)) {
448
		blog(LOG_INFO, "No scene file found, creating default scene");
449
		CreateDefaultScene(true);
450
		SaveProject();
451 452 453
		return;
	}

454 455 456 457 458
	disableSaving++;

	obs_data_t *data = obs_data_create_from_json_file_safe(file, "bak");
	if (!data) {
		disableSaving--;
459 460
		blog(LOG_ERROR, "Failed to load '%s', creating default scene",
				file);
461
		CreateDefaultScene(true);
J
jp9000 已提交
462
		SaveProject();
463
		return;
464
	}
465

466 467
	ClearSceneData();

J
jp9000 已提交
468
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
469
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
J
jp9000 已提交
470 471
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
J
jp9000 已提交
472 473 474 475 476 477 478

	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");
479
	obs_source_t     *curScene;
480

J
jp9000 已提交
481 482 483
	if (!name || !*name)
		name = curSceneCollection;

484 485 486 487 488 489
	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);

490 491
	obs_load_sources(sources);

J
jp9000 已提交
492 493 494
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

495 496 497 498 499
	curScene = obs_get_source_by_name(sceneName);
	obs_set_output_source(0, curScene);
	obs_source_release(curScene);

	obs_data_array_release(sources);
J
jp9000 已提交
500
	obs_data_array_release(sceneOrder);
J
jp9000 已提交
501 502 503 504 505 506 507 508 509

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

510
	obs_data_release(data);
J
jp9000 已提交
511 512

	disableSaving--;
513 514
}

J
jp9000 已提交
515
#define SERVICE_PATH "service.json"
516 517 518 519 520 521

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

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

528 529
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
530

531
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
532
	obs_data_set_obj(data, "settings", settings);
533

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

	obs_data_release(settings);
	obs_data_release(data);
}

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

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

551 552
	obs_data_t *data = obs_data_create_from_json_file_safe(serviceJsonPath,
			"bak");
553 554

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

557
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
558
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
559

560
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
561
			hotkey_data);
562
	obs_service_release(service);
563

P
Palana 已提交
564
	obs_data_release(hotkey_data);
565 566 567 568 569 570 571 572
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
P
Palana 已提交
573 574
	ProfileScope("OBSBasic::InitService");

575 576 577
	if (LoadService())
		return true;

578 579
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
580 581
	if (!service)
		return false;
582
	obs_service_release(service);
583 584 585 586

	return true;
}

587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602
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
};

603 604 605 606 607 608 609 610 611 612 613 614 615 616
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;

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

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

632
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
633

634 635
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
636 637
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
638 639
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
640
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
641 642 643 644
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
645 646 647 648
	config_set_default_string(basicConfig, "SimpleOutput", "RecQuality",
			"Stream");
	config_set_default_string(basicConfig, "SimpleOutput", "RecEncoder",
			SIMPLE_ENCODER_X264);
649

650 651
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
652 653 654 655 656 657 658 659
	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());
660
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
661 662
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
663
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
664 665 666
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

667 668 669 670 671
	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 已提交
672 673 674 675 676 677 678 679 680 681 682
	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);

683 684 685
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

686 687 688 689
	config_set_default_bool  (basicConfig, "Output", "DelayEnable", false);
	config_set_default_uint  (basicConfig, "Output", "DelaySec", 20);
	config_set_default_bool  (basicConfig, "Output", "DelayPreserve", true);

690 691 692 693
	config_set_default_bool  (basicConfig, "Output", "Reconnect", true);
	config_set_default_uint  (basicConfig, "Output", "RetryDelay", 10);
	config_set_default_uint  (basicConfig, "Output", "MaxRetries", 20);

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

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

	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 已提交
730 731
	ProfileScope("OBSBasic::InitBasicConfig");

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

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

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

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

765 766 767
	return InitBasicConfigDefaults();
}

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

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

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

J
jp9000 已提交
791
	obs_enter_graphics();
J
jp9000 已提交
792

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

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

J
jp9000 已提交
808
	obs_leave_graphics();
J
jp9000 已提交
809 810
}

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

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

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

828 829 830
#define MAIN_SEPARATOR \
	"====================================================================="

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

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

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

858
	ret = ResetVideo();
859 860 861 862 863

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
J
jp9000 已提交
864 865 866
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
867 868 869 870 871 872 873
	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";
	}

874
	InitOBSCallbacks();
P
Palana 已提交
875
	InitHotkeys();
876

877
	AddExtraModulePaths();
J
jp9000 已提交
878
	obs_load_all_modules();
J
jp9000 已提交
879

880 881
	blog(LOG_INFO, MAIN_SEPARATOR);

J
jp9000 已提交
882
	ResetOutputs();
883
	CreateHotkeys();
J
jp9000 已提交
884

885 886 887
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
888 889
	InitPrimitives();

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

J
jp9000 已提交
897
	TimedCheckForUpdates();
898
	loaded = true;
J
jp9000 已提交
899

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

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

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

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

931 932 933 934 935 936 937 938 939 940 941 942
#ifdef _WIN32
	show();
#endif

	bool alwaysOnTop = config_get_bool(App()->GlobalConfig(), "BasicWindow",
			"AlwaysOnTop");
	if (alwaysOnTop) {
		SetAlwaysOnTop(this, true);
		ui->actionAlwaysOnTop->setChecked(true);
	}

#ifndef _WIN32
943
	show();
944
#endif
J
jp9000 已提交
945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963

	QList<int> defSizes;

	int top = config_get_int(App()->GlobalConfig(), "BasicWindow",
			"splitterTop");
	int bottom = config_get_int(App()->GlobalConfig(), "BasicWindow",
			"splitterBottom");

	if (!top || !bottom) {
		defSizes = ui->mainSplitter->sizes();
		int total = defSizes[0] + defSizes[1];
		defSizes[0] = total * 75 / 100;
		defSizes[1] = total - defSizes[0];
	} else {
		defSizes.push_back(top);
		defSizes.push_back(bottom);
	}

	ui->mainSplitter->setSizes(defSizes);
964 965
}

P
Palana 已提交
966 967
void OBSBasic::InitHotkeys()
{
P
Palana 已提交
968 969
	ProfileScope("OBSBasic::InitHotkeys");

P
Palana 已提交
970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
	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"),
1012
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032

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

1033 1034
void OBSBasic::CreateHotkeys()
{
P
Palana 已提交
1035 1036
	ProfileScope("OBSBasic::CreateHotkeys");

1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
	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 已提交
1053 1054 1055 1056 1057 1058 1059 1060 1061
	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);
	};

1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
	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 已提交
1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
	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");

1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
	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 已提交
1130 1131 1132 1133
void OBSBasic::ClearHotkeys()
{
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
J
jp9000 已提交
1134
	obs_hotkey_unregister(forceStreamingStopHotkey);
J
jp9000 已提交
1135 1136
}

1137 1138
OBSBasic::~OBSBasic()
{
1139
	bool previewEnabled = obs_display_enabled(ui->preview->GetDisplay());
J
jp9000 已提交
1140

J
jp9000 已提交
1141 1142 1143 1144 1145 1146
	/* 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 已提交
1147
	delete cpuUsageTimer;
1148 1149
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
1150
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
J
jp9000 已提交
1151
	ClearHotkeys();
P
Palana 已提交
1152

1153
	service = nullptr;
J
jp9000 已提交
1154 1155
	outputHandler.reset();

J
John Bradley 已提交
1156 1157 1158
	if (interaction)
		delete interaction;

1159 1160 1161
	if (properties)
		delete properties;

J
jp9000 已提交
1162 1163 1164
	if (filters)
		delete filters;

1165 1166
	if (transformWindow)
		delete transformWindow;
1167

J
jp9000 已提交
1168 1169 1170
	if (advAudioWindow)
		delete advAudioWindow;

1171 1172 1173
	obs_display_remove_draw_callback(ui->preview->GetDisplay(),
			OBSBasic::RenderMain, this);

J
jp9000 已提交
1174
	obs_enter_graphics();
1175 1176
	gs_vertexbuffer_destroy(box);
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
1177
	obs_leave_graphics();
J
jp9000 已提交
1178

1179 1180 1181 1182 1183 1184 1185 1186 1187
	/* 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 已提交
1188 1189
	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
J
jp9000 已提交
1190 1191

	QRect lastGeom = normalGeometry();
J
jp9000 已提交
1192
	QList<int> splitterSizes = ui->mainSplitter->sizes();
1193
	bool alwaysOnTop = IsAlwaysOnTop(this);
J
jp9000 已提交
1194

J
jp9000 已提交
1195
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cx",
J
jp9000 已提交
1196
			lastGeom.width());
J
jp9000 已提交
1197
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cy",
J
jp9000 已提交
1198
			lastGeom.height());
J
jp9000 已提交
1199
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posx",
J
jp9000 已提交
1200
			lastGeom.x());
J
jp9000 已提交
1201
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posy",
J
jp9000 已提交
1202
			lastGeom.y());
J
jp9000 已提交
1203 1204 1205 1206
	config_set_int(App()->GlobalConfig(), "BasicWindow", "splitterTop",
			splitterSizes[0]);
	config_set_int(App()->GlobalConfig(), "BasicWindow", "splitterBottom",
			splitterSizes[1]);
J
jp9000 已提交
1207 1208
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
1209 1210
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "AlwaysOnTop",
			alwaysOnTop);
1211
	config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222

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

J
jp9000 已提交
1225 1226 1227 1228 1229 1230 1231 1232 1233
void OBSBasic::SaveProjectNow()
{
	if (disableSaving)
		return;

	projectChanged = true;
	SaveProjectDeferred();
}

J
jp9000 已提交
1234 1235
void OBSBasic::SaveProject()
{
J
jp9000 已提交
1236 1237 1238
	if (disableSaving)
		return;

J
jp9000 已提交
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253
	projectChanged = true;
	QMetaObject::invokeMethod(this, "SaveProjectDeferred",
			Qt::QueuedConnection);
}

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

	if (!projectChanged)
		return;

	projectChanged = false;

J
jp9000 已提交
1254 1255
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollectionFile");
1256
	char savePath[512];
J
jp9000 已提交
1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268
	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);
1269 1270 1271
	if (ret <= 0)
		return;

J
jp9000 已提交
1272 1273 1274
	Save(savePath);
}

J
jp9000 已提交
1275
OBSScene OBSBasic::GetCurrentScene()
1276
{
J
jp9000 已提交
1277
	QListWidgetItem *item = ui->scenes->currentItem();
P
Palana 已提交
1278
	return item ? GetOBSRef<OBSScene>(item) : nullptr;
1279 1280
}

1281
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
1282
{
P
Palana 已提交
1283
	return item ? GetOBSRef<OBSSceneItem>(item) : nullptr;
J
jp9000 已提交
1284 1285
}

1286 1287
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
1288
	return GetSceneItem(GetTopSelectedSourceItem());
1289 1290
}

1291 1292
void OBSBasic::UpdateSources(OBSScene scene)
{
1293
	ClearListItems(ui->sources);
1294 1295

	obs_scene_enum_items(scene,
1296
			[] (obs_scene_t *scene, obs_sceneitem_t *item, void *p)
1297 1298
			{
				OBSBasic *window = static_cast<OBSBasic*>(p);
1299
				window->InsertSceneItem(item);
J
jp9000 已提交
1300 1301

				UNUSED_PARAMETER(scene);
1302 1303 1304 1305
				return true;
			}, this);
}

1306
void OBSBasic::InsertSceneItem(obs_sceneitem_t *item)
1307
{
1308
	QListWidgetItem *listItem = new QListWidgetItem();
P
Palana 已提交
1309
	SetOBSRef(listItem, OBSSceneItem(item));
1310 1311

	ui->sources->insertItem(0, listItem);
1312
	ui->sources->setCurrentRow(0, QItemSelectionModel::ClearAndSelect);
1313

1314
	SetupVisibilityItem(ui->sources, listItem, item);
1315 1316
}

1317
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
1318 1319 1320 1321 1322 1323 1324 1325 1326
{
	if (interaction)
		interaction->close();

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

1327
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
1328 1329 1330 1331 1332 1333 1334
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
1337 1338 1339 1340 1341 1342 1343 1344 1345 1346
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

1347 1348 1349
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
1350
{
1351
	const char *name  = obs_source_get_name(source);
1352
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
1353 1354

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

P
Palana 已提交
1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369
	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));

1370
	signal_handler_t *handler = obs_source_get_signal_handler(source);
1371

J
jp9000 已提交
1372 1373 1374
	SignalContainer<OBSScene> container;
	container.ref = scene;
	container.handlers.assign({
1375 1376 1377 1378 1379 1380 1381 1382 1383 1384
		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),
J
jp9000 已提交
1385
	});
1386 1387

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

1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406
	/* 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 已提交
1407
	SaveProject();
1408 1409
}

1410
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
1411
{
P
Palana 已提交
1412 1413 1414 1415 1416 1417 1418 1419 1420
	obs_scene_t *scene = obs_scene_from_source(source);

	QListWidgetItem *sel = nullptr;
	int count = ui->scenes->count();
	for (int i = 0; i < count; i++) {
		auto item = ui->scenes->item(i);
		auto cur_scene = GetOBSRef<OBSScene>(item);
		if (cur_scene != scene)
			continue;
J
jp9000 已提交
1421

P
Palana 已提交
1422 1423 1424
		sel = item;
		break;
	}
J
jp9000 已提交
1425

J
jp9000 已提交
1426
	if (sel != nullptr) {
P
Palana 已提交
1427
		if (sel == ui->scenes->currentItem())
1428
			ClearListItems(ui->sources);
J
jp9000 已提交
1429
		delete sel;
J
jp9000 已提交
1430
	}
J
jp9000 已提交
1431 1432

	SaveProject();
1433 1434
}

1435
void OBSBasic::AddSceneItem(OBSSceneItem item)
1436
{
1437
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
J
jp9000 已提交
1438

1439 1440
	if (GetCurrentScene() == scene)
		InsertSceneItem(item);
J
jp9000 已提交
1441

J
jp9000 已提交
1442
	SaveProject();
1443 1444
}

1445
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
1446
{
1447
	obs_scene_t *scene = obs_sceneitem_get_scene(item);
1448

J
jp9000 已提交
1449
	if (GetCurrentScene() == scene) {
B
BtbN 已提交
1450
		for (int i = 0; i < ui->sources->count(); i++) {
J
jp9000 已提交
1451
			QListWidgetItem *listItem = ui->sources->item(i);
J
jp9000 已提交
1452

P
Palana 已提交
1453
			if (GetOBSRef<OBSSceneItem>(listItem) == item) {
1454
				DeleteListItem(ui->sources, listItem);
J
jp9000 已提交
1455 1456
				break;
			}
1457 1458
		}
	}
J
jp9000 已提交
1459

J
jp9000 已提交
1460
	SaveProject();
1461 1462
}

1463
void OBSBasic::UpdateSceneSelection(OBSSource source)
1464 1465
{
	if (source) {
1466
		obs_scene_t *scene = obs_scene_from_source(source);
1467
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
1468

1469 1470 1471
		if (!scene)
			return;

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

1475 1476 1477 1478 1479
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

J
jp9000 已提交
1480
			UpdateSources(scene);
1481
		}
J
jp9000 已提交
1482
	}
1483 1484
}

J
jp9000 已提交
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497
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);
1498 1499 1500 1501 1502

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

	SaveProject();
J
jp9000 已提交
1505 1506
}

1507 1508
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
J
jp9000 已提交
1509 1510
	SignalBlocker sourcesSignalBlocker(ui->sources);

1511
	if (scene != GetCurrentScene() || ignoreSelectionUpdate)
1512 1513 1514 1515
		return;

	for (int i = 0; i < ui->sources->count(); i++) {
		QListWidgetItem *witem = ui->sources->item(i);
P
Palana 已提交
1516 1517
		QVariant data =
			witem->data(static_cast<int>(QtDataRole::OBSRef));
1518 1519 1520 1521 1522 1523
		if (!data.canConvert<OBSSceneItem>())
			continue;

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

J
jp9000 已提交
1524
		witem->setSelected(select);
1525 1526 1527 1528
		break;
	}
}

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 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571
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());
}

1572 1573
void OBSBasic::ActivateAudioSource(OBSSource source)
{
1574 1575 1576 1577
	VolControl *vol = new VolControl(source, true);

	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593

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

1594
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
1595
{
1596
	const char *name  = obs_source_get_name(source);
1597 1598 1599

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

1601
	QMessageBox remove_source(this);
1602 1603 1604
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
1605 1606 1607 1608 1609 1610
	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();
1611
}
J
jp9000 已提交
1612

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

P
Palana 已提交
1615 1616 1617 1618 1619
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
1620 1621
void OBSBasic::TimedCheckForUpdates()
{
P
Palana 已提交
1622 1623 1624 1625
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
#else
J
jp9000 已提交
1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641
	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 已提交
1642
#endif
J
jp9000 已提交
1643 1644 1645 1646
}

void OBSBasic::CheckForUpdates()
{
P
Palana 已提交
1647 1648 1649
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
#else
J
jp9000 已提交
1650 1651
	ui->actionCheckForUpdates->setEnabled(false);

1652 1653 1654 1655
	if (updateCheckThread) {
		updateCheckThread->wait();
		delete updateCheckThread;
	}
1656

1657 1658 1659 1660 1661 1662
	RemoteTextThread *thread = new RemoteTextThread(
			"https://obsproject.com/obs2_update/basic.json");
	updateCheckThread = thread;
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::updateFileFinished);
	updateCheckThread->start();
P
Palana 已提交
1663
#endif
J
jp9000 已提交
1664 1665
}

J
jp9000 已提交
1666 1667
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
1668 1669
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
1670 1671 1672 1673
#else
#define VERSION_ENTRY "other"
#endif

1674
void OBSBasic::updateFileFinished(const QString &text, const QString &error)
J
jp9000 已提交
1675 1676 1677
{
	ui->actionCheckForUpdates->setEnabled(true);

1678 1679
	if (text.isEmpty()) {
		blog(LOG_WARNING, "Update check failed: %s", QT_TO_UTF8(error));
J
jp9000 已提交
1680 1681 1682
		return;
	}

1683
	obs_data_t *returnData  = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
1684
	obs_data_t *versionData = obs_data_get_obj(returnData, VERSION_ENTRY);
J
jp9000 已提交
1685 1686 1687
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
1688 1689

	if (returnData && versionData && description && download) {
J
jp9000 已提交
1690 1691 1692
		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 已提交
1693 1694
		long version = MAKE_SEMANTIC_VERSION(major, minor, patch);

1695 1696
		blog(LOG_INFO, "Update check: last known remote version "
				"is %ld.%ld.%ld",
J
jp9000 已提交
1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
				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);
1717
			config_save_safe(App()->GlobalConfig(), "tmp", nullptr);
J
jp9000 已提交
1718 1719 1720 1721 1722 1723 1724 1725 1726
		}
	} else {
		blog(LOG_WARNING, "Bad JSON file received from server");
	}

	obs_data_release(versionData);
	obs_data_release(returnData);
}

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 1774 1775 1776 1777 1778 1779 1780 1781 1782
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_scene_release(scene);

		obs_set_output_source(0, source);
		return;
	}
}

1783 1784 1785 1786
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1787
		obs_source_t *source = obs_scene_get_source(scene);
1788 1789 1790 1791 1792 1793 1794 1795 1796
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
1797
		obs_source_t *source = obs_sceneitem_get_source(item);
1798
		if (QueryRemoveSource(source))
J
jp9000 已提交
1799 1800 1801 1802
			obs_sceneitem_remove(item);
	}
}

1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816
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 已提交
1817
		OBSSceneItem sceneItem = GetOBSRef<OBSSceneItem>(listItem);
1818 1819 1820 1821 1822

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

1823
				listItem = TakeListItem(ui->sources, i);
1824 1825 1826
				if (listItem)  {
					ui->sources->insertItem(idx_inv,
							listItem);
1827 1828 1829
					SetupVisibilityItem(ui->sources,
							listItem, item);

1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844
					if (sel)
						ui->sources->setCurrentRow(
								idx_inv);
				}
			}

			break;
		}
	}
}

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

1845
	if (scene != GetCurrentScene() || ui->sources->IgnoreReorder())
1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857
		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 已提交
1858 1859

	SaveProject();
1860 1861
}

1862 1863
/* OBS Callbacks */

1864 1865 1866 1867 1868 1869 1870 1871 1872 1873
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)));
}

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

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

1880 1881
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1882 1883
}

1884
void OBSBasic::SceneItemRemoved(void *data, calldata_t *params)
1885
{
1886
	OBSBasic *window = static_cast<OBSBasic*>(data);
1887

1888
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
1889

1890 1891
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1892 1893
}

1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917
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));
}

1918
void OBSBasic::SourceLoaded(void *data, calldata_t *params)
1919
{
J
jp9000 已提交
1920
	OBSBasic *window = static_cast<OBSBasic*>(data);
1921
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1922

1923
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1924
		QMetaObject::invokeMethod(window,
1925 1926
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1927 1928
}

1929
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
1930
{
1931
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1932

1933
	if (obs_scene_from_source(source) != NULL)
1934 1935 1936
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1937 1938
}

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

1950
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
1951
{
1952
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1953 1954 1955 1956 1957 1958 1959 1960
	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)));
}

1961
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
1962 1963 1964 1965 1966 1967 1968 1969 1970 1971
{
	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)));
}

1972
void OBSBasic::ChannelChanged(void *data, calldata_t *params)
1973
{
1974
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1975
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1976 1977

	if (channel == 0)
1978 1979 1980
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1981 1982
}

1983 1984 1985 1986 1987
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1988
	gs_effect_t    *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
1989 1990
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
1991 1992 1993

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1994
	gs_effect_set_vec4(color, &colorVal);
1995

1996 1997
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1998 1999 2000 2001 2002 2003 2004 2005
	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();
2006 2007
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
2008 2009 2010 2011

	gs_load_vertexbuffer(nullptr);
}

2012 2013
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
2014
	OBSBasic *window = static_cast<OBSBasic*>(data);
2015 2016 2017 2018
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
2019 2020
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
2021 2022 2023

	gs_viewport_push();
	gs_projection_push();
2024 2025 2026

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

2027 2028
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
2029
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
2030
			window->previewCX, window->previewCY);
2031

2032 2033
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
2034
	obs_render_main_view();
2035
	gs_load_vertexbuffer(nullptr);
2036

2037 2038
	/* --------------------------------------- */

2039 2040 2041
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
2042 2043 2044 2045

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
2046
	gs_reset_viewport();
J
jp9000 已提交
2047 2048 2049

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

2050 2051
	/* --------------------------------------- */

2052 2053
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
2054 2055 2056

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
2057 2058
}

2059 2060
/* Main class functions */

2061
obs_service_t *OBSBasic::GetService()
2062
{
2063
	if (!service) {
2064 2065
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
2066 2067
		obs_service_release(service);
	}
2068 2069 2070
	return service;
}

2071
void OBSBasic::SetService(obs_service_t *newService)
2072
{
2073
	if (newService)
2074 2075 2076
		service = newService;
}

2077 2078 2079 2080 2081 2082 2083
bool OBSBasic::StreamingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

2084 2085 2086 2087 2088 2089
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

2090 2091
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
2092
	return obs_reset_video(ovi);
2093 2094
}

2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107
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;
}

2108 2109 2110 2111 2112 2113
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 已提交
2114 2115
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
2116 2117 2118 2119 2120 2121 2122 2123 2124
#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 已提交
2125
		return VIDEO_FORMAT_RGBA;
2126 2127
}

2128
int OBSBasic::ResetVideo()
J
jp9000 已提交
2129
{
P
Palana 已提交
2130 2131
	ProfileScope("OBSBasic::ResetVideo");

J
jp9000 已提交
2132
	struct obs_video_info ovi;
2133
	int ret;
J
jp9000 已提交
2134

2135
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
2136

2137 2138 2139 2140 2141 2142 2143
	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 已提交
2144
	ovi.graphics_module = App()->GetRenderModule();
2145
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2146
			"Video", "BaseCX");
2147
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2148
			"Video", "BaseCY");
2149
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2150
			"Video", "OutputCX");
2151
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
2152
			"Video", "OutputCY");
2153 2154 2155 2156 2157
	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 已提交
2158 2159
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
2160
	ovi.scale_type     = GetScaleType(basicConfig);
2161

2162
	ret = AttemptToResetVideo(&ovi);
2163 2164
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
		/* Try OpenGL if DirectX fails on windows */
2165
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
2166 2167 2168 2169
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
2170 2171
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
2172 2173
			ret = AttemptToResetVideo(&ovi);
		}
2174 2175
	} else if (ret == OBS_VIDEO_SUCCESS) {
		ResizePreview(ovi.base_width, ovi.base_height);
2176 2177
	}

2178
	return ret;
J
jp9000 已提交
2179
}
J
jp9000 已提交
2180

2181
bool OBSBasic::ResetAudio()
J
jp9000 已提交
2182
{
P
Palana 已提交
2183 2184
	ProfileScope("OBSBasic::ResetAudio");

2185
	struct obs_audio_info ai;
2186
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
2187 2188
			"SampleRate");

2189
	const char *channelSetupStr = config_get_string(basicConfig,
2190 2191 2192 2193 2194 2195 2196
			"Audio", "ChannelSetup");

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

2197
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
2198 2199

	return obs_reset_audio(&ai);
J
jp9000 已提交
2200 2201
}

2202
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceId,
2203
		const char *deviceDesc, int channel)
J
jp9000 已提交
2204
{
2205 2206
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
2207 2208 2209 2210
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
2211
		settings = obs_source_get_settings(source);
J
jp9000 已提交
2212
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223

		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) {
2224
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
2225
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
2226
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
2227
				sourceId, deviceDesc, settings, nullptr);
J
jp9000 已提交
2228 2229 2230 2231 2232 2233 2234
		obs_data_release(settings);

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

J
jp9000 已提交
2235
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
2236
{
2237
	QSize  targetSize;
J
jp9000 已提交
2238

2239
	/* resize preview panel to fix to the top section of the window */
2240
	targetSize = GetPixelSize(ui->preview);
2241
	GetScaleAndCenterPos(int(cx), int(cy),
2242 2243
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
2244
			previewX, previewY, previewScale);
J
jp9000 已提交
2245

2246 2247
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);
J
jp9000 已提交
2248 2249
}

2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
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 已提交
2267 2268
	disableSaving++;

2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290
	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);

J
jp9000 已提交
2291
	disableSaving--;
2292 2293 2294

	blog(LOG_INFO, "All scene data cleared");
	blog(LOG_INFO, "------------------------------------------------");
2295 2296
}

J
jp9000 已提交
2297
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
2298
{
2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309
	if (outputHandler && outputHandler->Active()) {
		QMessageBox::StandardButton button = QMessageBox::question(
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

2310 2311 2312 2313
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

2314 2315 2316 2317 2318
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
2319 2320
	signalHandlers.clear();

J
jp9000 已提交
2321
	SaveProjectNow();
J
jp9000 已提交
2322
	disableSaving++;
J
jp9000 已提交
2323

2324 2325 2326
	/* Clear all scene data (dialogs, widgets, widget sub-items, scenes,
	 * sources, etc) so that all references are released before shutdown */
	ClearSceneData();
2327 2328
}

J
jp9000 已提交
2329
void OBSBasic::changeEvent(QEvent *event)
2330
{
J
jp9000 已提交
2331 2332
	/* TODO */
	UNUSED_PARAMETER(event);
2333 2334
}

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

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

P
Palana 已提交
2354 2355 2356 2357 2358 2359
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

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

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

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

2375 2376 2377 2378 2379
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

2380 2381 2382 2383 2384
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

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

2390 2391 2392 2393
	if (sceneChanging)
		return;

	if (current) {
2394
		obs_scene_t *scene;
J
jp9000 已提交
2395

P
Palana 已提交
2396
		scene = GetOBSRef<OBSScene>(current);
2397
		source = obs_scene_get_source(scene);
2398 2399
	}

2400
	/* TODO: allow transitions */
2401
	obs_set_output_source(0, source);
2402 2403

	UNUSED_PARAMETER(prev);
2404 2405
}

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

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

J
jp9000 已提交
2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438
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 已提交
2439
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
2440
{
J
jp9000 已提交
2441
	QListWidgetItem *item = ui->scenes->itemAt(pos);
J
jp9000 已提交
2442
	QPointer<QMenu> sceneProjectorMenu;
J
jp9000 已提交
2443

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

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

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

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

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

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

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

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

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

			obs_source_release(source);
J
jp9000 已提交
2519
			on_actionAddScene_triggered();
2520 2521 2522
			return;
		}

2523
		obs_scene_t *scene = obs_scene_create(name.c_str());
2524
		source = obs_scene_get_source(scene);
2525
		AddScene(source);
2526
		obs_set_output_source(0, source);
2527
		obs_scene_release(scene);
2528
	}
2529 2530
}

J
jp9000 已提交
2531
void OBSBasic::on_actionRemoveScene_triggered()
2532
{
2533
	OBSScene     scene  = GetCurrentScene();
2534
	obs_source_t *source = obs_scene_get_source(scene);
2535 2536 2537

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
2538 2539
}

J
jp9000 已提交
2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559
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 已提交
2560
void OBSBasic::on_actionSceneUp_triggered()
2561
{
J
jp9000 已提交
2562
	ChangeSceneIndex(true, -1, 0);
2563 2564
}

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

J
jp9000 已提交
2581
void OBSBasic::on_sources_itemSelectionChanged()
2582
{
J
jp9000 已提交
2583
	SignalBlocker sourcesSignalBlocker(ui->sources);
2584

J
jp9000 已提交
2585
	auto updateItemSelection = [&]()
2586 2587
	{
		ignoreSelectionUpdate = true;
J
jp9000 已提交
2588 2589 2590 2591 2592 2593 2594
		for (int i = 0; i < ui->sources->count(); i++)
		{
			QListWidgetItem *wItem = ui->sources->item(i);
			OBSSceneItem item = GetOBSRef<OBSSceneItem>(wItem);

			obs_sceneitem_select(item, wItem->isSelected());
		}
2595 2596
		ignoreSelectionUpdate = false;
	};
J
jp9000 已提交
2597
	using updateItemSelection_t = decltype(updateItemSelection);
2598 2599

	obs_scene_atomic_update(GetCurrentScene(),
J
jp9000 已提交
2600
			[](void *data, obs_scene_t *)
2601
	{
J
jp9000 已提交
2602 2603
		(*static_cast<updateItemSelection_t*>(data))();
	}, static_cast<void*>(&updateItemSelection));
2604 2605
}

J
jp9000 已提交
2606 2607
void OBSBasic::EditSceneItemName()
{
2608
	QListWidgetItem *item = GetTopSelectedSourceItem();
2609
	Qt::ItemFlags flags   = item->flags();
P
Palana 已提交
2610
	OBSSceneItem sceneItem= GetOBSRef<OBSSceneItem>(item);
2611 2612
	obs_source_t *source  = obs_sceneitem_get_source(sceneItem);
	const char *name      = obs_source_get_name(source);
2613

2614
	item->setText(QT_UTF8(name));
2615
	item->setFlags(flags | Qt::ItemIsEditable);
2616
	ui->sources->removeItemWidget(item);
2617 2618
	ui->sources->editItem(item);
	item->setFlags(flags);
J
jp9000 已提交
2619 2620
}

J
jp9000 已提交
2621
void OBSBasic::CreateSourcePopupMenu(QListWidgetItem *item, bool preview)
2622
{
2623
	QMenu popup(this);
J
jp9000 已提交
2624 2625
	QPointer<QMenu> previewProjector;
	QPointer<QMenu> sourceProjector;
J
jp9000 已提交
2626 2627 2628 2629 2630 2631

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

J
jp9000 已提交
2635 2636 2637 2638 2639 2640
		previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
				SLOT(OpenPreviewProjector()));

		popup.addMenu(previewProjector);

J
jp9000 已提交
2641 2642 2643
		popup.addSeparator();
	}

J
jp9000 已提交
2644 2645 2646 2647 2648 2649 2650 2651
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

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

J
John Bradley 已提交
2652
		OBSSceneItem sceneItem = GetSceneItem(item);
2653
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
John Bradley 已提交
2654 2655
		QAction *action;

J
jp9000 已提交
2656 2657
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
2658 2659
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
2660
				DeleteKeys.front());
J
jp9000 已提交
2661 2662
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
2663
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
2664 2665 2666 2667 2668 2669 2670

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

		popup.addSeparator();
		popup.addMenu(sourceProjector);
J
jp9000 已提交
2671
		popup.addSeparator();
J
John Bradley 已提交
2672 2673 2674 2675 2676 2677 2678

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

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

J
jp9000 已提交
2679 2680
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
2681 2682 2683 2684 2685
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

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

J
jp9000 已提交
2688 2689 2690 2691 2692
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->itemAt(pos), false);
}

P
Palana 已提交
2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704
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 已提交
2705
void OBSBasic::AddSource(const char *id)
2706
{
2707 2708 2709
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
2710 2711
		if (sourceSelect.newSource)
			CreatePropertiesWindow(sourceSelect.newSource);
2712
	}
2713 2714
}

2715
QMenu *OBSBasic::CreateAddSourcePopupMenu()
2716
{
2717
	const char *type;
J
jp9000 已提交
2718 2719
	bool foundValues = false;
	size_t idx = 0;
2720

2721
	QMenu *popup = new QMenu(QTStr("Add"), this);
J
jp9000 已提交
2722
	while (obs_enum_input_types(idx++, &type)) {
2723
		const char *name = obs_source_get_display_name(
2724
				OBS_SOURCE_TYPE_INPUT, type);
2725

2726 2727 2728
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
2729 2730
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
2731 2732 2733
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
2734

J
jp9000 已提交
2735
		foundValues = true;
2736 2737
	}

2738 2739 2740
	if (!foundValues) {
		delete popup;
		popup = nullptr;
2741
	}
2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767

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

J
jp9000 已提交
2770
void OBSBasic::on_actionAddSource_triggered()
2771
{
J
jp9000 已提交
2772
	AddSourcePopupMenu(QCursor::pos());
2773 2774
}

J
jp9000 已提交
2775
void OBSBasic::on_actionRemoveSource_triggered()
2776
{
2777
	OBSSceneItem item   = GetCurrentSceneItem();
2778
	obs_source_t *source = obs_sceneitem_get_source(item);
2779 2780

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
2781
		obs_sceneitem_remove(item);
2782 2783
}

J
John Bradley 已提交
2784 2785 2786 2787 2788 2789 2790 2791 2792
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
2793
void OBSBasic::on_actionSourceProperties_triggered()
2794
{
2795
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2796
	OBSSource source = obs_sceneitem_get_source(item);
2797

2798 2799
	if (source)
		CreatePropertiesWindow(source);
2800 2801
}

J
jp9000 已提交
2802
void OBSBasic::on_actionSourceUp_triggered()
2803
{
J
jp9000 已提交
2804
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2805
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
2806
}
J
jp9000 已提交
2807

J
jp9000 已提交
2808
void OBSBasic::on_actionSourceDown_triggered()
2809
{
J
jp9000 已提交
2810
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2811
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
2812 2813
}

J
jp9000 已提交
2814 2815 2816
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2817
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
2818 2819 2820 2821 2822
}

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

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2829
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
2830 2831 2832 2833 2834
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2835
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
2836 2837
}

2838
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
2839
{
2840
	char logDir[512];
2841
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2842
		return nullptr;
J
jp9000 已提交
2843 2844 2845 2846 2847

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

2848
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
2849 2850 2851 2852 2853 2854 2855 2856
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

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

2859
	if (!fileString)
J
jp9000 已提交
2860 2861
		return;

2862
	if (!*fileString)
J
jp9000 已提交
2863 2864 2865 2866
		return;

	ui->menuLogFiles->setEnabled(false);

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

2870 2871 2872
	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 已提交
2873

2874
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
2875

2876 2877 2878 2879 2880 2881 2882 2883 2884 2885
	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 已提交
2886
	if (!json) {
2887 2888 2889 2890
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

F
fryshorts 已提交
2891 2892 2893
	QBuffer *postData = new QBuffer();
	postData->setData(json, (int) strlen(json));

2894 2895 2896 2897
	if (logUploadThread) {
		logUploadThread->wait();
		delete logUploadThread;
	}
F
fryshorts 已提交
2898

2899 2900 2901 2902 2903 2904 2905
	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 已提交
2906 2907
}

P
Palana 已提交
2908 2909
void OBSBasic::on_actionShowLogs_triggered()
{
2910
	char logDir[512];
2911
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2912 2913
		return;

P
Palana 已提交
2914 2915 2916 2917
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
2918 2919 2920 2921 2922 2923 2924 2925 2926 2927
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

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

2928 2929 2930
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
2931
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943
		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 已提交
2944 2945 2946 2947 2948
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

2949
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
2950 2951 2952
{
	ui->menuLogFiles->setEnabled(true);

2953
	if (text.isEmpty()) {
J
jp9000 已提交
2954 2955
		QMessageBox::information(this,
				QTStr("LogReturnDialog.ErrorUploadingLog"),
2956
				error);
J
jp9000 已提交
2957 2958 2959
		return;
	}

2960
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
2961
	QString logURL         = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
2962 2963 2964 2965 2966 2967
	obs_data_release(returnData);

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

J
jp9000 已提交
2968
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
2969
		obs_source_t *source, const string &name)
2970
{
2971 2972 2973 2974
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

2975
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
2976
	QListWidgetItem *listItem    = listWidget->currentItem();
2977

2978
	if (foundSource || name.empty()) {
2979
		listItem->setText(QT_UTF8(prevName));
2980

2981
		if (foundSource) {
2982 2983 2984 2985 2986 2987 2988 2989 2990
			QMessageBox::information(parent,
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
			QMessageBox::information(parent,
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

2991 2992 2993
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
2994
		obs_source_set_name(source, name.c_str());
2995 2996 2997
	}
}

J
jp9000 已提交
2998 2999 3000 3001 3002
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
3003
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
3004 3005 3006 3007

	if (!scene)
		return;

3008
	obs_source_t *source = obs_scene_get_source(scene);
3009
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
3010 3011 3012 3013 3014 3015 3016 3017 3018

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
3019
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
3020 3021 3022 3023

	if (!item)
		return;

3024
	obs_source_t *source = obs_sceneitem_get_source(item);
3025
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
3026

3027 3028 3029 3030
	QListWidgetItem *listItem = ui->sources->currentItem();
	listItem->setText(QString());
	SetupVisibilityItem(ui->sources, listItem, item);

J
jp9000 已提交
3031 3032 3033
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
3034 3035 3036 3037 3038 3039 3040 3041
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
3042 3043 3044 3045 3046 3047 3048 3049
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
3050 3051 3052 3053 3054 3055 3056 3057 3058
#define RECORDING_START \
	"==== Recording Start ==============================================="
#define RECORDING_STOP \
	"==== Recording Stop ================================================"
#define STREAMING_START \
	"==== Streaming Start ==============================================="
#define STREAMING_STOP \
	"==== Streaming Stop ================================================"

3059 3060 3061 3062
void OBSBasic::StartStreaming()
{
	SaveProject();

J
jp9000 已提交
3063 3064
	ui->streamButton->setEnabled(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
3065

J
jp9000 已提交
3066 3067 3068
	if (!outputHandler->StartStreaming(service)) {
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
3069 3070 3071 3072 3073 3074 3075 3076 3077
	}
}

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

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

3079
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
3080
		ui->profileMenu->setEnabled(true);
3081
		App()->DecrementSleepInhibition();
J
jp9000 已提交
3082
	}
3083 3084
}

J
jp9000 已提交
3085 3086 3087 3088 3089 3090 3091
void OBSBasic::ForceStopStreaming()
{
	SaveProject();

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

3092
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
3093
		ui->profileMenu->setEnabled(true);
3094
		App()->DecrementSleepInhibition();
J
jp9000 已提交
3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113
	}
}

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);
3114 3115 3116 3117 3118

	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
		App()->IncrementSleepInhibition();
	}
J
jp9000 已提交
3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138
}

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

3139
void OBSBasic::StreamingStart()
3140
{
3141
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
3142
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
3143
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
3144 3145 3146 3147 3148 3149

	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
		App()->IncrementSleepInhibition();
	}

3150
	blog(LOG_INFO, STREAMING_START);
3151 3152
}

3153
void OBSBasic::StreamingStop(int code)
3154
{
3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169
	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;

3170
	default:
3171 3172 3173 3174 3175 3176 3177 3178 3179 3180
	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 已提交
3181
	ui->statusbar->StreamStopped();
3182

3183
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
3184
	ui->streamButton->setEnabled(true);
3185

3186
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
3187
		ui->profileMenu->setEnabled(true);
3188 3189
		App()->DecrementSleepInhibition();
	}
3190 3191

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

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

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

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

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

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

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

3220
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
3221
		ui->profileMenu->setEnabled(true);
3222
		App()->DecrementSleepInhibition();
J
jp9000 已提交
3223
	}
3224 3225
}

P
Palana 已提交
3226 3227
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
3228
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
3229
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
3230 3231 3232 3233 3234 3235

	if (ui->profileMenu->isEnabled()) {
		ui->profileMenu->setEnabled(false);
		App()->IncrementSleepInhibition();
	}

3236
	blog(LOG_INFO, RECORDING_START);
P
Palana 已提交
3237 3238
}

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

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

J
jp9000 已提交
3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260
	} else if (code == OBS_OUTPUT_NO_SPACE) {
		QMessageBox::information(this,
				QTStr("Output.RecordNoSpace.Title"),
				QTStr("Output.RecordNoSpace.Msg"));

	} else if (code != OBS_OUTPUT_SUCCESS) {
		QMessageBox::information(this,
				QTStr("Output.RecordError.Title"),
				QTStr("Output.RecordError.Msg"));
	}

3261
	if (!outputHandler->Active() && !ui->profileMenu->isEnabled()) {
J
jp9000 已提交
3262
		ui->profileMenu->setEnabled(true);
3263
		App()->DecrementSleepInhibition();
J
jp9000 已提交
3264
	}
3265
}
3266

3267 3268
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
3269
	if (outputHandler->StreamingActive()) {
3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStoppingStream");

		if (confirm) {
			QMessageBox::StandardButton button =
				QMessageBox::question(this,
						QTStr("ConfirmStop.Title"),
						QTStr("ConfirmStop.Text"));

			if (button == QMessageBox::No)
				return;
		}

3283
		StopStreaming();
3284
	} else {
3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297
		bool confirm = config_get_bool(GetGlobalConfig(), "BasicWindow",
				"WarnBeforeStartingStream");

		if (confirm) {
			QMessageBox::StandardButton button =
				QMessageBox::question(this,
						QTStr("ConfirmStart.Title"),
						QTStr("ConfirmStart.Text"));

			if (button == QMessageBox::No)
				return;
		}

3298
		StartStreaming();
3299 3300 3301 3302 3303
	}
}

void OBSBasic::on_recordButton_clicked()
{
3304 3305 3306 3307
	if (outputHandler->RecordingActive())
		StopRecording();
	else
		StartRecording();
J
jp9000 已提交
3308 3309
}

J
jp9000 已提交
3310
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
3311
{
3312 3313
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
3314
}
3315

3316 3317 3318 3319 3320 3321
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341
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));
}

3342 3343 3344 3345 3346 3347 3348 3349 3350
QListWidgetItem *OBSBasic::GetTopSelectedSourceItem()
{
	QList<QListWidgetItem*> selectedItems = ui->sources->selectedItems();
	QListWidgetItem *topItem = nullptr;
	if (selectedItems.size() != 0)
		topItem = selectedItems[0];
	return topItem;
}

J
jp9000 已提交
3351 3352
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
3353
	CreateSourcePopupMenu(GetTopSelectedSourceItem(), true);
J
jp9000 已提交
3354 3355 3356 3357 3358 3359 3360 3361

	UNUSED_PARAMETER(pos);
}

void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
J
jp9000 已提交
3362
	QPointer<QMenu> previewProjector;
J
jp9000 已提交
3363 3364 3365 3366 3367

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

J
jp9000 已提交
3370 3371 3372 3373 3374
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));

	popup.addMenu(previewProjector);
J
jp9000 已提交
3375 3376 3377 3378 3379
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401
void OBSBasic::on_actionAlwaysOnTop_triggered()
{
	CloseDialogs();

	/* Make sure all dialogs are safely and successfully closed before
	 * switching the always on top mode due to the fact that windows all
	 * have to be recreated, so queue the actual toggle to happen after
	 * all events related to closing the dialogs have finished */
	QMetaObject::invokeMethod(this, "ToggleAlwaysOnTop",
			Qt::QueuedConnection);
}

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

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

	show();
}

3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447 3448 3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464
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);
}

3465
config_t *OBSBasic::Config() const
3466 3467 3468
{
	return basicConfig;
}
J
jp9000 已提交
3469 3470 3471

void OBSBasic::on_actionEditTransform_triggered()
{
3472 3473 3474
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
3475 3476
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
3477
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
3478 3479 3480 3481
}

void OBSBasic::on_actionResetTransform_triggered()
{
3482
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
3483 3484 3485 3486
	{
		if (!obs_sceneitem_selected(item))
			return true;

3487
		obs_transform_info info;
J
jp9000 已提交
3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504
		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);
}

3505
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
3506 3507 3508 3509 3510
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

3513
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
3514 3515 3516 3517
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
3518 3519
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
3520 3521
	};

3522 3523 3524 3525 3526 3527
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

3528
static vec3 GetItemTL(obs_sceneitem_t *item)
3529 3530 3531
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
3532 3533 3534
	return tl;
}

3535
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
3536 3537 3538 3539
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
3540
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
3541 3542 3543
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
3544
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
3545 3546
}

3547
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3548 3549 3550 3551 3552 3553 3554 3555 3556
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
3557
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
3558 3559
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
3560
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586

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

3587
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3588 3589 3590 3591 3592 3593 3594 3595 3596 3597
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
3598
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
3599
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
3600
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
3601 3602

	SetItemTL(item, tl);
J
jp9000 已提交
3603 3604

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
3605 3606 3607 3608 3609
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
3610 3611
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
3612 3613 3614 3615 3616 3617
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
3618 3619
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
3620 3621 3622 3623
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

3624
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3625 3626 3627 3628 3629 3630 3631 3632 3633 3634
		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);

3635
	obs_transform_info itemInfo;
J
jp9000 已提交
3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667
	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()
{
3668
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693
	{
		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 已提交
3694
		UNUSED_PARAMETER(param);
3695 3696 3697 3698
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
J
jp9000 已提交
3699
}
J
jp9000 已提交
3700 3701 3702

void OBSBasic::TogglePreview()
{
3703 3704
	bool enabled = !obs_display_enabled(ui->preview->GetDisplay());
	obs_display_set_enabled(ui->preview->GetDisplay(), enabled);
J
jp9000 已提交
3705 3706 3707
	ui->preview->setVisible(enabled);
	ui->previewDisabledLabel->setVisible(!enabled);
}
3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718

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 已提交
3719
		struct vec2 dir;
3720 3721
		struct vec2 pos;

J
jp9000 已提交
3722 3723
		vec2_set(&dir, 0.0f, 0.0f);

3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746
		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 已提交
3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787

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);
}
3788 3789 3790 3791 3792

void OBSBasic::UpdateTitleBar()
{
	stringstream name;

J
jp9000 已提交
3793 3794
	const char *profile = config_get_string(App()->GlobalConfig(),
			"Basic", "Profile");
J
jp9000 已提交
3795 3796 3797
	const char *sceneCollection = config_get_string(App()->GlobalConfig(),
			"Basic", "SceneCollection");

3798
	name << "OBS " << App()->GetVersionString();
J
jp9000 已提交
3799
	name << " - " << Str("TitleBar.Profile") << ": " << profile;
J
jp9000 已提交
3800
	name << " - " << Str("TitleBar.Scenes") << ": " << sceneCollection;
3801 3802 3803

	setWindowTitle(QT_UTF8(name.str().c_str()));
}
J
jp9000 已提交
3804 3805 3806 3807 3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823 3824 3825 3826 3827

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