window-basic-main.cpp 61.3 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"
P
Palana 已提交
41
#include "window-remux.hpp"
J
jp9000 已提交
42
#include "qt-wrappers.hpp"
43
#include "display-helpers.hpp"
44
#include "volume-control.hpp"
45

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

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

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

54 55
#define PREVIEW_EDGE_SIZE 10

56
using namespace std;
J
jp9000 已提交
57

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

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

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

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

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

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

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

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

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

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

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

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

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

130
	obs_data_t *data = obs_save_source(source);
131

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

	obs_data_release(data);
	obs_source_release(source);
}

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

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

	return saveData;
}

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

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

	volumes.clear();
}

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

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

183
static void LoadAudioDevice(const char *name, int channel, obs_data_t *parent)
184
{
185
	obs_data_t *data = obs_data_get_obj(parent, name);
186 187 188
	if (!data)
		return;

189
	obs_source_t *source = obs_load_source(data);
190 191 192 193 194 195 196 197 198 199
	if (source) {
		obs_set_output_source(channel, source);
		obs_source_release(source);
	}

	obs_data_release(data);
}

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

	obs_add_source(source);

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

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

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

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

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

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

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

255 256 257
static inline bool HasAudioDevices(const char *source_id)
{
	const char *output_id = source_id;
258
	obs_properties_t *props = obs_get_source_properties(
259
			OBS_SOURCE_TYPE_INPUT, output_id);
260 261 262 263 264
	size_t count = 0;

	if (!props)
		return false;

265
	obs_property_t *devices = obs_properties_get(props, "device_id");
266 267 268 269 270 271 272 273
	if (devices)
		count = obs_property_list_item_count(devices);

	obs_properties_destroy(props);

	return count != 0;
}

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

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

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

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

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

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

304 305 306 307 308 309 310 311 312 313 314
#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;

315 316
	obs_data_t *data     = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);
317

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

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

	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;

341
	obs_data_t *data = obs_data_create_from_json(jsonText);
342 343

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

346
	obs_data_t *settings = obs_data_get_obj(data, "settings");
347

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

	obs_data_release(settings);
	obs_data_release(data);

	return !!service;
}

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

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

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

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

378 379 380 381 382
	return true;
}

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

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

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

	if (!aac)
		return false;

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

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

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

	return true;
}

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

416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432
	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;

433
	/* TODO: temporary */
434 435
	config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
			GetDefaultVideoSavePath().c_str());
436 437 438
	config_set_default_uint  (basicConfig, "SimpleOutput", "VBitrate",
			2500);
	config_set_default_uint  (basicConfig, "SimpleOutput", "ABitrate", 128);
J
jp9000 已提交
439 440 441 442 443
	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 已提交
444 445
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseAdvanced",
			false);
446
	config_set_default_bool  (basicConfig, "SimpleOutput", "UseCBR", true);
J
jp9000 已提交
447 448
	config_set_default_string(basicConfig, "SimpleOutput", "Preset",
			"veryfast");
449

450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468
	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 已提交
469
	config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
470
			hasDesktopAudio ? "default" : "disabled");
J
jp9000 已提交
471 472 473
	config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice1",
474
			hasInputAudio ? "default" : "disabled");
J
jp9000 已提交
475 476 477 478 479
	config_set_default_string(basicConfig, "Audio", "AuxDevice2",
			"disabled");
	config_set_default_string(basicConfig, "Audio", "AuxDevice3",
			"disabled");

480 481 482 483 484 485 486
	return true;
}

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

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

	return InitBasicConfigDefaults();
}

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

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

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

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

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

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

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

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

549 550 551 552 553 554
	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 已提交
555 556 557
		throw "Failed to initialize video:  Required graphics API "
		      "functionality not found on these drivers or "
		      "unavailable on this equipment";
558 559 560 561 562 563 564
	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";
	}

565
	InitOBSCallbacks();
566

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

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

J
jp9000 已提交
577 578
	InitPrimitives();

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

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

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

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

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

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

	if (transformWindow)
		delete transformWindow;
609

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

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

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

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

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

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

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

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

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

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

657
void OBSBasic::InsertSceneItem(obs_sceneitem_t *item)
658
{
659
	obs_source_t *source = obs_sceneitem_get_source(item);
660
	const char   *name  = obs_source_get_name(source);
661 662

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

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

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

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

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

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

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

695 696 697
/* Qt callbacks for invokeMethod */

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

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

707
	signal_handler_t *handler = obs_source_get_signal_handler(source);
708
	signal_handler_connect(handler, "item_add",
J
jp9000 已提交
709
			OBSBasic::SceneItemAdded, this);
710
	signal_handler_connect(handler, "item_remove",
J
jp9000 已提交
711
			OBSBasic::SceneItemRemoved, this);
J
jp9000 已提交
712 713 714 715 716 717 718 719
	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);
720 721
}

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

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

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

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

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

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

748
void OBSBasic::RemoveSceneItem(OBSSceneItem item)
749
{
750
	obs_scene_t *scene = obs_sceneitem_get_scene(item);
751

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

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

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

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

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

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

781 782 783
		if (!scene)
			return;

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

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

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

J
jp9000 已提交
797 798 799 800 801 802 803 804 805 806 807 808 809 810
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);
811 812 813 814 815

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

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

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

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

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

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

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

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

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

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

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

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

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

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

P
Palana 已提交
894 895 896 897 898
#ifdef UPDATE_SPARKLE
void init_sparkle_updater(bool update_to_undeployed);
void trigger_sparkle_update();
#endif

