/*
 * changewangsetdata.h
 * Copyright 2017, Benjamin Trotter <bdtrotte@ucsc.edu>
 *
 * This file is part of Tiled.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program. If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once

#include "wangset.h"

#include <QUndoCommand>

namespace Tiled {

class Tileset;
class Tile;

namespace Internal {

class TilesetDocument;

class ChangeWangSetEdgeCount : public QUndoCommand
{
public:
    ChangeWangSetEdgeCount(TilesetDocument *TilesetDocument,
                           WangSet *wangSet,
                           int newValue);

    void undo() override;
    void redo() override;

private:
    struct WangColorChange {
        QSharedPointer<WangColor> wangColor;
        int index;
    };

    TilesetDocument *mTilesetDocument;
    WangSet *mWangSet;
    int mOldValue;
    int mNewValue;
    QVector<WangColorChange> mRemovedWangColors;
};

class ChangeWangSetCornerCount : public QUndoCommand
{
public:
    ChangeWangSetCornerCount(TilesetDocument *TilesetDocument,
                             WangSet *wangSet,
                             int newValue);

    void undo() override;
    void redo() override;

private:
    struct WangColorChange {
        QSharedPointer<WangColor> wangColor;
        int index;
    };

    TilesetDocument *mTilesetDocument;
    WangSet *mWangSet;
    int mOldValue;
    int mNewValue;
    QVector<WangColorChange> mRemovedWangColors;
};

class RemoveWangSetColor : public QUndoCommand
{
public:
    RemoveWangSetColor(TilesetDocument *tilesetDocumnet,
                       WangSet *wangSet,
                       int color,
                       bool isEdge);

    void undo() override;
    void redo() override;

private:
    TilesetDocument *mTilesetDocument;
    WangSet *mWangSet;
    int mColor;
    bool mIsEdge;
    QSharedPointer<WangColor> mRemovedWangColor;
    // When removing a color when there are two, both are actually removed,
    // this stores the extra if needed, and is null otherwise.
    QSharedPointer<WangColor> mExtraWangColor;
};

class SetWangSetImage : public QUndoCommand
{
public:
    SetWangSetImage(TilesetDocument *tilesetDocument,
                    WangSet *wangSet,
                    int tileId,
                    QUndoCommand *parent = nullptr);

    void undo() override;
    void redo() override;

private:
    TilesetDocument *mTilesetDocument;
    WangSet *mWangSet;
    int mOldImageTileId;
    int mNewImageTileId;
};

} // namespace Internal
} // namespace Tiled
