BloomPatched/src/Insight/UserInterfaces/InsightWindow/UiLoader.cpp

#include "UiLoader.hpp"

#include <QtUiTools>

// Custom widgets
#include "Widgets/Label.hpp"
#include "Widgets/RotatableLabel.hpp"
#include "Widgets/LabeledSeparator.hpp"
#include "Widgets/TextInput.hpp"
#include "Widgets/PushButton.hpp"
#include "Widgets/SvgWidget.hpp"
#include "Widgets/SvgToolButton.hpp"
#include "Widgets/ExpandingHeightScrollAreaWidget.hpp"
#include "Widgets/TargetWidgets/TargetPackageWidgetContainer.hpp"

namespace Bloom
{
    using namespace Bloom::Widgets;

    UiLoader::UiLoader(QObject* parent): QUiLoader(parent) {
        this->customWidgetConstructorsByWidgetName = {
            {
                "Label",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new Label(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "RotatableLabel",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new RotatableLabel("", parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "LabeledSeparator",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new LabeledSeparator(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "TextInput",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new TextInput(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "PushButton",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new PushButton(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "ExpandingHeightScrollAreaWidget",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new ExpandingHeightScrollAreaWidget(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "SvgWidget",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new SvgWidget(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "SvgToolButton",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new SvgToolButton(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
            {
                "TargetPackageWidgetContainer",
                [this] (QWidget* parent, const QString& name) {
                    auto* widget = new InsightTargetWidgets::TargetPackageWidgetContainer(parent);
                    widget->setObjectName(name);
                    widget->setStyleSheet(parent->styleSheet());
                    return widget;
                }
            },
        };
    }

    QWidget* UiLoader::createWidget(const QString& className, QWidget* parent, const QString& name) {
        const auto widgetContructorIt = this->customWidgetConstructorsByWidgetName.find(className);

        if (widgetContructorIt != this->customWidgetConstructorsByWidgetName.end()) {
            // This is a custom widget - call the mapped constructor
            return widgetContructorIt->second(parent, name);
        }

        return QUiLoader::createWidget(className, parent, name);
    }
}