liz/rawTherapee
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Amaze optimization, Issue 1835

Browse Source
This commit is contained in:
Ingo
2013-05-15 13:10:56 +02:00
parent 88c576dd67
commit acd2d20ef1
1 changed files with 259 additions and 191 deletions
Download Patch File Download Diff File Expand all files Collapse all files

450
rtengine/amaze_demosaic_RT.cc
View File

@@ -29,10 +29,14 @@
#include "rt_math.h" #include "rt_math.h"
#include "../rtgui/multilangmgr.h" #include "../rtgui/multilangmgr.h"
#include "procparams.h" #include "procparams.h"
#include "sleef.c"
using namespace rtengine; using namespace rtengine;
void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) { void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
// clock_t t1,t2;
// t1 = clock();
#define HCLIP(x) x //is this still necessary??? #define HCLIP(x) x //is this still necessary???
//min(clip_pt,x) //min(clip_pt,x)
@@ -42,7 +46,8 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
const float clip_pt = 1/initialGain; const float clip_pt = 1/initialGain;
#define TS 512 // Tile size; the image is processed in square tiles to lower memory requirements and facilitate multi-threading #define TS 512 // Tile size; the image is processed in square tiles to lower memory requirements and facilitate multi-threading
#define TSH 256
#define TS6 500
// local variables // local variables
@@ -79,7 +84,14 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
plistener->setProgressStr (Glib::ustring::compose(M("TP_RAW_DMETHOD_PROGRESSBAR"), RAWParams::methodstring[RAWParams::amaze])); plistener->setProgressStr (Glib::ustring::compose(M("TP_RAW_DMETHOD_PROGRESSBAR"), RAWParams::methodstring[RAWParams::amaze]));
plistener->setProgress (0.0); plistener->setProgress (0.0);
} }
struct s_mp {
float m;
float p;
};
struct s_hv {
float h;
float v;
};
#pragma omp parallel #pragma omp parallel
{ {
//position of top/left corner of the tile //position of top/left corner of the tile
@@ -118,71 +130,82 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
float (*delm); float (*delm);
// diagonal interpolation of R+B // diagonal interpolation of R+B
float (*rbint); float (*rbint);
s_hv (*Dgrb2);
// horizontal curvature of interpolated G (used to refine interpolation in Nyquist texture regions) // horizontal curvature of interpolated G (used to refine interpolation in Nyquist texture regions)
float (*Dgrbh2); // float (*Dgrbh2);
// vertical curvature of interpolated G // vertical curvature of interpolated G
float (*Dgrbv2); // float (*Dgrbv2);
// difference between up/down interpolations of G // difference between up/down interpolations of G
float (*dgintv); float (*dgintv);
// difference between left/right interpolations of G // difference between left/right interpolations of G
float (*dginth); float (*dginth);
// diagonal (plus) color difference R-B or G1-G2 // diagonal (plus) color difference R-B or G1-G2
float (*Dgrbp1); // float (*Dgrbp1);
// diagonal (minus) color difference R-B or G1-G2 // diagonal (minus) color difference R-B or G1-G2
float (*Dgrbm1); // float (*Dgrbm1);
s_mp (*Dgrbsq1);
// square of diagonal color difference // square of diagonal color difference
float (*Dgrbpsq1); // float (*Dgrbpsq1);
// square of diagonal color difference // square of diagonal color difference
float (*Dgrbmsq1); // float (*Dgrbmsq1);
// tile raw data // tile raw data
float (*cfa); float (*cfa);
// relative weight for combining plus and minus diagonal interpolations // relative weight for combining plus and minus diagonal interpolations
float (*pmwt); float (*pmwt);
// interpolated color difference R-B in minus and plus direction
s_mp (*rb);
// interpolated color difference R-B in plus direction // interpolated color difference R-B in plus direction
float (*rbp); // float (*rbp);
// interpolated color difference R-B in minus direction // interpolated color difference R-B in minus direction
float (*rbm); // float (*rbm);
// nyquist texture flag 1=nyquist, 0=not nyquist // nyquist texture flag 1=nyquist, 0=not nyquist
int (*nyquist); char (*nyquist);
#define CLF 1
// assign working space // assign working space
buffer = (char *) malloc((32*sizeof(float)+sizeof(int))*TS*TS); buffer = (char *) malloc(29*sizeof(float)*TS*TS - sizeof(float)*TS*TSH + sizeof(char)*TS*TSH+23*CLF*64);
char *data;
data = (char*)( ( uintptr_t(buffer) + uintptr_t(63)) / 64 * 64);
//merror(buffer,"amaze_interpolate()"); //merror(buffer,"amaze_interpolate()");
//memset(buffer,0,(34*sizeof(float)+sizeof(int))*TS*TS); //memset(buffer,0,(34*sizeof(float)+sizeof(int))*TS*TS);
// rgb array // rgb array
rgb = (float (*)[3]) buffer; //pointers to array rgb = (float (*)[3]) data; //pointers to array
delh = (float (*)) (buffer + 3*sizeof(float)*TS*TS); delh = (float (*)) (data + 3*sizeof(float)*TS*TS+1*CLF*64);
delv = (float (*)) (buffer + 4*sizeof(float)*TS*TS); delv = (float (*)) (data + 4*sizeof(float)*TS*TS+2*CLF*64);
delhsq = (float (*)) (buffer + 5*sizeof(float)*TS*TS); delhsq = (float (*)) (data + 5*sizeof(float)*TS*TS+3*CLF*64);
delvsq = (float (*)) (buffer + 6*sizeof(float)*TS*TS); delvsq = (float (*)) (data + 6*sizeof(float)*TS*TS+4*CLF*64);
dirwts = (float (*)[2]) (buffer + 7*sizeof(float)*TS*TS); dirwts = (float (*)[2]) (data + 7*sizeof(float)*TS*TS+5*CLF*64);
vcd = (float (*)) (buffer + 9*sizeof(float)*TS*TS); vcd = (float (*)) (data + 9*sizeof(float)*TS*TS+6*CLF*64);
hcd = (float (*)) (buffer + 10*sizeof(float)*TS*TS); hcd = (float (*)) (data + 10*sizeof(float)*TS*TS+7*CLF*64);
vcdalt = (float (*)) (buffer + 11*sizeof(float)*TS*TS); vcdalt = (float (*)) (data + 11*sizeof(float)*TS*TS+8*CLF*64);
hcdalt = (float (*)) (buffer + 12*sizeof(float)*TS*TS); hcdalt = (float (*)) (data + 12*sizeof(float)*TS*TS+9*CLF*64);
cddiffsq = (float (*)) (buffer + 13*sizeof(float)*TS*TS); cddiffsq = (float (*)) (data + 13*sizeof(float)*TS*TS+10*CLF*64);
hvwt = (float (*)) (buffer + 14*sizeof(float)*TS*TS); hvwt = (float (*)) (data + 14*sizeof(float)*TS*TS+11*CLF*64); //compressed 0.5 MB
Dgrb = (float (*)[2]) (buffer + 15*sizeof(float)*TS*TS); Dgrb = (float (*)[2]) (data + 15*sizeof(float)*TS*TS - sizeof(float)*TS*TSH+12*CLF*64);
delp = (float (*)) (buffer + 17*sizeof(float)*TS*TS); delp = (float (*)) (data + 17*sizeof(float)*TS*TS - sizeof(float)*TS*TSH+13*CLF*64); // compressed 0.5 MB
delm = (float (*)) (buffer + 18*sizeof(float)*TS*TS); delm = (float (*)) (data + 17*sizeof(float)*TS*TS+14*CLF*64); // compressed 0.5 MB
rbint = (float (*)) (buffer + 19*sizeof(float)*TS*TS); rbint = (float (*)) (data + 18*sizeof(float)*TS*TS - sizeof(float)*TS*TSH+15*CLF*64); // compressed 0.5 MB
Dgrbh2 = (float (*)) (buffer + 20*sizeof(float)*TS*TS); Dgrb2 = (s_hv (*)) (data + 18*sizeof(float)*TS*TS+16*CLF*64); // compressed 1.0 MB
Dgrbv2 = (float (*)) (buffer + 21*sizeof(float)*TS*TS); // Dgrbh2 = (float (*)) (data + 19*sizeof(float)*TS*TS);
dgintv = (float (*)) (buffer + 22*sizeof(float)*TS*TS); // Dgrbv2 = (float (*)) (data + 20*sizeof(float)*TS*TS);
dginth = (float (*)) (buffer + 23*sizeof(float)*TS*TS); dgintv = (float (*)) (data + 19*sizeof(float)*TS*TS+17*CLF*64);
Dgrbp1 = (float (*)) (buffer + 24*sizeof(float)*TS*TS); dginth = (float (*)) (data + 20*sizeof(float)*TS*TS+18*CLF*64);
Dgrbm1 = (float (*)) (buffer + 25*sizeof(float)*TS*TS); // Dgrbp1 = (float (*)) (data + 23*sizeof(float)*TS*TS); 1.0 MB
Dgrbpsq1 = (float (*)) (buffer + 26*sizeof(float)*TS*TS); // Dgrbm1 = (float (*)) (data + 23*sizeof(float)*TS*TS); 1.0 MB
Dgrbmsq1 = (float (*)) (buffer + 27*sizeof(float)*TS*TS); Dgrbsq1 = (s_mp (*)) (data + 21*sizeof(float)*TS*TS+19*CLF*64); // compressed 1.0 MB
cfa = (float (*)) (buffer + 28*sizeof(float)*TS*TS); // Dgrbpsq1 = (float (*)) (data + 23*sizeof(float)*TS*TS);
pmwt = (float (*)) (buffer + 29*sizeof(float)*TS*TS); // Dgrbmsq1 = (float (*)) (data + 24*sizeof(float)*TS*TS);
rbp = (float (*)) (buffer + 30*sizeof(float)*TS*TS); cfa = (float (*)) (data + 22*sizeof(float)*TS*TS+20*CLF*64);
rbm = (float (*)) (buffer + 31*sizeof(float)*TS*TS); pmwt = (float (*)) (data + 23*sizeof(float)*TS*TS+21*CLF*64); // compressed 0.5 MB
rb = (s_mp (*)) (data + 24*sizeof(float)*TS*TS - sizeof(float)*TS*TSH+22*CLF*64); // compressed 1.0 MB
nyquist = (int (*)) (buffer + 32*sizeof(int)*TS*TS); // rbp = (float (*)) (data + 30*sizeof(float)*TS*TS);
// rbm = (float (*)) (data + 31*sizeof(float)*TS*TS);
nyquist = (char (*)) (data + 25*sizeof(float)*TS*TS - sizeof(float)*TS*TSH+23*CLF*64); //compressed 0.875 MB
#undef CLF
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
/*double dt; /*double dt;
@@ -224,6 +247,8 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
#pragma omp for schedule(dynamic) collapse(2) nowait #pragma omp for schedule(dynamic) collapse(2) nowait
for (top=winy-16; top < winy+height; top += TS-32) for (top=winy-16; top < winy+height; top += TS-32)
for (left=winx-16; left < winx+width; left += TS-32) { for (left=winx-16; left < winx+width; left += TS-32) {
memset(nyquist, 0, sizeof(char)*TS*TSH);
memset(rbint, 0, sizeof(float)*TS*TSH);
//location of tile bottom edge //location of tile bottom edge
int bottom = min(top+TS,winy+height+16); int bottom = min(top+TS,winy+height+16);
//location of tile right edge //location of tile right edge
@@ -247,7 +272,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
//dummy indices //dummy indices
int i, j; int i, j;
// +1 or -1 // +1 or -1
int sgn; // int sgn;
//color ratios in up/down/left/right directions //color ratios in up/down/left/right directions
float cru, crd, crl, crr; float cru, crd, crl, crr;
@@ -263,23 +288,25 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
float Ginthar, Ginthha, Gintvar, Gintvha; float Ginthar, Ginthha, Gintvar, Gintvha;
//color difference (G-R or G-B) variance in up/down/left/right directions //color difference (G-R or G-B) variance in up/down/left/right directions
float Dgrbvvaru, Dgrbvvard, Dgrbhvarl, Dgrbhvarr; float Dgrbvvaru, Dgrbvvard, Dgrbhvarl, Dgrbhvarr;
float uave, dave, lave, rave;
//color difference variances in vertical and horizontal directions //color difference variances in vertical and horizontal directions
float vcdvar, hcdvar, vcdvar1, hcdvar1, hcdaltvar, vcdaltvar; float vcdvar, hcdvar, vcdvar1, hcdvar1, hcdaltvar, vcdaltvar;
//adaptive interpolation weight using variance of color differences //adaptive interpolation weight using variance of color differences
float varwt; float varwt; // 639 - 644
//adaptive interpolation weight using difference of left-right and up-down G interpolations //adaptive interpolation weight using difference of left-right and up-down G interpolations
float diffwt; float diffwt; // 640 - 644
//alternative adaptive weight for combining horizontal/vertical interpolations //alternative adaptive weight for combining horizontal/vertical interpolations
float hvwtalt; float hvwtalt; // 745 - 748
//temporary variables for combining interpolation weights at R and B sites //temporary variables for combining interpolation weights at R and B sites
float vo, ve; // float vo, ve;
//interpolation of G in four directions //interpolation of G in four directions
float gu, gd, gl, gr; float gu, gd, gl, gr;
//variance of G in vertical/horizontal directions //variance of G in vertical/horizontal directions
float gvarh, gvarv; float gvarh, gvarv;
//Nyquist texture test //Nyquist texture test
float nyqtest; float nyqtest; // 658 - 681
//accumulators for Nyquist texture interpolation //accumulators for Nyquist texture interpolation
float sumh, sumv, sumsqh, sumsqv, areawt; float sumh, sumv, sumsqh, sumsqv, areawt;
@@ -290,9 +317,9 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
//adaptive weights for combining diagonal interpolations //adaptive weights for combining diagonal interpolations
float wtse, wtnw, wtsw, wtne; float wtse, wtnw, wtsw, wtne;
//alternate weight for combining diagonal interpolations //alternate weight for combining diagonal interpolations
float pmwtalt; float pmwtalt; // 885 - 888
//variance of R-B in plus/minus directions //variance of R-B in plus/minus directions
float rbvarp, rbvarm; float rbvarm; // 843 - 848
@@ -402,32 +429,37 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=1; rr < rr1-1; rr++) for (rr=1; rr < rr1-1; rr++)
for (cc=1, indx=(rr)*TS+cc; cc < cc1-1; cc++, indx++) { for (cc=1, indx=(rr)*TS+cc; cc < cc1-1; cc++, indx++) {
delh[indx] = fabs(cfa[indx+1]-cfa[indx-1]); delh[indx] = fabsf(cfa[indx+1]-cfa[indx-1]);
delv[indx] = fabs(cfa[indx+v1]-cfa[indx-v1]); delv[indx] = fabsf(cfa[indx+v1]-cfa[indx-v1]);
delhsq[indx] = SQR(delh[indx]); delhsq[indx] = SQR(delh[indx]);
delvsq[indx] = SQR(delv[indx]); delvsq[indx] = SQR(delv[indx]);
delp[indx] = fabs(cfa[indx+p1]-cfa[indx-p1]); // delp[indx] = fabsf(cfa[indx+p1]-cfa[indx-p1]);
delm[indx] = fabs(cfa[indx+m1]-cfa[indx-m1]); // delm[indx] = fabsf(cfa[indx+m1]-cfa[indx-m1]);
} }
for (rr=2; rr < rr1-2; rr++) for (rr=2; rr < rr1-2; rr++)
for (cc=2,indx=(rr)*TS+cc; cc < cc1-2; cc++, indx++) { for (cc=2,indx=(rr)*TS+cc; cc < cc1-2; cc++, indx++) {
dirwts[indx][0] = eps+delv[indx+v1]+delv[indx-v1]+delv[indx];//+fabsf(cfa[indx+v2]-cfa[indx-v2]);
dirwts[indx][0] = eps+delv[indx+v1]+delv[indx-v1]+delv[indx];//+fabs(cfa[indx+v2]-cfa[indx-v2]);
//vert directional averaging weights //vert directional averaging weights
dirwts[indx][1] = eps+delh[indx+1]+delh[indx-1]+delh[indx];//+fabs(cfa[indx+2]-cfa[indx-2]); dirwts[indx][1] = eps+delh[indx+1]+delh[indx-1]+delh[indx];//+fabsf(cfa[indx+2]-cfa[indx-2]);
//horizontal weights //horizontal weights
if (FC(rr,cc)&1) {
//for later use in diagonal interpolation
//Dgrbp1[indx]=2*cfa[indx]-(cfa[indx-p1]+cfa[indx+p1]);
//Dgrbm1[indx]=2*cfa[indx]-(cfa[indx-m1]+cfa[indx+m1]);
Dgrbpsq1[indx]=(SQR(cfa[indx]-cfa[indx-p1])+SQR(cfa[indx]-cfa[indx+p1]));
Dgrbmsq1[indx]=(SQR(cfa[indx]-cfa[indx-m1])+SQR(cfa[indx]-cfa[indx+m1]));
}
} }
for (rr=6; rr < rr1-6; rr++)
for (cc=6+(FC(rr,2)&1), indx=(rr)*TS+cc; cc < cc1-6; cc+=2, indx+=2) {
delp[indx>>1] = fabsf(cfa[indx+p1]-cfa[indx-p1]);
delm[indx>>1] = fabsf(cfa[indx+m1]-cfa[indx-m1]);
}
for (rr=6; rr < rr1-6; rr++)
for (cc=6+(FC(rr,1)&1),indx=(rr)*TS+cc; cc < cc1-6; cc+=2, indx+=2) {
Dgrbsq1[indx>>1].p=(SQR(cfa[indx]-cfa[indx-p1])+SQR(cfa[indx]-cfa[indx+p1]));
Dgrbsq1[indx>>1].m=(SQR(cfa[indx]-cfa[indx-m1])+SQR(cfa[indx]-cfa[indx+m1]));
}
//t2_init += clock()-t1_init; //t2_init += clock()-t1_init;
// end of tile initialization // end of tile initialization
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
@@ -438,13 +470,13 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=4; rr<rr1-4; rr++) for (rr=4; rr<rr1-4; rr++)
//for (cc=4+(FC(rr,2)&1),indx=rr*TS+cc,c=FC(rr,cc); cc<cc1-4; cc+=2,indx+=2) { //for (cc=4+(FC(rr,2)&1),indx=rr*TS+cc,c=FC(rr,cc); cc<cc1-4; cc+=2,indx+=2) {
for (cc=4,indx=rr*TS+cc; cc<cc1-4; cc++,indx++) { for (cc=4,indx=rr*TS+cc; cc<cc1-4; cc++,indx++) {
c=FC(rr,cc); // c=FC(rr,cc);
if (c&1) {sgn=-1;} else {sgn=1;} // if (c&1) {sgn=-1;} else {sgn=1;}
//initialization of nyquist test //initialization of nyquist test
nyquist[indx]=0; // nyquist[indx]=0;
//preparation for diag interp //preparation for diag interp
rbint[indx]=0; // rbint[indx]=0;
//color ratios in each cardinal direction //color ratios in each cardinal direction
cru = cfa[indx-v1]*(dirwts[indx-v2][0]+dirwts[indx][0])/(dirwts[indx-v2][0]*(eps+cfa[indx])+dirwts[indx][0]*(eps+cfa[indx-v2])); cru = cfa[indx-v1]*(dirwts[indx-v2][0]+dirwts[indx][0])/(dirwts[indx-v2][0]*(eps+cfa[indx])+dirwts[indx][0]*(eps+cfa[indx-v2]));
@@ -452,15 +484,20 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
crl = cfa[indx-1]*(dirwts[indx-2][1]+dirwts[indx][1])/(dirwts[indx-2][1]*(eps+cfa[indx])+dirwts[indx][1]*(eps+cfa[indx-2])); crl = cfa[indx-1]*(dirwts[indx-2][1]+dirwts[indx][1])/(dirwts[indx-2][1]*(eps+cfa[indx])+dirwts[indx][1]*(eps+cfa[indx-2]));
crr = cfa[indx+1]*(dirwts[indx+2][1]+dirwts[indx][1])/(dirwts[indx+2][1]*(eps+cfa[indx])+dirwts[indx][1]*(eps+cfa[indx+2])); crr = cfa[indx+1]*(dirwts[indx+2][1]+dirwts[indx][1])/(dirwts[indx+2][1]*(eps+cfa[indx])+dirwts[indx][1]*(eps+cfa[indx+2]));
guha=HCLIP(cfa[indx-v1])+xdiv2f(cfa[indx]-cfa[indx-v2]);
gdha=HCLIP(cfa[indx+v1])+xdiv2f(cfa[indx]-cfa[indx+v2]);
glha=HCLIP(cfa[indx-1])+xdiv2f(cfa[indx]-cfa[indx-2]);
grha=HCLIP(cfa[indx+1])+xdiv2f(cfa[indx]-cfa[indx+2]);
/*
guha=HCLIP(cfa[indx-v1])+0.5*(cfa[indx]-cfa[indx-v2]); guha=HCLIP(cfa[indx-v1])+0.5*(cfa[indx]-cfa[indx-v2]);
gdha=HCLIP(cfa[indx+v1])+0.5*(cfa[indx]-cfa[indx+v2]); gdha=HCLIP(cfa[indx+v1])+0.5*(cfa[indx]-cfa[indx+v2]);
glha=HCLIP(cfa[indx-1])+0.5*(cfa[indx]-cfa[indx-2]); glha=HCLIP(cfa[indx-1])+0.5*(cfa[indx]-cfa[indx-2]);
grha=HCLIP(cfa[indx+1])+0.5*(cfa[indx]-cfa[indx+2]); grha=HCLIP(cfa[indx+1])+0.5*(cfa[indx]-cfa[indx+2]);
*/
if (fabs(1.0f-cru)<arthresh) {guar=cfa[indx]*cru;} else {guar=guha;} if (fabsf(1.0f-cru)<arthresh) {guar=cfa[indx]*cru;} else {guar=guha;}
if (fabs(1.0f-crd)<arthresh) {gdar=cfa[indx]*crd;} else {gdar=gdha;} if (fabsf(1.0f-crd)<arthresh) {gdar=cfa[indx]*crd;} else {gdar=gdha;}
if (fabs(1.0f-crl)<arthresh) {glar=cfa[indx]*crl;} else {glar=glha;} if (fabsf(1.0f-crl)<arthresh) {glar=cfa[indx]*crl;} else {glar=glha;}
if (fabs(1.0f-crr)<arthresh) {grar=cfa[indx]*crr;} else {grar=grha;} if (fabsf(1.0f-crr)<arthresh) {grar=cfa[indx]*crr;} else {grar=grha;}
hwt = dirwts[indx-1][1]/(dirwts[indx-1][1]+dirwts[indx+1][1]); hwt = dirwts[indx-1][1]/(dirwts[indx-1][1]+dirwts[indx+1][1]);
vwt = dirwts[indx-v1][0]/(dirwts[indx+v1][0]+dirwts[indx-v1][0]); vwt = dirwts[indx-v1][0]/(dirwts[indx+v1][0]+dirwts[indx-v1][0]);
@@ -471,11 +508,24 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
Gintvha = vwt*gdha+(1.0f-vwt)*guha; Gintvha = vwt*gdha+(1.0f-vwt)*guha;
Ginthha = hwt*grha+(1.0f-hwt)*glha; Ginthha = hwt*grha+(1.0f-hwt)*glha;
//interpolated color differences //interpolated color differences
if (FC(rr,cc)&1) {
vcd[indx] = cfa[indx]-Gintvar;
hcd[indx] = cfa[indx]-Ginthar;
vcdalt[indx] = cfa[indx]-Gintvha;
hcdalt[indx] = cfa[indx]-Ginthha;
} else {
//interpolated color differences
vcd[indx] = Gintvar-cfa[indx];
hcd[indx] = Ginthar-cfa[indx];
vcdalt[indx] = Gintvha-cfa[indx];
hcdalt[indx] = Ginthha-cfa[indx];
}
/*
vcd[indx] = sgn*(Gintvar-cfa[indx]); vcd[indx] = sgn*(Gintvar-cfa[indx]);
hcd[indx] = sgn*(Ginthar-cfa[indx]); hcd[indx] = sgn*(Ginthar-cfa[indx]);
vcdalt[indx] = sgn*(Gintvha-cfa[indx]); vcdalt[indx] = sgn*(Gintvha-cfa[indx]);
hcdalt[indx] = sgn*(Ginthha-cfa[indx]); hcdalt[indx] = sgn*(Ginthha-cfa[indx]);
*/
if (cfa[indx] > 0.8*clip_pt || Gintvha > 0.8*clip_pt || Ginthha > 0.8*clip_pt) { if (cfa[indx] > 0.8*clip_pt || Gintvha > 0.8*clip_pt || Ginthha > 0.8*clip_pt) {
//use HA if highlights are (nearly) clipped //use HA if highlights are (nearly) clipped
guar=guha; gdar=gdha; glar=glha; grar=grha; guar=guha; gdar=gdha; glar=glha; grar=grha;
@@ -529,7 +579,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
if (Gintv > clip_pt) vcd[indx]=-ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])+cfa[indx]; if (Gintv > clip_pt) vcd[indx]=-ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])+cfa[indx];
//if (Ginth > pre_mul[c]) hcd[indx]=-ULIM(Ginth,cfa[indx-1],cfa[indx+1])+cfa[indx];//for dcraw implementation //if (Ginth > pre_mul[c]) hcd[indx]=-ULIM(Ginth,cfa[indx-1],cfa[indx+1])+cfa[indx];//for dcraw implementation
//if (Gintv > pre_mul[c]) vcd[indx]=-ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])+cfa[indx]; //if (Gintv > pre_mul[c]) vcd[indx]=-ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])+cfa[indx];
} else {//R or B site } else {//R or B site
Ginth = hcd[indx]+cfa[indx];//interpolated G Ginth = hcd[indx]+cfa[indx];//interpolated G
@@ -556,9 +606,10 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
if (Gintv > clip_pt) vcd[indx]=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])-cfa[indx]; if (Gintv > clip_pt) vcd[indx]=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])-cfa[indx];
//if (Ginth > pre_mul[c]) hcd[indx]=ULIM(Ginth,cfa[indx-1],cfa[indx+1])-cfa[indx];//for dcraw implementation //if (Ginth > pre_mul[c]) hcd[indx]=ULIM(Ginth,cfa[indx-1],cfa[indx+1])-cfa[indx];//for dcraw implementation
//if (Gintv > pre_mul[c]) vcd[indx]=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])-cfa[indx]; //if (Gintv > pre_mul[c]) vcd[indx]=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1])-cfa[indx];
cddiffsq[indx] = SQR(vcd[indx]-hcd[indx]);
} }
cddiffsq[indx] = SQR(vcd[indx]-hcd[indx]); // cddiffsq[indx] = SQR(vcd[indx]-hcd[indx]);
} }
for (rr=6; rr<rr1-6; rr++) for (rr=6; rr<rr1-6; rr++)
@@ -566,10 +617,10 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
//compute color difference variances in cardinal directions //compute color difference variances in cardinal directions
float uave = vcd[indx]+vcd[indx-v1]+vcd[indx-v2]+vcd[indx-v3]; uave = vcd[indx]+vcd[indx-v1]+vcd[indx-v2]+vcd[indx-v3];
float dave = vcd[indx]+vcd[indx+v1]+vcd[indx+v2]+vcd[indx+v3]; dave = vcd[indx]+vcd[indx+v1]+vcd[indx+v2]+vcd[indx+v3];
float lave = (hcd[indx]+hcd[indx-1]+hcd[indx-2]+hcd[indx-3]); lave = (hcd[indx]+hcd[indx-1]+hcd[indx-2]+hcd[indx-3]);
float rave = (hcd[indx]+hcd[indx+1]+hcd[indx+2]+hcd[indx+3]); rave = (hcd[indx]+hcd[indx+1]+hcd[indx+2]+hcd[indx+3]);
Dgrbvvaru = SQR(vcd[indx]-uave)+SQR(vcd[indx-v1]-uave)+SQR(vcd[indx-v2]-uave)+SQR(vcd[indx-v3]-uave); Dgrbvvaru = SQR(vcd[indx]-uave)+SQR(vcd[indx-v1]-uave)+SQR(vcd[indx-v2]-uave)+SQR(vcd[indx-v3]-uave);
Dgrbvvard = SQR(vcd[indx]-dave)+SQR(vcd[indx+v1]-dave)+SQR(vcd[indx+v2]-dave)+SQR(vcd[indx+v3]-dave); Dgrbvvard = SQR(vcd[indx]-dave)+SQR(vcd[indx+v1]-dave)+SQR(vcd[indx+v2]-dave)+SQR(vcd[indx+v3]-dave);
@@ -597,7 +648,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
//if both agree on interpolation direction, choose the one with strongest directional discrimination; //if both agree on interpolation direction, choose the one with strongest directional discrimination;
//otherwise, choose the u/d and l/r difference fluctuation weights //otherwise, choose the u/d and l/r difference fluctuation weights
if ((0.5-varwt)*(0.5-diffwt)>0 && fabs(0.5-diffwt)<fabs(0.5-varwt)) {hvwt[indx]=varwt;} else {hvwt[indx]=diffwt;} if ((0.5-varwt)*(0.5-diffwt)>0 && fabsf(0.5-diffwt)<fabsf(0.5-varwt)) {hvwt[indx>>1]=varwt;} else {hvwt[indx>>1]=diffwt;}
//hvwt[indx]=varwt; //hvwt[indx]=varwt;
} }
@@ -634,19 +685,19 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
delhsq[indx-p2]+delvsq[indx-p2]+delhsq[indx+m2]+delvsq[indx+m2])); delhsq[indx-p2]+delvsq[indx-p2]+delhsq[indx+m2]+delvsq[indx+m2]));
if (nyqtest>0) {nyquist[indx]=1;}//nyquist=1 for nyquist region if (nyqtest>0) {nyquist[indx>>1]=1;}//nyquist=1 for nyquist region
} }
unsigned int nyquisttemp;
for (rr=8; rr<rr1-8; rr++) for (rr=8; rr<rr1-8; rr++)
for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) { for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) {
areawt=(nyquist[indx-v2]+nyquist[indx-m1]+nyquist[indx+p1]+ nyquisttemp=(nyquist[(indx-v2)>>1]+nyquist[(indx-m1)>>1]+nyquist[(indx+p1)>>1]+
nyquist[indx-2]+nyquist[indx]+nyquist[indx+2]+ nyquist[(indx-2)>>1]+nyquist[indx>>1]+nyquist[(indx+2)>>1]+
nyquist[indx-p1]+nyquist[indx+m1]+nyquist[indx+v2]); nyquist[(indx-p1)>>1]+nyquist[(indx+m1)>>1]+nyquist[(indx+v2)>>1]);
//if most of your neighbors are named Nyquist, it's likely that you're one too //if most of your neighbors are named Nyquist, it's likely that you're one too
if (areawt>4) nyquist[indx]=1; if (nyquisttemp>4) nyquist[indx>>1]=1;
//or not //or not
if (areawt<4) nyquist[indx]=0; if (nyquisttemp<4) nyquist[indx>>1]=0;
} }
//t2_nyqtest += clock() - t1_nyqtest; //t2_nyqtest += clock() - t1_nyqtest;
@@ -662,7 +713,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=8; rr<rr1-8; rr++) for (rr=8; rr<rr1-8; rr++)
for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) { for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) {
if (nyquist[indx]) { if (nyquist[indx>>1]) {
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
// area interpolation // area interpolation
@@ -670,19 +721,19 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (i=-6; i<7; i+=2) for (i=-6; i<7; i+=2)
for (j=-6; j<7; j+=2) { for (j=-6; j<7; j+=2) {
indx1=(rr+i)*TS+cc+j; indx1=(rr+i)*TS+cc+j;
if (nyquist[indx1]) { if (nyquist[indx1>>1]) {
sumh += cfa[indx1]-0.5*(cfa[indx1-1]+cfa[indx1+1]); sumh += cfa[indx1]-xdiv2f(cfa[indx1-1]+cfa[indx1+1]);
sumv += cfa[indx1]-0.5*(cfa[indx1-v1]+cfa[indx1+v1]); sumv += cfa[indx1]-xdiv2f(cfa[indx1-v1]+cfa[indx1+v1]);
sumsqh += 0.5*(SQR(cfa[indx1]-cfa[indx1-1])+SQR(cfa[indx1]-cfa[indx1+1])); sumsqh += xdiv2f(SQR(cfa[indx1]-cfa[indx1-1])+SQR(cfa[indx1]-cfa[indx1+1]));
sumsqv += 0.5*(SQR(cfa[indx1]-cfa[indx1-v1])+SQR(cfa[indx1]-cfa[indx1+v1])); sumsqv += xdiv2f(SQR(cfa[indx1]-cfa[indx1-v1])+SQR(cfa[indx1]-cfa[indx1+v1]));
areawt +=1; areawt +=1;
} }
} }
//horizontal and vertical color differences, and adaptive weight //horizontal and vertical color differences, and adaptive weight
hcdvar=epssq+fabs(areawt*sumsqh-sumh*sumh); hcdvar=epssq+fabsf(areawt*sumsqh-sumh*sumh);
vcdvar=epssq+fabs(areawt*sumsqv-sumv*sumv); vcdvar=epssq+fabsf(areawt*sumsqv-sumv*sumv);
hvwt[indx]=hcdvar/(vcdvar+hcdvar); hvwt[indx>>1]=hcdvar/(vcdvar+hcdvar);
// end of area interpolation // end of area interpolation
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
@@ -698,24 +749,26 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) { for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) {
//first ask if one gets more directional discrimination from nearby B/R sites //first ask if one gets more directional discrimination from nearby B/R sites
hvwtalt = 0.25*(hvwt[indx-m1]+hvwt[indx+p1]+hvwt[indx-p1]+hvwt[indx+m1]); hvwtalt = xdivf(hvwt[(indx-m1)>>1]+hvwt[(indx+p1)>>1]+hvwt[(indx-p1)>>1]+hvwt[(indx+m1)>>1],2);
vo=fabs(0.5-hvwt[indx]); // hvwtalt = 0.25*(hvwt[(indx-m1)>>1]+hvwt[(indx+p1)>>1]+hvwt[(indx-p1)>>1]+hvwt[(indx+m1)>>1]);
ve=fabs(0.5-hvwtalt); // vo=fabsf(0.5-hvwt[indx>>1]);
if (vo<ve) {hvwt[indx]=hvwtalt;}//a better result was obtained from the neighbors // ve=fabsf(0.5-hvwtalt);
if (fabsf(0.5-hvwt[indx>>1])<fabsf(0.5-hvwtalt)) {hvwt[indx>>1]=hvwtalt;}//a better result was obtained from the neighbors
// if (vo<ve) {hvwt[indx>>1]=hvwtalt;}//a better result was obtained from the neighbors
Dgrb[indx][0] = (hcd[indx]*(1.0f-hvwt[indx]) + vcd[indx]*hvwt[indx]);//evaluate color differences Dgrb[indx][0] = (hcd[indx]*(1.0f-hvwt[indx>>1]) + vcd[indx]*hvwt[indx>>1]);//evaluate color differences
//if (hvwt[indx]<0.5) Dgrb[indx][0]=hcd[indx]; //if (hvwt[indx]<0.5) Dgrb[indx][0]=hcd[indx];
//if (hvwt[indx]>0.5) Dgrb[indx][0]=vcd[indx]; //if (hvwt[indx]>0.5) Dgrb[indx][0]=vcd[indx];
rgb[indx][1] = cfa[indx] + Dgrb[indx][0];//evaluate G (finally!) rgb[indx][1] = cfa[indx] + Dgrb[indx][0];//evaluate G (finally!)
//local curvature in G (preparation for nyquist refinement step) //local curvature in G (preparation for nyquist refinement step)
if (nyquist[indx]) { if (nyquist[indx>>1]) {
Dgrbh2[indx] = SQR(rgb[indx][1] - 0.5*(rgb[indx-1][1]+rgb[indx+1][1])); Dgrb2[indx>>1].h = SQR(rgb[indx][1] - xdiv2f(rgb[indx-1][1]+rgb[indx+1][1]));
Dgrbv2[indx] = SQR(rgb[indx][1] - 0.5*(rgb[indx-v1][1]+rgb[indx+v1][1])); Dgrb2[indx>>1].v = SQR(rgb[indx][1] - xdiv2f(rgb[indx-v1][1]+rgb[indx+v1][1]));
} else { } else {
Dgrbh2[indx] = Dgrbv2[indx] = 0; Dgrb2[indx>>1].h = Dgrb2[indx>>1].v = 0;
} }
} }
@@ -730,16 +783,16 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=8; rr<rr1-8; rr++) for (rr=8; rr<rr1-8; rr++)
for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) { for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) {
if (nyquist[indx]) { if (nyquist[indx>>1]) {
//local averages (over Nyquist pixels only) of G curvature squared //local averages (over Nyquist pixels only) of G curvature squared
gvarh = epssq + (gquinc[0]*Dgrbh2[indx]+ gvarh = epssq + (gquinc[0]*Dgrb2[indx>>1].h+
gquinc[1]*(Dgrbh2[indx-m1]+Dgrbh2[indx+p1]+Dgrbh2[indx-p1]+Dgrbh2[indx+m1])+ gquinc[1]*(Dgrb2[(indx-m1)>>1].h+Dgrb2[(indx+p1)>>1].h+Dgrb2[(indx-p1)>>1].h+Dgrb2[(indx+m1)>>1].h)+
gquinc[2]*(Dgrbh2[indx-v2]+Dgrbh2[indx-2]+Dgrbh2[indx+2]+Dgrbh2[indx+v2])+ gquinc[2]*(Dgrb2[(indx-v2)>>1].h+Dgrb2[(indx-2)>>1].h+Dgrb2[(indx+2)>>1].h+Dgrb2[(indx+v2)>>1].h)+
gquinc[3]*(Dgrbh2[indx-m2]+Dgrbh2[indx+p2]+Dgrbh2[indx-p2]+Dgrbh2[indx+m2])); gquinc[3]*(Dgrb2[(indx-m2)>>1].h+Dgrb2[(indx+p2)>>1].h+Dgrb2[(indx-p2)>>1].h+Dgrb2[(indx+m2)>>1].h));
gvarv = epssq + (gquinc[0]*Dgrbv2[indx]+ gvarv = epssq + (gquinc[0]*Dgrb2[indx>>1].v+
gquinc[1]*(Dgrbv2[indx-m1]+Dgrbv2[indx+p1]+Dgrbv2[indx-p1]+Dgrbv2[indx+m1])+ gquinc[1]*(Dgrb2[(indx-m1)>>1].v+Dgrb2[(indx+p1)>>1].v+Dgrb2[(indx-p1)>>1].v+Dgrb2[(indx+m1)>>1].v)+
gquinc[2]*(Dgrbv2[indx-v2]+Dgrbv2[indx-2]+Dgrbv2[indx+2]+Dgrbv2[indx+v2])+ gquinc[2]*(Dgrb2[(indx-v2)>>1].v+Dgrb2[(indx-2)>>1].v+Dgrb2[(indx+2)>>1].v+Dgrb2[(indx+v2)>>1].v)+
gquinc[3]*(Dgrbv2[indx-m2]+Dgrbv2[indx+p2]+Dgrbv2[indx-p2]+Dgrbv2[indx+m2])); gquinc[3]*(Dgrb2[(indx-m2)>>1].v+Dgrb2[(indx+p2)>>1].v+Dgrb2[(indx-p2)>>1].v+Dgrb2[(indx+m2)>>1].v));
//use the results as weights for refined G interpolation //use the results as weights for refined G interpolation
Dgrb[indx][0] = (hcd[indx]*gvarv + vcd[indx]*gvarh)/(gvarv+gvarh); Dgrb[indx][0] = (hcd[indx]*gvarv + vcd[indx]*gvarh)/(gvarv+gvarh);
rgb[indx][1] = cfa[indx] + Dgrb[indx][0]; rgb[indx][1] = cfa[indx] + Dgrb[indx][0];
@@ -754,66 +807,75 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
// diagonal interpolation correction // diagonal interpolation correction
for (rr=8; rr<rr1-8; rr++) for (rr=8; rr<rr1-8; rr++)
for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-8; cc+=2,indx+=2) { for (cc=8+(FC(rr,2)&1),indx=rr*TS+cc,indx1=indx>>1; cc<cc1-8; cc+=2,indx+=2,indx1++) {
rbvarp = epssq + (gausseven[0]*(Dgrbpsq1[indx-v1]+Dgrbpsq1[indx-1]+Dgrbpsq1[indx+1]+Dgrbpsq1[indx+v1]) +
gausseven[1]*(Dgrbpsq1[indx-v2-1]+Dgrbpsq1[indx-v2+1]+Dgrbpsq1[indx-2-v1]+Dgrbpsq1[indx+2-v1]+
Dgrbpsq1[indx-2+v1]+Dgrbpsq1[indx+2+v1]+Dgrbpsq1[indx+v2-1]+Dgrbpsq1[indx+v2+1]));
rbvarm = epssq + (gausseven[0]*(Dgrbmsq1[indx-v1]+Dgrbmsq1[indx-1]+Dgrbmsq1[indx+1]+Dgrbmsq1[indx+v1]) +
gausseven[1]*(Dgrbmsq1[indx-v2-1]+Dgrbmsq1[indx-v2+1]+Dgrbmsq1[indx-2-v1]+Dgrbmsq1[indx+2-v1]+
Dgrbmsq1[indx-2+v1]+Dgrbmsq1[indx+2+v1]+Dgrbmsq1[indx+v2-1]+Dgrbmsq1[indx+v2+1]));
/*
rbvarp = epssq + (gausseven[0]*(Dgrbsq1[indx-v1].p+Dgrbsq1[indx-1].p+Dgrbsq1[indx+1].p+Dgrbsq1[indx+v1].p) +
gausseven[1]*(Dgrbsq1[indx-v2-1].p+Dgrbsq1[indx-v2+1].p+Dgrbsq1[indx-2-v1].p+Dgrbsq1[indx+2-v1].p+
Dgrbsq1[indx-2+v1].p+Dgrbsq1[indx+2+v1].p+Dgrbsq1[indx+v2-1].p+Dgrbsq1[indx+v2+1].p));
rbvarm = epssq + (gausseven[0]*(Dgrbsq1[indx-v1].m+Dgrbsq1[indx-1].m+Dgrbsq1[indx+1].m+Dgrbsq1[indx+v1].m) +
gausseven[1]*(Dgrbsq1[indx-v2-1].m+Dgrbsq1[indx-v2+1].m+Dgrbsq1[indx-2-v1].m+Dgrbsq1[indx+2-v1].m+
Dgrbsq1[indx-2+v1].m+Dgrbsq1[indx+2+v1].m+Dgrbsq1[indx+v2-1].m+Dgrbsq1[indx+v2+1].m));
*/
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
//diagonal color ratios //diagonal color ratios
crse=2.0f*(cfa[indx+m1])/(eps+cfa[indx]+(cfa[indx+m2])); crse=xmul2f(cfa[indx+m1])/(eps+cfa[indx]+(cfa[indx+m2]));
crnw=2.0f*(cfa[indx-m1])/(eps+cfa[indx]+(cfa[indx-m2])); crnw=xmul2f(cfa[indx-m1])/(eps+cfa[indx]+(cfa[indx-m2]));
crne=2.0f*(cfa[indx+p1])/(eps+cfa[indx]+(cfa[indx+p2])); crne=xmul2f(cfa[indx+p1])/(eps+cfa[indx]+(cfa[indx+p2]));
crsw=2.0f*(cfa[indx-p1])/(eps+cfa[indx]+(cfa[indx-p2])); crsw=xmul2f(cfa[indx-p1])/(eps+cfa[indx]+(cfa[indx-p2]));
//assign B/R at R/B sites //assign B/R at R/B sites
if (fabs(1.0f-crse)<arthresh) {rbse=cfa[indx]*crse;}//use this if more precise diag interp is necessary if (fabsf(1.0f-crse)<arthresh) {rbse=cfa[indx]*crse;}//use this if more precise diag interp is necessary
else {rbse=(cfa[indx+m1])+0.5*(cfa[indx]-cfa[indx+m2]);} else {rbse=(cfa[indx+m1])+xdiv2f(cfa[indx]-cfa[indx+m2]);}
if (fabs(1.0f-crnw)<arthresh) {rbnw=cfa[indx]*crnw;} if (fabsf(1.0f-crnw)<arthresh) {rbnw=cfa[indx]*crnw;}
else {rbnw=(cfa[indx-m1])+0.5*(cfa[indx]-cfa[indx-m2]);} else {rbnw=(cfa[indx-m1])+xdiv2f(cfa[indx]-cfa[indx-m2]);}
if (fabs(1.0f-crne)<arthresh) {rbne=cfa[indx]*crne;} if (fabsf(1.0f-crne)<arthresh) {rbne=cfa[indx]*crne;}
else {rbne=(cfa[indx+p1])+0.5*(cfa[indx]-cfa[indx+p2]);} else {rbne=(cfa[indx+p1])+xdiv2f(cfa[indx]-cfa[indx+p2]);}
if (fabs(1.0f-crsw)<arthresh) {rbsw=cfa[indx]*crsw;} if (fabsf(1.0f-crsw)<arthresh) {rbsw=cfa[indx]*crsw;}
else {rbsw=(cfa[indx-p1])+0.5*(cfa[indx]-cfa[indx-p2]);} else {rbsw=(cfa[indx-p1])+xdiv2f(cfa[indx]-cfa[indx-p2]);}
wtse= eps+delm[indx]+delm[indx+m1]+delm[indx+m2];//same as for wtu,wtd,wtl,wtr wtse= eps+delm[indx1]+delm[(indx+m1)>>1]+delm[(indx+m2)>>1];//same as for wtu,wtd,wtl,wtr
wtnw= eps+delm[indx]+delm[indx-m1]+delm[indx-m2]; wtnw= eps+delm[indx1]+delm[(indx-m1)>>1]+delm[(indx-m2)>>1];
wtne= eps+delp[indx]+delp[indx+p1]+delp[indx+p2]; wtne= eps+delp[indx1]+delp[(indx+p1)>>1]+delp[(indx+p2)>>1];
wtsw= eps+delp[indx]+delp[indx-p1]+delp[indx-p2]; wtsw= eps+delp[indx1]+delp[(indx-p1)>>1]+delp[(indx-p2)>>1];
rbm[indx] = (wtse*rbnw+wtnw*rbse)/(wtse+wtnw); rb[indx1].m = (wtse*rbnw+wtnw*rbse)/(wtse+wtnw);
rbp[indx] = (wtne*rbsw+wtsw*rbne)/(wtne+wtsw); rb[indx1].p = (wtne*rbsw+wtsw*rbne)/(wtne+wtsw);
/*
pmwt[indx] = rbvarm/(rbvarp+rbvarm); rbvarp = epssq + (gausseven[0]*(Dgrbsq1[indx-v1].p+Dgrbsq1[indx-1].p+Dgrbsq1[indx+1].p+Dgrbsq1[indx+v1].p) +
gausseven[1]*(Dgrbsq1[indx-v2-1].p+Dgrbsq1[indx-v2+1].p+Dgrbsq1[indx-2-v1].p+Dgrbsq1[indx+2-v1].p+
Dgrbsq1[indx-2+v1].p+Dgrbsq1[indx+2+v1].p+Dgrbsq1[indx+v2-1].p+Dgrbsq1[indx+v2+1].p));
*/
rbvarm = epssq + (gausseven[0]*(Dgrbsq1[(indx-v1)>>1].m+Dgrbsq1[(indx-1)>>1].m+Dgrbsq1[(indx+1)>>1].m+Dgrbsq1[(indx+v1)>>1].m) +
gausseven[1]*(Dgrbsq1[(indx-v2-1)>>1].m+Dgrbsq1[(indx-v2+1)>>1].m+Dgrbsq1[(indx-2-v1)>>1].m+Dgrbsq1[(indx+2-v1)>>1].m+
Dgrbsq1[(indx-2+v1)>>1].m+Dgrbsq1[(indx+2+v1)>>1].m+Dgrbsq1[(indx+v2-1)>>1].m+Dgrbsq1[(indx+v2+1)>>1].m));
pmwt[indx1] = rbvarm/((epssq + (gausseven[0]*(Dgrbsq1[(indx-v1)>>1].p+Dgrbsq1[(indx-1)>>1].p+Dgrbsq1[(indx+1)>>1].p+Dgrbsq1[(indx+v1)>>1].p) +
gausseven[1]*(Dgrbsq1[(indx-v2-1)>>1].p+Dgrbsq1[(indx-v2+1)>>1].p+Dgrbsq1[(indx-2-v1)>>1].p+Dgrbsq1[(indx+2-v1)>>1].p+
Dgrbsq1[(indx-2+v1)>>1].p+Dgrbsq1[(indx+2+v1)>>1].p+Dgrbsq1[(indx+v2-1)>>1].p+Dgrbsq1[(indx+v2+1)>>1].p)))+rbvarm);
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
//bound the interpolation in regions of high saturation //bound the interpolation in regions of high saturation
if (rbp[indx]<cfa[indx]) { if (rb[indx1].p<cfa[indx]) {
if (2.0*rbp[indx] < cfa[indx]) { if (xmul2f(rb[indx1].p) < cfa[indx]) {
rbp[indx] = ULIM(rbp[indx] ,cfa[indx-p1],cfa[indx+p1]); rb[indx1].p = ULIM(rb[indx1].p ,cfa[indx-p1],cfa[indx+p1]);
} else { } else {
pwt = 2.0*(cfa[indx]-rbp[indx])/(eps+rbp[indx]+cfa[indx]); pwt = xmul2f(cfa[indx]-rb[indx1].p)/(eps+rb[indx1].p+cfa[indx]);
rbp[indx]=pwt*rbp[indx] + (1.0f-pwt)*ULIM(rbp[indx],cfa[indx-p1],cfa[indx+p1]); rb[indx1].p=pwt*rb[indx1].p + (1.0f-pwt)*ULIM(rb[indx1].p,cfa[indx-p1],cfa[indx+p1]);
} }
} }
if (rbm[indx]<cfa[indx]) { if (rb[indx1].m<cfa[indx]) {
if (2.0*rbm[indx] < cfa[indx]) { if (xmul2f(rb[indx1].m) < cfa[indx]) {
rbm[indx] = ULIM(rbm[indx] ,cfa[indx-m1],cfa[indx+m1]); rb[indx1].m = ULIM(rb[indx1].m ,cfa[indx-m1],cfa[indx+m1]);
} else { } else {
mwt = 2.0*(cfa[indx]-rbm[indx])/(eps+rbm[indx]+cfa[indx]); mwt = xmul2f(cfa[indx]-rb[indx1].m)/(eps+rb[indx1].m+cfa[indx]);
rbm[indx]=mwt*rbm[indx] + (1.0f-mwt)*ULIM(rbm[indx],cfa[indx-m1],cfa[indx+m1]); rb[indx1].m=mwt*rb[indx1].m + (1.0f-mwt)*ULIM(rb[indx1].m,cfa[indx-m1],cfa[indx+m1]);
} }
} }
if (rbp[indx] > clip_pt) rbp[indx]=ULIM(rbp[indx],cfa[indx-p1],cfa[indx+p1]);//for RT implementation if (rb[indx1].p > clip_pt) rb[indx1].p=ULIM(rb[indx1].p,cfa[indx-p1],cfa[indx+p1]);//for RT implementation
if (rbm[indx] > clip_pt) rbm[indx]=ULIM(rbm[indx],cfa[indx-m1],cfa[indx+m1]); if (rb[indx1].m > clip_pt) rb[indx1].m=ULIM(rb[indx1].m,cfa[indx-m1],cfa[indx+m1]);
//c=2-FC(rr,cc);//for dcraw implementation //c=2-FC(rr,cc);//for dcraw implementation
//if (rbp[indx] > pre_mul[c]) rbp[indx]=ULIM(rbp[indx],cfa[indx-p1],cfa[indx+p1]); //if (rbp[indx] > pre_mul[c]) rbp[indx]=ULIM(rbp[indx],cfa[indx-p1],cfa[indx+p1]);
//if (rbm[indx] > pre_mul[c]) rbm[indx]=ULIM(rbm[indx],cfa[indx-m1],cfa[indx+m1]); //if (rbm[indx] > pre_mul[c]) rbm[indx]=ULIM(rbm[indx],cfa[indx-m1],cfa[indx+m1]);
@@ -825,39 +887,41 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=10; rr<rr1-10; rr++) for (rr=10; rr<rr1-10; rr++)
for (cc=10+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-10; cc+=2,indx+=2) { for (cc=10+(FC(rr,2)&1),indx=rr*TS+cc,indx1=indx>>1; cc<cc1-10; cc+=2,indx+=2,indx1++) {
//first ask if one gets more directional discrimination from nearby B/R sites //first ask if one gets more directional discrimination from nearby B/R sites
pmwtalt = 0.25*(pmwt[indx-m1]+pmwt[indx+p1]+pmwt[indx-p1]+pmwt[indx+m1]); pmwtalt = xdivf(pmwt[(indx-m1)>>1]+pmwt[(indx+p1)>>1]+pmwt[(indx-p1)>>1]+pmwt[(indx+m1)>>1],2);
vo=fabs(0.5-pmwt[indx]); // vo=fabsf(0.5-pmwt[indx1]);
ve=fabs(0.5-pmwtalt); // ve=fabsf(0.5-pmwtalt);
if (vo<ve) {pmwt[indx]=pmwtalt;}//a better result was obtained from the neighbors if (fabsf(0.5-pmwt[indx1])<fabsf(0.5-pmwtalt)) {pmwt[indx1]=pmwtalt;}//a better result was obtained from the neighbors
rbint[indx] = 0.5*(cfa[indx] + rbm[indx]*(1.0f-pmwt[indx]) + rbp[indx]*pmwt[indx]);//this is R+B, interpolated
// if (vo<ve) {pmwt[indx1]=pmwtalt;}//a better result was obtained from the neighbors
rbint[indx1] = xdiv2f(cfa[indx] + rb[indx1].m*(1.0f-pmwt[indx1]) + rb[indx1].p*pmwt[indx1]);//this is R+B, interpolated
} }
for (rr=12; rr<rr1-12; rr++) for (rr=12; rr<rr1-12; rr++)
for (cc=12+(FC(rr,2)&1),indx=rr*TS+cc; cc<cc1-12; cc+=2,indx+=2) { for (cc=12+(FC(rr,2)&1),indx=rr*TS+cc,indx1=indx>>1; cc<cc1-12; cc+=2,indx+=2,indx1++) {
if (fabs(0.5-pmwt[indx])<fabs(0.5-hvwt[indx]) ) continue; if (fabsf(0.5-pmwt[indx>>1])<fabsf(0.5-hvwt[indx>>1]) ) continue;
//now interpolate G vertically/horizontally using R+B values //now interpolate G vertically/horizontally using R+B values
//unfortunately, since G interpolation cannot be done diagonally this may lead to color shifts //unfortunately, since G interpolation cannot be done diagonally this may lead to color shifts
//color ratios for G interpolation //color ratios for G interpolation
cru = cfa[indx-v1]*2.0/(eps+rbint[indx]+rbint[indx-v2]); cru = cfa[indx-v1]*2.0/(eps+rbint[indx1]+rbint[(indx1-v1)]);
crd = cfa[indx+v1]*2.0/(eps+rbint[indx]+rbint[indx+v2]); crd = cfa[indx+v1]*2.0/(eps+rbint[indx1]+rbint[(indx1+v1)]);
crl = cfa[indx-1]*2.0/(eps+rbint[indx]+rbint[indx-2]); crl = cfa[indx-1]*2.0/(eps+rbint[indx1]+rbint[(indx1-1)]);
crr = cfa[indx+1]*2.0/(eps+rbint[indx]+rbint[indx+2]); crr = cfa[indx+1]*2.0/(eps+rbint[indx1]+rbint[(indx1+1)]);
//interpolated G via adaptive ratios or Hamilton-Adams in each cardinal direction //interpolated G via adaptive ratios or Hamilton-Adams in each cardinal direction
if (fabs(1.0f-cru)<arthresh) {gu=rbint[indx]*cru;} if (fabsf(1.0f-cru)<arthresh) {gu=rbint[indx1]*cru;}
else {gu=cfa[indx-v1]+0.5*(rbint[indx]-rbint[indx-v2]);} else {gu=cfa[indx-v1]+xdiv2f(rbint[indx1]-rbint[(indx1-v1)]);}
if (fabs(1.0f-crd)<arthresh) {gd=rbint[indx]*crd;} if (fabsf(1.0f-crd)<arthresh) {gd=rbint[indx1]*crd;}
else {gd=cfa[indx+v1]+0.5*(rbint[indx]-rbint[indx+v2]);} else {gd=cfa[indx+v1]+xdiv2f(rbint[indx1]-rbint[(indx1+v1)]);}
if (fabs(1.0f-crl)<arthresh) {gl=rbint[indx]*crl;} if (fabsf(1.0f-crl)<arthresh) {gl=rbint[indx1]*crl;}
else {gl=cfa[indx-1]+0.5*(rbint[indx]-rbint[indx-2]);} else {gl=cfa[indx-1]+xdiv2f(rbint[indx1]-rbint[(indx1-1)]);}
if (fabs(1.0f-crr)<arthresh) {gr=rbint[indx]*crr;} if (fabsf(1.0f-crr)<arthresh) {gr=rbint[indx1]*crr;}
else {gr=cfa[indx+1]+0.5*(rbint[indx]-rbint[indx+2]);} else {gr=cfa[indx+1]+xdiv2f(rbint[indx1]-rbint[(indx1+1)]);}
//gu=rbint[indx]*cru; //gu=rbint[indx]*cru;
//gd=rbint[indx]*crd; //gd=rbint[indx]*crd;
@@ -870,19 +934,19 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
//bound the interpolation in regions of high saturation //bound the interpolation in regions of high saturation
if (Gintv<rbint[indx]) { if (Gintv<rbint[indx1]) {
if (2*Gintv < rbint[indx]) { if (2*Gintv < rbint[indx1]) {
Gintv = ULIM(Gintv ,cfa[indx-v1],cfa[indx+v1]); Gintv = ULIM(Gintv ,cfa[indx-v1],cfa[indx+v1]);
} else { } else {
vwt = 2.0*(rbint[indx]-Gintv)/(eps+Gintv+rbint[indx]); vwt = 2.0*(rbint[indx1]-Gintv)/(eps+Gintv+rbint[indx1]);
Gintv=vwt*Gintv + (1.0f-vwt)*ULIM(Gintv,cfa[indx-v1],cfa[indx+v1]); Gintv=vwt*Gintv + (1.0f-vwt)*ULIM(Gintv,cfa[indx-v1],cfa[indx+v1]);
} }
} }
if (Ginth<rbint[indx]) { if (Ginth<rbint[indx1]) {
if (2*Ginth < rbint[indx]) { if (2*Ginth < rbint[indx1]) {
Ginth = ULIM(Ginth ,cfa[indx-1],cfa[indx+1]); Ginth = ULIM(Ginth ,cfa[indx-1],cfa[indx+1]);
} else { } else {
hwt = 2.0*(rbint[indx]-Ginth)/(eps+Ginth+rbint[indx]); hwt = 2.0*(rbint[indx1]-Ginth)/(eps+Ginth+rbint[indx1]);
Ginth=hwt*Ginth + (1.0f-hwt)*ULIM(Ginth,cfa[indx-1],cfa[indx+1]); Ginth=hwt*Ginth + (1.0f-hwt)*ULIM(Ginth,cfa[indx-1],cfa[indx+1]);
} }
} }
@@ -894,7 +958,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
//if (Gintv > pre_mul[c]) Gintv=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1]); //if (Gintv > pre_mul[c]) Gintv=ULIM(Gintv,cfa[indx-v1],cfa[indx+v1]);
// %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% // %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
rgb[indx][1] = Ginth*(1.0f-hvwt[indx]) + Gintv*hvwt[indx]; rgb[indx][1] = Ginth*(1.0f-hvwt[indx1]) + Gintv*hvwt[indx1];
//rgb[indx][1] = 0.5*(rgb[indx][1]+0.25*(rgb[indx-v1][1]+rgb[indx+v1][1]+rgb[indx-1][1]+rgb[indx+1][1])); //rgb[indx][1] = 0.5*(rgb[indx][1]+0.25*(rgb[indx-v1][1]+rgb[indx+v1][1]+rgb[indx-1][1]+rgb[indx+1][1]));
Dgrb[indx][0] = rgb[indx][1]-cfa[indx]; Dgrb[indx][0] = rgb[indx][1]-cfa[indx];
@@ -915,10 +979,10 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
} }
for (rr=12; rr<rr1-12; rr++) for (rr=12; rr<rr1-12; rr++)
for (cc=12+(FC(rr,2)&1),indx=rr*TS+cc,c=1-FC(rr,cc)/2; cc<cc1-12; cc+=2,indx+=2) { for (cc=12+(FC(rr,2)&1),indx=rr*TS+cc,c=1-FC(rr,cc)/2; cc<cc1-12; cc+=2,indx+=2) {
wtnw=1.0/(eps+fabs(Dgrb[indx-m1][c]-Dgrb[indx+m1][c])+fabs(Dgrb[indx-m1][c]-Dgrb[indx-m3][c])+fabs(Dgrb[indx+m1][c]-Dgrb[indx-m3][c])); wtnw=1.0/(eps+fabsf(Dgrb[indx-m1][c]-Dgrb[indx+m1][c])+fabsf(Dgrb[indx-m1][c]-Dgrb[indx-m3][c])+fabsf(Dgrb[indx+m1][c]-Dgrb[indx-m3][c]));
wtne=1.0/(eps+fabs(Dgrb[indx+p1][c]-Dgrb[indx-p1][c])+fabs(Dgrb[indx+p1][c]-Dgrb[indx+p3][c])+fabs(Dgrb[indx-p1][c]-Dgrb[indx+p3][c])); wtne=1.0/(eps+fabsf(Dgrb[indx+p1][c]-Dgrb[indx-p1][c])+fabsf(Dgrb[indx+p1][c]-Dgrb[indx+p3][c])+fabsf(Dgrb[indx-p1][c]-Dgrb[indx+p3][c]));
wtsw=1.0/(eps+fabs(Dgrb[indx-p1][c]-Dgrb[indx+p1][c])+fabs(Dgrb[indx-p1][c]-Dgrb[indx+m3][c])+fabs(Dgrb[indx+p1][c]-Dgrb[indx-p3][c])); wtsw=1.0/(eps+fabsf(Dgrb[indx-p1][c]-Dgrb[indx+p1][c])+fabsf(Dgrb[indx-p1][c]-Dgrb[indx+m3][c])+fabsf(Dgrb[indx+p1][c]-Dgrb[indx-p3][c]));
wtse=1.0/(eps+fabs(Dgrb[indx+m1][c]-Dgrb[indx-m1][c])+fabs(Dgrb[indx+m1][c]-Dgrb[indx-p3][c])+fabs(Dgrb[indx-m1][c]-Dgrb[indx+m3][c])); wtse=1.0/(eps+fabsf(Dgrb[indx+m1][c]-Dgrb[indx-m1][c])+fabsf(Dgrb[indx+m1][c]-Dgrb[indx-p3][c])+fabsf(Dgrb[indx-m1][c]-Dgrb[indx+m3][c]));
//Dgrb[indx][c]=(wtnw*Dgrb[indx-m1][c]+wtne*Dgrb[indx+p1][c]+wtsw*Dgrb[indx-p1][c]+wtse*Dgrb[indx+m1][c])/(wtnw+wtne+wtsw+wtse); //Dgrb[indx][c]=(wtnw*Dgrb[indx-m1][c]+wtne*Dgrb[indx+p1][c]+wtsw*Dgrb[indx-p1][c]+wtse*Dgrb[indx+m1][c])/(wtnw+wtne+wtsw+wtse);
@@ -930,9 +994,11 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
for (rr=12; rr<rr1-12; rr++) for (rr=12; rr<rr1-12; rr++)
for (cc=12+(FC(rr,1)&1),indx=rr*TS+cc,c=FC(rr,cc+1)/2; cc<cc1-12; cc+=2,indx+=2) for (cc=12+(FC(rr,1)&1),indx=rr*TS+cc,c=FC(rr,cc+1)/2; cc<cc1-12; cc+=2,indx+=2)
for(c=0;c<2;c++){ for(c=0;c<2;c++){
// Dgrb[indx][c]=((hvwt[indx-v1])*Dgrb[indx-v1][c]+(1.0f-hvwt[indx+1])*Dgrb[indx+1][c]+(1.0f-hvwt[indx-1])*Dgrb[indx-1][c]+(hvwt[indx+v1])*Dgrb[indx+v1][c])/
// ((hvwt[indx-v1])+(1.0f-hvwt[indx+1])+(1.0f-hvwt[indx-1])+(hvwt[indx+v1]));
Dgrb[indx][c]=((hvwt[indx-v1])*Dgrb[indx-v1][c]+(1.0f-hvwt[indx+1])*Dgrb[indx+1][c]+(1.0f-hvwt[indx-1])*Dgrb[indx-1][c]+(hvwt[indx+v1])*Dgrb[indx+v1][c])/ Dgrb[indx][c]=((hvwt[(indx-v1)>>1])*Dgrb[indx-v1][c]+(1.0f-hvwt[(indx+1)>>1])*Dgrb[indx+1][c]+(1.0f-hvwt[(indx-1)>>1])*Dgrb[indx-1][c]+(hvwt[(indx+v1)>>1])*Dgrb[indx+v1][c])/
((hvwt[indx-v1])+(1.0f-hvwt[indx+1])+(1.0f-hvwt[indx-1])+(hvwt[indx+v1])); ((hvwt[(indx-v1)>>1])+(1.0f-hvwt[(indx+1)>>1])+(1.0f-hvwt[(indx-1)>>1])+(hvwt[(indx+v1)>>1]));
} }
for(rr=12; rr<rr1-12; rr++) for(rr=12; rr<rr1-12; rr++)
@@ -985,5 +1051,7 @@ void RawImageSource::amaze_demosaic_RT(int winx, int winy, int winw, int winh) {
// done // done
#undef TS #undef TS
//t2 = clock() - t1;
//printf("Amaze took %d ms\n",t2);
} }