ratsnest_view_item.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KICAD, a free EDA CAD application.
 *
 * Copyright (C) 2013 CERN
 * Copyright (C) 2018-2019 KiCad Developers, see AUTHORS.txt for contributors.
 *
 * @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, 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 <ratsnest/ratsnest_view_item.h>

#include <gal/graphics_abstraction_layer.h>
#include <pcb_painter.h>
#include <ratsnest/ratsnest_data.h>

#include <layer_ids.h>
#include <pcb_base_frame.h>

#include <memory>
#include <utility>

#include <view/view.h>

RATSNEST_VIEW_ITEM::RATSNEST_VIEW_ITEM( std::shared_ptr<CONNECTIVITY_DATA> aData ) :
 EDA_ITEM( NOT_USED ), m_data( std::move(aData) )
{
}


const BOX2I RATSNEST_VIEW_ITEM::ViewBBox() const
{
 // Make it always visible
 BOX2I bbox;
 bbox.SetMaximum();

 return bbox;
}


void RATSNEST_VIEW_ITEM::ViewDraw( int aLayer, KIGFX::VIEW* aView ) const
{
 std::unique_lock<KISPINLOCK> lock( m_data->GetLock(), std::try_to_lock );

 if( !lock )
 return;

 constexpr int CROSS_SIZE = 200000;

 auto gal = aView->GetGAL();
 gal->SetIsStroke( true );
 gal->SetIsFill( false );
 gal->SetLineWidth( 1.0 );
 auto rs = static_cast<KIGFX::PCB_RENDER_SETTINGS*>( aView->GetPainter()->GetSettings() );

 COLOR4D defaultColor = rs->GetColor( nullptr, LAYER_RATSNEST );
 COLOR4D color = defaultColor;
 const bool colorByNet = rs->GetNetColorMode() != NET_COLOR_MODE::OFF;

 std::set<int> highlightedNets = rs->GetHighlightNetCodes();
 const std::set<int>& hiddenNets = rs->GetHiddenNets();

 std::map<int, KIGFX::COLOR4D>& netColors = rs->GetNetColorMap();
 std::map<wxString, KIGFX::COLOR4D>& netclassColors = rs->GetNetclassColorMap();
 const std::map<int, wxString>& netclassMap = m_data->GetNetclassMap();

 const bool onlyVisibleLayers = rs->GetRatsnestDisplayMode() == RATSNEST_MODE::VISIBLE;

 LSET visibleLayers;

 // If we are in "other layers off" mode, the active layer is the only visible layer
 if( rs->GetContrastModeDisplay() == HIGH_CONTRAST_MODE::HIDDEN )
 {
 visibleLayers.set( rs->GetPrimaryHighContrastLayer() );
 }
 else
 {
 for( PCB_LAYER_ID layer : LSET::AllCuMask().Seq() )
 if( aView->IsLayerVisible( layer ) )
 visibleLayers.set( layer );
 }

 const bool curved_ratsnest = rs->GetCurvedRatsnestLinesEnabled();

 // Draw the "dynamic" ratsnest (i.e. for objects that may be currently being moved)
 for( const RN_DYNAMIC_LINE& l : m_data->GetDynamicRatsnest() )
 {
 if( hiddenNets.count( l.netCode ) )
 continue;

 if( colorByNet && netColors.count( l.netCode ) )
 color = netColors.at( l.netCode );
 else if( colorByNet && netclassMap.count( l.netCode )
 && netclassColors.count( netclassMap.at( l.netCode ) ) )
 color = netclassColors.at( netclassMap.at( l.netCode ) );
 else
 color = defaultColor;

 if( color == COLOR4D::UNSPECIFIED )
 color = defaultColor;

 gal->SetStrokeColor( color.Brightened( 0.5 ) );

 if( l.a == l.b )
 {
 gal->DrawLine( VECTOR2I( l.a.x - CROSS_SIZE, l.a.y - CROSS_SIZE ),
 VECTOR2I( l.b.x + CROSS_SIZE, l.b.y + CROSS_SIZE ) );
 gal->DrawLine( VECTOR2I( l.a.x - CROSS_SIZE, l.a.y + CROSS_SIZE ),
 VECTOR2I( l.b.x + CROSS_SIZE, l.b.y - CROSS_SIZE ) );
 }
 else
 {
 if( curved_ratsnest )
 {
 auto dx = l.b.x - l.a.x;
 auto dy = l.b.y - l.a.y;
 const auto center = VECTOR2I( l.a.x + 0.5 * dx - 0.1 * dy,
 l.a.y + 0.5 * dy + 0.1 * dx );
 gal->DrawCurve( l.a, center, center, l.b );
 }
 else
 {
 gal->DrawLine( l.a, l.b );
 }
 }
 }

 for( int i = 1 /* skip "No Net" at [0] */; i < m_data->GetNetCount(); ++i )
 {
 if( hiddenNets.count( i ) )
 continue;

 RN_NET* net = m_data->GetRatsnestForNet( i );

 if( !net )
 continue;

 if( colorByNet && netColors.count( i ) )
 color = netColors.at( i );
 else if( colorByNet && netclassMap.count( i )
 && netclassColors.count( netclassMap.at( i ) ) )
 color = netclassColors.at( netclassMap.at( i ) );
 else
 color = defaultColor;

 if( color == COLOR4D::UNSPECIFIED )
 color = defaultColor;

 // Draw the "static" ratsnest
 if( highlightedNets.count( i ) )
 gal->SetStrokeColor( color.Brightened( 0.8 ) );
 else
 gal->SetStrokeColor( color ); // using the default ratsnest color for not highlighted

 for( const CN_EDGE& edge : net->GetUnconnected() )
 {
 //if ( !edge.IsVisible() )
 // continue;

 const CN_ANCHOR_PTR& sourceNode = edge.GetSourceNode();
 const CN_ANCHOR_PTR& targetNode = edge.GetTargetNode();
 const VECTOR2I source( sourceNode->Pos() );
 const VECTOR2I target( targetNode->Pos() );

 if( !sourceNode->Valid() || !targetNode->Valid() )
 continue;

 bool enable = !sourceNode->GetNoLine() && !targetNode->GetNoLine();
 bool show;

 // If the global ratsnest is currently enabled, the local ratsnest
 // should be easy to turn off, so either element can disable it
 // If the global ratsnest is disabled, the local ratsnest should be easy to turn on
 // so either element can enable it.
 if( rs->GetGlobalRatsnestLinesEnabled() )
 show = sourceNode->Parent()->GetLocalRatsnestVisible() &&
 targetNode->Parent()->GetLocalRatsnestVisible();
 else
 show = sourceNode->Parent()->GetLocalRatsnestVisible() ||
 targetNode->Parent()->GetLocalRatsnestVisible();

 if( onlyVisibleLayers && show )
 {
 LSET sourceLayers = sourceNode->Parent()->GetLayerSet();
 LSET targetLayers = targetNode->Parent()->GetLayerSet();

 if( !( sourceLayers & visibleLayers ).any() ||
 !( targetLayers & visibleLayers ).any() )
 show = false;
 }

 if ( enable && show )
 {
 if ( source == target )
 {
 gal->DrawLine( VECTOR2I( source.x - CROSS_SIZE, source.y - CROSS_SIZE ),
 VECTOR2I( source.x + CROSS_SIZE, source.y + CROSS_SIZE ) );
 gal->DrawLine( VECTOR2I( source.x - CROSS_SIZE, source.y + CROSS_SIZE ),
 VECTOR2I( source.x + CROSS_SIZE, source.y - CROSS_SIZE ) );
 }
 else
 {
 if( curved_ratsnest )
 {
 auto dx = target.x - source.x;
 auto dy = target.y - source.y;
 const auto center = VECTOR2I( source.x + 0.5 * dx - 0.1 * dy,
 source.y + 0.5 * dy + 0.1 * dx );
 gal->DrawCurve( source, center, center, target );
 }
 else
 {
 gal->DrawLine( source, target );
 }
 }
 }
 }
 }
}


void RATSNEST_VIEW_ITEM::ViewGetLayers( int aLayers[], int& aCount ) const
{
 aCount = 1;
 aLayers[0] = LAYER_RATSNEST;
}