i_mt_convo.h

Go to the documentation of this file.

/*
 * $Id: i_mt_convo.h 17393 2014-08-23 20:19:14Z predoehl $
 */

#ifndef I_CPP_I_MT_CONVO_H_INCLUDED_IVILAB
#define I_CPP_I_MT_CONVO_H_INCLUDED_IVILAB

#include <m_cpp/m_convolve.h>
#include <i_cpp/i_image.h>

namespace kjb
{


class Fftw_image_convolution
{
    Fftw_convolution_2d m_convo;

public:
    Fftw_image_convolution(
        int img_num_rows,
        int img_num_cols,
        int mask_max_rows,
        int mask_max_cols,
        int fft_alg_type = FFTW_MEASURE // Macro defined in m_convolve.h
    )
    :   m_convo(img_num_rows, img_num_cols,
                                mask_max_rows, mask_max_cols, fft_alg_type)
    {}


    void set_mask(const Matrix& m)
    {
        m_convo.set_mask(m);    // simple forwarding function
    }

    void set_gaussian_mask(double sigma)
    {
        m_convo.set_gaussian_mask(sigma);    // simple forwarding function
    }

    Fftw_convolution_2d::Work_buffer allocate_work_buffer() const
    {
        return m_convo.allocate_work_buffer();    // simple forwarding function
    }


/*  ABOVE THE LINE:  NOT THREAD SAFE
 * -----------------------------------------------------------------
 *  BELOW THE LINE:  THREAD SAFE
 */

    const Fftw_convolution_2d::Sizes& get_sizes() const
    {
        return m_convo.get_sizes();
    }

    bool is_mask_set() const
    {
        return m_convo.is_mask_set();
    }

    void reflect_and_convolve(
        const Image&,
        Image&,
        Fftw_convolution_2d::Work_buffer
    ) const;

    void convolve(
        const Image&,
        Image&,
        Fftw_convolution_2d::Work_buffer
    ) const;
};


}

#endif /* I_CPP_I_MT_CONVO_H_INCLUDED_IVILAB */