diff options
author | Joe Hunkeler <jhunkeler@gmail.com> | 2015-08-11 16:51:37 -0400 |
---|---|---|
committer | Joe Hunkeler <jhunkeler@gmail.com> | 2015-08-11 16:51:37 -0400 |
commit | 40e5a5811c6ffce9b0974e93cdd927cbcf60c157 (patch) | |
tree | 4464880c571602d54f6ae114729bf62a89518057 /math/gsurfit/gssolver.x | |
download | iraf-osx-40e5a5811c6ffce9b0974e93cdd927cbcf60c157.tar.gz |
Repatch (from linux) of OSX IRAF
Diffstat (limited to 'math/gsurfit/gssolver.x')
-rw-r--r-- | math/gsurfit/gssolver.x | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/math/gsurfit/gssolver.x b/math/gsurfit/gssolver.x new file mode 100644 index 00000000..5135298a --- /dev/null +++ b/math/gsurfit/gssolver.x @@ -0,0 +1,84 @@ +# Copyright(c) 1986 Association of Universities for Research in Astronomy Inc. + +include <math/gsurfit.h> +include "gsurfitdef.h" + +# GSSOLVE -- Solve the matrix normal equations of the form ca = b for a, +# where c is a symmetric, positive semi-definite, banded matrix with +# GS_NXCOEFF(sf) * GS_NYCOEFF(sf) rows and a and b are GS_NXCOEFF(sf) * +# GS_NYCOEFF(sf)-vectors. +# Initially c is stored in the matrix MATRIX +# and b is stored in VECTOR. +# The Cholesky factorization of MATRIX is calculated and stored in CHOFAC. +# Finally the coefficients are calculated by forward and back substitution +# and stored in COEFF. +# +# This version has two options: fit all the coefficients or fix the +# the zeroth coefficient at a specified reference point. + +procedure gssolve (sf, ier) + +pointer sf # curve descriptor +int ier # ier = OK, everything OK + # ier = SINGULAR, matrix is singular, 1 or more + # coefficients are 0. + # ier = NO_DEG_FREEDOM, too few points to solve matrix + +int i, ncoeff +pointer sp, vector, matrix + +real gseval() + +begin + if (IS_INDEFR(GS_XREF(sf)) || IS_INDEFR(GS_YREF(sf)) || + IS_INDEFR(GS_ZREF(sf))) + ncoeff = GS_NCOEFF(sf) + else + ncoeff = GS_NCOEFF(sf) - 1 + + # test for number of degrees of freedom + ier = OK + i = GS_NPTS(sf) - ncoeff + if (i < 0) { + ier = NO_DEG_FREEDOM + return + } + + if (ncoeff == GS_NCOEFF(sf)) { + vector = GS_VECTOR(sf) + matrix = GS_MATRIX(sf) + } else { + # allocate working space for the reduced vector and matrix + call smark (sp) + call salloc (vector, ncoeff, TY_REAL) + call salloc (matrix, ncoeff*ncoeff, TY_REAL) + + # eliminate the terms from the vector and matrix + call amovr (VECTOR(GS_VECTOR(sf)+1), Memr[vector], ncoeff) + do i = 0, ncoeff-1 + call amovr (MATRIX(GS_MATRIX(sf)+(i+1)*GS_NCOEFF(sf)), + Memr[matrix+i*ncoeff], ncoeff) + } + + # solve for the coefficients. + switch (GS_TYPE(sf)) { + case GS_LEGENDRE, GS_CHEBYSHEV, GS_POLYNOMIAL: + + # calculate the Cholesky factorization of the data matrix + call rgschofac (MATRIX(matrix), ncoeff, ncoeff, + CHOFAC(GS_CHOFAC(sf)), ier) + + # solve for the coefficients by forward and back substitution + call rgschoslv (CHOFAC(GS_CHOFAC(sf)), ncoeff, ncoeff, + VECTOR(vector), COEFF(GS_COEFF(sf)+GS_NCOEFF(sf)-ncoeff)) + + default: + call error (0, "GSSOLVE: Illegal surface type.") + } + + if (ncoeff != GS_NCOEFF(sf)) { + COEFF(GS_COEFF(sf)) = GS_ZREF(sf) - + gseval (sf, GS_XREF(sf), GS_YREF(sf)) + call sfree (sp) + } +end |