Instituting denoise branch to work on improvements to RT denoise. Most definitely a work in progress, but the patch is becoming quite complicated so it is definitely worth having it backed up on Google code.
This commit is contained in:
Emil Martinec
675
rtengine/curves.cc
Normal file
675
rtengine/curves.cc
Normal file
@@ -0,0 +1,675 @@
|
||||
/*
|
||||
* This file is part of RawTherapee.
|
||||
*
|
||||
* Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
|
||||
*
|
||||
* RawTherapee is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* RawTherapee is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with RawTherapee. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
#include <glib.h>
|
||||
#include <glib/gstdio.h>
|
||||
#include "curves.h"
|
||||
#include <cmath>
|
||||
#include <vector>
|
||||
#include "mytime.h"
|
||||
#include <cstring>
|
||||
|
||||
#include "array2D.h"
|
||||
#include "LUT.h"
|
||||
#include "color.h"
|
||||
|
||||
#undef CLIPD
|
||||
#define CLIPD(a) ((a)>0.0f?((a)<1.0f?(a):1.0f):0.0f)
|
||||
#define CLIP(a) ((a)<65535 ? (a) : (65535))
|
||||
|
||||
|
||||
namespace rtengine {
|
||||
|
||||
Curve::Curve () {
|
||||
x = 0;
|
||||
y = 0;
|
||||
ypp = 0;
|
||||
hash = NULL;
|
||||
hashSize = 1000; // has to be initiallised to the maximum value
|
||||
}
|
||||
|
||||
Curve::~Curve () {
|
||||
if (hash)
|
||||
delete [] hash;
|
||||
}
|
||||
|
||||
void Curve::AddPolygons ()
|
||||
{
|
||||
if (firstPointIncluded) {
|
||||
poly_x.push_back(x1);
|
||||
poly_y.push_back(y1);
|
||||
}
|
||||
for (int k=1; k<(nbr_points-1); k++) {
|
||||
double t = k*increment;
|
||||
double t2 = t*t;
|
||||
double tr = 1.-t;
|
||||
double tr2 = tr*tr;
|
||||
double tr2t = tr*2*t;
|
||||
|
||||
// adding a point to the polyline
|
||||
poly_x.push_back( tr2*x1 + tr2t*x2 + t2*x3);
|
||||
poly_y.push_back( tr2*y1 + tr2t*y2 + t2*y3);
|
||||
}
|
||||
// adding the last point of the sub-curve
|
||||
poly_x.push_back(x3);
|
||||
poly_y.push_back(y3);
|
||||
}
|
||||
|
||||
void Curve::fillHash() {
|
||||
hash = new unsigned short int[hashSize+2];
|
||||
|
||||
unsigned int polyIter = 0;
|
||||
double const increment = 1./hashSize;
|
||||
double milestone = 0.;
|
||||
|
||||
for (unsigned int i=0; i<(hashSize+1);) {
|
||||
while(poly_x[polyIter] <= milestone) ++polyIter;
|
||||
hash[i] = polyIter-1;
|
||||
milestone = (++i)*increment;
|
||||
}
|
||||
hash[hashSize+1] = poly_x.size()-1;
|
||||
|
||||
/*
|
||||
// Debug output to file
|
||||
FILE* f = fopen ("hash.txt", "wt");
|
||||
for (int i=0; i<(hashSize+2); i++)
|
||||
fprintf (f, "%d: %d > %.6f, %.6f\n", i, hash[i], poly_x[hash[i]], poly_y[hash[i]]);
|
||||
fprintf (f, "\nppn: %d\npoly_x: %d\n", ppn, poly_x.size());
|
||||
fclose (f);
|
||||
*/
|
||||
}
|
||||
|
||||
void fillCurveArray(DiagonalCurve* diagCurve, LUTf &outCurve, int skip, bool needed) {
|
||||
if (needed) {
|
||||
LUTf lutCurve (65536);
|
||||
|
||||
for (int i=0; i<=0xffff; i+= i<0xffff-skip ? skip : 1 ) {
|
||||
// change to [0,1] range
|
||||
double val = (double)i / 65535.0;
|
||||
// apply custom/parametric/NURBS curve, if any
|
||||
val = diagCurve->getVal (val);
|
||||
// store result in a temporary array
|
||||
lutCurve[i] = (val);
|
||||
}
|
||||
|
||||
// if skip>1, let apply linear interpolation in the skipped points of the curve
|
||||
if (skip > 1) {
|
||||
int prev = 0;
|
||||
for (int i=1; i<=0xffff-skip; i++) {
|
||||
if (i%skip==0) {
|
||||
prev+=skip;
|
||||
continue;
|
||||
}
|
||||
lutCurve[i] = ( lutCurve[prev] * (skip - i%skip) + lutCurve[prev+skip] * (i%skip) ) / skip;
|
||||
}
|
||||
}
|
||||
|
||||
for (int i=0; i<=0xffff; i++) {
|
||||
outCurve[i] = (65535.0 * lutCurve[i]);
|
||||
}
|
||||
}
|
||||
else {
|
||||
for (int i=0; i<=0xffff; i++) {
|
||||
outCurve[i] = (float)i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
void CurveFactory::complexsgnCurve (double saturation, bool satlimit, double satlimthresh, \
|
||||
const std::vector<double>& acurvePoints, const std::vector<double>& bcurvePoints, \
|
||||
LUTf & aoutCurve, LUTf & boutCurve, LUTf & satCurve, int skip) {
|
||||
|
||||
//colormult = chroma_scale for Lab manipulations
|
||||
|
||||
//-----------------------------------------------------
|
||||
|
||||
bool needed;
|
||||
DiagonalCurve* dCurve = NULL;
|
||||
|
||||
// check if contrast curve is needed
|
||||
needed = (saturation<-0.0001 || saturation>0.0001);
|
||||
|
||||
// Filling the curve if needed
|
||||
if (needed) {
|
||||
|
||||
//%%%%%%%%%%%%%%%%% Saturation curve's control points %%%%%%%%%%%%%%%%%
|
||||
std::vector<double> satcurvePoints;
|
||||
satcurvePoints.push_back((double)DCT_NURBS);
|
||||
if (saturation>0) {
|
||||
double satslope = (0.5+2*saturation/500.0)/(0.5-2*saturation/500.0);
|
||||
double scale = (satlimthresh/100.1);
|
||||
if (!satlimit) scale=100/100.1;
|
||||
|
||||
satcurvePoints.push_back(0); //black point. Value in [0 ; 1] range
|
||||
satcurvePoints.push_back(0); //black point. Value in [0 ; 1] range
|
||||
|
||||
//if (satlimit) {
|
||||
satcurvePoints.push_back(0.5-0.5*scale); //toe point
|
||||
satcurvePoints.push_back(0.5-0.5*scale); //value at toe point
|
||||
|
||||
satcurvePoints.push_back(0.5-(0.5/satslope)*scale); //toe point
|
||||
satcurvePoints.push_back(0.5-0.5*scale); //value at toe point
|
||||
|
||||
satcurvePoints.push_back(0.5+(0.5/satslope)*scale); //shoulder point
|
||||
satcurvePoints.push_back(0.5+0.5*scale); //value at shoulder point
|
||||
|
||||
satcurvePoints.push_back(0.5+0.5*scale); //shoulder point
|
||||
satcurvePoints.push_back(0.5+0.5*scale); //value at shoulder point
|
||||
/*} else {
|
||||
satcurvePoints.push_back(0.25+saturation/500.0); //toe point
|
||||
satcurvePoints.push_back(0.25-saturation/500.0); //value at toe point
|
||||
|
||||
satcurvePoints.push_back(0.75-saturation/500.0); //shoulder point
|
||||
satcurvePoints.push_back(0.75+saturation/500.0); //value at shoulder point
|
||||
}*/
|
||||
|
||||
satcurvePoints.push_back(1); // white point
|
||||
satcurvePoints.push_back(1); // value at white point
|
||||
} else {
|
||||
satcurvePoints.push_back(0);
|
||||
satcurvePoints.push_back(-(saturation/200.0));
|
||||
|
||||
satcurvePoints.push_back(1);
|
||||
satcurvePoints.push_back(1+saturation/200.0);
|
||||
}
|
||||
dCurve = new DiagonalCurve (satcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
fillCurveArray(dCurve, satCurve, skip, needed);
|
||||
|
||||
delete dCurve;
|
||||
dCurve = NULL;
|
||||
}
|
||||
else {
|
||||
fillCurveArray(NULL, satCurve, skip, needed);
|
||||
}
|
||||
|
||||
//-----------------------------------------------------
|
||||
|
||||
needed = false;
|
||||
// create a curve if needed
|
||||
if (acurvePoints.size()>0 && acurvePoints[0]!=0) {
|
||||
dCurve = new DiagonalCurve (acurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
if (dCurve && !dCurve->isIdentity())
|
||||
needed = true;
|
||||
}
|
||||
fillCurveArray(dCurve, aoutCurve, skip, needed);
|
||||
if (dCurve) {
|
||||
delete dCurve;
|
||||
dCurve = NULL;
|
||||
}
|
||||
|
||||
//-----------------------------------------------------
|
||||
|
||||
needed = false;
|
||||
if (bcurvePoints.size()>0 && bcurvePoints[0]!=0) {
|
||||
dCurve = new DiagonalCurve (bcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
if (dCurve && !dCurve->isIdentity())
|
||||
needed = true;
|
||||
}
|
||||
fillCurveArray(dCurve, boutCurve, skip, needed);
|
||||
if (dCurve) {
|
||||
delete dCurve;
|
||||
dCurve = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
void CurveFactory::complexCurve (double ecomp, double black, double hlcompr, double hlcomprthresh, \
|
||||
double shcompr, double br, double contr, double gamma_, bool igamma_, \
|
||||
const std::vector<double>& curvePoints, LUTu & histogram, LUTu & histogramCropped, \
|
||||
LUTf & hlCurve, LUTf & shCurve, LUTf & outCurve, \
|
||||
LUTu & outBeforeCCurveHistogram, int skip) {
|
||||
|
||||
|
||||
//double def_mul = pow (2.0, defmul);
|
||||
|
||||
/*printf ("def_mul= %f ecomp= %f black= %f hlcompr= %f shcompr= %f br= %f contr= %f defmul= %f \
|
||||
gamma= %f, skip= %d \n",def_mul,ecomp,black,hlcompr,shcompr,br,contr,defmul,gamma_,skip);*/
|
||||
|
||||
// compute parameters of the gamma curve
|
||||
/*double start = exp(gamma_*log( -0.099 / ((1.0/gamma_-1.0)*1.099 )));
|
||||
double slope = 1.099 * pow (start, 1.0/gamma_-1) - 0.099/start;
|
||||
double mul = 1.099;
|
||||
double add = 0.099;
|
||||
// gamma BT709*/
|
||||
|
||||
//normalize gamma to sRGB
|
||||
double start = exp(gamma_*log( -0.055 / ((1.0/gamma_-1.0)*1.055 )));
|
||||
double slope = 1.055 * pow (start, 1.0/gamma_-1) - 0.055/start;
|
||||
double mul = 1.055;
|
||||
double add = 0.055;
|
||||
|
||||
// a: slope of the curve, black: starting point at the x axis
|
||||
double a = pow (2.0, ecomp);
|
||||
|
||||
// curve without contrast
|
||||
LUTf dcurve(0x10000);
|
||||
|
||||
// check if inverse gamma is needed at the end
|
||||
bool needigamma = igamma_ && gamma_>0;
|
||||
|
||||
// clear array that stores histogram valid before applying the custom curve
|
||||
outBeforeCCurveHistogram.clear();
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
// tone curve base. a: slope (from exp.comp.), b: black, def_mul: max. x value (can be>1), hr,sr: highlight,shadow recovery
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
DiagonalCurve* brightcurve = NULL;
|
||||
|
||||
// check if brightness curve is needed
|
||||
if (br>0.00001 || br<-0.00001) {
|
||||
|
||||
std::vector<double> brightcurvePoints;
|
||||
brightcurvePoints.push_back((double)DCT_NURBS);
|
||||
|
||||
brightcurvePoints.push_back(0.); //black point. Value in [0 ; 1] range
|
||||
brightcurvePoints.push_back(0.); //black point. Value in [0 ; 1] range
|
||||
|
||||
if(br>0) {
|
||||
brightcurvePoints.push_back(0.1); //toe point
|
||||
brightcurvePoints.push_back(0.1+br/150.0); //value at toe point
|
||||
|
||||
brightcurvePoints.push_back(0.7); //shoulder point
|
||||
brightcurvePoints.push_back(MIN(1.0,0.7+br/300.0)); //value at shoulder point
|
||||
} else {
|
||||
brightcurvePoints.push_back(MAX(0.0,0.1-br/150.0)); //toe point
|
||||
brightcurvePoints.push_back(0.1); //value at toe point
|
||||
|
||||
brightcurvePoints.push_back(0.7-br/300.0); //shoulder point
|
||||
brightcurvePoints.push_back(0.7); //value at shoulder point
|
||||
}
|
||||
brightcurvePoints.push_back(1.); // white point
|
||||
brightcurvePoints.push_back(1.); // value at white point
|
||||
|
||||
brightcurve = new DiagonalCurve (brightcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
}
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
float exp_scale = a;
|
||||
float scale = 65536.0;
|
||||
float comp = (MAX(0,ecomp) + 1.0)*hlcompr/100.0;
|
||||
float shoulder = ((scale/MAX(1,exp_scale))*(hlcomprthresh/200.0))+0.1;
|
||||
//printf("shoulder = %e\n",shoulder);
|
||||
//printf ("exp_scale= %f comp= %f def_mul=%f a= %f \n",exp_scale,comp,def_mul,a);
|
||||
|
||||
for (int i=0; i<0x10000; i++) {
|
||||
|
||||
// change to [0,1] range
|
||||
float val = (float)i-shoulder;
|
||||
|
||||
if (comp>0.0)
|
||||
{
|
||||
if (val>0.0) {
|
||||
float R = val*comp/(scale-shoulder);
|
||||
hlCurve[i] = log(1.0+R*exp_scale)/R;
|
||||
} else {
|
||||
hlCurve[i]=exp_scale;
|
||||
}
|
||||
} else {
|
||||
hlCurve[i]=exp_scale;
|
||||
}
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
// change to [0,1] range
|
||||
if (i!=0) {
|
||||
val = (float)i / 65535.0f;
|
||||
} else {
|
||||
val = 1.0/65535.0;
|
||||
}
|
||||
|
||||
float val2 = basecurve (val, 1.0, black, 1.0, 0.0, 1.5*shcompr/100.0);
|
||||
shCurve[i] = CLIPD(val2)/val;
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
// change to [0,1] range
|
||||
val = (double)i / 65535.0;
|
||||
|
||||
// gamma correction
|
||||
if (gamma_>1)
|
||||
val = Color::gamma (val, gamma_, start, slope, mul, add);
|
||||
|
||||
// apply brightness curve
|
||||
if (brightcurve)
|
||||
val = brightcurve->getVal (val); // TODO: getVal(double) is very slow! Optimize with a LUTf
|
||||
|
||||
// store result in a temporary array
|
||||
dcurve[i] = CLIPD(val);
|
||||
}
|
||||
|
||||
if (brightcurve)
|
||||
delete brightcurve;
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// check if contrast curve is needed
|
||||
if (contr>0.00001 || contr<-0.00001) {
|
||||
|
||||
// compute mean luminance of the image with the curve applied
|
||||
int sum = 0;
|
||||
float avg = 0;
|
||||
//double sqavg = 0;
|
||||
for (int i=0; i<=0xffff; i++) {
|
||||
float fi=i;
|
||||
fi = hlCurve[fi]*fi;
|
||||
avg += dcurve[(int)(shCurve[fi]*fi)] * histogram[i];
|
||||
//sqavg += dcurve[i]*dcurve[i] * histogram[i];
|
||||
sum += histogram[i];
|
||||
}
|
||||
avg /= sum;
|
||||
//sqavg /= sum;
|
||||
//double stddev = sqrt(sqavg-avg*avg);
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
std::vector<double> contrastcurvePoints;
|
||||
contrastcurvePoints.push_back((double)DCT_NURBS);
|
||||
|
||||
contrastcurvePoints.push_back(0); //black point. Value in [0 ; 1] range
|
||||
contrastcurvePoints.push_back(0); //black point. Value in [0 ; 1] range
|
||||
|
||||
contrastcurvePoints.push_back(avg-avg*(0.6-contr/250.0)); //toe point
|
||||
contrastcurvePoints.push_back(avg-avg*(0.6+contr/250.0)); //value at toe point
|
||||
|
||||
contrastcurvePoints.push_back(avg+(1-avg)*(0.6-contr/250.0)); //shoulder point
|
||||
contrastcurvePoints.push_back(avg+(1-avg)*(0.6+contr/250.0)); //value at shoulder point
|
||||
|
||||
contrastcurvePoints.push_back(1); // white point
|
||||
contrastcurvePoints.push_back(1); // value at white point
|
||||
|
||||
DiagonalCurve* contrastcurve = new DiagonalCurve (contrastcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// apply contrast enhancement
|
||||
for (int i=0; i<=0xffff; i++) {
|
||||
dcurve[i] = contrastcurve->getVal (dcurve[i]);
|
||||
}
|
||||
|
||||
delete contrastcurve;
|
||||
}
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// create a curve if needed
|
||||
bool histNeeded = false;
|
||||
DiagonalCurve* tcurve = NULL;
|
||||
if (curvePoints.size()>0 && curvePoints[0]!=0) {
|
||||
tcurve = new DiagonalCurve (curvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
if (outBeforeCCurveHistogram /*&& histogramCropped*/)
|
||||
histNeeded = true;
|
||||
}
|
||||
if (tcurve && tcurve->isIdentity()) {
|
||||
delete tcurve;
|
||||
tcurve = NULL;
|
||||
}
|
||||
|
||||
for (int i=0; i<=0xffff; i++) {
|
||||
float val;
|
||||
|
||||
if (histNeeded) {
|
||||
float fi=i;
|
||||
float hval = hlCurve[i]*fi;
|
||||
hval = dcurve[shCurve[hval]*hval];
|
||||
//if (needigamma)
|
||||
// hval = igamma2 (hval);
|
||||
int hi = (int)(255.0*(hval));
|
||||
outBeforeCCurveHistogram[hi] += histogram/*Cropped*/[i] ;
|
||||
}
|
||||
|
||||
// apply custom/parametric/NURBS curve, if any
|
||||
if (tcurve) {
|
||||
val = tcurve->getVal (dcurve[i]); // TODO: getVal(double) is very slow! Optimize with a LUTf
|
||||
} else {
|
||||
val = (dcurve[i]);
|
||||
}
|
||||
|
||||
// if inverse gamma is needed, do it (standard sRGB inverse gamma is applied)
|
||||
if (needigamma)
|
||||
val = Color::igamma (val, gamma_, start, slope, mul, add);
|
||||
|
||||
outCurve[i] = (65535.0 * val);
|
||||
}
|
||||
|
||||
if (tcurve)
|
||||
delete tcurve;
|
||||
|
||||
/*if (outBeforeCCurveHistogram) {
|
||||
for (int i=0; i<256; i++) printf("i= %d bchist= %d \n",i,outBeforeCCurveHistogram[i]);
|
||||
}*/
|
||||
}
|
||||
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
void CurveFactory::complexLCurve (double br, double contr, const std::vector<double>& curvePoints, \
|
||||
LUTu & histogram, LUTu & histogramCropped, LUTf & outCurve, \
|
||||
LUTu & outBeforeCCurveHistogram, int skip) {
|
||||
|
||||
// curve without contrast
|
||||
LUTf dcurve(65536,0);
|
||||
|
||||
// clear array that stores histogram valid before applying the custom curve
|
||||
if (outBeforeCCurveHistogram)
|
||||
outBeforeCCurveHistogram.clear();
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
// tone curve base. a: slope (from exp.comp.), b: black, def_mul: max. x value (can be>1), hr,sr: highlight,shadow recovery
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// check if brightness curve is needed
|
||||
if (br>0.00001 || br<-0.00001) {
|
||||
|
||||
std::vector<double> brightcurvePoints;
|
||||
brightcurvePoints.push_back((double)((CurveType)DCT_NURBS));
|
||||
|
||||
brightcurvePoints.push_back(0.); // black point. Value in [0 ; 1] range
|
||||
brightcurvePoints.push_back(0.); // black point. Value in [0 ; 1] range
|
||||
|
||||
if (br>0) {
|
||||
brightcurvePoints.push_back(0.1); // toe point
|
||||
brightcurvePoints.push_back(0.1+br/150.0); //value at toe point
|
||||
|
||||
brightcurvePoints.push_back(0.7); // shoulder point
|
||||
brightcurvePoints.push_back(MIN(1.0,0.7+br/300.0)); //value at shoulder point
|
||||
} else {
|
||||
brightcurvePoints.push_back(0.1-br/150.0); // toe point
|
||||
brightcurvePoints.push_back(0.1); // value at toe point
|
||||
|
||||
brightcurvePoints.push_back(MIN(1.0,0.7-br/300.0)); // shoulder point
|
||||
brightcurvePoints.push_back(0.7); // value at shoulder point
|
||||
}
|
||||
brightcurvePoints.push_back(1.); // white point
|
||||
brightcurvePoints.push_back(1.); // value at white point
|
||||
|
||||
DiagonalCurve* brightcurve = new DiagonalCurve (brightcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// Applying brightness curve
|
||||
for (int i=0; i<32768; i++) { // L values range up to 32767, higher values are for highlight overflow
|
||||
|
||||
// change to [0,1] range
|
||||
float val = (float)i / 32767.0;
|
||||
|
||||
// apply brightness curve
|
||||
val = brightcurve->getVal (val);
|
||||
|
||||
// store result in a temporary array
|
||||
dcurve[i] = CLIPD(val);
|
||||
}
|
||||
delete brightcurve;
|
||||
}
|
||||
else {
|
||||
for (int i=0; i<32768; i++) { // L values range up to 32767, higher values are for highlight overflow
|
||||
// set the identity curve in the temporary array
|
||||
dcurve[i] = (float)i / 32767.0;
|
||||
}
|
||||
}
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// check if contrast curve is needed
|
||||
if (contr>0.00001 || contr<-0.00001) {
|
||||
|
||||
// compute mean luminance of the image with the curve applied
|
||||
int sum = 0;
|
||||
float avg = 0;
|
||||
//float sqavg = 0;
|
||||
for (int i=0; i<32768; i++) {
|
||||
avg += dcurve[i] * histogram[i];
|
||||
//sqavg += dcurve[i]*dcurve[i] * histogram[i];
|
||||
sum += histogram[i];
|
||||
}
|
||||
avg /= sum;
|
||||
//sqavg /= sum;
|
||||
//float stddev = sqrt(sqavg-avg*avg);
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
std::vector<double> contrastcurvePoints;
|
||||
contrastcurvePoints.push_back((double)((CurveType)DCT_NURBS));
|
||||
|
||||
contrastcurvePoints.push_back(0.); // black point. Value in [0 ; 1] range
|
||||
contrastcurvePoints.push_back(0.); // black point. Value in [0 ; 1] range
|
||||
|
||||
contrastcurvePoints.push_back(avg-avg*(0.6-contr/250.0)); // toe point
|
||||
contrastcurvePoints.push_back(avg-avg*(0.6+contr/250.0)); // value at toe point
|
||||
|
||||
contrastcurvePoints.push_back(avg+(1-avg)*(0.6-contr/250.0)); // shoulder point
|
||||
contrastcurvePoints.push_back(avg+(1-avg)*(0.6+contr/250.0)); // value at shoulder point
|
||||
|
||||
contrastcurvePoints.push_back(1.); // white point
|
||||
contrastcurvePoints.push_back(1.); // value at white point
|
||||
|
||||
DiagonalCurve* contrastcurve = new DiagonalCurve (contrastcurvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// apply contrast enhancement
|
||||
for (int i=0; i<32768; i++) {
|
||||
dcurve[i] = contrastcurve->getVal (dcurve[i]);
|
||||
}
|
||||
|
||||
delete contrastcurve;
|
||||
}
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// create a curve if needed
|
||||
DiagonalCurve* tcurve = NULL;
|
||||
bool histNeeded = false;
|
||||
if (curvePoints.size()>0 && curvePoints[0]!=0) {
|
||||
tcurve = new DiagonalCurve (curvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
if (outBeforeCCurveHistogram /*&& histogramCropped*/)
|
||||
histNeeded = true;
|
||||
}
|
||||
if (tcurve && tcurve->isIdentity()) {
|
||||
delete tcurve;
|
||||
tcurve = NULL;
|
||||
}
|
||||
|
||||
if (tcurve) {
|
||||
// L values go up to 32767, last stop is for highlight overflow
|
||||
for (int i=0; i<32768; i++) {
|
||||
float val;
|
||||
|
||||
if (histNeeded) {
|
||||
float hval = dcurve[i];
|
||||
int hi = (int)(255.0*CLIPD(hval));
|
||||
outBeforeCCurveHistogram[hi]+=histogram/*Cropped*/[i] ;
|
||||
}
|
||||
|
||||
// apply custom/parametric/NURBS curve, if any
|
||||
val = tcurve->getVal (dcurve[i]);
|
||||
|
||||
outCurve[i] = (32767.0 * val);
|
||||
}
|
||||
}
|
||||
else {
|
||||
// Skip the slow getval method if no curve is used (or an identity curve)
|
||||
// L values go up to 32767, last stop is for highlight overflow
|
||||
for (int i=0; i<32768; i++) {
|
||||
if (histNeeded) {
|
||||
float hval = dcurve[i];
|
||||
int hi = (int)(255.0*CLIPD(hval));
|
||||
outBeforeCCurveHistogram[hi]+=histogram/*Cropped*/[i] ;
|
||||
}
|
||||
|
||||
outCurve[i] = 32767.0*dcurve[i];
|
||||
}
|
||||
}
|
||||
for (int i=32768; i<65536; i++) outCurve[i]=(float)i;
|
||||
|
||||
if (tcurve)
|
||||
delete tcurve;
|
||||
|
||||
/*if (outBeforeCCurveHistogram) {
|
||||
for (int i=0; i<256; i++) printf("i= %d bchist= %d \n",i,outBeforeCCurveHistogram[i]);
|
||||
}*/
|
||||
}
|
||||
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
void CurveFactory::RGBCurve (const std::vector<double>& curvePoints, LUTf & outCurve, int skip) {
|
||||
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
// create a curve if needed
|
||||
DiagonalCurve* tcurve = NULL;
|
||||
bool histNeeded = false;
|
||||
if (curvePoints.size()>0 && curvePoints[0]!=0) {
|
||||
tcurve = new DiagonalCurve (curvePoints, CURVES_MIN_POLY_POINTS/skip);
|
||||
}
|
||||
if (tcurve && tcurve->isIdentity()) {
|
||||
delete tcurve;
|
||||
tcurve = NULL;
|
||||
}
|
||||
|
||||
if (tcurve) {
|
||||
for (int i=0; i<65536; i++) {
|
||||
// apply custom/parametric/NURBS curve, if any
|
||||
float val = tcurve->getVal ((float)i/65536.0f);
|
||||
outCurve[i] = (65536.0f * val);
|
||||
}
|
||||
}
|
||||
else {
|
||||
// Skip the slow getval method if no curve is used (or an identity curve)
|
||||
for (int i=0; i<65536; i++) {
|
||||
outCurve[i] = i;
|
||||
}
|
||||
}
|
||||
|
||||
if (tcurve)
|
||||
delete tcurve;
|
||||
|
||||
}
|
||||
|
||||
|
||||
//%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user