connectivity_rtree.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2018 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 PCBNEW_CONNECTIVITY_RTREE_H_
#define PCBNEW_CONNECTIVITY_RTREE_H_
 
#include <math/box2.h>
#include <router/pns_layerset.h>
 
#include <geometry/rtree.h>
 
 
template< class T >
class CN_RTREE
{
public:
 
 CN_RTREE()
 {
 this->m_tree = new RTree<T, int, 3, double>();
 }
 
 ~CN_RTREE()
 {
 delete this->m_tree;
 }
 
 void Insert( T aItem )
 {
 const BOX2I& bbox = aItem->BBox();
 const LAYER_RANGE layers = aItem->Layers();
 
 const int mmin[3] = { layers.Start(), bbox.GetX(), bbox.GetY() };
 const int mmax[3] = { layers.End(), bbox.GetRight(), bbox.GetBottom() };
 
 m_tree->Insert( mmin, mmax, aItem );
 }
 
 void Remove( T aItem )
 {
 
 // First, attempt to remove the item using its given BBox
 const BOX2I& bbox = aItem->BBox();
 const LAYER_RANGE layers = aItem->Layers();
 const int mmin[3] = { layers.Start(), bbox.GetX(), bbox.GetY() };
 const int mmax[3] = { layers.End(), bbox.GetRight(), bbox.GetBottom() };
 
 // If we are not successful ( 1 == not found ), then we expand
 // the search to the full tree
 if( m_tree->Remove( mmin, mmax, aItem ) )
 {
 // N.B. We must search the whole tree for the pointer to remove
 // because the item may have been moved before we have the chance to
 // delete it from the tree
 const int mmin2[3] = { INT_MIN, INT_MIN, INT_MIN };
 const int mmax2[3] = { INT_MAX, INT_MAX, INT_MAX };
 m_tree->Remove( mmin2, mmax2, aItem );
 }
 }
 
 void RemoveAll( )
 {
 m_tree->RemoveAll();
 }
 
 template <class Visitor>
 void Query( const BOX2I& aBounds, const LAYER_RANGE& aRange, Visitor& aVisitor ) const
  {
 const int mmin[3] = { aRange.Start(), aBounds.GetX(), aBounds.GetY() };
 const int mmax[3] = { aRange.End(), aBounds.GetRight(), aBounds.GetBottom() };
 
 m_tree->Search( mmin, mmax, aVisitor );
 }
 
private:
 
 RTree<T, int, 3, double>* m_tree;
};
 
 
#endif /* PCBNEW_CONNECTIVITY_RTREE_H_ */