teardrop.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2021 Jean-Pierre Charras, jp.charras at wanadoo.fr
 * Copyright (C) 2022 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 TEARDROP_H
#define TEARDROP_H
 
#include <pcb_edit_frame.h>
#include <board.h>
#include <footprint.h>
#include <pad.h>
#include <pcb_track.h>
#include <zone.h>
#include "teardrop_parameters.h"
 
#define MAGIC_TEARDROP_PADVIA_NAME "$teardrop_padvia$"
#define MAGIC_TEARDROP_TRACK_NAME "$teardrop_track$"
 
class TRACK_BUFFER;
 
struct VIAPAD
{
 VIAPAD( PCB_VIA* aVia );
 VIAPAD( PAD* aPad );
 VIAPAD( PCB_TRACK* aTrack, ENDPOINT_T aEndPoint );
 
 bool IsOnLayer( PCB_LAYER_ID aLayer ) const
 {
 return m_Parent->IsOnLayer( aLayer );
 }
 
 VECTOR2I m_Pos;
 int m_Width; // The diameter of a round shape, or the min size for others
 int m_Drill;
 int m_NetCode;
 bool m_IsRound; // true for round shapes, false for any other shape
 bool m_IsPad; // true for pads, false for vias and tracks
 BOARD_CONNECTED_ITEM* m_Parent;
};
 
 
class TEARDROP_MANAGER
{
 friend class TEARDROP_PARAMETERS;
 
public:
 enum TEARDROP_VARIANT
 {
 TD_TYPE_PADVIA, // Specify a teardrop on a pad via
 TD_TYPE_TRACKEND // specify a teardrop on a rond end of a wide track
 };
 
 TEARDROP_MANAGER( BOARD* aBoard, PCB_EDIT_FRAME* aFrame );
 
 int SetTeardrops( BOARD_COMMIT* aCommitter,
 bool aFollowTracks = true );
 
 
 int RemoveTeardrops( BOARD_COMMIT* aCommitter, bool aCommitAfterRemove );
 
 
private:
 void collectVias( std::vector< VIAPAD >& aList ) const;
 
 void collectPadsCandidate( std::vector< VIAPAD >& aList,
 bool aDrilledViaPad,
 bool aRoundShapesOnly,
 bool aIncludeNotDrilled ) const;
 
 void collectTeardrops( std::vector< ZONE* >& aList ) const;
 
 int addTeardropsOnTracks( BOARD_COMMIT* aCommitter );
 
 bool isViaAndTrackInSameZone( VIAPAD& aVia, PCB_TRACK* aTrack) const;
 
 void computeCurvedForRoundShape( TEARDROP_PARAMETERS* aCurrParams,
 std::vector<VECTOR2I>& aPoly,
 int aTrackHalfWidth,
 VECTOR2D aTrackDir, VIAPAD& aViaPad,
 std::vector<VECTOR2I>& aPts ) const;
 
 
 void computeCurvedForRectShape( TEARDROP_PARAMETERS* aCurrParams,
 std::vector<VECTOR2I>& aPoly, int aTdHeight,
 int aTrackHalfWidth, VIAPAD& aViaPad,
 std::vector<VECTOR2I>& aPts ) const;
 
 bool computeTeardropPolygonPoints( TEARDROP_PARAMETERS* aCurrParams,
 std::vector<VECTOR2I>& aCorners,
 PCB_TRACK* aTrack, VIAPAD& aVia,
 bool aFollowTracks,
 TRACK_BUFFER& aTrackLookupList) const;
 bool ComputePointsOnPadVia( TEARDROP_PARAMETERS* aCurrParams,
 PCB_TRACK* aTrack, VIAPAD& aViaPad,
 std::vector<VECTOR2I>& aPts ) const;
 
 PCB_TRACK* findTouchingTrack( EDA_ITEM_FLAGS& aMatchType, PCB_TRACK* aTrackRef,
 const VECTOR2I& aEndPoint,
 TRACK_BUFFER& aTrackLookupList ) const;
 
 ZONE* createTeardrop( TEARDROP_VARIANT aTeardropVariant,
 std::vector<VECTOR2I>& aPoints, PCB_TRACK* aTrack) const;
 
 void setTeardropPriorities();
 
 bool findAnchorPointsOnTrack( TEARDROP_PARAMETERS* aCurrParams,
 VECTOR2I& aStartPoint, VECTOR2I& aEndPoint,
 PCB_TRACK*& aTrack, VIAPAD& aViaPad,
 int* aEffectiveTeardropLen,
 bool aFollowTracks, TRACK_BUFFER& aTrackLookupList ) const;
 
private:
 int m_tolerance; // max distance between a track end point and a pad/via center to
 // see them connected to ut a teardrop
 BOARD* m_board;
 TEARDROP_PARAMETERS_LIST* m_prmsList; // the teardrop parameters list, from the board design settings
 std::vector<ZONE*> m_createdTdList; // list of new created teardrops
};
 
 
// A class to store tracks grouped by layer and netcode
class TRACK_BUFFER
{
public:
 TRACK_BUFFER() {}
 
 void AddTrack( PCB_TRACK* aTrack, int aLayer, int aNetcode );
 
 std::vector<PCB_TRACK*>* GetTrackList( int aLayer, int aNetcode )
 {
 return m_map_tracks.at( idxFromLayNet( aLayer, aNetcode ) );
 }
 
 std::map< int, std::vector<PCB_TRACK*>* >& GetBuffer() { return m_map_tracks ; }
 
 static void GetNetcodeAndLayerFromIndex( int aIdx, int* aLayer, int* aNetcode )
 {
 *aLayer = aIdx & 0xFF;
 *aNetcode = aIdx >> 8;
 }
 
private:
 // Build an index from the layer id and the netcode, to store a track in buffer
 int idxFromLayNet( int aLayer, int aNetcode ) const
 {
 return ( aNetcode << 8 ) + ( aLayer & 0xFF );
 }
 
 // Track buffer, tracks are grouped by layer+netcode
 std::map< int, std::vector<PCB_TRACK*>* > m_map_tracks;
};
 
#endif // ifndef TEARDROP_H