annotate src/samer/maths/opt/ConjGrad.java @ 8:5e3cbbf173aa tip

Reorganise some more
author samer
date Fri, 05 Apr 2019 22:41:58 +0100
parents bf79fb79ee13
children
rev   line source
samer@0 1 /*
samer@0 2 * Copyright (c) 2000, Samer Abdallah, King's College London.
samer@0 3 * All rights reserved.
samer@0 4 *
samer@0 5 * This software is provided AS iS and WITHOUT ANY WARRANTY;
samer@0 6 * without even the implied warranty of MERCHANTABILITY or
samer@0 7 * FITNESS FOR A PARTICULAR PURPOSE.
samer@0 8 */
samer@0 9
samer@0 10 package samer.maths.opt;
samer@0 11 import samer.maths.*;
samer@0 12 import samer.core.*;
samer@0 13
samer@0 14 public class ConjGrad
samer@0 15 {
samer@0 16 double[] dg;
samer@0 17 double gg, dgg;
samer@0 18 double invhess;
samer@0 19 State S;
samer@0 20
samer@0 21 public ConjGrad(State s)
samer@0 22 {
samer@0 23 S = s;
samer@0 24 dg = new double[S.n];
samer@0 25 invhess = 1;
samer@0 26 }
samer@0 27
samer@0 28 public void init() {
samer@0 29 Mathx.negate(S.P1.g,S.h);
samer@0 30 Mathx.mul(S.h,invhess);
samer@0 31 S.normh=Util.maxabs(S.h);
samer@0 32 }
samer@0 33
samer@0 34 public void update()
samer@0 35 {
samer@0 36 Mathx.sub(dg,S.P2.g,S.P1.g);
samer@0 37 gg = Mathx.norm2(S.P1.g);
samer@0 38 dgg = Mathx.dot(dg,S.P2.g);
samer@0 39
samer@0 40 Mathx.mul(S.h,dgg/gg);
samer@0 41 Mathx.sub(S.h,S.P2.g);
samer@0 42 Mathx.mul(S.h,invhess);
samer@0 43 S.normh=Util.maxabs(S.h);
samer@0 44
samer@0 45 double s2=Mathx.dot(S.P2.g,S.h);
samer@0 46 if (s2>0) {
samer@0 47 // Shell.trace("steepest descent");
samer@0 48 Mathx.negate(S.P2.g,S.h);
samer@0 49 S.normh=Util.maxabs(S.h);
samer@0 50 s2=Mathx.dot(S.P2.g,S.h);
samer@0 51 }
samer@0 52 S.P2.s=s2;
samer@0 53 }
samer@0 54 }