mass.hh

#ifndef VTESTBED_CORE_MASS_HH
#define VTESTBED_CORE_MASS_HH

#include <vector>

#include <vtestbed/base/blitz.hh>
#include <vtestbed/core/types.hh>

namespace vtb {

    namespace core {

        template <class T>
        struct Mass {

        public:
            using vec3 = blitz::TinyVector<T,3>;

        public:
            T m{};
            vec3 x{T{}};

            Mass() = default;
            inline Mass(T mass, const vec3& point): m(mass), x(point) {}

            inline T mass() const { return m; }
            inline const vec3& point() const { return x; }

        };

        template <class T>
        class Mass_array: public std::vector<Mass<T>> {

        private:
            using base_type = std::vector<Mass<T>>;

        public:
            using base_type::base_type;

            inline T
            sum() const {
                T s{};
                for (const auto& m : *this) { s += m.mass(); }
                return s;
            }

        };

        template <class T, class ... Args>
        Mass_array<T>
        flatten(const Args& ... args) {
            std::reference_wrapper<const Mass_array<T>> arrays[sizeof...(args)] = {args...};
            size_t n = 0;
            for (const auto& a : arrays) { n += a.get().size(); }
            Mass_array<T> result;
            result.reserve(n);
            for (const auto& a : arrays) {
                for (const auto& m : a.get()) { result.emplace_back(m); }
            }
            return result;
        }

    }

}

#endif // vim:filetype=cpp