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

    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
7
    the Free Software Foundation, either version 2 of the License, or
8 9 10
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
11
    but WITHOUT ANY WARRANTY; without even the implied warranty of
12 13 14 15 16 17 18
    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 已提交
19
#include <time.h>
J
jp9000 已提交
20
#include <obs.hpp>
J
jp9000 已提交
21
#include <QMessageBox>
22
#include <QShowEvent>
23
#include <QDesktopServices>
J
jp9000 已提交
24
#include <QFileDialog>
J
jp9000 已提交
25 26
#include <QNetworkRequest>
#include <QNetworkReply>
27

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

33
#include "obs-app.hpp"
34
#include "platform.hpp"
35
#include "window-basic-settings.hpp"
36
#include "window-namedialog.hpp"
J
jp9000 已提交
37
#include "window-basic-source-select.hpp"
J
jp9000 已提交
38
#include "window-basic-main.hpp"
39
#include "window-basic-properties.hpp"
J
jp9000 已提交
40
#include "window-log-reply.hpp"
J
jp9000 已提交
41
#include "qt-wrappers.hpp"
42
#include "display-helpers.hpp"
43
#include "volume-control.hpp"
44

J
jp9000 已提交
45
#include "ui_OBSBasic.h"
46

J
jp9000 已提交
47
#include <fstream>
48 49
#include <sstream>

50 51 52
#include <QScreen>
#include <QWindow>

53 54
#define PREVIEW_EDGE_SIZE 10

55
using namespace std;
J
jp9000 已提交
56

J
jp9000 已提交
57 58
Q_DECLARE_METATYPE(OBSScene);
Q_DECLARE_METATYPE(OBSSceneItem);
J
jp9000 已提交
59
Q_DECLARE_METATYPE(obs_order_movement);
J
jp9000 已提交
60

61 62
static void AddExtraModulePaths()
{
B
BtbN 已提交
63 64 65
	BPtr<char> base_module_dir =
		os_get_config_path("obs-studio/plugins/%module%");

66 67 68 69 70 71 72
	if (!base_module_dir)
		return;

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

73
OBSBasic::OBSBasic(QWidget *parent)
74
	: OBSMainWindow  (parent),
J
jp9000 已提交
75
	  ui             (new Ui::OBSBasic)
76 77
{
	ui->setupUi(this);
78 79 80 81 82 83 84

	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 已提交
85 86 87 88

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

J
jp9000 已提交
89
	blog(LOG_INFO, "%s", name.str().c_str());
J
jp9000 已提交
90
	setWindowTitle(QT_UTF8(name.str().c_str()));
J
jp9000 已提交
91 92 93 94 95 96 97 98 99 100 101 102 103 104

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

	cpuUsageInfo = os_cpu_usage_info_start();
J
jp9000 已提交
107 108 109 110
	cpuUsageTimer = new QTimer(this);
	connect(cpuUsageTimer, SIGNAL(timeout()),
			ui->statusbar, SLOT(UpdateCPUUsage()));
	cpuUsageTimer->start(3000);
111 112 113 114 115 116

#ifdef __APPLE__
	QList<QKeySequence> keys;
	keys.append(QKeySequence::Delete);
	keys.append(QKeySequence(Qt::Key_Backspace));
	ui->actionRemoveSource->setShortcuts(keys);
117 118 119

	ui->action_Settings->setMenuRole(QAction::PreferencesRole);
	ui->actionE_xit->setMenuRole(QAction::QuitRole);
120
#endif
121 122
}

123 124 125 126 127 128 129 130
static void SaveAudioDevice(const char *name, int channel, obs_data_t parent)
{
	obs_source_t source = obs_get_output_source(channel);
	if (!source)
		return;

	obs_data_t data = obs_save_source(source);

J
jp9000 已提交
131
	obs_data_set_obj(parent, name, data);
132 133 134 135 136

	obs_data_release(data);
	obs_source_release(source);
}

137 138 139 140 141
static obs_data_t GenerateSaveData()
{
	obs_data_t       saveData     = obs_data_create();
	obs_data_array_t sourcesArray = obs_save_sources();
	obs_source_t     currentScene = obs_get_output_source(0);
142
	const char       *sceneName   = obs_source_get_name(currentScene);
143

144 145 146 147 148 149
	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 已提交
150 151
	obs_data_set_string(saveData, "current_scene", sceneName);
	obs_data_set_array(saveData, "sources", sourcesArray);
152 153 154 155 156 157
	obs_data_array_release(sourcesArray);
	obs_source_release(currentScene);

	return saveData;
}

158 159 160 161 162 163 164 165 166 167 168 169
void OBSBasic::ClearVolumeControls()
{
	VolControl *control;

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

	volumes.clear();
}

170 171 172
void OBSBasic::Save(const char *file)
{
	obs_data_t saveData  = GenerateSaveData();
J
jp9000 已提交
173
	const char *jsonData = obs_data_get_json(saveData);
174 175 176 177 178 179 180 181

	/* TODO maybe a message box here? */
	if (!os_quick_write_utf8_file(file, jsonData, strlen(jsonData), false))
		blog(LOG_ERROR, "Could not save scene data to %s", file);

	obs_data_release(saveData);
}

182 183
static void LoadAudioDevice(const char *name, int channel, obs_data_t parent)
{
J
jp9000 已提交
184
	obs_data_t data = obs_data_get_obj(parent, name);
185 186 187 188 189 190 191 192 193 194 195 196 197 198
	if (!data)
		return;

	obs_source_t source = obs_load_source(data);
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

void OBSBasic::CreateDefaultScene()
{
199
	obs_scene_t  scene  = obs_scene_create(Str("Basic.Scene"));
200
	obs_source_t source = obs_scene_get_source(scene);
201 202 203 204 205

	obs_add_source(source);

#ifdef __APPLE__
	source = obs_source_create(OBS_SOURCE_TYPE_INPUT, "display_capture",
206
			Str("Basic.DisplayCapture"), NULL);
207 208 209 210 211 212 213 214

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

215
	obs_set_output_source(0, obs_scene_get_source(scene));
216 217 218
	obs_scene_release(scene);
}

219 220 221 222 223 224 225 226
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);
227 228
	if (!jsonData) {
		CreateDefaultScene();
229
		return;
230
	}
231 232

	obs_data_t       data       = obs_data_create_from_json(jsonData);
J
jp9000 已提交
233 234 235
	obs_data_array_t sources    = obs_data_get_array(data, "sources");
	const char       *sceneName = obs_data_get_string(data,
			"current_scene");
236 237
	obs_source_t     curScene;

238 239 240 241 242 243
	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);

244 245 246 247 248 249 250 251 252 253
	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);
}

254 255 256
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
257
	obs_properties_t props = obs_get_source_properties(
258
			OBS_SOURCE_TYPE_INPUT, output_id);
259 260 261 262 263 264 265 266 267 268 269 270 271 272
	size_t count = 0;

	if (!props)
		return false;

	obs_property_t devices = obs_properties_get(props, "device_id");
	if (devices)
		count = obs_property_list_item_count(devices);

	obs_properties_destroy(props);

	return count != 0;
}

