bilateral_separation.hpp

Go to the documentation of this file.

/*

PICCANTE
The hottest HDR imaging library!
http://vcg.isti.cnr.it/piccante

Copyright (C) 2014
Visual Computing Laboratory - ISTI CNR
http://vcg.isti.cnr.it
First author: Francesco Banterle

This Source Code Form is subject to the terms of the Mozilla Public
License, v. 2.0. If a copy of the MPL was not distributed with this
file, You can obtain one at http://mozilla.org/MPL/2.0/.

*/

#ifndef PIC_ALGORITHMS_BILATERAL_SEPARATION_HPP
#define PIC_ALGORITHMS_BILATERAL_SEPARATION_HPP

#include "../image.hpp"
#include "../filtering/filter_bilateral_2ds.hpp"
#include "../util/math.hpp"

namespace pic {

PIC_INLINE void bilateralSeparation(Image *imgIn, ImageVec &out,
                                         float sigma_s = -1.0f,
                                         float sigma_r = 0.4f,
                                         bool bLogDomain = false)
{
    if(imgIn == NULL) {
        return;
    }

    if(!imgIn->isValid()) {
        return;
    }

    if(out.size() < 2) {
        out.push_back(NULL);
        out.push_back(NULL);
    }

    if(sigma_s <= 0.0f) {
        sigma_s = MAX(imgIn->widthf, imgIn->heightf) * 0.02f;
    }

    if(sigma_r <= 0.0f) {
        sigma_r = 0.4f;
    }

    Image *img_tmp = imgIn->clone();

    img_tmp->applyFunction(log10fPlusEpsilon);

    Image *img_flt = FilterBilateral2DS::execute(img_tmp, NULL, sigma_s, sigma_r);

    if(!bLogDomain) {
        img_flt->applyFunction(powf10fMinusEpsilon);
    }

    Image *img_detail = img_tmp;

    if(bLogDomain) {
        *img_detail -= *img_flt;
    } else {
        *img_detail = imgIn;
        *img_detail /= *img_flt;
        img_detail->removeSpecials();
    }

    out[0] = img_flt;
    out[1] = img_detail;
}

} // end namespace pic

#endif /* PIC_ALGORITHMS_BILATERAL_SEPARATION_HPP */