package mypackage.hamilton;
/** 1次 Symplectic 法
 * Time-stamp: "2001-04-23 12:52:17 tkenichi" 
 *
 * Hamiltonian H(q,p) = V(p) + U(q) の形のときにだけ
 * 適用できる
 */

public class Symplectic1st extends HamiltonSolver{
    /** 独立変数 */
    private double t = 0;
    /** 従属変数(空間座標) */
    private double q[];
    /** 従属変数(運動量) */
    private double p[];

    /** フィールドの代入と初期値の設定 */
    public Symplectic1st(Hamilton hamilton,double step){
	this.hamilton = hamilton;
	this.step = step;
	t = hamilton.t0;
	q = new double[hamilton.dim];
	p = new double[hamilton.dim];
	for(int i=0;i<hamilton.dim;i++){
	    q[i] = hamilton.q0[i];      // 初期値の代入
	    p[i] = hamilton.p0[i];
	}
    }

    /** 次の逐次計算 */
    public void next(){
	for(int i=0;i<q.length;i++){
	    q[i] += step * hamilton.hp(i,q,p);
	}
	for(int i=0;i<q.length;i++){
	    p[i] -= step * hamilton.hq(i,q,p);
	}
	t += step;
    }

    /** 現在の独立変数の値を得る */
    public double getTime(){
	return t;
    }

    /** 現在の空間座標の値を得る */
    public double getQValue(int index){
	return q[index];
    }

    /** 現在の運動量の値を得る */
    public double getPValue(int index){
	return p[index];
    }

    /** q の index 番目の値を設定する */
    public void setQValue(int index,double value){
	q[index] = value;
    }
    
    /** p の index 番目の値を設定する */
    public void setPValue(int index,double value){
	p[index] = value;
    }
}