callback_utils_meta_init.hpp

Go to the documentation of this file.

 
#ifndef CALLBACK_UTILS_META_INIT_HPP
#define CALLBACK_UTILS_META_INIT_HPP
 
#include <cassert>
#include <functional>
#include <memory>
 
#include "callback.hpp"
#include "hdf5_utils.hpp"
 
namespace CallbackUtils {
 
namespace meta_init
{
 
    template<typename AFields>
    class MultiPrtMetaInitBase :
        virtual public Callback<AFields>
    {// {{{
        static constexpr size_t buf_size = 64UL;
        size_t N_inits = 0UL;
        std::pair<void *, std::function<void(void *, std::shared_ptr<H5::H5File>)>> inits[buf_size];
    protected :
        void register_init (void *obj, std::function<void(void *, std::shared_ptr<H5::H5File>)> fct)
        {
            inits[N_inits++] = std::make_pair(obj, fct);
            assert(N_inits < buf_size);
        }
    public :
        void read_prt_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            for (size_t ii=0; ii != N_inits; ++ii)
                inits[ii].second(inits[ii].first, fptr);
        }
    };// }}}
 
    template<typename AFields>
    class MultiGrpMetaInitBase :
        virtual public Callback<AFields>
    {// {{{
        static constexpr size_t buf_size = 64UL;
        size_t N_inits = 0UL;
        std::pair<void *, std::function<void(void *, std::shared_ptr<H5::H5File>)>> inits[buf_size];
    protected :
        void register_init (void *obj, std::function<void(void *, std::shared_ptr<H5::H5File>)> fct)
        {
            inits[N_inits++] = std::make_pair(obj, fct);
            assert(N_inits < buf_size);
        }
    public :
        void read_grp_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            for (size_t ii=0; ii != N_inits; ++ii)
                inits[ii].second(inits[ii].first, fptr);
        }
    };// }}}
 
    template<typename AFields, typename Child>
    class MultiPrtMetaInit :
        virtual public Callback<AFields>,
        virtual private MultiPrtMetaInitBase<AFields>
    {// {{{
        static void this_prt_meta_init_static (void *obj, std::shared_ptr<H5::H5File> fptr)
        {
            Child *p = (Child *)obj;
            p->this_prt_meta_init(fptr);
        }
    protected :
        virtual void this_prt_meta_init (std::shared_ptr<H5::H5File> fptr) = 0;
    public :
        MultiPrtMetaInit ()
        {
            MultiPrtMetaInitBase<AFields>::register_init(this, this_prt_meta_init_static);
        }
    };// }}}
 
    template<typename AFields, typename Child>
    class MultiGrpMetaInit :
        virtual public Callback<AFields>,
        virtual private MultiGrpMetaInitBase<AFields>
    {// {{{
        static void this_grp_meta_init_static (void *obj, std::shared_ptr<H5::H5File> fptr)
        {
            Child *p = (Child *)obj;
            p->this_grp_meta_init(fptr);
        }
    protected :
        virtual void this_grp_meta_init (std::shared_ptr<H5::H5File> fptr) = 0;
    public :
        MultiGrpMetaInit ()
        {
            MultiGrpMetaInitBase<AFields>::register_init(this, this_grp_meta_init_static);
        }
    };// }}}
 
    template<typename AFields>
    class IllustrisCosmology :
        virtual public Callback<AFields>,
        private MultiPrtMetaInit<AFields, IllustrisCosmology<AFields>>
    {// {{{
        friend MultiPrtMetaInit<AFields, IllustrisCosmology<AFields>>;
        void this_prt_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            #define READ(x) x = hdf5Utils::read_scalar_attr<double,double>(header, #x)
            auto header = fptr->openGroup("/Header");
            READ(HubbleParam);
            READ(Omega0);
            READ(OmegaLambda);
            READ(OmegaBaryon);
            READ(Redshift);
            READ(Time);
            header.close();
            #undef READ
        }
    protected :
        double HubbleParam, 
               Omega0, 
               OmegaLambda, 
               OmegaBaryon, 
               Redshift, 
               Time; 
    };// }}}
 
    template<typename AFields>
    class GadgetCosmology :
        virtual public Callback<AFields>,
        private MultiPrtMetaInit<AFields, GadgetCosmology<AFields>>
    {// {{{
        friend MultiPrtMetaInit<AFields, GadgetCosmology<AFields>>;
        void this_prt_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            #define READ(x) x = hdf5Utils::read_scalar_attr<double,double>(header, #x)
            auto header = fptr->openGroup("/Header");
            READ(HubbleParam);
            READ(Omega0);
            READ(OmegaLambda);
            READ(Redshift);
            READ(Time);
            header.close();
            #undef READ
        }
    protected :
        double HubbleParam, 
               Omega0, 
               OmegaLambda, 
               Redshift, 
               Time; 
    };// }}}
 
    template<typename AFields>
    class SIMBACosmology :
        virtual public Callback<AFields>,
        private MultiGrpMetaInit<AFields, SIMBACosmology<AFields>>
    {// {{{
        friend MultiGrpMetaInit<AFields, SIMBACosmology<AFields>>;
        void this_grp_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            // for some reason SIMBA does not store OmegaBaryon in the particle chunk headers,
            // and Redshift and Time are in the Header group of the group chunk...
            #define READ(x) x = hdf5Utils::read_scalar_attr<double,double>(header, #x)
 
            {
                auto header = fptr->openGroup("/Parameters");
                READ(HubbleParam);
                READ(Omega0);
                READ(OmegaLambda);
                READ(OmegaBaryon);
                header.close();
            }
 
            {
                auto header = fptr->openGroup("/Header");
                READ(Redshift);
                READ(Time);
                header.close();
            }
 
            #undef READ
        }
    protected :
        double HubbleParam, 
               Omega0, 
               OmegaLambda, 
               OmegaBaryon, 
               Redshift, 
               Time; 
    };// }}}
 
    template<typename AFields>
    class IllustrisMassTable :
        virtual public Callback<AFields>,
        private MultiPrtMetaInit<AFields, IllustrisMassTable<AFields>>
    {// {{{
        friend MultiPrtMetaInit<AFields, IllustrisMassTable<AFields>>;
        void this_prt_meta_init (std::shared_ptr<H5::H5File> fptr) override final
        {
            auto header = fptr->openGroup("/Header");
            Ntypes = hdf5Utils::read_vector_attr<double,double>(header, "MassTable", MassTable);
            header.close();
        }
    protected :
        double MassTable[16]; 
        size_t Ntypes; 
    };// }}}
 
} // namespace meta_init
 
 
} // namespace CallbackUtils
 
 
#endif // CALLBACK_UTILS_META_INIT_HPP