window-basic-main.cpp 84.8 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>
J
jp9000 已提交
30
#include <graphics/math-defs.h>
31

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

J
jp9000 已提交
49
#include "ui_OBSBasic.h"
50

J
jp9000 已提交
51
#include <fstream>
52 53
#include <sstream>

54 55 56
#include <QScreen>
#include <QWindow>

57 58
#define PREVIEW_EDGE_SIZE 10

59
using namespace std;
J
jp9000 已提交
60

J
jp9000 已提交
61 62
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
P
Palana 已提交
63
Q_DECLARE_METATYPE(OBSSource);
J
jp9000 已提交
64
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
65

66 67
static void AddExtraModulePaths()
{
68
	char base_module_dir[512];
69
	int ret = GetConfigPath(base_module_dir, sizeof(base_module_dir),
70
			"obs-studio/plugins/%module%");
B
BtbN 已提交
71

72
	if (ret <= 0)
73 74 75 76 77 78
		return;

	string path = (char*)base_module_dir;
	obs_add_module_path((path + "/bin").c_str(), (path + "/data").c_str());
}

79 80
static QList<QKeySequence> DeleteKeys;

81
OBSBasic::OBSBasic(QWidget *parent)
82
	: OBSMainWindow  (parent),
J
jp9000 已提交
83
	  ui             (new Ui::OBSBasic)
84 85
{
	ui->setupUi(this);
J
jp9000 已提交
86 87
	ui->previewDisabledLabel->setVisible(false);

S
Socapex 已提交
88
	copyActionsDynamicProperties();
89

90 91
	ui->sources->setItemDelegate(new VisibilityItemDelegate(ui->sources));

J
jp9000 已提交
92
	int width = config_get_int(App()->GlobalConfig(), "BasicWindow", "cx");
93 94 95

	// Check if no values are saved (new installation).
	if (width != 0) {
J
jp9000 已提交
96 97 98
		int height = config_get_int(App()->GlobalConfig(),
				"BasicWindow", "cy");
		int posx = config_get_int(App()->GlobalConfig(), "BasicWindow",
99
				"posx");
J
jp9000 已提交
100
		int posy = config_get_int(App()->GlobalConfig(), "BasicWindow",
101 102
				"posy");

103
		setGeometry(posx, posy, width, height);
104 105
	}

106
	char styleSheetPath[512];
107
	int ret = GetConfigPath(styleSheetPath, sizeof(styleSheetPath),
108 109
			"obs-studio/basic/stylesheet.qss");
	if (ret > 0) {
H
HomeWorld 已提交
110 111 112 113 114
		if (QFile::exists(styleSheetPath)) {
			QString path = QString("file:///") +
				QT_UTF8(styleSheetPath);
			App()->setStyleSheet(path);
		}
115 116
	}

P
Palana 已提交
117 118 119
	qRegisterMetaType<OBSScene>    ("OBSScene");
	qRegisterMetaType<OBSSceneItem>("OBSSceneItem");
	qRegisterMetaType<OBSSource>   ("OBSSource");
P
Palana 已提交
120
	qRegisterMetaType<obs_hotkey_id>("obs_hotkey_id");
P
Palana 已提交
121

P
Palana 已提交
122 123 124
	ui->scenes->setAttribute(Qt::WA_MacShowFocusRect, false);
	ui->sources->setAttribute(Qt::WA_MacShowFocusRect, false);

125 126 127 128 129 130
	connect(windowHandle(), &QWindow::screenChanged, [this]() {
		struct obs_video_info ovi;

		if (obs_get_video_info(&ovi))
			ResizePreview(ovi.base_width, ovi.base_height);
	});
J
jp9000 已提交
131 132 133 134

	stringstream name;
	name << "OBS " << App()->GetVersionString();

P
Palana 已提交
135 136
	installEventFilter(CreateShortcutFilter());

J
jp9000 已提交
137
	blog(LOG_INFO, "%s", name.str().c_str());
J
jp9000 已提交
138
	setWindowTitle(QT_UTF8(name.str().c_str()));
J
jp9000 已提交
139 140 141 142 143 144 145 146 147 148 149 150 151 152

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

	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
155 156 157 158
	cpuUsageTimer = new QTimer(this);
	connect(cpuUsageTimer, SIGNAL(timeout()),
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
159

160 161 162 163 164 165
	DeleteKeys =
#ifdef __APPLE__
		QList<QKeySequence>{{Qt::Key_Backspace}} <<
#endif
		QKeySequence::keyBindings(QKeySequence::Delete);

166
#ifdef __APPLE__
167 168
	ui->actionRemoveSource->setShortcuts(DeleteKeys);
	ui->actionRemoveScene->setShortcuts(DeleteKeys);
169 170 171

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
172
#endif
173 174 175 176 177 178 179 180 181 182 183 184 185 186

	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()));
187 188
}

189
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent)
190
{
191
	obs_source_t *source = obs_get_output_source(channel);
192 193 194
	if (!source)
		return;

195
	obs_data_t *data = obs_save_source(source);
196

J
jp9000 已提交
197
	obs_data_set_obj(parent, name, data);
198 199 200 201 202

	obs_data_release(data);
	obs_source_release(source);
}

J
jp9000 已提交
203
static obs_data_t *GenerateSaveData(obs_data_array_t *sceneOrder)
204
{
205 206 207
	obs_data_t       *saveData     = obs_data_create();
	obs_data_array_t *sourcesArray = obs_save_sources();
	obs_source_t     *currentScene = obs_get_output_source(0);
208
	const char       *sceneName   = obs_source_get_name(currentScene);
209

210 211 212 213 214 215
	SaveAudioDevice(DESKTOP_AUDIO_1, 1, saveData);
	SaveAudioDevice(DESKTOP_AUDIO_2, 2, saveData);
	SaveAudioDevice(AUX_AUDIO_1,     3, saveData);
	SaveAudioDevice(AUX_AUDIO_2,     4, saveData);
	SaveAudioDevice(AUX_AUDIO_3,     5, saveData);

J
jp9000 已提交
216
	obs_data_set_string(saveData, "current_scene", sceneName);
J
jp9000 已提交
217
	obs_data_set_array(saveData, "scene_order", sceneOrder);
J
jp9000 已提交
218
	obs_data_set_array(saveData, "sources", sourcesArray);
219 220 221 222 223 224
	obs_data_array_release(sourcesArray);
	obs_source_release(currentScene);

	return saveData;
}

S
Socapex 已提交
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
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));
		}
	}
}

245 246 247 248 249 250 251 252 253 254 255 256
void OBSBasic::ClearVolumeControls()
{
	VolControl *control;

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

	volumes.clear();
}

J
jp9000 已提交
257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
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;
}

272 273
void OBSBasic::Save(const char *file)
{
J
jp9000 已提交
274 275
	obs_data_array_t *sceneOrder = SaveSceneListOrder();
	obs_data_t *saveData  = GenerateSaveData(sceneOrder);
J
jp9000 已提交
276
	const char *jsonData = obs_data_get_json(saveData);
277

S
Socapex 已提交
278 279 280 281 282 283 284 285
	if (!!jsonData) {
		/* TODO: maybe a message box here? */
		bool success = os_quick_write_utf8_file(file, jsonData,
				strlen(jsonData), false);
		if (!success)
			blog(LOG_ERROR, "Could not save scene data to %s",
					file);
	}
286 287

	obs_data_release(saveData);
J
jp9000 已提交
288
	obs_data_array_release(sceneOrder);
289 290
}

291
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
292
{
293
	obs_data_t *data = obs_data_get_obj(parent, name);
294 295 296
	if (!data)
		return;

297
	obs_source_t *source = obs_load_source(data);
298 299 300 301 302 303 304 305 306 307
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

void OBSBasic::CreateDefaultScene()
{
308 309
	obs_scene_t  *scene  = obs_scene_create(Str("Basic.Scene"));
	obs_source_t *source = obs_scene_get_source(scene);
310 311 312 313 314

	obs_add_source(source);

#ifdef __APPLE__
	source = obs_source_create(OBS_SOURCE_TYPE_INPUT, "display_capture",
315
			Str("Basic.DisplayCapture"), NULL, nullptr);
316 317 318 319 320 321 322 323

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

324
	obs_set_output_source(0, obs_scene_get_source(scene));
325 326 327
	obs_scene_release(scene);
}

J
jp9000 已提交
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356
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);
	}
}

