item_modification_routine.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2023 KiCad Developers, see AUTHORS.txt for contributors.
 *
 * 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, you may find one here:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * or you may search the http://www.gnu.org website for the version 2 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */
 
#ifndef ITEM_MODIFICATION_ROUTINE_H_
#define ITEM_MODIFICATION_ROUTINE_H_
 
#include <functional>
#include <memory>
#include <optional>
#include <vector>
 
#include <board_item.h>
#include <pcb_shape.h>
 
#include <geometry/corner_operations.h>
 
class ITEM_MODIFICATION_ROUTINE
{
public:
    /*
    * Handlers for receiving changes from the tool
    *
    * These are used to allow the tool's caller to make changes to
    * affected board items using extra information that the tool
    * does not have access to (e.g. is this an FP editor, was
    * the line created from a rectangle and needs to be added, not
    * modified, etc).
    *
    * We can't store them up until the end, because modifications
    * need the old state to be known, so this allows the caller to
    * inject the dependencies for how to handle the changes.
    */
    class CHANGE_HANDLER
    {
    public:
        virtual ~CHANGE_HANDLER() = default;
 
        virtual void AddNewItem( std::unique_ptr<BOARD_ITEM> aItem ) = 0;
 
        virtual void MarkItemModified( BOARD_ITEM& aItem ) = 0;
 
        virtual void DeleteItem( BOARD_ITEM& aItem ) = 0;
    };
 
    class CALLABLE_BASED_HANDLER : public CHANGE_HANDLER
    {
    public:
        using CREATION_HANDLER = std::function<void( std::unique_ptr<BOARD_ITEM> )>;
 
        using MODIFICATION_HANDLER = std::function<void( BOARD_ITEM& )>;
 
        using DELETION_HANDLER = std::function<void( BOARD_ITEM& )>;
 
        CALLABLE_BASED_HANDLER( CREATION_HANDLER     aCreationHandler,
                                MODIFICATION_HANDLER aModificationHandler,
                                DELETION_HANDLER     aDeletionHandler ) :
                m_creationHandler( std::move( aCreationHandler ) ),
                m_modificationHandler( std::move( aModificationHandler ) ),
                m_deletionHandler( std::move( aDeletionHandler ) )
        {
        }
 
        void AddNewItem( std::unique_ptr<BOARD_ITEM> aItem ) override
        {
            m_creationHandler( std::move( aItem ) );
        }
 
        void MarkItemModified( BOARD_ITEM& aItem ) override { m_modificationHandler( aItem ); }
 
        void DeleteItem( BOARD_ITEM& aItem ) override { m_deletionHandler( aItem ); }
 
        CREATION_HANDLER     m_creationHandler;
        MODIFICATION_HANDLER m_modificationHandler;
        DELETION_HANDLER     m_deletionHandler;
    };
 
    ITEM_MODIFICATION_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            m_board( aBoard ),
            m_handler( aHandler ),
            m_numSuccesses( 0 ),
            m_numFailures( 0 )
    {
    }
 
    virtual ~ITEM_MODIFICATION_ROUTINE() = default;
 
    unsigned GetSuccesses() const { return m_numSuccesses; }
 
    unsigned GetFailures() const { return m_numFailures; }
 
    virtual wxString GetCommitDescription() const = 0;
 
    virtual std::optional<wxString> GetStatusMessage() const = 0;
 
protected:
    BOARD_ITEM* GetBoard() const { return m_board; }
 
    void AddSuccess() { ++m_numSuccesses; }
 
    void AddFailure() { ++m_numFailures; }
 
    bool ModifyLineOrDeleteIfZeroLength( PCB_SHAPE& aItem, const std::optional<SEG>& aSeg );
 
    CHANGE_HANDLER& GetHandler() { return m_handler; }
 
private:
    BOARD_ITEM*          m_board;
    CHANGE_HANDLER&      m_handler;
 
    unsigned m_numSuccesses;
    unsigned m_numFailures;
};
 
class PAIRWISE_LINE_ROUTINE : public ITEM_MODIFICATION_ROUTINE
{
public:
    PAIRWISE_LINE_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            ITEM_MODIFICATION_ROUTINE( aBoard, aHandler )
    {
    }
 
    virtual void ProcessLinePair( PCB_SHAPE& aLineA, PCB_SHAPE& aLineB ) = 0;
};
 
class LINE_FILLET_ROUTINE : public PAIRWISE_LINE_ROUTINE
{
public:
    LINE_FILLET_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler, int filletRadiusIU ) :
            PAIRWISE_LINE_ROUTINE( aBoard, aHandler ), m_filletRadiusIU( filletRadiusIU )
    {
    }
 
    wxString GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
    void ProcessLinePair( PCB_SHAPE& aLineA, PCB_SHAPE& aLineB ) override;
 
private:
    int m_filletRadiusIU;
};
 
