5410
|
1 ## Copyright (C) 1997 Kurt Hornik |
|
2 ## |
|
3 ## This file is part of Octave. |
|
4 ## |
|
5 ## Octave is free software; you can redistribute it and/or modify it |
|
6 ## under the terms of the GNU General Public License as published by |
7016
|
7 ## the Free Software Foundation; either version 3 of the License, or (at |
|
8 ## your option) any later version. |
5410
|
9 ## |
|
10 ## Octave is distributed in the hope that it will be useful, but |
|
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of |
|
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
13 ## General Public License for more details. |
|
14 ## |
|
15 ## You should have received a copy of the GNU General Public License |
7016
|
16 ## along with Octave; see the file COPYING. If not, see |
|
17 ## <http://www.gnu.org/licenses/>. |
5410
|
18 |
|
19 ## -*- texinfo -*- |
7002
|
20 ## @deftypefn {Function File} {} hygernd (@var{t}, @var{m}, @var{n}, @var{r}, @var{c}) |
|
21 ## @deftypefnx {Function File} {} hygernd (@var{t}, @var{m}, @var{n}, @var{sz}) |
7009
|
22 ## @deftypefnx {Function File} {} hygernd (@var{t}, @var{m}, @var{n}) |
5691
|
23 ## Return an @var{r} by @var{c} matrix of random samples from the |
7002
|
24 ## hypergeometric distribution with parameters @var{t}, @var{m}, |
5410
|
25 ## and @var{n}. |
|
26 ## |
7002
|
27 ## The parameters @var{t}, @var{m}, and @var{n} must positive integers |
5410
|
28 ## with @var{m} and @var{n} not greater than @var{t}. |
7009
|
29 ## |
|
30 ## The parameter @var{sz} must be scalar or a vector of matrix |
|
31 ## dimensions. If @var{sz} is scalar, then a @var{sz} by @var{sz} |
|
32 ## matrix of random samples is generated. |
5410
|
33 ## @end deftypefn |
|
34 |
7002
|
35 function rnd = hygernd (t, m, n, r, c) |
5410
|
36 |
|
37 if (nargin == 5) |
|
38 if (! (isscalar (r) && (r > 0) && (r == round (r)))) |
5411
|
39 error ("hygernd: r must be a positive integer"); |
5410
|
40 endif |
|
41 if (! (isscalar (c) && (c > 0) && (c == round (c)))) |
5411
|
42 error ("hygernd: c must be a positive integer"); |
5410
|
43 endif |
|
44 sz = [r, c]; |
|
45 elseif (nargin == 4) |
7009
|
46 if (isvector (r) && all (r > 0) && all (r == round (r))) |
|
47 if (isscalar (r)) |
|
48 sz = [r, r]; |
|
49 else |
|
50 sz = r(:)'; |
|
51 endif |
5410
|
52 else |
5411
|
53 error ("hygernd: r must be a vector of positive integers"); |
5410
|
54 endif |
7009
|
55 elseif (nargin != 3) |
6046
|
56 print_usage (); |
5410
|
57 endif |
|
58 |
7009
|
59 if (! isscalar (t) || ! isscalar (m) || ! isscalar (n)) |
|
60 [retval, t, m, n] = common_size (t, m, n); |
|
61 if (retval > 0) |
|
62 error ("hygernd: t, m and n must be of common size or scalar"); |
|
63 endif |
|
64 if (nargin > 3) |
|
65 if (any (sz != size (t))) |
|
66 error ("hygernd: t, m and n must have the same size as implied by r and c or must be scalar"); |
|
67 endif |
|
68 else |
|
69 sz = size (t); |
|
70 endif |
|
71 elseif (nargin == 3) |
|
72 sz = 1; |
5410
|
73 endif |
|
74 |
7009
|
75 ## NaN elements |
|
76 ne = (! (t >= 0) | ! (m >= 0) | ! (n > 0) | ! (t == round (t)) | ! (m == round (m)) | ! (n == round (n)) | ! (m <= t) | ! (n <= t)); |
|
77 |
|
78 if (! isscalar (t)) |
|
79 rnd = zeros (sz); |
|
80 rnd(ne) = NaN; |
|
81 rn = rand (sz); |
|
82 for i = find (! ne) |
|
83 v = 0 : n(i); |
|
84 p = hygepdf (v, t(i), m(i), n(i)); |
|
85 rnd(i) = v(lookup (cumsum (p(1 : end-1)) / sum (p), rn(i)) + 1); |
|
86 endfor |
5410
|
87 else |
7009
|
88 if (ne) |
|
89 rnd = NaN * ones (sz); |
|
90 else |
|
91 v = 0:n; |
|
92 p = hygepdf (v, t, m, n); |
|
93 rnd = v(lookup (cumsum (p(1:end-1)) / sum (p), rand (sz)) + 1); |
|
94 endif |
5410
|
95 endif |
|
96 |
|
97 endfunction |