/*------------------------------------------------------------------------------
スプロットレスラー方程式とヤコビ行列（線形化したレスラー方程式）
Sprott equations and the Jacobian (linearized Roessler equations)
solve_sprott_eqs.c
------------------------------------------------------------------------------*/
#include "sprott_lyap.h"
void sprott_eqs( double v[DIM], double dfu[DIM][DIM], double dvdt[DIM], double df_k[DIM][DIM] )
{
int i;
int j;
int k;
double d2;
double xx;
double yy;
double zz;
double df[DIM][DIM];
xx = v[0];
yy = v[1];
zz = v[2];
/*------------------------------------------------------------------*/
// スプロット方程式
// Sprott equations
dvdt[0] = yy;
dvdt[1] = zz;
dvdt[2] = -xx + yy * yy - A * zz;
/*------------------------------------------------------------------*/
// ヤコビ行列
// Jacobian matrix
df[0][0] = 0.0;
df[0][1] = 1.0;
df[0][2] = 0.0;
df[1][0] = 0.0;
df[1][1] = 0.0;
df[1][2] = 1.0;
df[2][0] = -1.0;
df[2][1] = 2.0 * yy;
df[2][2] = -A;
/*
｜ df[0][0] df[0][1] df[0][2] ｜
｜ ｜
df = ｜ df[1][0] df[1][1] df[1][2] ｜
｜ ｜
｜ df[2][0] df[2][1] df[2][2] ｜
*/
// ヤコビ行列を用いて微小変位ベクトルを計算
// calculate the micro-displacement vector using the Jacobian matrix
for ( i = 0; i < DIM; i++ ) {
for ( j = 0; j < DIM; j++ ) {
d2 = 0.0;
for ( k = 0; k < DIM; k++ ) {
d2 += df[i][k] * dfu[k][j];
}
df_k[i][j] = d2;
}
}
}
/*-----------------------------------------------------------------------------------------*/
void solve_sprott_eqs_using_RK4( double v[DIM], double u[DIM][DIM] )
{
int i;
int j;
double vv[DIM];
double v_k1[DIM], v_k2[DIM], v_k3[DIM], v_k4[DIM];
double dfu[DIM][DIM];
double df_k1[DIM][DIM], df_k2[DIM][DIM], df_k3[DIM][DIM], df_k4[DIM][DIM];
/*******************************************************/
for (i = 0; i < DIM; i++) {
vv[i] = v[i];
for (j = 0; j < DIM; j++) {
dfu[i][j] = u[i][j];
}
}
/*******************************************************/
// vvとdfuを渡し、v_k1とdf_k1の計算値を得る。
// pass vv and dfu, and get the calculated values of v_k1 and df_k1
sprott_eqs( vv, dfu, v_k1, df_k1 );
for (i = 0; i < DIM; i++) {
vv[i] = v[i] + DELTA_T * v_k1[i]/(double)2.0;
}
for (i = 0; i < DIM; i++) {
for (j = 0; j < DIM; j++) {
dfu[i][j] = u[i][j] + DELTA_T * df_k1[i][j]/(double)2.0;
}
}
/*******************************************************/
sprott_eqs( vv, dfu, v_k2, df_k2 );
for (i = 0; i < DIM; i++) {
vv[i] = v[i] + DELTA_T * v_k2[i]/(double)2.0;
}
for (i = 0; i < DIM; i++) {
for (j = 0; j < DIM; j++) {
dfu[i][j] = u[i][j] + DELTA_T * df_k2[i][j]/(double)2.0;
}
}
/*******************************************************/
sprott_eqs( vv, dfu, v_k3, df_k3 );
for (i = 0; i < DIM; i++) {
vv[i] = v[i] + DELTA_T * v_k3[i];
}
for (i = 0; i < DIM; i++) {
for (j = 0; j < DIM; j++) {
dfu[i][j] = u[i][j] + DELTA_T * df_k3[i][j];
}
}
/*******************************************************/
sprott_eqs( vv, dfu, v_k4, df_k4 );
for (i = 0; i < DIM; i++) {
v[i] += DELTA_T * (v_k1[i] + 2.0 * v_k2[i] + 2.0 * v_k3[i] + v_k4[i])/(double)6.0;
}
for (i = 0; i < DIM; i++) {
for (j = 0; j < DIM; j++) {
u[i][j] += DELTA_T * (df_k1[i][j] + 2.0 * df_k2[i][j] + 2.0 * df_k3[i][j] + df_k4[i][j])/(double)6.0;
}
}
/*******************************************************/
}