correlation.hpp

#pragma once
 
#if defined _WIN32
 
#  include "../_cexports/c_foundation.h"
 
#  include "../global.hpp"
#  include "../image.hpp"
#  include "../exception.hpp"
#  include "../analyze.hpp"
#  include "../rect.hpp"
#  include "../_detail/detail_pixel_list.hpp"
 
#  include <memory>
#  include <vector>
 
namespace Cvb
{
  CVB_BEGIN_INLINE_NS
 
  namespace Foundation
  {


    namespace Correlation
    {

      enum class CorrelationMethod
      {

        SumOfSquareDistances,

        CorrelationCoefficients,
        CrossCorrelation
      };
      enum class CorrelationMethod {…};


      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane,
                                              Rect<int> aoi, CorrelationMethod method)
      {
        return Internal::DoResCallObjectOut<Image>([&](void *&resimg) {
          switch (method)
          {
            case CorrelationMethod::CorrelationCoefficients:
              return CVB_CALL_CAPI(CalculateCorrelationCoefficients(
                  imagePlane.Parent().Handle(), imagePlane.Plane(), templatePlane.Parent().Handle(),
                  templatePlane.Plane(), aoi.Left(), aoi.Top(), aoi.Right(), aoi.Bottom(), resimg));
            case CorrelationMethod::CrossCorrelation:
              return CVB_CALL_CAPI(CalculateCrossCorrelations(
                  imagePlane.Parent().Handle(), imagePlane.Plane(), templatePlane.Parent().Handle(),
                  templatePlane.Plane(), aoi.Left(), aoi.Top(), aoi.Right(), aoi.Bottom(), resimg));
            case CorrelationMethod::SumOfSquareDistances:
              return CVB_CALL_CAPI(CalculateSumOfSquareDistances(
                  imagePlane.Parent().Handle(), imagePlane.Plane(), templatePlane.Parent().Handle(),
                  templatePlane.Plane(), aoi.Left(), aoi.Top(), aoi.Right(), aoi.Bottom(), resimg));
            default:
              throw std::invalid_argument("unknown correlation method");
          }
        });
      }
      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane, {…}


      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane,
                                              CorrelationMethod method)
      {
        return Correlate(imagePlane, templatePlane, imagePlane.Parent().Bounds(), method);
      }
      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane, {…}


      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane,
                                              Rect<int> aoi)
      {
        return Correlate(imagePlane, templatePlane, aoi, CorrelationMethod::CorrelationCoefficients);
      }
      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane, {…}


      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane)
      {
        return Correlate(imagePlane, templatePlane, imagePlane.Parent().Bounds());
      }
      inline std::unique_ptr<Image> Correlate(const ImagePlane &imagePlane, const ImagePlane &templatePlane) {…}


      inline std::vector<LocalMaximum> FindMatches(const ImagePlane &imagePlane, const ImagePlane &templatePlane,
                                                   double threshold, int locality, Rect<int> aoi,
                                                   SubPixelMode subPixMode   = SubPixelMode::ParabolicFast,
                                                   Neighborhood subPixRadius = Neighborhood::Use3x3)
      {
        auto pixelList = Internal::DoResCallObjectOut<Internal::PixelList>([&](void *&resobj) {
          return CVB_CALL_CAPI(FindCorrelationMatches(
              imagePlane.Parent().Handle(), imagePlane.Plane(), templatePlane.Parent().Handle(), templatePlane.Plane(),
              aoi.Left(), aoi.Top(), aoi.Right(), aoi.Bottom(), threshold, locality,
              static_cast<CExports::TSubPixelMode>(subPixMode), static_cast<CExports::cvbdim_t>(subPixRadius), resobj));
        });
        return pixelList->ToLocalMaximum();
      }
      inline std::vector<LocalMaximum> FindMatches(const ImagePlane &imagePlane, const ImagePlane &templatePlane, {…}


      inline std::vector<LocalMaximum> FindMatches(const ImagePlane &imagePlane, const ImagePlane &templatePlane,
                                                   double threshold, int locality,
                                                   SubPixelMode subPixMode   = SubPixelMode::ParabolicFast,
                                                   Neighborhood subPixRadius = Neighborhood::Use3x3)
      {
        return FindMatches(imagePlane, templatePlane, threshold, locality, imagePlane.Parent().Bounds(), subPixMode,
                           subPixRadius);
      }
      inline std::vector<LocalMaximum> FindMatches(const ImagePlane &imagePlane, const ImagePlane &templatePlane, {…}
 
    } /* namespace Correlation */
    namespace Correlation {…}
 
    using Correlation::CorrelationMethod;
 
    using Correlation::Correlate;
    using Correlation::FindMatches;
 
  } /* namespace Foundation */
  CVB_END_INLINE_NS
} /* namespace Cvb */
 
#endif