write.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 <pni/core/arrays/slice.hpp>
#include "../nxobject.hpp"
#include "../nxobject_traits.hpp"

namespace pni{
namespace io{
namespace nx{
    
    //------------------------------------------------------------------------
    template<
             template<nximp_code> class OTYPE,
             nximp_code IMPID,
             typename ATYPE
            >
    void write(const OTYPE<IMPID> &o,const ATYPE &a)
    {
        typedef nxobject_trait<IMPID> trait_type;
        typedef typename trait_type::field_type field_type;
        typedef typename trait_type::attribute_type attribute_type;

        static_assert(std::is_same<OTYPE<IMPID>,field_type>::value || 
                      std::is_same<OTYPE<IMPID>,attribute_type>::value,
                      "WRITE CAN ONLY BE USED WITH FIELDS AND "
                      "ATTRIBUTES!");
        o.write(a);
    }

    //------------------------------------------------------------------------
    template<
             template<nximp_code> class OTYPE,
             nximp_code IMPID,
             typename ATYPE,
             typename ...ITYPES
            >
    void write(const OTYPE<IMPID> &o,const ATYPE &a,ITYPES ...indices)
    {
        typedef nxobject_trait<IMPID> trait_type;
        typedef typename trait_type::field_type field_type;
        typedef typename trait_type::attribute_type attribute_type;

        static_assert(std::is_same<OTYPE<IMPID>,field_type>::value || 
                      std::is_same<OTYPE<IMPID>,attribute_type>::value,
                      "WRITE CAN ONLY BE USED WITH FIELDS AND "
                      "ATTRIBUTES!");
        
        o(indices...).write(a);
    }

    //-------------------------------------------------------------------------
    template<
             typename ATYPE,
             typename GTYPE,
             typename FTYPE,
             typename AATYPE,
             typename LTYPE
            > 
    class write_visitor : public boost::static_visitor<void>
    {
        public: 
            typedef std::vector<pni::core::slice> selection_t;
        private:
            const ATYPE &_data;
            selection_t _selection;
        public:
            using result_type = void;
            using group_type = GTYPE;
            using field_type = FTYPE;
            using attribute_type = AATYPE;
            using link_type = LTYPE;

            //-----------------------------------------------------------------
            write_visitor(const ATYPE &data,
                          const selection_t &s=selection_t()):
                _data(data),
                _selection(s)
            {}

            //-----------------------------------------------------------------
            result_type operator()(const group_type &) const
            {
                using namespace pni::core;
                throw type_error(EXCEPTION_RECORD,
                        "One cannot write data to a group object!");
            }

            //-----------------------------------------------------------------
            result_type operator()(const field_type &f) const
            {
                if(_selection.size())
                    f(_selection).write(_data);
                else
                    f.write(_data);
            }

            //-----------------------------------------------------------------
            result_type operator()(const attribute_type &a) const
            {
                if(_selection.size())
                    a(_selection).write(_data);
                else
                    a.write(_data);
            }
            
            //-----------------------------------------------------------------
            result_type operator()(const link_type &) const
            {
                using namespace pni::core;
                throw type_error(EXCEPTION_RECORD,
                        "Cannot write data to a link object!");
            }
    };

    //------------------------------------------------------------------------
    template<
             typename ATYPE,
             typename GTYPE,
             typename FTYPE,
             typename AATYPE,
             typename LTYPE,
             typename ...ITYPES 
            > 
    void write(nxobject<GTYPE,FTYPE,AATYPE,LTYPE> &&o,const ATYPE &a,
               ITYPES ...indices)
    {
        using visitor_type = write_visitor<ATYPE,GTYPE,FTYPE,AATYPE,LTYPE>;
        std::vector<pni::core::slice> sel{pni::core::slice(indices)...};
        return boost::apply_visitor(visitor_type(a,sel),o);
    }

    //------------------------------------------------------------------------
    template<
             typename ATYPE,
             typename GTYPE,
             typename FTYPE,
             typename AATYPE,
             typename LTYPE,
             typename ...ITYPES
            >
    void write(const nxobject<GTYPE,FTYPE,AATYPE,LTYPE> &o,const ATYPE &a,
               ITYPES ...indices)
    {
        using visitor_type = write_visitor<ATYPE,GTYPE,FTYPE,AATYPE,LTYPE>;
        std::vector<pni::core::slice> sel{pni::core::slice(indices)...};

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

    //------------------------------------------------------------------------
    template<
             typename ATYPE,
             typename GTYPE,
             typename FTYPE,
             typename AATYPE,
             typename LTYPE
            >
    void write(const nxobject<GTYPE,FTYPE,AATYPE,LTYPE> &o,const ATYPE &a,
               const std::vector<pni::core::slice> &sel)
    {
        using visitor_type =  write_visitor<ATYPE,GTYPE,FTYPE,AATYPE,LTYPE>;

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



//end of namespace
}
}
}