J
jp9000 已提交
899 900
void OBSBasic::TimedCheckForUpdates()
{
P
Palana 已提交
901 902 903 904
#ifdef UPDATE_SPARKLE
	init_sparkle_updater(config_get_bool(App()->GlobalConfig(), "General",
				"UpdateToUndeployed"));
#else
J
jp9000 已提交
905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920
	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 已提交
921
#endif
J
jp9000 已提交
922 923 924 925
}

void OBSBasic::CheckForUpdates()
{
P
Palana 已提交
926 927 928
#ifdef UPDATE_SPARKLE
	trigger_sparkle_update();
#else
J
jp9000 已提交
929 930
	ui->actionCheckForUpdates->setEnabled(false);

931 932 933 934 935 936 937 938
	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 已提交
939 940 941 942
	connect(updateReply, SIGNAL(finished()),
			this, SLOT(updateFileFinished()));
	connect(updateReply, SIGNAL(readyRead()),
			this, SLOT(updateFileRead()));
P
Palana 已提交
943
#endif
J
jp9000 已提交
944 945 946 947 948 949 950
}

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

J
jp9000 已提交
951 952
#ifdef __APPLE__
#define VERSION_ENTRY "mac"
J
jp9000 已提交
953 954
#elif _WIN32
#define VERSION_ENTRY "windows"
J
jp9000 已提交
955 956 957 958
#else
#define VERSION_ENTRY "other"
#endif

J
jp9000 已提交
959 960 961 962 963 964 965 966 967 968 969 970 971 972
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;

973 974
	obs_data_t *returnData   = obs_data_create_from_json(jsonReply);
	obs_data_t *versionData  = obs_data_get_obj(returnData, VERSION_ENTRY);
J
jp9000 已提交
975 976 977
	const char *description = obs_data_get_string(returnData,
			"description");
	const char *download    = obs_data_get_string(versionData, "download");
J
jp9000 已提交
978 979

	if (returnData && versionData && description && download) {
J
jp9000 已提交
980 981 982
		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 已提交
983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
		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 已提交
1014 1015

	updateReturnData.clear();
J
jp9000 已提交
1016 1017
}

1018 1019 1020 1021
void OBSBasic::RemoveSelectedScene()
{
	OBSScene scene = GetCurrentScene();
	if (scene) {
1022
		obs_source_t *source = obs_scene_get_source(scene);
1023 1024 1025 1026 1027 1028 1029 1030 1031
		if (QueryRemoveSource(source))
			obs_source_remove(source);
	}
}

void OBSBasic::RemoveSelectedSceneItem()
{
	OBSSceneItem item = GetCurrentSceneItem();
	if (item) {
1032
		obs_source_t *source = obs_sceneitem_get_source(item);
1033
		if (QueryRemoveSource(source))
J
jp9000 已提交
1034 1035 1036 1037
			obs_sceneitem_remove(item);
	}
}

1038 1039
/* OBS Callbacks */

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

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

1046 1047
	QMetaObject::invokeMethod(window, "AddSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
J
jp9000 已提交
1048 1049
}

1050
void OBSBasic::SceneItemRemoved(void *data, calldata_t *params)
1051
{
1052
	OBSBasic *window = static_cast<OBSBasic*>(data);
1053

1054
	obs_sceneitem_t *item = (obs_sceneitem_t*)calldata_ptr(params, "item");
1055

1056 1057
	QMetaObject::invokeMethod(window, "RemoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)));
1058 1059
}

1060
void OBSBasic::SourceAdded(void *data, calldata_t *params)
1061
{
J
jp9000 已提交
1062
	OBSBasic *window = static_cast<OBSBasic*>(data);
1063
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1064

1065
	if (obs_scene_from_source(source) != NULL)
J
jp9000 已提交
1066
		QMetaObject::invokeMethod(window,
1067 1068
				"AddScene",
				Q_ARG(OBSSource, OBSSource(source)));
1069 1070
}

1071
void OBSBasic::SourceRemoved(void *data, calldata_t *params)
1072
{
1073
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1074

1075
	if (obs_scene_from_source(source) != NULL)
1076 1077 1078
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"RemoveScene",
				Q_ARG(OBSSource, OBSSource(source)));
1079 1080
}

1081
void OBSBasic::SourceActivated(void *data, calldata_t *params)
1082
{
1083
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1084 1085 1086 1087 1088 1089 1090 1091
	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)));
}

1092
void OBSBasic::SourceDeactivated(void *data, calldata_t *params)
1093
{
1094
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1095 1096 1097 1098 1099 1100 1101 1102
	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)));
}

1103
void OBSBasic::SourceRenamed(void *data, calldata_t *params)
J
jp9000 已提交
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
{
	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)));
}

1114
void OBSBasic::ChannelChanged(void *data, calldata_t *params)
1115
{
1116
	obs_source_t *source = (obs_source_t*)calldata_ptr(params, "source");
1117
	uint32_t channel = (uint32_t)calldata_int(params, "channel");
1118 1119

	if (channel == 0)
1120 1121 1122
		QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
				"UpdateSceneSelection",
				Q_ARG(OBSSource, OBSSource(source)));
1123 1124
}

1125 1126 1127 1128 1129
void OBSBasic::DrawBackdrop(float cx, float cy)
{
	if (!box)
		return;

1130 1131 1132
	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");
1133 1134 1135

	vec4 colorVal;
	vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
1136
	gs_effect_set_vec4(color, &colorVal);
1137

1138 1139
	gs_technique_begin(tech);
	gs_technique_begin_pass(tech, 0);
1140 1141 1142 1143 1144 1145 1146 1147
	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();
1148 1149
	gs_technique_end_pass(tech);
	gs_technique_end(tech);
1150 1151 1152 1153

	gs_load_vertexbuffer(nullptr);
}

