get_size.hpp

Go to the documentation of this file.

//
// (c) Copyright 2013 DESY, Eugen Wintersberger <eugen.wintersberger@desy.de>
//
// This file is part of libpniio.
//
// libpniio 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.
//
// libpniio 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 libpniio.  If not, see <http://www.gnu.org/licenses/>.
// ===========================================================================
// Created on: Jul 5, 2013
//     Author: Eugen Wintersberger <eugen.wintersberger@desy.de>
//
#pragma once

#include <pni/core/types.hpp>
#include "../nximp_code.hpp"
#include "../nxobject.hpp"
#include "../nxobject_traits.hpp"

namespace pni{
namespace io{
namespace nx{

    template<
             template<nximp_code> class OTYPE,
             nximp_code IMPID
            >
    size_t get_size(const OTYPE<IMPID> &o)
    {
        using field_type = typename nxobject_trait<IMPID>::field_type;
        using attr_type  = typename nxobject_trait<IMPID>::attribute_type;
        using group_type = typename nxobject_trait<IMPID>::group_type;

        static_assert(std::is_same<field_type,OTYPE<IMPID>>::value ||
                      std::is_same<attr_type,OTYPE<IMPID>>::value ||
                      std::is_same<group_type,OTYPE<IMPID>>::value,
                      "Argument must be a group, field, or "
                      "attribute instance!");

        return o.size();
    }

    //-------------------------------------------------------------------------
    template<
             typename GTYPE,
             typename FTYPE,
             typename ATYPE,
             typename LTYPE
            > 
    class get_size_visitor : public boost::static_visitor<size_t>
    {
        public:
            using result_type = size_t;
            using group_type = GTYPE;
            using field_type =  FTYPE;
            using attribute_type =  ATYPE;
            using link_type = LTYPE;

            //-----------------------------------------------------------------
            result_type operator()(const group_type &g) const
            {
                return g.size();
            }

            //-----------------------------------------------------------------
            result_type operator()(const field_type &f) const
            {
                return f.size();
            }

            //-----------------------------------------------------------------
            result_type operator()(const attribute_type &a) const
            {
                return a.size();
            }

            //-----------------------------------------------------------------
            result_type operator()(const link_type &) const
            {
                using namespace pni::core;
                throw type_error(EXCEPTION_RECORD,
                        "Links do not have a size!");
                return 0;
            }
            
    };


    template<
             typename GTYPE,
             typename FTYPE,
             typename ATYPE,
             typename LTYPE
            > 
    size_t get_size(const nxobject<GTYPE,FTYPE,ATYPE,LTYPE> &o)
    {
        using visitor_type = get_size_visitor<GTYPE,FTYPE,ATYPE,LTYPE>;

        return boost::apply_visitor(visitor_type(),o);
    }

//end of namespace
}
}
}