| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485 | 
							- #include <string>
 
- #include <algorithm>
 
- #include <QMessageBox>
 
- #include "qt-wrappers.hpp"
 
- #include "audio-encoders.hpp"
 
- #include "window-basic-main.hpp"
 
- #include "window-basic-main-outputs.hpp"
 
- using namespace std;
 
- static void OBSStreamStarting(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	obs_output_t *obj = (obs_output_t*)calldata_ptr(params, "output");
 
- 	int sec = (int)obs_output_get_active_delay(obj);
 
- 	if (sec == 0)
 
- 		return;
 
- 	output->delayActive = true;
 
- 	QMetaObject::invokeMethod(output->main,
 
- 			"StreamDelayStarting", Q_ARG(int, sec));
 
- }
 
- static void OBSStreamStopping(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	obs_output_t *obj = (obs_output_t*)calldata_ptr(params, "output");
 
- 	int sec = (int)obs_output_get_active_delay(obj);
 
- 	if (sec == 0)
 
- 		QMetaObject::invokeMethod(output->main, "StreamStopping");
 
- 	else
 
- 		QMetaObject::invokeMethod(output->main,
 
- 				"StreamDelayStopping", Q_ARG(int, sec));
 
- }
 
- static void OBSStartStreaming(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	output->streamingActive = true;
 
- 	QMetaObject::invokeMethod(output->main, "StreamingStart");
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSStopStreaming(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	int code = (int)calldata_int(params, "code");
 
- 	const char *last_error = calldata_string(params, "last_error");
 
- 	QString arg_last_error = QString::fromUtf8(last_error);
 
- 	output->streamingActive = false;
 
- 	output->delayActive = false;
 
- 	QMetaObject::invokeMethod(output->main,
 
- 			"StreamingStop", Q_ARG(int, code), Q_ARG(QString, arg_last_error));
 
- }
 
- static void OBSStartRecording(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	output->recordingActive = true;
 
- 	QMetaObject::invokeMethod(output->main, "RecordingStart");
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSStopRecording(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	int code = (int)calldata_int(params, "code");
 
- 	output->recordingActive = false;
 
- 	QMetaObject::invokeMethod(output->main,
 
- 			"RecordingStop", Q_ARG(int, code));
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSRecordStopping(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	QMetaObject::invokeMethod(output->main, "RecordStopping");
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSStartReplayBuffer(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	output->replayBufferActive = true;
 
- 	QMetaObject::invokeMethod(output->main, "ReplayBufferStart");
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSStopReplayBuffer(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	int code = (int)calldata_int(params, "code");
 
- 	output->replayBufferActive = false;
 
- 	QMetaObject::invokeMethod(output->main,
 
- 			"ReplayBufferStop", Q_ARG(int, code));
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void OBSReplayBufferStopping(void *data, calldata_t *params)
 
- {
 
- 	BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
 
- 	QMetaObject::invokeMethod(output->main, "ReplayBufferStopping");
 
- 	UNUSED_PARAMETER(params);
 
- }
 
- static void FindBestFilename(string &strPath, bool noSpace)
 
- {
 
- 	int num = 2;
 
- 	if (!os_file_exists(strPath.c_str()))
 
- 		return;
 
- 	const char *ext = strrchr(strPath.c_str(), '.');
 
- 	if (!ext)
 
- 		return;
 
- 	int extStart = int(ext - strPath.c_str());
 
- 	for (;;) {
 
- 		string testPath = strPath;
 
- 		string numStr;
 
- 		numStr = noSpace ? "_" : " (";
 
- 		numStr += to_string(num++);
 
- 		if (!noSpace)
 
- 			numStr += ")";
 
- 		testPath.insert(extStart, numStr);
 
- 		if (!os_file_exists(testPath.c_str())) {
 
- 			strPath = testPath;
 
- 			break;
 
- 		}
 
- 	}
 
- }
 
- /* ------------------------------------------------------------------------ */
 
- static bool CreateAACEncoder(OBSEncoder &res, string &id, int bitrate,
 
- 		const char *name, size_t idx)
 
- {
 
- 	const char *id_ = GetAACEncoderForBitrate(bitrate);
 
- 	if (!id_) {
 
- 		id.clear();
 
- 		res = nullptr;
 
- 		return false;
 
- 	}
 
- 	if (id == id_)
 
- 		return true;
 
- 	id = id_;
 
- 	res = obs_audio_encoder_create(id_, name, nullptr, idx, nullptr);
 
- 	if (res) {
 
- 		obs_encoder_release(res);
 
- 		return true;
 
- 	}
 
- 	return false;
 
- }
 
- /* ------------------------------------------------------------------------ */
 
- struct SimpleOutput : BasicOutputHandler {
 
- 	OBSEncoder             aacStreaming;
 
- 	OBSEncoder             h264Streaming;
 
- 	OBSEncoder             aacRecording;
 
- 	OBSEncoder             h264Recording;
 
- 	string                 aacRecEncID;
 
- 	string                 aacStreamEncID;
 
- 	string                 videoEncoder;
 
- 	string                 videoQuality;
 
- 	bool                   usingRecordingPreset = false;
 
- 	bool                   recordingConfigured = false;
 
- 	bool                   ffmpegOutput = false;
 
- 	bool                   lowCPUx264 = false;
 
- 	SimpleOutput(OBSBasic *main_);
 
- 	int CalcCRF(int crf);
 
- 	void UpdateStreamingSettings_amd(obs_data_t *settings, int bitrate);
 
- 	void UpdateRecordingSettings_x264_crf(int crf);
 
- 	void UpdateRecordingSettings_qsv11(int crf);
 
- 	void UpdateRecordingSettings_nvenc(int cqp);
 
- 	void UpdateRecordingSettings_amd_cqp(int cqp);
 
- 	void UpdateRecordingSettings();
 
- 	void UpdateRecordingAudioSettings();
 
- 	virtual void Update() override;
 
- 	void SetupOutputs();
 
- 	int GetAudioBitrate() const;
 
- 	void LoadRecordingPreset_h264(const char *encoder);
 
- 	void LoadRecordingPreset_Lossless();
 
- 	void LoadRecordingPreset();
 
- 	void LoadStreamingPreset_h264(const char *encoder);
 
- 	void UpdateRecording();
 
- 	bool ConfigureRecording(bool useReplayBuffer);
 
- 	virtual bool StartStreaming(obs_service_t *service) override;
 
- 	virtual bool StartRecording() override;
 
- 	virtual bool StartReplayBuffer() override;
 
- 	virtual void StopStreaming(bool force) override;
 
- 	virtual void StopRecording(bool force) override;
 
- 	virtual void StopReplayBuffer(bool force) override;
 
- 	virtual bool StreamingActive() const override;
 
- 	virtual bool RecordingActive() const override;
 
- 	virtual bool ReplayBufferActive() const override;
 
- };
 
- void SimpleOutput::LoadRecordingPreset_Lossless()
 
- {
 
- 	fileOutput = obs_output_create("ffmpeg_output",
 
- 			"simple_ffmpeg_output", nullptr, nullptr);
 
- 	if (!fileOutput)
 
- 		throw "Failed to create recording FFmpeg output "
 
- 		      "(simple output)";
 
- 	obs_output_release(fileOutput);
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "format_name", "avi");
 
- 	obs_data_set_string(settings, "video_encoder", "utvideo");
 
- 	obs_data_set_string(settings, "audio_encoder", "pcm_s16le");
 
- 	obs_output_update(fileOutput, settings);
 
- 	obs_data_release(settings);
 
- }
 
- void SimpleOutput::LoadRecordingPreset_h264(const char *encoderId)
 
- {
 
- 	h264Recording = obs_video_encoder_create(encoderId,
 
- 			"simple_h264_recording", nullptr, nullptr);
 
- 	if (!h264Recording)
 
- 		throw "Failed to create h264 recording encoder (simple output)";
 
- 	obs_encoder_release(h264Recording);
 
- }
 
- void SimpleOutput::LoadStreamingPreset_h264(const char *encoderId)
 
- {
 
- 	h264Streaming = obs_video_encoder_create(encoderId,
 
- 			"simple_h264_stream", nullptr, nullptr);
 
- 	if (!h264Streaming)
 
- 		throw "Failed to create h264 streaming encoder (simple output)";
 
- 	obs_encoder_release(h264Streaming);
 
- }
 
- void SimpleOutput::LoadRecordingPreset()
 
- {
 
- 	const char *quality = config_get_string(main->Config(), "SimpleOutput",
 
- 			"RecQuality");
 
- 	const char *encoder = config_get_string(main->Config(), "SimpleOutput",
 
- 			"RecEncoder");
 
- 	videoEncoder = encoder;
 
- 	videoQuality = quality;
 
- 	ffmpegOutput = false;
 
- 	if (strcmp(quality, "Stream") == 0) {
 
- 		h264Recording = h264Streaming;
 
- 		aacRecording = aacStreaming;
 
- 		usingRecordingPreset = false;
 
- 		return;
 
- 	} else if (strcmp(quality, "Lossless") == 0) {
 
- 		LoadRecordingPreset_Lossless();
 
- 		usingRecordingPreset = true;
 
- 		ffmpegOutput = true;
 
- 		return;
 
- 	} else {
 
- 		lowCPUx264 = false;
 
- 		if (strcmp(encoder, SIMPLE_ENCODER_X264) == 0) {
 
- 			LoadRecordingPreset_h264("obs_x264");
 
- 		} else if (strcmp(encoder, SIMPLE_ENCODER_X264_LOWCPU) == 0) {
 
- 			LoadRecordingPreset_h264("obs_x264");
 
- 			lowCPUx264 = true;
 
- 		} else if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
 
- 			LoadRecordingPreset_h264("obs_qsv11");
 
- 		} else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
 
- 			LoadRecordingPreset_h264("amd_amf_h264");
 
- 		} else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
 
- 			LoadRecordingPreset_h264("ffmpeg_nvenc");
 
- 		}
 
- 		usingRecordingPreset = true;
 
- 		if (!CreateAACEncoder(aacRecording, aacRecEncID, 192,
 
- 					"simple_aac_recording", 0))
 
- 			throw "Failed to create aac recording encoder "
 
- 			      "(simple output)";
 
- 	}
 
- }
 
- SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
 
- {
 
- 	streamOutput = obs_output_create("rtmp_output", "simple_stream",
 
- 			nullptr, nullptr);
 
- 	if (!streamOutput)
 
- 		throw "Failed to create stream output (simple output)";
 
- 	obs_output_release(streamOutput);
 
- 	const char *encoder = config_get_string(main->Config(), "SimpleOutput",
 
- 			"StreamEncoder");
 
- 	if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0)
 
- 		LoadStreamingPreset_h264("obs_qsv11");
 
- 	else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0)
 
- 		LoadStreamingPreset_h264("amd_amf_h264");
 
- 	else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0)
 
