Ball.h

Go to the documentation of this file.

/*
 *  Copyright (C) 2012 Josh Bialkowski (jbialk@mit.edu)
 *
 *  This file is part of mpblocks.
 *
 *  mpblocks 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.
 *
 *  mpblocks 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 mpblocks.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef MPBLOCKS_KD_TREE_EUCLIDEAN_BLOCKS_BALL_H_
#define MPBLOCKS_KD_TREE_EUCLIDEAN_BLOCKS_BALL_H_

namespace  mpblocks {
namespace   kd_tree {
namespace euclidean {
namespace    blocks {

template <class Traits,
            template<class> class Allocator = std::allocator >
class Ball
{
    public:
        typedef typename Traits::Format_t   Format_t;
        typedef typename Traits::Node       Node_t;

        typedef Eigen::Matrix<Format_t,Traits::NDim,1>  Point_t;

        typedef Ball<Traits,Allocator>                  This_t;
        typedef euclidean::Ball<Traits,Allocator>       Search_t;
        typedef Tree<Traits>                            Tree_t;

    protected:
        Tree_t*  m_tree;
        Search_t m_search;

    public:
        sigc::signal<void,Node_t*>  sig_result;

        Ball(Tree_t* tree=0):
            m_tree(tree)
        {}

        void setTree( Tree_t* tree )
            { m_tree = tree; }

        void search( const Point_t& center, Format_t radius )
        {
            m_search.reset(center,radius);
            if(m_tree)
            {
                m_tree->findRange(m_search);
                const typename Search_t::List_t& list = m_search.result();
                typename Search_t::List_t::const_iterator ipNode;
                for( ipNode = list.begin(); ipNode != list.end(); ++ipNode )
                {
                    sig_result.emit(*ipNode);
                }
            }
        }

        sigc::slot<void,const Point_t&,Format_t> slot_search()
            { return sigc::mem_fun(this,This_t::search); }

};





} // namespace blocks
} // namespace euclidean
} // namespace kd_tree
} // namespace mpblocks








#endif // NEARESTNEIGHBOR_H_