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

---

TestWatershedMarkers.h

Go to the documentation of this file.

#ifndef Impala_Core_Test_TestWatershedMarkers_h
#define Impala_Core_Test_TestWatershedMarkers_h

#include "Core/Test/TestImage.h"
#include "Core/Array/WatershedMarkers.h"

namespace Impala
{
namespace Core
{
namespace Test
{


class TestWatershedMarkers : public TestImage
{
public:

    TestWatershedMarkers(String srcName, String maskName, int conn, int doLabel)
    {
        mTestNameBase = "WatershedMarkers";
        mSrcName = srcName;
        mMaskName = maskName;
        mConn = conn;
        mDoLabel = doLabel;
    }

    virtual void
    Run()
    {
        Array2dScalarInt32* src = OpenSrcArrayAsInt(mSrcName);
        Array2dScalarInt32* mask = OpenSrcArrayAsInt(mMaskName);
        Array2dScalarInt32* res = 0;
        WatershedMarkers(res, src, mask, mConn, mDoLabel);

        int id = mImSet->GetFileId(mSrcName);
        String suffix = FileNameCtor("", mTestNameBase, mMaskName,
                                                mConn, mDoLabel, ".raw");
        if (1) // 1 to compare to the reference, 0 to generate the reference
        {
            Compare(res, mImSet->GetFilePathReference(id, suffix, false, false));
        }
        else
        {
            WriteRaw(res, mImSet->GetFilePathReference(id, suffix, true, false),
                     mImSet->GetDatabase(), true);
        }

        delete src;
        delete mask;
        delete res;
    }

    virtual String
    Name() const
    {
        return FileNameCtor(mTestNameBase, mSrcName, mMaskName, mConn,
                            mDoLabel, "");
    }

private:

    String mTestNameBase;
    String mSrcName;
    String mMaskName;
    int    mConn;
    int    mDoLabel;

};


static TestBase* gWatershedMarkersSuite[] =
    {
        new TestWatershedMarkers("0a.png", "0amask.png", 8, 1),
        new TestWatershedMarkers("0a.png", "0amask3.png", 8, 1),
        (TestBase*)0
    };

} // namespace Test
} // namespace Core
} // namespace Impala

#endif

---