summaryrefslogtreecommitdiff
path: root/boost/compute/interop/vtk/points.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'boost/compute/interop/vtk/points.hpp')
-rw-r--r--boost/compute/interop/vtk/points.hpp55
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