X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?p=CreaPhase.git;a=blobdiff_plain;f=octave_packages%2Fm%2Fstatistics%2Fbase%2Fmeansq.m;fp=octave_packages%2Fm%2Fstatistics%2Fbase%2Fmeansq.m;h=08875894d4c05ce08555ed02224fc5afc6ba71fd;hp=0000000000000000000000000000000000000000;hb=1c0469ada9531828709108a4882a751d2816994a;hpb=63de9f36673d49121015e3695f2c336ea92bc278 diff --git a/octave_packages/m/statistics/base/meansq.m b/octave_packages/m/statistics/base/meansq.m new file mode 100644 index 0000000..0887589 --- /dev/null +++ b/octave_packages/m/statistics/base/meansq.m @@ -0,0 +1,89 @@ +## Copyright (C) 1995-2012 Kurt Hornik +## Copyright (C) 2009 Jaroslav Hajek +## +## 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 +## . + +## -*- texinfo -*- +## @deftypefn {Function File} {} meansq (@var{x}) +## @deftypefnx {Function File} {} meansq (@var{x}, @var{dim}) +## Compute the mean square of the elements of the vector @var{x}. +## @tex +## $$ +## {\rm meansq} (x) = {\sum_{i=1}^N {x_i}^2 \over N} +## $$ +## where $\bar{x}$ is the mean value of $x$. +## @end tex +## @ifnottex +## +## @example +## @group +## std (x) = 1/N SUM_i x(i)^2 +## @end group +## @end example +## +## @end ifnottex +## For matrix arguments, return a row vector containing the mean square +## of each column. +## +## If the optional argument @var{dim} is given, operate along this dimension. +## @seealso{var, std, moment} +## @end deftypefn + +## Author: KH +## Description: Compute mean square + +function y = meansq (x, dim) + + if (nargin != 1 && nargin != 2) + print_usage (); + endif + + if (! (isnumeric (x) || islogical (x))) + error ("mean: X must be a numeric vector or matrix"); + endif + + nd = ndims (x); + sz = size (x); + if (nargin < 2) + ## Find the first non-singleton dimension. + (dim = find (sz > 1, 1)) || (dim = 1); + else + if (!(isscalar (dim) && dim == fix (dim)) + || !(1 <= dim && dim <= nd)) + error ("mean: DIM must be an integer and a valid dimension"); + endif + endif + + y = sumsq (x, dim) / sz(dim); + +endfunction + + +%!assert(meansq (1:5), 11); +%!assert(meansq (single(1:5)), single(11)); +%!assert(meansq (magic (4)), [94.5, 92.5, 92.5, 94.5]); +%!assert(meansq (magic (4), 2), [109.5; 77.5; 77.5; 109.5]); + +%% Test input validation +%!error meansq () +%!error meansq (1, 2, 3) +%!error meansq (['A'; 'B']); +%!error meansq (1, ones(2,2)) +%!error meansq (1, 1.5) +%!error meansq (1, 0) +%!error meansq (1, 3) +