Mercurial > octave
view libinterp/corefcn/colloc.cc @ 20909:03e4ddd49396
omit unnecessary nargout checks
* paramdemo.cc, __ichol__.cc, __ilu__.cc, balance.cc, dassl.cc,
eig.cc, ellipj.cc, filter.cc, givens.cc, hess.cc, lsode.cc, lu.cc,
mgorth.cc, nproc.cc, pr-output.cc, quad.cc, quadcc.cc, sylvester.cc,
urlwrite.cc, variables.cc, ccolamd.cc, colamd.cc, qr.cc: Don't call
print_usage based on value of nargout. If nargout doesn't alter
function behavior and returning extra values is inexpensive, simply
return them instead of checking nargout.
author | John W. Eaton <jwe@octave.org> |
---|---|
date | Wed, 16 Dec 2015 11:41:11 -0500 |
parents | f1b2a2dbc0e1 |
children | 6f0bd96f93c0 |
line wrap: on
line source
/* Copyright (C) 1996-2015 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 3 of the License, 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, see <http://www.gnu.org/licenses/>. */ #ifdef HAVE_CONFIG_H #include <config.h> #endif #include <string> #include "CollocWt.h" #include "lo-mappers.h" #include "defun.h" #include "error.h" #include "oct-obj.h" #include "utils.h" DEFUN (colloc, args, , "-*- texinfo -*-\n\ @deftypefn {} {[@var{r}, @var{amat}, @var{bmat}, @var{q}] =} colloc (@var{n}, \"left\", \"right\")\n\ Compute derivative and integral weight matrices for orthogonal collocation.\n\ \n\ Reference: @nospell{J. Villadsen}, @nospell{M. L. Michelsen},\n\ @cite{Solution of Differential Equation Models by Polynomial Approximation}.\n\ @end deftypefn") { octave_value_list retval; int nargin = args.length (); if (nargin < 1 || nargin > 3) print_usage (); if (! args(0).is_scalar_type ()) error ("colloc: N must be a scalar"); double tmp = args(0).double_value (); if (xisnan (tmp)) error ("colloc: N cannot be NaN"); octave_idx_type ncol = NINTbig (tmp); if (ncol < 0) error ("colloc: N must be positive"); octave_idx_type ntot = ncol; octave_idx_type left = 0; octave_idx_type right = 0; for (int i = 1; i < nargin; i++) { std::string s = args(i).xstring_value ("colloc: optional arguments must be strings"); 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: string argument must be \"left\" or \"right\""); } ntot += left + right; if (ntot < 1) error ("colloc: the total number of roots must be positive"); CollocWt wts (ncol, left, right); ColumnVector r = wts.roots (); Matrix A = wts.first (); Matrix B = wts.second (); ColumnVector q = wts.quad_weights (); retval = ovl (r, A, B, q); return retval; }