1154 1155
void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
{
J
jp9000 已提交
1156
	OBSBasic *window = static_cast<OBSBasic*>(data);
1157 1158 1159 1160
	obs_video_info ovi;

	obs_get_video_info(&ovi);

J
jp9000 已提交
1161 1162
	window->previewCX = int(window->previewScale * float(ovi.base_width));
	window->previewCY = int(window->previewScale * float(ovi.base_height));
1163 1164 1165

	gs_viewport_push();
	gs_projection_push();
1166 1167 1168

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

1169 1170
	gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
			-100.0f, 100.0f);
1171
	gs_set_viewport(window->previewX, window->previewY,
J
jp9000 已提交
1172
			window->previewCX, window->previewCY);
1173

1174 1175
	window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));

J
jp9000 已提交
1176
	obs_render_main_view();
1177
	gs_load_vertexbuffer(nullptr);
1178

1179 1180
	/* --------------------------------------- */

1181 1182 1183
	QSize previewSize = GetPixelSize(window->ui->preview);
	float right  = float(previewSize.width())  - window->previewX;
	float bottom = float(previewSize.height()) - window->previewY;
1184 1185 1186 1187

	gs_ortho(-window->previewX, right,
	         -window->previewY, bottom,
	         -100.0f, 100.0f);
1188
	gs_reset_viewport();
J
jp9000 已提交
1189 1190 1191

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

1192 1193
	/* --------------------------------------- */

1194 1195
	gs_projection_pop();
	gs_viewport_pop();
J
jp9000 已提交
1196 1197 1198

	UNUSED_PARAMETER(cx);
	UNUSED_PARAMETER(cy);
1199 1200
}

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

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

1219
void OBSBasic::SceneItemMoveTop(void *data, calldata_t *params)
J
jp9000 已提交
1220
{
1221
	OBSSceneItem item = (obs_sceneitem_t*)calldata_ptr(params, "item");
J
jp9000 已提交
1222 1223 1224
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1225
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_TOP));
J
jp9000 已提交
1226 1227
}

1228
void OBSBasic::SceneItemMoveBottom(void *data, calldata_t *params)
J
jp9000 已提交
1229
{
1230
	OBSSceneItem item = (obs_sceneitem_t*)calldata_ptr(params, "item");
J
jp9000 已提交
1231 1232 1233
	QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
			"MoveSceneItem",
			Q_ARG(OBSSceneItem, OBSSceneItem(item)),
J
jp9000 已提交
1234
			Q_ARG(obs_order_movement, OBS_ORDER_MOVE_BOTTOM));
J
jp9000 已提交
1235 1236
}

1237 1238
/* Main class functions */

1239
obs_service_t *OBSBasic::GetService()
1240 1241 1242 1243 1244 1245
{
	if (!service)
		service = obs_service_create("rtmp_common", NULL, NULL);
	return service;
}

1246
void OBSBasic::SetService(obs_service_t *newService)
1247 1248 1249 1250 1251 1252 1253 1254
{
	if (newService) {
		if (service)
			obs_service_destroy(service);
		service = newService;
	}
}

1255 1256 1257 1258 1259 1260
#ifdef _WIN32
#define IS_WIN32 1
#else
#define IS_WIN32 0
#endif

1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281
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;
}

1282
int OBSBasic::ResetVideo()
J
jp9000 已提交
1283 1284
{
	struct obs_video_info ovi;
1285
	int ret;
J
jp9000 已提交
1286

1287
	GetConfigFPS(ovi.fps_num, ovi.fps_den);
J
jp9000 已提交
1288 1289

	ovi.graphics_module = App()->GetRenderModule();
1290
	ovi.base_width     = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1291
			"Video", "BaseCX");
1292
	ovi.base_height    = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1293
			"Video", "BaseCY");
1294
	ovi.output_width   = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1295
			"Video", "OutputCX");
1296
	ovi.output_height  = (uint32_t)config_get_uint(basicConfig,
J
jp9000 已提交
1297
			"Video", "OutputCY");
1298
	ovi.output_format  = VIDEO_FORMAT_NV12;
J
jp9000 已提交
1299 1300
	ovi.adapter        = 0;
	ovi.gpu_conversion = true;
1301

J
jp9000 已提交
1302
	QTToGSWindow(ui->preview->winId(), ovi.window);
J
jp9000 已提交
1303 1304

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

1307
	QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1308 1309
	ovi.window_width  = size.width();
	ovi.window_height = size.height();
J
jp9000 已提交
1310

1311
	ret = AttemptToResetVideo(&ovi);
1312 1313 1314 1315 1316 1317 1318 1319
	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);
		}
	}

1320 1321
	if (ret == OBS_VIDEO_SUCCESS)
		obs_add_draw_callback(OBSBasic::RenderMain, this);
1322

1323
	return ret;
J
jp9000 已提交
1324
}
J
jp9000 已提交
1325

1326
bool OBSBasic::ResetAudio()
J
jp9000 已提交
1327
{
J
jp9000 已提交
1328
	struct audio_output_info ai;
1329 1330 1331
	ai.name = "Main Audio Track";
	ai.format = AUDIO_FORMAT_FLOAT;

1332
	ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
1333 1334
			"SampleRate");

1335
	const char *channelSetupStr = config_get_string(basicConfig,
1336 1337 1338 1339 1340 1341 1342
			"Audio", "ChannelSetup");

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

1343
	ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
J
jp9000 已提交
1344 1345

	return obs_reset_audio(&ai);