- 		LoadStreamingPreset_h264("ffmpeg_nvenc");
 
- 	else
 
- 		LoadStreamingPreset_h264("obs_x264");
 
- 	if (!CreateAACEncoder(aacStreaming, aacStreamEncID, GetAudioBitrate(),
 
- 				"simple_aac", 0))
 
- 		throw "Failed to create aac streaming encoder (simple output)";
 
- 	streamDelayStarting.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"starting", OBSStreamStarting, this);
 
- 	streamStopping.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"stopping", OBSStreamStopping, this);
 
- 	startStreaming.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"start", OBSStartStreaming, this);
 
- 	stopStreaming.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"stop", OBSStopStreaming, this);
 
- 	LoadRecordingPreset();
 
- 	if (!ffmpegOutput) {
 
- 		bool useReplayBuffer = config_get_bool(main->Config(),
 
- 				"SimpleOutput", "RecRB");
 
- 		if (useReplayBuffer) {
 
- 			const char *str = config_get_string(main->Config(),
 
- 					"Hotkeys", "ReplayBuffer");
 
- 			obs_data_t *hotkey = obs_data_create_from_json(str);
 
- 			replayBuffer = obs_output_create("replay_buffer",
 
- 					Str("ReplayBuffer"), nullptr, hotkey);
 
- 			obs_data_release(hotkey);
 
- 			if (!replayBuffer)
 
- 				throw "Failed to create replay buffer output "
 
- 				      "(simple output)";
 
- 			obs_output_release(replayBuffer);
 
- 			signal_handler_t *signal =
 
- 				obs_output_get_signal_handler(replayBuffer);
 
- 			startReplayBuffer.Connect(signal, "start",
 
- 					OBSStartReplayBuffer, this);
 
- 			stopReplayBuffer.Connect(signal, "stop",
 
- 					OBSStopReplayBuffer, this);
 
- 			replayBufferStopping.Connect(signal, "stopping",
 
- 					OBSReplayBufferStopping, this);
 
- 		}
 
- 		fileOutput = obs_output_create("ffmpeg_muxer",
 
- 				"simple_file_output", nullptr, nullptr);
 
- 		if (!fileOutput)
 
- 			throw "Failed to create recording output "
 
- 			      "(simple output)";
 
- 		obs_output_release(fileOutput);
 
- 	}
 
- 	startRecording.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"start", OBSStartRecording, this);
 
- 	stopRecording.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"stop", OBSStopRecording, this);
 
- 	recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"stopping", OBSRecordStopping, this);
 
- }
 
- int SimpleOutput::GetAudioBitrate() const
 
- {
 
- 	int bitrate = (int)config_get_uint(main->Config(), "SimpleOutput",
 
- 			"ABitrate");
 
- 	return FindClosestAvailableAACBitrate(bitrate);
 
- }
 
- void SimpleOutput::Update()
 
- {
 
- 	obs_data_t *h264Settings = obs_data_create();
 
- 	obs_data_t *aacSettings  = obs_data_create();
 
- 	int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
 
- 			"VBitrate");
 
- 	int audioBitrate = GetAudioBitrate();
 
