sph_data_containers.h

Go to the documentation of this file.

#ifndef SPH_DATA_CONTAINERS_H
#define SPH_DATA_CONTAINERS_H

#include "base_data_package.h"
#include "base_data_type.h"

namespace SPH
{
    class BaseMaterial;
    class SPHBody;
    class RealBody;
    class SolidBody;
    class BodyPart;
    class FictitiousBody;
    class CellList;
    class BaseParticles;

    using MaterialVector = StdVec<BaseMaterial *>;
    using SPHBodyVector = StdVec<SPHBody *>;
    using SolidBodyVector = StdVec<SolidBody *>;
    using RealBodyVector = StdVec<RealBody *>;
    using BodyPartVector = StdVec<BodyPart *>;
    using FictitiousBodyVector = StdVec<FictitiousBody *>;

    using IndexVector = StdVec<size_t>;
    using ConcurrentIndexVector = LargeVec<size_t>;

    using ListData = std::pair<size_t, Vecd>;
    using ListDataVector = StdLargeVec<ListData>;
    using CellLists = StdLargeVec<CellList *>;

    template <class DataType>
    using ConcurrentVector = LargeVec<DataType>;
    using ConcurrentCellLists = LargeVec<CellList *>;
    using SplitCellLists = StdVec<ConcurrentCellLists>;
    using PositionsVolumes = StdVec<std::pair<Vecd, Real>>;

    typedef GeneralDataAssemble<StdLargeVec> ParticleData;
    typedef std::array<std::map<std::string, size_t>, 4> ParticleDataMap;
    typedef std::array<StdVec<std::pair<std::string, size_t>>, 4> ParticleVariableList;
        
    template <typename VariableType>
    struct loopParticleDataMap
    {
        template <typename VariableOperation>
        void operator()(ParticleData &particle_data,
                        ParticleDataMap &particle_data_map, VariableOperation &variable_operation) const
        {
            constexpr int type_index = DataTypeIndex<VariableType>::value;
            for (auto const &name_index : particle_data_map[type_index])
            {
                std::string variable_name = name_index.first;
                StdLargeVec<VariableType> &variable = *(std::get<type_index>(particle_data)[name_index.second]);
                variable_operation(variable_name, variable);
            }
        };
    };

    template <typename VariableType>
    struct loopVariableNameList
    {
        template <typename VariableOperation>
        void operator()(ParticleData &particle_data,
                        ParticleVariableList &variable_name_list, VariableOperation &variable_operation) const
        {
            constexpr int type_index = DataTypeIndex<VariableType>::value;
            for (std::pair<std::string, size_t> &name_index : variable_name_list[type_index])
            {
                std::string variable_name = name_index.first;
                StdLargeVec<VariableType> &variable = *(std::get<type_index>(particle_data)[name_index.second]);
                variable_operation(variable_name, variable);
            }
        };
    };  
}
#endif // SPH_DATA_CONTAINERS_H