J
jp9000 已提交
1346 1347
}

J
jp9000 已提交
1348
void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
1349
		const char *deviceDesc, int channel)
J
jp9000 已提交
1350 1351 1352
{
	const char *deviceId = config_get_string(basicConfig, "Audio",
			deviceName);
1353 1354
	obs_source_t *source;
	obs_data_t *settings;
J
jp9000 已提交
1355 1356 1357 1358
	bool same = false;

	source = obs_get_output_source(channel);
	if (source) {
1359
		settings = obs_source_get_settings(source);
J
jp9000 已提交
1360
		const char *curId = obs_data_get_string(settings, "device_id");
J
jp9000 已提交
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371

		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) {
1372
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
1373
		obs_data_set_string(settings, "device_id", deviceId);
J
jp9000 已提交
1374
		source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
1375
				sourceId, deviceDesc, settings);
J
jp9000 已提交
1376 1377 1378 1379 1380 1381 1382 1383
		obs_data_release(settings);

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

void OBSBasic::ResetAudioDevices()
J
jp9000 已提交
1384
{
1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
	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 已提交
1395 1396
}

J
jp9000 已提交
1397
void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
1398
{
1399
	QSize  targetSize;
J
jp9000 已提交
1400

1401
	/* resize preview panel to fix to the top section of the window */
1402
	targetSize = GetPixelSize(ui->preview);
1403
	GetScaleAndCenterPos(int(cx), int(cy),
1404 1405
			targetSize.width()  - PREVIEW_EDGE_SIZE * 2,
			targetSize.height() - PREVIEW_EDGE_SIZE * 2,
1406
			previewX, previewY, previewScale);
J
jp9000 已提交
1407

1408 1409 1410
	previewX += float(PREVIEW_EDGE_SIZE);
	previewY += float(PREVIEW_EDGE_SIZE);

J
jp9000 已提交
1411 1412 1413 1414 1415
	if (isVisible()) {
		if (resizeTimer)
			killTimer(resizeTimer);
		resizeTimer = startTimer(100);
	}
J
jp9000 已提交
1416 1417
}

J
jp9000 已提交
1418
void OBSBasic::closeEvent(QCloseEvent *event)
J
jp9000 已提交
1419
{
1420 1421 1422 1423 1424 1425 1426
	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);
1427 1428
}

J
jp9000 已提交
1429
void OBSBasic::changeEvent(QEvent *event)
1430
{
J
jp9000 已提交
1431 1432
	/* TODO */
	UNUSED_PARAMETER(event);
1433 1434
}

J
jp9000 已提交
1435
void OBSBasic::resizeEvent(QResizeEvent *event)
1436
{
J
jp9000 已提交
1437 1438 1439 1440
	struct obs_video_info ovi;

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

	UNUSED_PARAMETER(event);
1443 1444
}

J
jp9000 已提交
1445 1446 1447 1448 1449 1450
void OBSBasic::timerEvent(QTimerEvent *event)
{
	if (event->timerId() == resizeTimer) {
		killTimer(resizeTimer);
		resizeTimer = 0;

1451
		QSize size = GetPixelSize(ui->preview);
J
jp9000 已提交
1452 1453 1454 1455
		obs_resize(size.width(), size.height());
	}
}

J
jp9000 已提交
1456
void OBSBasic::on_action_New_triggered()
1457
{
J
jp9000 已提交
1458
	/* TODO */
1459 1460
}

J
jp9000 已提交
1461
void OBSBasic::on_action_Open_triggered()
1462
{
J
jp9000 已提交
1463
	/* TODO */
1464 1465
}

J
jp9000 已提交
1466
void OBSBasic::on_action_Save_triggered()
1467
{
J
jp9000 已提交
1468
	/* TODO */
1469 1470
}

1471 1472 1473 1474 1475 1476 1477
void OBSBasic::on_actionShow_Recordings_triggered()
{
	const char *path = config_get_string(basicConfig,
			"SimpleOutput", "FilePath");
	QDesktopServices::openUrl(QUrl::fromLocalFile(path));
}

P
Palana 已提交
1478 1479 1480 1481 1482 1483 1484 1485
void OBSBasic::on_actionRemux_triggered()
{
	const char *path = config_get_string(basicConfig,
			"SimpleOutput", "FilePath");
	OBSRemux remux(path, this);
	remux.exec();
}

P
Palana 已提交
1486 1487 1488 1489 1490 1491
void OBSBasic::on_action_Settings_triggered()
{
	OBSBasicSettings settings(this);
	settings.exec();
}

1492 1493
void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1494
{
1495
	obs_source_t *source = NULL;
J
jp9000 已提交
1496

1497 1498 1499 1500
	if (sceneChanging)
		return;

	if (current) {
1501
		obs_scene_t *scene;
J
jp9000 已提交
1502

1503
		scene = current->data(Qt::UserRole).value<OBSScene>();
1504
		source = obs_scene_get_source(scene);
1505 1506
	}

1507
	/* TODO: allow transitions */
1508
	obs_set_output_source(0, source);
1509 1510

	UNUSED_PARAMETER(prev);
1511 1512
}

J
jp9000 已提交
1513 1514 1515 1516 1517
void OBSBasic::EditSceneName()
{
	ui->scenes->editItem(ui->scenes->currentItem());
}

J
jp9000 已提交
1518
void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
1519
{
J
jp9000 已提交
1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530
	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());
1531 1532
}

