// SPDX-FileCopyrightText: 2002-2024 PCSX2 Dev Team
// SPDX-License-Identifier: LGPL-3.0+

#include "PrecompiledHeader.h"
#include "SavedAddressesModel.h"

SavedAddressesModel::SavedAddressesModel(DebugInterface& cpu, QObject* parent)
	: QAbstractTableModel(parent)
	, m_cpu(cpu)
{
}

QVariant SavedAddressesModel::data(const QModelIndex& index, int role) const
{
	if (role == Qt::CheckStateRole)
	{
		return QVariant();
	}

	if (role == Qt::DisplayRole || role == Qt::EditRole)
	{
		SavedAddress savedAddress = m_savedAddresses.at(index.row());
		switch (index.column())
		{
			case HeaderColumns::ADDRESS:
				return QString::number(savedAddress.address, 16).toUpper();
			case HeaderColumns::LABEL:
				return savedAddress.label;
			case HeaderColumns::DESCRIPTION:
				return savedAddress.description;
		}
	}
	if (role == Qt::UserRole)
	{
		SavedAddress savedAddress = m_savedAddresses.at(index.row());
		switch (index.column())
		{
			case HeaderColumns::ADDRESS:
				return savedAddress.address;
			case HeaderColumns::LABEL:
				return savedAddress.label;
			case HeaderColumns::DESCRIPTION:
				return savedAddress.description;
		}
	}
	return QVariant();
}

bool SavedAddressesModel::setData(const QModelIndex& index, const QVariant& value, int role)
{
	if (role == Qt::CheckStateRole)
	{
		return false;
	}

	if (role == Qt::EditRole)
	{
		SavedAddress addressToEdit = m_savedAddresses.at(index.row());
		if (index.column() == HeaderColumns::ADDRESS)
		{
			bool ok = false;
			const u32 address = value.toString().toUInt(&ok, 16);
			if (ok)
				addressToEdit.address = address;
			else
				return false;
		}
		if (index.column() == HeaderColumns::DESCRIPTION)
			addressToEdit.description = value.toString();
		if (index.column() == HeaderColumns::LABEL)
			addressToEdit.label = value.toString();
		m_savedAddresses.at(index.row()) = addressToEdit;

		emit dataChanged(index, index, QList<int>(role));
		return true;
	}
	else if (role == Qt::UserRole)
	{
		SavedAddress addressToEdit = m_savedAddresses.at(index.row());
		if (index.column() == HeaderColumns::ADDRESS)
		{
			const u32 address = value.toUInt();
			addressToEdit.address = address;
		}
		if (index.column() == HeaderColumns::DESCRIPTION)
			addressToEdit.description = value.toString();
		if (index.column() == HeaderColumns::LABEL)
			addressToEdit.label = value.toString();
		m_savedAddresses.at(index.row()) = addressToEdit;

		emit dataChanged(index, index, QList<int>(role));
		return true;
	}
	return false;
}

QVariant SavedAddressesModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	if ((role != Qt::DisplayRole && role != Qt::EditRole) || orientation != Qt::Horizontal)
		return QVariant();

	switch (section)
	{
		case SavedAddressesModel::ADDRESS:
			return tr("MEMORY ADDRESS");
		case SavedAddressesModel::LABEL:
			return tr("LABEL");
		case SavedAddressesModel::DESCRIPTION:
			return tr("DESCRIPTION");
		default:
			return QVariant();
	}
}

Qt::ItemFlags SavedAddressesModel::flags(const QModelIndex& index) const
{
	return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
}

void SavedAddressesModel::addRow()
{
	const SavedAddress defaultNewAddress = {0, "Name", "Description"};
	addRow(defaultNewAddress);
}

void SavedAddressesModel::addRow(SavedAddress addresstoSave)
{
	const int newRowIndex = m_savedAddresses.size();
	beginInsertRows(QModelIndex(), newRowIndex, newRowIndex);
	m_savedAddresses.push_back(addresstoSave);
	endInsertRows();
}

bool SavedAddressesModel::removeRows(int row, int count, const QModelIndex& parent)
{
	if (row + count > m_savedAddresses.size() || row < 0 || count < 1)
		return false;
	beginRemoveRows(parent, row, row + count - 1);
	m_savedAddresses.erase(m_savedAddresses.begin() + row, m_savedAddresses.begin() + row + count);
	endRemoveRows();
	return true;
}

int SavedAddressesModel::rowCount(const QModelIndex&) const
{
	return m_savedAddresses.size();
}

int SavedAddressesModel::columnCount(const QModelIndex&) const
{
	return HeaderColumns::COLUMN_COUNT;
}