summaryrefslogtreecommitdiff
path: root/boost/compute/detail/buffer_value.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'boost/compute/detail/buffer_value.hpp')
-rw-r--r--boost/compute/detail/buffer_value.hpp178
1 files changed, 178 insertions, 0 deletions
diff --git a/boost/compute/detail/buffer_value.hpp b/boost/compute/detail/buffer_value.hpp
new file mode 100644
index 0000000000..6a4e78fc19
--- /dev/null
+++ b/boost/compute/detail/buffer_value.hpp
@@ -0,0 +1,178 @@
+//---------------------------------------------------------------------------//
+// Copyright (c) 2013 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_DETAIL_BUFFER_VALUE_HPP
+#define BOOST_COMPUTE_DETAIL_BUFFER_VALUE_HPP
+
+#include <boost/compute/context.hpp>
+#include <boost/compute/command_queue.hpp>
+#include <boost/compute/detail/device_ptr.hpp>
+#include <boost/compute/detail/read_write_single_value.hpp>
+
+namespace boost {
+namespace compute {
+namespace detail {
+
+template<class T>
+class buffer_value
+{
+public:
+ typedef T value_type;
+
+ buffer_value()
+ {
+ }
+
+ buffer_value(const value_type &value)
+ : m_value(value)
+ {
+ }
+
+ // creates a reference for the value in buffer at index (in bytes).
+ buffer_value(const buffer &buffer, size_t index)
+ : m_buffer(buffer.get(), false),
+ m_index(index)
+ {
+ }
+
+ buffer_value(const buffer_value<T> &other)
+ : m_buffer(other.m_buffer.get(), false),
+ m_index(other.m_index)
+ {
+ }
+
+ ~buffer_value()
+ {
+ // set buffer to null so that its reference count will
+ // not be decremented when its destructor is called
+ m_buffer.get() = 0;
+ }
+
+ operator value_type() const
+ {
+ if(m_buffer.get()){
+ const context &context = m_buffer.get_context();
+ const device &device = context.get_device();
+ command_queue queue(context, device);
+
+ return detail::read_single_value<T>(m_buffer, m_index / sizeof(T), queue);
+ }
+ else {
+ return m_value;
+ }
+ }
+
+ buffer_value<T> operator-() const
+ {
+ return -T(*this);
+ }
+
+ bool operator<(const T &value) const
+ {
+ return T(*this) < value;
+ }
+
+ bool operator>(const T &value) const
+ {
+ return T(*this) > value;
+ }
+
+ bool operator<=(const T &value) const
+ {
+ return T(*this) <= value;
+ }
+
+ bool operator>=(const T &value) const
+ {
+ return T(*this) <= value;
+ }
+
+ bool operator==(const T &value) const
+ {
+ return T(*this) == value;
+ }
+
+ bool operator==(const buffer_value<T> &other) const
+ {
+ if(m_buffer.get() != other.m_buffer.get()){
+ return false;
+ }
+
+ if(m_buffer.get()){
+ return m_index == other.m_index;
+ }
+ else {
+ return m_value == other.m_value;
+ }
+ }
+
+ bool operator!=(const T &value) const
+ {
+ return T(*this) != value;
+ }
+
+ buffer_value<T>& operator=(const T &value)
+ {
+ if(m_buffer.get()){
+ const context &context = m_buffer.get_context();
+ command_queue queue(context, context.get_device());
+
+ detail::write_single_value<T>(value, m_buffer, m_index / sizeof(T), queue);
+
+ return *this;
+ }
+ else {
+ m_value = value;
+ return *this;
+ }
+ }
+
+ buffer_value<T>& operator=(const buffer_value<T> &value)
+ {
+ return operator=(T(value));
+ }
+
+ detail::device_ptr<T> operator&() const
+ {
+ return detail::device_ptr<T>(m_buffer, m_index);
+ }
+
+ buffer_value<T>& operator++()
+ {
+ if(m_buffer.get()){
+ T value = T(*this);
+ value++;
+ *this = value;
+ }
+ else {
+ m_value++;
+ }
+
+ return *this;
+ }
+
+ buffer_value<T> operator++(int)
+ {
+ buffer_value<T> result(*this);
+ ++(*this);
+ return result;
+ }
+
+private:
+ const buffer m_buffer;
+ size_t m_index;
+ value_type m_value;
+};
+
+} // end detail namespace
+} // end compute namespace
+} // end boost namespace
+
+#endif // BOOST_COMPUTE_DETAIL_BUFFER_VALUE_HPP