357 358 359 360 361 362 363 364
void OBSBasic::Load(const char *file)
{
	if (!file) {
		blog(LOG_ERROR, "Could not find file %s", file);
		return;
	}

	BPtr<char> jsonData = os_quick_read_utf8_file(file);
365 366
	if (!jsonData) {
		CreateDefaultScene();
367
		return;
368
	}
369

370
	obs_data_t       *data       = obs_data_create_from_json(jsonData);
J
jp9000 已提交
371
	obs_data_array_t *sceneOrder = obs_data_get_array(data, "scene_order");
372
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
J
jp9000 已提交
373 374
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
375
	obs_source_t     *curScene;
376

377 378 379 380 381 382
	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);

383 384
	obs_load_sources(sources);

J
jp9000 已提交
385 386 387
	if (sceneOrder)
		LoadSceneListOrder(sceneOrder);

388 389 390 391 392
	curScene = obs_get_source_by_name(sceneName);
	obs_set_output_source(0, curScene);
	obs_source_release(curScene);

	obs_data_array_release(sources);
J
jp9000 已提交
393
	obs_data_array_release(sceneOrder);
394 395 396
	obs_data_release(data);
}

397 398 399
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
400
	obs_properties_t *props = obs_get_source_properties(
401
			OBS_SOURCE_TYPE_INPUT, output_id);
402 403 404 405 406
	size_t count = 0;

	if (!props)
		return false;

407
	obs_property_t *devices = obs_properties_get(props, "device_id");
408 409 410 411 412 413 414 415
	if (devices)
		count = obs_property_list_item_count(devices);

	obs_properties_destroy(props);

	return count != 0;
}

416 417 418 419 420 421 422
#define SERVICE_PATH "obs-studio/basic/service.json"

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

423
	char serviceJsonPath[512];
424
	int ret = GetConfigPath(serviceJsonPath, sizeof(serviceJsonPath),
425 426
			SERVICE_PATH);
	if (ret <= 0)
427 428
		return;

429 430
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
431

432
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
433
	obs_data_set_obj(data, "settings", settings);
434

J
jp9000 已提交
435
	const char *json = obs_data_get_json(data);
436 437 438 439 440 441 442 443 444 445 446

	os_quick_write_utf8_file(serviceJsonPath, json, strlen(json), false);

	obs_data_release(settings);
	obs_data_release(data);
}

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

447
	char serviceJsonPath[512];
448
	int ret = GetConfigPath(serviceJsonPath, sizeof(serviceJsonPath),
449 450
			SERVICE_PATH);
	if (ret <= 0)
451 452 453 454 455 456
		return false;

	BPtr<char> jsonText = os_quick_read_utf8_file(serviceJsonPath);
	if (!jsonText)
		return false;

457
	obs_data_t *data = obs_data_create_from_json(jsonText);
458 459

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

462
	obs_data_t *settings = obs_data_get_obj(data, "settings");
P
Palana 已提交
463
	obs_data_t *hotkey_data = obs_data_get_obj(data, "hotkeys");
464

465
	service = obs_service_create(type, "default_service", settings,
P
Palana 已提交
466
			hotkey_data);
467
	obs_service_release(service);
468

P
Palana 已提交
469
	obs_data_release(hotkey_data);
470 471 472 473 474 475 476 477 478 479 480
	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitService()
{
	if (LoadService())
		return true;

481 482
	service = obs_service_create("rtmp_common", "default_service", nullptr,
			nullptr);
483 484
	if (!service)
		return false;
485
	obs_service_release(service);
486 487 488 489

	return true;
}

490 491
bool OBSBasic::InitBasicConfigDefaults()
{
492 493 494
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

495 496 497 498 499 500 501 502 503 504 505 506
	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;

507
	config_set_default_string(basicConfig, "Output", "Mode", "Simple");
J
jp9000 已提交
508

509 510
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
511 512
	config_set_default_string(basicConfig, "SimpleOutput", "RecFormat",
			"flv");
513 514
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
515
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 160);
J
jp9000 已提交
516 517
	config_set_default_bool  (basicConfig, "SimpleOutput", "Reconnect",
			true);
518 519
	config_set_default_uint  (basicConfig, "SimpleOutput", "RetryDelay",
			10);
J
jp9000 已提交
520 521
	config_set_default_uint  (basicConfig, "SimpleOutput", "MaxRetries",
			20);
J
jp9000 已提交
522 523
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
524
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseCBR", true);
525 526 527
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseBufsize",
			false);
	config_set_default_int   (basicConfig, "SimpleOutput", "Bufsize", 2500);
J
jp9000 已提交
528 529
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
530

J
jp9000 已提交
531
	config_set_default_bool  (basicConfig, "AdvOut", "Reconnect", true);
532
	config_set_default_uint  (basicConfig, "AdvOut", "RetryDelay", 10);
J
jp9000 已提交
533
	config_set_default_uint  (basicConfig, "AdvOut", "MaxRetries", 20);
534 535
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
536 537 538 539 540 541 542 543 544 545
	config_set_default_bool  (basicConfig, "AdvOut", "UseRescale", false);
	config_set_default_bool  (basicConfig, "AdvOut", "Multitrack", false);
	config_set_default_uint  (basicConfig, "AdvOut", "TrackIndex", 1);
	config_set_default_uint  (basicConfig, "AdvOut", "TrackCount", 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());
546
	config_set_default_string(basicConfig, "AdvOut", "RecFormat", "flv");
J
jp9000 已提交
547 548 549 550
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
	config_set_default_bool  (basicConfig, "AdvOut", "RecMultitrack",
			false);
551
	config_set_default_uint  (basicConfig, "AdvOut", "RecTracks", (1<<0));
J
jp9000 已提交
552 553 554 555 556 557 558 559 560 561 562 563 564 565
	config_set_default_string(basicConfig, "AdvOut", "RecEncoder",
			"none");

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

566 567 568 569 570 571 572 573 574 575 576 577 578
	config_set_default_uint  (basicConfig, "Video", "BaseCX",   cx);
	config_set_default_uint  (basicConfig, "Video", "BaseCY",   cy);

	cx = cx * 10 / 15;
	cy = cy * 10 / 15;
	config_set_default_uint  (basicConfig, "Video", "OutputCX", cx);
	config_set_default_uint  (basicConfig, "Video", "OutputCY", cy);

	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);
579
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
580 581 582 583
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
	config_set_default_string(basicConfig, "Video", "ColorSpace", "709");
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
584 585 586 587 588 589

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

J
jp9000 已提交
590
	config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
591
			hasDesktopAudio ? "default" : "disabled");
J
jp9000 已提交
592 593 594
	config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice1",
595
			hasInputAudio ? "default" : "disabled");
J
jp9000 已提交
596 597 598 599 600
	config_set_default_string(basicConfig, "Audio", "AuxDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice3",
			"disabled");

601 602 603 604 605
	return true;
}

bool OBSBasic::InitBasicConfig()
{
606
	char configPath[512];
607
	int ret = GetConfigPath(configPath, sizeof(configPath),
608 609 610 611 612
			"obs-studio/basic/basic.ini");
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
613

614 615 616
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
617 618 619 620 621 622
		return false;
	}

	return InitBasicConfigDefaults();
}

623 624
void OBSBasic::InitOBSCallbacks()
{
P
Palana 已提交
625 626
	signalHandlers.reserve(signalHandlers.size() + 6);
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_add",
627
			OBSBasic::SourceAdded, this);
P
Palana 已提交
628
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_remove",
629
			OBSBasic::SourceRemoved, this);
P
Palana 已提交
630
	signalHandlers.emplace_back(obs_get_signal_handler(), "channel_change",
631
			OBSBasic::ChannelChanged, this);
P
Palana 已提交
632
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_activate",
633
			OBSBasic::SourceActivated, this);
