openrw/rwviewer/views/ObjectViewer.cpp

#include "ObjectViewer.hpp"
#include <QDebug>
#include <QMenu>
#include <models/ObjectListModel.hpp>
#include "ViewerWidget.hpp"

ObjectViewer::ObjectViewer(ViewerWidget* viewer, QWidget* parent,
                           Qt::WindowFlags f)
    : ViewerInterface(parent, f) {
    mainLayout = new QHBoxLayout;

    objectList = new QTableView;

    objectMenu = new QMenu(objectList);
    objectList->setContextMenuPolicy(Qt::CustomContextMenu);
    auto viewModelAction = new QAction("View Model", objectMenu);
    objectMenu->addAction(viewModelAction);
    connect(viewModelAction, SIGNAL(triggered()), this, SLOT(menuViewModel()));
    connect(objectList, SIGNAL(customContextMenuRequested(QPoint)), this,
            SLOT(onCustomContextMenu(QPoint)));

    mainLayout->addWidget(objectList);

    previewWidget = viewer;
    previewWidget->setMinimumSize(250, 250);

    infoLayout = new QGridLayout;

    previewID = new QLabel;
    previewModel = new QLabel;
    previewClass = new QLabel;
    infoLayout->addWidget(new QLabel("ID"), 1, 0);
    infoLayout->addWidget(previewID, 1, 1);
    infoLayout->addWidget(new QLabel("Type"), 2, 0);
    infoLayout->addWidget(previewClass, 2, 1);
    infoLayout->addWidget(new QLabel("Model"), 3, 0);
    infoLayout->addWidget(previewModel, 3, 1);

    mainLayout->addLayout(infoLayout);

    this->setLayout(mainLayout);

    setViewerWidget(previewWidget);
}

void ObjectViewer::setViewerWidget(ViewerWidget* widget) {
    // widgetLayout->removeWidget(previewWidget);
    previewWidget = widget;
    infoLayout->addWidget(previewWidget, 0, 0, 1, 2);
}

void ObjectViewer::worldChanged() {
    if (objectList->model()) {
        delete objectList->model();
    }

    objectList->setModel(new ObjectListModel(world()->data, objectList));
    connect(objectList->selectionModel(),
            SIGNAL(currentChanged(QModelIndex, QModelIndex)), this,
            SLOT(showItem(QModelIndex)));
}

void ObjectViewer::showItem(qint16 item) {
    auto def = world()->data->modelinfo[item].get();

    if (def) {
        previewID->setText(QString::number(def->id()));
        previewClass->setText(
            QString::fromStdString(BaseModelInfo::getTypeName(def->type())));
        previewModel->setText(QString::fromStdString(def->name));
        previewWidget->showObject(item);
    }
}

void ObjectViewer::showItem(QModelIndex model) {
    showItem(model.internalId());
}

void ObjectViewer::onCustomContextMenu(const QPoint& p) {
    contextMenuIndex = objectList->indexAt(p);
    if (contextMenuIndex.isValid()) {
        objectMenu->exec(objectList->mapToGlobal(p));
    }
}

void ObjectViewer::menuViewModel() {
    if (contextMenuIndex.isValid()) {
        auto id = contextMenuIndex.internalId();
        showObjectModel(id);
    }
}