|| || || || || || || ||

---

GeodesicDistanceTransform.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_GeodesicDistanceTransform_h
#define Impala_Core_Array_GeodesicDistanceTransform_h

#include "Core/Array/Pattern/PatQueueBased.h"
#include "Core/Array/Trait/QbWatershed.h"
#include "Core/Array/PixMinMax.h"
#include "Core/Matrix/MatSet.h"

namespace Impala
{
namespace Core
{
namespace Array
{


template<class DstArrayT, class SrcArrayT>
inline void
GeodesicDistanceTransform(DstArrayT*& dst, SrcArrayT* src, int conn)
{
    typedef typename SrcArrayT::ArithType ArithT; // expected to be int

    // to speedup compute here the min and max gray values 
    ArithT minVal, maxVal;
    PixMinMax(src, &minVal, &maxVal);

    // The result (distImage) is actually a side-effect of the watershed algo...
    Array2dScalarInt32 distImage(src->CW(), src->CH(), 0, 0);
    MatSet(&distImage, 0);

    Trait::QbWatershed<DstArrayT, SrcArrayT, SrcArrayT>
        qb(src->CW(), src->CH(), minVal, maxVal, &distImage);
    SrcArrayT* src2 = ArrayClone<SrcArrayT>(src);
    Pattern::PatQueueBased(dst, src, src2, qb, conn);
    delete src2;

    Set(dst, &distImage);
}

} // namespace Array
} // namespace Core
} // namespace Impala

#endif

---