multi_array_adapter.hpp

//  Copyright (c) 2012-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_ADAPTER_HPP
#define ECHELON_MULTI_ARRAY_ADAPTER_HPP

#include <echelon/range.hpp>

#include <echelon/hdf5/container_adaption.hpp>
#include <echelon/dataset.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 Container>
class multi_array_adapter
{
public:
    typedef typename Container::value_type value_type;

    multi_array_adapter(Container& container_, std::vector<std::size_t> shape_)
    : container_(container_), 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 container_[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 container_[detail::map_indices(shape_, indices...)];
    }

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

    typename Container::iterator begin()
    {
        return container_.begin();
    }

    typename Container::iterator end()
    {
        return container_.end();
    }

    typename Container::const_iterator begin() const
    {
        return container_.begin();
    }

    typename Container::const_iterator end() const
    {
        return container_.end();
    }

    const value_type* data() const
    {
        using ::echelon::hdf5::data;

        return data(container_);
    }

    value_type* data()
    {
        using ::echelon::hdf5::data;

        return data(container_);
    }

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

private:
    Container& container_;
    std::vector<std::size_t> shape_;
};
}

#endif