- 	bool advanced = config_get_bool(main->Config(), "SimpleOutput",
 
- 			"UseAdvanced");
 
- 	bool enforceBitrate = config_get_bool(main->Config(), "SimpleOutput",
 
- 			"EnforceBitrate");
 
- 	const char *custom = config_get_string(main->Config(),
 
- 			"SimpleOutput", "x264Settings");
 
- 	const char *encoder = config_get_string(main->Config(), "SimpleOutput",
 
- 			"StreamEncoder");
 
- 	const char *presetType;
 
- 	const char *preset;
 
- 	if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
 
- 		presetType = "QSVPreset";
 
- 	} else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
 
- 		presetType = "AMDPreset";
 
- 		UpdateStreamingSettings_amd(h264Settings, videoBitrate);
 
- 	} else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
 
- 		presetType = "NVENCPreset";
 
- 	} else {
 
- 		presetType = "Preset";
 
- 	}
 
- 	preset = config_get_string(main->Config(), "SimpleOutput", presetType);
 
- 	obs_data_set_string(h264Settings, "rate_control", "CBR");
 
- 	obs_data_set_int(h264Settings, "bitrate", videoBitrate);
 
- 	if (advanced) {
 
- 		obs_data_set_string(h264Settings, "preset", preset);
 
- 		obs_data_set_string(h264Settings, "x264opts", custom);
 
- 	}
 
- 	obs_data_set_string(aacSettings, "rate_control", "CBR");
 
- 	obs_data_set_int(aacSettings, "bitrate", audioBitrate);
 
- 	obs_service_apply_encoder_settings(main->GetService(),
 
- 			h264Settings, aacSettings);
 
- 	if (advanced && !enforceBitrate) {
 
- 		obs_data_set_int(h264Settings, "bitrate", videoBitrate);
 
- 		obs_data_set_int(aacSettings, "bitrate", audioBitrate);
 
- 	}
 
- 	video_t *video = obs_get_video();
 
- 	enum video_format format = video_output_get_format(video);
 
- 	if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
 
- 		obs_encoder_set_preferred_video_format(h264Streaming,
 
- 				VIDEO_FORMAT_NV12);
 
- 	obs_encoder_update(h264Streaming, h264Settings);
 
- 	obs_encoder_update(aacStreaming,  aacSettings);
 
- 	obs_data_release(h264Settings);
 
- 	obs_data_release(aacSettings);
 
- }
 
- void SimpleOutput::UpdateRecordingAudioSettings()
 
- {
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_int(settings, "bitrate", 192);
 
- 	obs_data_set_string(settings, "rate_control", "CBR");
 
- 	obs_encoder_update(aacRecording, settings);
 
- 	obs_data_release(settings);
 
- }
 
- #define CROSS_DIST_CUTOFF 2000.0
 
- int SimpleOutput::CalcCRF(int crf)
 
- {
 
- 	int cx = config_get_uint(main->Config(), "Video", "OutputCX");
 
- 	int cy = config_get_uint(main->Config(), "Video", "OutputCY");
 
- 	double fCX = double(cx);
 
- 	double fCY = double(cy);
 
- 	if (lowCPUx264)
 
- 		crf -= 2;
 
- 	double crossDist = sqrt(fCX * fCX + fCY * fCY);
 
- 	double crfResReduction =
 
- 		fmin(CROSS_DIST_CUTOFF, crossDist) / CROSS_DIST_CUTOFF;
 
- 	crfResReduction = (1.0 - crfResReduction) * 10.0;
 
- 	return crf - int(crfResReduction);
 
- }
 
- void SimpleOutput::UpdateRecordingSettings_x264_crf(int crf)
 
- {
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_int(settings, "crf", crf);
 
- 	obs_data_set_bool(settings, "use_bufsize", true);
 
- 	obs_data_set_string(settings, "rate_control", "CRF");
 
- 	obs_data_set_string(settings, "profile", "high");
 
- 	obs_data_set_string(settings, "preset",
 
- 			lowCPUx264 ? "ultrafast" : "veryfast");
 
- 	obs_encoder_update(h264Recording, settings);
 
- 	obs_data_release(settings);
 
- }
 
- static bool icq_available(obs_encoder_t *encoder)
 
- {
 
- 	obs_properties_t *props = obs_encoder_properties(encoder);
 
- 	obs_property_t *p = obs_properties_get(props, "rate_control");
 
- 	bool icq_found = false;
 
- 	size_t num = obs_property_list_item_count(p);
 
- 	for (size_t i = 0; i < num; i++) {
 
- 		const char *val = obs_property_list_item_string(p, i);
 
- 		if (strcmp(val, "ICQ") == 0) {
 
- 			icq_found = true;
 
- 			break;
 
- 		}
 
- 	}
 
- 	obs_properties_destroy(props);
 
- 	return icq_found;
 
- }
 
- void SimpleOutput::UpdateRecordingSettings_qsv11(int crf)
 
- {
 
- 	bool icq = icq_available(h264Recording);
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "profile", "high");
 
- 	if (icq) {
 
- 		obs_data_set_string(settings, "rate_control", "ICQ");
 
- 		obs_data_set_int(settings, "icq_quality", crf);
 
- 	} else {
 
- 		obs_data_set_string(settings, "rate_control", "CQP");
 
- 		obs_data_set_int(settings, "qpi", crf);
 
- 		obs_data_set_int(settings, "qpp", crf);
 
- 		obs_data_set_int(settings, "qpb", crf);
 
- 	}
 
- 	obs_encoder_update(h264Recording, settings);
 
- 	obs_data_release(settings);
 
- }
 
- void SimpleOutput::UpdateRecordingSettings_nvenc(int cqp)
 
- {
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "rate_control", "CQP");
 
- 	obs_data_set_string(settings, "profile", "high");
 
- 	obs_data_set_string(settings, "preset", "hq");
 
- 	obs_data_set_int(settings, "cqp", cqp);
 
- 	obs_encoder_update(h264Recording, settings);
 
- 	obs_data_release(settings);
 
- }
 
- void SimpleOutput::UpdateStreamingSettings_amd(obs_data_t *settings,
 
- 		int bitrate)
 
- {
 
- 	// Static Properties
 
- 	obs_data_set_int(settings, "Usage", 0);
 
- 	obs_data_set_int(settings, "Profile", 100); // High
 
- 	obs_data_set_string(settings, "profile", "high"); // High
 
- 	
 
- 	// Rate Control Properties
 
- 	obs_data_set_int(settings, "RateControlMethod", 1);
 
- 	obs_data_set_string(settings, "rate_control", "CBR");
 
- 	obs_data_set_int(settings, "Bitrate.Target", bitrate);
 
- 	obs_data_set_int(settings, "bitrate", bitrate);
 
- 	obs_data_set_int(settings, "FillerData", 1);
 
- 	obs_data_set_int(settings, "VBVBuffer", 1);
 
- 	obs_data_set_int(settings, "VBVBuffer.Size", bitrate);
 
- 	
 
- 	// Picture Control Properties
 
- 	obs_data_set_double(settings, "KeyframeInterval", 2.0);
 
- 	obs_data_set_int(settings, "keyint_sec", 2);
 
- }
 
