Mercurial > hg > octave-lyh
diff src/DLD-FUNCTIONS/colloc.cc @ 2928:295f037b4b3e
[project @ 1997-05-05 05:32:33 by jwe]
author | jwe |
---|---|
date | Mon, 05 May 1997 05:33:54 +0000 |
parents | |
children | 38de16594cb4 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/src/DLD-FUNCTIONS/colloc.cc @@ -0,0 +1,141 @@ +/* + +Copyright (C) 1996, 1997 John W. Eaton + +This file is part of Octave. + +Octave is free software; you can redistribute it and/or modify it +under the terms of the GNU General Public License as published by the +Free Software Foundation; either version 2, or (at your option) any +later version. + +Octave is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +for more details. + +You should have received a copy of the GNU General Public License +along with Octave; see the file COPYING. If not, write to the Free +Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + +*/ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <string> + +#include "CollocWt.h" +#include "lo-mappers.h" + +#include "defun-dld.h" +#include "error.h" +#include "help.h" +#include "oct-obj.h" +#include "utils.h" + +DEFUN_DLD (colloc, args, , + "[R, A, B, Q] = colloc (N [, \"left\"] [, \"right\"]): collocation weights") +{ + octave_value_list retval; + + int nargin = args.length (); + + if (nargin < 1 || nargin > 3) + { + print_usage ("colloc"); + return retval; + } + + if (! args(0).is_scalar_type ()) + { + error ("colloc: first argument must be a scalar"); + return retval; + } + + double tmp = args(0).double_value (); + + if (error_state) + return retval; + + if (xisnan (tmp)) + { + error ("colloc: NaN is invalid as NCOL"); + return retval; + } + + int ncol = NINT (tmp); + if (ncol < 0) + { + error ("colloc: first argument must be non-negative"); + return retval; + } + + int ntot = ncol; + int left = 0; + int right = 0; + + for (int i = 1; i < nargin; i++) + { + if (args(i).is_defined ()) + { + if (! args(i).is_string ()) + { + error ("colloc: expecting string argument"); + return retval; + } + + string s = args(i).string_value (); + + if ((s.length () == 1 && (s[0] == 'R' || s[0] == 'r')) + || s == "right") + { + right = 1; + } + else if ((s.length () == 1 && (s[0] == 'L' || s[0] == 'l')) + || s == "left") + { + left = 1; + } + else + { + error ("colloc: unrecognized argument"); + return retval; + } + } + else + { + error ("colloc: unexpected empty argument"); + return retval; + } + } + + ntot += left + right; + if (ntot < 1) + { + error ("colloc: the total number of roots must be positive"); + return retval; + } + + CollocWt wts (ncol, left, right); + + ColumnVector r = wts.roots (); + Matrix A = wts.first (); + Matrix B = wts.second (); + ColumnVector q = wts.quad_weights (); + + retval(3) = q; + retval(2) = B; + retval(1) = A; + retval(0) = r; + + return retval; +} + +/* +;;; Local Variables: *** +;;; mode: C++ *** +;;; End: *** +*/ +