J
jp9000 已提交
1533
void OBSBasic::on_actionAddScene_triggered()
1534
{
1535
	string name;
S
Socapex 已提交
1536
	QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
P
Palana 已提交
1537 1538 1539

	int i = 1;
	QString placeHolderText = format.arg(i);
1540
	obs_source_t *source = nullptr;
P
Palana 已提交
1541 1542
	while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
		obs_source_release(source);
P
Palana 已提交
1543
		placeHolderText = format.arg(++i);
P
Palana 已提交
1544
	}
S
Socapex 已提交
1545

J
jp9000 已提交
1546
	bool accepted = NameDialog::AskForName(this,
1547 1548
			QTStr("Basic.Main.AddSceneDlg.Title"),
			QTStr("Basic.Main.AddSceneDlg.Text"),
S
Socapex 已提交
1549 1550
			name,
			placeHolderText);
1551

J
jp9000 已提交
1552
	if (accepted) {
J
jp9000 已提交
1553 1554
		if (name.empty()) {
			QMessageBox::information(this,
1555 1556
					QTStr("NoNameEntered.Title"),
					QTStr("NoNameEntered.Text"));
J
jp9000 已提交
1557 1558 1559 1560
			on_actionAddScene_triggered();
			return;
		}

1561
		obs_source_t *source = obs_get_source_by_name(name.c_str());
1562
		if (source) {
J
jp9000 已提交
1563
			QMessageBox::information(this,
1564 1565
					QTStr("NameExists.Title"),
					QTStr("NameExists.Text"));
1566 1567

			obs_source_release(source);
J
jp9000 已提交
1568
			on_actionAddScene_triggered();
1569 1570 1571
			return;
		}

1572
		obs_scene_t *scene = obs_scene_create(name.c_str());
1573
		source = obs_scene_get_source(scene);
1574
		obs_add_source(source);
1575
		obs_scene_release(scene);
1576 1577

		obs_set_output_source(0, source);
1578
	}
1579 1580
}

J
jp9000 已提交
1581
void OBSBasic::on_actionRemoveScene_triggered()
1582
{
1583
	OBSScene     scene  = GetCurrentScene();
1584
	obs_source_t *source = obs_scene_get_source(scene);
1585 1586 1587

	if (source && QueryRemoveSource(source))
		obs_source_remove(source);
1588 1589
}

J
jp9000 已提交
1590
void OBSBasic::on_actionSceneProperties_triggered()
1591
{
J
jp9000 已提交
1592
	/* TODO */
1593 1594
}

J
jp9000 已提交
1595
void OBSBasic::on_actionSceneUp_triggered()
1596
{
J
jp9000 已提交
1597
	/* TODO */
1598 1599
}

J
jp9000 已提交
1600
void OBSBasic::on_actionSceneDown_triggered()
1601
{
J
jp9000 已提交
1602
	/* TODO */
1603 1604
}

1605 1606
void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
		QListWidgetItem *prev)
1607
{
1608
	auto select_one = [] (obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
1609 1610
			void *param)
	{
1611
		obs_sceneitem_t *selectedItem =
J
jp9000 已提交
1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622
			*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>();
1623
	obs_source_t *source = obs_sceneitem_get_source(item);
1624 1625 1626
	if ((obs_source_get_output_flags(source) & OBS_SOURCE_VIDEO) == 0)
		return;

J
jp9000 已提交
1627 1628
	obs_scene_enum_items(GetCurrentScene(), select_one, &item);

1629
	UNUSED_PARAMETER(prev);
1630 1631
}

J
jp9000 已提交
1632 1633 1634 1635 1636
void OBSBasic::EditSceneItemName()
{
	ui->sources->editItem(ui->sources->currentItem());
}

J
jp9000 已提交
1637
void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
1638
{
J
jp9000 已提交
1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
	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 已提交
1650
		OBSSceneItem sceneItem = GetSceneItem(item);
1651
		obs_source_t *source = obs_sceneitem_get_source(sceneItem);
J
John Bradley 已提交
1652 1653
		QAction *action;

J
jp9000 已提交
1654 1655
		popup.addAction(QTStr("Rename"), this,
				SLOT(EditSceneItemName()));
1656 1657 1658
		popup.addAction(QTStr("Remove"), this,
				SLOT(on_actionRemoveSource_triggered()),
				QKeySequence::Delete);
J
jp9000 已提交
1659 1660
		popup.addSeparator();
		popup.addMenu(ui->orderMenu);
J
jp9000 已提交
1661 1662
		popup.addMenu(ui->transformMenu);
		popup.addSeparator();
J
John Bradley 已提交
1663 1664 1665 1666 1667 1668 1669

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

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

J
jp9000 已提交
1670 1671 1672 1673 1674
		popup.addAction(QTStr("Properties"), this,
				SLOT(on_actionSourceProperties_triggered()));
	}

	popup.exec(QCursor::pos());
1675 1676
}

J
jp9000 已提交
1677
void OBSBasic::AddSource(const char *id)
1678
{
1679 1680 1681 1682
	if (id && *id) {
		OBSBasicSourceSelect sourceSelect(this, id);
		sourceSelect.exec();
	}
1683 1684
}