- void SimpleOutput::UpdateRecordingSettings_amd_cqp(int cqp)
 
- {
 
- 	obs_data_t *settings = obs_data_create();
 
- 	// Static Properties
 
- 	obs_data_set_int(settings, "Usage", 0);
 
- 	obs_data_set_int(settings, "Profile", 100); // High
 
- 	obs_data_set_string(settings, "profile", "high"); // High
 
- 	// Rate Control Properties
 
- 	obs_data_set_int(settings, "RateControlMethod", 0);
 
- 	obs_data_set_string(settings, "rate_control", "CQP");
 
- 	obs_data_set_int(settings, "QP.IFrame", cqp);
 
- 	obs_data_set_int(settings, "QP.PFrame", cqp);
 
- 	obs_data_set_int(settings, "QP.BFrame", cqp);
 
- 	obs_data_set_int(settings, "VBVBuffer", 1);
 
- 	obs_data_set_int(settings, "VBVBuffer.Size", 100000);
 
- 	// Picture Control Properties
 
- 	obs_data_set_double(settings, "KeyframeInterval", 2.0);
 
- 	obs_data_set_int(settings, "keyint_sec", 2);
 
- 	// Update and release
 
- 	obs_encoder_update(h264Recording, settings);
 
- 	obs_data_release(settings);
 
- }
 
- void SimpleOutput::UpdateRecordingSettings()
 
- {
 
- 	bool ultra_hq = (videoQuality == "HQ");
 
- 	int crf = CalcCRF(ultra_hq ? 16 : 23);
 
- 	if (astrcmp_n(videoEncoder.c_str(), "x264", 4) == 0) {
 
- 		UpdateRecordingSettings_x264_crf(crf);
 
- 	} else if (videoEncoder == SIMPLE_ENCODER_QSV) {
 
- 		UpdateRecordingSettings_qsv11(crf);
 
- 	} else if (videoEncoder == SIMPLE_ENCODER_AMD) {
 
- 		UpdateRecordingSettings_amd_cqp(crf);
 
- 	} else if (videoEncoder == SIMPLE_ENCODER_NVENC) {
 
- 		UpdateRecordingSettings_nvenc(crf);
 
- 	}
 
- }
 
- inline void SimpleOutput::SetupOutputs()
 
- {
 
- 	SimpleOutput::Update();
 
- 	obs_encoder_set_video(h264Streaming, obs_get_video());
 
- 	obs_encoder_set_audio(aacStreaming,  obs_get_audio());
 
- 	if (usingRecordingPreset) {
 
- 		if (ffmpegOutput) {
 
- 			obs_output_set_media(fileOutput, obs_get_video(),
 
- 					obs_get_audio());
 
- 		} else {
 
- 			obs_encoder_set_video(h264Recording, obs_get_video());
 
- 			obs_encoder_set_audio(aacRecording,  obs_get_audio());
 
- 		}
 
- 	}
 
- }
 
- bool SimpleOutput::StartStreaming(obs_service_t *service)
 
- {
 
- 	if (!Active())
 
- 		SetupOutputs();
 
- 	obs_output_set_video_encoder(streamOutput, h264Streaming);
 
- 	obs_output_set_audio_encoder(streamOutput, aacStreaming, 0);
 
- 	obs_output_set_service(streamOutput, service);
 
- 	bool reconnect = config_get_bool(main->Config(), "Output",
 
- 			"Reconnect");
 
- 	int retryDelay = config_get_uint(main->Config(), "Output",
 
- 			"RetryDelay");
 
- 	int maxRetries = config_get_uint(main->Config(), "Output",
 
- 			"MaxRetries");
 
- 	bool useDelay = config_get_bool(main->Config(), "Output",
 
- 			"DelayEnable");
 
- 	int delaySec = config_get_int(main->Config(), "Output",
 
- 			"DelaySec");
 
- 	bool preserveDelay = config_get_bool(main->Config(), "Output",
 
- 			"DelayPreserve");
 
- 	const char *bindIP = config_get_string(main->Config(), "Output",
 
- 			"BindIP");
 
- 	bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
 
- 			"NewSocketLoopEnable");
 
- 	bool enableLowLatencyMode = config_get_bool(main->Config(), "Output",
 
- 			"LowLatencyEnable");
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "bind_ip", bindIP);
 
- 	obs_data_set_bool(settings, "new_socket_loop_enabled",
 
- 			enableNewSocketLoop);
 
- 	obs_data_set_bool(settings, "low_latency_mode_enabled",
 
- 			enableLowLatencyMode);
 
- 	obs_output_update(streamOutput, settings);
 
- 	obs_data_release(settings);
 
- 	if (!reconnect)
 
- 		maxRetries = 0;
 
- 	obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
 
- 			preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
 
- 	obs_output_set_reconnect_settings(streamOutput, maxRetries,
 
- 			retryDelay);
 
- 	if (obs_output_start(streamOutput)) {
 
- 		return true;
 
- 	}
 
- 	return false;
 
- }
 
- static void remove_reserved_file_characters(string &s)
 
- {
 
- 	replace(s.begin(), s.end(), '/', '_');
 
- 	replace(s.begin(), s.end(), '\\', '_');
 
- 	replace(s.begin(), s.end(), '*', '_');
 
- 	replace(s.begin(), s.end(), '?', '_');
 
- 	replace(s.begin(), s.end(), '"', '_');
 
- 	replace(s.begin(), s.end(), '|', '_');
 
- 	replace(s.begin(), s.end(), ':', '_');
 
- 	replace(s.begin(), s.end(), '>', '_');
 
- 	replace(s.begin(), s.end(), '<', '_');
 
- }
 
- static void ensure_directory_exists(string &path)
 
- {
 
- 	replace(path.begin(), path.end(), '\\', '/');
 
- 	size_t last = path.rfind('/');
 
- 	if (last == string::npos)
 
- 		return;
 
- 	string directory = path.substr(0, last);
 
- 	os_mkdirs(directory.c_str());
 
- }
 
- void SimpleOutput::UpdateRecording()
 
- {
 
- 	if (replayBufferActive || recordingActive)
 
- 		return;
 
- 	if (usingRecordingPreset) {
 
- 		if (!ffmpegOutput)
 
- 			UpdateRecordingSettings();
 
- 	} else if (!obs_output_active(streamOutput)) {
 
- 		Update();
 
- 	}
 
- 	if (!Active())
 
- 		SetupOutputs();
 
- 	if (!ffmpegOutput) {
 
- 		obs_output_set_video_encoder(fileOutput, h264Recording);
 
- 		obs_output_set_audio_encoder(fileOutput, aacRecording, 0);
 
- 	}
 
- 	if (replayBuffer) {
 
- 		obs_output_set_video_encoder(replayBuffer, h264Recording);
 
- 		obs_output_set_audio_encoder(replayBuffer, aacRecording, 0);
 
- 	}
 
- 	recordingConfigured = true;
 
- }
 
- bool SimpleOutput::ConfigureRecording(bool updateReplayBuffer)
 
