window-basic-filters.cpp 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230
  1. /******************************************************************************
  2. Copyright (C) 2015 by Hugh Bailey <[email protected]>
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 2 of the License, or
  6. (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ******************************************************************************/
  14. #include "properties-view.hpp"
  15. #include "window-namedialog.hpp"
  16. #include "window-basic-main.hpp"
  17. #include "window-basic-filters.hpp"
  18. #include "display-helpers.hpp"
  19. #include "qt-wrappers.hpp"
  20. #include "visibility-item-widget.hpp"
  21. #include "item-widget-helpers.hpp"
  22. #include "obs-app.hpp"
  23. #include "undo-stack-obs.hpp"
  24. #include <QMessageBox>
  25. #include <QCloseEvent>
  26. #include <obs-data.h>
  27. #include <obs.h>
  28. #include <util/base.h>
  29. #include <vector>
  30. #include <string>
  31. #include <QMenu>
  32. #include <QVariant>
  33. using namespace std;
  34. Q_DECLARE_METATYPE(OBSSource);
  35. OBSBasicFilters::OBSBasicFilters(QWidget *parent, OBSSource source_)
  36. : QDialog(parent),
  37. ui(new Ui::OBSBasicFilters),
  38. source(source_),
  39. addSignal(obs_source_get_signal_handler(source), "filter_add",
  40. OBSBasicFilters::OBSSourceFilterAdded, this),
  41. removeSignal(obs_source_get_signal_handler(source), "filter_remove",
  42. OBSBasicFilters::OBSSourceFilterRemoved, this),
  43. reorderSignal(obs_source_get_signal_handler(source),
  44. "reorder_filters", OBSBasicFilters::OBSSourceReordered,
  45. this),
  46. removeSourceSignal(obs_source_get_signal_handler(source), "remove",
  47. OBSBasicFilters::SourceRemoved, this),
  48. renameSourceSignal(obs_source_get_signal_handler(source), "rename",
  49. OBSBasicFilters::SourceRenamed, this),
  50. noPreviewMargin(13)
  51. {
  52. main = reinterpret_cast<OBSBasic *>(parent);
  53. setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
  54. ui->setupUi(this);
  55. UpdateFilters();
  56. ui->asyncFilters->setItemDelegate(
  57. new VisibilityItemDelegate(ui->asyncFilters));
  58. ui->effectFilters->setItemDelegate(
  59. new VisibilityItemDelegate(ui->effectFilters));
  60. const char *name = obs_source_get_name(source);
  61. setWindowTitle(QTStr("Basic.Filters.Title").arg(QT_UTF8(name)));
  62. #ifndef QT_NO_SHORTCUT
  63. ui->actionRemoveFilter->setShortcut(
  64. QApplication::translate("OBSBasicFilters", "Del", nullptr));
  65. #endif // QT_NO_SHORTCUT
  66. addAction(ui->actionRemoveFilter);
  67. addAction(ui->actionMoveUp);
  68. addAction(ui->actionMoveDown);
  69. installEventFilter(CreateShortcutFilter());
  70. connect(ui->asyncFilters->itemDelegate(),
  71. SIGNAL(closeEditor(QWidget *,
  72. QAbstractItemDelegate::EndEditHint)),
  73. this,
  74. SLOT(AsyncFilterNameEdited(
  75. QWidget *, QAbstractItemDelegate::EndEditHint)));
  76. connect(ui->effectFilters->itemDelegate(),
  77. SIGNAL(closeEditor(QWidget *,
  78. QAbstractItemDelegate::EndEditHint)),
  79. this,
  80. SLOT(EffectFilterNameEdited(
  81. QWidget *, QAbstractItemDelegate::EndEditHint)));
  82. QPushButton *close = ui->buttonBox->button(QDialogButtonBox::Close);
  83. connect(close, SIGNAL(clicked()), this, SLOT(close()));
  84. close->setDefault(true);
  85. ui->buttonBox->button(QDialogButtonBox::Reset)
  86. ->setText(QTStr("Defaults"));
  87. connect(ui->buttonBox->button(QDialogButtonBox::Reset),
  88. SIGNAL(clicked()), this, SLOT(ResetFilters()));
  89. uint32_t caps = obs_source_get_output_flags(source);
  90. bool audio = (caps & OBS_SOURCE_AUDIO) != 0;
  91. bool audioOnly = (caps & OBS_SOURCE_VIDEO) == 0;
  92. bool async = (caps & OBS_SOURCE_ASYNC) != 0;
  93. if (!async && !audio) {
  94. ui->asyncWidget->setVisible(false);
  95. ui->separatorLine->setVisible(false);
  96. }
  97. if (audioOnly) {
  98. ui->effectWidget->setVisible(false);
  99. ui->separatorLine->setVisible(false);
  100. }
  101. if (audioOnly || (audio && !async))
  102. ui->asyncLabel->setText(QTStr("Basic.Filters.AudioFilters"));
  103. auto addDrawCallback = [this]() {
  104. obs_display_add_draw_callback(ui->preview->GetDisplay(),
  105. OBSBasicFilters::DrawPreview,
  106. this);
  107. };
  108. enum obs_source_type type = obs_source_get_type(source);
  109. bool drawable_type = type == OBS_SOURCE_TYPE_INPUT ||
  110. type == OBS_SOURCE_TYPE_SCENE;
  111. if ((caps & OBS_SOURCE_VIDEO) != 0) {
  112. ui->rightLayout->setContentsMargins(0, 0, 0, 0);
  113. ui->preview->show();
  114. if (drawable_type)
  115. connect(ui->preview, &OBSQTDisplay::DisplayCreated,
  116. addDrawCallback);
  117. } else {
  118. ui->rightLayout->setContentsMargins(0, noPreviewMargin, 0, 0);
  119. ui->rightContainerLayout->insertStretch(1);
  120. ui->preview->hide();
  121. }
  122. QAction *renameAsync = new QAction(ui->asyncWidget);
  123. renameAsync->setShortcutContext(Qt::WidgetWithChildrenShortcut);
  124. connect(renameAsync, SIGNAL(triggered()), this,
  125. SLOT(RenameAsyncFilter()));
  126. ui->asyncWidget->addAction(renameAsync);
  127. QAction *renameEffect = new QAction(ui->effectWidget);
  128. renameEffect->setShortcutContext(Qt::WidgetWithChildrenShortcut);
  129. connect(renameEffect, SIGNAL(triggered()), this,
  130. SLOT(RenameEffectFilter()));
  131. ui->effectWidget->addAction(renameEffect);
  132. #ifdef __APPLE__
  133. renameAsync->setShortcut({Qt::Key_Return});
  134. renameEffect->setShortcut({Qt::Key_Return});
  135. #else
  136. renameAsync->setShortcut({Qt::Key_F2});
  137. renameEffect->setShortcut({Qt::Key_F2});
  138. #endif
  139. }
  140. OBSBasicFilters::~OBSBasicFilters()
  141. {
  142. ClearListItems(ui->asyncFilters);
  143. ClearListItems(ui->effectFilters);
  144. }
  145. void OBSBasicFilters::Init()
  146. {
  147. show();
  148. }
  149. inline OBSSource OBSBasicFilters::GetFilter(int row, bool async)
  150. {
  151. if (row == -1)
  152. return OBSSource();
  153. QListWidget *list = async ? ui->asyncFilters : ui->effectFilters;
  154. QListWidgetItem *item = list->item(row);
  155. if (!item)
  156. return OBSSource();
  157. QVariant v = item->data(Qt::UserRole);
  158. return v.value<OBSSource>();
  159. }
  160. void OBSBasicFilters::UpdatePropertiesView(int row, bool async)
  161. {
  162. if (view) {
  163. updatePropertiesSignal.Disconnect();
  164. ui->rightLayout->removeWidget(view);
  165. view->deleteLater();
  166. view = nullptr;
  167. }
  168. OBSSource filter = GetFilter(row, async);
  169. if (!filter)
  170. return;
  171. obs_data_t *settings = obs_source_get_settings(filter);
  172. auto filter_change = [](void *vp, obs_data_t *nd_old_settings,
  173. obs_data_t *new_settings) {
  174. obs_source_t *source = reinterpret_cast<obs_source_t *>(vp);
  175. obs_source_t *parent = obs_filter_get_parent(source);
  176. const char *source_name = obs_source_get_name(source);
  177. OBSBasic *main = OBSBasic::Get();
  178. obs_data_t *redo_wrapper = obs_data_create();
  179. obs_data_set_string(redo_wrapper, "name", source_name);
  180. obs_data_set_string(redo_wrapper, "settings",
  181. obs_data_get_json(new_settings));
  182. obs_data_set_string(redo_wrapper, "parent",
  183. obs_source_get_name(parent));
  184. obs_data_t *filter_settings = obs_source_get_settings(source);
  185. obs_data_t *old_settings =
  186. obs_data_get_defaults(filter_settings);
  187. obs_data_apply(old_settings, nd_old_settings);
  188. obs_data_t *undo_wrapper = obs_data_create();
  189. obs_data_set_string(undo_wrapper, "name", source_name);
  190. obs_data_set_string(undo_wrapper, "settings",
  191. obs_data_get_json(old_settings));
  192. obs_data_set_string(undo_wrapper, "parent",
  193. obs_source_get_name(parent));
  194. auto undo_redo = [](const std::string &data) {
  195. obs_data_t *dat =
  196. obs_data_create_from_json(data.c_str());
  197. obs_source_t *parent_source = obs_get_source_by_name(
  198. obs_data_get_string(dat, "parent"));
  199. const char *filter_name =
  200. obs_data_get_string(dat, "name");
  201. obs_source_t *filter = obs_source_get_filter_by_name(
  202. parent_source, filter_name);
  203. obs_data_t *settings = obs_data_create_from_json(
  204. obs_data_get_string(dat, "settings"));
  205. obs_source_update(filter, settings);
  206. obs_source_update_properties(filter);
  207. obs_data_release(dat);
  208. obs_data_release(settings);
  209. obs_source_release(filter);
  210. obs_source_release(parent_source);
  211. };
  212. std::string name = std::string(obs_source_get_name(source));
  213. std::string undo_data = obs_data_get_json(undo_wrapper);
  214. std::string redo_data = obs_data_get_json(redo_wrapper);
  215. main->undo_s.add_action(QTStr("Undo.Filters").arg(name.c_str()),
  216. undo_redo, undo_redo, undo_data,
  217. redo_data, NULL);
  218. obs_data_release(redo_wrapper);
  219. obs_data_release(undo_wrapper);
  220. obs_data_release(old_settings);
  221. obs_data_release(filter_settings);
  222. obs_source_update(source, new_settings);
  223. main->undo_s.enable_undo_redo();
  224. };
  225. auto disabled_undo = [](void *vp, obs_data_t *settings) {
  226. OBSBasic *main =
  227. reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  228. main->undo_s.disable_undo_redo();
  229. obs_source_t *source = reinterpret_cast<obs_source_t *>(vp);
  230. obs_source_update(source, settings);
  231. };
  232. view = new OBSPropertiesView(
  233. settings, filter,
  234. (PropertiesReloadCallback)obs_source_properties,
  235. (PropertiesUpdateCallback)filter_change,
  236. (PropertiesVisualUpdateCb)disabled_undo);
  237. updatePropertiesSignal.Connect(obs_source_get_signal_handler(filter),
  238. "update_properties",
  239. OBSBasicFilters::UpdateProperties, this);
  240. obs_data_release(settings);
  241. view->setMaximumHeight(250);
  242. view->setMinimumHeight(150);
  243. ui->rightLayout->addWidget(view);
  244. view->show();
  245. }
  246. void OBSBasicFilters::UpdateProperties(void *data, calldata_t *)
  247. {
  248. QMetaObject::invokeMethod(static_cast<OBSBasicFilters *>(data)->view,
  249. "ReloadProperties");
  250. }
  251. void OBSBasicFilters::AddFilter(OBSSource filter, bool focus)
  252. {
  253. uint32_t flags = obs_source_get_output_flags(filter);
  254. bool async = (flags & OBS_SOURCE_ASYNC) != 0;
  255. QListWidget *list = async ? ui->asyncFilters : ui->effectFilters;
  256. QListWidgetItem *item = new QListWidgetItem();
  257. Qt::ItemFlags itemFlags = item->flags();
  258. item->setFlags(itemFlags | Qt::ItemIsEditable);
  259. item->setData(Qt::UserRole, QVariant::fromValue(filter));
  260. list->addItem(item);
  261. if (focus)
  262. list->setCurrentItem(item);
  263. SetupVisibilityItem(list, item, filter);
  264. }
  265. void OBSBasicFilters::RemoveFilter(OBSSource filter)
  266. {
  267. uint32_t flags = obs_source_get_output_flags(filter);
  268. bool async = (flags & OBS_SOURCE_ASYNC) != 0;
  269. QListWidget *list = async ? ui->asyncFilters : ui->effectFilters;
  270. for (int i = 0; i < list->count(); i++) {
  271. QListWidgetItem *item = list->item(i);
  272. QVariant v = item->data(Qt::UserRole);
  273. OBSSource curFilter = v.value<OBSSource>();
  274. if (filter == curFilter) {
  275. DeleteListItem(list, item);
  276. break;
  277. }
  278. }
  279. const char *filterName = obs_source_get_name(filter);
  280. const char *sourceName = obs_source_get_name(source);
  281. if (!sourceName || !filterName)
  282. return;
  283. const char *filterId = obs_source_get_id(filter);
  284. blog(LOG_INFO, "User removed filter '%s' (%s) from source '%s'",
  285. filterName, filterId, sourceName);
  286. main->SaveProject();
  287. }
  288. struct FilterOrderInfo {
  289. int asyncIdx = 0;
  290. int effectIdx = 0;
  291. OBSBasicFilters *window;
  292. inline FilterOrderInfo(OBSBasicFilters *window_) : window(window_) {}
  293. };
  294. void OBSBasicFilters::ReorderFilter(QListWidget *list, obs_source_t *filter,
  295. size_t idx)
  296. {
  297. int count = list->count();
  298. for (int i = 0; i < count; i++) {
  299. QListWidgetItem *listItem = list->item(i);
  300. QVariant v = listItem->data(Qt::UserRole);
  301. OBSSource filterItem = v.value<OBSSource>();
  302. if (filterItem == filter) {
  303. if ((int)idx != i) {
  304. bool sel = (list->currentRow() == i);
  305. listItem = TakeListItem(list, i);
  306. if (listItem) {
  307. list->insertItem((int)idx, listItem);
  308. SetupVisibilityItem(list, listItem,
  309. filterItem);
  310. if (sel)
  311. list->setCurrentRow((int)idx);
  312. }
  313. }
  314. break;
  315. }
  316. }
  317. }
  318. void OBSBasicFilters::ReorderFilters()
  319. {
  320. FilterOrderInfo info(this);
  321. obs_source_enum_filters(
  322. source,
  323. [](obs_source_t *, obs_source_t *filter, void *p) {
  324. FilterOrderInfo *info =
  325. reinterpret_cast<FilterOrderInfo *>(p);
  326. uint32_t flags;
  327. bool async;
  328. flags = obs_source_get_output_flags(filter);
  329. async = (flags & OBS_SOURCE_ASYNC) != 0;
  330. if (async) {
  331. info->window->ReorderFilter(
  332. info->window->ui->asyncFilters, filter,
  333. info->asyncIdx++);
  334. } else {
  335. info->window->ReorderFilter(
  336. info->window->ui->effectFilters, filter,
  337. info->effectIdx++);
  338. }
  339. },
  340. &info);
  341. }
  342. void OBSBasicFilters::UpdateFilters()
  343. {
  344. if (!source)
  345. return;
  346. ClearListItems(ui->effectFilters);
  347. ClearListItems(ui->asyncFilters);
  348. obs_source_enum_filters(
  349. source,
  350. [](obs_source_t *, obs_source_t *filter, void *p) {
  351. OBSBasicFilters *window =
  352. reinterpret_cast<OBSBasicFilters *>(p);
  353. window->AddFilter(filter, false);
  354. },
  355. this);
  356. if (ui->asyncFilters->count() > 0) {
  357. ui->asyncFilters->setCurrentItem(ui->asyncFilters->item(0));
  358. } else if (ui->effectFilters->count() > 0) {
  359. ui->effectFilters->setCurrentItem(ui->effectFilters->item(0));
  360. }
  361. main->SaveProject();
  362. }
  363. static bool filter_compatible(bool async, uint32_t sourceFlags,
  364. uint32_t filterFlags)
  365. {
  366. bool filterVideo = (filterFlags & OBS_SOURCE_VIDEO) != 0;
  367. bool filterAsync = (filterFlags & OBS_SOURCE_ASYNC) != 0;
  368. bool filterAudio = (filterFlags & OBS_SOURCE_AUDIO) != 0;
  369. bool audio = (sourceFlags & OBS_SOURCE_AUDIO) != 0;
  370. bool audioOnly = (sourceFlags & OBS_SOURCE_VIDEO) == 0;
  371. bool asyncSource = (sourceFlags & OBS_SOURCE_ASYNC) != 0;
  372. if (async && ((audioOnly && filterVideo) || (!audio && !asyncSource)))
  373. return false;
  374. return (async && (filterAudio || filterAsync)) ||
  375. (!async && !filterAudio && !filterAsync);
  376. }
  377. QMenu *OBSBasicFilters::CreateAddFilterPopupMenu(bool async)
  378. {
  379. uint32_t sourceFlags = obs_source_get_output_flags(source);
  380. const char *type_str;
  381. bool foundValues = false;
  382. size_t idx = 0;
  383. struct FilterInfo {
  384. string type;
  385. string name;
  386. inline FilterInfo(const char *type_, const char *name_)
  387. : type(type_), name(name_)
  388. {
  389. }
  390. };
  391. vector<FilterInfo> types;
  392. while (obs_enum_filter_types(idx++, &type_str)) {
  393. const char *name = obs_source_get_display_name(type_str);
  394. uint32_t caps = obs_get_source_output_flags(type_str);
  395. if ((caps & OBS_SOURCE_DEPRECATED) != 0)
  396. continue;
  397. if ((caps & OBS_SOURCE_CAP_DISABLED) != 0)
  398. continue;
  399. if ((caps & OBS_SOURCE_CAP_OBSOLETE) != 0)
  400. continue;
  401. auto it = types.begin();
  402. for (; it != types.end(); ++it) {
  403. if (it->name >= name)
  404. break;
  405. }
  406. types.emplace(it, type_str, name);
  407. }
  408. QMenu *popup = new QMenu(QTStr("Add"), this);
  409. for (FilterInfo &type : types) {
  410. uint32_t filterFlags =
  411. obs_get_source_output_flags(type.type.c_str());
  412. if (!filter_compatible(async, sourceFlags, filterFlags))
  413. continue;
  414. QAction *popupItem =
  415. new QAction(QT_UTF8(type.name.c_str()), this);
  416. popupItem->setData(QT_UTF8(type.type.c_str()));
  417. connect(popupItem, SIGNAL(triggered(bool)), this,
  418. SLOT(AddFilterFromAction()));
  419. popup->addAction(popupItem);
  420. foundValues = true;
  421. }
  422. if (!foundValues) {
  423. delete popup;
  424. popup = nullptr;
  425. }
  426. return popup;
  427. }
  428. void OBSBasicFilters::AddNewFilter(const char *id)
  429. {
  430. if (id && *id) {
  431. obs_source_t *existing_filter;
  432. string name = obs_source_get_display_name(id);
  433. QString placeholder = QString::fromStdString(name);
  434. QString text{placeholder};
  435. int i = 2;
  436. while ((existing_filter = obs_source_get_filter_by_name(
  437. source, QT_TO_UTF8(text)))) {
  438. obs_source_release(existing_filter);
  439. text = QString("%1 %2").arg(placeholder).arg(i++);
  440. }
  441. bool success = NameDialog::AskForName(
  442. this, QTStr("Basic.Filters.AddFilter.Title"),
  443. QTStr("Basic.Filters.AddFilter.Text"), name, text);
  444. if (!success)
  445. return;
  446. if (name.empty()) {
  447. OBSMessageBox::warning(this,
  448. QTStr("NoNameEntered.Title"),
  449. QTStr("NoNameEntered.Text"));
  450. AddNewFilter(id);
  451. return;
  452. }
  453. existing_filter =
  454. obs_source_get_filter_by_name(source, name.c_str());
  455. if (existing_filter) {
  456. OBSMessageBox::warning(this, QTStr("NameExists.Title"),
  457. QTStr("NameExists.Text"));
  458. obs_source_release(existing_filter);
  459. AddNewFilter(id);
  460. return;
  461. }
  462. obs_data_t *wrapper = obs_data_create();
  463. obs_data_set_string(wrapper, "sname",
  464. obs_source_get_name(source));
  465. obs_data_set_string(wrapper, "fname", name.c_str());
  466. std::string scene_name = obs_source_get_name(
  467. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  468. ->GetCurrentSceneSource());
  469. auto undo = [scene_name](const std::string &data) {
  470. obs_source_t *ssource =
  471. obs_get_source_by_name(scene_name.c_str());
  472. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  473. ->SetCurrentScene(ssource);
  474. obs_source_release(ssource);
  475. obs_data_t *dat =
  476. obs_data_create_from_json(data.c_str());
  477. obs_source_t *source = obs_get_source_by_name(
  478. obs_data_get_string(dat, "sname"));
  479. obs_source_t *filter = obs_source_get_filter_by_name(
  480. source, obs_data_get_string(dat, "fname"));
  481. obs_source_filter_remove(source, filter);
  482. obs_data_release(dat);
  483. obs_source_release(source);
  484. obs_source_release(filter);
  485. };
  486. obs_data_t *rwrapper = obs_data_create();
  487. obs_data_set_string(rwrapper, "sname",
  488. obs_source_get_name(source));
  489. auto redo = [scene_name, id = std::string(id),
  490. name](const std::string &data) {
  491. obs_source_t *ssource =
  492. obs_get_source_by_name(scene_name.c_str());
  493. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  494. ->SetCurrentScene(ssource);
  495. obs_source_release(ssource);
  496. obs_data_t *dat =
  497. obs_data_create_from_json(data.c_str());
  498. obs_source_t *source = obs_get_source_by_name(
  499. obs_data_get_string(dat, "sname"));
  500. obs_source_t *filter = obs_source_create(
  501. id.c_str(), name.c_str(), nullptr, nullptr);
  502. if (filter) {
  503. obs_source_filter_add(source, filter);
  504. obs_source_release(filter);
  505. }
  506. obs_data_release(dat);
  507. obs_source_release(source);
  508. };
  509. std::string undo_data(obs_data_get_json(wrapper));
  510. std::string redo_data(obs_data_get_json(rwrapper));
  511. main->undo_s.add_action(QTStr("Undo.Add").arg(name.c_str()),
  512. undo, redo, undo_data, redo_data, NULL);
  513. obs_data_release(wrapper);
  514. obs_data_release(rwrapper);
  515. obs_source_t *filter =
  516. obs_source_create(id, name.c_str(), nullptr, nullptr);
  517. if (filter) {
  518. const char *sourceName = obs_source_get_name(source);
  519. blog(LOG_INFO,
  520. "User added filter '%s' (%s) "
  521. "to source '%s'",
  522. name.c_str(), id, sourceName);
  523. obs_source_filter_add(source, filter);
  524. obs_source_release(filter);
  525. }
  526. }
  527. }
  528. void OBSBasicFilters::AddFilterFromAction()
  529. {
  530. QAction *action = qobject_cast<QAction *>(sender());
  531. if (!action)
  532. return;
  533. AddNewFilter(QT_TO_UTF8(action->data().toString()));
  534. }
  535. void OBSBasicFilters::closeEvent(QCloseEvent *event)
  536. {
  537. QDialog::closeEvent(event);
  538. if (!event->isAccepted())
  539. return;
  540. obs_display_remove_draw_callback(ui->preview->GetDisplay(),
  541. OBSBasicFilters::DrawPreview, this);
  542. main->SaveProject();
  543. }
  544. /* OBS Signals */
  545. void OBSBasicFilters::OBSSourceFilterAdded(void *param, calldata_t *data)
  546. {
  547. OBSBasicFilters *window = reinterpret_cast<OBSBasicFilters *>(param);
  548. obs_source_t *filter = (obs_source_t *)calldata_ptr(data, "filter");
  549. QMetaObject::invokeMethod(window, "AddFilter",
  550. Q_ARG(OBSSource, OBSSource(filter)));
  551. }
  552. void OBSBasicFilters::OBSSourceFilterRemoved(void *param, calldata_t *data)
  553. {
  554. OBSBasicFilters *window = reinterpret_cast<OBSBasicFilters *>(param);
  555. obs_source_t *filter = (obs_source_t *)calldata_ptr(data, "filter");
  556. QMetaObject::invokeMethod(window, "RemoveFilter",
  557. Q_ARG(OBSSource, OBSSource(filter)));
  558. }
  559. void OBSBasicFilters::OBSSourceReordered(void *param, calldata_t *data)
  560. {
  561. QMetaObject::invokeMethod(reinterpret_cast<OBSBasicFilters *>(param),
  562. "ReorderFilters");
  563. UNUSED_PARAMETER(data);
  564. }
  565. void OBSBasicFilters::SourceRemoved(void *param, calldata_t *data)
  566. {
  567. UNUSED_PARAMETER(data);
  568. QMetaObject::invokeMethod(static_cast<OBSBasicFilters *>(param),
  569. "close");
  570. }
  571. void OBSBasicFilters::SourceRenamed(void *param, calldata_t *data)
  572. {
  573. const char *name = calldata_string(data, "new_name");
  574. QString title = QTStr("Basic.Filters.Title").arg(QT_UTF8(name));
  575. QMetaObject::invokeMethod(static_cast<OBSBasicFilters *>(param),
  576. "setWindowTitle", Q_ARG(QString, title));
  577. }
  578. void OBSBasicFilters::DrawPreview(void *data, uint32_t cx, uint32_t cy)
  579. {
  580. OBSBasicFilters *window = static_cast<OBSBasicFilters *>(data);
  581. if (!window->source)
  582. return;
  583. uint32_t sourceCX = max(obs_source_get_width(window->source), 1u);
  584. uint32_t sourceCY = max(obs_source_get_height(window->source), 1u);
  585. int x, y;
  586. int newCX, newCY;
  587. float scale;
  588. GetScaleAndCenterPos(sourceCX, sourceCY, cx, cy, x, y, scale);
  589. newCX = int(scale * float(sourceCX));
  590. newCY = int(scale * float(sourceCY));
  591. gs_viewport_push();
  592. gs_projection_push();
  593. const bool previous = gs_set_linear_srgb(true);
  594. gs_ortho(0.0f, float(sourceCX), 0.0f, float(sourceCY), -100.0f, 100.0f);
  595. gs_set_viewport(x, y, newCX, newCY);
  596. obs_source_video_render(window->source);
  597. gs_set_linear_srgb(previous);
  598. gs_projection_pop();
  599. gs_viewport_pop();
  600. }
  601. /* Qt Slots */
  602. static bool QueryRemove(QWidget *parent, obs_source_t *source)
  603. {
  604. const char *name = obs_source_get_name(source);
  605. QString text = QTStr("ConfirmRemove.Text");
  606. text.replace("$1", QT_UTF8(name));
  607. QMessageBox remove_source(parent);
  608. remove_source.setText(text);
  609. QAbstractButton *Yes =
  610. remove_source.addButton(QTStr("Yes"), QMessageBox::YesRole);
  611. remove_source.addButton(QTStr("No"), QMessageBox::NoRole);
  612. remove_source.setIcon(QMessageBox::Question);
  613. remove_source.setWindowTitle(QTStr("ConfirmRemove.Title"));
  614. remove_source.exec();
  615. return Yes == remove_source.clickedButton();
  616. }
  617. void OBSBasicFilters::on_addAsyncFilter_clicked()
  618. {
  619. QScopedPointer<QMenu> popup(CreateAddFilterPopupMenu(true));
  620. if (popup)
  621. popup->exec(QCursor::pos());
  622. }
  623. void OBSBasicFilters::on_removeAsyncFilter_clicked()
  624. {
  625. OBSSource filter = GetFilter(ui->asyncFilters->currentRow(), true);
  626. if (filter) {
  627. if (QueryRemove(this, filter))
  628. obs_source_filter_remove(source, filter);
  629. }
  630. }
  631. void OBSBasicFilters::on_moveAsyncFilterUp_clicked()
  632. {
  633. OBSSource filter = GetFilter(ui->asyncFilters->currentRow(), true);
  634. if (filter)
  635. obs_source_filter_set_order(source, filter, OBS_ORDER_MOVE_UP);
  636. }
  637. void OBSBasicFilters::on_moveAsyncFilterDown_clicked()
  638. {
  639. OBSSource filter = GetFilter(ui->asyncFilters->currentRow(), true);
  640. if (filter)
  641. obs_source_filter_set_order(source, filter,
  642. OBS_ORDER_MOVE_DOWN);
  643. }
  644. void OBSBasicFilters::on_asyncFilters_GotFocus()
  645. {
  646. UpdatePropertiesView(ui->asyncFilters->currentRow(), true);
  647. isAsync = true;
  648. }
  649. void OBSBasicFilters::on_asyncFilters_currentRowChanged(int row)
  650. {
  651. UpdatePropertiesView(row, true);
  652. }
  653. void OBSBasicFilters::on_addEffectFilter_clicked()
  654. {
  655. QScopedPointer<QMenu> popup(CreateAddFilterPopupMenu(false));
  656. if (popup)
  657. popup->exec(QCursor::pos());
  658. }
  659. void OBSBasicFilters::on_removeEffectFilter_clicked()
  660. {
  661. OBSSource filter = GetFilter(ui->effectFilters->currentRow(), false);
  662. if (filter) {
  663. if (QueryRemove(this, filter)) {
  664. obs_data_t *wrapper = obs_save_source(filter);
  665. std::string parent_name(obs_source_get_name(source));
  666. obs_data_set_string(wrapper, "undo_name",
  667. parent_name.c_str());
  668. std::string scene_name = obs_source_get_name(
  669. reinterpret_cast<OBSBasic *>(
  670. App()->GetMainWindow())
  671. ->GetCurrentSceneSource());
  672. auto undo = [scene_name](const std::string &data) {
  673. obs_source_t *ssource = obs_get_source_by_name(
  674. scene_name.c_str());
  675. reinterpret_cast<OBSBasic *>(
  676. App()->GetMainWindow())
  677. ->SetCurrentScene(ssource);
  678. obs_source_release(ssource);
  679. obs_data_t *dat =
  680. obs_data_create_from_json(data.c_str());
  681. obs_source_t *source = obs_get_source_by_name(
  682. obs_data_get_string(dat, "undo_name"));
  683. obs_source_t *filter = obs_load_source(dat);
  684. obs_source_filter_add(source, filter);
  685. obs_data_release(dat);
  686. obs_source_release(source);
  687. obs_source_release(filter);
  688. };
  689. obs_data_t *rwrapper = obs_data_create();
  690. obs_data_set_string(rwrapper, "fname",
  691. obs_source_get_name(filter));
  692. obs_data_set_string(rwrapper, "sname",
  693. parent_name.c_str());
  694. auto redo = [scene_name](const std::string &data) {
  695. obs_source_t *ssource = obs_get_source_by_name(
  696. scene_name.c_str());
  697. reinterpret_cast<OBSBasic *>(
  698. App()->GetMainWindow())
  699. ->SetCurrentScene(ssource);
  700. obs_source_release(ssource);
  701. obs_data_t *dat =
  702. obs_data_create_from_json(data.c_str());
  703. obs_source_t *source = obs_get_source_by_name(
  704. obs_data_get_string(dat, "sname"));
  705. obs_source_t *filter =
  706. obs_source_get_filter_by_name(
  707. source, obs_data_get_string(
  708. dat, "fname"));
  709. obs_source_filter_remove(source, filter);
  710. obs_data_release(dat);
  711. obs_source_release(filter);
  712. obs_source_release(source);
  713. };
  714. std::string undo_data(obs_data_get_json(wrapper));
  715. std::string redo_data(obs_data_get_json(rwrapper));
  716. main->undo_s.add_action(
  717. QTStr("Undo.Delete")
  718. .arg(obs_source_get_name(filter)),
  719. undo, redo, undo_data, redo_data, NULL);
  720. obs_source_filter_remove(source, filter);
  721. obs_data_release(wrapper);
  722. obs_data_release(rwrapper);
  723. }
  724. }
  725. }
  726. void OBSBasicFilters::on_moveEffectFilterUp_clicked()
  727. {
  728. OBSSource filter = GetFilter(ui->effectFilters->currentRow(), false);
  729. if (filter)
  730. obs_source_filter_set_order(source, filter, OBS_ORDER_MOVE_UP);
  731. }
  732. void OBSBasicFilters::on_moveEffectFilterDown_clicked()
  733. {
  734. OBSSource filter = GetFilter(ui->effectFilters->currentRow(), false);
  735. if (filter)
  736. obs_source_filter_set_order(source, filter,
  737. OBS_ORDER_MOVE_DOWN);
  738. }
  739. void OBSBasicFilters::on_effectFilters_GotFocus()
  740. {
  741. UpdatePropertiesView(ui->effectFilters->currentRow(), false);
  742. isAsync = false;
  743. }
  744. void OBSBasicFilters::on_effectFilters_currentRowChanged(int row)
  745. {
  746. UpdatePropertiesView(row, false);
  747. }
  748. void OBSBasicFilters::on_actionRemoveFilter_triggered()
  749. {
  750. if (ui->asyncFilters->hasFocus())
  751. on_removeAsyncFilter_clicked();
  752. else if (ui->effectFilters->hasFocus())
  753. on_removeEffectFilter_clicked();
  754. }
  755. void OBSBasicFilters::on_actionMoveUp_triggered()
  756. {
  757. if (ui->asyncFilters->hasFocus())
  758. on_moveAsyncFilterUp_clicked();
  759. else if (ui->effectFilters->hasFocus())
  760. on_moveEffectFilterUp_clicked();
  761. }
  762. void OBSBasicFilters::on_actionMoveDown_triggered()
  763. {
  764. if (ui->asyncFilters->hasFocus())
  765. on_moveAsyncFilterDown_clicked();
  766. else if (ui->effectFilters->hasFocus())
  767. on_moveEffectFilterDown_clicked();
  768. }
  769. void OBSBasicFilters::CustomContextMenu(const QPoint &pos, bool async)
  770. {
  771. QListWidget *list = async ? ui->asyncFilters : ui->effectFilters;
  772. QListWidgetItem *item = list->itemAt(pos);
  773. QMenu popup(window());
  774. QPointer<QMenu> addMenu = CreateAddFilterPopupMenu(async);
  775. if (addMenu)
  776. popup.addMenu(addMenu);
  777. if (item) {
  778. const char *dulpicateSlot =
  779. async ? SLOT(DuplicateAsyncFilter())
  780. : SLOT(DuplicateEffectFilter());
  781. const char *renameSlot = async ? SLOT(RenameAsyncFilter())
  782. : SLOT(RenameEffectFilter());
  783. const char *removeSlot =
  784. async ? SLOT(on_removeAsyncFilter_clicked())
  785. : SLOT(on_removeEffectFilter_clicked());
  786. popup.addSeparator();
  787. popup.addAction(QTStr("Duplicate"), this, dulpicateSlot);
  788. popup.addSeparator();
  789. popup.addAction(QTStr("Rename"), this, renameSlot);
  790. popup.addAction(QTStr("Remove"), this, removeSlot);
  791. popup.addSeparator();
  792. QAction *copyAction = new QAction(QTStr("Copy"));
  793. connect(copyAction, SIGNAL(triggered()), this,
  794. SLOT(CopyFilter()));
  795. copyAction->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_C));
  796. ui->effectWidget->addAction(copyAction);
  797. ui->asyncWidget->addAction(copyAction);
  798. popup.addAction(copyAction);
  799. }
  800. QAction *pasteAction = new QAction(QTStr("Paste"));
  801. pasteAction->setEnabled(main->copyFilter);
  802. connect(pasteAction, SIGNAL(triggered()), this, SLOT(PasteFilter()));
  803. pasteAction->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_V));
  804. ui->effectWidget->addAction(pasteAction);
  805. ui->asyncWidget->addAction(pasteAction);
  806. popup.addAction(pasteAction);
  807. popup.exec(QCursor::pos());
  808. }
  809. void OBSBasicFilters::EditItem(QListWidgetItem *item, bool async)
  810. {
  811. if (editActive)
  812. return;
  813. Qt::ItemFlags flags = item->flags();
  814. OBSSource filter = item->data(Qt::UserRole).value<OBSSource>();
  815. const char *name = obs_source_get_name(filter);
  816. QListWidget *list = async ? ui->asyncFilters : ui->effectFilters;
  817. item->setText(QT_UTF8(name));
  818. item->setFlags(flags | Qt::ItemIsEditable);
  819. list->removeItemWidget(item);
  820. list->editItem(item);
  821. item->setFlags(flags);
  822. editActive = true;
  823. }
  824. void OBSBasicFilters::DuplicateItem(QListWidgetItem *item)
  825. {
  826. OBSSource filter = item->data(Qt::UserRole).value<OBSSource>();
  827. string name = obs_source_get_name(filter);
  828. obs_source_t *existing_filter;
  829. QString placeholder = QString::fromStdString(name);
  830. QString text{placeholder};
  831. int i = 2;
  832. while ((existing_filter = obs_source_get_filter_by_name(
  833. source, QT_TO_UTF8(text)))) {
  834. obs_source_release(existing_filter);
  835. text = QString("%1 %2").arg(placeholder).arg(i++);
  836. }
  837. bool success = NameDialog::AskForName(
  838. this, QTStr("Basic.Filters.AddFilter.Title"),
  839. QTStr("Basic.Filters.AddFilter.Text"), name, text);
  840. if (!success)
  841. return;
  842. if (name.empty()) {
  843. OBSMessageBox::warning(this, QTStr("NoNameEntered.Title"),
  844. QTStr("NoNameEntered.Text"));
  845. DuplicateItem(item);
  846. return;
  847. }
  848. existing_filter = obs_source_get_filter_by_name(source, name.c_str());
  849. if (existing_filter) {
  850. OBSMessageBox::warning(this, QTStr("NameExists.Title"),
  851. QTStr("NameExists.Text"));
  852. obs_source_release(existing_filter);
  853. DuplicateItem(item);
  854. return;
  855. }
  856. bool enabled = obs_source_enabled(filter);
  857. obs_source_t *new_filter =
  858. obs_source_duplicate(filter, name.c_str(), false);
  859. if (new_filter) {
  860. const char *sourceName = obs_source_get_name(source);
  861. const char *id = obs_source_get_id(new_filter);
  862. blog(LOG_INFO,
  863. "User duplicated filter '%s' (%s) from '%s' "
  864. "to source '%s'",
  865. name.c_str(), id, name.c_str(), sourceName);
  866. obs_source_set_enabled(new_filter, enabled);
  867. obs_source_filter_add(source, new_filter);
  868. obs_source_release(new_filter);
  869. }
  870. }
  871. void OBSBasicFilters::on_asyncFilters_customContextMenuRequested(
  872. const QPoint &pos)
  873. {
  874. CustomContextMenu(pos, true);
  875. }
  876. void OBSBasicFilters::on_effectFilters_customContextMenuRequested(
  877. const QPoint &pos)
  878. {
  879. CustomContextMenu(pos, false);
  880. }
  881. void OBSBasicFilters::RenameAsyncFilter()
  882. {
  883. EditItem(ui->asyncFilters->currentItem(), true);
  884. }
  885. void OBSBasicFilters::RenameEffectFilter()
  886. {
  887. EditItem(ui->effectFilters->currentItem(), false);
  888. }
  889. void OBSBasicFilters::DuplicateAsyncFilter()
  890. {
  891. DuplicateItem(ui->asyncFilters->currentItem());
  892. }
  893. void OBSBasicFilters::DuplicateEffectFilter()
  894. {
  895. DuplicateItem(ui->effectFilters->currentItem());
  896. }
  897. void OBSBasicFilters::FilterNameEdited(QWidget *editor, QListWidget *list)
  898. {
  899. QListWidgetItem *listItem = list->currentItem();
  900. OBSSource filter = listItem->data(Qt::UserRole).value<OBSSource>();
  901. QLineEdit *edit = qobject_cast<QLineEdit *>(editor);
  902. string name = QT_TO_UTF8(edit->text().trimmed());
  903. const char *prevName = obs_source_get_name(filter);
  904. bool sameName = (name == prevName);
  905. obs_source_t *foundFilter = nullptr;
  906. if (!sameName)
  907. foundFilter =
  908. obs_source_get_filter_by_name(source, name.c_str());
  909. if (foundFilter || name.empty() || sameName) {
  910. listItem->setText(QT_UTF8(prevName));
  911. if (foundFilter) {
  912. OBSMessageBox::information(window(),
  913. QTStr("NameExists.Title"),
  914. QTStr("NameExists.Text"));
  915. obs_source_release(foundFilter);
  916. } else if (name.empty()) {
  917. OBSMessageBox::information(window(),
  918. QTStr("NoNameEntered.Title"),
  919. QTStr("NoNameEntered.Text"));
  920. }
  921. } else {
  922. const char *sourceName = obs_source_get_name(source);
  923. blog(LOG_INFO,
  924. "User renamed filter '%s' on source '%s' to '%s'",
  925. prevName, sourceName, name.c_str());
  926. listItem->setText(QT_UTF8(name.c_str()));
  927. obs_source_set_name(filter, name.c_str());
  928. std::string scene_name = obs_source_get_name(
  929. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  930. ->GetCurrentSceneSource());
  931. auto undo = [scene_name, prev = std::string(prevName),
  932. name](const std::string &data) {
  933. obs_source_t *ssource =
  934. obs_get_source_by_name(scene_name.c_str());
  935. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  936. ->SetCurrentScene(ssource);
  937. obs_source_release(ssource);
  938. obs_source_t *source =
  939. obs_get_source_by_name(data.c_str());
  940. obs_source_t *filter = obs_source_get_filter_by_name(
  941. source, name.c_str());
  942. obs_source_set_name(filter, prev.c_str());
  943. obs_source_release(source);
  944. obs_source_release(filter);
  945. };
  946. auto redo = [scene_name, prev = std::string(prevName),
  947. name](const std::string &data) {
  948. obs_source_t *ssource =
  949. obs_get_source_by_name(scene_name.c_str());
  950. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  951. ->SetCurrentScene(ssource);
  952. obs_source_release(ssource);
  953. obs_source_t *source =
  954. obs_get_source_by_name(data.c_str());
  955. obs_source_t *filter = obs_source_get_filter_by_name(
  956. source, prev.c_str());
  957. obs_source_set_name(filter, name.c_str());
  958. obs_source_release(source);
  959. obs_source_release(filter);
  960. };
  961. std::string undo_data(sourceName);
  962. std::string redo_data(sourceName);
  963. main->undo_s.add_action(QTStr("Undo.Rename").arg(name.c_str()),
  964. undo, redo, undo_data, redo_data, NULL);
  965. }
  966. listItem->setText(QString());
  967. SetupVisibilityItem(list, listItem, filter);
  968. editActive = false;
  969. }
  970. void OBSBasicFilters::AsyncFilterNameEdited(
  971. QWidget *editor, QAbstractItemDelegate::EndEditHint endHint)
  972. {
  973. FilterNameEdited(editor, ui->asyncFilters);
  974. UNUSED_PARAMETER(endHint);
  975. }
  976. void OBSBasicFilters::EffectFilterNameEdited(
  977. QWidget *editor, QAbstractItemDelegate::EndEditHint endHint)
  978. {
  979. FilterNameEdited(editor, ui->effectFilters);
  980. UNUSED_PARAMETER(endHint);
  981. }
  982. void OBSBasicFilters::ResetFilters()
  983. {
  984. QListWidget *list = isAsync ? ui->asyncFilters : ui->effectFilters;
  985. int row = list->currentRow();
  986. OBSSource filter = GetFilter(row, isAsync);
  987. if (!filter)
  988. return;
  989. obs_data_t *settings = obs_source_get_settings(filter);
  990. obs_data_clear(settings);
  991. obs_data_release(settings);
  992. if (!view->DeferUpdate())
  993. obs_source_update(filter, nullptr);
  994. view->RefreshProperties();
  995. }
  996. void OBSBasicFilters::CopyFilter()
  997. {
  998. OBSSource filter = nullptr;
  999. if (isAsync)
  1000. filter = GetFilter(ui->asyncFilters->currentRow(), true);
  1001. else
  1002. filter = GetFilter(ui->effectFilters->currentRow(), false);
  1003. main->copyFilter = OBSGetWeakRef(filter);
  1004. }
  1005. void OBSBasicFilters::PasteFilter()
  1006. {
  1007. OBSSource filter = OBSGetStrongRef(main->copyFilter);
  1008. if (!filter)
  1009. return;
  1010. obs_source_copy_single_filter(source, filter);
  1011. }