1685
QMenu *OBSBasic::CreateAddSourcePopupMenu()
1686
{
1687
	const char *type;
J
jp9000 已提交
1688 1689
	bool foundValues = false;
	size_t idx = 0;
1690

1691
	QMenu *popup = new QMenu(QTStr("Add"));
J
jp9000 已提交
1692
	while (obs_enum_input_types(idx++, &type)) {
1693
		const char *name = obs_source_get_display_name(
1694
				OBS_SOURCE_TYPE_INPUT, type);
1695

1696 1697 1698
		if (strcmp(type, "scene") == 0)
			continue;

J
jp9000 已提交
1699 1700
		QAction *popupItem = new QAction(QT_UTF8(name), this);
		popupItem->setData(QT_UTF8(type));
1701 1702 1703
		connect(popupItem, SIGNAL(triggered(bool)),
				this, SLOT(AddSourceFromAction()));
		popup->addAction(popupItem);
1704

J
jp9000 已提交
1705
		foundValues = true;
1706 1707
	}

1708 1709 1710
	if (!foundValues) {
		delete popup;
		popup = nullptr;
1711
	}
1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737

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

J
jp9000 已提交
1740
void OBSBasic::on_actionAddSource_triggered()
1741
{
J
jp9000 已提交
1742
	AddSourcePopupMenu(QCursor::pos());
1743 1744
}

J
jp9000 已提交
1745
void OBSBasic::on_actionRemoveSource_triggered()
1746
{
1747
	OBSSceneItem item   = GetCurrentSceneItem();
1748
	obs_source_t *source = obs_sceneitem_get_source(item);
1749 1750

	if (source && QueryRemoveSource(source))
J
jp9000 已提交
1751
		obs_sceneitem_remove(item);
1752 1753
}

J
John Bradley 已提交
1754 1755 1756 1757 1758 1759 1760 1761 1762
void OBSBasic::on_actionInteract_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
	OBSSource source = obs_sceneitem_get_source(item);

	if (source)
		CreateInteractionWindow(source);
}

J
jp9000 已提交
1763
void OBSBasic::on_actionSourceProperties_triggered()
1764
{
1765
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1766
	OBSSource source = obs_sceneitem_get_source(item);
1767

1768 1769
	if (source)
		CreatePropertiesWindow(source);
1770 1771
}

J
jp9000 已提交
1772
void OBSBasic::on_actionSourceUp_triggered()
1773
{
J
jp9000 已提交
1774
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1775
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
1776
}
J
jp9000 已提交
1777

J
jp9000 已提交
1778
void OBSBasic::on_actionSourceDown_triggered()
1779
{
J
jp9000 已提交
1780
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1781
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
1782 1783
}

J
jp9000 已提交
1784 1785 1786
void OBSBasic::on_actionMoveUp_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1787
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_UP);
J
jp9000 已提交
1788 1789 1790 1791 1792
}

void OBSBasic::on_actionMoveDown_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1793
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_DOWN);
J
jp9000 已提交
1794 1795 1796 1797 1798
}

void OBSBasic::on_actionMoveToTop_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1799
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_TOP);
J
jp9000 已提交
1800 1801 1802 1803 1804
}

void OBSBasic::on_actionMoveToBottom_triggered()
{
	OBSSceneItem item = GetCurrentSceneItem();
J
jp9000 已提交
1805
	obs_sceneitem_set_order(item, OBS_ORDER_MOVE_BOTTOM);
J
jp9000 已提交
1806 1807
}

1808
static BPtr<char> ReadLogFile(const char *log)
J
jp9000 已提交
1809 1810 1811 1812 1813 1814 1815
{
	BPtr<char> logDir(os_get_config_path("obs-studio/logs"));

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

1816
	BPtr<char> file = os_quick_read_utf8_file(path.c_str());
J
jp9000 已提交
1817 1818 1819 1820 1821 1822 1823 1824
	if (!file)
		blog(LOG_WARNING, "Failed to read log file %s", path.c_str());

	return file;
}

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

1827
	if (!fileString)
J
jp9000 已提交
1828 1829
		return;

1830
	if (!*fileString)
J
jp9000 已提交
1831 1832 1833 1834
		return;

	ui->menuLogFiles->setEnabled(false);

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

1838 1839 1840
	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 已提交
1841

1842
	obs_data_set_string(content.get(), "content", fileString);
J
jp9000 已提交
1843

1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
	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 已提交
1854
	if (!json) {
1855 1856 1857 1858
		blog(LOG_ERROR, "Failed to get JSON data for log upload");
		return;
	}

J
jp9000 已提交
1859
	logUploadPostData.setData(json, (int)strlen(json));
J
jp9000 已提交
1860 1861 1862 1863 1864 1865 1866 1867 1868 1869

	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 已提交
1870 1871 1872 1873 1874 1875 1876
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 已提交
1877 1878 1879 1880 1881 1882 1883 1884 1885 1886
void OBSBasic::on_actionUploadCurrentLog_triggered()
{
	UploadLog(App()->GetCurrentLog());
}

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

J
jp9000 已提交
1887 1888 1889 1890 1891
void OBSBasic::on_actionCheckForUpdates_triggered()
{
	CheckForUpdates();
}

J
jp9000 已提交
1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911
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;

1912
	obs_data_t *returnData = obs_data_create_from_json(jsonReply);
J
jp9000 已提交
1913
	QString logURL = obs_data_get_string(returnData, "html_url");
J
jp9000 已提交
1914 1915 1916 1917
	obs_data_release(returnData);

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

	logUploadReturnData.clear();
J
jp9000 已提交
1920 1921
}

J
jp9000 已提交
1922
static void RenameListItem(OBSBasic *parent, QListWidget *listWidget,
1923
		obs_source_t *source, const string &name)
1924
{
1925
	const char      *prevName   = obs_source_get_name(source);
1926
	obs_source_t    *foundSource = obs_get_source_by_name(name.c_str());
1927 1928 1929 1930
	QListWidgetItem *listItem   = listWidget->currentItem();

	if (foundSource || name.compare(prevName) == 0 || name.empty()) {
		listItem->setText(QT_UTF8(prevName));
1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941

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

1942 1943 1944
		obs_source_release(foundSource);
	} else {
		listItem->setText(QT_UTF8(name.c_str()));
1945
		obs_source_set_name(source, name.c_str());
1946 1947 1948
	}
}

