pns_dragger.h 3.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
/*
 * KiRouter - a push-and-(sometimes-)shove PCB router
 *
 * Copyright (C) 2013-2014 CERN
 * Author: Tomasz Wlostowski <tomasz.wlostowski@cern.ch>
 *
 * 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 3 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/>.
 */

#ifndef __PNS_DRAGGER_H
#define __PNS_DRAGGER_H

#include <math/vector2d.h>

#include "pns_node.h"
#include "pns_via.h"
#include "pns_line.h"
#include "pns_algo_base.h"
#include "pns_itemset.h"

class PNS_ROUTER;
class PNS_SHOVE;
class PNS_OPTIMIZER;
class PNS_ROUTER_BASE;

/**
 * Class PNS_DRAGGER
 *
 * Via, segment and corner dragging algorithm.
 */
class PNS_DRAGGER : public PNS_ALGO_BASE
{
public:
Maciej Suminski's avatar
Maciej Suminski committed
45
     PNS_DRAGGER( PNS_ROUTER* aRouter );
46 47 48 49 50 51 52
    ~PNS_DRAGGER();

    /**
     * Function SetWorld()
     *
     * Sets the board to work on.
     */
53
    void SetWorld( PNS_NODE* aWorld );
54 55 56

    /**
     * Function Start()
Maciej Suminski's avatar
Maciej Suminski committed
57
     *
58 59 60
     * Starts routing a single track at point aP, taking item aStartItem as anchor
     * (unless NULL). Returns true if a dragging operation has started.
     */
61
    bool Start( const VECTOR2I& aP, PNS_ITEM* aStartItem );
62 63 64 65 66 67 68

    /**
     * Function Drag()
     *
     * Drags the current segment/corner/via to the point aP.
     * @return true, if dragging finished with success.
     */
69
    bool Drag( const VECTOR2I& aP );
Maciej Suminski's avatar
Maciej Suminski committed
70

71 72 73
    /**
     * Function FixRoute()
     *
Maciej Suminski's avatar
Maciej Suminski committed
74
     * Checks if the result of current dragging operation is correct
75 76 77
     * and eventually commits it to the world.
     * @return true, if dragging finished with success.
     */
78
    bool FixRoute();
Maciej Suminski's avatar
Maciej Suminski committed
79 80

    /**
81 82 83 84 85 86
     * Function CurrentNode()
     *
     * Returns the most recent world state, including all
     * items changed due to dragging operation.
     */
    PNS_NODE* CurrentNode() const;
Maciej Suminski's avatar
Maciej Suminski committed
87

88 89 90 91 92 93 94 95
    /**
     * Function Traces()
     *
     * Returns the set of dragged items.
     */
    const PNS_ITEMSET Traces();

    /// @copydoc PNS_ALGO_BASE::Logger()
96
    virtual PNS_LOGGER* Logger();
Maciej Suminski's avatar
Maciej Suminski committed
97

98
private:
Maciej Suminski's avatar
Maciej Suminski committed
99
    typedef std::pair<PNS_LINE*, PNS_LINE*> LinePair;
100 101
    typedef std::vector<LinePair> LinePairVec;

Maciej Suminski's avatar
Maciej Suminski committed
102 103 104 105 106
    enum DragMode {
        CORNER = 0,
        SEGMENT,
        VIA
    };
107

108 109
    bool dragMarkObstacles( const VECTOR2I& aP );
    bool dragShove(const VECTOR2I& aP );
Maciej Suminski's avatar
Maciej Suminski committed
110 111 112 113 114 115 116 117 118 119 120
    bool startDragSegment( const VECTOR2D& aP, PNS_SEGMENT* aSeg );
    bool startDragVia( const VECTOR2D& aP, PNS_VIA* aVia );
    void dumbDragVia( PNS_VIA* aVia, PNS_NODE* aNode, const VECTOR2I& aP );

    PNS_NODE*   m_world;
    PNS_NODE*   m_lastNode;
    DragMode    m_mode;
    PNS_LINE*   m_draggedLine;
    PNS_VIA*    m_draggedVia;
    PNS_LINE    m_lastValidDraggedLine;
    PNS_SHOVE*  m_shove;
121 122
    int         m_draggedSegmentIndex;
    bool        m_dragStatus;
123
    PNS_MODE    m_currentMode;
124 125 126 127
    std::vector<PNS_LINE>   m_origViaConnections;
    std::vector<PNS_LINE>   m_draggedViaConnections;
    PNS_VIA*                m_initialVia;
    PNS_ITEMSET             m_draggedItems;
128 129 130
};

#endif