diff options
Diffstat (limited to 'boost/compute/detail/buffer_value.hpp')
-rw-r--r-- | boost/compute/detail/buffer_value.hpp | 178 |
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 |