P
Palana 已提交
634
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_deactivate",
635
			OBSBasic::SourceDeactivated, this);
P
Palana 已提交
636
	signalHandlers.emplace_back(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
637
			OBSBasic::SourceRenamed, this);
638 639
}

J
jp9000 已提交
640 641
void OBSBasic::InitPrimitives()
{
J
jp9000 已提交
642
	obs_enter_graphics();
J
jp9000 已提交
643

644
	gs_render_start(true);
J
jp9000 已提交
645 646 647 648 649
	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);
650
	box = gs_render_save();
J
jp9000 已提交
651

652
	gs_render_start(true);
J
jp9000 已提交
653 654 655 656
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
657
	circle = gs_render_save();
J
jp9000 已提交
658

J
jp9000 已提交
659
	obs_leave_graphics();
J
jp9000 已提交
660 661
}

J
jp9000 已提交
662 663
void OBSBasic::ResetOutputs()
{
J
jp9000 已提交
664 665 666
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	bool advOut = astrcmpi(mode, "Advanced") == 0;

J
jp9000 已提交
667 668
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
669 670 671
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
J
jp9000 已提交
672 673 674 675 676
	} else {
		outputHandler->Update();
	}
}

677 678
void OBSBasic::OBSInit()
{
679
	char savePath[512];
680
	int ret = GetConfigPath(savePath, sizeof(savePath),
681 682 683
			"obs-studio/basic/scenes.json");
	if (ret <= 0)
		throw "Failed to get scenes.json file path";
684

685 686 687 688
	/* make sure it's fully displayed before doing any initialization */
	show();
	App()->processEvents();

689
	if (!obs_startup(App()->GetLocale()))
690
		throw "Failed to initialize libobs";
691 692
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
693
	if (!ResetAudio())
694 695
		throw "Failed to initialize audio";

696
	ret = ResetVideo();
697 698 699 700 701

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
J
jp9000 已提交
702 703 704
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
705 706 707 708 709 710 711
	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";
	}

712
	InitOBSCallbacks();
P
Palana 已提交
713
	InitHotkeys();
714

715
	AddExtraModulePaths();
J
jp9000 已提交
716
	obs_load_all_modules();
J
jp9000 已提交
717

J
jp9000 已提交
718
	ResetOutputs();
719
	CreateHotkeys();
J
jp9000 已提交
720

721 722 723
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
724 725
	InitPrimitives();

726
	Load(savePath);
J
jp9000 已提交
727
	ResetAudioDevices();
728

J
jp9000 已提交
729
	TimedCheckForUpdates();
730
	loaded = true;
J
jp9000 已提交
731

732 733 734
	saveTimer = new QTimer(this);
	connect(saveTimer, SIGNAL(timeout()), this, SLOT(SaveProject()));
	saveTimer->start(20000);
J
jp9000 已提交
735 736 737 738 739 740

	bool previewEnabled = config_get_bool(App()->GlobalConfig(),
			"BasicWindow", "PreviewEnabled");
	if (!previewEnabled)
		QMetaObject::invokeMethod(this, "TogglePreview",
				Qt::QueuedConnection);
741 742 743 744 745 746 747 748 749

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

P
Palana 已提交
752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795
void OBSBasic::InitHotkeys()
{
	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"),
796
			Str("Push-to-mute"), Str("Push-to-talk"));
P
Palana 已提交
797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816

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

817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887
void OBSBasic::CreateHotkeys()
{
	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;
	};

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

	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
}

888 889
OBSBasic::~OBSBasic()
{
J
jp9000 已提交
890 891
	bool previewEnabled = obs_preview_enabled();

J
jp9000 已提交
892 893 894 895 896 897
	/* 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 已提交
898
	delete cpuUsageTimer;
899 900
	os_cpu_usage_info_destroy(cpuUsageInfo);

P
Palana 已提交
901
	obs_hotkey_set_callback_routing_func(nullptr, nullptr);
902 903
	obs_hotkey_pair_unregister(streamingHotkeys);
	obs_hotkey_pair_unregister(recordingHotkeys);
P
Palana 已提交
904

905
	service = nullptr;
J
jp9000 已提交
906 907
	outputHandler.reset();

J
John Bradley 已提交
908 909 910
	if (interaction)
		delete interaction;

911 912 913
	if (properties)
		delete properties;

J
jp9000 已提交
914 915 916
	if (filters)
		delete filters;

917 918
	if (transformWindow)
		delete transformWindow;
919

J
jp9000 已提交
920 921 922
	if (advAudioWindow)
		delete advAudioWindow;

J
jp9000 已提交
923
	obs_enter_graphics();
924 925
	gs_vertexbuffer_destroy(box);
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
926
	obs_leave_graphics();
J
jp9000 已提交
927

928 929 930 931 932 933 934 935 936
	/* 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);

937
	obs_shutdown();
J
jp9000 已提交
938 939 940

	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
J
jp9000 已提交
941 942 943

	QRect lastGeom = normalGeometry();

J
jp9000 已提交
944
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cx",
J
jp9000 已提交
945
			lastGeom.width());
J
jp9000 已提交
946
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cy",
J
jp9000 已提交
947
			lastGeom.height());
J
jp9000 已提交
948
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posx",
J
jp9000 已提交
949
			lastGeom.x());
J
jp9000 已提交
950
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posy",
J
jp9000 已提交
951
			lastGeom.y());
J
jp9000 已提交
952 953
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
J
jp9000 已提交
954
	config_save(App()->GlobalConfig());
955 956 957 958 959 960 961 962 963 964 965

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

J
jp9000 已提交
968 969
void OBSBasic::SaveProject()
{
970
	char savePath[512];
971
	int ret = GetConfigPath(savePath, sizeof(savePath),
972 973 974 975
			"obs-studio/basic/scenes.json");
	if (ret <= 0)
		return;

J
jp9000 已提交
976 977 978
	Save(savePath);
}

J
jp9000 已提交
979
OBSScene OBSBasic::GetCurrentScene()
980
{
J
jp9000 已提交
981
	QListWidgetItem *item = ui->scenes->currentItem();
J
jp9000 已提交
982
	return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
983 984
}

985
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
986
{
J
jp9000 已提交
987
	return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
J
jp9000 已提交
988 989
}

990 991 992 993 994
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
	return GetSceneItem(ui->sources->currentItem());
}

995 996 997 998 999
void OBSBasic::UpdateSources(OBSScene scene)
{
	ui->sources->clear();

	obs_scene_enum_items(scene,
1000
			[] (obs_scene_t *scene, obs_sceneitem_t *item, void *p)
1001 1002
			{
				OBSBasic *window = static_cast<OBSBasic*>(p);
1003
				window->InsertSceneItem(item);
J
jp9000 已提交
1004 1005

				UNUSED_PARAMETER(scene);
1006 1007 1008 1009
				return true;
			}, this);
}

1010
void OBSBasic::InsertSceneItem(obs_sceneitem_t *item)
1011
{
1012
	obs_source_t *source = obs_sceneitem_get_source(item);
1013

1014
	QListWidgetItem *listItem = new QListWidgetItem();
1015 1016 1017 1018
	listItem->setData(Qt::UserRole,
			QVariant::fromValue(OBSSceneItem(item)));

	ui->sources->insertItem(0, listItem);
1019 1020
	ui->sources->setCurrentRow(0);

1021 1022
	SetupVisibilityItem(ui->sources, listItem, item);

1023
	/* if the source was just created, open properties dialog */
1024 1025 1026 1027
	if (sourceSceneRefs[source] == 0 && loaded)
		CreatePropertiesWindow(source);
}

1028
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
1029 1030 1031 1032 1033 1034 1035 1036 1037
{
	if (interaction)
		interaction->close();

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

1038
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
1039 1040 1041 1042 1043 1044 1045
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

1058 1059 1060
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
1061
{
1062
	const char *name  = obs_source_get_name(source);
1063
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
1064 1065

	QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
J
jp9000 已提交
1066
	item->setData(Qt::UserRole, QVariant::fromValue(OBSScene(scene)));
J
jp9000 已提交
1067
	ui->scenes->addItem(item);
1068

P
Palana 已提交
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
	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));

