window-basic-main.cpp 74.2 KB
Newer Older
1
/******************************************************************************
2
    Copyright (C) 2013-2015 by Hugh Bailey <obs.jim@gmail.com>
J
jp9000 已提交
3
                               Zachary Lund <admin@computerquip.com>
4
                               Philippe Groarke <philippe.groarke@gmail.com>
5 6 7

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
8
    the Free Software Foundation, either version 2 of the License, or
9 10 11
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
12
    but WITHOUT ANY WARRANTY; without even the implied warranty of
13 14 15 16 17 18 19
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/

J
jp9000 已提交
20
#include <time.h>
J
jp9000 已提交
21
#include <obs.hpp>
J
jp9000 已提交
22
#include <QMessageBox>
23
#include <QShowEvent>
24
#include <QDesktopServices>
J
jp9000 已提交
25
#include <QFileDialog>
J
jp9000 已提交
26 27
#include <QNetworkRequest>
#include <QNetworkReply>
28

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

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

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

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

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

58 59
#define PREVIEW_EDGE_SIZE 10

60
using namespace std;
J
jp9000 已提交
61

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

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

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

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

80 81
static QList<QKeySequence> DeleteKeys;

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

S
Socapex 已提交
89
	copyActionsDynamicProperties();
90

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

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

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

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

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

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

122 123 124 125 126 127
	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 已提交
128 129 130 131

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

J
jp9000 已提交
132
	blog(LOG_INFO, "%s", name.str().c_str());
J
jp9000 已提交
133
	setWindowTitle(QT_UTF8(name.str().c_str()));
J
jp9000 已提交
134 135 136 137 138 139 140 141 142 143 144 145 146 147

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

	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
150 151 152 153
	cpuUsageTimer = new QTimer(this);
	connect(cpuUsageTimer, SIGNAL(timeout()),
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
154

155 156 157 158 159 160
	DeleteKeys =
#ifdef __APPLE__
		QList<QKeySequence>{{Qt::Key_Backspace}} <<
#endif
		QKeySequence::keyBindings(QKeySequence::Delete);

161
#ifdef __APPLE__
162 163
	ui->actionRemoveSource->setShortcuts(DeleteKeys);
	ui->actionRemoveScene->setShortcuts(DeleteKeys);
164 165 166

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
167
#endif
168 169 170 171 172 173 174 175 176 177 178 179 180 181

	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()));
182 183
}

184
static void SaveAudioDevice(const char *name, int channel, obs_data_t *parent)
185
{
186
	obs_source_t *source = obs_get_output_source(channel);
187 188 189
	if (!source)
		return;

190
	obs_data_t *data = obs_save_source(source);
191

J
jp9000 已提交
192
	obs_data_set_obj(parent, name, data);
193 194 195 196 197

	obs_data_release(data);
	obs_source_release(source);
}

198
static obs_data_t *GenerateSaveData()
199
{
200 201 202
	obs_data_t       *saveData     = obs_data_create();
	obs_data_array_t *sourcesArray = obs_save_sources();
	obs_source_t     *currentScene = obs_get_output_source(0);
203
	const char       *sceneName   = obs_source_get_name(currentScene);
204

205 206 207 208 209 210
	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 已提交
211 212
	obs_data_set_string(saveData, "current_scene", sceneName);
	obs_data_set_array(saveData, "sources", sourcesArray);
213 214 215 216 217 218
	obs_data_array_release(sourcesArray);
	obs_source_release(currentScene);

	return saveData;
}

S
Socapex 已提交
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238
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));
		}
	}
}

239 240 241 242 243 244 245 246 247 248 249 250
void OBSBasic::ClearVolumeControls()
{
	VolControl *control;

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

	volumes.clear();
}

251 252
void OBSBasic::Save(const char *file)
{
253
	obs_data_t *saveData  = GenerateSaveData();
J
jp9000 已提交
254
	const char *jsonData = obs_data_get_json(saveData);
255

S
Socapex 已提交
256 257 258 259 260 261 262 263
	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);
	}
264 265 266 267

	obs_data_release(saveData);
}

268
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
269
{
270
	obs_data_t *data = obs_data_get_obj(parent, name);
271 272 273
	if (!data)
		return;

274
	obs_source_t *source = obs_load_source(data);
275 276 277 278 279 280 281 282 283 284
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

void OBSBasic::CreateDefaultScene()
{
285 286
	obs_scene_t  *scene  = obs_scene_create(Str("Basic.Scene"));
	obs_source_t *source = obs_scene_get_source(scene);
287 288 289 290 291

	obs_add_source(source);

#ifdef __APPLE__
	source = obs_source_create(OBS_SOURCE_TYPE_INPUT, "display_capture",
292
			Str("Basic.DisplayCapture"), NULL);
293 294 295 296 297 298 299 300

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

301
	obs_set_output_source(0, obs_scene_get_source(scene));
302 303 304
	obs_scene_release(scene);
}

305 306 307 308 309 310 311 312
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);
313 314
	if (!jsonData) {
		CreateDefaultScene();
315
		return;
316
	}
317

318 319
	obs_data_t       *data       = obs_data_create_from_json(jsonData);
	obs_data_array_t *sources    = obs_data_get_array(data, "sources");
J
jp9000 已提交
320 321
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
322
	obs_source_t     *curScene;
323

324 325 326 327 328 329
	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);

330 331 332 333 334 335 336 337 338 339
	obs_load_sources(sources);

	curScene = obs_get_source_by_name(sceneName);
	obs_set_output_source(0, curScene);
	obs_source_release(curScene);

	obs_data_array_release(sources);
	obs_data_release(data);
}

340 341 342
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
343
	obs_properties_t *props = obs_get_source_properties(
344
			OBS_SOURCE_TYPE_INPUT, output_id);
345 346 347 348 349
	size_t count = 0;

	if (!props)
		return false;

350
	obs_property_t *devices = obs_properties_get(props, "device_id");
351 352 353 354 355 356 357 358
	if (devices)
		count = obs_property_list_item_count(devices);

	obs_properties_destroy(props);

	return count != 0;
}

359 360 361 362 363 364 365
#define SERVICE_PATH "obs-studio/basic/service.json"

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

366 367 368 369
	char serviceJsonPath[512];
	int ret = os_get_config_path(serviceJsonPath, sizeof(serviceJsonPath),
			SERVICE_PATH);
	if (ret <= 0)
370 371
		return;

372 373
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
374

375
	obs_data_set_string(data, "type", obs_service_get_type(service));
J
jp9000 已提交
376
	obs_data_set_obj(data, "settings", settings);
377

J
jp9000 已提交
378
	const char *json = obs_data_get_json(data);
379 380 381 382 383 384 385 386 387 388 389

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

	obs_data_release(settings);
	obs_data_release(data);
}

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

390 391 392 393 394
	if (service) {
		obs_service_destroy(service);
		service = nullptr;
	}

395 396 397 398
	char serviceJsonPath[512];
	int ret = os_get_config_path(serviceJsonPath, sizeof(serviceJsonPath),
			SERVICE_PATH);
	if (ret <= 0)
399 400 401 402 403 404
		return false;

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

405
	obs_data_t *data = obs_data_create_from_json(jsonText);
406 407

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

410
	obs_data_t *settings = obs_data_get_obj(data, "settings");
411

412
	service = obs_service_create(type, "default_service", settings);
413 414 415 416 417 418 419 420 421 422 423 424

	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

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

425
	service = obs_service_create("rtmp_common", "default_service", nullptr);
426 427 428 429 430 431
	if (!service)
		return false;

	return true;
}

432 433
bool OBSBasic::InitBasicConfigDefaults()
{
434 435 436
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

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

J
jp9000 已提交
449 450
	config_set_default_string(basicConfig, "Output", "Type", "Simple");

451 452
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
453 454 455
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 128);
J
jp9000 已提交
456 457
	config_set_default_bool  (basicConfig, "SimpleOutput", "Reconnect",
			true);
458 459
	config_set_default_uint  (basicConfig, "SimpleOutput", "RetryDelay",
			10);
J
jp9000 已提交
460 461
	config_set_default_uint  (basicConfig, "SimpleOutput", "MaxRetries",
			20);
J
jp9000 已提交
462 463
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
464
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseCBR", true);
465 466 467
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseBufsize",
			false);
	config_set_default_int   (basicConfig, "SimpleOutput", "Bufsize", 2500);
J
jp9000 已提交
468 469
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
470

J
jp9000 已提交
471
	config_set_default_bool  (basicConfig, "AdvOut", "Reconnect", true);
472
	config_set_default_uint  (basicConfig, "AdvOut", "RetryDelay", 10);
J
jp9000 已提交
473
	config_set_default_uint  (basicConfig, "AdvOut", "MaxRetries", 20);
474 475
	config_set_default_bool  (basicConfig, "AdvOut", "ApplyServiceSettings",
			true);
J
jp9000 已提交
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
	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());
	config_set_default_bool  (basicConfig, "AdvOut", "RecUseRescale",
			false);
	config_set_default_bool  (basicConfig, "AdvOut", "RecMultitrack",
			false);
	config_set_default_uint  (basicConfig, "AdvOut", "RecTrackIndex", 1);
	config_set_default_uint  (basicConfig, "AdvOut", "RecTrackCount", 1);
	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);

506 507 508 509 510 511 512 513 514 515 516 517 518
	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);
519
	config_set_default_string(basicConfig, "Video", "ScaleType", "bicubic");
520 521 522 523
	config_set_default_string(basicConfig, "Video", "ColorFormat", "NV12");
	config_set_default_string(basicConfig, "Video", "ColorSpace", "709");
	config_set_default_string(basicConfig, "Video", "ColorRange",
			"Partial");
524 525 526 527 528 529

	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 已提交
530
	config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
531
			hasDesktopAudio ? "default" : "disabled");
J
jp9000 已提交
532 533 534
	config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice1",
535
			hasInputAudio ? "default" : "disabled");
J
jp9000 已提交
536 537 538 539 540
	config_set_default_string(basicConfig, "Audio", "AuxDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice3",
			"disabled");

541 542 543 544 545
	return true;
}

bool OBSBasic::InitBasicConfig()
{
546 547 548 549 550 551 552
	char configPath[512];
	int ret = os_get_config_path(configPath, sizeof(configPath),
			"obs-studio/basic/basic.ini");
	if (ret <= 0) {
		OBSErrorBox(nullptr, "Failed to get base.ini path");
		return false;
	}
553

554 555 556
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
557 558 559 560 561 562
		return false;
	}

	return InitBasicConfigDefaults();
}

563 564
void OBSBasic::InitOBSCallbacks()
{
565
	signal_handler_connect(obs_get_signal_handler(), "source_add",
566
			OBSBasic::SourceAdded, this);
567
	signal_handler_connect(obs_get_signal_handler(), "source_remove",
568
			OBSBasic::SourceRemoved, this);
569
	signal_handler_connect(obs_get_signal_handler(), "channel_change",
570
			OBSBasic::ChannelChanged, this);
571
	signal_handler_connect(obs_get_signal_handler(), "source_activate",
572
			OBSBasic::SourceActivated, this);
573
	signal_handler_connect(obs_get_signal_handler(), "source_deactivate",
574
			OBSBasic::SourceDeactivated, this);
575
	signal_handler_connect(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
576
			OBSBasic::SourceRenamed, this);
577 578
}

J
jp9000 已提交
579 580
void OBSBasic::InitPrimitives()
{
J
jp9000 已提交
581
	obs_enter_graphics();
J
jp9000 已提交
582

583
	gs_render_start(true);
J
jp9000 已提交
584 585 586 587 588
	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);
589
	box = gs_render_save();
J
jp9000 已提交
590

591
	gs_render_start(true);
J
jp9000 已提交
592 593 594 595
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
596
	circle = gs_render_save();
J
jp9000 已提交
597

J
jp9000 已提交
598
	obs_leave_graphics();
J
jp9000 已提交
599 600
}

J
jp9000 已提交
601 602
void OBSBasic::ResetOutputs()
{
J
jp9000 已提交
603 604 605
	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	bool advOut = astrcmpi(mode, "Advanced") == 0;

J
jp9000 已提交
606 607
	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
J
jp9000 已提交
608 609 610
		outputHandler.reset(advOut ?
			CreateAdvancedOutputHandler(this) :
			CreateSimpleOutputHandler(this));
J
jp9000 已提交
611 612 613 614 615
	} else {
		outputHandler->Update();
	}
}

616 617
void OBSBasic::OBSInit()
{
618 619 620 621 622
	char savePath[512];
	int ret = os_get_config_path(savePath, sizeof(savePath),
			"obs-studio/basic/scenes.json");
	if (ret <= 0)
		throw "Failed to get scenes.json file path";
623

624 625 626 627
	/* make sure it's fully displayed before doing any initialization */
	show();
	App()->processEvents();