273 274 275 276 277 278 279 280 281
static void OBSStartStreaming(void *data, calldata_t params)
{
	UNUSED_PARAMETER(params);
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"StreamingStart");
}

static void OBSStopStreaming(void *data, calldata_t params)
{
282
	int code = (int)calldata_int(params, "code");
283 284 285 286
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"StreamingStop", Q_ARG(int, code));
}

P
Palana 已提交
287 288 289 290 291 292 293 294
static void OBSStartRecording(void *data, calldata_t params)
{
	UNUSED_PARAMETER(params);

	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"RecordingStart");
}

295 296 297 298 299 300 301 302
static void OBSStopRecording(void *data, calldata_t params)
{
	UNUSED_PARAMETER(params);

	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"RecordingStop");
}

303 304 305 306 307 308 309 310 311 312 313 314 315 316
#define SERVICE_PATH "obs-studio/basic/service.json"

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

	BPtr<char> serviceJsonPath(os_get_config_path(SERVICE_PATH));
	if (!serviceJsonPath)
		return;

	obs_data_t data     = obs_data_create();
	obs_data_t settings = obs_service_get_settings(service);

J
jp9000 已提交
317 318
	obs_data_set_string(data, "type", obs_service_gettype(service));
	obs_data_set_obj(data, "settings", settings);
319

J
jp9000 已提交
320
	const char *json = obs_data_get_json(data);
321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342

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

	obs_data_release(settings);
	obs_data_release(data);
}

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

	BPtr<char> serviceJsonPath(os_get_config_path(SERVICE_PATH));
	if (!serviceJsonPath)
		return false;

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

	obs_data_t data = obs_data_create_from_json(jsonText);

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

J
jp9000 已提交
345
	obs_data_t settings = obs_data_get_obj(data, "settings");
346

347
	service = obs_service_create(type, "default_service", settings);
348 349 350 351 352 353 354 355 356

	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

bool OBSBasic::InitOutputs()
{
357 358
	fileOutput = obs_output_create("flv_output", "default_file_output",
			nullptr);
359 360 361
	if (!fileOutput)
		return false;

362 363
	streamOutput = obs_output_create("rtmp_output", "default_stream",
			nullptr);
364 365 366
	if (!streamOutput)
		return false;

367
	signal_handler_connect(obs_output_get_signal_handler(streamOutput),
368
			"start", OBSStartStreaming, this);
369
	signal_handler_connect(obs_output_get_signal_handler(streamOutput),
370 371
			"stop", OBSStopStreaming, this);

P
Palana 已提交
372 373
	signal_handler_connect(obs_output_get_signal_handler(fileOutput),
			"start", OBSStartRecording, this);
374
	signal_handler_connect(obs_output_get_signal_handler(fileOutput),
375 376
			"stop", OBSStopRecording, this);

377 378 379 380 381
	return true;
}

bool OBSBasic::InitEncoders()
{
382
	x264 = obs_video_encoder_create("obs_x264", "default_h264", nullptr);
383 384 385
	if (!x264)
		return false;

386
	aac = obs_audio_encoder_create("libfdk_aac", "default_aac", nullptr);
B
BtbN 已提交
387

388 389 390
	if (!aac)
		aac = obs_audio_encoder_create("ffmpeg_aac", "default_aac",
				nullptr);
B
BtbN 已提交
391 392 393 394

	if (!aac)
		return false;

395 396 397 398 399 400 401 402
	return true;
}

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

403
	service = obs_service_create("rtmp_common", "default_service", nullptr);
404 405 406 407 408 409
	if (!service)
		return false;

	return true;
}

410 411
bool OBSBasic::InitBasicConfigDefaults()
{
412 413 414
	bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
	bool hasInputAudio   = HasAudioDevices(App()->InputAudioSource());

415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
	config_set_default_int(basicConfig, "Window", "PosX",  -1);
	config_set_default_int(basicConfig, "Window", "PosY",  -1);
	config_set_default_int(basicConfig, "Window", "SizeX", -1);
	config_set_default_int(basicConfig, "Window", "SizeY", -1);

	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;

432
	/* TODO: temporary */
433 434
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
435 436 437
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 128);
J
jp9000 已提交
438 439 440 441 442
	config_set_default_bool  (basicConfig, "SimpleOutput", "Reconnect",
			true);
	config_set_default_uint  (basicConfig, "SimpleOutput", "RetryDelay", 2);
	config_set_default_uint  (basicConfig, "SimpleOutput", "MaxRetries",
			20);
J
jp9000 已提交
443 444
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
445
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseCBR", true);
J
jp9000 已提交
446 447
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
448

449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
	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);

	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 已提交
468
	config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
469
			hasDesktopAudio ? "default" : "disabled");
J
jp9000 已提交
470 471 472
	config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice1",
473
			hasInputAudio ? "default" : "disabled");
J
jp9000 已提交
474 475 476 477 478
	config_set_default_string(basicConfig, "Audio", "AuxDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice3",
			"disabled");

479 480 481 482 483 484 485
	return true;
}

bool OBSBasic::InitBasicConfig()
{
	BPtr<char> configPath(os_get_config_path("obs-studio/basic/basic.ini"));

486 487 488
	int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
	if (code != CONFIG_SUCCESS) {
		OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
489 490 491 492 493 494
		return false;
	}

	return InitBasicConfigDefaults();
}

495 496
void OBSBasic::InitOBSCallbacks()
{
497
	signal_handler_connect(obs_get_signal_handler(), "source_add",
498
			OBSBasic::SourceAdded, this);
499
	signal_handler_connect(obs_get_signal_handler(), "source_remove",
500
			OBSBasic::SourceRemoved, this);
501
	signal_handler_connect(obs_get_signal_handler(), "channel_change",
502
			OBSBasic::ChannelChanged, this);
503
	signal_handler_connect(obs_get_signal_handler(), "source_activate",
504
			OBSBasic::SourceActivated, this);
505
	signal_handler_connect(obs_get_signal_handler(), "source_deactivate",
506
			OBSBasic::SourceDeactivated, this);
507
	signal_handler_connect(obs_get_signal_handler(), "source_rename",
J
jp9000 已提交
508
			OBSBasic::SourceRenamed, this);
509 510
}

J
jp9000 已提交
511 512
void OBSBasic::InitPrimitives()
{
J
jp9000 已提交
513
	obs_enter_graphics();
J
jp9000 已提交
514

515
	gs_render_start(true);
J
jp9000 已提交
516 517 518 519 520
	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);
521
	box = gs_render_save();
J
jp9000 已提交
522

523
	gs_render_start(true);
J
jp9000 已提交
524 525 526 527
	for (int i = 0; i <= 360; i += (360/20)) {
		float pos = RAD(float(i));
		gs_vertex2f(cosf(pos), sinf(pos));
	}
528
	circle = gs_render_save();
J
jp9000 已提交
529

J
jp9000 已提交
530
	obs_leave_graphics();
J
jp9000 已提交
531 532
}

