drill_enumerator.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright The 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, see <https://www.gnu.org/licenses/>.
 */
 
#include <drill/drill_enumerator.h>
 
#include <board.h>
#include <board_design_settings.h>
#include <footprint.h>
#include <pad.h>
#include <pcb_track.h>
 
#include <set>
#include <utility>
 
 
namespace
{
 
DRILL_POST_MACHINING fromPadstack( const PADSTACK::POST_MACHINING_PROPS& aProps )
{
    DRILL_POST_MACHINING pm;
 
    pm.m_Mode = aProps.mode.value_or( PAD_DRILL_POST_MACHINING_MODE::UNKNOWN );
    pm.m_Size = aProps.size;
    pm.m_Depth = aProps.depth;
    pm.m_Angle = aProps.angle;
 
    return pm;
}
 
 
bool isBackdrillSet( const PADSTACK::DRILL_PROPS& aDrill )
{
    return aDrill.start != UNDEFINED_LAYER && aDrill.end != UNDEFINED_LAYER
           && ( aDrill.size.x > 0 || aDrill.size.y > 0 );
}
 
 
std::optional<DRILL_OPERATION> backdrillOperation( const PADSTACK::DRILL_PROPS& aDrill, DRILL_OP_KIND aKind,
                                                   const DRILL_SPAN& aSpan, BOARD_ITEM* aItem )
{
    if( !isBackdrillSet( aDrill ) || aDrill.start != aSpan.DrillStartLayer()
        || aDrill.end != aSpan.DrillEndLayer() )
    {
        return std::nullopt;
    }
 
    DRILL_OPERATION op;
    op.m_Kind = aKind;
    op.m_SourceItem = aItem;
    op.m_SourceId = aItem->m_Uuid;
    op.m_NotPlated = true;
    op.m_TopLayer = aSpan.TopLayer();
    op.m_BottomLayer = aSpan.BottomLayer();
    op.m_DrillStart = aDrill.start;
    op.m_DrillEnd = aDrill.end;
    op.m_SizeXY = aDrill.size;
    op.m_Diameter = aDrill.size.x;
 
    if( aDrill.size.y > 0 )
        op.m_Diameter = op.m_Diameter > 0 ? std::min( op.m_Diameter, aDrill.size.y ) : aDrill.size.y;
 
    op.m_IsSlot = aDrill.shape != PAD_DRILL_SHAPE::CIRCLE && aDrill.size.x != aDrill.size.y;
    return op;
}
 
 
void enumerateVia( PCB_VIA* aVia, const BOARD& aBoard, const DRILL_QUERY& aQuery,
                   std::vector<DRILL_OPERATION>& aOut )
{
    const DRILL_SPAN&            span = aQuery.m_Span;
    const BOARD_DESIGN_SETTINGS& bds = aBoard.GetDesignSettings();
 
    auto boardSide =
            [&]( PCB_LAYER_ID aLayer, bool aFront, bool aBack )
            {
                return aLayer == F_Cu ? aFront : aLayer == B_Cu ? aBack : false;
            };
 
    auto covered =
            [&]( PCB_LAYER_ID aLayer )
            {
                bool fallback = boardSide( aLayer, bds.m_CoverViasFront, bds.m_CoverViasBack );
                return aVia->Padstack().IsCovered( aLayer ).value_or( fallback );
            };
 
    auto plugged =
            [&]( PCB_LAYER_ID aLayer )
            {
                bool fallback = boardSide( aLayer, bds.m_PlugViasFront, bds.m_PlugViasBack );
                return aVia->Padstack().IsPlugged( aLayer ).value_or( fallback );
            };
 
    auto tented =
            [&]( PCB_LAYER_ID aLayer )
            {
                bool fallback = boardSide( aLayer, bds.m_TentViasFront, bds.m_TentViasBack );
                return aVia->Padstack().IsTented( aLayer ).value_or( fallback );
            };
 
    auto setProtection =
            [&]( DRILL_OPERATION& aOp, std::optional<bool> aFilled, std::optional<bool> aCapped )
            {
                aOp.m_Filled = aFilled.value_or( bds.m_FillVias );
                aOp.m_Capped = aCapped.value_or( bds.m_CapVias );
                aOp.m_TopCovered = covered( aOp.m_TopLayer );
                aOp.m_BottomCovered = covered( aOp.m_BottomLayer );
                aOp.m_TopPlugged = plugged( aOp.m_TopLayer );
                aOp.m_BottomPlugged = plugged( aOp.m_BottomLayer );
                aOp.m_TopTented = tented( aOp.m_TopLayer );
                aOp.m_BottomTented = tented( aOp.m_BottomLayer );
            };
 
    auto stubLength =
            [&]( PCB_LAYER_ID aStart, PCB_LAYER_ID aEnd ) -> std::optional<int>
            {
                if( aStart == UNDEFINED_LAYER || aEnd == UNDEFINED_LAYER )
                    return std::optional<int>();
 
                const BOARD_STACKUP& stackup = aBoard.GetDesignSettings().GetStackupDescriptor();
 
                return std::optional<int>( stackup.GetDrillDepth( aStart, aEnd ) );
            };
 
    if( span.m_IsBackdrill )
    {
        auto tryBackdrill =
                [&]( const PADSTACK::DRILL_PROPS& aDrill, DRILL_OP_KIND aKind )
                {
                    std::optional<DRILL_OPERATION> op = backdrillOperation( aDrill, aKind, span, aVia );
 
                    if( !op )
                        return;
 
                    op->m_Attribute = HOLE_ATTRIBUTE::HOLE_VIA_BACKDRILL;
                    op->m_Orientation = ANGLE_0;
                    op->m_Position = aVia->GetStart();
                    op->m_StubLength = stubLength( aDrill.start, aDrill.end );
                    // Backdrills remove the via barrel, so via surface protection is not inherited
                    op->m_Filled = aDrill.is_filled.value_or( false );
                    op->m_Capped = aDrill.is_capped.value_or( false );
                    aOut.push_back( std::move( *op ) );
                };
 
        tryBackdrill( aVia->Padstack().SecondaryDrill(), DRILL_OP_KIND::SECONDARY_DRILL );
        tryBackdrill( aVia->Padstack().TertiaryDrill(), DRILL_OP_KIND::TERTIARY_DRILL );
        return;
    }
 
    int holeSize = aVia->GetDrillValue();
 
    if( holeSize == 0 )
        return;
 
    PCB_LAYER_ID topLayer;
    PCB_LAYER_ID bottomLayer;
    aVia->LayerPair( &topLayer, &bottomLayer );
 
    if( DRILL_LAYER_PAIR( topLayer, bottomLayer ) != span.Pair()
        && DRILL_LAYER_PAIR( bottomLayer, topLayer ) != span.Pair() )
    {
        return;
    }
 
    DRILL_OPERATION op;
    op.m_SourceItem = aVia;
    op.m_SourceId = aVia->m_Uuid;
 
    if( span.Pair() == DRILL_LAYER_PAIR( F_Cu, B_Cu ) )
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_VIA_THROUGH;
    else
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_VIA_BURIED;
 
    op.m_Orientation = ANGLE_0;
    op.m_Diameter = holeSize;
    op.m_NotPlated = false;
    op.m_SizeXY = VECTOR2I( holeSize, holeSize );
    op.m_Position = aVia->GetStart();
    op.m_TopLayer = topLayer;
    op.m_BottomLayer = bottomLayer;
    setProtection( op, aVia->Padstack().IsFilled(), aVia->Padstack().IsCapped() );
    op.m_FrontPostMachining = fromPadstack( aVia->Padstack().FrontPostMachining() );
    op.m_BackPostMachining = fromPadstack( aVia->Padstack().BackPostMachining() );
    op.m_DrillStart = aVia->Padstack().Drill().start;
    op.m_DrillEnd = bottomLayer;
 
    aOut.push_back( op );
}
 
 
void enumeratePad( PAD* aPad, const DRILL_QUERY& aQuery, std::vector<DRILL_OPERATION>& aOut )
{
    if( aQuery.m_Span.m_IsBackdrill )
    {
        if( !aQuery.m_PadMachining )
            return;
 
        auto addBackdrill = [&]( const PADSTACK::DRILL_PROPS& aDrill, DRILL_OP_KIND aKind )
        {
            std::optional<DRILL_OPERATION> op = backdrillOperation( aDrill, aKind, aQuery.m_Span, aPad );
 
            if( !op )
                return;
 
            op->m_Attribute = HOLE_ATTRIBUTE::HOLE_PAD_BACKDRILL;
            op->m_Position = aPad->GetPosition();
            op->m_Orientation = aPad->GetOrientation();
            aOut.push_back( std::move( *op ) );
        };
 
        addBackdrill( aPad->Padstack().SecondaryDrill(), DRILL_OP_KIND::SECONDARY_DRILL );
        addBackdrill( aPad->Padstack().TertiaryDrill(), DRILL_OP_KIND::TERTIARY_DRILL );
        return;
    }
 
    if( !aQuery.m_MergePTHNPTH )
    {
        if( !aQuery.m_NonPlatedOnly && aPad->GetAttribute() == PAD_ATTRIB::NPTH )
            return;
 
        if( aQuery.m_NonPlatedOnly && aPad->GetAttribute() != PAD_ATTRIB::NPTH )
            return;
    }
 
    if( aPad->GetDrillSize().x == 0 )
        return;
 
    DRILL_OPERATION op;
    op.m_SourceItem = aPad;
    op.m_SourceId = aPad->m_Uuid;
    op.m_NotPlated = ( aPad->GetAttribute() == PAD_ATTRIB::NPTH );
 
    if( op.m_NotPlated )
    {
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_MECHANICAL;
    }
    else if( aPad->GetProperty() == PAD_PROP::CASTELLATED )
    {
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_PAD_CASTELLATED;
    }
    else if( aPad->GetProperty() == PAD_PROP::PRESSFIT )
    {
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_PAD_PRESSFIT;
    }
    else
    {
        op.m_Attribute = HOLE_ATTRIBUTE::HOLE_PAD;
    }
 
    op.m_Orientation = aPad->GetOrientation();
    op.m_Diameter = std::min( aPad->GetDrillSize().x, aPad->GetDrillSize().y );
 
    op.m_IsSlot = IsDrillSlot( *aPad );
 
    op.m_SizeXY = aPad->GetDrillSize();
    op.m_Position = aPad->GetPosition();
    op.m_BottomLayer = B_Cu;
    op.m_TopLayer = F_Cu;
 
    if( aQuery.m_PadMachining )
    {
        op.m_FrontPostMachining = fromPadstack( aPad->Padstack().FrontPostMachining() );
        op.m_BackPostMachining = fromPadstack( aPad->Padstack().BackPostMachining() );
    }
 
    aOut.push_back( op );
}
 
} // namespace
 
 
bool IsDrillSlot( const PAD& aPad )
{
    return aPad.HasHole()
            && aPad.GetDrillShape() != PAD_DRILL_SHAPE::CIRCLE
            && aPad.GetDrillSize().x != aPad.GetDrillSize().y;
}
 
 
std::vector<DRILL_OPERATION> EnumerateDrillOperations( const BOARD& aBoard, const DRILL_QUERY& aQuery )
{
    std::vector<DRILL_OPERATION> operations;
 
    wxASSERT( IsCopperLayerLowerThan( aQuery.m_Span.BottomLayer(), aQuery.m_Span.TopLayer() )
                || aQuery.m_Span.BottomLayer() == aQuery.m_Span.TopLayer() );
 
    if( !aQuery.m_NonPlatedOnly )
    {
        for( PCB_TRACK* track : aBoard.Tracks() )
        {
            if( track->Type() != PCB_VIA_T )
                continue;
 
            enumerateVia( static_cast<PCB_VIA*>( track ), aBoard, aQuery, operations );
        }
    }
 
    if( ( aQuery.m_Span.m_IsBackdrill && aQuery.m_PadMachining )
        || ( !aQuery.m_Span.m_IsBackdrill && aQuery.m_Span.Pair() == DRILL_LAYER_PAIR( F_Cu, B_Cu ) ) )
    {
        for( FOOTPRINT* footprint : aBoard.Footprints() )
        {
            for( PAD* pad : footprint->Pads() )
                enumeratePad( pad, aQuery, operations );
        }
    }
 
    return operations;
}
 
 
std::vector<DRILL_SPAN> EnumerateDrillSpans( const BOARD& aBoard, bool aPadBackdrills )
{
    std::set<DRILL_SPAN> unique;
 
    auto addBackdrill =
            [&]( const PADSTACK::DRILL_PROPS& aDrill )
            {
                if( isBackdrillSet( aDrill ) )
                    unique.emplace( aDrill.start, aDrill.end, true, false );
            };
 
    for( PCB_TRACK* track : aBoard.Tracks() )
    {
        if( track->Type() != PCB_VIA_T )
            continue;
 
        PCB_VIA* via = static_cast<PCB_VIA*>( track );
        PCB_LAYER_ID top;
        PCB_LAYER_ID bottom;
        via->LayerPair( &top, &bottom );
 
        if( DRILL_LAYER_PAIR( top, bottom ) != DRILL_LAYER_PAIR( F_Cu, B_Cu ) )
            unique.emplace( top, bottom, false, false );
 
        addBackdrill( via->Padstack().SecondaryDrill() );
        addBackdrill( via->Padstack().TertiaryDrill() );
    }
 
    if( aPadBackdrills )
    {
        for( FOOTPRINT* footprint : aBoard.Footprints() )
        {
            for( PAD* pad : footprint->Pads() )
            {
                addBackdrill( pad->Padstack().SecondaryDrill() );
                addBackdrill( pad->Padstack().TertiaryDrill() );
            }
        }
    }
 
    std::vector<DRILL_SPAN> spans;
    spans.emplace_back( F_Cu, B_Cu, false, false );
 
    for( const DRILL_SPAN& span : unique )
    {
        if( span.m_IsBackdrill || span.Pair() != DRILL_LAYER_PAIR( F_Cu, B_Cu ) )
            spans.push_back( span );
    }
 
    return spans;
}