628
	if (!obs_startup(App()->GetLocale()))
629
		throw "Failed to initialize libobs";
630 631
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
632
	if (!ResetAudio())
633 634
		throw "Failed to initialize audio";

635
	ret = ResetVideo();
636 637 638 639 640

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
J
jp9000 已提交
641 642 643
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
644 645 646 647 648 649 650
	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";
	}

651
	InitOBSCallbacks();
652

653
	AddExtraModulePaths();
J
jp9000 已提交
654
	obs_load_all_modules();
J
jp9000 已提交
655

J
jp9000 已提交
656 657
	ResetOutputs();

658 659 660
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
661 662
	InitPrimitives();

663
	Load(savePath);
J
jp9000 已提交
664
	ResetAudioDevices();
665

J
jp9000 已提交
666
	TimedCheckForUpdates();
667
	loaded = true;
J
jp9000 已提交
668

669 670 671
	saveTimer = new QTimer(this);
	connect(saveTimer, SIGNAL(timeout()), this, SLOT(SaveProject()));
	saveTimer->start(20000);
J
jp9000 已提交
672 673 674 675 676 677

	bool previewEnabled = config_get_bool(App()->GlobalConfig(),
			"BasicWindow", "PreviewEnabled");
	if (!previewEnabled)
		QMetaObject::invokeMethod(this, "TogglePreview",
				Qt::QueuedConnection);
678 679 680 681
}