- {
 
- 	const char *path = config_get_string(main->Config(),
 
- 			"SimpleOutput", "FilePath");
 
- 	const char *format = config_get_string(main->Config(),
 
- 			"SimpleOutput", "RecFormat");
 
- 	const char *mux = config_get_string(main->Config(), "SimpleOutput",
 
- 			"MuxerCustom");
 
- 	bool noSpace = config_get_bool(main->Config(), "SimpleOutput",
 
- 			"FileNameWithoutSpace");
 
- 	const char *filenameFormat = config_get_string(main->Config(), "Output",
 
- 				"FilenameFormatting");
 
- 	bool overwriteIfExists = config_get_bool(main->Config(), "Output",
 
- 				"OverwriteIfExists");
 
- 	const char *rbPrefix = config_get_string(main->Config(), "SimpleOutput",
 
- 				"RecRBPrefix");
 
- 	const char *rbSuffix = config_get_string(main->Config(), "SimpleOutput",
 
- 				"RecRBSuffix");
 
- 	int rbTime = config_get_int(main->Config(), "SimpleOutput",
 
- 			"RecRBTime");
 
- 	int rbSize = config_get_int(main->Config(), "SimpleOutput",
 
- 			"RecRBSize");
 
- 	os_dir_t *dir = path ? os_opendir(path) : nullptr;
 
- 	if (!dir) {
 
- 		if (main->isVisible())
 
- 			OBSMessageBox::information(main,
 
- 					QTStr("Output.BadPath.Title"),
 
- 					QTStr("Output.BadPath.Text"));
 
- 		else
 
- 			main->SysTrayNotify(QTStr("Output.BadPath.Text"),
 
- 					QSystemTrayIcon::Warning);
 
- 		return false;
 
- 	}
 
- 	os_closedir(dir);
 
- 	string strPath;
 
- 	strPath += path;
 
- 	char lastChar = strPath.back();
 
- 	if (lastChar != '/' && lastChar != '\\')
 
- 		strPath += "/";
 
- 	strPath += GenerateSpecifiedFilename(ffmpegOutput ? "avi" : format,
 
- 			noSpace, filenameFormat);
 
- 	ensure_directory_exists(strPath);
 
- 	if (!overwriteIfExists)
 
- 		FindBestFilename(strPath, noSpace);
 
- 	obs_data_t *settings = obs_data_create();
 
- 	if (updateReplayBuffer) {
 
- 		string f;
 
- 		if (rbPrefix && *rbPrefix) {
 
- 			f += rbPrefix;
 
- 			if (f.back() != ' ')
 
- 				f += " ";
 
- 		}
 
- 		f += filenameFormat;
 
- 		if (rbSuffix && *rbSuffix) {
 
- 			if (*rbSuffix != ' ')
 
- 				f += " ";
 
- 			f += rbSuffix;
 
- 		}
 
- 		remove_reserved_file_characters(f);
 
- 		obs_data_set_string(settings, "directory", path);
 
- 		obs_data_set_string(settings, "format", f.c_str());
 
- 		obs_data_set_string(settings, "extension", format);
 
- 		obs_data_set_int(settings, "max_time_sec", rbTime);
 
- 		obs_data_set_int(settings, "max_size_mb",
 
- 				usingRecordingPreset ? rbSize : 0);
 
- 	} else {
 
- 		obs_data_set_string(settings, ffmpegOutput ? "url" : "path",
 
- 				strPath.c_str());
 
- 	}
 
- 	obs_data_set_string(settings, "muxer_settings", mux);
 
- 	if (updateReplayBuffer)
 
- 		obs_output_update(replayBuffer, settings);
 
- 	else
 
- 		obs_output_update(fileOutput, settings);
 
- 	obs_data_release(settings);
 
- 	return true;
 
- }
 
- bool SimpleOutput::StartRecording()
 
- {
 
- 	UpdateRecording();
 
- 	if (!ConfigureRecording(false))
 
- 		return false;
 
- 	if (!obs_output_start(fileOutput))  {
 
- 		QMessageBox::critical(main,
 
- 				QTStr("Output.StartRecordingFailed"),
 
- 				QTStr("Output.StartFailedGeneric"));
 
- 		return false;
 
- 	}
 
- 	return true;
 
- }
 
- bool SimpleOutput::StartReplayBuffer()
 
- {
 
- 	UpdateRecording();
 
- 	if (!ConfigureRecording(true))
 
- 		return false;
 
- 	if (!obs_output_start(replayBuffer)) {
 
- 		QMessageBox::critical(main,
 
- 				QTStr("Output.StartReplayFailed"),
 
- 				QTStr("Output.StartFailedGeneric"));
 
- 		return false;
 
- 	}
 
- 	return true;
 
- }
 
- void SimpleOutput::StopStreaming(bool force)
 
- {
 
- 	if (force)
 
- 		obs_output_force_stop(streamOutput);
 
- 	else
 
- 		obs_output_stop(streamOutput);
 
- }
 
- void SimpleOutput::StopRecording(bool force)
 
- {
 
- 	if (force)
 
- 		obs_output_force_stop(fileOutput);
 
- 	else
 
- 		obs_output_stop(fileOutput);
 
- }
 
- void SimpleOutput::StopReplayBuffer(bool force)
 
- {
 
- 	if (force)
 
- 		obs_output_force_stop(replayBuffer);
 
- 	else
 
- 		obs_output_stop(replayBuffer);
 
- }
 
- bool SimpleOutput::StreamingActive() const
 
- {
 
- 	return obs_output_active(streamOutput);
 
- }
 
- bool SimpleOutput::RecordingActive() const
 
- {
 
- 	return obs_output_active(fileOutput);
 
- }
 
- bool SimpleOutput::ReplayBufferActive() const
 
- {
 
- 	return obs_output_active(replayBuffer);
 
- }
 
- /* ------------------------------------------------------------------------ */
 
- struct AdvancedOutput : BasicOutputHandler {
 
- 	OBSEncoder             aacTrack[MAX_AUDIO_MIXES];
 
- 	OBSEncoder             h264Streaming;
 
- 	OBSEncoder             h264Recording;
 
- 	bool                   ffmpegOutput;
 
- 	bool                   ffmpegRecording;
 
- 	bool                   useStreamEncoder;
 
- 	string                 aacEncoderID[MAX_AUDIO_MIXES];
 
- 	AdvancedOutput(OBSBasic *main_);
 
- 	inline void UpdateStreamSettings();
 
- 	inline void UpdateRecordingSettings();
 
- 	inline void UpdateAudioSettings();
 
- 	virtual void Update() override;
 
- 	inline void SetupStreaming();
 
- 	inline void SetupRecording();
 
- 	inline void SetupFFmpeg();
 
- 	void SetupOutputs();
 
- 	int GetAudioBitrate(size_t i) const;
 
- 	virtual bool StartStreaming(obs_service_t *service) override;
 
- 	virtual bool StartRecording() override;
 
- 	virtual void StopStreaming(bool force) override;
 
- 	virtual void StopRecording(bool force) override;
 
- 	virtual bool StreamingActive() const override;
 
- 	virtual bool RecordingActive() const override;
 
- };
 