1081
	signal_handler_t *handler = obs_source_get_signal_handler(source);
1082
	signal_handler_connect(handler, "item_add",
J
jp9000 已提交
1083
			OBSBasic::SceneItemAdded, this);
1084
	signal_handler_connect(handler, "item_remove",
J
jp9000 已提交
1085
			OBSBasic::SceneItemRemoved, this);
1086 1087 1088
	signal_handler_connect(handler, "item_select",
			OBSBasic::SceneItemSelected, this);
	signal_handler_connect(handler, "item_deselect",
J
jp9000 已提交
1089
			OBSBasic::SceneItemDeselected, this);
1090 1091
	signal_handler_connect(handler, "reorder",
			OBSBasic::SceneReordered, this);
1092 1093
}

1094
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
1095
{
1096
	const char *name = obs_source_get_name(source);
J
jp9000 已提交
1097

J
jp9000 已提交
1098 1099 1100
	QListWidgetItem *sel = ui->scenes->currentItem();
	QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
			Qt::MatchExactly);
J
jp9000 已提交
1101

J
jp9000 已提交
1102 1103 1104 1105
	if (sel != nullptr) {
		if (items.contains(sel))
			ui->sources->clear();
		delete sel;
J
jp9000 已提交
1106
	}
1107 1108
}

1109
void OBSBasic::AddSceneItem(OBSSceneItem item)
1110
{
1111 1112
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
1113

1114 1115
	if (GetCurrentScene() == scene)
		InsertSceneItem(item);
J
jp9000 已提交
1116 1117

	sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
1118 1119
}

1120
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
1121
{
1122
	obs_scene_t *scene = obs_sceneitem_get_scene(item);
1123

J
jp9000 已提交
1124
	if (GetCurrentScene() == scene) {
B
BtbN 已提交
1125
		for (int i = 0; i < ui->sources->count(); i++) {
J
jp9000 已提交
1126 1127
			QListWidgetItem *listItem = ui->sources->item(i);
			QVariant userData = listItem->data(Qt::UserRole);
J
jp9000 已提交
1128

J
jp9000 已提交
1129
			if (userData.value<OBSSceneItem>() == item) {
J
jp9000 已提交
1130
				delete listItem;
J
jp9000 已提交
1131 1132
				break;
			}
1133 1134
		}
	}
J
jp9000 已提交
1135

1136
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
1137 1138

	int scenes = sourceSceneRefs[source] - 1;
1139 1140
	sourceSceneRefs[source] = scenes;

J
jp9000 已提交
1141 1142 1143 1144
	if (scenes == 0) {
		obs_source_remove(source);
		sourceSceneRefs.erase(source);
	}
1145 1146
}

1147
void OBSBasic::UpdateSceneSelection(OBSSource source)
1148 1149
{
	if (source) {
1150
		obs_scene_t *scene = obs_scene_from_source(source);
1151
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
1152

1153 1154 1155
		if (!scene)
			return;

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

1159 1160 1161 1162 1163
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

J
jp9000 已提交
1164
			UpdateSources(scene);
1165
		}
J
jp9000 已提交
1166
	}
1167 1168
}

J
jp9000 已提交
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
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);
1182 1183 1184 1185 1186

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

1189 1190
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
1191
	if (scene != GetCurrentScene())
1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202
		return;

	for (int i = 0; i < ui->sources->count(); i++) {
		QListWidgetItem *witem = ui->sources->item(i);
		QVariant data = witem->data(Qt::UserRole);
		if (!data.canConvert<OBSSceneItem>())
			continue;

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

1203 1204 1205 1206 1207 1208
		if (select) {
			ui->sources->setCurrentItem(witem);
		} else if (ui->sources->currentItem() == witem) {
			ui->sources->setCurrentItem(nullptr);
		}

1209 1210 1211 1212
		break;
	}
}

J
jp9000 已提交
1213
void OBSBasic::MoveSceneItem(OBSSceneItem item, obs_order_movement movement)
J
jp9000 已提交
1214
{
J
jp9000 已提交
1215
	OBSScene scene = obs_sceneitem_get_scene(item);
J
jp9000 已提交
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
	if (scene != GetCurrentScene())
		return;

	int curRow = ui->sources->currentRow();
	if (curRow == -1)
		return;

	QListWidgetItem *listItem = ui->sources->takeItem(curRow);

	switch (movement) {
J
jp9000 已提交
1226
	case OBS_ORDER_MOVE_UP:
J
jp9000 已提交
1227 1228 1229 1230
		if (curRow > 0)
			curRow--;
		break;

J
jp9000 已提交
1231
	case OBS_ORDER_MOVE_DOWN:
J
jp9000 已提交
1232 1233 1234 1235
		if (curRow < ui->sources->count())
			curRow++;
		break;

J
jp9000 已提交
1236
	case OBS_ORDER_MOVE_TOP:
J
jp9000 已提交
1237 1238 1239
		curRow = 0;
		break;

J
jp9000 已提交
1240
	case OBS_ORDER_MOVE_BOTTOM:
J
jp9000 已提交
1241 1242 1243 1244 1245 1246 1247 1248
		curRow = ui->sources->count();
		break;
	}

	ui->sources->insertItem(curRow, listItem);
	ui->sources->setCurrentRow(curRow);
}

1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291
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());
}

1292 1293
void OBSBasic::ActivateAudioSource(OBSSource source)
{
1294 1295 1296 1297
	VolControl *vol = new VolControl(source, true);

	connect(vol, &VolControl::ConfigClicked,
			this, &OBSBasic::VolControlContextMenu);
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313

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

1314
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
1315
{
1316
	const char *name  = obs_source_get_name(source);
1317 1318 1319

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

1321
	QMessageBox remove_source(this);
1322 1323 1324
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
1325 1326 1327 1328 1329 1330
	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();
1331
}
J
jp9000 已提交
1332

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

P
Palana 已提交
1335 1336 1337 1338 1339
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
1340 1341
void OBSBasic::TimedCheckForUpdates()
{
P
Palana 已提交
1342 1343 1344 1345
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
#else
J
jp9000 已提交
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361
	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 已提交
1362
#endif
J
jp9000 已提交
1363 1364 1365 1366
}

void OBSBasic::CheckForUpdates()
{
P
Palana 已提交
1367 1368 1369
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
#else
J
jp9000 已提交
1370 1371
	ui->actionCheckForUpdates->setEnabled(false);

1372 1373 1374 1375
	if (updateCheckThread) {
		updateCheckThread->wait();
		delete updateCheckThread;
	}
1376

1377 1378 1379 1380 1381 1382
	RemoteTextThread *thread = new RemoteTextThread(
			"https://obsproject.com/obs2_update/basic.json");
	updateCheckThread = thread;
	connect(thread, &RemoteTextThread::Result,
			this, &OBSBasic::updateFileFinished);
	updateCheckThread->start();
P
Palana 已提交
1383
#endif
J
jp9000 已提交
1384 1385
}

J
jp9000 已提交
1386 1387
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
1388 1389
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
1390 1391 1392 1393
#else
#define VERSION_ENTRY "other"
#endif

1394
void OBSBasic::updateFileFinished(const QString &text, const QString &error)
J
jp9000 已提交
1395 1396 1397
{
	ui->actionCheckForUpdates->setEnabled(true);

1398 1399
	if (text.isEmpty()) {
		blog(LOG_WARNING, "Update check failed: %s", QT_TO_UTF8(error));
J
jp9000 已提交
1400 1401 1402
		return;
	}

1403
	obs_data_t *returnData  = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
1404
	obs_data_t *versionData = obs_data_get_obj(returnData, VERSION_ENTRY);
J
jp9000 已提交
1405 1406 1407
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
1408 1409

	if (returnData && versionData && description && download) {
J
jp9000 已提交
1410 1411 1412
		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 已提交
1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445
		long version = MAKE_SEMANTIC_VERSION(major, minor, patch);

		blog(LOG_INFO, "Update check: latest version is: %ld.%ld.%ld",
				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);
			config_save(App()->GlobalConfig());
		}
	} else {
		blog(LOG_WARNING, "Bad JSON file received from server");
	}

	obs_data_release(versionData);
	obs_data_release(returnData);
}