OBSBasic::~OBSBasic()
{
J
jp9000 已提交
682 683
	bool previewEnabled = obs_preview_enabled();

J
jp9000 已提交
684 685 686 687 688 689
	/* 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 已提交
690
	delete cpuUsageTimer;
691 692
	os_cpu_usage_info_destroy(cpuUsageInfo);

J
jp9000 已提交
693 694
	outputHandler.reset();

J
John Bradley 已提交
695 696 697
	if (interaction)
		delete interaction;

698 699 700
	if (properties)
		delete properties;

J
jp9000 已提交
701 702 703
	if (filters)
		delete filters;

704 705
	if (transformWindow)
		delete transformWindow;
706

J
jp9000 已提交
707 708 709
	if (advAudioWindow)
		delete advAudioWindow;

J
jp9000 已提交
710
	obs_enter_graphics();
711 712
	gs_vertexbuffer_destroy(box);
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
713
	obs_leave_graphics();
J
jp9000 已提交
714

715 716 717 718 719 720 721 722 723
	/* 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);

724
	obs_shutdown();
J
jp9000 已提交
725 726 727

	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
J
jp9000 已提交
728 729 730

	QRect lastGeom = normalGeometry();

J
jp9000 已提交
731
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cx",
J
jp9000 已提交
732
			lastGeom.width());
J
jp9000 已提交
733
	config_set_int(App()->GlobalConfig(), "BasicWindow", "cy",
J
jp9000 已提交
734
			lastGeom.height());
J
jp9000 已提交
735
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posx",
J
jp9000 已提交
736
			lastGeom.x());
J
jp9000 已提交
737
	config_set_int(App()->GlobalConfig(), "BasicWindow", "posy",
J
jp9000 已提交
738
			lastGeom.y());
J
jp9000 已提交
739 740
	config_set_bool(App()->GlobalConfig(), "BasicWindow", "PreviewEnabled",
			previewEnabled);
J
jp9000 已提交
741
	config_save(App()->GlobalConfig());
742 743
}

J
jp9000 已提交
744 745
void OBSBasic::SaveProject()
{
746 747 748 749 750 751
	char savePath[512];
	int ret = os_get_config_path(savePath, sizeof(savePath),
			"obs-studio/basic/scenes.json");
	if (ret <= 0)
		return;

J
jp9000 已提交
752 753 754
	Save(savePath);
}

J
jp9000 已提交
755
OBSScene OBSBasic::GetCurrentScene()
756
{
J
jp9000 已提交
757
	QListWidgetItem *item = ui->scenes->currentItem();
J
jp9000 已提交
758
	return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
759 760
}

761
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
762
{
J
jp9000 已提交
763
	return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
J
jp9000 已提交
764 765
}

766 767 768 769 770
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
	return GetSceneItem(ui->sources->currentItem());
}

771 772 773 774 775
void OBSBasic::UpdateSources(OBSScene scene)
{
	ui->sources->clear();

	obs_scene_enum_items(scene,
776
			[] (obs_scene_t *scene, obs_sceneitem_t *item, void *p)
777 778
			{
				OBSBasic *window = static_cast<OBSBasic*>(p);
779
				window->InsertSceneItem(item);
J
jp9000 已提交
780 781

				UNUSED_PARAMETER(scene);
782 783 784 785
				return true;
			}, this);
}

786
void OBSBasic::InsertSceneItem(obs_sceneitem_t *item)
787
{
788
	obs_source_t *source = obs_sceneitem_get_source(item);
789

790
	QListWidgetItem *listItem = new QListWidgetItem();
791 792 793 794
	listItem->setData(Qt::UserRole,
			QVariant::fromValue(OBSSceneItem(item)));

	ui->sources->insertItem(0, listItem);
795 796
	ui->sources->setCurrentRow(0);

797 798
	SetupVisibilityItem(ui->sources, listItem, item);

799
	/* if the source was just created, open properties dialog */
800 801 802 803
	if (sourceSceneRefs[source] == 0 && loaded)
		CreatePropertiesWindow(source);
}

804
void OBSBasic::CreateInteractionWindow(obs_source_t *source)
J
John Bradley 已提交
805 806 807 808 809 810 811 812 813
{
	if (interaction)
		interaction->close();

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

814
void OBSBasic::CreatePropertiesWindow(obs_source_t *source)
815 816 817 818 819 820 821
{
	if (properties)
		properties->close();

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

J
jp9000 已提交
824 825 826 827 828 829 830 831 832 833
void OBSBasic::CreateFiltersWindow(obs_source_t *source)
{
	if (filters)
		filters->close();

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

834 835 836
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
837
{
838
	const char *name  = obs_source_get_name(source);
839
	obs_scene_t *scene = obs_scene_from_source(source);
J
jp9000 已提交
840 841

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

845
	signal_handler_t *handler = obs_source_get_signal_handler(source);
846
	signal_handler_connect(handler, "item_add",
J
jp9000 已提交
847
			OBSBasic::SceneItemAdded, this);
848
	signal_handler_connect(handler, "item_remove",
J
jp9000 已提交
849
			OBSBasic::SceneItemRemoved, this);
850 851 852
	signal_handler_connect(handler, "item_select",
			OBSBasic::SceneItemSelected, this);
	signal_handler_connect(handler, "item_deselect",
J
jp9000 已提交
853
			OBSBasic::SceneItemDeselected, this);
854 855
	signal_handler_connect(handler, "reorder",
			OBSBasic::SceneReordered, this);
856 857
}

858
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
859
{
860
	const char *name = obs_source_get_name(source);
J
jp9000 已提交
861

J
jp9000 已提交
862 863 864
	QListWidgetItem *sel = ui->scenes->currentItem();
	QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
			Qt::MatchExactly);
J
jp9000 已提交
865

J
jp9000 已提交
866 867 868 869
	if (sel != nullptr) {
		if (items.contains(sel))
			ui->sources->clear();
		delete sel;
J
jp9000 已提交
870
	}
871 872
}

873
void OBSBasic::AddSceneItem(OBSSceneItem item)
874
{
875 876
	obs_scene_t  *scene  = obs_sceneitem_get_scene(item);
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
877

878 879
	if (GetCurrentScene() == scene)
		InsertSceneItem(item);
J
jp9000 已提交
880 881

	sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
882 883
}

884
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
885
{
886
	obs_scene_t *scene = obs_sceneitem_get_scene(item);
887

J
jp9000 已提交
888
	if (GetCurrentScene() == scene) {
B
BtbN 已提交
889
		for (int i = 0; i < ui->sources->count(); i++) {
J
jp9000 已提交
890 891
			QListWidgetItem *listItem = ui->sources->item(i);
			QVariant userData = listItem->data(Qt::UserRole);
J
jp9000 已提交
892

J
jp9000 已提交
893
			if (userData.value<OBSSceneItem>() == item) {
J
jp9000 已提交
894
				delete listItem;
J
jp9000 已提交
895 896
				break;
			}
897 898
		}
	}
J
jp9000 已提交
899

900
	obs_source_t *source = obs_sceneitem_get_source(item);
J
jp9000 已提交
901 902

	int scenes = sourceSceneRefs[source] - 1;
903 904
	sourceSceneRefs[source] = scenes;

J
jp9000 已提交
905 906 907 908
	if (scenes == 0) {
		obs_source_remove(source);
		sourceSceneRefs.erase(source);
	}
909 910
}

911
void OBSBasic::UpdateSceneSelection(OBSSource source)
912 913
{
	if (source) {
914
		obs_scene_t *scene = obs_scene_from_source(source);
915
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
916

917 918 919
		if (!scene)
			return;

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

923 924 925 926 927
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

J
jp9000 已提交
928
			UpdateSources(scene);
929
		}
J
jp9000 已提交
930
	}
931 932
}

J
jp9000 已提交
933 934 935 936 937 938 939 940 941 942 943 944 945
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);
946 947 948 949 950

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