533 534
void OBSBasic::OBSInit()
{
535 536
	BPtr<char> savePath(os_get_config_path("obs-studio/basic/scenes.json"));

537 538 539 540
	/* make sure it's fully displayed before doing any initialization */
	show();
	App()->processEvents();

541
	if (!obs_startup(App()->GetLocale()))
542
		throw "Failed to initialize libobs";
543 544
	if (!InitBasicConfig())
		throw "Failed to load basic.ini";
545
	if (!ResetAudio())
546 547
		throw "Failed to initialize audio";

548 549 550 551 552 553
	int ret = ResetVideo();

	switch (ret) {
	case OBS_VIDEO_MODULE_NOT_FOUND:
		throw "Failed to initialize video:  Graphics module not found";
	case OBS_VIDEO_NOT_SUPPORTED:
J
jp9000 已提交
554 555 556
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
557 558 559 560 561 562 563
	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";
	}

564
	InitOBSCallbacks();
565

566
	AddExtraModulePaths();
J
jp9000 已提交
567
	obs_load_all_modules();
J
jp9000 已提交
568

569 570 571 572 573 574 575
	if (!InitOutputs())
		throw "Failed to initialize outputs";
	if (!InitEncoders())
		throw "Failed to initialize encoders";
	if (!InitService())
		throw "Failed to initialize service";

J
jp9000 已提交
576 577
	InitPrimitives();

578
	Load(savePath);
J
jp9000 已提交
579
	ResetAudioDevices();
580

J
jp9000 已提交
581
	TimedCheckForUpdates();
582
	loaded = true;
583 584 585 586
}

OBSBasic::~OBSBasic()
{
587
	BPtr<char> savePath(os_get_config_path("obs-studio/basic/scenes.json"));
588
	SaveService();
589
	Save(savePath);
590

J
jp9000 已提交
591 592 593 594 595 596
	/* 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 已提交
597
	delete cpuUsageTimer;
598 599
	os_cpu_usage_info_destroy(cpuUsageInfo);

J
John Bradley 已提交
600 601 602
	if (interaction)
		delete interaction;

603 604 605 606 607
	if (properties)
		delete properties;

	if (transformWindow)
		delete transformWindow;
608

609
	ClearVolumeControls();
610 611
	ui->sources->clear();
	ui->scenes->clear();
J
jp9000 已提交
612

J
jp9000 已提交
613
	obs_enter_graphics();
614 615
	gs_vertexbuffer_destroy(box);
	gs_vertexbuffer_destroy(circle);
J
jp9000 已提交
616
	obs_leave_graphics();
J
jp9000 已提交
617

618
	obs_shutdown();
J
jp9000 已提交
619 620 621 622

	config_set_int(App()->GlobalConfig(), "General", "LastVersion",
			LIBOBS_API_VER);
	config_save(App()->GlobalConfig());
623 624
}

J
jp9000 已提交
625
OBSScene OBSBasic::GetCurrentScene()
626
{
J
jp9000 已提交
627
	QListWidgetItem *item = ui->scenes->currentItem();
J
jp9000 已提交
628
	return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
629 630
}

631
OBSSceneItem OBSBasic::GetSceneItem(QListWidgetItem *item)
J
jp9000 已提交
632
{
J
jp9000 已提交
633
	return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
J
jp9000 已提交
634 635
}

636 637 638 639 640
OBSSceneItem OBSBasic::GetCurrentSceneItem()
{
	return GetSceneItem(ui->sources->currentItem());
}

641 642 643 644 645 646 647 648
void OBSBasic::UpdateSources(OBSScene scene)
{
	ui->sources->clear();

	obs_scene_enum_items(scene,
			[] (obs_scene_t scene, obs_sceneitem_t item, void *p)
			{
				OBSBasic *window = static_cast<OBSBasic*>(p);
649
				window->InsertSceneItem(item);
J
jp9000 已提交
650 651

				UNUSED_PARAMETER(scene);
652 653 654 655
				return true;
			}, this);
}

656 657
void OBSBasic::InsertSceneItem(obs_sceneitem_t item)
{
J
jp9000 已提交
658
	obs_source_t source = obs_sceneitem_get_source(item);
659
	const char   *name  = obs_source_get_name(source);
660 661

	QListWidgetItem *listItem = new QListWidgetItem(QT_UTF8(name));
J
jp9000 已提交
662
	listItem->setFlags(listItem->flags() | Qt::ItemIsEditable);
663 664 665 666
	listItem->setData(Qt::UserRole,
			QVariant::fromValue(OBSSceneItem(item)));

	ui->sources->insertItem(0, listItem);
667 668 669
	ui->sources->setCurrentRow(0);

	/* if the source was just created, open properties dialog */
670 671 672 673
	if (sourceSceneRefs[source] == 0 && loaded)
		CreatePropertiesWindow(source);
}

J
John Bradley 已提交
674 675 676 677 678 679 680 681 682 683
void OBSBasic::CreateInteractionWindow(obs_source_t source)
{
	if (interaction)
		interaction->close();

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

684 685 686 687 688 689 690 691
void OBSBasic::CreatePropertiesWindow(obs_source_t source)
{
	if (properties)
		properties->close();

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

694 695 696
/* Qt callbacks for invokeMethod */

void OBSBasic::AddScene(OBSSource source)
697
{
698 699
	const char *name  = obs_source_get_name(source);
	obs_scene_t scene = obs_scene_from_source(source);
J
jp9000 已提交
700 701

	QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
J
jp9000 已提交
702
	item->setFlags(item->flags() | Qt::ItemIsEditable);
J
jp9000 已提交
703
	item->setData(Qt::UserRole, QVariant::fromValue(OBSScene(scene)));
J
jp9000 已提交
704
	ui->scenes->addItem(item);
705

706
	signal_handler_t handler = obs_source_get_signal_handler(source);
707
	signal_handler_connect(handler, "item_add",
J
jp9000 已提交
708
			OBSBasic::SceneItemAdded, this);
709
	signal_handler_connect(handler, "item_remove",
J
jp9000 已提交
710
			OBSBasic::SceneItemRemoved, this);
J
jp9000 已提交
711 712 713 714 715 716 717 718
	signal_handler_connect(handler, "item_move_up",
			OBSBasic::SceneItemMoveUp, this);
	signal_handler_connect(handler, "item_move_down",
			OBSBasic::SceneItemMoveDown, this);
	signal_handler_connect(handler, "item_move_top",
			OBSBasic::SceneItemMoveTop, this);
	signal_handler_connect(handler, "item_move_bottom",
			OBSBasic::SceneItemMoveBottom, this);
719 720
}

721
void OBSBasic::RemoveScene(OBSSource source)
J
jp9000 已提交
722
{
723
	const char *name = obs_source_get_name(source);
J
jp9000 已提交
724

J
jp9000 已提交
725 726 727
	QListWidgetItem *sel = ui->scenes->currentItem();
	QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
			Qt::MatchExactly);
J
jp9000 已提交
728

J
jp9000 已提交
729 730 731 732
	if (sel != nullptr) {
		if (items.contains(sel))
			ui->sources->clear();
		delete sel;
J
jp9000 已提交
733
	}
734 735
}

736
void OBSBasic::AddSceneItem(OBSSceneItem item)
737
{
J
jp9000 已提交
738 739
	obs_scene_t  scene  = obs_sceneitem_get_scene(item);
	obs_source_t source = obs_sceneitem_get_source(item);
J
jp9000 已提交
740

741 742
	if (GetCurrentScene() == scene)
		InsertSceneItem(item);
J
jp9000 已提交
743 744

	sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
745 746
}

747
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
748
{
J
jp9000 已提交
749
	obs_scene_t scene = obs_sceneitem_get_scene(item);
750

J
jp9000 已提交
751
	if (GetCurrentScene() == scene) {
B
BtbN 已提交
752
		for (int i = 0; i < ui->sources->count(); i++) {
J
jp9000 已提交
753 754
			QListWidgetItem *listItem = ui->sources->item(i);
			QVariant userData = listItem->data(Qt::UserRole);
J
jp9000 已提交
755

J
jp9000 已提交
756
			if (userData.value<OBSSceneItem>() == item) {
J
jp9000 已提交
757
				delete listItem;
J
jp9000 已提交
758 759
				break;
			}
760 761
		}
	}
J
jp9000 已提交
762

J
jp9000 已提交
763
	obs_source_t source = obs_sceneitem_get_source(item);
J
jp9000 已提交
764 765

	int scenes = sourceSceneRefs[source] - 1;
766 767
	sourceSceneRefs[source] = scenes;

J
jp9000 已提交
768 769 770 771
	if (scenes == 0) {
		obs_source_remove(source);
		sourceSceneRefs.erase(source);
	}
772 773
}

774
void OBSBasic::UpdateSceneSelection(OBSSource source)
775 776
{
	if (source) {
777 778
		obs_scene_t scene = obs_scene_from_source(source);
		const char *name = obs_source_get_name(source);
J
jp9000 已提交
779

780 781 782
		if (!scene)
			return;

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

786 787 788 789 790
		if (items.count()) {
			sceneChanging = true;
			ui->scenes->setCurrentItem(items.first());
			sceneChanging = false;

J
jp9000 已提交
791
			UpdateSources(scene);
792
		}
J
jp9000 已提交
793
	}
794 795
}

J
jp9000 已提交
796 797 798 799 800 801 802 803 804 805 806 807 808 809
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);
	RenameListValues(ui->sources, newName, prevName);
