eq.hpp

/*
 * eq.hpp
 *
 *  Created on: Oct 5, 2015
 *      Author: i-bird
 */

#ifndef OPENFPM_NUMERICS_SRC_FINITEDIFFERENCE_EQ_HPP_
#define OPENFPM_NUMERICS_SRC_FINITEDIFFERENCE_EQ_HPP_

#define EQS_FIELD 0
#define EQS_POS 1

//#define PERIODIC true
//#define NON_PERIODIC false

#include "util/util_num.hpp"
#include "Matrix/SparseMatrix.hpp"

template<typename expr1,typename expr2,typename Sys_eqs>
class Eq
{
    template<unsigned int ord=EQS_FIELD> static void value(const grid_key_dx<Sys_eqs::dims> & pos)
    {
        if (EQS_FIELD)
            value_f(pos);
        else
            value_s(pos);
    }

    static openfpm::vector<cval<typename Sys_eqs::stype>> value_s(grid_key_dx<Sys_eqs::dims> & it)
    {
        return expr1::value_s(it) - expr2::value_s(it);
    }

    static void value_f(grid_key_dx<Sys_eqs::dims> & it)
    {
        return expr1::value_s(it) - expr2::value_s(it);
    }
};


// spatial position + value

template<unsigned int dim,typename T>
struct pos_val
{
    pos_val()
    {
        value = 0.0;
    }

    grid_key_dx<dim> pos;
    T value;
};

template<unsigned int f, typename Sys_eqs>
class Field
{
    typedef typename stub_or_real<Sys_eqs,Sys_eqs::dims,typename Sys_eqs::stype,typename Sys_eqs::b_grid::decomposition::extended_type::extended_type>::type map_grid;

public:

    static void value(const map_grid & g_map, grid_dist_key_dx<Sys_eqs::dims> & kmap, const grid_sm<Sys_eqs::dims,void> & gs, typename Sys_eqs::stype (& spacing )[Sys_eqs::dims] , std::unordered_map<long int,typename Sys_eqs::stype > & cols, typename Sys_eqs::stype coeff)
    {
        cols[g_map.template get<0>(kmap)*Sys_eqs::nvar + f] += coeff;
    }

    static grid_key_dx<Sys_eqs::dims> position(grid_key_dx<Sys_eqs::dims> & pos, const grid_sm<Sys_eqs::dims,void> & gs, const comb<Sys_eqs::dims> (& s_pos)[Sys_eqs::nvar])
    {
            return grid_key_dx<Sys_eqs::dims>(s_pos[f]);
    }
};

class ConstField
{

};

inline size_t mat_factor(size_t nvar, size_t sz, const size_t ord)
{
    return nvar;
}

#include "mul.hpp"
#include "Average.hpp"
#include "Derivative.hpp"
#include "sum.hpp"
#include "Laplacian.hpp"

#endif /* OPENFPM_NUMERICS_SRC_FINITEDIFFERENCE_EQ_HPP_ */