Model_MD1D.h

Go to the documentation of this file.

#ifndef MODEL_MD1D_H
#define MODEL_MD1D_H
 
#include "kids/Kernel.h"
#include "kids/Kernel_Random.h"
#include "kids/hash_fnv1a.h"
#include "kids/macro_utils.h"
 
namespace PROJECT_NS {
 
 
class Model_HO final : public Kernel {
   public:
    virtual const std::string getName() { return "Model_HO"; }
 
    virtual int getType() const { return utils::hash(FUNCTION_NAME); }
 
   private:
    int     size;
    double *x, *p;
    double *x_init, *p_init;
    double *f;
    double *m, *w;
 
    double beta;
 
    virtual void setInputParam_impl(Param *PM) {
        size = PM->get_int("N", LOC());             //
        beta = PM->get_double("beta", LOC(), 1.0);  //
    }
 
    virtual void setInputDataSet_impl(DataSet *DS) {
        // external
        x = DS->def_real("integrator.x", size);
        p = DS->def_real("integrator.p", size);
        f = DS->def_real("integrator.f", size);
 
        x_init = DS->def_real("init.x", size);
        p_init = DS->def_real("init.p", size);
 
        // shared
        w = DS->def_real("model.w", size);
        m = DS->def_real("model.mass", size);
        for (int i = 0; i < size; ++i) m[i] = 1, w[i] = 1;
    }
 
    virtual Status &initializeKernel_impl(Status &stat) {
        for (int i = 0; i < size; ++i) {
            double Qi_div_beta = 0.5f * w[i] / std::tanh(0.5f * beta * w[i]);
            double xi_sigma    = std::sqrt(Qi_div_beta / (m[i] * w[i] * w[i]));
            double pi_sigma    = std::sqrt(Qi_div_beta * m[i]);
 
            double u1, u2;
            Kernel_Random::rand_gaussian(&u1);
            Kernel_Random::rand_gaussian(&u2);
            x_init[i] = u1 * xi_sigma;
            p_init[i] = u2 * pi_sigma;
        }
        // copy to external
        for (int i = 0; i < size; ++i) {
            x[i] = x_init[i];
            p[i] = p_init[i];
        }
        return stat;
    }
 
    virtual Status &executeKernel_impl(Status &stat) {
        for (int i = 0; i < size; ++i) f[i] = m[i] * w[i] * w[i] * x[i];
        return stat;
    }
};
 
class MODEL_MD1D final : public Kernel {
   public:
    virtual const std::string getName() { return "Model_HO"; }
 
    virtual int getType() const { return utils::hash(FUNCTION_NAME); }
 
   private:
    int     size;
    double *x, *f;
    double *m, *w;
 
    virtual void setInputParam_impl(Param *PM) {
        size = PM->get_int("N", LOC());  //
    }
 
    virtual void setInputDataSet_impl(DataSet *DS) {
        // external
        x = DS->def_real("integrator.x", size);
        f = DS->def_real("integrator.f", size);
        // shared
        w = DS->def_real("model.w", size);
        m = DS->def_real("model.mass", size);
        for (int i = 0; i < size; ++i) m[i] = 1, w[i] = 1;
    }
 
    virtual Status &executeKernel_impl(Status &stat) {
        for (int i = 0; i < size; ++i) f[i] = m[i] * w[i] * w[i] * x[i];
        return stat;
    }
};
};  // namespace PROJECT_NS
 
#endif  // MODEL_MD1D_H