Path: blob/master/src/java.desktop/share/native/libmlib_image/mlib_ImageConv_32nw.c
41152 views
/*1* Copyright (c) 2000, 2020, Oracle and/or its affiliates. All rights reserved.2* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.3*4* This code is free software; you can redistribute it and/or modify it5* under the terms of the GNU General Public License version 2 only, as6* published by the Free Software Foundation. Oracle designates this7* particular file as subject to the "Classpath" exception as provided8* by Oracle in the LICENSE file that accompanied this code.9*10* This code is distributed in the hope that it will be useful, but WITHOUT11* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or12* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License13* version 2 for more details (a copy is included in the LICENSE file that14* accompanied this code).15*16* You should have received a copy of the GNU General Public License version17* 2 along with this work; if not, write to the Free Software Foundation,18* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.19*20* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA21* or visit www.oracle.com if you need additional information or have any22* questions.23*/242526/*27* FUNCTION28* Internal functions for mlib_ImageConv* on S32 type and29* MLIB_EDGE_DST_NO_WRITE mask30*31*/3233#include "mlib_image.h"34#include "mlib_ImageConv.h"3536/***************************************************************/37#define CACHE_SIZE (64*1024)3839/***************************************************************/40#define CONV_FUNC(KERN) mlib_conv##KERN##nw_s324142/***************************************************************/43#ifndef MLIB_USE_FTOI_CLAMPING4445#define CLAMP_S32(dst, src) \46if (src > (mlib_d64)MLIB_S32_MAX) src = (mlib_d64)MLIB_S32_MAX; \47if (src < (mlib_d64)MLIB_S32_MIN) src = (mlib_d64)MLIB_S32_MIN; \48dst = (mlib_s32)src4950#else5152#define CLAMP_S32(dst, src) dst = (mlib_s32)(src)5354#endif /* MLIB_USE_FTOI_CLAMPING */5556/***************************************************************/57#define GET_SRC_DST_PARAMETERS(type) \58mlib_s32 hgt = mlib_ImageGetHeight(src); \59mlib_s32 wid = mlib_ImageGetWidth(src); \60mlib_s32 sll = mlib_ImageGetStride(src) / sizeof(type); \61mlib_s32 dll = mlib_ImageGetStride(dst) / sizeof(type); \62type* adr_src = mlib_ImageGetData(src); \63type* adr_dst = mlib_ImageGetData(dst); \64mlib_s32 chan1 = mlib_ImageGetChannels(src)65/* mlib_s32 chan2 = chan1 + chan1 */6667/***************************************************************/68#define DEF_VARS(type) \69GET_SRC_DST_PARAMETERS(type); \70type *sl, *sp, *sl1, *dl, *dp; \71mlib_d64 *pbuff = buff, *buff0, *buff1, *buff2, *buffT; \72mlib_s32 i, j, c; \73mlib_d64 scalef, d0, d17475/***************************************************************/76#define DEF_VARS_MxN(type) \77GET_SRC_DST_PARAMETERS(type); \78type *sl, *sp = NULL, *dl, *dp = NULL; \79mlib_d64 *pbuff = buff; \80mlib_s32 i, j, c8182/***************************************************************/83#define FTYPE mlib_d6484#define DTYPE mlib_s328586#define BUFF_SIZE 16008788static mlib_status mlib_ImageConv1xN(mlib_image *dst,89const mlib_image *src,90const mlib_d64 *k,91mlib_s32 n,92mlib_s32 dn,93mlib_s32 cmask)94{95FTYPE buff[BUFF_SIZE];96mlib_s32 off, kh;97const FTYPE *pk;98FTYPE k0, k1, k2, k3, d0, d1;99FTYPE p0, p1, p2, p3, p4;100DTYPE *sl_c, *dl_c, *sl0;101mlib_s32 l, hsize, max_hsize;102DEF_VARS_MxN(DTYPE);103104hgt -= (n - 1);105adr_dst += dn*dll;106107max_hsize = (CACHE_SIZE/sizeof(DTYPE))/sll;108109if (!max_hsize) max_hsize = 1;110111if (max_hsize > BUFF_SIZE) {112pbuff = mlib_malloc(sizeof(FTYPE)*max_hsize);113}114115sl_c = adr_src;116dl_c = adr_dst;117118for (l = 0; l < hgt; l += hsize) {119hsize = hgt - l;120121if (hsize > max_hsize) hsize = max_hsize;122123for (c = 0; c < chan1; c++) {124if (!(cmask & (1 << (chan1 - 1 - c)))) continue;125126sl = sl_c + c;127dl = dl_c + c;128129for (j = 0; j < hsize; j++) pbuff[j] = 0.0;130131for (i = 0; i < wid; i++) {132sl0 = sl;133134for (off = 0; off < (n - 4); off += 4) {135pk = k + off;136sp = sl0;137138k0 = pk[0]; k1 = pk[1]; k2 = pk[2]; k3 = pk[3];139p2 = sp[0]; p3 = sp[sll]; p4 = sp[2*sll];140sp += 3*sll;141142for (j = 0; j < hsize; j += 2) {143p0 = p2; p1 = p3; p2 = p4;144p3 = sp[0];145p4 = sp[sll];146147pbuff[j ] += p0*k0 + p1*k1 + p2*k2 + p3*k3;148pbuff[j + 1] += p1*k0 + p2*k1 + p3*k2 + p4*k3;149150sp += 2*sll;151}152153sl0 += 4*sll;154}155156pk = k + off;157sp = sl0;158159k0 = pk[0]; k1 = pk[1]; k2 = pk[2]; k3 = pk[3];160p2 = sp[0]; p3 = sp[sll]; p4 = sp[2*sll];161162dp = dl;163kh = n - off;164165if (kh == 4) {166sp += 3*sll;167168for (j = 0; j <= (hsize - 2); j += 2) {169p0 = p2; p1 = p3; p2 = p4;170p3 = sp[0];171p4 = sp[sll];172173d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + pbuff[j];174d1 = p1*k0 + p2*k1 + p3*k2 + p4*k3 + pbuff[j + 1];175CLAMP_S32(dp[0 ], d0);176CLAMP_S32(dp[dll], d1);177178pbuff[j] = 0;179pbuff[j + 1] = 0;180181sp += 2*sll;182dp += 2*dll;183}184185if (j < hsize) {186p0 = p2; p1 = p3; p2 = p4;187p3 = sp[0];188189d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + pbuff[j];190CLAMP_S32(dp[0], d0);191192pbuff[j] = 0;193}194195} else if (kh == 3) {196sp += 2*sll;197198for (j = 0; j <= (hsize - 2); j += 2) {199p0 = p2; p1 = p3;200p2 = sp[0];201p3 = sp[sll];202203d0 = p0*k0 + p1*k1 + p2*k2 + pbuff[j];204d1 = p1*k0 + p2*k1 + p3*k2 + pbuff[j + 1];205CLAMP_S32(dp[0 ], d0);206CLAMP_S32(dp[dll], d1);207208pbuff[j] = 0;209pbuff[j + 1] = 0;210211sp += 2*sll;212dp += 2*dll;213}214215if (j < hsize) {216p0 = p2; p1 = p3;217p2 = sp[0];218219d0 = p0*k0 + p1*k1 + p2*k2 + pbuff[j];220CLAMP_S32(dp[0], d0);221222pbuff[j] = 0;223}224225} else if (kh == 2) {226sp += sll;227228for (j = 0; j <= (hsize - 2); j += 2) {229p0 = p2;230p1 = sp[0];231p2 = sp[sll];232233d0 = p0*k0 + p1*k1 + pbuff[j];234d1 = p1*k0 + p2*k1 + pbuff[j + 1];235CLAMP_S32(dp[0 ], d0);236CLAMP_S32(dp[dll], d1);237238pbuff[j] = 0;239pbuff[j + 1] = 0;240241sp += 2*sll;242dp += 2*dll;243}244245if (j < hsize) {246p0 = p2;247p1 = sp[0];248249d0 = p0*k0 + p1*k1 + pbuff[j];250CLAMP_S32(dp[0], d0);251252pbuff[j] = 0;253}254255} else /* if (kh == 1) */ {256for (j = 0; j < hsize; j++) {257p0 = sp[0];258259d0 = p0*k0 + pbuff[j];260CLAMP_S32(dp[0], d0);261262pbuff[j] = 0;263264sp += sll;265dp += dll;266}267}268269sl += chan1;270dl += chan1;271}272}273274sl_c += max_hsize*sll;275dl_c += max_hsize*dll;276}277278if (pbuff != buff) mlib_free(pbuff);279280return MLIB_SUCCESS;281}282283/***************************************************************/284#define MAX_KER 7285286#define MAX_N 15287288#undef BUFF_SIZE289#define BUFF_SIZE 1500290291mlib_status CONV_FUNC(MxN)(mlib_image *dst,292const mlib_image *src,293const mlib_s32 *kernel,294mlib_s32 m,295mlib_s32 n,296mlib_s32 dm,297mlib_s32 dn,298mlib_s32 scale,299mlib_s32 cmask)300{301mlib_d64 buff[BUFF_SIZE], *buffs_arr[2*(MAX_N + 1)];302mlib_d64 **buffs = buffs_arr, *buffd;303mlib_d64 akernel[256], *k = akernel, fscale = 1.0;304mlib_s32 l, off, kw, bsize, buff_ind, mn;305mlib_d64 d0, d1;306mlib_d64 k0, k1, k2, k3, k4, k5, k6;307mlib_d64 p0, p1, p2, p3, p4, p5, p6, p7;308DEF_VARS_MxN(mlib_s32);309mlib_s32 chan2 = chan1 + chan1;310311mlib_status status = MLIB_SUCCESS;312313if (scale > 30) {314fscale *= 1.0/(1 << 30);315scale -= 30;316}317318fscale /= (1 << scale);319320mn = m*n;321322if (mn > 256) {323k = mlib_malloc(mn*sizeof(mlib_d64));324325if (k == NULL) return MLIB_FAILURE;326}327328for (i = 0; i < mn; i++) {329k[i] = kernel[i]*fscale;330}331332if (m == 1) {333status = mlib_ImageConv1xN(dst, src, k, n, dn, cmask);334FREE_AND_RETURN_STATUS;335}336337bsize = (n + 2)*wid;338339if ((bsize > BUFF_SIZE) || (n > MAX_N)) {340pbuff = mlib_malloc(sizeof(mlib_d64)*bsize + sizeof(mlib_d64*)*2*(n + 1));341342if (pbuff == NULL) {343status = MLIB_FAILURE;344FREE_AND_RETURN_STATUS;345}346buffs = (mlib_d64**)(pbuff + bsize);347}348349for (l = 0; l < (n + 1); l++) buffs[l] = pbuff + l*wid;350for (l = 0; l < (n + 1); l++) buffs[l + (n + 1)] = buffs[l];351buffd = buffs[n] + wid;352353wid -= (m - 1);354hgt -= (n - 1);355adr_dst += dn*dll + dm*chan1;356357for (c = 0; c < chan1; c++) {358if (!(cmask & (1 << (chan1 - 1 - c)))) continue;359360sl = adr_src + c;361dl = adr_dst + c;362363for (l = 0; l < n; l++) {364mlib_d64 *buff = buffs[l];365366for (i = 0; i < wid + (m - 1); i++) {367buff[i] = (mlib_d64)sl[i*chan1];368}369370sl += sll;371}372373buff_ind = 0;374375for (i = 0; i < wid; i++) buffd[i] = 0.0;376377for (j = 0; j < hgt; j++) {378mlib_d64 **buffc = buffs + buff_ind;379mlib_d64 *buffn = buffc[n];380mlib_d64 *pk = k;381382for (l = 0; l < n; l++) {383mlib_d64 *buff_l = buffc[l];384385for (off = 0; off < m;) {386mlib_d64 *buff = buff_l + off;387388kw = m - off;389390if (kw > 2*MAX_KER) kw = MAX_KER; else391if (kw > MAX_KER) kw = kw/2;392off += kw;393394sp = sl;395dp = dl;396397p2 = buff[0]; p3 = buff[1]; p4 = buff[2];398p5 = buff[3]; p6 = buff[4]; p7 = buff[5];399400k0 = pk[0]; k1 = pk[1]; k2 = pk[2]; k3 = pk[3];401k4 = pk[4]; k5 = pk[5]; k6 = pk[6];402pk += kw;403404if (kw == 7) {405406if (l < (n - 1) || off < m) {407for (i = 0; i <= (wid - 2); i += 2) {408p0 = p2; p1 = p3; p2 = p4; p3 = p5; p4 = p6; p5 = p7;409410p6 = buff[i + 6]; p7 = buff[i + 7];411412buffd[i ] += p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4 + p5*k5 + p6*k6;413buffd[i + 1] += p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4 + p6*k5 + p7*k6;414}415416} else {417for (i = 0; i <= (wid - 2); i += 2) {418p0 = p2; p1 = p3; p2 = p4; p3 = p5; p4 = p6; p5 = p7;419420p6 = buff[i + 6]; p7 = buff[i + 7];421422buffn[i ] = (mlib_d64)sp[0];423buffn[i + 1] = (mlib_d64)sp[chan1];424425d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4 + p5*k5 + p6*k6 + buffd[i ];426d1 = p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4 + p6*k5 + p7*k6 + buffd[i + 1];427428CLAMP_S32(dp[0], d0);429CLAMP_S32(dp[chan1], d1);430431buffd[i ] = 0.0;432buffd[i + 1] = 0.0;433434sp += chan2;435dp += chan2;436}437}438439} else if (kw == 6) {440441if (l < (n - 1) || off < m) {442for (i = 0; i <= (wid - 2); i += 2) {443p0 = p2; p1 = p3; p2 = p4; p3 = p5; p4 = p6;444445p5 = buff[i + 5]; p6 = buff[i + 6];446447buffd[i ] += p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4 + p5*k5;448buffd[i + 1] += p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4 + p6*k5;449}450451} else {452for (i = 0; i <= (wid - 2); i += 2) {453p0 = p2; p1 = p3; p2 = p4; p3 = p5; p4 = p6;454455p5 = buff[i + 5]; p6 = buff[i + 6];456457buffn[i ] = (mlib_d64)sp[0];458buffn[i + 1] = (mlib_d64)sp[chan1];459460d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4 + p5*k5 + buffd[i ];461d1 = p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4 + p6*k5 + buffd[i + 1];462463CLAMP_S32(dp[0], d0);464CLAMP_S32(dp[chan1], d1);465466buffd[i ] = 0.0;467buffd[i + 1] = 0.0;468469sp += chan2;470dp += chan2;471}472}473474} else if (kw == 5) {475476if (l < (n - 1) || off < m) {477for (i = 0; i <= (wid - 2); i += 2) {478p0 = p2; p1 = p3; p2 = p4; p3 = p5;479480p4 = buff[i + 4]; p5 = buff[i + 5];481482buffd[i ] += p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4;483buffd[i + 1] += p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4;484}485486} else {487for (i = 0; i <= (wid - 2); i += 2) {488p0 = p2; p1 = p3; p2 = p4; p3 = p5;489490p4 = buff[i + 4]; p5 = buff[i + 5];491492buffn[i ] = (mlib_d64)sp[0];493buffn[i + 1] = (mlib_d64)sp[chan1];494495d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + p4*k4 + buffd[i ];496d1 = p1*k0 + p2*k1 + p3*k2 + p4*k3 + p5*k4 + buffd[i + 1];497498CLAMP_S32(dp[0], d0);499CLAMP_S32(dp[chan1], d1);500501buffd[i ] = 0.0;502buffd[i + 1] = 0.0;503504sp += chan2;505dp += chan2;506}507}508509} else if (kw == 4) {510511if (l < (n - 1) || off < m) {512for (i = 0; i <= (wid - 2); i += 2) {513p0 = p2; p1 = p3; p2 = p4;514515p3 = buff[i + 3]; p4 = buff[i + 4];516517buffd[i ] += p0*k0 + p1*k1 + p2*k2 + p3*k3;518buffd[i + 1] += p1*k0 + p2*k1 + p3*k2 + p4*k3;519}520521} else {522for (i = 0; i <= (wid - 2); i += 2) {523p0 = p2; p1 = p3; p2 = p4;524525p3 = buff[i + 3]; p4 = buff[i + 4];526527buffn[i ] = (mlib_d64)sp[0];528buffn[i + 1] = (mlib_d64)sp[chan1];529530d0 = p0*k0 + p1*k1 + p2*k2 + p3*k3 + buffd[i ];531d1 = p1*k0 + p2*k1 + p3*k2 + p4*k3 + buffd[i + 1];532533CLAMP_S32(dp[0], d0);534CLAMP_S32(dp[chan1], d1);535536buffd[i ] = 0.0;537buffd[i + 1] = 0.0;538539sp += chan2;540dp += chan2;541}542}543544} else if (kw == 3) {545546if (l < (n - 1) || off < m) {547for (i = 0; i <= (wid - 2); i += 2) {548p0 = p2; p1 = p3;549550p2 = buff[i + 2]; p3 = buff[i + 3];551552buffd[i ] += p0*k0 + p1*k1 + p2*k2;553buffd[i + 1] += p1*k0 + p2*k1 + p3*k2;554}555556} else {557for (i = 0; i <= (wid - 2); i += 2) {558p0 = p2; p1 = p3;559560p2 = buff[i + 2]; p3 = buff[i + 3];561562buffn[i ] = (mlib_d64)sp[0];563buffn[i + 1] = (mlib_d64)sp[chan1];564565d0 = p0*k0 + p1*k1 + p2*k2 + buffd[i ];566d1 = p1*k0 + p2*k1 + p3*k2 + buffd[i + 1];567568CLAMP_S32(dp[0], d0);569CLAMP_S32(dp[chan1], d1);570571buffd[i ] = 0.0;572buffd[i + 1] = 0.0;573574sp += chan2;575dp += chan2;576}577}578579} else { /* kw == 2 */580581if (l < (n - 1) || off < m) {582for (i = 0; i <= (wid - 2); i += 2) {583p0 = p2;584585p1 = buff[i + 1]; p2 = buff[i + 2];586587buffd[i ] += p0*k0 + p1*k1;588buffd[i + 1] += p1*k0 + p2*k1;589}590591} else {592for (i = 0; i <= (wid - 2); i += 2) {593p0 = p2;594595p1 = buff[i + 1]; p2 = buff[i + 2];596597buffn[i ] = (mlib_d64)sp[0];598buffn[i + 1] = (mlib_d64)sp[chan1];599600d0 = p0*k0 + p1*k1 + buffd[i ];601d1 = p1*k0 + p2*k1 + buffd[i + 1];602603CLAMP_S32(dp[0], d0);604CLAMP_S32(dp[chan1], d1);605606buffd[i ] = 0.0;607buffd[i + 1] = 0.0;608609sp += chan2;610dp += chan2;611}612}613}614}615}616617/* last pixels */618for (; i < wid; i++) {619mlib_d64 *pk = k, s = 0;620mlib_s32 x;621622for (l = 0; l < n; l++) {623mlib_d64 *buff = buffc[l] + i;624625for (x = 0; x < m; x++) s += buff[x] * (*pk++);626}627628CLAMP_S32(dp[0], s);629630buffn[i] = (mlib_d64)sp[0];631632sp += chan1;633dp += chan1;634}635636for (l = 0; l < (m - 1); l++) buffn[wid + l] = sp[l*chan1];637638/* next line */639sl += sll;640dl += dll;641642buff_ind++;643644if (buff_ind >= n + 1) buff_ind = 0;645}646}647648FREE_AND_RETURN_STATUS;649}650651/***************************************************************/652653654