|
@@ -15,172 +15,12 @@
|
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
******************************************************************************/
|
|
|
|
|
|
-#include "moc_window-importer.cpp"
|
|
|
+#include "ImporterModel.hpp"
|
|
|
|
|
|
-#include "obs-app.hpp"
|
|
|
+#include <OBSApp.hpp>
|
|
|
+#include <importers/importers.hpp>
|
|
|
|
|
|
-#include <QPushButton>
|
|
|
-#include <QLineEdit>
|
|
|
-#include <QToolButton>
|
|
|
-#include <QMimeData>
|
|
|
-#include <QStyledItemDelegate>
|
|
|
-#include <QDirIterator>
|
|
|
-#include <QDropEvent>
|
|
|
-#include <qt-wrappers.hpp>
|
|
|
-
|
|
|
-#include "importers/importers.hpp"
|
|
|
-
|
|
|
-extern bool SceneCollectionExists(const char *findName);
|
|
|
-
|
|
|
-enum ImporterColumn {
|
|
|
- Selected,
|
|
|
- Name,
|
|
|
- Path,
|
|
|
- Program,
|
|
|
-
|
|
|
- Count
|
|
|
-};
|
|
|
-
|
|
|
-enum ImporterEntryRole { EntryStateRole = Qt::UserRole, NewPath, AutoPath, CheckEmpty };
|
|
|
-
|
|
|
-/**********************************************************
|
|
|
- Delegate - Presents cells in the grid.
|
|
|
-**********************************************************/
|
|
|
-
|
|
|
-ImporterEntryPathItemDelegate::ImporterEntryPathItemDelegate() : QStyledItemDelegate() {}
|
|
|
-
|
|
|
-QWidget *ImporterEntryPathItemDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem & /* option */,
|
|
|
- const QModelIndex &index) const
|
|
|
-{
|
|
|
- bool empty = index.model()
|
|
|
- ->index(index.row(), ImporterColumn::Path)
|
|
|
- .data(ImporterEntryRole::CheckEmpty)
|
|
|
- .value<bool>();
|
|
|
-
|
|
|
- QSizePolicy buttonSizePolicy(QSizePolicy::Policy::Minimum, QSizePolicy::Policy::Expanding,
|
|
|
- QSizePolicy::ControlType::PushButton);
|
|
|
-
|
|
|
- QWidget *container = new QWidget(parent);
|
|
|
-
|
|
|
- auto browseCallback = [this, container]() {
|
|
|
- const_cast<ImporterEntryPathItemDelegate *>(this)->handleBrowse(container);
|
|
|
- };
|
|
|
-
|
|
|
- auto clearCallback = [this, container]() {
|
|
|
- const_cast<ImporterEntryPathItemDelegate *>(this)->handleClear(container);
|
|
|
- };
|
|
|
-
|
|
|
- QHBoxLayout *layout = new QHBoxLayout();
|
|
|
- layout->setContentsMargins(0, 0, 0, 0);
|
|
|
- layout->setSpacing(0);
|
|
|
-
|
|
|
- QLineEdit *text = new QLineEdit();
|
|
|
- text->setObjectName(QStringLiteral("text"));
|
|
|
- text->setSizePolicy(QSizePolicy(QSizePolicy::Policy::Expanding, QSizePolicy::Policy::Expanding,
|
|
|
- QSizePolicy::ControlType::LineEdit));
|
|
|
- layout->addWidget(text);
|
|
|
-
|
|
|
- QObject::connect(text, &QLineEdit::editingFinished, this, &ImporterEntryPathItemDelegate::updateText);
|
|
|
-
|
|
|
- QToolButton *browseButton = new QToolButton();
|
|
|
- browseButton->setText("...");
|
|
|
- browseButton->setSizePolicy(buttonSizePolicy);
|
|
|
- layout->addWidget(browseButton);
|
|
|
-
|
|
|
- container->connect(browseButton, &QToolButton::clicked, browseCallback);
|
|
|
-
|
|
|
- // The "clear" button is not shown in output cells
|
|
|
- // or the insertion point's input cell.
|
|
|
- if (!empty) {
|
|
|
- QToolButton *clearButton = new QToolButton();
|
|
|
- clearButton->setText("X");
|
|
|
- clearButton->setSizePolicy(buttonSizePolicy);
|
|
|
- layout->addWidget(clearButton);
|
|
|
-
|
|
|
- container->connect(clearButton, &QToolButton::clicked, clearCallback);
|
|
|
- }
|
|
|
-
|
|
|
- container->setLayout(layout);
|
|
|
- container->setFocusProxy(text);
|
|
|
- return container;
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
|
|
|
-{
|
|
|
- QLineEdit *text = editor->findChild<QLineEdit *>();
|
|
|
- text->setText(index.data().toString());
|
|
|
- editor->setProperty(PATH_LIST_PROP, QVariant());
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::setModelData(QWidget *editor, QAbstractItemModel *model,
|
|
|
- const QModelIndex &index) const
|
|
|
-{
|
|
|
- // We use the PATH_LIST_PROP property to pass a list of
|
|
|
- // path strings from the editor widget into the model's
|
|
|
- // NewPathsToProcessRole. This is only used when paths
|
|
|
- // are selected through the "browse" or "delete" buttons
|
|
|
- // in the editor. If the user enters new text in the
|
|
|
- // text box, we simply pass that text on to the model
|
|
|
- // as normal text data in the default role.
|
|
|
- QVariant pathListProp = editor->property(PATH_LIST_PROP);
|
|
|
- if (pathListProp.isValid()) {
|
|
|
- QStringList list = editor->property(PATH_LIST_PROP).toStringList();
|
|
|
- model->setData(index, list, ImporterEntryRole::NewPath);
|
|
|
- } else {
|
|
|
- QLineEdit *lineEdit = editor->findChild<QLineEdit *>();
|
|
|
- model->setData(index, lineEdit->text());
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option,
|
|
|
- const QModelIndex &index) const
|
|
|
-{
|
|
|
- QStyleOptionViewItem localOption = option;
|
|
|
- initStyleOption(&localOption, index);
|
|
|
-
|
|
|
- QApplication::style()->drawControl(QStyle::CE_ItemViewItem, &localOption, painter);
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::handleBrowse(QWidget *container)
|
|
|
-{
|
|
|
- QString Pattern = "(*.json *.bpres *.xml *.xconfig)";
|
|
|
-
|
|
|
- QLineEdit *text = container->findChild<QLineEdit *>();
|
|
|
-
|
|
|
- QString currentPath = text->text();
|
|
|
-
|
|
|
- bool isSet = false;
|
|
|
- QStringList paths = OpenFiles(container, QTStr("Importer.SelectCollection"), currentPath,
|
|
|
- QTStr("Importer.Collection") + QString(" ") + Pattern);
|
|
|
-
|
|
|
- if (!paths.empty()) {
|
|
|
- container->setProperty(PATH_LIST_PROP, paths);
|
|
|
- isSet = true;
|
|
|
- }
|
|
|
-
|
|
|
- if (isSet)
|
|
|
- emit commitData(container);
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::handleClear(QWidget *container)
|
|
|
-{
|
|
|
- // An empty string list will indicate that the entry is being
|
|
|
- // blanked and should be deleted.
|
|
|
- container->setProperty(PATH_LIST_PROP, QStringList());
|
|
|
-
|
|
|
- emit commitData(container);
|
|
|
-}
|
|
|
-
|
|
|
-void ImporterEntryPathItemDelegate::updateText()
|
|
|
-{
|
|
|
- QLineEdit *lineEdit = dynamic_cast<QLineEdit *>(sender());
|
|
|
- QWidget *editor = lineEdit->parentWidget();
|
|
|
- emit commitData(editor);
|
|
|
-}
|
|
|
-
|
|
|
-/**
|
|
|
- Model
|
|
|
-**/
|
|
|
+#include "moc_ImporterModel.cpp"
|
|
|
|
|
|
int ImporterModel::rowCount(const QModelIndex &) const
|
|
|
{
|
|
@@ -364,207 +204,3 @@ QVariant ImporterModel::headerData(int section, Qt::Orientation orientation, int
|
|
|
|
|
|
return result;
|
|
|
}
|
|
|
-
|
|
|
-/**
|
|
|
- Window
|
|
|
-**/
|
|
|
-
|
|
|
-OBSImporter::OBSImporter(QWidget *parent) : QDialog(parent), optionsModel(new ImporterModel), ui(new Ui::OBSImporter)
|
|
|
-{
|
|
|
- setAcceptDrops(true);
|
|
|
-
|
|
|
- setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
|
|
|
-
|
|
|
- ui->setupUi(this);
|
|
|
-
|
|
|
- ui->tableView->setModel(optionsModel);
|
|
|
- ui->tableView->setItemDelegateForColumn(ImporterColumn::Path, new ImporterEntryPathItemDelegate());
|
|
|
- ui->tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeMode::ResizeToContents);
|
|
|
- ui->tableView->horizontalHeader()->setSectionResizeMode(ImporterColumn::Path, QHeaderView::ResizeMode::Stretch);
|
|
|
-
|
|
|
- connect(optionsModel, &ImporterModel::dataChanged, this, &OBSImporter::dataChanged);
|
|
|
-
|
|
|
- ui->tableView->setEditTriggers(QAbstractItemView::EditTrigger::CurrentChanged);
|
|
|
-
|
|
|
- ui->buttonBox->button(QDialogButtonBox::Ok)->setText(QTStr("Import"));
|
|
|
- ui->buttonBox->button(QDialogButtonBox::Open)->setText(QTStr("Add"));
|
|
|
-
|
|
|
- connect(ui->buttonBox->button(QDialogButtonBox::Ok), &QPushButton::clicked, this,
|
|
|
- &OBSImporter::importCollections);
|
|
|
- connect(ui->buttonBox->button(QDialogButtonBox::Open), &QPushButton::clicked, this, &OBSImporter::browseImport);
|
|
|
- connect(ui->buttonBox->button(QDialogButtonBox::Close), &QPushButton::clicked, this, &OBSImporter::close);
|
|
|
-
|
|
|
- ImportersInit();
|
|
|
-
|
|
|
- bool autoSearchPrompt = config_get_bool(App()->GetUserConfig(), "General", "AutoSearchPrompt");
|
|
|
-
|
|
|
- if (!autoSearchPrompt) {
|
|
|
- QMessageBox::StandardButton button = OBSMessageBox::question(
|
|
|
- parent, QTStr("Importer.AutomaticCollectionPrompt"), QTStr("Importer.AutomaticCollectionText"));
|
|
|
-
|
|
|
- if (button == QMessageBox::Yes) {
|
|
|
- config_set_bool(App()->GetUserConfig(), "General", "AutomaticCollectionSearch", true);
|
|
|
- } else {
|
|
|
- config_set_bool(App()->GetUserConfig(), "General", "AutomaticCollectionSearch", false);
|
|
|
- }
|
|
|
-
|
|
|
- config_set_bool(App()->GetUserConfig(), "General", "AutoSearchPrompt", true);
|
|
|
- }
|
|
|
-
|
|
|
- bool autoSearch = config_get_bool(App()->GetUserConfig(), "General", "AutomaticCollectionSearch");
|
|
|
-
|
|
|
- OBSImporterFiles f;
|
|
|
- if (autoSearch)
|
|
|
- f = ImportersFindFiles();
|
|
|
-
|
|
|
- for (size_t i = 0; i < f.size(); i++) {
|
|
|
- QString path = f[i].c_str();
|
|
|
- path.replace("\\", "/");
|
|
|
- addImportOption(path, true);
|
|
|
- }
|
|
|
-
|
|
|
- f.clear();
|
|
|
-
|
|
|
- ui->tableView->resizeColumnsToContents();
|
|
|
-
|
|
|
- QModelIndex index = optionsModel->createIndex(optionsModel->rowCount() - 1, 2);
|
|
|
- QMetaObject::invokeMethod(ui->tableView, "setCurrentIndex", Qt::QueuedConnection,
|
|
|
- Q_ARG(const QModelIndex &, index));
|
|
|
-}
|
|
|
-
|
|
|
-void OBSImporter::addImportOption(QString path, bool automatic)
|
|
|
-{
|
|
|
- QStringList list;
|
|
|
-
|
|
|
- list.append(path);
|
|
|
-
|
|
|
- QModelIndex insertIndex = optionsModel->index(optionsModel->rowCount() - 1, ImporterColumn::Path);
|
|
|
-
|
|
|
- optionsModel->setData(insertIndex, list, automatic ? ImporterEntryRole::AutoPath : ImporterEntryRole::NewPath);
|
|
|
-}
|
|
|
-
|
|
|
-void OBSImporter::dropEvent(QDropEvent *ev)
|
|
|
-{
|
|
|
- for (QUrl url : ev->mimeData()->urls()) {
|
|
|
- QFileInfo fileInfo(url.toLocalFile());
|
|
|
- if (fileInfo.isDir()) {
|
|
|
-
|
|
|
- QDirIterator dirIter(fileInfo.absoluteFilePath(), QDir::Files);
|
|
|
-
|
|
|
- while (dirIter.hasNext()) {
|
|
|
- addImportOption(dirIter.next(), false);
|
|
|
- }
|
|
|
- } else {
|
|
|
- addImportOption(fileInfo.canonicalFilePath(), false);
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-void OBSImporter::dragEnterEvent(QDragEnterEvent *ev)
|
|
|
-{
|
|
|
- if (ev->mimeData()->hasUrls())
|
|
|
- ev->accept();
|
|
|
-}
|
|
|
-
|
|
|
-void OBSImporter::browseImport()
|
|
|
-{
|
|
|
- QString Pattern = "(*.json *.bpres *.xml *.xconfig)";
|
|
|
-
|
|
|
- QStringList paths = OpenFiles(this, QTStr("Importer.SelectCollection"), "",
|
|
|
- QTStr("Importer.Collection") + QString(" ") + Pattern);
|
|
|
-
|
|
|
- if (!paths.empty()) {
|
|
|
- for (int i = 0; i < paths.count(); i++) {
|
|
|
- addImportOption(paths[i], false);
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-bool GetUnusedName(std::string &name)
|
|
|
-{
|
|
|
- OBSBasic *basic = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
|
|
|
-
|
|
|
- if (!basic->GetSceneCollectionByName(name)) {
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- std::string newName;
|
|
|
- int inc = 2;
|
|
|
- do {
|
|
|
- newName = name;
|
|
|
- newName += " ";
|
|
|
- newName += std::to_string(inc++);
|
|
|
- } while (basic->GetSceneCollectionByName(newName));
|
|
|
-
|
|
|
- name = newName;
|
|
|
- return true;
|
|
|
-}
|
|
|
-
|
|
|
-constexpr std::string_view OBSSceneCollectionPath = "obs-studio/basic/scenes/";
|
|
|
-
|
|
|
-void OBSImporter::importCollections()
|
|
|
-{
|
|
|
- setEnabled(false);
|
|
|
-
|
|
|
- const std::filesystem::path sceneCollectionLocation =
|
|
|
- App()->userScenesLocation / std::filesystem::u8path(OBSSceneCollectionPath);
|
|
|
-
|
|
|
- for (int i = 0; i < optionsModel->rowCount() - 1; i++) {
|
|
|
- int selected = optionsModel->index(i, ImporterColumn::Selected).data(Qt::CheckStateRole).value<int>();
|
|
|
-
|
|
|
- if (selected == Qt::Unchecked)
|
|
|
- continue;
|
|
|
-
|
|
|
- std::string pathStr = optionsModel->index(i, ImporterColumn::Path)
|
|
|
- .data(Qt::DisplayRole)
|
|
|
- .value<QString>()
|
|
|
- .toStdString();
|
|
|
- std::string nameStr = optionsModel->index(i, ImporterColumn::Name)
|
|
|
- .data(Qt::DisplayRole)
|
|
|
- .value<QString>()
|
|
|
- .toStdString();
|
|
|
-
|
|
|
- json11::Json res;
|
|
|
- ImportSC(pathStr, nameStr, res);
|
|
|
-
|
|
|
- if (res != json11::Json()) {
|
|
|
- json11::Json::object out = res.object_items();
|
|
|
- std::string name = res["name"].string_value();
|
|
|
- std::string file;
|
|
|
-
|
|
|
- if (GetUnusedName(name)) {
|
|
|
- json11::Json::object newOut = out;
|
|
|
- newOut["name"] = name;
|
|
|
- out = newOut;
|
|
|
- }
|
|
|
-
|
|
|
- std::string fileName;
|
|
|
- if (!GetFileSafeName(name.c_str(), fileName)) {
|
|
|
- blog(LOG_WARNING, "Failed to create safe file name for '%s'", fileName.c_str());
|
|
|
- }
|
|
|
-
|
|
|
- std::string collectionFile;
|
|
|
- collectionFile.reserve(sceneCollectionLocation.u8string().size() + fileName.size());
|
|
|
- collectionFile.append(sceneCollectionLocation.u8string()).append(fileName);
|
|
|
-
|
|
|
- if (!GetClosestUnusedFileName(collectionFile, "json")) {
|
|
|
- blog(LOG_WARNING, "Failed to get closest file name for %s", fileName.c_str());
|
|
|
- }
|
|
|
-
|
|
|
- std::string out_str = json11::Json(out).dump();
|
|
|
-
|
|
|
- bool success = os_quick_write_utf8_file(collectionFile.c_str(), out_str.c_str(), out_str.size(),
|
|
|
- false);
|
|
|
-
|
|
|
- blog(LOG_INFO, "Import Scene Collection: %s (%s) - %s", name.c_str(), fileName.c_str(),
|
|
|
- success ? "SUCCESS" : "FAILURE");
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- close();
|
|
|
-}
|
|
|
-
|
|
|
-void OBSImporter::dataChanged()
|
|
|
-{
|
|
|
- ui->tableView->resizeColumnToContents(ImporterColumn::Name);
|
|
|
-}
|