filter_radial_basis_function.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_FILTERING_FILTER_RADIAL_BASIS_FUNCTION
#define PIC_FILTERING_FILTER_RADIAL_BASIS_FUNCTION

namespace pic {

#include "../algorithms/radial_basis_function.hpp"

#include "../filtering/filter.hpp"

class FilterRadialBasisFunction: public Filter
{
protected:

    RadialBasisFunction *rbf;

    void ProcessBBox(Image *dst, ImageVec src, BBox *box)
    {
        if(rbf == NULL) {
            return;
        }

        int channels = src[0]->channels;

        if(rbf->nDim != channels) {
            return;
        }

        for(int j = box->y0; j < box->y1; j++) {

            for(int i = box->x0; i < box->x1; i++) {

                float *dataIn  = (*src[0]) (i, j);
                float *dataOut = (*dst)    (i, j);

                dataOut[0] = rbf->eval(dataIn);
            }
        }
    }

public:

    FilterRadialBasisFunction() : Filter()
    {
        rbf = NULL;
    }

    void OutputSize(ImageVec imgIn, int &width, int &height, int &channels, int &frames)
    {
        width    = imgIn[0]->width;
        height   = imgIn[0]->height;
        channels = 1;
        frames   = imgIn[0]->frames;
    }

    void update(RadialBasisFunction *rbf)
    {
        if(rbf != NULL) {
            this->rbf = rbf;
        }
    }
};

} // end namespace pic

#endif /* PIC_FILTERING_FILTER_RADIAL_BASIS_FUNCTION */