953 954
void OBSBasic::SelectSceneItem(OBSScene scene, OBSSceneItem item, bool select)
{
955
	if (scene != GetCurrentScene())
956 957 958 959 960 961 962 963 964 965 966
		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;

967 968 969 970 971 972
		if (select) {
			ui->sources->setCurrentItem(witem);
		} else if (ui->sources->currentItem() == witem) {
			ui->sources->setCurrentItem(nullptr);
		}

973 974 975 976
		break;
	}
}

J
jp9000 已提交
977
void OBSBasic::MoveSceneItem(OBSSceneItem item, obs_order_movement movement)
J
jp9000 已提交
978
{
J
jp9000 已提交
979
	OBSScene scene = obs_sceneitem_get_scene(item);
J
jp9000 已提交
980 981 982 983 984 985 986 987 988 989
	if (scene != GetCurrentScene())
		return;

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

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

	switch (movement) {
J
jp9000 已提交
990
	case OBS_ORDER_MOVE_UP:
J
jp9000 已提交
991 992 993 994
		if (curRow > 0)
			curRow--;
		break;

J
jp9000 已提交
995
	case OBS_ORDER_MOVE_DOWN:
J
jp9000 已提交
996 997 998 999
		if (curRow < ui->sources->count())
			curRow++;
		break;

J
jp9000 已提交
1000
	case OBS_ORDER_MOVE_TOP:
J
jp9000 已提交
1001 1002 1003
		curRow = 0;
		break;

J
jp9000 已提交
1004
	case OBS_ORDER_MOVE_BOTTOM:
J
jp9000 已提交
1005 1006 1007 1008 1009 1010 1011 1012
		curRow = ui->sources->count();
		break;
	}

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

1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
void OBSBasic::ActivateAudioSource(OBSSource source)
{
	VolControl *vol = new VolControl(source);

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

1032
bool OBSBasic::QueryRemoveSource(obs_source_t *source)
J
jp9000 已提交
1033
{
1034
	const char *name  = obs_source_get_name(source);
1035 1036 1037

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

1039
	QMessageBox remove_source(this);
1040 1041 1042
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
1043 1044 1045 1046 1047 1048
	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();
1049
}
J
jp9000 已提交
1050

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

P
Palana 已提交
1053 1054 1055 1056 1057
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
1058 1059
void OBSBasic::TimedCheckForUpdates()
{
P
Palana 已提交
1060 1061 1062 1063
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
#else
J
jp9000 已提交
1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079
	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 已提交
1080
#endif
J
jp9000 已提交
1081 1082 1083 1084
}

void OBSBasic::CheckForUpdates()
{
P
Palana 已提交
1085 1086 1087
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
#else
J
jp9000 已提交
1088 1089
	ui->actionCheckForUpdates->setEnabled(false);

1090 1091 1092 1093 1094 1095 1096
	string versionString("obs-basic ");
	versionString += App()->GetVersionString();

	QNetworkRequest request;
	request.setUrl(QUrl("https://obsproject.com/obs2_update/basic.json"));
	request.setRawHeader("User-Agent", versionString.c_str());

F
fryshorts 已提交
1097 1098
	QNetworkReply *reply = networkManager.get(request);
	connect(reply, SIGNAL(finished()),
J
jp9000 已提交
1099
			this, SLOT(updateFileFinished()));
P
Palana 已提交
1100
#endif
J
jp9000 已提交
1101 1102
}

J
jp9000 已提交
1103 1104
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
1105 1106
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
1107 1108 1109 1110
#else
#define VERSION_ENTRY "other"
#endif

J
jp9000 已提交
1111 1112 1113 1114
void OBSBasic::updateFileFinished()
{
	ui->actionCheckForUpdates->setEnabled(true);

F
fryshorts 已提交
1115 1116
	QNetworkReply *reply = qobject_cast<QNetworkReply *>(sender());
	if (!reply || reply->error()) {
J
jp9000 已提交
1117
		blog(LOG_WARNING, "Update check failed: %s",
F
fryshorts 已提交
1118
				QT_TO_UTF8(reply->errorString()));
J
jp9000 已提交
1119 1120 1121
		return;
	}

F
fryshorts 已提交
1122 1123
	QByteArray raw = reply->readAll();
	if (!raw.length())
J
jp9000 已提交
1124 1125
		return;

F
fryshorts 已提交
1126 1127
	obs_data_t *returnData  = obs_data_create_from_json(raw.constData());
	obs_data_t *versionData = obs_data_get_obj(returnData, VERSION_ENTRY);
J
jp9000 已提交
1128 1129 1130
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
1131 1132

	if (returnData && versionData && description && download) {
J
jp9000 已提交
1133 1134 1135
		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 已提交
1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166
		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);
J
jp9000 已提交
1167

F
fryshorts 已提交
1168
	reply->deleteLater();
J
jp9000 已提交
1169 1170
}

1171 1172 1173 1174
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1175
		obs_source_t *source = obs_scene_get_source(scene);
1176 1177 1178 1179 1180 1181 1182 1183 1184
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
1185
		obs_source_t *source = obs_sceneitem_get_source(item);
1186
		if (QueryRemoveSource(source))
J
jp9000 已提交
1187 1188 1189 1190
			obs_sceneitem_remove(item);
	}
}

1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215
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);
1216 1217 1218
					SetupVisibilityItem(ui->sources,
							listItem, item);

1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248
					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);
}

1249 1250
/* OBS Callbacks */

1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
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)));
}

1261
void OBSBasic::SceneItemAdded(void *data, calldata_t *params)
1262 1263 1264
{
	OBSBasic *window = static_cast<OBSBasic*>(data);

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

1267 1268
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1269 1270
}

1271
void OBSBasic::SceneItemRemoved(void *data, calldata_t *params)
1272
{
1273
	OBSBasic *window = static_cast<OBSBasic*>(data);
1274

1275
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
1276

1277 1278
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1279 1280
}

1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
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));
}

1305
void OBSBasic::SourceAdded(void *data, calldata_t *params)
1306
{
J
jp9000 已提交
1307
	OBSBasic *window = static_cast<OBSBasic*>(data);
1308
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1309

1310
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1311
		QMetaObject::invokeMethod(window,
1312 1313
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1314 1315
}

1316
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
1317
{
1318
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1319

1320
	if (obs_scene_from_source(source) != NULL)
1321 1322 1323
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1324 1325
}

1326
void OBSBasic::SourceActivated(void *data, calldata_t *params)
1327
{
1328
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1329 1330 1331 1332 1333 1334 1335 1336
	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)));
}

1337
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
1338
{
1339
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1340 1341 1342 1343 1344 1345 1346 1347
	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)));
}

1348
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
1349 1350 1351 1352 1353 1354 1355 1356 1357 1358
{
	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)));
}

1359
void OBSBasic::ChannelChanged(void *data, calldata_t *params)
1360
{
1361
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1362
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1363 1364

	if (channel == 0)
1365 1366 1367
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1368 1369
}

1370 1371 1372 1373 1374
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1375 1376 1377
	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");
1378 1379 1380

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1381
	gs_effect_set_vec4(color, &colorVal);
1382

1383 1384
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1385 1386 1387 1388 1389 1390 1391 1392
	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();
1393 1394
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1395 1396 1397 1398

	gs_load_vertexbuffer(nullptr);
}