J
jp9000 已提交
1949 1950 1951 1952 1953
void OBSBasic::SceneNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSScene  scene = GetCurrentScene();
	QLineEdit *edit = qobject_cast<QLineEdit*>(editor);
1954
	string    text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
1955 1956 1957 1958

	if (!scene)
		return;

1959
	obs_source_t *source = obs_scene_get_source(scene);
1960
	RenameListItem(this, ui->scenes, source, text);
J
jp9000 已提交
1961 1962 1963 1964 1965 1966 1967 1968 1969

	UNUSED_PARAMETER(endHint);
}

void OBSBasic::SceneItemNameEdited(QWidget *editor,
		QAbstractItemDelegate::EndEditHint endHint)
{
	OBSSceneItem item  = GetCurrentSceneItem();
	QLineEdit    *edit = qobject_cast<QLineEdit*>(editor);
1970
	string       text  = QT_TO_UTF8(edit->text().trimmed());
J
jp9000 已提交
1971 1972 1973 1974

	if (!item)
		return;

1975
	obs_source_t *source = obs_sceneitem_get_source(item);
1976
	RenameListItem(this, ui->sources, source, text);
J
jp9000 已提交
1977 1978 1979 1980

	UNUSED_PARAMETER(endHint);
}

1981
void OBSBasic::StreamingStart()
1982
{
1983
	ui->streamButton->setText(QTStr("Basic.Main.StopStreaming"));
J
jp9000 已提交
1984
	ui->streamButton->setEnabled(true);
J
jp9000 已提交
1985
	ui->statusbar->StreamStarted(streamOutput);
1986 1987
}

1988
void OBSBasic::StreamingStop(int code)
1989
{
1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004
	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;

2005
	default:
2006 2007 2008 2009 2010 2011 2012 2013 2014 2015
	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");
	}

2016
	activeRefs--;
J
jp9000 已提交
2017
	ui->statusbar->StreamStopped();
2018

2019
	ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
J
jp9000 已提交
2020
	ui->streamButton->setEnabled(true);
2021 2022 2023 2024 2025

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

P
Palana 已提交
2028 2029 2030 2031 2032
void OBSBasic::RecordingStart()
{
	ui->statusbar->RecordingStarted(fileOutput);
}

2033
void OBSBasic::RecordingStop()
2034
{
P
Palana 已提交
2035
	ui->statusbar->RecordingStopped();
2036 2037 2038
	activeRefs--;
	ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
}
2039

2040 2041 2042
void OBSBasic::SetupEncoders()
{
	if (activeRefs == 0) {
2043 2044
		obs_data_t *x264Settings = obs_data_create();
		obs_data_t *aacSettings  = obs_data_create();
2045 2046

		int videoBitrate = config_get_uint(basicConfig, "SimpleOutput",
2047
				"VBitrate");
2048
		int audioBitrate = config_get_uint(basicConfig, "SimpleOutput",
2049
				"ABitrate");
J
jp9000 已提交
2050 2051
		bool advanced = config_get_bool(basicConfig, "SimpleOutput",
				"UseAdvanced");
2052 2053
		bool useCBR = config_get_bool(basicConfig, "SimpleOutput",
				"UseCBR");
J
jp9000 已提交
2054 2055 2056 2057
		const char *preset = config_get_string(basicConfig,
				"SimpleOutput", "Preset");
		const char *custom = config_get_string(basicConfig,
				"SimpleOutput", "x264Settings");
2058

J
jp9000 已提交
2059 2060
		obs_data_set_int(x264Settings, "bitrate", videoBitrate);
		obs_data_set_int(x264Settings, "buffer_size", videoBitrate);
2061

J
jp9000 已提交
2062 2063 2064
		if (advanced) {
			obs_data_set_string(x264Settings, "preset", preset);
			obs_data_set_string(x264Settings, "x264opts", custom);
2065 2066 2067
			obs_data_set_bool(x264Settings, "cbr", useCBR);
		} else {
			obs_data_set_bool(x264Settings, "cbr", true);
J
jp9000 已提交
2068 2069
		}

J
jp9000 已提交
2070
		obs_data_set_int(aacSettings, "bitrate", audioBitrate);
2071

2072 2073
		obs_encoder_update(x264, x264Settings);
		obs_encoder_update(aac,  aacSettings);
J
jp9000 已提交
2074

2075 2076
		obs_data_release(x264Settings);
		obs_data_release(aacSettings);
2077

2078 2079
		obs_encoder_set_video(x264, obs_get_video());
		obs_encoder_set_audio(aac,  obs_get_audio());
2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091
	}
}

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

		SaveService();
		SetupEncoders();

2092 2093 2094
		obs_output_set_video_encoder(streamOutput, x264);
		obs_output_set_audio_encoder(streamOutput, aac);
		obs_output_set_service(streamOutput, service);
2095

J
jp9000 已提交
2096 2097 2098 2099 2100 2101 2102 2103 2104
		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 已提交
2105 2106
		obs_output_set_reconnect_settings(streamOutput, maxRetries,
				retryDelay);
J
jp9000 已提交
2107

2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126
		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");

2127
		os_dir_t *dir = path ? os_opendir(path) : nullptr;
2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151

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

2152
		obs_data_t *settings = obs_data_create();
J
jp9000 已提交
2153
		obs_data_set_string(settings, "path", strPath.c_str());
2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164

		obs_output_update(fileOutput, settings);

		obs_data_release(settings);

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

			ui->recordButton->setText(
					QTStr("Basic.Main.StopRecording"));
		}
