sparsegrid_util_common.hpp

#ifndef SPARSEGRID_UTIL_COMMON
#define SPARSEGRID_UTIL_COMMON

template<typename base>
struct sparse_grid_bck_wrapper_impl
{
    base & b;

    typedef typename base::value_type bt_type;

    sparse_grid_bck_wrapper_impl(base & b)
    :b(b)
    {}

    auto operator[](int i) -> decltype(b[0])
    {
        b[0];
    }

    template<typename T>
    sparse_grid_bck_wrapper_impl<base> & operator=(T c)
    {
        for (int i = 0; i < b.size() ; i++)
        {
            b[i] = c;
        }

        return *this;
    }

    operator bt_type() const
    {
        return b[0];
    }
};

// openfpm::detail::multi_array::sub_array_openfpm<T,N1,vmpl>

template<typename T,typename vmpl>
struct sparse_grid_bck_wrapper_impl<openfpm::detail::multi_array::sub_array_openfpm<T, 1, vmpl>>
{
    openfpm::detail::multi_array::sub_array_openfpm<T, 1, vmpl > b;

    
    sparse_grid_bck_wrapper_impl(openfpm::detail::multi_array::sub_array_openfpm<T, 1, vmpl > b)
    :b(b)
    {}

    auto operator[](int i) -> decltype(b[i][0])
    {
        return b[i][0];
    }

    auto operator[](int i) const -> decltype(b[i][0])
    {
        return b[i][0];
    }
};

template<typename chunk_def>
struct sparse_grid_bck_value
{
    chunk_def bck;

    sparse_grid_bck_value(chunk_def bck)
    :bck(bck)
    {}

    template<unsigned int p>
    auto get() -> sparse_grid_bck_wrapper_impl<typename std::remove_reference<decltype(bck.template get<p>())>::type>
    {
        return sparse_grid_bck_wrapper_impl<typename std::remove_reference<decltype(bck.template get<p>())>::type>(bck.template get<p>());
    }

    template<unsigned int p>
    auto get() const -> sparse_grid_bck_wrapper_impl<typename std::remove_reference<decltype(bck.template get<p>())>::type>
    {
        return sparse_grid_bck_wrapper_impl<typename std::remove_reference<decltype(bck.template get<p>())>::type>(bck.template get<p>());
    }
};

#endif