class LINE_CHAMFER_ROUTINE : public PAIRWISE_LINE_ROUTINE
{
public:
    LINE_CHAMFER_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler,
                          CHAMFER_PARAMS aChamferParams ) :
            PAIRWISE_LINE_ROUTINE( aBoard, aHandler ),
            m_chamferParams( std::move( aChamferParams ) )
    {
    }
 
    wxString GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
    void ProcessLinePair( PCB_SHAPE& aLineA, PCB_SHAPE& aLineB ) override;
 
private:
    const CHAMFER_PARAMS m_chamferParams;
};
 
class LINE_EXTENSION_ROUTINE : public PAIRWISE_LINE_ROUTINE
{
public:
    LINE_EXTENSION_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            PAIRWISE_LINE_ROUTINE( aBoard, aHandler )
    {
    }
 
    wxString GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
    void ProcessLinePair( PCB_SHAPE& aLineA, PCB_SHAPE& aLineB ) override;
};
 
class DOGBONE_CORNER_ROUTINE : public PAIRWISE_LINE_ROUTINE
{
public:
    struct PARAMETERS
    {
        int  DogboneRadiusIU;
        bool AddSlots;
    };
 
    DOGBONE_CORNER_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler, PARAMETERS aParams ) :
            PAIRWISE_LINE_ROUTINE( aBoard, aHandler ), m_params( std::move( aParams ) ),
            m_haveNarrowMouths( false )
    {
    }
 
    wxString                GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
    void ProcessLinePair( PCB_SHAPE& aLineA, PCB_SHAPE& aLineB ) override;
 
private:
    PARAMETERS m_params;
    bool m_haveNarrowMouths;
};
 
 
class POLYGON_BOOLEAN_ROUTINE : public ITEM_MODIFICATION_ROUTINE
{
public:
    POLYGON_BOOLEAN_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            ITEM_MODIFICATION_ROUTINE( aBoard, aHandler )
    {
    }
 
    void ProcessShape( PCB_SHAPE& aPcbShape );
 
    void Finalize();
 
protected:
    SHAPE_POLY_SET& GetWorkingPolygons() { return m_workingPolygons; }
 
    virtual bool ProcessSubsequentPolygon( const SHAPE_POLY_SET& aPolygon ) = 0;
 
private:
    SHAPE_POLY_SET m_workingPolygons;
 
    bool         m_firstPolygon = true;
    int          m_width = 0;
    PCB_LAYER_ID m_layer = PCB_LAYER_ID::UNDEFINED_LAYER;
    bool         m_filled = false;
};
 
class POLYGON_MERGE_ROUTINE : public POLYGON_BOOLEAN_ROUTINE
{
public:
    POLYGON_MERGE_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            POLYGON_BOOLEAN_ROUTINE( aBoard, aHandler )
    {
    }
 
    wxString                GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
private:
    bool ProcessSubsequentPolygon( const SHAPE_POLY_SET& aPolygon ) override;
};
 
 
class POLYGON_SUBTRACT_ROUTINE : public POLYGON_BOOLEAN_ROUTINE
{
public:
    POLYGON_SUBTRACT_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            POLYGON_BOOLEAN_ROUTINE( aBoard, aHandler )
    {
    }
 
    wxString                GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
private:
    bool ProcessSubsequentPolygon( const SHAPE_POLY_SET& aPolygon ) override;
};
 
 
class POLYGON_INTERSECT_ROUTINE : public POLYGON_BOOLEAN_ROUTINE
{
public:
    POLYGON_INTERSECT_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler ) :
            POLYGON_BOOLEAN_ROUTINE( aBoard, aHandler )
    {
    }
 
    wxString                GetCommitDescription() const override;
    std::optional<wxString> GetStatusMessage() const override;
 
private:
    bool ProcessSubsequentPolygon( const SHAPE_POLY_SET& aPolygon ) override;
};
 
 
class OUTSET_ROUTINE : public ITEM_MODIFICATION_ROUTINE
{
public:
    struct PARAMETERS
    {
        int                outsetDistance;
        bool               roundCorners;
        bool               useSourceLayers;
        bool               useSourceWidths;
        PCB_LAYER_ID       layer;
        int                lineWidth;
        std::optional<int> gridRounding;
        bool               deleteSourceItems;
    };
 
    OUTSET_ROUTINE( BOARD_ITEM* aBoard, CHANGE_HANDLER& aHandler, const PARAMETERS& aParams ) :
            ITEM_MODIFICATION_ROUTINE( aBoard, aHandler ), m_params( aParams )
    {
    }
 
    wxString GetCommitDescription() const override;
 
    std::optional<wxString> GetStatusMessage() const override;
 
    void ProcessItem( BOARD_ITEM& aItem );
 
private:
    const PARAMETERS m_params;
};
 
#endif /* ITEM_MODIFICATION_ROUTINE_H_ */