810 811 812 813 814

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

J
jp9000 已提交
817
void OBSBasic::MoveSceneItem(OBSSceneItem item, obs_order_movement movement)
J
jp9000 已提交
818
{
J
jp9000 已提交
819
	OBSScene scene = obs_sceneitem_get_scene(item);
J
jp9000 已提交
820 821 822 823 824 825 826 827 828 829
	if (scene != GetCurrentScene())
		return;

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

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

	switch (movement) {
J
jp9000 已提交
830
	case OBS_ORDER_MOVE_UP:
J
jp9000 已提交
831 832 833 834
		if (curRow > 0)
			curRow--;
		break;

J
jp9000 已提交
835
	case OBS_ORDER_MOVE_DOWN:
J
jp9000 已提交
836 837 838 839
		if (curRow < ui->sources->count())
			curRow++;
		break;

J
jp9000 已提交
840
	case OBS_ORDER_MOVE_TOP:
J
jp9000 已提交
841 842 843
		curRow = 0;
		break;

J
jp9000 已提交
844
	case OBS_ORDER_MOVE_BOTTOM:
J
jp9000 已提交
845 846 847 848 849 850 851 852
		curRow = ui->sources->count();
		break;
	}

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

853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871
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;
		}
	}
}

872
bool OBSBasic::QueryRemoveSource(obs_source_t source)
J
jp9000 已提交
873
{
874
	const char *name  = obs_source_get_name(source);
875 876 877

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

J
Jkoan 已提交
879
	QMessageBox remove_source;
880 881 882
	remove_source.setText(text);
	QAbstractButton *Yes = remove_source.addButton(QTStr("Yes"),
			QMessageBox::YesRole);
J
Jkoan 已提交
883 884 885 886 887 888
	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();
889
}
J
jp9000 已提交
890

J
jp9000 已提交
891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916
#define UPDATE_CHECK_INTERVAL (60*60*24*4) /* 4 days */

void OBSBasic::TimedCheckForUpdates()
{
	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();
}

void OBSBasic::CheckForUpdates()
{
	ui->actionCheckForUpdates->setEnabled(false);

917 918 919 920 921 922 923 924
	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());

	updateReply = networkManager.get(request);
J
jp9000 已提交
925 926 927 928 929 930 931 932 933 934 935
	connect(updateReply, SIGNAL(finished()),
			this, SLOT(updateFileFinished()));
	connect(updateReply, SIGNAL(readyRead()),
			this, SLOT(updateFileRead()));
}

void OBSBasic::updateFileRead()
{
	updateReturnData.push_back(updateReply->readAll());
}

J
jp9000 已提交
936 937
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
938 939
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
940 941 942 943
#else
#define VERSION_ENTRY "other"
#endif

J
jp9000 已提交
944 945 946 947 948 949 950 951 952 953 954 955 956 957 958
void OBSBasic::updateFileFinished()
{
	ui->actionCheckForUpdates->setEnabled(true);

	if (updateReply->error()) {
		blog(LOG_WARNING, "Update check failed: %s",
				QT_TO_UTF8(updateReply->errorString()));
		return;
	}

	const char *jsonReply = updateReturnData.constData();
	if (!jsonReply || !*jsonReply)
		return;

	obs_data_t returnData   = obs_data_create_from_json(jsonReply);
J
jp9000 已提交
959 960 961 962
	obs_data_t versionData  = obs_data_get_obj(returnData, VERSION_ENTRY);
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
963 964

	if (returnData && versionData && description && download) {
J
jp9000 已提交
965 966 967
		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 已提交
968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998
		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 已提交
999 1000

	updateReturnData.clear();
J
jp9000 已提交
1001 1002
}

1003 1004 1005 1006
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1007
		obs_source_t source = obs_scene_get_source(scene);
1008 1009 1010 1011 1012 1013 1014 1015 1016
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
J
jp9000 已提交
1017
		obs_source_t source = obs_sceneitem_get_source(item);
1018
		if (QueryRemoveSource(source))
J
jp9000 已提交
1019 1020 1021 1022
			obs_sceneitem_remove(item);
	}
}

1023 1024 1025 1026 1027 1028 1029
/* OBS Callbacks */

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

	obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
J
jp9000 已提交
1030

1031 1032
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1033 1034
}

1035
void OBSBasic::SceneItemRemoved(void *data, calldata_t params)
1036
{
1037
	OBSBasic *window = static_cast<OBSBasic*>(data);
1038

1039 1040
	obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");

1041 1042
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1043 1044 1045 1046
}

void OBSBasic::SourceAdded(void *data, calldata_t params)
{
J
jp9000 已提交
1047
	OBSBasic *window = static_cast<OBSBasic*>(data);
1048
	obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
1049

1050
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1051
		QMetaObject::invokeMethod(window,
1052 1053
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1054 1055
}

J
jp9000 已提交
1056
void OBSBasic::SourceRemoved(void *data, calldata_t params)
1057
{
1058
	obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
1059

1060
	if (obs_scene_from_source(source) != NULL)
1061 1062 1063
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1064 1065
}

1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
void OBSBasic::SourceActivated(void *data, calldata_t params)
{
	obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
	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)));
}