1446 1447 1448 1449
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1450
		obs_source_t *source = obs_scene_get_source(scene);
1451 1452 1453 1454 1455 1456 1457 1458 1459
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
1460
		obs_source_t *source = obs_sceneitem_get_source(item);
1461
		if (QueryRemoveSource(source))
J
jp9000 已提交
1462 1463 1464 1465
			obs_sceneitem_remove(item);
	}
}

1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
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);
		QVariant v = listItem->data(Qt::UserRole);
		OBSSceneItem sceneItem = v.value<OBSSceneItem>();

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

				listItem = ui->sources->takeItem(i);
				if (listItem)  {
					ui->sources->insertItem(idx_inv,
							listItem);
1491 1492 1493
					SetupVisibilityItem(ui->sources,
							listItem, item);

1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523
					if (sel)
						ui->sources->setCurrentRow(
								idx_inv);
				}
			}

			break;
		}
	}
}

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

	if (scene != GetCurrentScene())
		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);
}

1524 1525
/* OBS Callbacks */

1526 1527 1528 1529 1530 1531 1532 1533 1534 1535
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)));
}

1536
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
1537 1538 1539
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

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

1542 1543
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1544 1545
}

1546
void OBSBasic::SceneItemRemoved(void *data, calldata_t *params)
1547
{
1548
	OBSBasic *window = static_cast<OBSBasic*>(data);
1549

1550
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
1551

1552 1553
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1554 1555
}

1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
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));
}

1580
void OBSBasic::SourceAdded(void *data, calldata_t *params)
1581
{
J
jp9000 已提交
1582
	OBSBasic *window = static_cast<OBSBasic*>(data);
1583
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1584

1585
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1586
		QMetaObject::invokeMethod(window,
1587 1588
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1589 1590
}

1591
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
1592
{
1593
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1594

1595
	if (obs_scene_from_source(source) != NULL)
1596 1597 1598
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1599 1600
}

1601
void OBSBasic::SourceActivated(void *data, calldata_t *params)
1602
{
1603
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1604 1605 1606 1607 1608 1609 1610 1611
	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)));
}

1612
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
1613
{
1614
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1615 1616 1617 1618 1619 1620 1621 1622
	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)));
}

1623
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
1624 1625 1626 1627 1628 1629 1630 1631 1632 1633
{
	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)));
}

1634
void OBSBasic::ChannelChanged(void *data, calldata_t *params)
1635
{
1636
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1637
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1638 1639

	if (channel == 0)
1640 1641 1642
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1643 1644
}

1645 1646 1647 1648 1649
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1650 1651 1652
	gs_effect_t    *solid = obs_get_solid_effect();
	gs_eparam_t    *color = gs_effect_get_param_by_name(solid, "color");
	gs_technique_t *tech  = gs_effect_get_technique(solid, "Solid");
1653 1654 1655

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1656
	gs_effect_set_vec4(color, &colorVal);
1657

1658 1659
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1660 1661 1662 1663 1664 1665 1666 1667
	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();
1668 1669
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1670 1671 1672 1673

	gs_load_vertexbuffer(nullptr);
}

1674 1675
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
1676
	OBSBasic *window = static_cast<OBSBasic*>(data);
1677 1678 1679 1680
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
1681 1682
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
1683 1684 1685

	gs_viewport_push();
	gs_projection_push();
1686 1687 1688

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

1689 1690
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
1691
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
1692
			window->previewCX, window->previewCY);
1693

1694 1695
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
1696
	obs_render_main_view();
1697
	gs_load_vertexbuffer(nullptr);
1698

1699 1700
	/* --------------------------------------- */

1701 1702 1703
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
1704 1705 1706 1707

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
1708
	gs_reset_viewport();
J
jp9000 已提交
1709 1710 1711

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

1712 1713
	/* --------------------------------------- */

1714 1715
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
1716 1717 1718

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
1719 1720
}

1721 1722
/* Main class functions */

1723
obs_service_t *OBSBasic::GetService()
1724
{
1725
	if (!service) {
1726 1727
		service = obs_service_create("rtmp_common", NULL, NULL,
				nullptr);
1728 1729
		obs_service_release(service);
	}
1730 1731 1732
	return service;
}

1733
void OBSBasic::SetService(obs_service_t *newService)
1734
{
1735
	if (newService)
1736 1737 1738
		service = newService;
}

1739 1740 1741 1742 1743 1744 1745
bool OBSBasic::StreamingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

1746 1747 1748 1749 1750 1751
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772
static inline int AttemptToResetVideo(struct obs_video_info *ovi)
{
	int ret = obs_reset_video(ovi);
	if (ret == OBS_VIDEO_INVALID_PARAM) {
		struct obs_video_info new_params = *ovi;

		if (new_params.window_width == 0)
			new_params.window_width = 512;
		if (new_params.window_height == 0)
			new_params.window_height = 512;

		new_params.output_width  = new_params.window_width;
		new_params.output_height = new_params.window_height;
		new_params.base_width    = new_params.window_width;
		new_params.base_height   = new_params.window_height;
		ret = obs_reset_video(&new_params);
	}

	return ret;
}

1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
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;
}

1786 1787 1788 1789 1790 1791
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 已提交
1792 1793
	else if (astrcmpi(name, "I444") == 0)
		return VIDEO_FORMAT_I444;
1794 1795 1796 1797 1798 1799 1800 1801 1802
#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 已提交
1803
		return VIDEO_FORMAT_RGBA;
1804 1805
}

1806
int OBSBasic::ResetVideo()
J
jp9000 已提交
1807 1808
{
	struct obs_video_info ovi;
1809
	int ret;
J
jp9000 已提交
1810

1811
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
1812

1813 1814 1815 1816 1817 1818 1819
	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 已提交
1820
	ovi.graphics_module = App()->GetRenderModule();
1821
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1822
			"Video", "BaseCX");
1823
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1824
			"Video", "BaseCY");
1825
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1826
			"Video", "OutputCX");
1827
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1828
			"Video", "OutputCY");
1829 1830 1831 1832 1833
	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 已提交
1834 1835
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
1836
	ovi.scale_type     = GetScaleType(basicConfig);
1837

J
jp9000 已提交
1838
	QTToGSWindow(ui->preview->winId(), ovi.window);
J
jp9000 已提交
1839 1840

	//required to make opengl display stuff on osx(?)
J
jp9000 已提交
1841
	ResizePreview(ovi.base_width, ovi.base_height);
J
jp9000 已提交
1842

1843
	QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1844 1845
	ovi.window_width  = size.width();
	ovi.window_height = size.height();
J
jp9000 已提交
1846

1847
	ret = AttemptToResetVideo(&ovi);
1848 1849
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
		/* Try OpenGL if DirectX fails on windows */
1850
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
1851 1852 1853 1854
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
1855 1856
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
1857 1858 1859 1860
			ret = AttemptToResetVideo(&ovi);
		}
	}

1861 1862
	if (ret == OBS_VIDEO_SUCCESS)
		obs_add_draw_callback(OBSBasic::RenderMain, this);
1863

1864
	return ret;
J
jp9000 已提交
1865
}
J
jp9000 已提交
1866

1867
bool OBSBasic::ResetAudio()
J
jp9000 已提交
1868
{
1869
	struct obs_audio_info ai;
1870
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
1871 1872
			"SampleRate");

1873
	const char *channelSetupStr = config_get_string(basicConfig,
1874 1875 1876 1877 1878 1879 1880
			"Audio", "ChannelSetup");

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

1881
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
1882 1883

	return obs_reset_audio(&ai);
J
jp9000 已提交
1884 1885
}

J
jp9000 已提交
1886
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
1887
		const char *deviceDesc, int channel)
J
jp9000 已提交
1888 1889 1890
{
	const char *deviceId = config_get_string(basicConfig, "Audio",
			deviceName);
1891 1892
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
1893 1894 1895 1896
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
1897
		settings = obs_source_get_settings(source);
J
jp9000 已提交
1898
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909

		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) {
1910
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
1911
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
1912
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
1913
				sourceId, deviceDesc, settings, nullptr);
