Mercurial > octave-nkf
annotate scripts/general/bicubic.m @ 8507:cadc73247d65
style fixes
| author | John W. Eaton <jwe@octave.org> |
|---|---|
| date | Tue, 13 Jan 2009 14:08:36 -0500 |
| parents | bc982528de11 |
| children | 8463d1a2e544 |
| rev | line source |
|---|---|
| 7017 | 1 ## Copyright (C) 2005, 2006, 2007 Hoxide Ma |
| 5837 | 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. | |
| 5837 | 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/>. | |
| 5837 | 18 |
| 19 ## -*- texinfo -*- | |
| 6702 | 20 ## @deftypefn {Function File} {@var{zi}=} bicubic (@var{x}, @var{y}, @var{z}, @var{xi}, @var{yi}, @var{extrapval}) |
| 5837 | 21 ## |
| 6653 | 22 ## Return a matrix @var{zi} corresponding to the bicubic |
| 5838 | 23 ## interpolations at @var{xi} and @var{yi} of the data supplied |
| 6702 | 24 ## as @var{x}, @var{y} and @var{z}. Points outside the grid are set |
| 25 ## to @var{extrapval} | |
| 5837 | 26 ## |
| 6702 | 27 ## See @url{http://wiki.woodpecker.org.cn/moin/Octave/Bicubic} |
| 28 ## for further information. | |
| 5838 | 29 ## @seealso{interp2} |
| 5837 | 30 ## @end deftypefn |
| 31 | |
| 32 ## Bicubic interpolation method. | |
| 33 ## Author: Hoxide Ma <hoxide_dirac@yahoo.com.cn> | |
| 34 | |
| 6702 | 35 function F = bicubic (X, Y, Z, XI, YI, extrapval, spline_alpha) |
| 5837 | 36 |
| 6702 | 37 if (nargin < 1 || nargin > 7) |
| 5837 | 38 print_usage (); |
| 39 endif | |
| 40 | |
| 6702 | 41 if (nargin == 7 && isscalar(spline_alpha)) |
| 5837 | 42 a = spline_alpha |
| 43 else | |
| 44 a = 0.5; | |
| 45 endif | |
| 46 | |
| 6702 | 47 if (nargin < 6) |
| 48 extrapval = NaN; | |
| 49 endif | |
| 50 | |
|
7795
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
51 if (isa (X, "single") || isa (Y, "single") || isa (Z, "single") || |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
52 isa (XI, "single") || isa (YI, "single")) |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
53 myeps = eps("single"); |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
54 else |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
55 myeps = eps; |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
56 endif |
|
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
57 |
| 5838 | 58 if (nargin <= 2) |
| 59 ## bicubic (Z) or bicubic (Z, 2) | |
| 5837 | 60 if (nargin == 1) |
| 61 n = 1; | |
| 62 else | |
| 63 n = Y; | |
| 64 endif | |
| 5838 | 65 Z = X; |
| 5837 | 66 X = []; |
| 5838 | 67 [rz, cz] = size (Z); |
| 68 s = linspace (1, cz, (cz-1)*pow2(n)+1); | |
| 69 t = linspace (1, rz, (rz-1)*pow2(n)+1); | |
| 5837 | 70 elseif (nargin == 3) |
| 5838 | 71 if (! isvector (X) || ! isvector (Y)) |
| 5837 | 72 error ("XI and YI must be vector"); |
| 73 endif | |
| 74 s = Y; | |
| 75 t = Z; | |
| 76 Z = X; | |
| 5838 | 77 [rz, cz] = size (Z); |
| 5837 | 78 elseif (nargin == 5 || nargin == 6) |
| 79 [rz, cz] = size (Z) ; | |
| 80 if (isvector (X) && isvector (Y)) | |
| 81 if (rz != length (Y) || cz != length (X)) | |
| 82 error ("length of X and Y must match the size of Z"); | |
| 83 endif | |
| 6157 | 84 elseif (size_equal (X, Y) && size_equal (X, Z)) |
| 5837 | 85 X = X(1,:); |
| 86 Y = Y(:,1); | |
| 87 else | |
| 88 error ("X, Y and Z must be martrices of same size"); | |
| 89 endif | |
| 90 | |
| 8506 | 91 ## Mark values outside the lookup table. |
| 5837 | 92 xfirst_ind = find (XI < X(1)); |
| 93 xlast_ind = find (XI > X(cz)); | |
| 94 yfirst_ind = find (YI < Y(1)); | |
| 95 ylast_ind = find (YI > Y(rz)); | |
| 8506 | 96 ## Set value outside the table preliminary to min max index. |
| 5837 | 97 XI(xfirst_ind) = X(1); |
| 98 XI(xlast_ind) = X(cz); | |
| 99 YI(yfirst_ind) = Y(1); | |
| 100 YI(ylast_ind) = Y(rz); | |
| 101 | |
| 102 | |
| 5838 | 103 X = reshape (X, 1, cz); |
|
7795
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
104 X(cz) *= 1 + sign (X(cz))*myeps; |
| 5837 | 105 if (X(cz) == 0) |
|
7795
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
106 X(cz) = myeps; |
| 5837 | 107 endif; |
| 5838 | 108 XI = reshape (XI, 1, length (XI)); |
| 109 [m, i] = sort ([X, XI]); | |
| 110 o = cumsum (i <= cz); | |
| 111 xidx = o(find (i > cz)); | |
| 5837 | 112 |
| 5838 | 113 Y = reshape (Y, rz, 1); |
|
7795
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
114 Y(rz) *= 1 + sign (Y(rz))*myeps; |
| 5837 | 115 if (Y(rz) == 0) |
|
7795
df9519e9990c
Handle single precision eps values
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
116 Y(rz) = myeps; |
| 5837 | 117 endif; |
| 5838 | 118 YI = reshape (YI, length (YI), 1); |
| 119 [m, i] = sort ([Y; YI]); | |
| 120 o = cumsum (i <= rz); | |
| 8507 | 121 yidx = o([find(i > rz)]); |
| 5837 | 122 |
| 8506 | 123 ## Set s and t used follow codes. |
| 5837 | 124 s = xidx + ((XI .- X(xidx))./(X(xidx+1) .- X(xidx))); |
| 125 t = yidx + ((YI - Y(yidx))./(Y(yidx+1) - Y(yidx))); | |
| 126 else | |
| 127 print_usage (); | |
| 128 endif | |
| 129 | |
| 130 if (rz < 3 || cz < 3) | |
| 131 error ("Z at least a 3 by 3 matrices"); | |
| 132 endif | |
| 133 | |
| 5838 | 134 inds = floor (s); |
| 135 d = find (s == cz); | |
| 136 s = s - floor (s); | |
| 5837 | 137 inds(d) = cz-1; |
| 138 s(d) = 1.0; | |
| 139 | |
| 140 d = []; | |
| 5838 | 141 indt = floor (t); |
| 142 d = find (t == rz); | |
| 143 t = t - floor (t); | |
| 5837 | 144 indt(d) = rz-1; |
| 145 t(d) = 1.0; | |
| 146 d = []; | |
| 147 | |
| 5838 | 148 p = zeros (size (Z) + 2); |
| 5837 | 149 p(2:rz+1,2:cz+1) = Z; |
| 8507 | 150 p(1,:) = (6*(1-a))*p(2,:) - 3*p(3,:) + (6*a-2)*p(4,:); |
| 151 p(rz+2,:) = (6*(1-a))*p(rz+1,:) - 3*p(rz,:) + (6*a-2)*p(rz-1,:); | |
| 152 p(:,1) = (6*(1-a))*p(:,2) - 3*p(:,3) + (6*a-2)*p(:,4); | |
| 153 p(:,cz+2) = (6*(1-a))*p(:,cz+1) - 3*p(:,cz) + (6*a-2)*p(:,cz-1); | |
| 5837 | 154 |
| 8506 | 155 ## Calculte the C1(t) C2(t) C3(t) C4(t) and C1(s) C2(s) C3(s) C4(s). |
| 8507 | 156 t2 = t.*t; |
| 157 t3 = t2.*t; | |
| 5837 | 158 |
| 8507 | 159 ct0 = -a .* t3 + (2 * a) .* t2 - a .* t ; # -a G0 |
| 5837 | 160 ct1 = (2-a) .* t3 + (-3+a) .* t2 + 1 ; # F0 - a G1 |
| 161 ct2 = (a-2) .* t3 + (-2 *a + 3) .* t2 + a .* t ; # F1 + a G0 | |
| 162 ct3 = a .* t3 - a .* t2; # a G1 | |
| 8507 | 163 t = []; t2 = []; t3 = []; |
| 5837 | 164 |
| 8507 | 165 s2 = s.*s; |
| 166 s3 = s2.*s; | |
| 5837 | 167 |
| 8507 | 168 cs0 = -a .* s3 + (2 * a) .* s2 - a .*s ; # -a G0 |
| 5837 | 169 cs1 = (2-a) .* s3 + (-3 + a) .* s2 + 1 ; # F0 - a G1 |
| 170 cs2 = (a-2) .* s3 + (-2 *a + 3) .* s2 + a .*s ; # F1 + a G0 | |
| 171 cs3 = a .* s3 - a .* s2; # a G1 | |
| 8507 | 172 s = []; s2 = []; s3 = []; |
| 5837 | 173 |
| 174 cs0 = cs0([1,1,1,1],:); | |
| 175 cs1 = cs1([1,1,1,1],:); | |
| 176 cs2 = cs2([1,1,1,1],:); | |
| 177 cs3 = cs3([1,1,1,1],:); | |
| 178 | |
| 5838 | 179 lent = length (ct0); |
| 180 lens = length (cs0); | |
| 181 F = zeros (lent, lens); | |
| 5837 | 182 |
| 183 for i = 1:lent | |
| 184 it = indt(i); | |
| 185 int = [it, it+1, it+2, it+3]; | |
| 8507 | 186 F(i,:) = ([ct0(i),ct1(i),ct2(i),ct3(i)] |
| 187 * (p(int,inds) .* cs0 + p(int,inds+1) .* cs1 | |
| 188 + p(int,inds+2) .* cs2 + p(int,inds+3) .* cs3)); | |
| 5837 | 189 endfor |
| 190 | |
| 8506 | 191 ## Set points outside the table to extrapval. |
| 5837 | 192 if (! (isempty (xfirst_ind) && isempty (xlast_ind))) |
| 6702 | 193 F(:, [xfirst_ind, xlast_ind]) = extrapval; |
| 5837 | 194 endif |
| 195 if (! (isempty (yfirst_ind) && isempty (ylast_ind))) | |
| 6702 | 196 F([yfirst_ind; ylast_ind], :) = extrapval; |
| 5837 | 197 endif |
| 198 | |
| 199 endfunction | |
| 200 | |
| 201 %!demo | |
| 202 %! A=[13,-1,12;5,4,3;1,6,2]; | |
| 203 %! x=[0,1,4]+10; y=[-10,-9,-8]; | |
| 204 %! xi=linspace(min(x),max(x),17); | |
| 6702 | 205 %! yi=linspace(min(y),max(y),26)'; |
| 5837 | 206 %! mesh(xi,yi,bicubic(x,y,A,xi,yi)); |
| 207 %! [x,y] = meshgrid(x,y); | |
| 208 %! hold on; plot3(x(:),y(:),A(:),"b*"); hold off; |