multi_array_view.hpp

//  Copyright (c) 2014 Christopher Hinz
//
//  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)

#ifndef ECHELON_MULTI_ARRAY_VIEW_HPP
#define ECHELON_MULTI_ARRAY_VIEW_HPP

#include <echelon/range.hpp>

#include <echelon/detail/map_indices.hpp>
#include <echelon/detail/all_integral.hpp>

#include <algorithm>
#include <functional>
#include <vector>
#include <utility>
#include <type_traits>

namespace echelon
{

template <typename T>
class multi_array_view
{
public:
    using value_type = T;

    multi_array_view(T* data_, std::vector<std::size_t> shape_)
    : data_(data_), shape_(std::move(shape_))
    {
    }

    template <typename... Indices, typename Enabler = typename std::enable_if<
                                       detail::all_integral<Indices...>::value>::type>
    const value_type& operator()(Indices... indices) const
    {
        static_assert(detail::all_integral<Indices...>::value,
                      "All indices must be of integral type.");

        return data_[detail::map_indices(shape_, indices...)];
    }

    template <typename... Indices, typename Enabler = typename std::enable_if<
                                       detail::all_integral<Indices...>::value>::type>
    value_type& operator()(Indices... indices)
    {
        static_assert(detail::all_integral<Indices...>::value,
                      "All indices must be of integral type.");

        return data_[detail::map_indices(shape_, indices...)];
    }

    template <typename... Args, typename Enabler = typename std::enable_if<
                                    !detail::all_integral<Args...>::value>::type>
    echelon::hdf5::array_slice<T> operator()(Args... args)
    {
        return echelon::make_slice(*this, std::forward<Args>(args)...);
    }

    const value_type* data() const
    {
        return data_;
    }

    value_type* data()
    {
        return data_;
    }

    const std::vector<std::size_t>& shape() const
    {
        return shape_;
    }

private:
    T* data_;
    std::vector<std::size_t> shape_;
};
}

#endif