/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#pragma once

#include <AudioControl.h>

#include <AzCore/std/containers/map.h>
#include <AzCore/std/containers/vector.h>
#include <AzCore/std/smart_ptr/shared_ptr.h>

#include <QATLControlsTreeModel.h>
#include <Undo/IUndoObject.h>

#include <QAbstractItemModel>
#include <QString>

class QStandardItem;

namespace AudioControls
{
    typedef AZStd::vector<int> TPath;

    //-------------------------------------------------------------------------------------------//
    class IUndoControlOperation
        : public IUndoObject
    {
    protected:
        IUndoControlOperation() {}
        void AddStoredControl();
        void RemoveStoredControl();

        TPath m_path;
        CID m_id;
        AZStd::shared_ptr<CATLControl> m_pStoredControl;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoControlAdd
        : public IUndoControlOperation
    {
    public:
        explicit CUndoControlAdd(CID id);
    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Control Add"; }

        void Undo(bool bUndo) override;
        void Redo() override;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoControlRemove
        : public IUndoControlOperation
    {
    public:
        explicit CUndoControlRemove(AZStd::shared_ptr<CATLControl>& pControl);
    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Control Remove"; }

        void Undo(bool bUndo) override;
        void Redo() override;
    };

    //-------------------------------------------------------------------------------------------//
    class IUndoFolderOperation
        : public IUndoObject
    {
    protected:
        explicit IUndoFolderOperation(QStandardItem* pItem);
        void AddFolderItem();
        void RemoveItem();

        TPath m_path;
        QString m_sName;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoFolderRemove
        : public IUndoFolderOperation
    {
    public:
        explicit CUndoFolderRemove(QStandardItem* pItem);
    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Folder Remove"; }

        void Undo(bool bUndo) override;
        void Redo() override;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoFolderAdd
        : public IUndoFolderOperation
    {
    public:
        explicit CUndoFolderAdd(QStandardItem* pItem);
    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Folder Add"; }

        void Undo(bool bUndo) override;
        void Redo() override;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoControlModified
        : public IUndoObject
    {
    public:
        explicit CUndoControlModified(CID id);
    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Control Modify"; }

        void SwapData();
        void Undo(bool bUndo) override;
        void Redo() override;

        CID m_id;
        AZStd::string m_name;
        AZStd::string m_scope;
        bool m_isAutoLoad;
        AZStd::vector<TConnectionPtr> m_connectedControls;
    };

    //-------------------------------------------------------------------------------------------//
    class CUndoItemMove
        : public IUndoObject
    {
    public:
        CUndoItemMove();

    protected:
        int GetSize() override { return sizeof(*this); }
        QString GetDescription() override { return "Undo ATL Control Move"; }

        void Undo(bool bUndo) override;
        void Redo() override;
        void Copy(QStandardItem* pSource, QStandardItem* pDest);

        QATLTreeModel m_original;
        QATLTreeModel m_modified;

        bool bModifiedInitialised;
    };
} //namespace AudioControls