J
jp9000 已提交
2165 2166 2167
	}
}

J
jp9000 已提交
2168
void OBSBasic::on_settingsButton_clicked()
J
jp9000 已提交
2169
{
2170 2171
	OBSBasicSettings settings(this);
	settings.exec();
J
jp9000 已提交
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 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236

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

2237
config_t *OBSBasic::Config() const
2238 2239 2240
{
	return basicConfig;
}
J
jp9000 已提交
2241 2242 2243

void OBSBasic::on_actionEditTransform_triggered()
{
2244 2245 2246
	if (transformWindow)
		transformWindow->close();

J
jp9000 已提交
2247 2248
	transformWindow = new OBSBasicTransform(this);
	transformWindow->show();
2249
	transformWindow->setAttribute(Qt::WA_DeleteOnClose, true);
J
jp9000 已提交
2250 2251 2252 2253
}

void OBSBasic::on_actionResetTransform_triggered()
{
2254
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
J
jp9000 已提交
2255 2256 2257 2258
	{
		if (!obs_sceneitem_selected(item))
			return true;

2259
		obs_transform_info info;
J
jp9000 已提交
2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276
		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);
}

2277
static void GetItemBox(obs_sceneitem_t *item, vec3 &tl, vec3 &br)
J
jp9000 已提交
2278 2279 2280 2281 2282
{
	matrix4 boxTransform;
	obs_sceneitem_get_box_transform(item, &boxTransform);

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

2285
	auto GetMinPos = [&] (float x, float y)
J
jp9000 已提交
2286 2287 2288 2289
	{
		vec3 pos;
		vec3_set(&pos, x, y, 0.0f);
		vec3_transform(&pos, &pos, &boxTransform);
2290 2291
		vec3_min(&tl, &tl, &pos);
		vec3_max(&br, &br, &pos);
J
jp9000 已提交
2292 2293
	};

2294 2295 2296 2297 2298 2299
	GetMinPos(0.0f, 0.0f);
	GetMinPos(1.0f, 0.0f);
	GetMinPos(0.0f, 1.0f);
	GetMinPos(1.0f, 1.0f);
}

2300
static vec3 GetItemTL(obs_sceneitem_t *item)
2301 2302 2303
{
	vec3 tl, br;
	GetItemBox(item, tl, br);
J
jp9000 已提交
2304 2305 2306
	return tl;
}

2307
static void SetItemTL(obs_sceneitem_t *item, const vec3 &tl)
J
jp9000 已提交
2308 2309 2310 2311
{
	vec3 newTL;
	vec2 pos;

J
jp9000 已提交
2312
	obs_sceneitem_get_pos(item, &pos);
J
jp9000 已提交
2313 2314 2315
	newTL = GetItemTL(item);
	pos.x += tl.x - newTL.x;
	pos.y += tl.y - newTL.y;
J
jp9000 已提交
2316
	obs_sceneitem_set_pos(item, &pos);
J
jp9000 已提交
2317 2318
}

2319
static bool RotateSelectedSources(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2320 2321 2322 2323 2324 2325 2326 2327 2328
		void *param)
{
	if (!obs_sceneitem_selected(item))
		return true;

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

	vec3 tl = GetItemTL(item);

J
jp9000 已提交
2329
	rot += obs_sceneitem_get_rot(item);
J
jp9000 已提交
2330 2331
	if (rot >= 360.0f)       rot -= 360.0f;
	else if (rot <= -360.0f) rot += 360.0f;
J
jp9000 已提交
2332
	obs_sceneitem_set_rot(item, rot);
J
jp9000 已提交
2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358

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

2359
static bool MultiplySelectedItemScale(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2360 2361 2362 2363 2364 2365 2366 2367 2368 2369
		void *param)
{
	vec2 &mul = *reinterpret_cast<vec2*>(param);

	if (!obs_sceneitem_selected(item))
		return true;

	vec3 tl = GetItemTL(item);

	vec2 scale;
J
jp9000 已提交
2370
	obs_sceneitem_get_scale(item, &scale);
J
jp9000 已提交
2371
	vec2_mul(&scale, &scale, &mul);
J
jp9000 已提交
2372
	obs_sceneitem_set_scale(item, &scale);
J
jp9000 已提交
2373 2374

	SetItemTL(item, tl);
J
jp9000 已提交
2375 2376

	UNUSED_PARAMETER(scene);
J
jp9000 已提交
2377 2378 2379 2380 2381
	return true;
}

void OBSBasic::on_actionFlipHorizontal_triggered()
{
J
jp9000 已提交
2382 2383
	vec2 scale;
	vec2_set(&scale, -1.0f, 1.0f);
J
jp9000 已提交
2384 2385 2386 2387 2388 2389
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

void OBSBasic::on_actionFlipVertical_triggered()
{
J
jp9000 已提交
2390 2391
	vec2 scale;
	vec2_set(&scale, 1.0f, -1.0f);
J
jp9000 已提交
2392 2393 2394 2395
	obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
			&scale);
}

2396
static bool CenterAlignSelectedItems(obs_scene_t *scene, obs_sceneitem_t *item,
J
jp9000 已提交
2397 2398 2399 2400 2401 2402 2403 2404 2405 2406
		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);

2407
	obs_transform_info itemInfo;
J
jp9000 已提交
2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439
	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()
{
2440
	auto func = [] (obs_scene_t *scene, obs_sceneitem_t *item, void *param)
2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465
	{
		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 已提交
2466
		UNUSED_PARAMETER(param);
2467 2468 2469 2470
		return true;
	};

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