ComplexIntlabMethod.java

/*
 * Created on 2008/01/31
 * Copyright (C) 2008 Koga Laboratory. All rights reserved.
 *
 */
package org.mklab.cga.linear;

import java.util.ArrayList;
import java.util.List;

import org.mklab.cga.interval.matrix.IntervalComplexNumericalMatrix;
import org.mklab.cga.interval.matrix.IntervalRealNumericalMatrix;
import org.mklab.cga.interval.scalar.IntervalComplexNumericalScalar;
import org.mklab.cga.interval.scalar.IntervalRealNumericalScalar;
import org.mklab.cga.util.IntervalUtil;
import org.mklab.nfc.matrix.ComplexNumericalMatrix;
import org.mklab.nfc.matrix.NormType;
import org.mklab.nfc.matrix.RealNumericalMatrix;
import org.mklab.nfc.scalar.ComplexNumericalScalar;
import org.mklab.nfc.scalar.NumericalScalar;
import org.mklab.nfc.scalar.RealNumericalScalar;
import org.mklab.nfc.util.RoundMode;
import org.mklab.nfc.util.RoundModeManager;


/**
 * 線形方程式の精度保証付き解をINTLABに実装されている方法で求めるクラスです。
 * 
 * {@link NumericalScalar}を用いているので多倍長精度に対応しています。
 * 
 * @author yano
 * @version $Revision: 1.7 $, 2008/01/31
 * @param <RIS> 実区間スカラーの型
 * @param <RIM> 実区間行列の型
 * @param <CIS> 複素区間スカラーの型
 * @param <CIM> 複素区間行列の型
 * @param <RS> 実スカラーの型
 * @param <RM> 実行列の型
 * @param <CS> 複素スカラーの型
 * @param <CM> 複素行列の型
 */
