openrw/rwviewer/models/ObjectListModel.cpp

#include "ObjectListModel.hpp"

ObjectListModel::ObjectListModel(GameWorld *world, QObject *parent) :
	QAbstractTableModel(parent), _world( world )
{
}

int ObjectListModel::rowCount(const QModelIndex &parent) const
{
	return _world->objectTypes.size();
}

int ObjectListModel::columnCount(const QModelIndex &parent) const
{
	return 3;
}

static std::map<ObjectInformation::ObjectClass, QString> gDataType =
{
	{ ObjectInformation::_class("OBJS"), "Object" },
	{ ObjectInformation::_class("CARS"), "Vehicle" },
	{ ObjectInformation::_class("PEDS"), "Pedestrian" },
	{ ObjectInformation::_class("HIER"), "Cutscene" }
};

QVariant ObjectListModel::data(const QModelIndex &index, int role) const
{
	if ( role == Qt::DisplayRole ) {
		auto id = index.internalId();
		if( id == -1 ) return QVariant::Invalid;
		if( index.column() == 0 )
		{
			return id;
		}
		else if ( index.column() == 1 )
		{
			auto object = world()->objectTypes[id];
			if( gDataType[object->class_type].isEmpty() )
			{
				return QString("Unknown");
			}
			return gDataType[object->class_type];
		}
		else if( index.column() == 2 )
		{
			auto object = world()->objectTypes[id];
			if(object->class_type == ObjectData::class_id)
			{
				auto v = std::static_pointer_cast<ObjectData>(object);
				return QString::fromStdString(v->modelName);
			}
			else if(object->class_type == VehicleData::class_id)
			{
				auto v = std::static_pointer_cast<VehicleData>(object);
				return QString::fromStdString(v->modelName);
			}
			else if(object->class_type == CharacterData::class_id)
			{
				auto v = std::static_pointer_cast<CharacterData>(object);
				return QString::fromStdString(v->modelName);
			}
		}
	}
	return QVariant::Invalid;
}

QVariant ObjectListModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	if(role == Qt::DisplayRole && orientation == Qt::Horizontal) {
		switch(section) {
		case 0:
			return "ID";
		case 1:
			return "Type";
		case 2:
			return "Model";
		}
	}
	return QVariant::Invalid;
}

QModelIndex ObjectListModel::index(int row, int column, const QModelIndex &parent) const
{
	auto it = world()->objectTypes.begin();
	for(int i = 0; i < row; i++) it++;
	auto id = it->second->ID;

	return hasIndex(row, column, parent) ? createIndex(row, column, id) : QModelIndex();
}