PointSet.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_DUBINS_CURVES_CUDA2_POINTSET_H_
#define MPBLOCKS_DUBINS_CURVES_CUDA2_POINTSET_H_


#include <map>
#include <string>

namespace    mpblocks {
namespace      dubins {
namespace curves_cuda {


template <typename Format_t>
class ResultBlock
{
    public:
        typedef unsigned int uint_t;

    private:
        uint_t m_rows;
        uint_t m_cols;

        Format_t* m_buf;

    public:
        ResultBlock( uint_t rows=0, uint_t cols=0 ):
            m_rows(0),
            m_cols(0),
            m_buf(0)
        {
            allocate( rows, cols );
        }

        ~ResultBlock()
        {
            if( m_buf )
                delete [] m_buf;
        }

        void allocate( uint_t rows, uint_t cols );

        Format_t* ptr() const { return m_buf; }
        uint_t rows()  const { return m_rows; }
        uint_t cols()  const { return m_cols; }
        uint_t pitch() const { return m_cols*sizeof(Format_t); }

        Format_t operator()( int i, int j )
        {
            return m_buf[ i*m_cols + j ];
        }
};



template <typename Format_t>
class PointSet
{
    public:
        typedef unsigned int uint_t;
        typedef std::map< std::string, cuda::FuncAttributes >  fattrMap_t;
        typedef cuda::bitonic::Sorter<Format_t,Format_t> Sorter_t;

    private:
        Params<Format_t> m_params;   
        uint_t    m_dbAlloc;    
        uint_t    m_dbAlloc2;   
        uint_t    m_dbSize;     
        size_t    m_pitchIn;    
        size_t    m_pitchOut;   
        Format_t* m_g_in;       
        Format_t* m_g_out;      
        Format_t* m_g_sorted;   

        Sorter_t  m_sorter;     

        uint_t  m_threadsPerBlock;  
        uint_t  m_nSM;              

    public:
        PointSet(uint_t n=10, Format_t r=1);
        ~PointSet();

        void deallocate();

        void allocate(uint_t n);

        void set_r( Format_t r );

        void clear(bool clearmem=false);

        void config();

        void config(int dev);

    private:
        void computeGrid( uint_t& blocks, uint_t& threads );

    public:
        int  insert( Format_t q[3] );

        void distance_to_set(  Format_t q[3], ResultBlock<Format_t>& out );

        void distance_from_set(  Format_t q[3], ResultBlock<Format_t>& out  );

        void nearest_children( Format_t q[3], ResultBlock<Format_t>& out  );

        void nearest_parents(  Format_t q[3], ResultBlock<Format_t>& out );

        void group_distance_to_set( Format_t q[3], ResultBlock<Format_t>& out );

        void group_distance_neighbors( Format_t q[3], ResultBlock<Format_t>& out );

        static void get_fattr( fattrMap_t&  );
};



} // curves
} // dubins
} // mpblocks

#endif // POINTSET_H_