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

---

Watershed.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_Watershed_h
#define Impala_Core_Array_Watershed_h

#include "Core/Array/Pattern/PatQueueBased.h"
#include "Core/Array/Trait/QbWatershed.h"
#include "Core/Array/Pattern/PatNgbOp2d.h"
#include "Core/Array/Trait/NgbLWshed.h"
#include "Core/Array/PixMinMax.h"

namespace Impala
{
namespace Core
{
namespace Array
{


template<class DstArrayT, class SrcArrayT>
inline void
Watershed(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);
    // for more speedup, use a histogram to indicate the existing gray levels

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

    //because the LucVincent algorithm doesn't get all the watershed points
    //this neigborhoud operator replaces the smallest label with wshed value
    Trait::NgbLWshed<DstArrayT, DstArrayT> lwshed(conn, Trait::WSHEDVAL);
    Pattern::PatNgbOp2d(dst, dst, lwshed);
}

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

#endif

---