commit.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright 2016-2017 CERN
 * Copyright The KiCad Developers, see AUTHORS.txt for contributors.
 *
 * @author Tomasz Wlostowski <[email protected]>
 * @author Maciej Suminski <[email protected]>
 *
 * 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 <https://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include <set>
#include <vector>
#include <wx/string.h>
#include <undo_redo_container.h>
#include <kiid.h>
#include <eda_item.h>
 
class EDA_GROUP;
class BASE_SCREEN;

enum CHANGE_TYPE {
    CHT_ADD     = 1,
    CHT_REMOVE  = 2,
    CHT_MODIFY  = 4,
    CHT_TYPE    = CHT_ADD | CHT_REMOVE | CHT_MODIFY,
 
    CHT_DONE    = 32,             
    CHT_FLAGS   = CHT_DONE
};
 
template<typename T>
CHANGE_TYPE operator|( CHANGE_TYPE aTypeA, T aTypeB )
{
    return CHANGE_TYPE( (int) aTypeA | (int) aTypeB );
}
 
template<typename T>
CHANGE_TYPE operator&( CHANGE_TYPE aTypeA, T aTypeB )
{
    return CHANGE_TYPE( (int) aTypeA & (int) aTypeB );
}
 

class COMMIT
{
public:
    COMMIT();
    virtual ~COMMIT();

    COMMIT& Add( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr )
    {
        return Stage( aItem, CHT_ADD, aScreen );
    }

    COMMIT& Added( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr )
    {
        return Stage( aItem, CHT_ADD | CHT_DONE, aScreen );
    }

    COMMIT& Remove( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr )
    {
        return Stage( aItem, CHT_REMOVE, aScreen );
    }

    COMMIT& Removed( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr )
    {
        return Stage( aItem, CHT_REMOVE | CHT_DONE, aScreen );
    }

    COMMIT& Modify( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr,
                    RECURSE_MODE aRecurse = RECURSE_MODE::NO_RECURSE )
    {
        return Stage( aItem, CHT_MODIFY, aScreen, aRecurse );
    }

    COMMIT& Modified( EDA_ITEM* aItem, EDA_ITEM* aCopy, BASE_SCREEN *aScreen = nullptr );

    virtual COMMIT& Stage( EDA_ITEM* aItem, CHANGE_TYPE aChangeType, BASE_SCREEN *aScreen = nullptr,
                           RECURSE_MODE aRecurse = RECURSE_MODE::NO_RECURSE );
 
    virtual COMMIT& Stage( std::vector<EDA_ITEM*>& container, CHANGE_TYPE aChangeType,
                           BASE_SCREEN *aScreen = nullptr );
 
    virtual COMMIT& Stage( const PICKED_ITEMS_LIST& aItems, UNDO_REDO aModFlag = UNDO_REDO::UNSPECIFIED,
                           BASE_SCREEN *aScreen = nullptr );
 
    void Unstage( EDA_ITEM* aItem, BASE_SCREEN* aScreen );
    void Unmodify( EDA_ITEM* aItem, BASE_SCREEN* aScreen );

    virtual void Push( const wxString& aMessage = wxT( "A commit" ), int aFlags = 0 ) = 0;

    virtual void Revert() = 0;
 
    bool Empty() const
    {
        return m_entries.empty();
    }

    int GetStatus( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr );
 
    EDA_ITEM* GetFirst() const { return m_entries.empty() ? nullptr : m_entries[0].m_item; }
 
protected:
    struct COMMIT_LINE
    {
        EDA_ITEM*    m_item;                
        EDA_ITEM*    m_copy;                
        CHANGE_TYPE  m_type;                
        BASE_SCREEN* m_screen;
    };

    void clear()
    {
        m_addedItems.clear();
        m_changedItems.clear();
        m_deletedItems.clear();
        m_entries.clear();
    }
 
    virtual void makeEntry( EDA_ITEM* aItem, CHANGE_TYPE aType, EDA_ITEM* aCopy = nullptr,
                            BASE_SCREEN *aScreen = nullptr );

    COMMIT_LINE* findEntry( EDA_ITEM* aItem, BASE_SCREEN *aScreen = nullptr );
 
    virtual EDA_ITEM* undoLevelItem( EDA_ITEM* aItem ) const = 0;
 
    virtual EDA_ITEM* makeImage( EDA_ITEM* aItem ) const = 0;
 
    CHANGE_TYPE convert( UNDO_REDO aType ) const;
    UNDO_REDO convert( CHANGE_TYPE aType ) const;
 
protected:
    std::set<std::pair<EDA_ITEM*, BASE_SCREEN*>> m_addedItems;
    std::set<std::pair<EDA_ITEM*, BASE_SCREEN*>> m_changedItems;
    std::set<std::pair<EDA_ITEM*, BASE_SCREEN*>> m_deletedItems;
    std::vector<COMMIT_LINE>                     m_entries;
};