343 lines
12 KiB
C++
343 lines
12 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2017-2019 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
|
|
*/
|
|
|
|
#include <tools/zone_create_helper.h>
|
|
#include <tool/tool_manager.h>
|
|
#include <class_zone.h>
|
|
#include <class_drawsegment.h>
|
|
#include <class_edge_mod.h>
|
|
#include <board_commit.h>
|
|
#include <pcb_painter.h>
|
|
#include <tools/pcb_actions.h>
|
|
#include <tools/selection_tool.h>
|
|
#include <zone_filler.h>
|
|
|
|
ZONE_CREATE_HELPER::ZONE_CREATE_HELPER( DRAWING_TOOL& aTool, PARAMS& aParams ):
|
|
m_tool( aTool ),
|
|
m_params( aParams ),
|
|
m_parentView( *aTool.getView() )
|
|
{
|
|
m_parentView.Add( &m_previewItem );
|
|
}
|
|
|
|
|
|
ZONE_CREATE_HELPER::~ZONE_CREATE_HELPER()
|
|
{
|
|
// remove the preview from the view
|
|
m_parentView.SetVisible( &m_previewItem, false );
|
|
m_parentView.Remove( &m_previewItem );
|
|
}
|
|
|
|
|
|
std::unique_ptr<ZONE_CONTAINER> ZONE_CREATE_HELPER::createNewZone( bool aKeepout )
|
|
{
|
|
PCB_BASE_EDIT_FRAME* frame = m_tool.getEditFrame<PCB_BASE_EDIT_FRAME>();
|
|
BOARD* board = frame->GetBoard();
|
|
BOARD_ITEM_CONTAINER* parent = m_tool.m_frame->GetModel();
|
|
KIGFX::VIEW_CONTROLS* controls = m_tool.GetManager()->GetViewControls();
|
|
std::set<int> highlightedNets = board->GetHighLightNetCodes();
|
|
|
|
// Get the current default settings for zones
|
|
ZONE_SETTINGS zoneInfo = frame->GetZoneSettings();
|
|
zoneInfo.m_Layers.reset().set( m_params.m_layer ); // TODO(JE) multilayer defaults?
|
|
zoneInfo.m_NetcodeSelection = highlightedNets.empty() ? -1 : *highlightedNets.begin();
|
|
zoneInfo.SetIsKeepout( m_params.m_keepout );
|
|
zoneInfo.m_Zone_45_Only = ( m_params.m_leaderMode == POLYGON_GEOM_MANAGER::LEADER_MODE::DEG45 );
|
|
|
|
// If we don't have a net from highlighing, maybe we can get one from the selection
|
|
SELECTION_TOOL* selectionTool = m_tool.GetManager()->GetTool<SELECTION_TOOL>();
|
|
|
|
if( selectionTool && !selectionTool->GetSelection().Empty()
|
|
&& zoneInfo.m_NetcodeSelection == -1 )
|
|
{
|
|
EDA_ITEM* item = *selectionTool->GetSelection().GetItems().begin();
|
|
|
|
if( BOARD_CONNECTED_ITEM* bci = dynamic_cast<BOARD_CONNECTED_ITEM*>( item ) )
|
|
zoneInfo.m_NetcodeSelection = bci->GetNetCode();
|
|
}
|
|
|
|
if( m_params.m_mode != ZONE_MODE::GRAPHIC_POLYGON )
|
|
{
|
|
// Get the current default settings for zones
|
|
|
|
// Show options dialog
|
|
int dialogResult;
|
|
|
|
if( m_params.m_keepout )
|
|
dialogResult = InvokeKeepoutAreaEditor( frame, &zoneInfo );
|
|
else
|
|
{
|
|
// TODO(JE) combine these dialogs?
|
|
if( ( zoneInfo.m_Layers & LSET::AllCuMask() ).any() )
|
|
dialogResult = InvokeCopperZonesEditor( frame, &zoneInfo );
|
|
else
|
|
dialogResult = InvokeNonCopperZonesEditor( frame, &zoneInfo );
|
|
}
|
|
|
|
if( dialogResult == wxID_CANCEL )
|
|
return nullptr;
|
|
|
|
controls->WarpCursor( controls->GetCursorPosition(), true );
|
|
}
|
|
|
|
// The new zone is a ZONE_CONTAINER if created in the board editor
|
|
// and a MODULE_ZONE_CONTAINER if created in the footprint editor
|
|
wxASSERT( !m_tool.m_editModules || ( parent->Type() == PCB_MODULE_T ) );
|
|
|
|
std::unique_ptr<ZONE_CONTAINER> newZone = m_tool.m_editModules ?
|
|
std::make_unique<MODULE_ZONE_CONTAINER>( parent ) :
|
|
std::make_unique<ZONE_CONTAINER>( parent );
|
|
|
|
// Apply the selected settings
|
|
zoneInfo.ExportSetting( *newZone );
|
|
|
|
return newZone;
|
|
}
|
|
|
|
|
|
std::unique_ptr<ZONE_CONTAINER> ZONE_CREATE_HELPER::createZoneFromExisting(
|
|
const ZONE_CONTAINER& aSrcZone )
|
|
{
|
|
auto& board = *m_tool.getModel<BOARD>();
|
|
|
|
auto newZone = std::make_unique<ZONE_CONTAINER>( &board );
|
|
|
|
ZONE_SETTINGS zoneSettings;
|
|
zoneSettings << aSrcZone;
|
|
|
|
zoneSettings.ExportSetting( *newZone );
|
|
|
|
return newZone;
|
|
}
|
|
|
|
|
|
void ZONE_CREATE_HELPER::performZoneCutout( ZONE_CONTAINER& aZone, ZONE_CONTAINER& aCutout )
|
|
{
|
|
BOARD_COMMIT commit( &m_tool );
|
|
BOARD* board = m_tool.getModel<BOARD>();
|
|
std::vector<ZONE_CONTAINER*> newZones;
|
|
|
|
// Clear the selection before removing the old zone
|
|
auto toolMgr = m_tool.GetManager();
|
|
toolMgr->RunAction( PCB_ACTIONS::selectionClear, true );
|
|
|
|
SHAPE_POLY_SET originalOutline( *aZone.Outline() );
|
|
originalOutline.BooleanSubtract( *aCutout.Outline(), SHAPE_POLY_SET::PM_FAST );
|
|
|
|
// After substracting the hole, originalOutline can have more than one
|
|
// main outline.
|
|
// But a zone can have only one main outline, so create as many zones as
|
|
// originalOutline contains main outlines:
|
|
for( int outline = 0; outline < originalOutline.OutlineCount(); outline++ )
|
|
{
|
|
auto newZoneOutline = new SHAPE_POLY_SET;
|
|
newZoneOutline->AddOutline( originalOutline.Outline( outline ) );
|
|
|
|
// Add holes (if any) to thez new zone outline:
|
|
for (int hole = 0; hole < originalOutline.HoleCount( outline ) ; hole++ )
|
|
newZoneOutline->AddHole( originalOutline.CHole( outline, hole ) );
|
|
|
|
auto newZone = new ZONE_CONTAINER( aZone );
|
|
newZone->SetOutline( newZoneOutline );
|
|
newZone->SetLocalFlags( 1 );
|
|
newZone->HatchBorder();
|
|
newZones.push_back( newZone );
|
|
commit.Add( newZone );
|
|
}
|
|
|
|
commit.Remove( &aZone );
|
|
|
|
ZONE_FILLER filler( board, &commit );
|
|
if( !filler.Fill( newZones ) )
|
|
{
|
|
commit.Revert();
|
|
return;
|
|
}
|
|
|
|
commit.Push( _( "Add a zone cutout" ) );
|
|
|
|
// Select the new zone and set it as the source for the next cutout
|
|
toolMgr->RunAction( PCB_ACTIONS::selectItem, true, newZones[0] );
|
|
m_params.m_sourceZone = newZones[0];
|
|
|
|
}
|
|
|
|
|
|
void ZONE_CREATE_HELPER::commitZone( std::unique_ptr<ZONE_CONTAINER> aZone )
|
|
{
|
|
switch ( m_params.m_mode )
|
|
{
|
|
case ZONE_MODE::CUTOUT:
|
|
// For cutouts, subtract from the source
|
|
performZoneCutout( *m_params.m_sourceZone, *aZone );
|
|
break;
|
|
|
|
case ZONE_MODE::ADD:
|
|
case ZONE_MODE::SIMILAR:
|
|
{
|
|
BOARD_COMMIT bCommit( &m_tool );
|
|
|
|
aZone->HatchBorder();
|
|
bCommit.Add( aZone.get() );
|
|
|
|
if( !m_params.m_keepout )
|
|
{
|
|
ZONE_FILLER filler( m_tool.getModel<BOARD>(), &bCommit );
|
|
std::vector<ZONE_CONTAINER*> toFill = { aZone.get() };
|
|
|
|
if( !filler.Fill( toFill ) )
|
|
{
|
|
bCommit.Revert();
|
|
break;
|
|
}
|
|
}
|
|
|
|
bCommit.Push( _( "Add a zone" ) );
|
|
m_tool.GetManager()->RunAction( PCB_ACTIONS::selectItem, true, aZone.release() );
|
|
break;
|
|
}
|
|
|
|
case ZONE_MODE::GRAPHIC_POLYGON:
|
|
{
|
|
BOARD_COMMIT bCommit( &m_tool );
|
|
BOARD_ITEM_CONTAINER* parent = m_tool.m_frame->GetModel();
|
|
LSET graphicPolygonsLayers = LSET::AllLayersMask();
|
|
|
|
graphicPolygonsLayers.reset( Edge_Cuts ).reset( F_CrtYd ).reset( B_CrtYd );
|
|
|
|
if( graphicPolygonsLayers.Contains( m_params.m_layer ) )
|
|
{
|
|
auto poly = m_tool.m_editModules ? new EDGE_MODULE( (MODULE *) parent )
|
|
: new DRAWSEGMENT();
|
|
poly->SetShape ( S_POLYGON );
|
|
poly->SetLayer( m_params.m_layer );
|
|
poly->SetPolyShape ( *aZone->Outline() );
|
|
bCommit.Add( poly );
|
|
m_tool.GetManager()->RunAction( PCB_ACTIONS::selectItem, true, poly );
|
|
}
|
|
else
|
|
{
|
|
auto outline = aZone->Outline();
|
|
|
|
for( auto seg = outline->CIterateSegments( 0 ); seg; seg++ )
|
|
{
|
|
auto new_seg = m_tool.m_editModules ? new EDGE_MODULE( (MODULE *) parent )
|
|
: new DRAWSEGMENT();
|
|
new_seg->SetShape( S_SEGMENT );
|
|
new_seg->SetLayer( m_params.m_layer );
|
|
new_seg->SetStart( wxPoint( seg.Get().A.x, seg.Get().A.y ) );
|
|
new_seg->SetEnd( wxPoint( seg.Get().B.x, seg.Get().B.y ) );
|
|
bCommit.Add( new_seg );
|
|
}
|
|
}
|
|
|
|
bCommit.Push( _( "Add a graphical polygon" ) );
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
bool ZONE_CREATE_HELPER::OnFirstPoint( POLYGON_GEOM_MANAGER& aMgr )
|
|
{
|
|
// if we don't have a zone, create one
|
|
// the user's choice here can affect things like the colour of the preview
|
|
if( !m_zone )
|
|
{
|
|
if( m_params.m_sourceZone )
|
|
m_zone = createZoneFromExisting( *m_params.m_sourceZone );
|
|
else
|
|
m_zone = createNewZone( m_params.m_keepout );
|
|
|
|
if( m_zone )
|
|
{
|
|
m_tool.GetManager()->RunAction( PCB_ACTIONS::selectionClear, true );
|
|
|
|
// set up poperties from zone
|
|
const auto& settings = *m_parentView.GetPainter()->GetSettings();
|
|
COLOR4D color = settings.GetColor( nullptr, m_zone->GetLayer() );
|
|
|
|
m_previewItem.SetStrokeColor( COLOR4D::WHITE );
|
|
m_previewItem.SetFillColor( color.WithAlpha( 0.2 ) );
|
|
|
|
m_parentView.SetVisible( &m_previewItem, true );
|
|
|
|
aMgr.SetLeaderMode( m_zone->GetHV45() ? POLYGON_GEOM_MANAGER::LEADER_MODE::DEG45
|
|
: POLYGON_GEOM_MANAGER::LEADER_MODE::DIRECT );
|
|
}
|
|
}
|
|
|
|
return m_zone != nullptr;
|
|
}
|
|
|
|
|
|
void ZONE_CREATE_HELPER::OnGeometryChange( const POLYGON_GEOM_MANAGER& aMgr )
|
|
{
|
|
// send the points to the preview item
|
|
m_previewItem.SetPoints( aMgr.GetLockedInPoints(), aMgr.GetLeaderLinePoints() );
|
|
m_parentView.Update( &m_previewItem, KIGFX::GEOMETRY );
|
|
}
|
|
|
|
|
|
void ZONE_CREATE_HELPER::OnComplete( const POLYGON_GEOM_MANAGER& aMgr )
|
|
{
|
|
auto& finalPoints = aMgr.GetLockedInPoints();
|
|
|
|
if( finalPoints.PointCount() < 3 )
|
|
{
|
|
// just scrap the zone in progress
|
|
m_zone = nullptr;
|
|
}
|
|
else
|
|
{
|
|
// if m_params.m_mode == DRAWING_TOOL::ZONE_MODE::CUTOUT, m_zone
|
|
// will be merged to the existing zone as a new hole.
|
|
m_zone->Outline()->NewOutline();
|
|
auto* outline = m_zone->Outline();
|
|
|
|
for( int i = 0; i < finalPoints.PointCount(); ++i )
|
|
outline->Append( finalPoints.CPoint( i ) );
|
|
|
|
// In DEG45 mode, we may have intermediate points in the leader that should be
|
|
// included as they are shown in the preview. These typically maintain the
|
|
// 45 constraint
|
|
if( aMgr.GetLeaderMode() == POLYGON_GEOM_MANAGER::LEADER_MODE::DEG45 )
|
|
{
|
|
const auto& pts = aMgr.GetLeaderLinePoints();
|
|
for( int i = 1; i < pts.PointCount(); i++ )
|
|
outline->Append( pts.CPoint( i ) );
|
|
}
|
|
|
|
outline->Outline( 0 ).SetClosed( true );
|
|
outline->RemoveNullSegments();
|
|
outline->Simplify( SHAPE_POLY_SET::PM_FAST );
|
|
|
|
// hand the zone over to the committer
|
|
commitZone( std::move( m_zone ) );
|
|
m_zone = nullptr;
|
|
}
|
|
|
|
m_parentView.SetVisible( &m_previewItem, false );
|
|
}
|