terrainarraymodel.cpp

/*-
 * Copyright (c) 2026 Nikolay Burkov <nbrk@linklevel.net>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include "terrainarraymodel.h"

TerrainArrayModel::TerrainArrayModel(QObject *parent)
    : QAbstractTableModel(parent)
    , m_hasMap(false)
{}

void TerrainArrayModel::setUnderlyingMap(std::unique_ptr<tacopsdatmap> dat)
{
        m_dat = std::move(dat);
        m_hasMap = true;
}

QVariant TerrainArrayModel::headerData(int section, Qt::Orientation orientation, int role) const
{
        // FIXME: Implement me!
        // switch (section) {
        // case 0:
        //         return QVariant{""};
        //         break;
        // default:
        //         break;
        // }
        return QVariant{};
}

int TerrainArrayModel::rowCount(const QModelIndex &parent) const
{
        if (parent.isValid() || !m_hasMap)
                return 0;

        return m_dat->terrain_nrows();
}

int TerrainArrayModel::columnCount(const QModelIndex &parent) const
{
        if (parent.isValid() || !m_hasMap)
                return 0;

        return m_dat->terrain_ncolumns();
}

QVariant TerrainArrayModel::data(const QModelIndex &index, int role) const
{
        if (!index.isValid() || !m_hasMap)
                return QVariant();

        auto const tidx = m_dat->column_row_to_index(index.column(), index.row());
        auto const cell = m_dat->terrain_cells()[tidx];

        switch (role) {
        case TerrainTypeRole:
                switch (tacopsdatmap::cell_type(cell)) {
                case tacopsdatmap::terrain_type::clear:
                        return QVariant{"clear"};
                case tacopsdatmap::terrain_type::nogo_wheeled:
                        return QVariant{"nogo_wheeled"};
                case tacopsdatmap::terrain_type::nogo_tracked:
                        return QVariant{"nogo_tracked"};
                case tacopsdatmap::terrain_type::nogo_dismount:
                        return QVariant{"nogo_dismount"};
                case tacopsdatmap::terrain_type::rough_1:
                        return QVariant{"rough_1"};
                case tacopsdatmap::terrain_type::rough_2:
                        return QVariant{"rough_2"};
                case tacopsdatmap::terrain_type::rough_3:
                        return QVariant{"rough_3"};
                case tacopsdatmap::terrain_type::rough_4:
                        return QVariant{"rough_4"};
                case tacopsdatmap::terrain_type::water:
                        return QVariant{"water"};
                }
                break;

        case ElevationRole:
                return QVariant{tacopsdatmap::cell_has_elevation(cell)};
        case RoadRole:
                return QVariant{tacopsdatmap::cell_has_road(cell)};
        case WoodsRole:
                return QVariant{tacopsdatmap::cell_has_woods(cell)};
        case TownRole:
                return QVariant{tacopsdatmap::cell_has_town(cell)};
        default:
                break;
        }

        return QVariant();
}

QHash<int, QByteArray> TerrainArrayModel::roleNames() const
{
        return {
            {TerrainTypeRole, "terrainType"},
            {ElevationRole, "elevation"},
            {RoadRole, "road"},
            {WoodsRole, "woods"},
            {TownRole, "town"},
        };
}