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

Add background color support, and tentatively disable density blurring.

Browse Source
This commit is contained in:
Steven Robertson
2011-06-19 00:30:54 -04:00
parent 883de380fc
commit c66cb463d4
2 changed files with 39 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
cuburn/code/filtering.py
View File

@ -2,17 +2,20 @@
from cuburn.code.util import *
class ColorClip(HunkOCode):
defs = """
def __init__(self, features):
self.defs = self.defs_tmpl.substitute(features=features)
defs_tmpl = Template("""
__global__
void colorclip(float4 *pixbuf, float gamma, float vibrancy, float highpow,
float linrange, float lingam) {
float linrange, float lingam, float3 bkgd) {
// TODO: test if over an edge of the framebuffer - currently gutters are
// used and up to 256 pixels are ignored, which breaks when width<256
int i = blockDim.x * blockIdx.x + threadIdx.x;
float4 pix = pixbuf[i];
if (pix.w <= 0) {
pixbuf[i] = make_float4(0, 0, 0, 0);
pixbuf[i] = make_float4(bkgd.x, bkgd.y, bkgd.z, 0);
return;
}
@ -57,6 +60,18 @@ void colorclip(float4 *pixbuf, float gamma, float vibrancy, float highpow,
pix.y += (1.0f - vibrancy) * powf(opix.y, gamma);
pix.z += (1.0f - vibrancy) * powf(opix.z, gamma);
{{if features.alpha_output_channel}}
float 1_alpha = 1 / alpha;
pix.x *= 1_alpha;
pix.y *= 1_alpha;
pix.z *= 1_alpha;
{{else}}
pix.x += (1.0f - alpha) * bkgd.x;
pix.y += (1.0f - alpha) * bkgd.y;
pix.z += (1.0f - alpha) * bkgd.z;
{{endif}}
pix.w = alpha;
// Clamp values. I think this is superfluous, but I'm not certain.
pix.x = fminf(1.0f, pix.x);
pix.y = fminf(1.0f, pix.y);
@ -64,7 +79,7 @@ void colorclip(float4 *pixbuf, float gamma, float vibrancy, float highpow,
pixbuf[i] = pix;
}
"""
""")
class DensityEst(HunkOCode):
"""