48 lines
1.2 KiB
C++
48 lines
1.2 KiB
C++
#include "CLayerModel.h"
|
|
#include "../UICommon.h"
|
|
|
|
CLayerModel::CLayerModel(QObject *pParent) : QAbstractListModel(pParent)
|
|
{
|
|
mpArea = nullptr;
|
|
mHasGenerateLayer = false;
|
|
}
|
|
|
|
CLayerModel::~CLayerModel()
|
|
{
|
|
}
|
|
|
|
int CLayerModel::rowCount(const QModelIndex& /*parent*/) const
|
|
{
|
|
if (!mpArea) return 0;
|
|
if (mHasGenerateLayer) return mpArea->GetScriptLayerCount() + 1;
|
|
else return mpArea->GetScriptLayerCount();
|
|
}
|
|
|
|
QVariant CLayerModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (mpArea && (role == Qt::DisplayRole) && (index.row() < rowCount(QModelIndex())))
|
|
return TO_QSTRING(Layer(index)->Name());
|
|
|
|
return QVariant::Invalid;
|
|
}
|
|
|
|
void CLayerModel::SetArea(CGameArea *pArea)
|
|
{
|
|
mpArea = pArea;
|
|
mHasGenerateLayer = (pArea->GetGeneratorLayer() != nullptr);
|
|
emit layoutChanged();
|
|
}
|
|
|
|
CScriptLayer* CLayerModel::Layer(const QModelIndex& index) const
|
|
{
|
|
if (!mpArea) return nullptr;
|
|
u32 NumLayers = mpArea->GetScriptLayerCount();
|
|
|
|
if (index.row() < (int) NumLayers)
|
|
return mpArea->GetScriptLayer(index.row());
|
|
if (mHasGenerateLayer && (index.row() == NumLayers))
|
|
return mpArea->GetGeneratorLayer();
|
|
|
|
return nullptr;
|
|
}
|