J. Duke | 319a3b9 | 2007-12-01 00:00:00 +0000 | [diff] [blame^] | 1 | /* |
| 2 | * Copyright 2003 Sun Microsystems, Inc. All Rights Reserved. |
| 3 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| 4 | * |
| 5 | * This code is free software; you can redistribute it and/or modify it |
| 6 | * under the terms of the GNU General Public License version 2 only, as |
| 7 | * published by the Free Software Foundation. Sun designates this |
| 8 | * particular file as subject to the "Classpath" exception as provided |
| 9 | * by Sun in the LICENSE file that accompanied this code. |
| 10 | * |
| 11 | * This code is distributed in the hope that it will be useful, but WITHOUT |
| 12 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| 13 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| 14 | * version 2 for more details (a copy is included in the LICENSE file that |
| 15 | * accompanied this code). |
| 16 | * |
| 17 | * You should have received a copy of the GNU General Public License version |
| 18 | * 2 along with this work; if not, write to the Free Software Foundation, |
| 19 | * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| 20 | * |
| 21 | * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, |
| 22 | * CA 95054 USA or visit www.sun.com if you need additional information or |
| 23 | * have any questions. |
| 24 | */ |
| 25 | |
| 26 | |
| 27 | /* |
| 28 | * FUNCTION |
| 29 | * mlib_ImageConvMxN - image convolution with edge condition |
| 30 | * |
| 31 | * SYNOPSIS |
| 32 | * mlib_status mlib_ImageConvMxN(mlib_image *dst, |
| 33 | * const mlib_image *src, |
| 34 | * const mlib_s32 *kernel, |
| 35 | * mlib_s32 m, |
| 36 | * mlib_s32 n, |
| 37 | * mlib_s32 dm, |
| 38 | * mlib_s32 dn, |
| 39 | * mlib_s32 scale, |
| 40 | * mlib_s32 cmask, |
| 41 | * mlib_edge edge) |
| 42 | * |
| 43 | * ARGUMENTS |
| 44 | * dst Pointer to destination image. |
| 45 | * src Pointer to source image. |
| 46 | * m Kernel width (m must be not less than 1). |
| 47 | * n Kernel height (n must be not less than 1). |
| 48 | * dm, dn Position of key element in convolution kernel. |
| 49 | * kernel Pointer to convolution kernel. |
| 50 | * scale The scaling factor to convert the input integer |
| 51 | * coefficients into floating-point coefficients: |
| 52 | * floating-point coefficient = integer coefficient * 2^(-scale) |
| 53 | * cmask Channel mask to indicate the channels to be convolved. |
| 54 | * Each bit of which represents a channel in the image. The |
| 55 | * channels corresponded to 1 bits are those to be processed. |
| 56 | * edge Type of edge condition. |
| 57 | * |
| 58 | * DESCRIPTION |
| 59 | * 2-D convolution, MxN kernel. |
| 60 | * |
| 61 | * The center of the source image is mapped to the center of the |
| 62 | * destination image. |
| 63 | * The unselected channels are not overwritten. If both src and dst have |
| 64 | * just one channel, cmask is ignored. |
| 65 | * |
| 66 | * The edge condition can be one of the following: |
| 67 | * MLIB_EDGE_DST_NO_WRITE (default) |
| 68 | * MLIB_EDGE_DST_FILL_ZERO |
| 69 | * MLIB_EDGE_DST_COPY_SRC |
| 70 | * MLIB_EDGE_SRC_EXTEND |
| 71 | * |
| 72 | * RESTRICTION |
| 73 | * The src and the dst must be the same type and have same number |
| 74 | * of channels (1, 2, 3, or 4). |
| 75 | * m >= 1, n >= 1, |
| 76 | * 0 <= dm < m, 0 <= dn < n. |
| 77 | * For data type MLIB_BYTE: 16 <= scale <= 31 (to be compatible with VIS version) |
| 78 | * For data type MLIB_USHORT: 17 <= scale <= 32 (to be compatible with VIS version) |
| 79 | * For data type MLIB_SHORT: 17 <= scale <= 32 (to be compatible with VIS version) |
| 80 | * For data type MLIB_INT: scale >= 0 |
| 81 | */ |
| 82 | |
| 83 | #include "mlib_image.h" |
| 84 | #include "mlib_ImageConv.h" |
| 85 | |
| 86 | /***************************************************************/ |
| 87 | static void mlib_ImageConvMxNMulAdd_S32(mlib_d64 *dst, |
| 88 | const mlib_s32 *src, |
| 89 | const mlib_d64 *dkernel, |
| 90 | mlib_s32 n, |
| 91 | mlib_s32 m, |
| 92 | mlib_s32 nch); |
| 93 | |
| 94 | static void mlib_ImageConvMxNMedian_S32(mlib_s32 *dst, |
| 95 | mlib_d64 *src, |
| 96 | mlib_s32 n, |
| 97 | mlib_s32 nch); |
| 98 | |
| 99 | static void mlib_ImageConvMxNS322S32_ext(mlib_s32 *dst, |
| 100 | const mlib_s32 *src, |
| 101 | mlib_s32 n, |
| 102 | mlib_s32 nch, |
| 103 | mlib_s32 dx_l, |
| 104 | mlib_s32 dx_r); |
| 105 | |
| 106 | /***************************************************************/ |
| 107 | #ifdef MLIB_USE_FTOI_CLAMPING |
| 108 | |
| 109 | #define CLAMP_S32(dst, src) \ |
| 110 | dst = (mlib_s32)(src) |
| 111 | |
| 112 | #else |
| 113 | |
| 114 | #define CLAMP_S32(dst, src) { \ |
| 115 | mlib_d64 s0 = (mlib_d64)(src); \ |
| 116 | if (s0 > (mlib_d64)MLIB_S32_MAX) s0 = (mlib_d64)MLIB_S32_MAX; \ |
| 117 | if (s0 < (mlib_d64)MLIB_S32_MIN) s0 = (mlib_d64)MLIB_S32_MIN; \ |
| 118 | dst = (mlib_s32)s0; \ |
| 119 | } |
| 120 | |
| 121 | #endif /* MLIB_USE_FTOI_CLAMPING */ |
| 122 | |
| 123 | /***************************************************************/ |
| 124 | void mlib_ImageConvMxNMulAdd_S32(mlib_d64 *dst, |
| 125 | const mlib_s32 *src, |
| 126 | const mlib_d64 *dkernel, |
| 127 | mlib_s32 n, |
| 128 | mlib_s32 m, |
| 129 | mlib_s32 nch) |
| 130 | { |
| 131 | mlib_d64 *dst1 = dst + 1; |
| 132 | mlib_s32 i, j; |
| 133 | |
| 134 | for (j = 0; j < m; j += 3, src += 3 * nch, dkernel += 3) { |
| 135 | const mlib_s32 *src2 = src + 2 * nch; |
| 136 | mlib_d64 hval0 = dkernel[0]; |
| 137 | mlib_d64 hval1 = dkernel[1]; |
| 138 | mlib_d64 hval2 = dkernel[2]; |
| 139 | mlib_d64 val0 = src[0]; |
| 140 | mlib_d64 val1 = src[nch]; |
| 141 | mlib_d64 dval = dst[0]; |
| 142 | |
| 143 | if (j == m - 2) { |
| 144 | hval2 = 0.f; |
| 145 | } |
| 146 | else if (j == m - 1) { |
| 147 | hval1 = 0.f; |
| 148 | hval2 = 0.f; |
| 149 | } |
| 150 | |
| 151 | #ifdef __SUNPRO_C |
| 152 | #pragma pipeloop(0) |
| 153 | #endif /* __SUNPRO_C */ |
| 154 | for (i = 0; i < n; i++) { |
| 155 | mlib_d64 dval0 = val0 * hval0 + dval; |
| 156 | mlib_d64 val2 = src2[i * nch]; |
| 157 | |
| 158 | dval = dst1[i]; |
| 159 | dval0 += val1 * hval1; |
| 160 | dval0 += val2 * hval2; |
| 161 | val0 = val1; |
| 162 | val1 = val2; |
| 163 | |
| 164 | dst[i] = dval0; |
| 165 | } |
| 166 | } |
| 167 | } |
| 168 | |
| 169 | /***************************************************************/ |
| 170 | void mlib_ImageConvMxNMedian_S32(mlib_s32 *dst, |
| 171 | mlib_d64 *src, |
| 172 | mlib_s32 n, |
| 173 | mlib_s32 nch) |
| 174 | { |
| 175 | mlib_s32 i; |
| 176 | |
| 177 | #ifdef __SUNPRO_C |
| 178 | #pragma pipeloop(0) |
| 179 | #endif /* __SUNPRO_C */ |
| 180 | for (i = 0; i < n; i++) { |
| 181 | mlib_s32 res; |
| 182 | |
| 183 | CLAMP_S32(res, src[i]); |
| 184 | src[i] = 0.5; |
| 185 | dst[i * nch] = res; |
| 186 | } |
| 187 | } |
| 188 | |
| 189 | /***************************************************************/ |
| 190 | void mlib_ImageConvMxNS322S32_ext(mlib_s32 *dst, |
| 191 | const mlib_s32 *src, |
| 192 | mlib_s32 n, |
| 193 | mlib_s32 nch, |
| 194 | mlib_s32 dx_l, |
| 195 | mlib_s32 dx_r) |
| 196 | { |
| 197 | mlib_s32 i; |
| 198 | mlib_d64 val = src[0]; |
| 199 | |
| 200 | for (i = 0; i < dx_l; i++) |
| 201 | dst[i] = (mlib_s32) val; |
| 202 | #ifdef __SUNPRO_C |
| 203 | #pragma pipeloop(0) |
| 204 | #endif /* __SUNPRO_C */ |
| 205 | for (; i < n - dx_r; i++) |
| 206 | dst[i] = src[nch * (i - dx_l)]; |
| 207 | val = dst[n - dx_r - 1]; |
| 208 | for (; i < n; i++) |
| 209 | dst[i] = (mlib_s32) val; |
| 210 | } |
| 211 | |
| 212 | /***************************************************************/ |
| 213 | mlib_status mlib_convMxNext_s32(mlib_image *dst, |
| 214 | const mlib_image *src, |
| 215 | const mlib_s32 *kernel, |
| 216 | mlib_s32 m, |
| 217 | mlib_s32 n, |
| 218 | mlib_s32 dx_l, |
| 219 | mlib_s32 dx_r, |
| 220 | mlib_s32 dy_t, |
| 221 | mlib_s32 dy_b, |
| 222 | mlib_s32 scale, |
| 223 | mlib_s32 cmask) |
| 224 | { |
| 225 | mlib_d64 dspace[1024], *dsa = dspace; |
| 226 | mlib_d64 akernel[256], *dkernel = akernel, fscale = 1.0; |
| 227 | mlib_s32 wid_e = mlib_ImageGetWidth(src); |
| 228 | mlib_d64 *dsh, *dsv; |
| 229 | mlib_s32 *isa; |
| 230 | mlib_s32 *da = mlib_ImageGetData(dst); |
| 231 | mlib_s32 *sa = mlib_ImageGetData(src); |
| 232 | mlib_s32 dlb = mlib_ImageGetStride(dst) >> 2; |
| 233 | mlib_s32 slb = mlib_ImageGetStride(src) >> 2; |
| 234 | mlib_s32 dw = mlib_ImageGetWidth(dst); |
| 235 | mlib_s32 dh = mlib_ImageGetHeight(dst); |
| 236 | mlib_s32 nch = mlib_ImageGetChannels(dst); |
| 237 | mlib_s32 i, j, j1, k, mn; |
| 238 | |
| 239 | /* internal buffer */ |
| 240 | |
| 241 | if (3 * wid_e + m > 1024) { |
| 242 | dsa = mlib_malloc((3 * wid_e + m) * sizeof(mlib_d64)); |
| 243 | |
| 244 | if (dsa == NULL) |
| 245 | return MLIB_FAILURE; |
| 246 | } |
| 247 | |
| 248 | isa = (mlib_s32 *) dsa; |
| 249 | |
| 250 | /* load kernel */ |
| 251 | mn = m * n; |
| 252 | |
| 253 | if (mn > 256) { |
| 254 | dkernel = mlib_malloc(mn * sizeof(mlib_d64)); |
| 255 | |
| 256 | if (dkernel == NULL) |
| 257 | return MLIB_FAILURE; |
| 258 | } |
| 259 | |
| 260 | while (scale > 30) { |
| 261 | fscale /= (1 << 30); |
| 262 | scale -= 30; |
| 263 | } |
| 264 | |
| 265 | fscale /= (1 << scale); |
| 266 | |
| 267 | for (i = 0; i < mn; i++) { |
| 268 | dkernel[i] = ((mlib_s32 *) kernel)[i] * fscale; |
| 269 | } |
| 270 | |
| 271 | dsh = dsa + dw + m; |
| 272 | dsv = dsh + dw; |
| 273 | |
| 274 | for (i = 0; i < dw; i++) { |
| 275 | dsh[i] = 0.5; |
| 276 | dsv[i] = 0.5; |
| 277 | } |
| 278 | |
| 279 | for (j = 0; j < dh; j++, da += dlb) { |
| 280 | for (k = 0; k < nch; k++) |
| 281 | if (cmask & (1 << (nch - 1 - k))) { |
| 282 | mlib_s32 *sa1 = sa + k; |
| 283 | mlib_d64 *dkernel1 = dkernel; |
| 284 | |
| 285 | for (j1 = 0; j1 < n; j1++, dkernel1 += m) { |
| 286 | mlib_ImageConvMxNS322S32_ext(isa, sa1, dw + m - 1, nch, dx_l, dx_r); |
| 287 | mlib_ImageConvMxNMulAdd_S32(dsh, isa, dkernel1, dw, m, 1); |
| 288 | |
| 289 | if ((j + j1 >= dy_t) && (j + j1 < dh + n - dy_b - 2)) |
| 290 | sa1 += slb; |
| 291 | } |
| 292 | |
| 293 | mlib_ImageConvMxNMedian_S32(da + k, dsh, dw, nch); |
| 294 | } |
| 295 | |
| 296 | if ((j >= dy_t) && (j < dh + n - dy_b - 2)) |
| 297 | sa += slb; |
| 298 | } |
| 299 | |
| 300 | if (dkernel != akernel) |
| 301 | mlib_free(dkernel); |
| 302 | if (dsa != dspace) |
| 303 | mlib_free(dsa); |
| 304 | return MLIB_SUCCESS; |
| 305 | } |
| 306 | |
| 307 | /***************************************************************/ |