1399 1400
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
1401
	OBSBasic *window = static_cast<OBSBasic*>(data);
1402 1403 1404 1405
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
1406 1407
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
1408 1409 1410

	gs_viewport_push();
	gs_projection_push();
1411 1412 1413

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

1414 1415
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
1416
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
1417
			window->previewCX, window->previewCY);
1418

1419 1420
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
1421
	obs_render_main_view();
1422
	gs_load_vertexbuffer(nullptr);
1423

1424 1425
	/* --------------------------------------- */

1426 1427 1428
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
1429 1430 1431 1432

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
1433
	gs_reset_viewport();
J
jp9000 已提交
1434 1435 1436

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

1437 1438
	/* --------------------------------------- */

1439 1440
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
1441 1442 1443

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
1444 1445
}

1446 1447
/* Main class functions */

1448
obs_service_t *OBSBasic::GetService()
1449 1450 1451 1452 1453 1454
{
	if (!service)
		service = obs_service_create("rtmp_common", NULL, NULL);
	return service;
}

1455
void OBSBasic::SetService(obs_service_t *newService)
1456 1457 1458 1459 1460 1461 1462 1463
{
	if (newService) {
		if (service)
			obs_service_destroy(service);
		service = newService;
	}
}

1464 1465 1466 1467 1468 1469 1470
bool OBSBasic::StreamingActive()
{
	if (!outputHandler)
		return false;
	return outputHandler->StreamingActive();
}

1471 1472 1473 1474 1475 1476
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497
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;
}

1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510
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;
}

1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525
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;
#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 已提交
1526
		return VIDEO_FORMAT_RGBA;
1527 1528
}

1529
int OBSBasic::ResetVideo()
J
jp9000 已提交
1530 1531
{
	struct obs_video_info ovi;
1532
	int ret;
J
jp9000 已提交
1533

1534
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
1535

1536 1537 1538 1539 1540 1541 1542
	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 已提交
1543
	ovi.graphics_module = App()->GetRenderModule();
1544
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1545
			"Video", "BaseCX");
1546
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1547
			"Video", "BaseCY");
1548
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1549
			"Video", "OutputCX");
1550
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1551
			"Video", "OutputCY");
1552 1553 1554 1555 1556
	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 已提交
1557 1558
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
1559
	ovi.scale_type     = GetScaleType(basicConfig);
1560

J
jp9000 已提交
1561
	QTToGSWindow(ui->preview->winId(), ovi.window);
J
jp9000 已提交
1562 1563

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

1566
	QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1567 1568
	ovi.window_width  = size.width();
	ovi.window_height = size.height();
J
jp9000 已提交
1569

1570
	ret = AttemptToResetVideo(&ovi);
1571 1572
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
		/* Try OpenGL if DirectX fails on windows */
1573
		if (astrcmpi(ovi.graphics_module, DL_OPENGL) != 0) {
1574 1575 1576 1577
			blog(LOG_WARNING, "Failed to initialize obs video (%d) "
					  "with graphics_module='%s', retrying "
					  "with graphics_module='%s'",
					  ret, ovi.graphics_module,
1578 1579
					  DL_OPENGL);
			ovi.graphics_module = DL_OPENGL;
1580 1581 1582 1583
			ret = AttemptToResetVideo(&ovi);
		}
	}

1584 1585
	if (ret == OBS_VIDEO_SUCCESS)
		obs_add_draw_callback(OBSBasic::RenderMain, this);
1586

1587
	return ret;
J
jp9000 已提交
1588
}
J
jp9000 已提交
1589

1590
bool OBSBasic::ResetAudio()
J
jp9000 已提交
1591
{
1592
	struct obs_audio_info ai;
1593
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
1594 1595
			"SampleRate");

1596
	const char *channelSetupStr = config_get_string(basicConfig,
1597 1598 1599 1600 1601 1602 1603
			"Audio", "ChannelSetup");

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

1604
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
1605 1606

	return obs_reset_audio(&ai);
J
jp9000 已提交
1607 1608
}

J
jp9000 已提交
1609
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
1610
		const char *deviceDesc, int channel)
J
jp9000 已提交
1611 1612 1613
{
	const char *deviceId = config_get_string(basicConfig, "Audio",
			deviceName);
1614 1615
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
1616 1617 1618 1619
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
1620
		settings = obs_source_get_settings(source);
J
jp9000 已提交
1621
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632

		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) {
1633
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
1634
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
1635
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
1636
				sourceId, deviceDesc, settings);
J
jp9000 已提交
1637 1638 1639 1640 1641 1642 1643 1644
		obs_data_release(settings);

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

void OBSBasic::ResetAudioDevices()
J
jp9000 已提交
1645
{
1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
	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 已提交
1656 1657
}

J
jp9000 已提交
1658
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
1659
{
1660
	QSize  targetSize;
J
jp9000 已提交
1661

1662
	/* resize preview panel to fix to the top section of the window */
1663
	targetSize = GetPixelSize(ui->preview);
1664
	GetScaleAndCenterPos(int(cx), int(cy),
1665 1666
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
1667
			previewX, previewY, previewScale);
J
jp9000 已提交
1668

1669 1670 1671
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);

J
jp9000 已提交
1672 1673 1674 1675 1676
	if (isVisible()) {
		if (resizeTimer)
			killTimer(resizeTimer);
		resizeTimer = startTimer(100);
	}
J
jp9000 已提交
1677 1678
}

