/*----------------------------------------------------------------------------------
エノン・ハイレス方程式とヤコビ行列（線形化したエノン・ハイレス方程式）
Henon-Heiles equations and the Jacobian (linearized Henon-Heiles equations)
solve_henon-heiles_eqs.c
----------------------------------------------------------------------------------*/
#include "henon-heiles_lyap.h"
void henon_heiles_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 ww;
double df[DIM][DIM];
xx = v[0];
yy = v[1];
zz = v[2];
ww = v[3];
/*------------------------------------------------------------------*/
// エノン・ハイレス方程式
// Henon-Heiles equations
dvdt[0] = zz;
dvdt[1] = ww;
dvdt[2] = -xx - 2.0 * xx * yy;
dvdt[3] = -yy - xx * xx + yy * yy;
/*------------------------------------------------------------------*/
// ヤコビ行列
// Jacobian matrix
df[0][0] = 0.0;
df[0][1] = 0.0;
df[0][2] = 1.0;
df[0][3] = 0.0;
df[1][0] = 0.0;
df[1][1] = 0.0;
df[1][2] = 0.0;
df[1][3] = 1.0;
df[2][0] = -1.0 - 2.0 * yy;
df[2][1] = -2.0 * xx;
df[2][2] = 0.0;
df[2][3] = 0.0;
df[3][0] = -2.0 * xx;
df[3][1] = -1.0 + 2.0 * yy;
df[3][2] = 0.0;
df[3][3] = 0.0;
/*
｜ df[0][0] df[0][1] df[0][2] df[0][3] ｜
｜ ｜
｜ df[1][0] df[1][1] df[1][2] df[1][3] ｜
df = ｜ ｜
｜ df[2][0] df[2][1] df[2][2] df[2][3] ｜
｜ ｜
｜ df[3][0] df[3][1] df[3][2] df[3][3] ｜
*/
// ヤコビ行列を用いて微小変位ベクトルを計算
// calculate the micro-displacement vectors 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_henon_heiles_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
henon_heiles_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;
}
}
/*******************************************************/
henon_heiles_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;
}
}
/*******************************************************/
henon_heiles_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];
}
}
/*******************************************************/
henon_heiles_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;
}
}
/*******************************************************/
}