mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Experimental: real Sobel gradient detection

Browse Source
This commit is contained in:
Steven Robertson 2011-10-28 08:25:00 -04:00
parent 9b2b3ba011
commit 15f88383b1
1 changed files with 33 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
cuburn/code/filtering.py
View File

@ -153,14 +153,32 @@ void density_est(float4 *pixbuf, float4 *outbuf,
float den; float den;
read_pix(in, den); read_pix(in, den);
float *dens = reinterpret_cast<float*>(pixbuf);
float top = dens[(idx-{{info.acc_stride}})*4+3],
bottom = dens[(idx+{{info.acc_stride}})*4+3];
den = fmaxf(den, fabsf(top-bottom));
float left = dens[idx*4-1], right = dens[idx*4+7];
den = fmaxf(den, fabsf(left-right));
if (in.w > 0 && den > 0) { if (in.w > 0 && den > 0) {
// Compute a Sobel derivative, The Terrible Way! Because I'm Lazy.
// Or more to the point, because even this sloppy mess is such a
// small part of the overall runtime that it's not yet worth
// optimizing. Using a [3, 10, 3] Sobel, normalized to represent
// the density change over a single pixel.
const float corner = 0.0625f; // (3/16)/3
const float main = 0.208333333f; // (10/16)/3
float *dens = reinterpret_cast<float*>(pixbuf);
int didx = idx * 4 + 3;
float v = 0.0f, h = 0.0f, p;
p = corner * dens[didx-{{info.acc_stride*4}}-4];
v -= p; h -= p;
v -= main * dens[didx-{{info.acc_stride*4}}];
p = corner * dens[didx-{{info.acc_stride*4}}+4];
v -= p; h += p;
h -= main * dens[didx-4];
h += main * dens[didx+4];
p = corner * dens[didx+{{info.acc_stride*4}}-4];
v += p; h -= p;
v += main * dens[didx+{{info.acc_stride*4}}];
p = corner * dens[didx+{{info.acc_stride*4}}-4];
v += p; h += p;
float sobel_mag = sqrtf(v*v + h*h);
float ls = k1 * logf(1.0f + in.w * k2) / in.w; float ls = k1 * logf(1.0f + in.w * k2) / in.w;
in.x *= ls; in.x *= ls;
in.y *= ls; in.y *= ls;
@ -174,8 +192,14 @@ void density_est(float4 *pixbuf, float4 *outbuf,
// then scaled in inverse proportion to the density of the point // then scaled in inverse proportion to the density of the point
// being scaled. // being scaled.
float sd = est_sd * powf(den+1.0f, neg_est_curve); float sd = est_sd * powf(den+1.0f, neg_est_curve);
// Clamp the final standard deviation. Things will go badly if the // And for the Sobel...
// minimum is undershot. float sobel_sd = est_sd * powf(sobel_mag+1.0f, neg_est_curve);
// If the Sobel SD is smaller than the minimum SD, we're probably
// on a strong edge; blur with a standard deviation of 1px.
if (sobel_sd < fmaxf(est_min, MIN_SD)) sd = fminf(sd, 0.5f);
// Clamp the final standard deviation.
sd = fminf(MAX_SD, fmaxf(sd, est_min)); sd = fminf(MAX_SD, fmaxf(sd, est_min));
// Below a certain threshold, only one coeffecient would be // Below a certain threshold, only one coeffecient would be