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

fixed crashes in histogram matching

Browse Source
This commit is contained in:
Alberto Griggio
2018-01-18 14:17:51 +01:00
parent 931ee95620
commit abfeb4ca1d
1 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
rtengine/histmatching.cc
View File

@@ -96,31 +96,35 @@ void mappingToCurve(const std::vector<int> &mapping, std::vector<double> &curve)
auto coord = [](int v) -> double { return double(v)/255.0; }; auto coord = [](int v) -> double { return double(v)/255.0; };
auto doit = auto doit =
[&](int start, int stop, int step) -> void [&](int start, int stop, int step, bool addstart) -> void
{ {
int prev = start; int prev = start;
if (addstart) {
curve.push_back(coord(start));
curve.push_back(coord(mapping[start]));
}
for (int i = start; i < stop; ++i) { for (int i = start; i < stop; ++i) {
int v = mapping[i]; int v = mapping[i];
bool change = i > 0 && v != mapping[i-1]; bool change = i > 0 && v != mapping[i-1];
int diff = i - prev; int diff = i - prev;
if (change && std::abs(diff - step) <= 1) { if (change && std::abs(diff - step) <= 1) {
curve.emplace_back(coord(i)); curve.push_back(coord(i));
curve.emplace_back(coord(v)); curve.push_back(coord(v));
prev = i; prev = i;
} }
} }
}; };
doit(0, idx, idx > step ? step : idx / 2); doit(0, idx, idx > step ? step : idx / 2, true);
doit(idx, int(mapping.size()), step); doit(idx, int(mapping.size()), step, idx - step > step / 2);
if (curve[1] > 0.01) { if (curve.size() <= 2 || curve.back() < 0.99 || (1 - curve[curve.size()-2] > step / 512.0 && curve.back() < coord(mapping.back()))) {
curve.insert(curve.begin(), 0.0);
curve.insert(curve.begin(), 0.0);
}
if (curve.back() < 0.99 || (1 - curve[curve.size()-2] > step / 512.0 && curve.back() < coord(mapping.back()))) {
curve.emplace_back(1.0); curve.emplace_back(1.0);
curve.emplace_back(coord(mapping.back())); curve.emplace_back(coord(mapping.back()));
} }
if (curve.size() < 4) {
curve = { DCT_Linear }; // not enough points, fall back to linear
} else {
curve.insert(curve.begin(), DCT_Spline); curve.insert(curve.begin(), DCT_Spline);
}
} }
} // namespace } // namespace
@@ -134,6 +138,8 @@ void RawImageSource::getAutoMatchedToneCurve(std::vector<double> &outCurve)
std::cout << "performing histogram matching for " << getFileName() << " on the embedded thumbnail" << std::endl; std::cout << "performing histogram matching for " << getFileName() << " on the embedded thumbnail" << std::endl;
} }
outCurve = { DCT_Linear };
ProcParams neutral; ProcParams neutral;
std::unique_ptr<IImage8> target; std::unique_ptr<IImage8> target;
{ {
@@ -168,6 +174,12 @@ void RawImageSource::getAutoMatchedToneCurve(std::vector<double> &outCurve)
eSensorType sensor_type; eSensorType sensor_type;
int w, h; int w, h;
std::unique_ptr<Thumbnail> thumb(Thumbnail::loadQuickFromRaw(getFileName(), rml, sensor_type, w, h, 1, false, true)); std::unique_ptr<Thumbnail> thumb(Thumbnail::loadQuickFromRaw(getFileName(), rml, sensor_type, w, h, 1, false, true));
if (!thumb) {
if (settings->verbose) {
std::cout << "histogram matching: no thumbnail found, generating a neutral curve" << std::endl;
}
return;
}
source.reset(thumb->quickProcessImage(neutral, target->getHeight(), TI_Nearest)); source.reset(thumb->quickProcessImage(neutral, target->getHeight(), TI_Nearest));
if (settings->verbose) { if (settings->verbose) {
@@ -186,7 +198,7 @@ void RawImageSource::getAutoMatchedToneCurve(std::vector<double> &outCurve)
int j = 0; int j = 0;
for (size_t i = 0; i < tcdf.size(); ++i) { for (size_t i = 0; i < tcdf.size(); ++i) {
j = findMatch(tcdf[i], scdf, j); j = findMatch(tcdf[i], scdf, j);
mapping.emplace_back(j); mapping.push_back(j);
} }
mappingToCurve(mapping, outCurve); mappingToCurve(mapping, outCurve);