J
jp9000 已提交
1679
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
1680
{
1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691
	if (outputHandler && outputHandler->Active()) {
		QMessageBox::StandardButton button = QMessageBox::question(
				this, QTStr("ConfirmExit.Title"),
				QTStr("ConfirmExit.Text"));

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

1692 1693 1694 1695
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

1696 1697 1698 1699 1700 1701 1702 1703 1704 1705
	/* 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 已提交
1706 1707 1708 1709 1710
	for (QPointer<QWidget> &projector : projectors) {
		delete projector;
		projector.clear();
	}

1711 1712 1713
	// remove draw callback in case our drawable surfaces go away before
	// the destructor gets called
	obs_remove_draw_callback(OBSBasic::RenderMain, this);
1714 1715 1716 1717 1718

	/* Delete the save timer so it doesn't trigger after this point while
	 * the program data is being freed */
	delete saveTimer;
	SaveProject();
J
jp9000 已提交
1719 1720 1721 1722 1723 1724 1725

	/* 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();
1726 1727
}

J
jp9000 已提交
1728
void OBSBasic::changeEvent(QEvent *event)
1729
{
J
jp9000 已提交
1730 1731
	/* TODO */
	UNUSED_PARAMETER(event);
1732 1733
}

J
jp9000 已提交
1734
void OBSBasic::resizeEvent(QResizeEvent *event)
1735
{
J
jp9000 已提交
1736 1737 1738 1739
	struct obs_video_info ovi;

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

	UNUSED_PARAMETER(event);
1742 1743
}

J
jp9000 已提交
1744 1745 1746 1747 1748 1749
void OBSBasic::timerEvent(QTimerEvent *event)
{
	if (event->timerId() == resizeTimer) {
		killTimer(resizeTimer);
		resizeTimer = 0;

1750
		QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1751 1752 1753 1754
		obs_resize(size.width(), size.height());
	}
}

J
jp9000 已提交
1755
void OBSBasic::on_action_New_triggered()
1756
{
J
jp9000 已提交
1757
	/* TODO */
1758 1759
}

J
jp9000 已提交
1760
void OBSBasic::on_action_Open_triggered()
1761
{
J
jp9000 已提交
1762
	/* TODO */
1763 1764
}

J
jp9000 已提交
1765
void OBSBasic::on_action_Save_triggered()
1766
{
J
jp9000 已提交
1767
	/* TODO */
1768 1769
}

1770 1771 1772 1773 1774 1775 1776
void OBSBasic::on_actionShow_Recordings_triggered()
{
	const char *path = config_get_string(basicConfig,
			"SimpleOutput", "FilePath");
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
1777 1778 1779 1780 1781 1782 1783 1784
void OBSBasic::on_actionRemux_triggered()
{
	const char *path = config_get_string(basicConfig,
			"SimpleOutput", "FilePath");
	OBSRemux remux(path, this);
	remux.exec();
}

P
Palana 已提交
1785 1786 1787 1788 1789 1790
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

J
jp9000 已提交
1791 1792
void OBSBasic::on_actionAdvAudioProperties_triggered()
{
1793 1794 1795 1796 1797
	if (advAudioWindow != nullptr) {
		advAudioWindow->raise();
		return;
	}

J
jp9000 已提交
1798 1799 1800
	advAudioWindow = new OBSBasicAdvAudio(this);
	advAudioWindow->show();
	advAudioWindow->setAttribute(Qt::WA_DeleteOnClose, true);
1801 1802 1803

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

1806 1807 1808 1809 1810
void OBSBasic::on_advAudioProps_clicked()
{
	on_actionAdvAudioProperties_triggered();
}

1811 1812 1813 1814 1815
void OBSBasic::on_advAudioProps_destroyed()
{
	advAudioWindow = nullptr;
}

1816 1817
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1818
{
1819
	obs_source_t *source = NULL;
J
jp9000 已提交
1820

1821 1822 1823 1824
	if (sceneChanging)
		return;

	if (current) {
1825
		obs_scene_t *scene;
J
jp9000 已提交
1826

1827
		scene = current->data(Qt::UserRole).value<OBSScene>();
1828
		source = obs_scene_get_source(scene);
1829 1830
	}

1831
	/* TODO: allow transitions */
1832
	obs_set_output_source(0, source);
1833 1834

	UNUSED_PARAMETER(prev);
1835 1836
}

J
jp9000 已提交
1837 1838
void OBSBasic::EditSceneName()
{
1839 1840 1841 1842 1843 1844
	QListWidgetItem *item = ui->scenes->currentItem();
	Qt::ItemFlags flags   = item->flags();

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

J
jp9000 已提交
1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869
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 已提交
1870
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
1871
{
J
jp9000 已提交
1872
	QListWidgetItem *item = ui->scenes->itemAt(pos);
J
jp9000 已提交
1873
	QPointer<QMenu> sceneProjectorMenu;
J
jp9000 已提交
1874

1875
	QMenu popup(this);
J
jp9000 已提交
1876 1877 1878
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

P
Palana 已提交
1879 1880 1881 1882
	if (item) {
		popup.addSeparator();
		popup.addAction(QTStr("Rename"),
				this, SLOT(EditSceneName()));
J
jp9000 已提交
1883
		popup.addAction(QTStr("Remove"),
1884 1885
				this, SLOT(RemoveSelectedScene()),
				DeleteKeys.front());
J
jp9000 已提交
1886
		popup.addSeparator();
J
jp9000 已提交
1887 1888 1889 1890 1891
		sceneProjectorMenu = new QMenu(QTStr("SceneProjector"));
		AddProjectorMenuMonitors(sceneProjectorMenu, this,
				SLOT(OpenSceneProjector()));
		popup.addMenu(sceneProjectorMenu);
		popup.addSeparator();
J
jp9000 已提交
1892 1893
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenSceneFilters()));
P
Palana 已提交
1894
	}
J
jp9000 已提交
1895 1896

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

J
jp9000 已提交
1899
void OBSBasic::on_actionAddScene_triggered()
1900
{
1901
	string name;
S
Socapex 已提交
1902
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
1903 1904 1905

	int i = 1;
	QString placeHolderText = format.arg(i);
1906
	obs_source_t *source = nullptr;
P
Palana 已提交
1907 1908
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
1909
		placeHolderText = format.arg(++i);
P
Palana 已提交
1910
	}
S
Socapex 已提交
1911

J
jp9000 已提交
1912
	bool accepted = NameDialog::AskForName(this,
1913 1914
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
1915 1916
			name,
			placeHolderText);
1917

J
jp9000 已提交
1918
	if (accepted) {
J
jp9000 已提交
1919 1920
		if (name.empty()) {
			QMessageBox::information(this,
1921 1922
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
1923 1924 1925 1926
			on_actionAddScene_triggered();
			return;
		}

1927
		obs_source_t *source = obs_get_source_by_name(name.c_str());
1928
		if (source) {
J
jp9000 已提交
1929
			QMessageBox::information(this,
1930 1931
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
1932 1933

			obs_source_release(source);
J
jp9000 已提交
1934
			on_actionAddScene_triggered();
1935 1936 1937
			return;
		}

1938
		obs_scene_t *scene = obs_scene_create(name.c_str());
1939
		source = obs_scene_get_source(scene);
1940
		obs_add_source(source);
1941
		obs_scene_release(scene);
1942 1943

		obs_set_output_source(0, source);
1944
	}
1945 1946
}

J
jp9000 已提交
1947
void OBSBasic::on_actionRemoveScene_triggered()
1948
{
1949
	OBSScene     scene  = GetCurrentScene();
1950
	obs_source_t *source = obs_scene_get_source(scene);
1951 1952 1953

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
1954 1955
}

J
jp9000 已提交
1956
void OBSBasic::on_actionSceneProperties_triggered()
1957
{
J
jp9000 已提交
1958
	/* TODO */
1959 1960
}

J
jp9000 已提交
1961
void OBSBasic::on_actionSceneUp_triggered()
1962
{
J
jp9000 已提交
1963
	/* TODO */
1964 1965
}

J
jp9000 已提交
1966
void OBSBasic::on_actionSceneDown_triggered()
1967
{
J
jp9000 已提交
1968
	/* TODO */
1969 1970
}

1971 1972
void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1973
{
1974
	auto select_one = [] (obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
1975 1976
			void *param)
	{
1977
		obs_sceneitem_t *selectedItem =
J
jp9000 已提交
1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988
			*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>();
1989
	obs_source_t *source = obs_sceneitem_get_source(item);
1990 1991 1992
	if ((obs_source_get_output_flags(source) & OBS_SOURCE_VIDEO) == 0)
		return;

J
jp9000 已提交
1993 1994
	obs_scene_enum_items(GetCurrentScene(), select_one, &item);

1995
	UNUSED_PARAMETER(prev);
1996 1997
}

J
jp9000 已提交
1998 1999
void OBSBasic::EditSceneItemName()
{
2000 2001
	QListWidgetItem *item = ui->sources->currentItem();
	Qt::ItemFlags flags   = item->flags();
2002 2003 2004
	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);
2005

2006
	item->setText(QT_UTF8(name));
2007
	item->setFlags(flags | Qt::ItemIsEditable);
2008
	ui->sources->removeItemWidget(item);
2009 2010
	ui->sources->editItem(item);
	item->setFlags(flags);
J
jp9000 已提交
2011 2012
}

J
jp9000 已提交
2013
void OBSBasic::CreateSourcePopupMenu(QListWidgetItem *item, bool preview)
2014
{
2015
	QMenu popup(this);
J
jp9000 已提交
2016 2017
	QPointer<QMenu> previewProjector;
	QPointer<QMenu> sourceProjector;
J
jp9000 已提交
2018 2019 2020 2021 2022 2023 2024 2025

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

J
jp9000 已提交
2026 2027 2028 2029 2030 2031
		previewProjector = new QMenu(QTStr("PreviewProjector"));
		AddProjectorMenuMonitors(previewProjector, this,
				SLOT(OpenPreviewProjector()));

		popup.addMenu(previewProjector);

J
jp9000 已提交
2032 2033 2034
		popup.addSeparator();
	}

J
jp9000 已提交
2035 2036 2037 2038 2039 2040 2041 2042
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

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

J
John Bradley 已提交
2043
		OBSSceneItem sceneItem = GetSceneItem(item);
2044
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
John Bradley 已提交
2045 2046
		QAction *action;

J
jp9000 已提交
2047 2048
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
2049 2050
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
2051
				DeleteKeys.front());
J
jp9000 已提交
2052 2053
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
2054
		popup.addMenu(ui->transformMenu);
J
jp9000 已提交
2055 2056 2057 2058 2059 2060 2061

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

		popup.addSeparator();
		popup.addMenu(sourceProjector);
J
jp9000 已提交
2062
		popup.addSeparator();
J
John Bradley 已提交
2063 2064 2065 2066 2067 2068 2069

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

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

J
jp9000 已提交
2070 2071
		popup.addAction(QTStr("Filters"), this,
				SLOT(OpenFilters()));
J
jp9000 已提交
2072 2073 2074 2075 2076
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

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

J
jp9000 已提交
2079 2080 2081 2082 2083
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
{
	CreateSourcePopupMenu(ui->sources->itemAt(pos), false);
}

P
Palana 已提交
2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095
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 已提交
2096
void OBSBasic::AddSource(const char *id)
2097
{
2098 2099 2100 2101
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
	}
2102 2103
}

2104
QMenu *OBSBasic::CreateAddSourcePopupMenu()
2105
{
2106
	const char *type;
J
jp9000 已提交
2107 2108
	bool foundValues = false;
	size_t idx = 0;
2109

2110
	QMenu *popup = new QMenu(QTStr("Add"), this);
J
jp9000 已提交
2111
	while (obs_enum_input_types(idx++, &type)) {
2112
		const char *name = obs_source_get_display_name(
2113
				OBS_SOURCE_TYPE_INPUT, type);
2114

2115 2116 2117
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
2118 2119
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
2120 2121 2122
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
2123

J
jp9000 已提交
2124
		foundValues = true;
2125 2126
	}

2127 2128 2129
	if (!foundValues) {
		delete popup;
		popup = nullptr;
2130
	}
2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156

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

J
jp9000 已提交
2159
void OBSBasic::on_actionAddSource_triggered()
2160
{
J
jp9000 已提交
2161
	AddSourcePopupMenu(QCursor::pos());
2162 2163
}

J
jp9000 已提交
2164
void OBSBasic::on_actionRemoveSource_triggered()
2165
{
2166
	OBSSceneItem item   = GetCurrentSceneItem();
2167
	obs_source_t *source = obs_sceneitem_get_source(item);
2168 2169

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
2170
		obs_sceneitem_remove(item);
2171 2172
}

J
John Bradley 已提交
2173 2174 2175 2176 2177 2178 2179 2180 2181
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
2182
void OBSBasic::on_actionSourceProperties_triggered()
2183
{
2184
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2185
	OBSSource source = obs_sceneitem_get_source(item);
2186

2187 2188
	if (source)
		CreatePropertiesWindow(source);
2189 2190
}

J
jp9000 已提交
2191
void OBSBasic::on_actionSourceUp_triggered()
2192
{
J
jp9000 已提交
2193
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2194
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
2195
}
J
jp9000 已提交
2196

J
jp9000 已提交
2197
void OBSBasic::on_actionSourceDown_triggered()
2198
{
J
jp9000 已提交
2199
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2200
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
2201 2202
}

J
jp9000 已提交
2203 2204 2205
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2206
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
2207 2208 2209 2210 2211
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2212
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
2213 2214 2215 2216 2217
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2218
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
2219 2220 2221 2222 2223
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
2224
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
2225 2226
}

2227
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
2228
{
2229 2230 2231
	char logDir[512];
	if (os_get_config_path(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
		return nullptr;
J
jp9000 已提交
2232 2233 2234 2235 2236

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

2237
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
2238 2239 2240 2241 2242 2243 2244 2245
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

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

2248
	if (!fileString)
J
jp9000 已提交
2249 2250
		return;

2251
	if (!*fileString)
J
jp9000 已提交
2252 2253 2254 2255
		return;

	ui->menuLogFiles->setEnabled(false);

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

2259 2260 2261
	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 已提交
2262

2263
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
2264

2265 2266 2267 2268 2269 2270 2271 2272 2273 2274
	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 已提交
2275
	if (!json) {
2276 2277 2278 2279
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

F
fryshorts 已提交
2280 2281 2282 2283 2284 2285 2286 2287
	QBuffer *postData = new QBuffer();
	postData->setData(json, (int) strlen(json));

	QNetworkRequest postReq(QUrl("https://api.github.com/gists"));
	postReq.setHeader(QNetworkRequest::ContentTypeHeader,
			"application/json");

	QNetworkReply *reply = networkManager.post(postReq, postData);
J
jp9000 已提交
2288

F
fryshorts 已提交
2289 2290 2291 2292
	/* set the reply as parent, so the buffer is deleted with the reply */
	postData->setParent(reply);

	connect(reply, SIGNAL(finished()),
J
jp9000 已提交
2293 2294 2295
			this, SLOT(logUploadFinished()));
}

P
Palana 已提交
2296 2297
void OBSBasic::on_actionShowLogs_triggered()
{
2298 2299 2300 2301
	char logDir[512];
	if (os_get_config_path(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
		return;

P
Palana 已提交
2302 2303 2304 2305
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
2306 2307 2308 2309 2310 2311 2312 2313 2314 2315
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

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

2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331
void OBSBasic::on_actionViewCurrentLog_triggered()
{
	char logDir[512];
	if (os_get_config_path(logDir, sizeof(logDir), "obs-studio/logs") <= 0)
		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 已提交
2332 2333 2334 2335 2336
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

J
jp9000 已提交
2337 2338 2339 2340
void OBSBasic::logUploadFinished()
{
	ui->menuLogFiles->setEnabled(true);

F
fryshorts 已提交
2341 2342
	QNetworkReply *reply = qobject_cast<QNetworkReply *>(sender());
	if (!reply || reply->error()) {
J
jp9000 已提交
2343 2344
		QMessageBox::information(this,
				QTStr("LogReturnDialog.ErrorUploadingLog"),
F
fryshorts 已提交
2345
				reply->errorString());
J
jp9000 已提交
2346 2347 2348
		return;
	}

F
fryshorts 已提交
2349 2350
	QByteArray raw = reply->readAll();
	if (!raw.length())
J
jp9000 已提交
2351 2352
		return;

F
fryshorts 已提交
2353 2354
	obs_data_t *returnData = obs_data_create_from_json(raw.constData());
	QString logURL         = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
2355 2356 2357 2358
	obs_data_release(returnData);

	OBSLogReply logDialog(this, logURL);
	logDialog.exec();
J
jp9000 已提交
2359

F
fryshorts 已提交
2360
	reply->deleteLater();
J
jp9000 已提交
2361 2362
}

J
jp9000 已提交
2363
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
2364
		obs_source_t *source, const string &name)
2365
{
2366 2367 2368 2369
	const char *prevName = obs_source_get_name(source);
	if (name == prevName)
		return;

2370
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
2371
	QListWidgetItem *listItem    = listWidget->currentItem();
2372

2373
	if (foundSource || name.empty()) {
2374
		listItem->setText(QT_UTF8(prevName));
2375

2376
		if (foundSource) {
2377 2378 2379 2380 2381 2382 2383 2384 2385
			QMessageBox::information(parent,
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
			QMessageBox::information(parent,
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

2386 2387 2388
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
2389
		obs_source_set_name(source, name.c_str());
2390 2391 2392
	}
}

J
jp9000 已提交
2393 2394 2395 2396 2397
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
2398
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
2399 2400 2401 2402

	if (!scene)
		return;

2403
	obs_source_t *source = obs_scene_get_source(scene);
2404
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
2405 2406 2407 2408 2409 2410 2411 2412 2413

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
2414
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
2415 2416 2417 2418

	if (!item)
		return;

2419
	obs_source_t *source = obs_sceneitem_get_source(item);
2420
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
2421

2422 2423 2424 2425
	QListWidgetItem *listItem = ui->sources->currentItem();
	listItem->setText(QString());
	SetupVisibilityItem(ui->sources, listItem, item);

J
jp9000 已提交
2426 2427 2428
	UNUSED_PARAMETER(endHint);
}

J
jp9000 已提交
2429 2430 2431 2432 2433 2434 2435 2436
void OBSBasic::OpenFilters()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	CreateFiltersWindow(source);
}

J
jp9000 已提交
2437 2438 2439 2440 2441 2442 2443 2444
void OBSBasic::OpenSceneFilters()
{
	OBSScene scene = GetCurrentScene();
	OBSSource source = obs_scene_get_source(scene);

	CreateFiltersWindow(source);
}

2445
void OBSBasic::StreamingStart()
2446
{
2447
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
2448
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
2449
	ui->statusbar->StreamStarted(outputHandler->streamOutput);
2450 2451
}

2452
void OBSBasic::StreamingStop(int code)
2453
{
2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468
	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;

2469
	default:
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479
	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 已提交
2480
	ui->statusbar->StreamStopped();
2481

2482
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
2483
	ui->streamButton->setEnabled(true);
2484 2485 2486 2487 2488

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

P
Palana 已提交
2491 2492
void OBSBasic::RecordingStart()
{
J
jp9000 已提交
2493
	ui->statusbar->RecordingStarted(outputHandler->fileOutput);
P
Palana 已提交
2494 2495
}

2496
void OBSBasic::RecordingStop()
2497
{
P
Palana 已提交
2498
	ui->statusbar->RecordingStopped();
2499 2500
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
}
2501

2502 2503
void OBSBasic::on_streamButton_clicked()
{
J
jp9000 已提交
2504 2505
	SaveProject();

J
jp9000 已提交
2506 2507
	if (outputHandler->StreamingActive()) {
		outputHandler->StopStreaming();
2508
	} else {
J
jp9000 已提交
2509
		if (outputHandler->StartStreaming(service)) {
2510 2511 2512 2513 2514 2515 2516 2517 2518
			ui->streamButton->setEnabled(false);
			ui->streamButton->setText(
					QTStr("Basic.Main.Connecting"));
		}
	}
}

void OBSBasic::on_recordButton_clicked()
{
J
jp9000 已提交
2519 2520
	SaveProject();

J
jp9000 已提交
2521 2522
	if (outputHandler->RecordingActive()) {
		outputHandler->StopRecording();
2523 2524
	} else {

J
jp9000 已提交
2525
		if (outputHandler->StartRecording()) {
2526 2527 2528
			ui->recordButton->setText(
					QTStr("Basic.Main.StopRecording"));
		}
J
jp9000 已提交
2529 2530 2531
	}
}

J
jp9000 已提交
2532
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
2533
{
2534 2535
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
2536
}
2537

J
jp9000 已提交
2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548
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 已提交
2549
	QPointer<QMenu> previewProjector;
J
jp9000 已提交
2550 2551 2552 2553 2554 2555 2556

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

J
jp9000 已提交
2557 2558 2559 2560 2561
	previewProjector = new QMenu(QTStr("PreviewProjector"));
	AddProjectorMenuMonitors(previewProjector, this,
			SLOT(OpenPreviewProjector()));

	popup.addMenu(previewProjector);
J
jp9000 已提交
2562 2563 2564 2565 2566
	popup.exec(QCursor::pos());

	UNUSED_PARAMETER(pos);
}

2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629
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);
}

2630
config_t *OBSBasic::Config() const
2631 2632 2633
{
	return basicConfig;
}
J
jp9000 已提交
2634 2635 2636

void OBSBasic::on_actionEditTransform_triggered()
{
2637 2638 2639
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
2640 2641
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
2642
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
2643 2644 2645 2646
}

void OBSBasic::on_actionResetTransform_triggered()
{
2647
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
2648 2649 2650 2651
	{
		if (!obs_sceneitem_selected(item))
			return true;

2652
		obs_transform_info info;
J
jp9000 已提交
2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669
		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);
}

2670
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
2671 2672 2673 2674 2675
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

2678
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
2679 2680 2681 2682
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
2683 2684
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
2685 2686
	};

2687 2688 2689 2690 2691 2692
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

2693
static vec3 GetItemTL(obs_sceneitem_t *item)
2694 2695 2696
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
2697 2698 2699
	return tl;
}

2700
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
2701 2702 2703 2704
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
2705
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
2706 2707 2708
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
2709
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
2710 2711
}

2712
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2713 2714 2715 2716 2717 2718 2719 2720 2721
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
2722
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
2723 2724
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
2725
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751

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

2752
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2753 2754 2755 2756 2757 2758 2759 2760 2761 2762
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
2763
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
2764
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
2765
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
2766 2767

	SetItemTL(item, tl);
J
jp9000 已提交
2768 2769

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
2770 2771 2772 2773 2774
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
2775 2776
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
2777 2778 2779 2780 2781 2782
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
2783 2784
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
2785 2786 2787 2788
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

2789
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2790 2791 2792 2793 2794 2795 2796 2797 2798 2799
		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);

2800
	obs_transform_info itemInfo;
J
jp9000 已提交
2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832
	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()
{
2833
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858
	{
		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 已提交
2859
		UNUSED_PARAMETER(param);
2860 2861 2862 2863
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
J
jp9000 已提交
2864
}
J
jp9000 已提交
2865 2866 2867 2868 2869 2870 2871 2872

void OBSBasic::TogglePreview()
{
	bool enabled = !obs_preview_enabled();
	obs_preview_set_enabled(enabled);
	ui->preview->setVisible(enabled);
	ui->previewDisabledLabel->setVisible(!enabled);
}
2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883

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 已提交
2884
		struct vec2 dir;
2885 2886
		struct vec2 pos;

J
jp9000 已提交
2887 2888
		vec2_set(&dir, 0.0f, 0.0f);

2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911
		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 已提交
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

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