public class ComplexIntlabMethod<RIS extends IntervalRealNumericalScalar<RIS,RIM,CIS,CIM,RS,RM,CS,CM>, RIM extends IntervalRealNumericalMatrix<RIS,RIM,CIS,CIM,RS,RM,CS,CM>, CIS extends IntervalComplexNumericalScalar<RIS,RIM,CIS,CIM,RS,RM,CS,CM>, CIM extends IntervalComplexNumericalMatrix<RIS,RIM,CIS,CIM,RS,RM,CS,CM>, RS extends RealNumericalScalar<RS,RM,CS,CM>, RM extends RealNumericalMatrix<RS,RM,CS,CM>, CS extends ComplexNumericalScalar<RS,RM,CS,CM>, CM extends ComplexNumericalMatrix<RS,RM,CS,CM>>
    implements LinearEquationVerifier<CIS, CIM, CS, CM> {

  /** 区間係数行列 */
  private CIM intA;

  /** 区間右辺ベクトル */
  private CIM intB;

  /** 精度保証付き解 */
  private CIM solution;

  /** 計算途中に使うパラメータ。何なのかよくわからない。 */
  private static int INTERVAL_RESIDUAL = 1;

  /** 計算途中に使うパラメータ。何なのかよくわからない。 */
  private static boolean INTERVAL_FIRST_SECOND_STAGE = true;

  //  /**
  //   * 新しく生成された<code>NumericalMatrixElementIntlabMethod</code>オブジェクトを初期化します。
  //   */
  //  public IntlabMethod() {
  //    //
  //  }
  //
  //  /**
  //   * 新しく生成された<code>NumericalMatrixElementIntlabMethod</code>オブジェクトを初期化します。
  //   * 
  //   * @param A 係数行列
  //   * @param b 右辺係数ベクトル
  //   */
  //  public IntlabMethod(final M A, final M b) {
  //    this.intA = IntervalMatrixFactory.toInterval(A);
  //    this.intB = IntervalMatrixFactory.toInterval(b);
  //  }
  //
  //  /**
  //   * 新しく生成された<code>NumericalMatrixElementIntlabMethod</code>オブジェクトを初期化します。
  //   * 
  //   * @param A 係数行列
  //   * @param b 右辺係数区間ベクトル
  //   */
  //  public IntlabMethod(final M A, final IM b) {
  //    this.intA = IntervalMatrixFactory.toInterval(A);
  //    this.intB = b.clone();
  //  }
  //
  //  /**
  //   * 新しく生成された<code>NumericalMatrixElementIntlabMethod</code>オブジェクトを初期化します。
  //   * 
  //   * @param A 係数区間行列
  //   * @param b 右辺係数ベクトル
  //   */
  //  public IntlabMethod(final IM A, final M b) {
  //    this.intA = A.clone();
  //    this.intB = IntervalMatrixFactory.toInterval(b);
  //  }
  //

  //  /**
  //   * {@inheritDoc}
  //   */
  //  public void solve(final M A, final M b) {
  //    this.intA = IntervalMatrixFactory.toInterval(A);
  //    this.intB = IntervalMatrixFactory.toInterval(b);
  //    solve();
  //  }
  //
  //  /**
  //   * {@inheritDoc}
  //   */
  //  public void solve(final M A, final IM b) {
  //    this.intA = IntervalMatrixFactory.toInterval(A);
  //    this.intB = b.clone();
  //    solve();
  //  }
  //
  //  /**
  //   * {@inheritDoc}
  //   */
  //  public void solve(final IM A, final M b) {
  //    this.intA = A.clone();
  //    this.intB = IntervalMatrixFactory.toInterval(b);
  //    solve();
  //  }

  /**
   * 新しく生成された<code>NumericalMatrixElementIntlabMethod</code>オブジェクトを初期化します。
   * 
   * @param A 係数区間行列
   * @param b 右辺係数区間ベクトル
   */
  public ComplexIntlabMethod(final CIM A, final CIM b) {
    this.intA = A;
    this.intB = b;
  }

  /**
   * {@inheritDoc}
   */
  public void solve() {
//    this.intA = A;
//    this.intB = b;

    CM midA = this.intA.getMiddle();
    CM midb = this.intB.getMiddle();

    //    String message;
    //    if (((message = Abcdchk.abcdchk(midA, midb)).length()) > 0) {
    //      throw new IllegalArgumentException(message);
    //    }

    int k = midA.getColumnSize();
    int n = midA.getColumnSize();

    CM R = midA.inverse();
    CM xs = R.multiply(midb);// 近似解の計算。

    // Improve residual calculation
    // Interval iteration
    List<CIM> ires_Z_RA = intervalIteration(this.intA, this.intB, R, xs);
    CIM ires = ires_Z_RA.get(0);
    CIM Z = ires_Z_RA.get(1);
    CIM RA = ires_Z_RA.get(2);

    this.solution = stage12(ires, Z, RA, R, xs, n, k);
  }

  //  /**
  //   * @see org.mklab.cga.linear.LinearEquationVerifier#getSolution()
  //   */
  //  public LinearEquationVerifiedSolution<E> getSolution() {
  //    return new LinearEquationVerifiedSolution(this.solution);
  //  }

  /**
   * Creates interval matrix
   * 
   * @param matrix matrix
   * @return interval matrix
   */
  private CIM createInterval(CM matrix) {
    return this.intA.create(matrix);
  }

  /**
   * Creates interval with middle and radius.
   * 
   * @param mid middle
   * @param rad radius
   * @return interval matrxi
   */
  private CIM createMidRad(CM mid, CM rad) {
    RoundModeManager manager = RoundModeManager.getManager();
    RoundMode oldRoundMode = manager.getRoundMode();

    manager.setRoundMode(RoundMode.ROUND_DOWN);
    CM inf = mid.subtract(rad);

    manager.setRoundMode(RoundMode.ROUND_UP);
    CM sup = mid.add(rad);

    CIM ans = this.intA.createInfSup(inf, sup);

    manager.setRoundMode(oldRoundMode);

    return ans;
  }

  /**
   * @param A A
   * @param b b
   * @param R R
   * @param xs xs
   * @return ??
   */
  private List<CIM> intervalIteration(CIM A, CIM b, CM R, CM xs) {
    CIM ires = b.subtract(A.multiply(createInterval(xs)));
    CIM Z = createInterval(R).multiply(ires);
    CIM RA = createInterval(R).multiply(A);

    List<CIM> ans = new ArrayList<>();
    ans.add(ires);
    ans.add(Z);
    ans.add(RA);
    return ans;
  }

  /**
   * @param midA Aの中心
   * @param midb Bの中心
   * @param R R
   * @param xs xs
   * @return ??
   */
  private List<Object> improveResidual(CM midA, CM midb, CM R, CM xs) {
    boolean improvedresidual = (INTERVAL_RESIDUAL == 1) | ((INTERVAL_RESIDUAL == 2) & (midb.getColumnSize() > 1));

    CM localxs = xs;

    // IMPROVE RESIDUAL CALCULATION
    if (improvedresidual) {
      // TODO 表現可能な最大数を返すメソッドがいる?(これよりは小さい値のはず？？ 精度が落ちるから駄目?)
      CS resnorm = midA.getElement(1).getInfinity(); //transformFrom(new MPFloat(1).divide(new MPFloat(0)));

      int i = 0;

      while (i < 15) {
        i = i + 1;
        CS resnormold = resnorm.clone();
        CM res = R.multiply(ResidualImprover.lssresidual(midA, xs, midb));

        // TODO ここで計算失敗（０に値が近いから？？）
        resnorm = (res).norm(NormType.TWO);// 最大特異値
        if (resnorm.isLessThan(resnormold)) {
          localxs = localxs.add(res);
        }

        // TODO 精度が落ちそう
        if (resnorm.isGreaterThanOrEquals(resnormold.divide(10))) {// 0の剰余に注意
          break;
        }
      }

    } else if (INTERVAL_RESIDUAL == 2) {// quadruple precision residual
      // calculation??
      // TODO 最大値を返すメソッドがいる？(この値より小さいはず？ 精度に影響がある？？)
      //S resnorm = (midA).getElement(1).transformFrom(Double.POSITIVE_INFINITY);
      CS resnorm = (midA).getElement(1);
      int i = 0;

      while (i < 15) {
        i = i + 1;

        CS resnormold = resnorm.clone();
        CM res = R.multiply(midb.unaryMinus().add(midA.multiply(localxs)));
        resnorm = (res).norm(NormType.TWO);

        if (resnorm.isLessThan(resnormold)) {
          localxs = localxs.subtract(res);
        }

        // TODO 精度に影響しない？
        if (resnorm.isLessThanOrEquals(resnormold.multiply(1e-1))) {// beware of zero residual
          break;
        }
      }

    } else {
      localxs = localxs.add(R.multiply(midb.subtract(midA.multiply(localxs))));
    }

    List<Object> list = new ArrayList<>();
    list.add(localxs);
    list.add(Boolean.valueOf(improvedresidual));
    return list;
  }

  /**
   * intbalMethod内で使用する計算。
   * 
   * @param ires ires
   * @param Z Z
   * @param RA RA
   * @param R R
   * @param xs xs
   * @param n n
   * @param k k
   * @return result
   */
  private CIM stage12(CIM ires, CIM Z, CIM RA, CM R, CM xs, int n, int k) {
    RoundModeManager manager = RoundModeManager.getManager();
    RoundMode oldRoundMode = manager.getRoundMode();

    manager.setRoundMode(RoundMode.ROUND_NEAR);

    // FIRST STAGE
    CIM X = null;
    if (INTERVAL_FIRST_SECOND_STAGE) {
      CS unit = R.getElement(1, 1).abs();

      // TODO 精度にいぞんする？これより小さい値もありうるよね？
      CS realmin = unit.create(Double.MIN_VALUE); // Z.getMiddle().getElement(1).transformFrom(Double.MIN_VALUE);
      CIM C = createInterval(R.createUnit(n)).subtract(RA);

      CIS type = this.intA.getElement(1, 1);
      CIM Y = Z;
      // TODO 具体的なMPFloatを使わない(精度を落とさずにできるの？)
      CIS minusPlusOneInterval = type.createInfSup(unit.createUnit().unaryMinus(), unit.createUnit());
      CIS disk = type.createInfSup(realmin.multiply(10).unaryMinus(), realmin.multiply(10));
      CM radius = Y.getSupremum().subtract(Y.getInfimum()).divide(2);
      CIM yRadius = createInterval(radius.divide(10));
      CIM E = yRadius.multiply(minusPlusOneInterval).addElementWise(disk);

      int localk = 0;
      final int kmax = 7;
      boolean ready = false;

      while ((!ready) & (localk < kmax) & (!Y.isNanElementWise().anyTrue())) {
        localk = localk + 1;
        X = Y.add(E);
        Y = Z.add(C.multiply(X));
        ready = IntervalUtil.in0(Y, X).allTrue();
      }

      if (ready) {// SUCCESS FIRST STAGE
        X = Y.add(createInterval(xs));
        manager.setRoundMode(oldRoundMode);
        return X;
      }
    }

    // SECOND STAGE STARTS
    CIM b = createInterval(R).multiply(ires);
    CIM dRA = RA.diagonalToVector();
    CM A = RA.comparisonMatrix();
    CM B = A.inverse();
    CM v = (B.multiply(A.createOnes(n, 1))).absElementWise();

    manager.setRoundMode(RoundMode.ROUND_DOWN);

    CM u = A.multiply(v);

    if ((u).minColumnWise().compareElementWise(".>", A.createOnes(1, u.getColumnSize())).allTrue()) { //$NON-NLS-1$
      CM dAc = A.diagonalToVector();
      A = A.multiply(B).subtract(A.createUnit(n));

      manager.setRoundMode(RoundMode.ROUND_UP);

      CM w = (A.unaryMinus().divideElementWise(u.multiply(A.createOnes(1, n)))).maxColumnWise().transpose();
      CM dlow = v.multiplyElementWise(w.conjugateTranspose()).subtract(B.diagonalToVector());
      dlow = (dlow.unaryMinus()).maxElementWise(A.createZero(dlow.getRowSize(), dlow.getColumnSize()));

      B = B.add(v.multiply(w));
      u = B.multiply(b.abssElementWise());
      CM d = B.diagonalToVector();
      CM alpha = dAc.add(A.createOnes(d.getRowSize(), d.getColumnSize()).unaryMinus().divideElementWise(d));

      int localk = b.getColumnSize();

      CIM Ixs = createInterval(xs);

      if (localk == 1) {
        CM beta = u.divideElementWise(dlow).subtract(b.abssElementWise());
        CM mid1 = A.createZero(beta.getRowSize(), beta.getColumnSize());
        CM mid2 = A.createZero(alpha.getRowSize(), alpha.getColumnSize());
        X = Ixs.add((b.add(createMidRad(mid1, beta))).divideElementWise((dRA.add(createMidRad(mid2, alpha)))));
      } else {
        // d and dRA adapted for multiple r.h.s.
        v = A.createOnes(1, localk);
        CM beta = u.divideElementWise(d.multiply(v)).subtract(b.abssElementWise());
        CM mid1 = A.createZero(beta.getRowSize(), beta.getColumnSize());
        CM mid2 = A.createZero(alpha.getRowSize(), alpha.getColumnSize());
        X = Ixs.add(b.add(createMidRad(mid1, beta)).divideElementWise((dRA.add(createMidRad(mid2, alpha)).multiply(createInterval(v)))));

      }
    }

    manager.setRoundMode(oldRoundMode);
    return X;
  }

  /**
   * {@inheritDoc}
   */
  public CIM getSolution() {
    return this.solution;
  }

  /**
   * Returns A.
   * 
   * @return A
   */
  protected CIM getA() {
    return this.intA;
  }
  
  /**
   * Returns B.
   * 
   * @return B
   */
  protected CIM getB() {
    return this.intB;
  }
}