J
jp9000 已提交
1914 1915 1916 1917 1918 1919 1920 1921
		obs_data_release(settings);

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

void OBSBasic::ResetAudioDevices()
J
jp9000 已提交
1922
{
1923 1924 1925 1926 1927 1928 1929 1930 1931 1932
	ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice1",
			Str("Basic.DesktopDevice1"), 1);
	ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice2",
			Str("Basic.DesktopDevice2"), 2);
	ResetAudioDevice(App()->InputAudioSource(),  "AuxDevice1",
			Str("Basic.AuxDevice1"), 3);
	ResetAudioDevice(App()->InputAudioSource(),  "AuxDevice2",
			Str("Basic.AuxDevice2"), 4);
	ResetAudioDevice(App()->InputAudioSource(),  "AuxDevice3",
			Str("Basic.AuxDevice3"), 5);
J
jp9000 已提交
1933 1934
}

J
jp9000 已提交
1935
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
1936
{
1937
	QSize  targetSize;
J
jp9000 已提交
1938

1939
	/* resize preview panel to fix to the top section of the window */
1940
	targetSize = GetPixelSize(ui->preview);
1941
	GetScaleAndCenterPos(int(cx), int(cy),
1942 1943
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
1944
			previewX, previewY, previewScale);
J
jp9000 已提交
1945

1946 1947 1948
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);

J
jp9000 已提交
1949 1950 1951 1952 1953
	if (isVisible()) {
		if (resizeTimer)
			killTimer(resizeTimer);
		resizeTimer = startTimer(100);
	}
J
jp9000 已提交
1954 1955
}

J
jp9000 已提交
1956
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
1957
{
1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968
	if (outputHandler && outputHandler->Active()) {
		QMessageBox::StandardButton button = QMessageBox::question(
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

1969 1970 1971 1972
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

1973 1974 1975 1976 1977
	if (updateCheckThread)
		updateCheckThread->wait();
	if (logUploadThread)
		logUploadThread->wait();

P
Palana 已提交
1978 1979
	signalHandlers.clear();

1980 1981 1982 1983 1984 1985 1986 1987 1988 1989
	/* Check all child dialogs and ensure they run their proper closeEvent
	 * methods before exiting the application.  Otherwise Qt doesn't send
	 * the proper QCloseEvent messages. */
	QList<QDialog*> childDialogs = this->findChildren<QDialog *>();
	if (!childDialogs.isEmpty()) {
		for (int i = 0; i < childDialogs.size(); ++i) {
			childDialogs.at(i)->close();
		}
	}

J
jp9000 已提交
1990 1991 1992 1993 1994
	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}

1995 1996 1997
	// remove draw callback in case our drawable surfaces go away before
	// the destructor gets called
	obs_remove_draw_callback(OBSBasic::RenderMain, this);
1998 1999 2000 2001 2002

	/* Delete the save timer so it doesn't trigger after this point while
	 * the program data is being freed */
	delete saveTimer;
	SaveProject();
J
jp9000 已提交
2003 2004 2005 2006 2007 2008 2009

	/* Clear the list boxes in ::closeEvent to ensure that we can process
	 * any ->deleteLater events in this window created by Qt in relation to
	 * their internal data */
	ClearVolumeControls();
	ui->sources->clear();
	ui->scenes->clear();
2010 2011
}

J
jp9000 已提交
2012
void OBSBasic::changeEvent(QEvent *event)
2013
{
J
jp9000 已提交
2014 2015
	/* TODO */
	UNUSED_PARAMETER(event);
2016 2017
}

J
jp9000 已提交
2018
void OBSBasic::resizeEvent(QResizeEvent *event)
2019
{
J
jp9000 已提交
2020 2021 2022 2023
	struct obs_video_info ovi;

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

	UNUSED_PARAMETER(event);
2026 2027
}

J
jp9000 已提交
2028 2029 2030 2031 2032 2033
void OBSBasic::timerEvent(QTimerEvent *event)
{
	if (event->timerId() == resizeTimer) {
		killTimer(resizeTimer);
		resizeTimer = 0;

2034
		QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
2035 2036 2037 2038
		obs_resize(size.width(), size.height());
	}
}

2039 2040
void OBSBasic::on_actionShow_Recordings_triggered()
{
2041 2042 2043 2044
	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");
2045 2046 2047
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
2048 2049
void OBSBasic::on_actionRemux_triggered()
{
2050 2051 2052 2053
	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 已提交
2054 2055 2056 2057
	OBSRemux remux(path, this);
	remux.exec();
}

P
Palana 已提交
2058 2059 2060 2061 2062 2063
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

J
jp9000 已提交
2064 2065
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
2066 2067 2068 2069 2070
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
2071 2072 2073
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
2074 2075 2076

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

2079 2080 2081 2082 2083
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

2084 2085 2086 2087 2088
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

2089 2090
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
2091
{
2092
	obs_source_t *source = NULL;
J
jp9000 已提交
2093

2094 2095 2096 2097
	if (sceneChanging)
		return;

	if (current) {
2098
		obs_scene_t *scene;
J
jp9000 已提交
2099

2100
		scene = current->data(Qt::UserRole).value<OBSScene>();
2101
		source = obs_scene_get_source(scene);
2102 2103
	}

2104
	/* TODO: allow transitions */
2105
	obs_set_output_source(0, source);
2106 2107

	UNUSED_PARAMETER(prev);
2108 2109
}

J
jp9000 已提交
2110 2111
void OBSBasic::EditSceneName()
{
2112 2113 2114 2115 2116 2117
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

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

J
jp9000 已提交
2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142
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 已提交
2143
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
2144
{
J
jp9000 已提交
2145
	QListWidgetItem *item = ui->scenes->itemAt(pos);
J
jp9000 已提交
2146
	QPointer<QMenu> sceneProjectorMenu;
J
jp9000 已提交
2147

2148
	QMenu popup(this);
J
jp9000 已提交
2149
	QMenu order(QTStr("Basic.MainMenu.Edit.Order"), this);
J
jp9000 已提交
2150 2151 2152
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
2153 2154 2155 2156
	if (item) {
		popup.addSeparator();
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
2157
		popup.addAction(QTStr("Remove"),
2158 2159
				this, SLOT(RemoveSelectedScene()),
				DeleteKeys.front());
J
jp9000 已提交
2160
		popup.addSeparator();
J
jp9000 已提交
2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173

		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 已提交
2174 2175 2176 2177 2178
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
		popup.addSeparator();
J
jp9000 已提交
2179 2180
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
P
Palana 已提交
2181
	}
J
jp9000 已提交
2182 2183

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

J
jp9000 已提交
2186
void OBSBasic::on_actionAddScene_triggered()
2187
{
2188
	string name;
S
Socapex 已提交
2189
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
2190 2191 2192

	int i = 1;
	QString placeHolderText = format.arg(i);
2193
	obs_source_t *source = nullptr;
P
Palana 已提交
2194 2195
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
2196
		placeHolderText = format.arg(++i);
P
Palana 已提交
2197
	}
S
Socapex 已提交
2198

J
jp9000 已提交
2199
	bool accepted = NameDialog::AskForName(this,
2200 2201
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
2202 2203
			name,
			placeHolderText);
2204

J
jp9000 已提交
2205
	if (accepted) {
J
jp9000 已提交
2206 2207
		if (name.empty()) {
			QMessageBox::information(this,
2208 2209
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
2210 2211 2212 2213
			on_actionAddScene_triggered();
			return;
		}

2214
		obs_source_t *source = obs_get_source_by_name(name.c_str());
2215
		if (source) {
J
jp9000 已提交
2216
			QMessageBox::information(this,
2217 2218
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
2219 2220

			obs_source_release(source);
J
jp9000 已提交
2221
			on_actionAddScene_triggered();
2222 2223 2224
			return;
		}

2225
		obs_scene_t *scene = obs_scene_create(name.c_str());
2226
		source = obs_scene_get_source(scene);
2227
		obs_add_source(source);
2228
		obs_scene_release(scene);
2229 2230

		obs_set_output_source(0, source);
2231
	}
2232 2233
}

J
jp9000 已提交
2234
void OBSBasic::on_actionRemoveScene_triggered()
2235
{
2236
	OBSScene     scene  = GetCurrentScene();
2237
	obs_source_t *source = obs_scene_get_source(scene);
2238 2239 2240

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
2241 2242
}

J
jp9000 已提交
2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262
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 已提交
2263
void OBSBasic::on_actionSceneUp_triggered()
2264
{
J
jp9000 已提交
2265
	ChangeSceneIndex(true, -1, 0);
2266 2267
}

J
jp9000 已提交
2268
void OBSBasic::on_actionSceneDown_triggered()
2269
{
J
jp9000 已提交
2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281
	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);
2282 2283
}

2284 2285
void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
2286
{
2287
	auto select_one = [] (obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2288 2289
			void *param)
	{
2290
		obs_sceneitem_t *selectedItem =
J
jp9000 已提交
2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301
			*reinterpret_cast<OBSSceneItem*>(param);
		obs_sceneitem_select(item, (selectedItem == item));

		UNUSED_PARAMETER(scene);
		return true;
	};

	if (!current)
		return;

	OBSSceneItem item = current->data(Qt::UserRole).value<OBSSceneItem>();
2302
	obs_source_t *source = obs_sceneitem_get_source(item);
2303 2304 2305
	if ((obs_source_get_output_flags(source) & OBS_SOURCE_VIDEO) == 0)
		return;

J
jp9000 已提交
2306 2307
	obs_scene_enum_items(GetCurrentScene(), select_one, &item);

2308
	UNUSED_PARAMETER(prev);
2309 2310
}

J
jp9000 已提交
2311 2312
void OBSBasic::EditSceneItemName()
{
2313 2314
	QListWidgetItem *item = ui->sources->currentItem();
	Qt::ItemFlags flags   = item->flags();
2315 2316 2317
	OBSSceneItem sceneItem= item->data(Qt::UserRole).value<OBSSceneItem>();
	obs_source_t *source  = obs_sceneitem_get_source(sceneItem);
	const char *name      = obs_source_get_name(source);
2318

2319
	item->setText(QT_UTF8(name));
2320
	item->setFlags(flags | Qt::ItemIsEditable);
2321
	ui->sources->removeItemWidget(item);
2322 2323
	ui->sources->editItem(item);
	item->setFlags(flags);
J
jp9000 已提交
2324 2325
}

J
jp9000 已提交
2326
void OBSBasic::CreateSourcePopupMenu(QListWidgetItem *item, bool preview)
2327
{
2328
	QMenu popup(this);
J
jp9000 已提交
2329 2330
	QPointer<QMenu> previewProjector;
	QPointer<QMenu> sourceProjector;
J
jp9000 已提交
2331 2332 2333 2334 2335 2336 2337 2338

	if (preview) {
		QAction *action = popup.addAction(
				QTStr("Basic.Main.PreviewConextMenu.Enable"),
				this, SLOT(TogglePreview()));
		action->setCheckable(true);
		action->setChecked(obs_preview_enabled());

J
jp9000 已提交
2339 2340 2341 2342 2343 2344
		previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
				SLOT(OpenPreviewProjector()));

		popup.addMenu(previewProjector);

J
jp9000 已提交
2345 2346 2347
		popup.addSeparator();
	}

J
jp9000 已提交
2348 2349 2350 2351 2352 2353 2354 2355
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

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

J
John Bradley 已提交
2356
		OBSSceneItem sceneItem = GetSceneItem(item);
2357
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
John Bradley 已提交
2358 2359
		QAction *action;

J
jp9000 已提交
2360 2361
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
2362 2363
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
2364
				DeleteKeys.front());
J
jp9000 已提交
2365 2366
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
2367
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
2368 2369 2370 2371 2372 2373 2374

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

		popup.addSeparator();
		popup.addMenu(sourceProjector);
J
jp9000 已提交
2375
		popup.addSeparator();
J
John Bradley 已提交
2376 2377 2378 2379 2380 2381 2382

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

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

J
jp9000 已提交
2383 2384
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
2385 2386 2387 2388 2389
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

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

J
jp9000 已提交
2392 2393 2394 2395 2396
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->itemAt(pos), false);
}

P
Palana 已提交
2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408
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 已提交
2409
void OBSBasic::AddSource(const char *id)
2410
{
2411 2412 2413 2414
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
	}
2415 2416
}