- static OBSData GetDataFromJsonFile(const char *jsonFile)
 
- {
 
- 	char fullPath[512];
 
- 	int ret = GetProfilePath(fullPath, sizeof(fullPath), jsonFile);
 
- 	if (ret > 0) {
 
- 		BPtr<char> jsonData = os_quick_read_utf8_file(fullPath);
 
- 		if (!!jsonData) {
 
- 			obs_data_t *data = obs_data_create_from_json(jsonData);
 
- 			OBSData dataRet(data);
 
- 			obs_data_release(data);
 
- 			return dataRet;
 
- 		}
 
- 	}
 
- 	return nullptr;
 
- }
 
- AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
 
- {
 
- 	const char *recType = config_get_string(main->Config(), "AdvOut",
 
- 			"RecType");
 
- 	const char *streamEncoder = config_get_string(main->Config(), "AdvOut",
 
- 			"Encoder");
 
- 	const char *recordEncoder = config_get_string(main->Config(), "AdvOut",
 
- 			"RecEncoder");
 
- 	ffmpegOutput = astrcmpi(recType, "FFmpeg") == 0;
 
- 	ffmpegRecording = ffmpegOutput &&
 
- 		config_get_bool(main->Config(), "AdvOut", "FFOutputToFile");
 
- 	useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
 
- 	OBSData streamEncSettings = GetDataFromJsonFile("streamEncoder.json");
 
- 	OBSData recordEncSettings = GetDataFromJsonFile("recordEncoder.json");
 
- 	streamOutput = obs_output_create("rtmp_output", "adv_stream",
 
- 			nullptr, nullptr);
 
- 	if (!streamOutput)
 
- 		throw "Failed to create stream output (advanced output)";
 
- 	obs_output_release(streamOutput);
 
- 	if (ffmpegOutput) {
 
- 		fileOutput = obs_output_create("ffmpeg_output",
 
- 				"adv_ffmpeg_output", nullptr, nullptr);
 
- 		if (!fileOutput)
 
- 			throw "Failed to create recording FFmpeg output "
 
- 			      "(advanced output)";
 
- 		obs_output_release(fileOutput);
 
- 	} else {
 
- 		fileOutput = obs_output_create("ffmpeg_muxer",
 
- 				"adv_file_output", nullptr, nullptr);
 
- 		if (!fileOutput)
 
- 			throw "Failed to create recording output "
 
- 			      "(advanced output)";
 
- 		obs_output_release(fileOutput);
 
- 		if (!useStreamEncoder) {
 
- 			h264Recording = obs_video_encoder_create(recordEncoder,
 
- 					"recording_h264", recordEncSettings,
 
- 					nullptr);
 
- 			if (!h264Recording)
 
- 				throw "Failed to create recording h264 "
 
- 				      "encoder (advanced output)";
 
- 			obs_encoder_release(h264Recording);
 
- 		}
 
- 	}
 
- 	h264Streaming = obs_video_encoder_create(streamEncoder,
 
- 			"streaming_h264", streamEncSettings, nullptr);
 
- 	if (!h264Streaming)
 
- 		throw "Failed to create streaming h264 encoder "
 
- 		      "(advanced output)";
 
- 	obs_encoder_release(h264Streaming);
 
- 	for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
 
- 		char name[9];
 
- 		sprintf(name, "adv_aac%d", i);
 
- 		if (!CreateAACEncoder(aacTrack[i], aacEncoderID[i],
 
- 					GetAudioBitrate(i), name, i))
 
- 			throw "Failed to create audio encoder "
 
- 			      "(advanced output)";
 
- 	}
 
- 	streamDelayStarting.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"starting", OBSStreamStarting, this);
 
- 	streamStopping.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"stopping", OBSStreamStopping, this);
 
- 	startStreaming.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"start", OBSStartStreaming, this);
 
- 	stopStreaming.Connect(obs_output_get_signal_handler(streamOutput),
 
- 			"stop", OBSStopStreaming, this);
 
- 	startRecording.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"start", OBSStartRecording, this);
 
- 	stopRecording.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"stop", OBSStopRecording, this);
 
- 	recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
 
- 			"stopping", OBSRecordStopping, this);
 
- }
 
- void AdvancedOutput::UpdateStreamSettings()
 
- {
 
- 	bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
 
- 			"ApplyServiceSettings");
 
- 	OBSData settings = GetDataFromJsonFile("streamEncoder.json");
 
- 	if (applyServiceSettings)
 
- 		obs_service_apply_encoder_settings(main->GetService(),
 
- 				settings, nullptr);
 
- 	video_t *video = obs_get_video();
 
- 	enum video_format format = video_output_get_format(video);
 
- 	if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
 
- 		obs_encoder_set_preferred_video_format(h264Streaming,
 
- 				VIDEO_FORMAT_NV12);
 
- 	obs_encoder_update(h264Streaming, settings);
 
- }
 
- inline void AdvancedOutput::UpdateRecordingSettings()
 
- {
 
- 	OBSData settings = GetDataFromJsonFile("recordEncoder.json");
 
- 	obs_encoder_update(h264Recording, settings);
 
- }
 
- void AdvancedOutput::Update()
 
- {
 
- 	UpdateStreamSettings();
 
- 	if (!useStreamEncoder && !ffmpegOutput)
 
- 		UpdateRecordingSettings();
 
- 	UpdateAudioSettings();
 
- }
 
- inline void AdvancedOutput::SetupStreaming()
 
- {
 
- 	bool rescale = config_get_bool(main->Config(), "AdvOut",
 
- 			"Rescale");
 
- 	const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
 
- 			"RescaleRes");
 
- 	bool multitrack = config_get_bool(main->Config(), "AdvOut",
 
- 			"Multitrack");
 
- 	int trackIndex = config_get_int(main->Config(), "AdvOut",
 
- 			"TrackIndex");
 
- 	int trackCount = config_get_int(main->Config(), "AdvOut",
 
- 			"TrackCount");
 
- 	unsigned int cx = 0;
 
- 	unsigned int cy = 0;
 
- 	if (rescale && rescaleRes && *rescaleRes) {
 
- 		if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
 
- 			cx = 0;
 
- 			cy = 0;
 
- 		}
 
- 	}
 
- 	obs_encoder_set_scaled_size(h264Streaming, cx, cy);
 
- 	obs_encoder_set_video(h264Streaming, obs_get_video());
 
- 	obs_output_set_video_encoder(streamOutput, h264Streaming);
 
- 	if (multitrack) {
 
- 		int i = 0;
 
- 		for (; i < trackCount; i++)
 
- 			obs_output_set_audio_encoder(streamOutput, aacTrack[i],
 
- 					i);
 
- 		for (; i < MAX_AUDIO_MIXES; i++)
 
- 			obs_output_set_audio_encoder(streamOutput, nullptr, i);
 
- 	} else {
 
- 		obs_output_set_audio_encoder(streamOutput,
 
- 				aacTrack[trackIndex - 1], 0);
 
- 	}
 
