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

Speedup for guided filter

Browse Source
This commit is contained in:
Ingo Weyrich
2019-09-17 15:11:12 +02:00
parent c88a74e83a
commit 991fc94d89
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
rtengine/guidedfilter.cc
View File

@@ -207,9 +207,6 @@ void guidedFilter(const array2D<float> &guide, const array2D<float> &src, array2
apply(SUBMUL, b, a, meanI, meanp);
DEBUG_DUMP(b);
meanI.free(); // frees w * h * 4 byte
meanp.free(); // frees w * h * 4 byte
array2D<float> &meana = a;
f_mean(meana, a, r1);
DEBUG_DUMP(meana);
@@ -218,18 +215,25 @@ void guidedFilter(const array2D<float> &guide, const array2D<float> &src, array2
f_mean(meanb, b, r1);
DEBUG_DUMP(meanb);
blur_buf.resize(0); // frees w * h * 4 byte
const int Ws = meana.width();
const int Hs = meana.height();
const int Wd = q.width();
const int Hd = q.height();
array2D<float> meanA(W, H);
f_upsample(meanA, meana);
DEBUG_DUMP(meanA);
float col_scale = float (Ws) / float (Wd);
float row_scale = float (Hs) / float (Hd);
array2D<float> &meanB = q;
f_upsample(meanB, meanb);
DEBUG_DUMP(meanB);
#ifdef _OPENMP
#pragma omp parallel for if (multithread)
#endif
apply(ADDMUL, q, meanA, I, meanB);
DEBUG_DUMP(q);
for (int y = 0; y < Hd; ++y) {
float ymrs = y * row_scale;
for (int x = 0; x < Wd; ++x) {
q[y][x] = getBilinearValue(meana, x * col_scale, ymrs) * I[y][x] + getBilinearValue(meanb, x * col_scale, ymrs);
}
}
}
} // namespace rtengine