diff options
Diffstat (limited to 'boost/compute/interop/vtk/points.hpp')
-rw-r--r-- | boost/compute/interop/vtk/points.hpp | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/boost/compute/interop/vtk/points.hpp b/boost/compute/interop/vtk/points.hpp new file mode 100644 index 0000000000..fefbbb9874 --- /dev/null +++ b/boost/compute/interop/vtk/points.hpp @@ -0,0 +1,55 @@ +//---------------------------------------------------------------------------// +// Copyright (c) 2013-2014 Kyle Lutz <kyle.r.lutz@gmail.com> +// +// Distributed under the Boost Software License, Version 1.0 +// See accompanying file LICENSE_1_0.txt or copy at +// http://www.boost.org/LICENSE_1_0.txt +// +// See http://boostorg.github.com/compute for more information. +//---------------------------------------------------------------------------// + +#ifndef BOOST_COMPUTE_INTEROP_VTK_POINTS_HPP +#define BOOST_COMPUTE_INTEROP_VTK_POINTS_HPP + +#include <vector> + +#include <vtkPoints.h> + +#include <boost/compute/system.hpp> +#include <boost/compute/command_queue.hpp> +#include <boost/compute/algorithm/copy.hpp> +#include <boost/compute/iterator/buffer_iterator.hpp> + +namespace boost { +namespace compute { + +/// Copies \p points to \p buffer. +/// +/// For example, to copy from a \c vtkPoints object to a \c vector<float4_>: +/// \code +/// vtkPoints *points = ... +/// vector<float4_> vector(points->GetNumberOfPoints(), context); +/// vtk_copy_points_to_buffer(points, vector.begin(), queue); +/// \endcode +template<class PointType> +inline void vtk_copy_points_to_buffer(const vtkPoints *points, + buffer_iterator<PointType> buffer, + command_queue &queue = system::default_queue()) +{ + vtkPoints *points_ = const_cast<vtkPoints *>(points); + + // copy points to aligned buffer + std::vector<PointType> tmp(points_->GetNumberOfPoints()); + for(vtkIdType i = 0; i < points_->GetNumberOfPoints(); i++){ + double *p = points_->GetPoint(i); + tmp[i] = PointType(p[0], p[1], p[2], 1); + } + + // copy data to device + copy(tmp.begin(), tmp.end(), buffer, queue); +} + +} // end compute namespace +} // end boost namespace + +#endif // BOOST_COMPUTE_INTEROP_VTK_POINTS_HPP |