void OBSBasic::SourceDeactivated(void *data, calldata_t params)
{
	obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
	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)));
}

J
jp9000 已提交
1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
void OBSBasic::SourceRenamed(void *data, calldata_t params)
{
	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)));
}

1099 1100 1101
void OBSBasic::ChannelChanged(void *data, calldata_t params)
{
	obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
1102
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1103 1104

	if (channel == 0)
1105 1106 1107
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1108 1109
}

1110 1111 1112 1113 1114
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1115 1116 1117
	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");
1118 1119 1120

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1121
	gs_effect_set_vec4(color, &colorVal);
1122

1123 1124
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1125 1126 1127 1128 1129 1130 1131 1132
	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();
1133 1134
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1135 1136 1137 1138

	gs_load_vertexbuffer(nullptr);
}

1139 1140
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
1141
	OBSBasic *window = static_cast<OBSBasic*>(data);
1142 1143 1144 1145
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
1146 1147
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
1148 1149 1150

	gs_viewport_push();
	gs_projection_push();
1151 1152 1153

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

1154 1155
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
1156
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
1157
			window->previewCX, window->previewCY);
1158

1159 1160
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
1161
	obs_render_main_view();
1162
	gs_load_vertexbuffer(nullptr);
1163

1164 1165
	/* --------------------------------------- */

1166 1167 1168
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
1169 1170 1171 1172

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
1173
	gs_reset_viewport();
J
jp9000 已提交
1174 1175 1176

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

1177 1178
	/* --------------------------------------- */

1179 1180
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
1181 1182 1183

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
1184 1185
}

J
jp9000 已提交
1186 1187 1188 1189 1190 1191
void OBSBasic::SceneItemMoveUp(void *data, calldata_t params)
{
	OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1192
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_UP));
J
jp9000 已提交
1193 1194 1195 1196 1197 1198 1199 1200
}

void OBSBasic::SceneItemMoveDown(void *data, calldata_t params)
{
	OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1201
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_DOWN));
J
jp9000 已提交
1202 1203 1204 1205 1206 1207 1208 1209
}

void OBSBasic::SceneItemMoveTop(void *data, calldata_t params)
{
	OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1210
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_TOP));
J
jp9000 已提交
1211 1212 1213 1214 1215 1216 1217 1218
}

void OBSBasic::SceneItemMoveBottom(void *data, calldata_t params)
{
	OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1219
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_BOTTOM));
J
jp9000 已提交
1220 1221
}

1222 1223
/* Main class functions */

1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
obs_service_t OBSBasic::GetService()
{
	if (!service)
		service = obs_service_create("rtmp_common", NULL, NULL);
	return service;
}

void OBSBasic::SetService(obs_service_t newService)
{
	if (newService) {
		if (service)
			obs_service_destroy(service);
		service = newService;
	}
}

1240 1241 1242 1243 1244 1245
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266
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;
}

1267
int OBSBasic::ResetVideo()
J
jp9000 已提交
1268 1269
{
	struct obs_video_info ovi;
1270
	int ret;
J
jp9000 已提交
1271

1272
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
1273 1274

	ovi.graphics_module = App()->GetRenderModule();
1275
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1276
			"Video", "BaseCX");
1277
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1278
			"Video", "BaseCY");
1279
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1280
			"Video", "OutputCX");
1281
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1282
			"Video", "OutputCY");
1283
	ovi.output_format  = VIDEO_FORMAT_NV12;
J
jp9000 已提交
1284 1285
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
1286

J
jp9000 已提交
1287
	QTToGSWindow(ui->preview->winId(), ovi.window);
J
jp9000 已提交
1288 1289

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

1292
	QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1293 1294
	ovi.window_width  = size.width();
	ovi.window_height = size.height();
J
jp9000 已提交
1295

1296
	ret = AttemptToResetVideo(&ovi);
1297 1298 1299 1300 1301 1302 1303 1304
	if (IS_WIN32 && ret != OBS_VIDEO_SUCCESS) {
		/* Try OpenGL if DirectX fails on windows */
		if (astrcmpi(ovi.graphics_module, "libobs-opengl") != 0) {
			ovi.graphics_module = "libobs-opengl";
			ret = AttemptToResetVideo(&ovi);
		}
	}

1305 1306
	if (ret == OBS_VIDEO_SUCCESS)
		obs_add_draw_callback(OBSBasic::RenderMain, this);
1307

1308
	return ret;
J
jp9000 已提交
1309
}
J
jp9000 已提交
1310

1311
bool OBSBasic::ResetAudio()
J
jp9000 已提交
1312
{
J
jp9000 已提交
1313
	struct audio_output_info ai;
1314 1315 1316
	ai.name = "Main Audio Track";
	ai.format = AUDIO_FORMAT_FLOAT;

1317
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
1318 1319
			"SampleRate");

1320
	const char *channelSetupStr = config_get_string(basicConfig,
1321 1322 1323 1324 1325 1326 1327
			"Audio", "ChannelSetup");

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

1328
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
1329 1330

	return obs_reset_audio(&ai);
J
jp9000 已提交
1331 1332
}

J
jp9000 已提交
1333
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
1334
		const char *deviceDesc, int channel)
J
jp9000 已提交
1335 1336 1337 1338 1339 1340 1341 1342 1343
{
	const char *deviceId = config_get_string(basicConfig, "Audio",
			deviceName);
	obs_source_t source;
	obs_data_t settings;
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
1344
		settings = obs_source_get_settings(source);
J
jp9000 已提交
1345
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357

		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) {
		obs_data_t settings = obs_data_create();
J
jp9000 已提交
1358
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
1359
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
1360
				sourceId, deviceDesc, settings);
J
jp9000 已提交
1361 1362 1363 1364 1365 1366 1367 1368
		obs_data_release(settings);

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

void OBSBasic::ResetAudioDevices()
J
jp9000 已提交
1369
{
1370 1371 1372 1373 1374 1375 1376 1377 1378 1379
	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 已提交
1380 1381
}

J
jp9000 已提交
1382
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
1383
{
1384
	QSize  targetSize;
J
jp9000 已提交
1385

1386
	/* resize preview panel to fix to the top section of the window */
1387
	targetSize = GetPixelSize(ui->preview);
1388
	GetScaleAndCenterPos(int(cx), int(cy),
1389 1390
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
1391
			previewX, previewY, previewScale);
J
jp9000 已提交
1392

1393 1394 1395
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);

J
jp9000 已提交
1396 1397 1398 1399 1400
	if (isVisible()) {
		if (resizeTimer)
			killTimer(resizeTimer);
		resizeTimer = startTimer(100);
	}
J
jp9000 已提交
1401 1402
}

J
jp9000 已提交
1403
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
1404
{
1405 1406 1407 1408 1409 1410 1411
	QWidget::closeEvent(event);
	if (!event->isAccepted())
		return;

	// remove draw callback in case our drawable surfaces go away before
	// the destructor gets called
	obs_remove_draw_callback(OBSBasic::RenderMain, this);
1412 1413
}

J
jp9000 已提交
1414
void OBSBasic::changeEvent(QEvent *event)
1415
{
J
jp9000 已提交
1416 1417
	/* TODO */
	UNUSED_PARAMETER(event);
1418 1419
}

