Mercurial > hg > jslab
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 } |