imageXCorrDiscrete_test.cpp

Go to the documentation of this file.

/** \file imageXCorrDiscrete_test.cpp
 */
#include "../../catch2/catch.hpp"
 
#include <vector>
#include <Eigen/Dense>
 
#define MX_NO_ERROR_REPORTS
 
#include "../../../include/math/func/gaussian.hpp"
#include "../../../include/improc/imageXCorrDiscrete.hpp"
#include "../../../include/improc/eigenCube.hpp"
#include "../../../include/ioutils/fits/fitsFile.hpp"
 
/** Scenario: centroiding Gaussians with center of light
 *
 * Verify center of light calculation
 *
 * \anchor tests_improc_imageUtils_imageXCorrDiscrete
 */
SCENARIO( "Verify X-Corr with center of light calculation", "[improc::imageXCorrDiscrete]" )
{
    GIVEN( "two Gaussians" )
    {
        WHEN( "1 at geometric center" )
        {
            mx::improc::eigenImage<double> im0, im2;
            im0.resize( 64, 64 );
            im2.resize( 64, 64 );
 
            mx::math::func::gaussian2D<double>( im0.data(), im0.rows(), im0.cols(), 0., 1.0, 31.5, 31.5, 2 );
            mx::math::func::gaussian2D<double>( im2.data(), im2.rows(), im2.cols(), 0., 1.0, 31.5 + 4, 31.5 + 4, 2 );
 
            double x, y;
            mx::improc::imageXCorrDiscrete<mx::improc::eigenImage<double>> xcf( 5 );
 
            mx::improc::eigenImage<double> refIm = im0.block( 10, 10, im0.rows() - 11, im0.cols() - 11 );
            xcf.refIm( refIm );
            xcf.m_peakMethod = mx::improc::xcorrPeakMethod::centroid;
 
            xcf( x, y, im2 );
 
            mx::fits::fitsFile<double> ff;
            ff.write( "refIm.fits", xcf.m_refIm );
            ff.write( "ccIm.fits", xcf.m_ccIm );
 
            std::cerr << x << " " << y << "\n";
 
            REQUIRE( fabs( x - 2 ) < 1e-8 );
            REQUIRE( fabs( y - 2 ) < 1e-8 );
        }
        WHEN( "geometric quarter" )
        {
            mx::improc::eigenImage<double> im;
            im.resize( 64, 64 );
 
            mx::math::func::gaussian2D<double>( im.data(), im.rows(), im.cols(), 0., 1.0, 15.5, 15.5, 2 );
 
            double x, y;
            mx::improc::imageCenterOfLight( x, y, im );
 
            REQUIRE( fabs( x - 15.5 ) < 1e-8 );
            REQUIRE( fabs( y - 15.5 ) < 1e-8 );
        }
    }
}