J
jp9000 已提交
1420
void OBSBasic::resizeEvent(QResizeEvent *event)
1421
{
J
jp9000 已提交
1422 1423 1424 1425
	struct obs_video_info ovi;

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

	UNUSED_PARAMETER(event);
1428 1429
}

J
jp9000 已提交
1430 1431 1432 1433 1434 1435
void OBSBasic::timerEvent(QTimerEvent *event)
{
	if (event->timerId() == resizeTimer) {
		killTimer(resizeTimer);
		resizeTimer = 0;

1436
		QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1437 1438 1439 1440
		obs_resize(size.width(), size.height());
	}
}

J
jp9000 已提交
1441
void OBSBasic::on_action_New_triggered()
1442
{
J
jp9000 已提交
1443
	/* TODO */
1444 1445
}

J
jp9000 已提交
1446
void OBSBasic::on_action_Open_triggered()
1447
{
J
jp9000 已提交
1448
	/* TODO */
1449 1450
}

J
jp9000 已提交
1451
void OBSBasic::on_action_Save_triggered()
1452
{
J
jp9000 已提交
1453
	/* TODO */
1454 1455
}

1456 1457 1458 1459 1460 1461 1462
void OBSBasic::on_actionShow_Recordings_triggered()
{
	const char *path = config_get_string(basicConfig,
			"SimpleOutput", "FilePath");
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
1463 1464 1465 1466 1467 1468
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

1469 1470
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1471 1472
{
	obs_source_t source = NULL;
J
jp9000 已提交
1473

1474 1475 1476 1477
	if (sceneChanging)
		return;

	if (current) {
J
jp9000 已提交
1478 1479
		obs_scene_t scene;

1480
		scene = current->data(Qt::UserRole).value<OBSScene>();
1481
		source = obs_scene_get_source(scene);
1482 1483
	}

1484
	/* TODO: allow transitions */
1485
	obs_set_output_source(0, source);
1486 1487

	UNUSED_PARAMETER(prev);
1488 1489
}

J
jp9000 已提交
1490 1491 1492 1493 1494
void OBSBasic::EditSceneName()
{
	ui->scenes->editItem(ui->scenes->currentItem());
}

J
jp9000 已提交
1495
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
1496
{
J
jp9000 已提交
1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507
	QListWidgetItem *item = ui->scenes->itemAt(pos);

	QMenu popup;
	popup.addAction(QTStr("Add"),
			this, SLOT(on_actionAddScene_triggered()));

	if (item)
		popup.addAction(QTStr("Remove"),
				this, SLOT(RemoveSelectedScene()));

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

J
jp9000 已提交
1510
void OBSBasic::on_actionAddScene_triggered()
1511
{
1512
	string name;
S
Socapex 已提交
1513
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
1514 1515 1516

	int i = 1;
	QString placeHolderText = format.arg(i);
P
Palana 已提交
1517 1518 1519
	obs_source_t source = nullptr;
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
1520
		placeHolderText = format.arg(++i);
P
Palana 已提交
1521
	}
S
Socapex 已提交
1522

J
jp9000 已提交
1523
	bool accepted = NameDialog::AskForName(this,
1524 1525
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
1526 1527
			name,
			placeHolderText);
1528

J
jp9000 已提交
1529
	if (accepted) {
J
jp9000 已提交
1530 1531
		if (name.empty()) {
			QMessageBox::information(this,
1532 1533
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
1534 1535 1536 1537
			on_actionAddScene_triggered();
			return;
		}

1538 1539
		obs_source_t source = obs_get_source_by_name(name.c_str());
		if (source) {
J
jp9000 已提交
1540
			QMessageBox::information(this,
1541 1542
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
1543 1544

			obs_source_release(source);
J
jp9000 已提交
1545
			on_actionAddScene_triggered();
1546 1547 1548
			return;
		}

1549
		obs_scene_t scene = obs_scene_create(name.c_str());
1550
		source = obs_scene_get_source(scene);
1551
		obs_add_source(source);
1552
		obs_scene_release(scene);
1553 1554

		obs_set_output_source(0, source);
1555
	}
1556 1557
}

J
jp9000 已提交
1558
void OBSBasic::on_actionRemoveScene_triggered()
1559
{
1560
	OBSScene     scene  = GetCurrentScene();
1561
	obs_source_t source = obs_scene_get_source(scene);
1562 1563 1564

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
1565 1566
}

J
jp9000 已提交
1567
void OBSBasic::on_actionSceneProperties_triggered()
1568
{
J
jp9000 已提交
1569
	/* TODO */
1570 1571
}

J
jp9000 已提交
1572
void OBSBasic::on_actionSceneUp_triggered()
1573
{
J
jp9000 已提交
1574
	/* TODO */
1575 1576
}

J
jp9000 已提交
1577
void OBSBasic::on_actionSceneDown_triggered()
1578
{
J
jp9000 已提交
1579
	/* TODO */
1580 1581
}

1582 1583
void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1584
{
J
jp9000 已提交
1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599
	auto select_one = [] (obs_scene_t scene, obs_sceneitem_t item,
			void *param)
	{
		obs_sceneitem_t selectedItem =
			*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>();
J
jp9000 已提交
1600
	obs_source_t source = obs_sceneitem_get_source(item);
1601 1602 1603
	if ((obs_source_get_output_flags(source) & OBS_SOURCE_VIDEO) == 0)
		return;

J
jp9000 已提交
1604 1605
	obs_scene_enum_items(GetCurrentScene(), select_one, &item);

1606
	UNUSED_PARAMETER(prev);
1607 1608
}

J
jp9000 已提交
1609 1610 1611 1612 1613
void OBSBasic::EditSceneItemName()
{
	ui->sources->editItem(ui->sources->currentItem());
}

J
jp9000 已提交
1614
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
1615
{
J
jp9000 已提交
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626
	QListWidgetItem *item = ui->sources->itemAt(pos);

	QMenu popup;
	QPointer<QMenu> addSourceMenu = CreateAddSourcePopupMenu();
	if (addSourceMenu)
		popup.addMenu(addSourceMenu);

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

J
John Bradley 已提交
1627 1628 1629 1630
		OBSSceneItem sceneItem = GetSceneItem(item);
		obs_source_t source = obs_sceneitem_get_source(sceneItem);
		QAction *action;

J
jp9000 已提交
1631 1632
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
1633 1634 1635
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
				QKeySequence::Delete);
J
jp9000 已提交
1636 1637
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
1638 1639
		popup.addMenu(ui->transformMenu);
		popup.addSeparator();
J
John Bradley 已提交
1640 1641 1642 1643 1644 1645 1646

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

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

J
jp9000 已提交
1647 1648 1649 1650 1651
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

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

J
jp9000 已提交
1654
void OBSBasic::AddSource(const char *id)
1655
{
1656 1657 1658 1659
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
	}
1660 1661
}

1662
QMenu *OBSBasic::CreateAddSourcePopupMenu()
1663
{
1664
	const char *type;
J
jp9000 已提交
1665 1666
	bool foundValues = false;
	size_t idx = 0;
1667

1668
	QMenu *popup = new QMenu(QTStr("Add"));
J
jp9000 已提交
1669
	while (obs_enum_input_types(idx++, &type)) {
1670
		const char *name = obs_source_get_display_name(
1671
				OBS_SOURCE_TYPE_INPUT, type);
1672

1673 1674 1675
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
1676 1677
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
1678 1679 1680
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
1681

J
jp9000 已提交
1682
		foundValues = true;
1683 1684
	}

1685 1686 1687
	if (!foundValues) {
		delete popup;
		popup = nullptr;
1688
	}
1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714

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

J
jp9000 已提交
1717
void OBSBasic::on_actionAddSource_triggered()
1718
{
J
jp9000 已提交
1719
	AddSourcePopupMenu(QCursor::pos());
1720 1721
}

J
jp9000 已提交
1722
void OBSBasic::on_actionRemoveSource_triggered()
1723
{
1724
	OBSSceneItem item   = GetCurrentSceneItem();
J
jp9000 已提交
1725
	obs_source_t source = obs_sceneitem_get_source(item);
1726 1727

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
1728
		obs_sceneitem_remove(item);
1729 1730
}

J
John Bradley 已提交
1731 1732 1733 1734 1735 1736 1737 1738 1739
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
1740
void OBSBasic::on_actionSourceProperties_triggered()
1741
{
1742
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1743
	OBSSource source = obs_sceneitem_get_source(item);
1744

1745 1746
	if (source)
		CreatePropertiesWindow(source);
1747 1748
}

J
jp9000 已提交
1749
void OBSBasic::on_actionSourceUp_triggered()
1750
{
J
jp9000 已提交
1751
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1752
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
1753
}
J
jp9000 已提交
1754

J
jp9000 已提交
1755
void OBSBasic::on_actionSourceDown_triggered()
1756
{
J
jp9000 已提交
1757
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1758
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
1759 1760
}

J
jp9000 已提交
1761 1762 1763
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1764
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
1765 1766 1767 1768 1769
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1770
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
1771 1772 1773 1774 1775
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1776
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
1777 1778 1779 1780 1781
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1782
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
1783 1784
}

1785
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
1786 1787 1788 1789 1790 1791 1792
{
	BPtr<char> logDir(os_get_config_path("obs-studio/logs"));

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

1793
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
1794 1795 1796 1797 1798 1799 1800 1801
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

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

1804
	if (!fileString)
J
jp9000 已提交
1805 1806
		return;

1807
	if (!*fileString)
J
jp9000 已提交
1808 1809 1810 1811
		return;

	ui->menuLogFiles->setEnabled(false);

1812 1813
	auto data_deleter = [](obs_data_t d) { obs_data_release(d); };
	using data_t = unique_ptr<struct obs_data, decltype(data_deleter)>;
J
jp9000 已提交
1814

1815 1816 1817
	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 已提交
1818

1819
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
1820

1821 1822 1823 1824 1825 1826 1827 1828 1829 1830
	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 已提交
1831
	if (!json) {
1832 1833 1834 1835
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

J
jp9000 已提交
1836
	logUploadPostData.setData(json, (int)strlen(json));
J
jp9000 已提交
1837 1838 1839 1840 1841 1842 1843 1844 1845 1846

	QUrl url("https://api.github.com/gists");
	logUploadReply = networkManager.post(QNetworkRequest(url),
			&logUploadPostData);
	connect(logUploadReply, SIGNAL(finished()),
			this, SLOT(logUploadFinished()));
	connect(logUploadReply, SIGNAL(readyRead()),
			this, SLOT(logUploadRead()));
}

P
Palana 已提交
1847 1848 1849 1850 1851 1852 1853
void OBSBasic::on_actionShowLogs_triggered()
{
	BPtr<char> logDir(os_get_config_path("obs-studio/logs"));
	QUrl url = QUrl::fromLocalFile(QT_UTF8(logDir));
	QDesktopServices::openUrl(url);
}

J
jp9000 已提交
1854 1855 1856 1857 1858 1859 1860 1861 1862 1863
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

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

J
jp9000 已提交
1864 1865 1866 1867 1868
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

J
jp9000 已提交
1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889
void OBSBasic::logUploadRead()
{
	logUploadReturnData.push_back(logUploadReply->readAll());
}

void OBSBasic::logUploadFinished()
{
	ui->menuLogFiles->setEnabled(true);

	if (logUploadReply->error()) {
		QMessageBox::information(this,
				QTStr("LogReturnDialog.ErrorUploadingLog"),
				logUploadReply->errorString());
		return;
	}

	const char *jsonReply = logUploadReturnData.constData();
	if (!jsonReply || !*jsonReply)
		return;

	obs_data_t returnData = obs_data_create_from_json(jsonReply);
J
jp9000 已提交
1890
	QString logURL = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
1891 1892 1893 1894
	obs_data_release(returnData);

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

	logUploadReturnData.clear();
J
jp9000 已提交
1897 1898
}

J
jp9000 已提交
1899 1900
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
		obs_source_t source, const string &name)
1901
{
1902
	const char      *prevName   = obs_source_get_name(source);
1903 1904 1905 1906 1907
	obs_source_t    foundSource = obs_get_source_by_name(name.c_str());
	QListWidgetItem *listItem   = listWidget->currentItem();

	if (foundSource || name.compare(prevName) == 0 || name.empty()) {
		listItem->setText(QT_UTF8(prevName));
1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918

		if (foundSource || name.compare(prevName) == 0) {
			QMessageBox::information(parent,
				QTStr("NameExists.Title"),
				QTStr("NameExists.Text"));
		} else if (name.empty()) {
			QMessageBox::information(parent,
				QTStr("NoNameEntered.Title"),
				QTStr("NoNameEntered.Text"));
		}

1919 1920 1921
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
1922
		obs_source_set_name(source, name.c_str());
1923 1924 1925
	}
}

J
jp9000 已提交
1926 1927 1928 1929 1930
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
1931
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
1932 1933 1934 1935

	if (!scene)
		return;

1936
	obs_source_t source = obs_scene_get_source(scene);
1937
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
1938 1939 1940 1941 1942 1943 1944 1945 1946

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
1947
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
1948 1949 1950 1951

	if (!item)
		return;

J
jp9000 已提交
1952
	obs_source_t source = obs_sceneitem_get_source(item);
1953
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
1954 1955 1956 1957

	UNUSED_PARAMETER(endHint);
}

1958
void OBSBasic::StreamingStart()
1959
{
1960
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
1961
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
1962
	ui->statusbar->StreamStarted(streamOutput);
1963 1964
}

1965
void OBSBasic::StreamingStop(int code)
1966
{
1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981
	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;

1982
	default:
1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
	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");
	}

1993
	activeRefs--;
J
jp9000 已提交
1994
	ui->statusbar->StreamStopped();
1995

1996
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
1997
	ui->streamButton->setEnabled(true);
1998 1999 2000 2001 2002

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

P
Palana 已提交
2005 2006 2007 2008 2009
void OBSBasic::RecordingStart()
{
	ui->statusbar->RecordingStarted(fileOutput);
}

2010
void OBSBasic::RecordingStop()
2011
{
P
Palana 已提交
2012
	ui->statusbar->RecordingStopped();
2013 2014 2015
	activeRefs--;
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
}
2016

2017 2018 2019
void OBSBasic::SetupEncoders()
{
	if (activeRefs == 0) {
2020 2021 2022 2023
		obs_data_t x264Settings = obs_data_create();
		obs_data_t aacSettings  = obs_data_create();

		int videoBitrate = config_get_uint(basicConfig, "SimpleOutput",
2024
				"VBitrate");
2025
		int audioBitrate = config_get_uint(basicConfig, "SimpleOutput",
2026
				"ABitrate");
J
jp9000 已提交
2027 2028
		bool advanced = config_get_bool(basicConfig, "SimpleOutput",
				"UseAdvanced");
2029 2030
		bool useCBR = config_get_bool(basicConfig, "SimpleOutput",
				"UseCBR");
J
jp9000 已提交
2031 2032 2033 2034
		const char *preset = config_get_string(basicConfig,
				"SimpleOutput", "Preset");
		const char *custom = config_get_string(basicConfig,
				"SimpleOutput", "x264Settings");
2035

J
jp9000 已提交
2036 2037
		obs_data_set_int(x264Settings, "bitrate", videoBitrate);
		obs_data_set_int(x264Settings, "buffer_size", videoBitrate);
2038

J
jp9000 已提交
2039 2040 2041
		if (advanced) {
			obs_data_set_string(x264Settings, "preset", preset);
			obs_data_set_string(x264Settings, "x264opts", custom);
2042 2043 2044
			obs_data_set_bool(x264Settings, "cbr", useCBR);
		} else {
			obs_data_set_bool(x264Settings, "cbr", true);
J
jp9000 已提交
2045 2046
		}

J
jp9000 已提交
2047
		obs_data_set_int(aacSettings, "bitrate", audioBitrate);
2048

2049 2050
		obs_encoder_update(x264, x264Settings);
		obs_encoder_update(aac,  aacSettings);
J
jp9000 已提交
2051

2052 2053
		obs_data_release(x264Settings);
		obs_data_release(aacSettings);
2054

2055 2056
		obs_encoder_set_video(x264, obs_get_video());
		obs_encoder_set_audio(aac,  obs_get_audio());
2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068
	}
}

void OBSBasic::on_streamButton_clicked()
{
	if (obs_output_active(streamOutput)) {
		obs_output_stop(streamOutput);
	} else {

		SaveService();
		SetupEncoders();

2069 2070 2071
		obs_output_set_video_encoder(streamOutput, x264);
		obs_output_set_audio_encoder(streamOutput, aac);
		obs_output_set_service(streamOutput, service);
2072

J
jp9000 已提交
2073 2074 2075 2076 2077 2078 2079 2080 2081
		bool reconnect = config_get_bool(basicConfig, "SimpleOutput",
				"Reconnect");
		int retryDelay = config_get_uint(basicConfig, "SimpleOutput",
				"RetryDelay");
		int maxRetries = config_get_uint(basicConfig, "SimpleOutput",
				"MaxRetries");
		if (!reconnect)
			maxRetries = 0;

J
jp9000 已提交
2082 2083
		obs_output_set_reconnect_settings(streamOutput, maxRetries,
				retryDelay);
J
jp9000 已提交
2084

2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129
		if (obs_output_start(streamOutput)) {
			activeRefs++;

			ui->streamButton->setEnabled(false);
			ui->streamButton->setText(
					QTStr("Basic.Main.Connecting"));
		}
	}
}

void OBSBasic::on_recordButton_clicked()
{
	if (obs_output_active(fileOutput)) {
		obs_output_stop(fileOutput);
	} else {

		const char *path = config_get_string(basicConfig,
				"SimpleOutput", "FilePath");

		os_dir_t dir = path ? os_opendir(path) : nullptr;

		if (!dir) {
			QMessageBox::information(this,
					QTStr("Output.BadPath.Title"),
					QTStr("Output.BadPath.Text"));
			return;
		}

		os_closedir(dir);

		string strPath;
		strPath += path;

		char lastChar = strPath.back();
		if (lastChar != '/' && lastChar != '\\')
			strPath += "/";

		strPath += GenerateTimeDateFilename("flv");

		SetupEncoders();

		obs_output_set_video_encoder(fileOutput, x264);
		obs_output_set_audio_encoder(fileOutput, aac);

		obs_data_t settings = obs_data_create();
J
jp9000 已提交
2130
		obs_data_set_string(settings, "path", strPath.c_str());
2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141

		obs_output_update(fileOutput, settings);

		obs_data_release(settings);

		if (obs_output_start(fileOutput)) {
			activeRefs++;

			ui->recordButton->setText(
					QTStr("Basic.Main.StopRecording"));
		}
J
jp9000 已提交
2142 2143 2144
	}
}

J
jp9000 已提交
2145
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
2146
{
2147 2148
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
2149
}
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217

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

config_t OBSBasic::Config() const
{
	return basicConfig;
}
J
jp9000 已提交
2218 2219 2220

void OBSBasic::on_actionEditTransform_triggered()
{
2221 2222 2223
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
2224 2225
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
2226
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
2227 2228 2229 2230 2231 2232 2233 2234 2235
}

void OBSBasic::on_actionResetTransform_triggered()
{
	auto func = [] (obs_scene_t scene, obs_sceneitem_t item, void *param)
	{
		if (!obs_sceneitem_selected(item))
			return true;

2236
		obs_transform_info info;
J
jp9000 已提交
2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253
		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);
}

2254
static void GetItemBox(obs_sceneitem_t item, vec3 &tl, vec3 &br)
J
jp9000 已提交
2255 2256 2257 2258 2259
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

2262
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
2263 2264 2265 2266
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
2267 2268
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
2269 2270
	};

2271 2272 2273 2274 2275 2276 2277 2278 2279 2280
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

static vec3 GetItemTL(obs_sceneitem_t item)
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
2281 2282 2283 2284 2285 2286 2287 2288
	return tl;
}

static void SetItemTL(obs_sceneitem_t item, const vec3 &tl)
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
2289
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
2290 2291 2292
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
2293
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305
}

static bool RotateSelectedSources(obs_scene_t scene, obs_sceneitem_t item,
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
2306
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
2307 2308
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
2309
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346

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

static bool MultiplySelectedItemScale(obs_scene_t scene, obs_sceneitem_t item,
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
2347
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
2348
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
2349
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
2350 2351

	SetItemTL(item, tl);
J
jp9000 已提交
2352 2353

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
2354 2355 2356 2357 2358
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
2359 2360
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
2361 2362 2363 2364 2365 2366
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
2367 2368
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

static bool CenterAlignSelectedItems(obs_scene_t scene, obs_sceneitem_t item,
		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);

2384
	obs_transform_info itemInfo;
J
jp9000 已提交
2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416
	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()
{
2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442
	auto func = [] (obs_scene_t scene, obs_sceneitem_t item, void *param)
	{
		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 已提交
2443
		UNUSED_PARAMETER(param);
2444 2445 2446 2447
		return true;
	};

	obs_scene_enum_items(GetCurrentScene(), func, nullptr);
J
jp9000 已提交
2448
}