opencv/modules/photo/src/fast_nlmeans_denoising_invoker_commons.hpp

/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                        Intel License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000, Intel Corporation, all rights reserved.
// Third party copyrights are property of their respective icvers.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of Intel Corporation may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#ifndef __OPENCV_FAST_NLMEANS_DENOISING_INVOKER_COMMONS_HPP__
#define __OPENCV_FAST_NLMEANS_DENOISING_INVOKER_COMMONS_HPP__

using namespace cv;

template <typename T> struct pixelInfo_
{
    static const int channels = 1;
    typedef T sampleType;
};

template <typename ET, int n> struct pixelInfo_<Vec<ET, n> >
{
    static const int channels = n;
    typedef ET sampleType;
};

template <typename T> struct pixelInfo: public pixelInfo_<T>
{
    using typename pixelInfo_<T>::sampleType;

    static inline sampleType sampleMax()
    {
        return std::numeric_limits<sampleType>::max();
    }

    static inline sampleType sampleMin()
    {
        return std::numeric_limits<sampleType>::min();
    }

    static inline size_t sampleBytes()
    {
        return sizeof(sampleType);
    }

    static inline size_t sampleBits()
    {
        return 8*sampleBytes();
    }
};

class DistAbs
{
    template <typename T, typename IT> struct calcDist_
    {
        static inline IT f(const T a, const T b)
        {
            return std::abs((IT)(a-b));
        }
    };

    template <typename ET, typename IT> struct calcDist_<Vec<ET, 2>, IT>
    {
        static inline IT f(const Vec<ET, 2> a, const Vec<ET, 2> b)
        {
            return std::abs((IT)(a[0]-b[0])) + std::abs((IT)(a[1]-b[1]));
        }
    };

    template <typename ET, typename IT> struct calcDist_<Vec<ET, 3>, IT>
    {
        static inline IT f(const Vec<ET, 3> a, const Vec<ET, 3> b)
        {
            return
                std::abs((IT)(a[0]-b[0])) +
                std::abs((IT)(a[1]-b[1])) +
                std::abs((IT)(a[2]-b[2]));
        }
    };

public:
    template <typename T, typename IT> static inline IT calcDist(const T a, const T b)
    {
        return calcDist_<T, IT>::f(a, b);
    }

    template <typename T, typename IT>
    static inline IT calcDist(const Mat& m, int i1, int j1, int i2, int j2)
    {
        const T a = m.at<T>(i1, j1);
        const T b = m.at<T>(i2, j2);
        return calcDist<T, IT>(a,b);
    }

    template <typename T, typename IT>
    static inline IT calcUpDownDist(T a_up, T a_down, T b_up, T b_down)
    {
        return calcDist<T, IT>(a_down, b_down) - calcDist<T, IT>(a_up, b_up);
    };

    template <typename T>
    static double calcWeight(double dist, double h)
    {
        return std::exp(-dist*dist / (h * h * pixelInfo<T>::channels));
    }

    template <typename T, typename IT>
    static double maxDist()
    {
        return (IT)pixelInfo<T>::sampleMax() * (IT)pixelInfo<T>::channels;
    }
};

class DistSquared
{
    template <typename T, typename IT> struct calcDist_
    {
        static inline IT f(const T a, const T b)
        {
            return (IT)(a-b) * (IT)(a-b);
        }
    };

    template <typename ET, typename IT> struct calcDist_<Vec<ET, 2>, IT>
    {
        static inline IT f(const Vec<ET, 2> a, const Vec<ET, 2> b)
        {
            return (IT)(a[0]-b[0])*(IT)(a[0]-b[0]) + (IT)(a[1]-b[1])*(IT)(a[1]-b[1]);
        }
    };

    template <typename ET, typename IT> struct calcDist_<Vec<ET, 3>, IT>
    {
        static inline IT f(const Vec<ET, 3> a, const Vec<ET, 3> b)
        {
            return
                (IT)(a[0]-b[0])*(IT)(a[0]-b[0]) +
                (IT)(a[1]-b[1])*(IT)(a[1]-b[1]) +
                (IT)(a[2]-b[2])*(IT)(a[2]-b[2]);
        }
    };

