Density.h

Go to the documentation of this file.

#ifndef DENSITY_H
#define DENSITY_H
 
#define DENSITY_VERSION "4"
 
#include <iostream>
#include <memory>
#include <string>
#include <valarray>
 
#include "../../src/GenericManagingModule.h"
#include "../../src/ModuleFactory.h"
 
template <typename T>
class Density : public GenericManagingModule<T>
{
    private:
        static bool m_registered;
 
        std::string m_ini_filepath{"density/Density.ini"};
 
    public:
        Density(SimulationClassBase<T>* sim);
 
        bool setup(std::vector<std::shared_ptr<GenericSubmodule<T>>> all_submodules) override;
        
        bool exec(std::string_view param) override;  // main functionality of the module
 
        bool init() override;
 
        bool preTimeStep() override;
 
        bool postTimeStep() override { return true; };
 
        bool output() override { return true; };
 
        void writeToSim();
 
        std::vector<std::string> getChainInsertion() const override
        {
            return this->ini_file_data.getStringVectorParameters("chain_insertion");
        };
 
        static std::shared_ptr<GenericManagingModule<T>> createMethode(SimulationClassBase<T>* sim)
        {
            return std::make_shared<Density<T>>(sim);
        };
 
        static std::string getName() { return "Density"; };
 
        std::string_view getNameLocal() const override { return "Density"; };
};
 
// ================= Implementation =================
 
template <typename T>
Density<T>::Density(SimulationClassBase<T>* sim) : GenericManagingModule<T>(sim)
{
    try
    {
        std::string ini_folder_path{
            this->sim->m_simulation_config.getStringParameters("ini_folder_path")};
        this->ini_file_data.loadUserInput(ini_folder_path + m_ini_filepath);
        this->m_generic_submodules = this->ini_file_data.getStringVectorParameters("submodules");
    }
    catch (const BadInput& e)
    {
        std::cerr << e.what() << '\n';
    }
}
 
template <typename T>
bool Density<T>::setup(std::vector<std::shared_ptr<GenericSubmodule<T>>> all_submodules)
{
    this->sim->m_field_map.insert(
        {"Density",
         std::valarray<T>(this->sim->m_simulation_config.getIntParameters("numerical_layers")
                          * this->sim->m_simulation_config.getIntParameters("number_of_facets"))});
    this->module_field = std::make_shared<std::valarray<T>>(
        this->sim->m_simulation_config.getIntParameters("numerical_layers")
        * this->sim->m_simulation_config.getIntParameters("number_of_facets"));
    this->setSubmodules(all_submodules);
    return true;
}
 
template <typename T>
bool Density<T>::exec(std::string_view param)
{
    if (param == "InitChain")
    {
        return init();
    }
    if (param == "PreTimeStepChain")
    {
        return preTimeStep();
    }
    if (param == "PostTimeStepChain")
    {
        return postTimeStep();
    }
    if (param == "OutputChain")
    {
        return output();
    }
    return false;
}
 
template <typename T>
void Density<T>::writeToSim()
{
    this->sim->m_field_map["Density"] = (*this->module_field);
}
 
template <typename T>
bool Density<T>::init()
{
    this->sub_module_chain.runChain("InitChain");
    writeToSim();
    return true;
}
 
template <typename T>
bool Density<T>::preTimeStep()
{
    this->sub_module_chain.runChain("PreTimeStepChain");
    writeToSim();
    return true;
}
 
template <typename T>
bool Density<T>::m_registered = ModuleFactory<T>::registerModule(getName(), createMethode);
 
#endif // DENSITY_H