1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
|
########################################################################
##
## Copyright (C) 2008-2024 The Octave Project Developers
##
## See the file COPYRIGHT.md in the top-level directory of this
## distribution or <https://octave.org/copyright/>.
##
## 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
## <https://www.gnu.org/licenses/>.
##
########################################################################
## -*- texinfo -*-
## @deftypefn {} {} nargoutchk (@var{minargs}, @var{maxargs})
## @deftypefnx {} {@var{msgstr} =} nargoutchk (@var{minargs}, @var{maxargs}, @var{nargs})
## @deftypefnx {} {@var{msgstr} =} nargoutchk (@var{minargs}, @var{maxargs}, @var{nargs}, "string")
## @deftypefnx {} {@var{msgstruct} =} nargoutchk (@var{minargs}, @var{maxargs}, @var{nargs}, "struct")
## Check for correct number of output arguments.
##
## In the first form, return an error if the number of arguments is not between
## @var{minargs} and @var{maxargs}. Otherwise, do nothing. Note that this
## function evaluates the value of @code{nargout} on the caller so its value
## must have not been tampered with.
##
## Both @var{minargs} and @var{maxargs} must be numeric scalars. Zero, Inf,
## and negative are all valid, and they can have the same value.
##
## For backwards compatibility, the other forms return an appropriate error
## message string (or structure) if the number of outputs requested is
## invalid.
##
## This is useful for checking to that the number of output arguments supplied
## to a function is within an acceptable range.
## @seealso{narginchk, error, nargout, nargin}
## @end deftypefn
function msg = nargoutchk (minargs, maxargs, nargs, outtype)
## Before Matlab 2011b, nargoutchk would return an error message (just the
## message in a string). With 2011b, it no longer returns anything, it
## simply gives an error if the args number is incorrect.
## To try to keep compatibility with both versions, check nargout and nargin
## to guess if the caller is expecting a value (old syntax) or none
## (new syntax).
if (nargout == 1 && (nargin == 3 || nargin == 4))
if (minargs > maxargs)
error ("nargoutchk: MINARGS must be <= MAXARGS");
elseif (nargin == 3)
outtype = "string";
elseif (! any (strcmpi (outtype, {"string" "struct"})))
error ("nargoutchk: output type must be either string or struct");
elseif (! (isscalar (minargs) && isscalar (maxargs) && isscalar (nargs)))
error ("nargoutchk: MINARGS, MAXARGS, and NARGS must be scalars");
endif
msg = struct ("message", "", "identifier", "");
if (nargs < minargs)
msg.message = "not enough output arguments";
msg.identifier = "Octave:nargoutchk:not-enough-outputs";
elseif (nargs > maxargs)
msg.message = "too many output arguments";
msg.identifier = "Octave:nargoutchk:too-many-outputs";
endif
if (strcmpi (outtype, "string"))
msg = msg.message;
elseif (isempty (msg.message))
## Compatibility: Matlab returns a 0x1 empty struct when nargoutchk passes
msg = resize (msg, 0, 1);
endif
elseif (nargout == 0 && nargin == 2)
if (! isnumeric (minargs) || ! isscalar (minargs))
error ("nargoutchk: MINARGS must be a numeric scalar");
elseif (! isnumeric (maxargs) || ! isscalar (maxargs))
error ("nargoutchk: MAXARGS must be a numeric scalar");
elseif (minargs > maxargs)
error ("nargoutchk: MINARGS cannot be larger than MAXARGS");
endif
args = evalin ("caller", "nargout;");
if (args < minargs)
error ("nargoutchk: Not enough output arguments");
elseif (args > maxargs)
error ("nargoutchk: Too many output arguments");
endif
else
print_usage ();
endif
endfunction
%!shared stnul, stmin, stmax
%! stnul = resize (struct ("message", "", "identifier", ""), 0, 1);
%! stmin = struct ("message", "not enough output arguments",
%! "identifier", "Octave:nargoutchk:not-enough-outputs");
%! stmax = struct ("message", "too many output arguments",
%! "identifier", "Octave:nargoutchk:too-many-outputs");
%!assert (nargoutchk (0, 1, 0), "")
%!assert (nargoutchk (0, 1, 1), "")
%!assert (nargoutchk (1, 1, 0), "not enough output arguments")
%!assert (nargoutchk (0, 1, 2), "too many output arguments")
%!assert (nargoutchk (0, 1, 2, "string"), "too many output arguments")
## Struct outputs
%!assert (nargoutchk (0, 1, 0, "struct"), stnul)
%!assert (nargoutchk (0, 1, 1, "struct"), stnul)
%!assert (nargoutchk (1, 1, 0, "struct"), stmin)
%!assert (nargoutchk (0, 1, 2, "struct"), stmax)
|