2417
QMenu *OBSBasic::CreateAddSourcePopupMenu()
2418
{
2419
	const char *type;
J
jp9000 已提交
2420 2421
	bool foundValues = false;
	size_t idx = 0;
2422

2423
	QMenu *popup = new QMenu(QTStr("Add"), this);
J
jp9000 已提交
2424
	while (obs_enum_input_types(idx++, &type)) {
2425
		const char *name = obs_source_get_display_name(
2426
				OBS_SOURCE_TYPE_INPUT, type);
2427

2428 2429 2430
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
2431 2432
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
2433 2434 2435
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
2436

J
jp9000 已提交
2437
		foundValues = true;
2438 2439
	}

2440 2441 2442
	if (!foundValues) {
		delete popup;
		popup = nullptr;
2443
	}
2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469

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

J
jp9000 已提交
2472
void OBSBasic::on_actionAddSource_triggered()
2473
{
J
jp9000 已提交
2474
	AddSourcePopupMenu(QCursor::pos());
2475 2476
}

J
jp9000 已提交
2477
void OBSBasic::on_actionRemoveSource_triggered()
2478
{
2479
	OBSSceneItem item   = GetCurrentSceneItem();
2480
	obs_source_t *source = obs_sceneitem_get_source(item);
2481 2482

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
2483
		obs_sceneitem_remove(item);
2484 2485
}

J
John Bradley 已提交
2486 2487 2488 2489 2490 2491 2492 2493 2494
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
2495
void OBSBasic::on_actionSourceProperties_triggered()
2496
{
2497
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2498
	OBSSource source = obs_sceneitem_get_source(item);
2499

2500 2501
	if (source)
		CreatePropertiesWindow(source);
2502 2503
}

J
jp9000 已提交
2504
void OBSBasic::on_actionSourceUp_triggered()
2505
{
J
jp9000 已提交
2506
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2507
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
2508
}
J
jp9000 已提交
2509

J
jp9000 已提交
2510
void OBSBasic::on_actionSourceDown_triggered()
2511
{
J
jp9000 已提交
2512
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2513
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
2514 2515
}

J
jp9000 已提交
2516 2517 2518
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2519
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
2520 2521 2522 2523 2524
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2525
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
2526 2527 2528 2529 2530
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2531
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
2532 2533 2534 2535 2536
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2537
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
2538 2539
}

