HeatCapacityVariableCustomPolynomial.h

Go to the documentation of this file.

#ifndef HEAT_CAPACITY_VARIABLE_CUSTOM_POLYNOMIAL_H
#define HEAT_CAPACITY_VARIABLE_CUSTOM_POLYNOMIAL_H
 
#define VERSION "1"
 
#include <iostream>
#include <memory>
#include <string>
#include <valarray>
 
#include "../../src/GenericSubmodule.h"
#include "../../src/ModuleFactory.h"
 
template <typename T>
class HeatCapacityVariableCustomPolynomial : public GenericSubmodule<T>
{
    private:
        static bool m_registered;
        std::string m_ini_filepath{"heat_capacity/HeatCapacityVariableCustomPolynomial.ini"};
        T m_coefficient_A{};
        T m_coefficient_B{};
        T m_coefficient_C{};
        T m_coefficient_D{};
 
    public:
        HeatCapacityVariableCustomPolynomial(SimulationClassBase<T>* sim);
        bool setup(std::vector<std::shared_ptr<GenericSubmodule<T>>> all_submodules)
            override;  // loads module into necessary module chains (auto load for calculation
                       // chains)
        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 calculateHeatCapacity();
 
        void setFieldPtr(std::shared_ptr<std::valarray<T>> field_ptr) override
        {
            this->module_field = field_ptr;
        }
 
        static std::shared_ptr<GenericSubmodule<T>> createMethode(SimulationClassBase<T>* sim)
        {
            return std::make_shared<HeatCapacityVariableCustomPolynomial<T>>(sim);
        }
        static std::string getName() { return "HeatCapacityVariableCustomPolynomial"; }
        std::string_view getNameLocal() const override
        {
            return "HeatCapacityVariableCustomPolynomial";
        };
        std::vector<std::string> getDependencies() const override
        {
            return this->ini_file_data.getStringVectorParameters("dependencies");
        };
};
 
// ================= Implementation =================
 
template <typename T>
HeatCapacityVariableCustomPolynomial<T>::HeatCapacityVariableCustomPolynomial(
    SimulationClassBase<T>* sim)
    : GenericSubmodule<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 HeatCapacityVariableCustomPolynomial<T>::setup(
    std::vector<std::shared_ptr<GenericSubmodule<T>>> all_submodules)
{
    m_coefficient_A = this->ini_file_data.getDoubleParameters("coefficient_A");
    m_coefficient_B = this->ini_file_data.getDoubleParameters("coefficient_B");
    m_coefficient_C = this->ini_file_data.getDoubleParameters("coefficient_C");
    m_coefficient_D = this->ini_file_data.getDoubleParameters("coefficient_D");
 
    return true;
}
 
template <typename T>
bool HeatCapacityVariableCustomPolynomial<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 HeatCapacityVariableCustomPolynomial<T>::calculateHeatCapacity()
{
    const std::valarray<T>& temperature{this->sim->getField("Temperature")};
    for (int i = 0; i < temperature.size(); ++i)
    {
        (*this->module_field)[i] = m_coefficient_A + m_coefficient_B * temperature[i]
                                   + m_coefficient_C * std::pow(temperature[i], 2.)
                                   + m_coefficient_D * std::pow(temperature[i], 3.);
    }
}
 
template <typename T>
bool HeatCapacityVariableCustomPolynomial<T>::preTimeStep()
{
    calculateHeatCapacity();
    return true;
}
 
template <typename T>
bool HeatCapacityVariableCustomPolynomial<T>::init()
{
    calculateHeatCapacity();
    return true;
}
 
template <typename T>
bool HeatCapacityVariableCustomPolynomial<T>::m_registered =
    ModuleFactory<T>::registerModule(getName(), createMethode);
 
#endif  // HEAT_CAPACITY_CONSTANT_CUSTOM_H