    template <typename T, typename IT> struct calcUpDownDist_
    {
        static inline IT f(T a_up, T a_down, T b_up, T b_down)
        {
            IT A = a_down - b_down;
            IT B = a_up - b_up;
            return (A-B)*(A+B);
        }
    };

    template <typename ET, int n, typename IT> struct calcUpDownDist_<Vec<ET, n>, IT>
    {
    private:
        typedef Vec<ET, n> T;
    public:
        static inline IT f(T a_up, T a_down, T b_up, T b_down)
        {
            return calcDist<T, IT>(a_down, b_down) - calcDist<T, IT>(a_up, b_up);
        }
    };

public:
    template <typename T, typename IT> static inline IT calcDist(const T a, const T b)
    {
        return calcDist_<T, IT>::f(a, b);
    }

    template <typename T, typename IT>
    static inline IT calcDist(const Mat& m, int i1, int j1, int i2, int j2)
    {
        const T a = m.at<T>(i1, j1);
        const T b = m.at<T>(i2, j2);
        return calcDist<T, IT>(a,b);
    }

    template <typename T, typename IT>
    static inline IT calcUpDownDist(T a_up, T a_down, T b_up, T b_down)
    {
        return calcUpDownDist_<T, IT>::f(a_up, a_down, b_up, b_down);
    };

    template <typename T>
    static double calcWeight(double dist, double h)
    {
        return std::exp(-dist / (h * h * pixelInfo<T>::channels));
    }

    template <typename T, typename IT>
    static double maxDist()
    {
        return (IT)pixelInfo<T>::sampleMax() * (IT)pixelInfo<T>::sampleMax() *
            (IT)pixelInfo<T>::channels;
    }
};

template <typename T, typename IT> struct incWithWeight_
{
    static inline void f(IT* estimation, IT weight, T p)
    {
        estimation[0] += weight * p;
    }
};

template <typename ET, typename IT> struct incWithWeight_<Vec<ET, 2>, IT>
{
    static inline void f(IT* estimation, IT weight, Vec<ET, 2> p)
    {
        estimation[0] += weight * p[0];
        estimation[1] += weight * p[1];
    }
};

template <typename ET, typename IT> struct incWithWeight_<Vec<ET, 3>, IT>
{
    static inline void f(IT* estimation, IT weight, Vec<ET, 3> p)
    {
        estimation[0] += weight * p[0];
        estimation[1] += weight * p[1];
        estimation[2] += weight * p[2];
    }
};

template <typename T, typename IT>
static inline void incWithWeight(IT* estimation, IT weight, T p)
{
    return incWithWeight_<T, IT>::f(estimation, weight, p);
}

template <typename T, typename IT> struct saturateCastFromArray_
{
    static inline T f(IT* estimation)
    {
        return saturate_cast<T>(estimation[0]);
    }
};

template <typename ET, typename IT> struct saturateCastFromArray_<Vec<ET, 2>, IT>
{
    static inline Vec<ET, 2> f(IT* estimation)
    {
        Vec<ET, 2> res;
        res[0] = saturate_cast<ET>(estimation[0]);
        res[1] = saturate_cast<ET>(estimation[1]);
        return res;
    }
};

template <typename ET, typename IT> struct saturateCastFromArray_<Vec<ET, 3>, IT>
{
    static inline Vec<ET, 3> f(IT* estimation)
    {
        Vec<ET, 3> res;
        res[0] = saturate_cast<ET>(estimation[0]);
        res[1] = saturate_cast<ET>(estimation[1]);
        res[2] = saturate_cast<ET>(estimation[2]);
        return res;
    }
};

template <typename T, typename IT> static inline T saturateCastFromArray(IT* estimation)
{
    return saturateCastFromArray_<T, IT>::f(estimation);
}

#endif