- }
 
- inline void AdvancedOutput::SetupRecording()
 
- {
 
- 	const char *path = config_get_string(main->Config(), "AdvOut",
 
- 			"RecFilePath");
 
- 	const char *mux = config_get_string(main->Config(), "AdvOut",
 
- 			"RecMuxerCustom");
 
- 	bool rescale = config_get_bool(main->Config(), "AdvOut",
 
- 			"RecRescale");
 
- 	const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
 
- 			"RecRescaleRes");
 
- 	int tracks = config_get_int(main->Config(), "AdvOut", "RecTracks");
 
- 	obs_data_t *settings = obs_data_create();
 
- 	unsigned int cx = 0;
 
- 	unsigned int cy = 0;
 
- 	int idx = 0;
 
- 	if (useStreamEncoder) {
 
- 		obs_output_set_video_encoder(fileOutput, h264Streaming);
 
- 	} else {
 
- 		if (rescale && rescaleRes && *rescaleRes) {
 
- 			if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
 
- 				cx = 0;
 
- 				cy = 0;
 
- 			}
 
- 		}
 
- 		obs_encoder_set_scaled_size(h264Recording, cx, cy);
 
- 		obs_encoder_set_video(h264Recording, obs_get_video());
 
- 		obs_output_set_video_encoder(fileOutput, h264Recording);
 
- 	}
 
- 	for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
 
- 		if ((tracks & (1<<i)) != 0) {
 
- 			obs_output_set_audio_encoder(fileOutput, aacTrack[i],
 
- 					idx++);
 
- 		}
 
- 	}
 
- 	obs_data_set_string(settings, "path", path);
 
- 	obs_data_set_string(settings, "muxer_settings", mux);
 
- 	obs_output_update(fileOutput, settings);
 
- 	obs_data_release(settings);
 
- }
 
- inline void AdvancedOutput::SetupFFmpeg()
 
- {
 
- 	const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
 
- 	int vBitrate = config_get_int(main->Config(), "AdvOut",
 
- 			"FFVBitrate");
 
- 	int gopSize = config_get_int(main->Config(), "AdvOut",
 
- 			"FFVGOPSize");
 
- 	bool rescale = config_get_bool(main->Config(), "AdvOut",
 
- 			"FFRescale");
 
- 	const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
 
- 			"FFRescaleRes");
 
- 	const char *formatName = config_get_string(main->Config(), "AdvOut",
 
- 			"FFFormat");
 
- 	const char *mimeType = config_get_string(main->Config(), "AdvOut",
 
- 			"FFFormatMimeType");
 
- 	const char *muxCustom = config_get_string(main->Config(), "AdvOut",
 
- 			"FFMCustom");
 
- 	const char *vEncoder = config_get_string(main->Config(), "AdvOut",
 
- 			"FFVEncoder");
 
- 	int vEncoderId = config_get_int(main->Config(), "AdvOut",
 
- 			"FFVEncoderId");
 
- 	const char *vEncCustom = config_get_string(main->Config(), "AdvOut",
 
- 			"FFVCustom");
 
- 	int aBitrate = config_get_int(main->Config(), "AdvOut",
 
- 			"FFABitrate");
 
- 	int aTrack = config_get_int(main->Config(), "AdvOut",
 
- 			"FFAudioTrack");
 
- 	const char *aEncoder = config_get_string(main->Config(), "AdvOut",
 
- 			"FFAEncoder");
 
- 	int aEncoderId = config_get_int(main->Config(), "AdvOut",
 
- 			"FFAEncoderId");
 
- 	const char *aEncCustom = config_get_string(main->Config(), "AdvOut",
 
- 			"FFACustom");
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "url", url);
 
- 	obs_data_set_string(settings, "format_name", formatName);
 
- 	obs_data_set_string(settings, "format_mime_type", mimeType);
 
- 	obs_data_set_string(settings, "muxer_settings", muxCustom);
 
- 	obs_data_set_int(settings, "gop_size", gopSize);
 
- 	obs_data_set_int(settings, "video_bitrate", vBitrate);
 
- 	obs_data_set_string(settings, "video_encoder", vEncoder);
 
- 	obs_data_set_int(settings, "video_encoder_id", vEncoderId);
 
- 	obs_data_set_string(settings, "video_settings", vEncCustom);
 
- 	obs_data_set_int(settings, "audio_bitrate", aBitrate);
 
- 	obs_data_set_string(settings, "audio_encoder", aEncoder);
 
- 	obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
 
- 	obs_data_set_string(settings, "audio_settings", aEncCustom);
 
- 	if (rescale && rescaleRes && *rescaleRes) {
 
- 		int width;
 
- 		int height;
 
- 		int val = sscanf(rescaleRes, "%dx%d", &width, &height);
 
- 		if (val == 2 && width && height) {
 
- 			obs_data_set_int(settings, "scale_width", width);
 
- 			obs_data_set_int(settings, "scale_height", height);
 
- 		}
 
- 	}
 
- 	obs_output_set_mixer(fileOutput, aTrack - 1);
 
- 	obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
 
- 	obs_output_update(fileOutput, settings);
 
- 	obs_data_release(settings);
 
- }
 
- static inline void SetEncoderName(obs_encoder_t *encoder, const char *name,
 
- 		const char *defaultName)
 
- {
 
- 	obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
 
- }
 
- inline void AdvancedOutput::UpdateAudioSettings()
 
- {
 
- 	bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
 
- 			"ApplyServiceSettings");
 
- 	int streamTrackIndex = config_get_int(main->Config(), "AdvOut",
 
- 			"TrackIndex");
 
- 	obs_data_t *settings[MAX_AUDIO_MIXES];
 
- 	for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
 
- 		settings[i] = obs_data_create();
 
- 		obs_data_set_int(settings[i], "bitrate", GetAudioBitrate(i));
 
- 	}
 
- 	for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
 
- 		string cfg_name = "Track";
 
- 		cfg_name += to_string((int)i + 1);
 
- 		cfg_name += "Name";
 
- 		const char *name = config_get_string(main->Config(), "AdvOut",
 
- 				cfg_name.c_str());
 
- 		string def_name = "Track";
 
- 		def_name += to_string((int)i + 1);
 
- 		SetEncoderName(aacTrack[i], name, def_name.c_str());
 
- 	}
 
- 	for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
 
- 		if (applyServiceSettings && (int)(i + 1) == streamTrackIndex)
 
- 			obs_service_apply_encoder_settings(main->GetService(),
 
- 					nullptr, settings[i]);
 
- 		obs_encoder_update(aacTrack[i], settings[i]);
 
- 		obs_data_release(settings[i]);
 
- 	}
 
- }
 
- void AdvancedOutput::SetupOutputs()
 
