static_index_map.hpp

//
// (c) Copyright 2013 DESY, Eugen Wintersberger <eugen.wintersberger@desy.de>
//
// This file is part of libpnicore.
//
// libpnicore is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 2 of the License, or
// (at your option) any later version.
//
// libpnicore is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with libpnicore.  If not, see <http://www.gnu.org/licenses/>.
//
// ============================================================================
//
// Created on: Oct 25, 2013
//     Author: Eugen Wintersberger <eugen.wintersberger@desy.de>
//

#pragma once
#include <algorithm>
#include <numeric>
#include "../../utilities/container_utils.hpp"

namespace pni{
namespace core{


    template<
             typename MAP_IMP,
             size_t... DIMS
            > 
    class static_index_map
    {
        public:
            //=================public types====================================
            typedef std::array<size_t,sizeof...(DIMS)> storage_type;
            typedef MAP_IMP     implementation_type;
            typedef storage_type index_type;
            typedef typename storage_type::const_iterator const_iterator;
        private:
            constexpr static storage_type _shape{{DIMS...}};
        public:

            //-----------------------------------------------------------------
            size_t max_elements() const
            {
                return std::accumulate(_shape.begin(),_shape.end(),size_t(1),
                                       std::multiplies<size_t>()); 
            }

            //-----------------------------------------------------------------
            constexpr size_t rank() const { return sizeof...(DIMS); }

            //-----------------------------------------------------------------
            constexpr size_t size() const { return rank(); }


            //-----------------------------------------------------------------

            template<typename CTYPE,
                     typename = typename std::enable_if<
                     std::is_compound<
                     typename std::remove_const<
                     typename std::remove_reference<CTYPE>::type
                     >::type
                     >::value
                         >::type
                    > 
            size_t offset(const CTYPE &index) const
            {
                return implementation_type::template offset(_shape,index);
            }
            
            //-----------------------------------------------------------------
            template<typename CTYPE,
                     typename = typename std::enable_if<
                     std::is_compound<
                     typename std::remove_const<
                     typename std::remove_reference<CTYPE>::type 
                     >::type
                     >::value
                        >::type
                     >
            size_t offset(const array_selection &s,const CTYPE &index) 
            const
            {
                return implementation_type::template offset(s,_shape,index);
            }

            //-----------------------------------------------------------------

            template<typename CTYPE> CTYPE index(size_t offset) const
            {
                CTYPE index = container_utils<CTYPE>::create(rank()); 
                implementation_type::template index(_shape,index,offset);
                return index;
            }

            //-----------------------------------------------------------------
            const_iterator begin() const { return _shape.begin(); }

            //-----------------------------------------------------------------
            const_iterator end() const { return _shape.end(); }

    };
    
template<typename MAP_IMP,size_t... DIMS> 
    constexpr typename static_index_map<MAP_IMP,DIMS...>::storage_type 
    static_index_map<MAP_IMP,DIMS...>::_shape;
//end of namespace
}
}