combine_pDs.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef cMHN_UTILITY_COMBINE_PDS_H
#define cMHN_UTILITY_COMBINE_PDS_H
 
#include <map>
#include <vector>
 
#include <prc.hpp>
 
namespace cMHN
{
    template<class T, class S>
    static inline auto combinePDs(std::vector<std::map<S, T>> const &pDs,
        std::vector<pRC::Index> const &lengths)
    {
        std::map<S, T> res_pD;
 
        T sum = pRC::zero<T>();
 
        for(pRC::Index i = 0; i < pDs.size(); ++i)
        {
            auto const pD = pDs[i];
            auto const length = lengths[i];
            for(auto const &[k, v] : pD)
            {
                res_pD.try_emplace(k, pRC::zero<T>());
                res_pD[k] += T(length) * v;
                sum += T(length) * v;
            }
        }
 
        for(auto &[k, v] : res_pD)
        {
            v = v / sum;
        }
 
        return res_pD;
    }
} // namespace cMHN
 
#endif // cMHN_UTILITY_COMBINE_PDS_H