- {
 
- 	obs_encoder_set_video(h264Streaming, obs_get_video());
 
- 	if (h264Recording)
 
- 		obs_encoder_set_video(h264Recording, obs_get_video());
 
- 	for (size_t i = 0; i < MAX_AUDIO_MIXES; i++)
 
- 		obs_encoder_set_audio(aacTrack[i], obs_get_audio());
 
- 	SetupStreaming();
 
- 	if (ffmpegOutput)
 
- 		SetupFFmpeg();
 
- 	else
 
- 		SetupRecording();
 
- }
 
- int AdvancedOutput::GetAudioBitrate(size_t i) const
 
- {
 
- 	static const char *names[] = {
 
- 		"Track1Bitrate", "Track2Bitrate",
 
- 		"Track3Bitrate", "Track4Bitrate",
 
- 		"Track5Bitrate", "Track6Bitrate",
 
- 	};
 
- 	int bitrate = (int)config_get_uint(main->Config(), "AdvOut", names[i]);
 
- 	return FindClosestAvailableAACBitrate(bitrate);
 
- }
 
- bool AdvancedOutput::StartStreaming(obs_service_t *service)
 
- {
 
- 	if (!useStreamEncoder ||
 
- 	    (!ffmpegOutput && !obs_output_active(fileOutput))) {
 
- 		UpdateStreamSettings();
 
- 	}
 
- 	UpdateAudioSettings();
 
- 	if (!Active())
 
- 		SetupOutputs();
 
- 	obs_output_set_service(streamOutput, service);
 
- 	bool reconnect = config_get_bool(main->Config(), "Output", "Reconnect");
 
- 	int retryDelay = config_get_int(main->Config(), "Output", "RetryDelay");
 
- 	int maxRetries = config_get_int(main->Config(), "Output", "MaxRetries");
 
- 	bool useDelay = config_get_bool(main->Config(), "Output",
 
- 			"DelayEnable");
 
- 	int delaySec = config_get_int(main->Config(), "Output",
 
- 			"DelaySec");
 
- 	bool preserveDelay = config_get_bool(main->Config(), "Output",
 
- 			"DelayPreserve");
 
- 	const char *bindIP = config_get_string(main->Config(), "Output",
 
- 			"BindIP");
 
- 	bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
 
- 			"NewSocketLoopEnable");
 
- 	bool enableLowLatencyMode = config_get_bool(main->Config(), "Output",
 
- 			"LowLatencyEnable");
 
- 	obs_data_t *settings = obs_data_create();
 
- 	obs_data_set_string(settings, "bind_ip", bindIP);
 
- 	obs_data_set_bool(settings, "new_socket_loop_enabled",
 
- 			enableNewSocketLoop);
 
- 	obs_data_set_bool(settings, "low_latency_mode_enabled",
 
- 			enableLowLatencyMode);
 
- 	obs_output_update(streamOutput, settings);
 
- 	obs_data_release(settings);
 
- 	if (!reconnect)
 
- 		maxRetries = 0;
 
- 	obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
 
- 			preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
 
- 	obs_output_set_reconnect_settings(streamOutput, maxRetries,
 
- 			retryDelay);
 
- 	if (obs_output_start(streamOutput)) {
 
- 		return true;
 
- 	}
 
- 	return false;
 
- }
 
- bool AdvancedOutput::StartRecording()
 
- {
 
- 	const char *path;
 
- 	const char *recFormat;
 
- 	const char *filenameFormat;
 
- 	bool noSpace = false;
 
- 	bool overwriteIfExists = false;
 
- 	if (!useStreamEncoder) {
 
- 		if (!ffmpegOutput) {
 
- 			UpdateRecordingSettings();
 
- 		}
 
- 	} else if (!obs_output_active(streamOutput)) {
 
- 		UpdateStreamSettings();
 
- 	}
 
- 	UpdateAudioSettings();
 
- 	if (!Active())
 
- 		SetupOutputs();
 
- 	if (!ffmpegOutput || ffmpegRecording) {
 
- 		path = config_get_string(main->Config(), "AdvOut",
 
- 				ffmpegRecording ? "FFFilePath" : "RecFilePath");
 
- 		recFormat = config_get_string(main->Config(), "AdvOut",
 
- 				ffmpegRecording ? "FFExtension" : "RecFormat");
 
- 		filenameFormat = config_get_string(main->Config(), "Output",
 
- 				"FilenameFormatting");
 
- 		overwriteIfExists = config_get_bool(main->Config(), "Output",
 
- 				"OverwriteIfExists");
 
- 		noSpace = config_get_bool(main->Config(), "AdvOut",
 
- 				ffmpegRecording ?
 
- 				"FFFileNameWithoutSpace" :
 
- 				"RecFileNameWithoutSpace");
 
- 		os_dir_t *dir = path ? os_opendir(path) : nullptr;
 
- 		if (!dir) {
 
- 			if (main->isVisible())
 
- 				OBSMessageBox::information(main,
 
- 						QTStr("Output.BadPath.Title"),
 
- 						QTStr("Output.BadPath.Text"));
 
- 			else
 
- 				main->SysTrayNotify(QTStr("Output.BadPath.Text"),
 
- 						QSystemTrayIcon::Warning);
 
- 			return false;
 
- 		}
 
- 		os_closedir(dir);
 
- 		string strPath;
 
- 		strPath += path;
 
- 		char lastChar = strPath.back();
 
- 		if (lastChar != '/' && lastChar != '\\')
 
- 			strPath += "/";
 
- 		strPath += GenerateSpecifiedFilename(recFormat, noSpace,
 
- 							filenameFormat);
 
- 		ensure_directory_exists(strPath);
 
- 		if (!overwriteIfExists)
 
- 			FindBestFilename(strPath, noSpace);
 
- 		obs_data_t *settings = obs_data_create();
 
- 		obs_data_set_string(settings,
 
- 				ffmpegRecording ? "url" : "path",
 
- 				strPath.c_str());
 
- 		obs_output_update(fileOutput, settings);
 
- 		obs_data_release(settings);
 
- 	}
 
- 	if (!obs_output_start(fileOutput)) {
 
- 		QMessageBox::critical(main,
 
- 				QTStr("Output.StartRecordingFailed"),
 
- 				QTStr("Output.StartFailedGeneric"));
 
- 		return false;
 
- 	}
 
- 	return true;
 
- }
 
- void AdvancedOutput::StopStreaming(bool force)
 
- {
 
- 	if (force)
 
- 		obs_output_force_stop(streamOutput);
 
- 	else
 
- 		obs_output_stop(streamOutput);
 
- }
 
- void AdvancedOutput::StopRecording(bool force)
 
- {
 
- 	if (force)
 
- 		obs_output_force_stop(fileOutput);
 
- 	else
 
- 		obs_output_stop(fileOutput);
 
- }
 
- bool AdvancedOutput::StreamingActive() const
 
- {
 
- 	return obs_output_active(streamOutput);
 
- }
 
- bool AdvancedOutput::RecordingActive() const
 
- {
 
- 	return obs_output_active(fileOutput);
 
- }
 
- /* ------------------------------------------------------------------------ */
 
- BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
 
- {
 
- 	return new SimpleOutput(main);
 
- }
 
- BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
 
- {
 
- 	return new AdvancedOutput(main);
 
- }
 
 
  |