kicad/pcbnew/router/pns_itemset.h

228 lines
5.0 KiB
C
Raw Normal View History

/*
* KiRouter - a push-and-(sometimes-)shove PCB router
*
* Copyright (C) 2013-2014 CERN
* Author: Tomasz Wlostowski <tomasz.wlostowski@cern.ch>
2013-09-26 21:53:54 +00:00
*
* 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.
2013-09-26 21:53:54 +00:00
*
* 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.
2013-09-26 21:53:54 +00:00
*
* 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_ITEMSET_H
#define __PNS_ITEMSET_H
#include <vector>
#include <boost/foreach.hpp>
#include "pns_item.h"
/**
* Class PNS_ITEMSET
*
* Holds a list of board items, that can be filtered against net, kinds,
2013-09-26 21:53:54 +00:00
* layers, etc.
**/
class PNS_LINE;
class PNS_ITEMSET
{
public:
struct ENTRY {
ENTRY( PNS_ITEM* aItem, bool aOwned = false ) :
item( aItem ),
owned( aOwned )
{}
ENTRY( const ENTRY& aOther )
{
owned = aOther.owned;
if( aOther.owned )
item = aOther.item->Clone();
else
item = aOther.item;
}
~ENTRY()
{
if( owned )
delete item;
}
bool operator== ( const ENTRY& b ) const
{
return item == b.item;
}
bool operator< ( const ENTRY& b ) const
{
return item < b.item;
}
ENTRY& operator= ( const ENTRY& aOther )
{
owned = aOther.owned;
if( aOther.owned )
item = aOther.item->Clone();
else
item = aOther.item;
return *this;
}
operator PNS_ITEM* () const
{
return item;
}
PNS_ITEM *item;
bool owned;
};
typedef std::vector<ENTRY> ENTRIES;
PNS_ITEMSET( PNS_ITEM* aInitialItem = NULL, bool aBecomeOwner = false )
{
if( aInitialItem )
{
m_items.push_back( ENTRY( aInitialItem, aBecomeOwner ) );
}
}
PNS_ITEMSET( const PNS_ITEMSET& aOther )
{
m_items = aOther.m_items;
}
~PNS_ITEMSET();
const PNS_ITEMSET& operator=( const PNS_ITEMSET& aOther )
{
m_items = aOther.m_items;
return *this;
}
2014-11-14 19:19:00 +00:00
int Count( int aKindMask = -1 ) const
{
int n = 0;
2014-11-14 19:19:00 +00:00
BOOST_FOREACH( PNS_ITEM* item, m_items )
{
if( item->Kind() & aKindMask )
n++;
}
return n;
}
bool Empty() const
{
return m_items.empty();
}
ENTRIES& Items() { return m_items; }
const ENTRIES& CItems() const { return m_items; }
PNS_ITEMSET& FilterLayers( int aStart, int aEnd = -1, bool aInvert = false );
PNS_ITEMSET& FilterKinds( int aKindMask, bool aInvert = false );
PNS_ITEMSET& FilterNet( int aNet, bool aInvert = false );
PNS_ITEMSET& FilterMarker( int aMarker, bool aInvert = false );
PNS_ITEMSET& ExcludeLayers( int aStart, int aEnd = -1 )
{
return FilterLayers( aStart, aEnd, true );
}
PNS_ITEMSET& ExcludeKinds( int aKindMask )
{
2014-11-14 19:19:00 +00:00
return FilterKinds( aKindMask, true );
}
PNS_ITEMSET& ExcludeNet( int aNet )
{
2014-11-14 19:19:00 +00:00
return FilterNet( aNet, true );
}
2014-11-14 19:19:00 +00:00
PNS_ITEMSET& ExcludeItem( const PNS_ITEM* aItem );
2014-11-14 19:19:00 +00:00
int Size() const
{
return m_items.size();
}
void Add( const PNS_LINE& aLine );
void Prepend( const PNS_LINE& aLine );
PNS_ITEM* operator[] ( int index ) const
{
return m_items[index].item;
}
void Add( PNS_ITEM* aItem, bool aBecomeOwner = false )
2014-11-14 19:19:00 +00:00
{
m_items.push_back( ENTRY( aItem, aBecomeOwner ) );
2013-09-26 21:53:54 +00:00
}
void Prepend( PNS_ITEM* aItem, bool aBecomeOwner = false )
{
m_items.insert( m_items.begin(), ENTRY( aItem, aBecomeOwner ) );
}
void Clear()
{
m_items.clear();
}
bool Contains( PNS_ITEM* aItem ) const
{
const ENTRY ent( aItem );
return std::find( m_items.begin(), m_items.end(), ent ) != m_items.end();
}
2014-11-14 19:19:00 +00:00
void Erase( PNS_ITEM* aItem )
{
ENTRY ent( aItem );
ENTRIES::iterator f = std::find( m_items.begin(), m_items.end(), ent );
if( f != m_items.end() )
2014-11-14 19:19:00 +00:00
m_items.erase( f );
}
2014-11-14 19:19:00 +00:00
template<class T>
2015-08-12 19:12:12 +00:00
T* FindByKind( PNS_ITEM::PnsKind kind, int index = 0 )
{
int n = 0;
BOOST_FOREACH( const PNS_ITEM* item, m_items )
2015-08-12 19:12:12 +00:00
{
if( item->OfKind( kind ) )
{
2015-08-12 19:12:12 +00:00
if( index == n )
return static_cast<T*>( item );
else
n++;
}
}
2015-08-12 19:12:12 +00:00
return NULL;
}
private:
ENTRIES m_items;
};
#endif