2540
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
2541
{
2542
	char logDir[512];
2543
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2544
		return nullptr;
J
jp9000 已提交
2545 2546 2547 2548 2549

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

2550
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
2551 2552 2553 2554 2555 2556 2557 2558
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

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

2561
	if (!fileString)
J
jp9000 已提交
2562 2563
		return;

2564
	if (!*fileString)
J
jp9000 已提交
2565 2566 2567 2568
		return;

	ui->menuLogFiles->setEnabled(false);

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

2572 2573 2574
	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 已提交
2575

2576
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
2577

2578 2579 2580 2581 2582 2583 2584 2585 2586 2587
	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 已提交
2588
	if (!json) {
2589 2590 2591 2592
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

F
fryshorts 已提交
2593 2594 2595
	QBuffer *postData = new QBuffer();
	postData->setData(json, (int) strlen(json));

2596 2597 2598 2599
	if (logUploadThread) {
		logUploadThread->wait();
		delete logUploadThread;
	}
F
fryshorts 已提交
2600

2601 2602 2603 2604 2605 2606 2607
	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 已提交
2608 2609
}

P
Palana 已提交
2610 2611
void OBSBasic::on_actionShowLogs_triggered()
{
2612
	char logDir[512];
2613
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2614 2615
		return;

P
Palana 已提交
2616 2617 2618 2619
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
2620 2621 2622 2623 2624 2625 2626 2627 2628 2629
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

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

2630 2631 2632
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
2633
	if (GetConfigPath(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645
		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 已提交
2646 2647 2648 2649 2650
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

2651
void OBSBasic::logUploadFinished(const QString &text, const QString &error)
J
jp9000 已提交
2652 2653 2654
{
	ui->menuLogFiles->setEnabled(true);

2655
	if (text.isEmpty()) {
J
jp9000 已提交
2656 2657
		QMessageBox::information(this,
				QTStr("LogReturnDialog.ErrorUploadingLog"),
2658
				error);
J
jp9000 已提交
2659 2660 2661
		return;
	}

2662
	obs_data_t *returnData = obs_data_create_from_json(QT_TO_UTF8(text));
F
fryshorts 已提交
2663
	QString logURL         = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
2664 2665 2666 2667 2668 2669
	obs_data_release(returnData);

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

J
jp9000 已提交
2670
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
2671
		obs_source_t *source, const string &name)
2672
{
2673 2674 2675 2676
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

2677
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
2678
	QListWidgetItem *listItem    = listWidget->currentItem();
2679

2680
	if (foundSource || name.empty()) {
2681
		listItem->setText(QT_UTF8(prevName));
2682

2683
		if (foundSource) {
2684 2685 2686 2687 2688 2689 2690 2691 2692
			QMessageBox::information(parent,
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
			QMessageBox::information(parent,
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

2693 2694 2695
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
2696
		obs_source_set_name(source, name.c_str());
2697 2698 2699
	}
}

J
jp9000 已提交
2700 2701 2702 2703 2704
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
2705
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
2706 2707 2708 2709

	if (!scene)
		return;

2710
	obs_source_t *source = obs_scene_get_source(scene);
2711
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
2712 2713 2714 2715 2716 2717 2718 2719 2720

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
2721
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
2722 2723 2724 2725

	if (!item)
		return;

2726
	obs_source_t *source = obs_sceneitem_get_source(item);
2727
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
2728

2729 2730 2731 2732
	QListWidgetItem *listItem = ui->sources->currentItem();
	listItem->setText(QString());
	SetupVisibilityItem(ui->sources, listItem, item);

J
jp9000 已提交
2733 2734 2735
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
2736 2737 2738 2739 2740 2741 2742 2743
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
2744 2745 2746 2747 2748 2749 2750 2751
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772
void OBSBasic::StartStreaming()
{
	SaveProject();

	if (outputHandler->StreamingActive())
		return;

	if (outputHandler->StartStreaming(service)) {
		ui->streamButton->setEnabled(false);
		ui->streamButton->setText(QTStr("Basic.Main.Connecting"));
	}
}

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

	if (outputHandler->StreamingActive())
		outputHandler->StopStreaming();
}

2773
void OBSBasic::StreamingStart()
2774
{
2775
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
2776
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
2777
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
2778 2779
}

2780
void OBSBasic::StreamingStop(int code)
2781
{
2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796
	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;

2797
	default:
2798 2799 2800 2801 2802 2803 2804 2805 2806 2807
	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 已提交
2808
	ui->statusbar->StreamStopped();
2809

2810
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
2811
	ui->streamButton->setEnabled(true);
2812 2813 2814 2815 2816

	if (code != OBS_OUTPUT_SUCCESS)
		QMessageBox::information(this,
				QTStr("Output.ConnectFail.Title"),
				QT_UTF8(errorMessage));
J
jp9000 已提交
2817 2818
}

2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
void OBSBasic::StartRecording()
{
	SaveProject();

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

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

	if (outputHandler->RecordingActive())
		outputHandler->StopRecording();
}

P
Palana 已提交
2835 2836
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
2837
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
2838
	ui->recordButton->setText(QTStr("Basic.Main.StopRecording"));
P
Palana 已提交
2839 2840
}

2841
void OBSBasic::RecordingStop(int code)
2842
{
P
Palana 已提交
2843
	ui->statusbar->RecordingStopped();
2844
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
2845 2846 2847 2848 2849

	if (code == OBS_OUTPUT_UNSUPPORTED)
		QMessageBox::information(this,
				QTStr("Output.RecordFail.Title"),
				QTStr("Output.RecordFail.Unsupported"));
2850
}
2851

2852 2853
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
2854
	if (outputHandler->StreamingActive()) {
2855
		StopStreaming();
2856
	} else {
2857
		StartStreaming();
2858 2859 2860 2861 2862
	}
}

void OBSBasic::on_recordButton_clicked()
{
2863 2864 2865 2866
	if (outputHandler->RecordingActive())
		StopRecording();
	else
		StartRecording();
J
jp9000 已提交
2867 2868
}

J
jp9000 已提交
2869
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
2870
{
2871 2872
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
2873
}
2874

2875 2876 2877 2878 2879 2880
void OBSBasic::on_actionWebsite_triggered()
{
	QUrl url = QUrl("https://obsproject.com", QUrl::TolerantMode);
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891
void OBSBasic::on_preview_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->currentItem(), true);

	UNUSED_PARAMETER(pos);
}

void OBSBasic::on_previewDisabledLabel_customContextMenuRequested(
		const QPoint &pos)
{
	QMenu popup(this);
J
jp9000 已提交
2892
	QPointer<QMenu> previewProjector;
J
jp9000 已提交
2893 2894 2895 2896 2897 2898 2899

	QAction *action = popup.addAction(
			QTStr("Basic.Main.PreviewConextMenu.Enable"),
			this, SLOT(TogglePreview()));
	action->setCheckable(true);
	action->setChecked(obs_preview_enabled());

J
jp9000 已提交
2900 2901 2902 2903 2904
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));

	popup.addMenu(previewProjector);
J
jp9000 已提交
2905 2906 2907 2908 2909
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972
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);
}

2973
config_t *OBSBasic::Config() const
2974 2975 2976
{
	return basicConfig;
}
J
jp9000 已提交
2977 2978 2979

void OBSBasic::on_actionEditTransform_triggered()
{
2980 2981 2982
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
2983 2984
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
2985
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
2986 2987 2988 2989
}

void OBSBasic::on_actionResetTransform_triggered()
{
2990
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
2991 2992 2993 2994
	{
		if (!obs_sceneitem_selected(item))
			return true;

2995
		obs_transform_info info;
J
jp9000 已提交
2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012
		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);
}

3013
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
3014 3015 3016 3017 3018
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

3021
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
3022 3023 3024 3025
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
3026 3027
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
3028 3029
	};

3030 3031 3032 3033 3034 3035
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

3036
static vec3 GetItemTL(obs_sceneitem_t *item)
3037 3038 3039
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
3040 3041 3042
	return tl;
}

3043
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
3044 3045 3046 3047
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
3048
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
3049 3050 3051
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
3052
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
3053 3054
}

3055
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3056 3057 3058 3059 3060 3061 3062 3063 3064
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
3065
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
3066 3067
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
3068
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094

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

3095
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3096 3097 3098 3099 3100 3101 3102 3103 3104 3105
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
3106
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
3107
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
3108
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
3109 3110

	SetItemTL(item, tl);
J
jp9000 已提交
3111 3112

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
3113 3114 3115 3116 3117
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
3118 3119
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
3120 3121 3122 3123 3124 3125
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
3126 3127
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
3128 3129 3130 3131
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

3132
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
3133 3134 3135 3136 3137 3138 3139 3140 3141 3142
		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);

3143
	obs_transform_info itemInfo;
J
jp9000 已提交
3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175
	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()
{
3176
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201
	{
		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 已提交
3202
		UNUSED_PARAMETER(param);
3203 3204 3205 3206
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
J
jp9000 已提交
3207
}
J
jp9000 已提交
3208 3209 3210 3211 3212 3213 3214 3215

void OBSBasic::TogglePreview()
{
	bool enabled = !obs_preview_enabled();
	obs_preview_set_enabled(enabled);
	ui->preview->setVisible(enabled);
	ui->previewDisabledLabel->setVisible(!enabled);
}
3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226

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 已提交
3227
		struct vec2 dir;
3228 3229
		struct vec2 pos;

J
jp9000 已提交
3230 3231
		vec2_set(&dir, 0.0f, 0.0f);

3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254
		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 已提交
3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295

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