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

---

TestColorSegmentation.h

Go to the documentation of this file.

#ifndef Impala_Core_Test_TestColorSegmentation_h
#define Impala_Core_Test_TestColorSegmentation_h

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

#pragma warning (disable : 4800)

namespace Impala
{
namespace Core
{
namespace Test
{


class TestColorSegmentation : public TestImage
{
public:
    typedef Array::ColorSegmentationAlgorithm ColorSegmentationAlgorithm;
    typedef Array::ColorSegmentationInvariant ColorSegmentationInvariant;

    TestColorSegmentation(String srcName, ColorSegmentationAlgorithm segAlg,
                          ColorSegmentationInvariant invType,
                          double fraction, double threshold, bool useGauss)
    {
        mTestNameBase = "ColorSegmentation";
        mSrcName = srcName;
        mSegAlg = segAlg;
        mInvType = invType;
        mFraction = fraction;
        mThreshold = threshold;
        mUseGauss = useGauss;
    }

    virtual void
    Run()
    {
        Array2dVec3Real64* src = OpenSrcArrayAsVec3Real64(mSrcName);
        Array2dVec3UInt8* res = 0;
        Array::ColorSegmentation(res, src, mSegAlg, mInvType, mFraction,
                                 mThreshold, mUseGauss, 0);

        int id = mImSet->GetFileId(mSrcName);
        String suffix = FileNameCtor("", mTestNameBase, mSegAlg, mInvType,
                                     mFraction, mThreshold, mUseGauss, ".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 res;
    }

    virtual String
    Name() const
    {
        return FileNameCtor(mTestNameBase, mSrcName, mSegAlg, mInvType,
                            mFraction, mThreshold, mUseGauss, "");
    }

private:

    String mTestNameBase;
    String mSrcName;
    ColorSegmentationAlgorithm mSegAlg;
    ColorSegmentationInvariant mInvType;
    double mFraction;
    double mThreshold;
    bool   mUseGauss;

};


static TestBase* gColorSegmentationSuite[] =
    {
        //new TestColorSegmentation("mosaic.raw", OnlyZero, C, 0.05, 7.5, false),
        //new TestColorSegmentation("mosaic.raw", TextureAddZero, C, 0.05, 7.5, false),
        //new TestColorSegmentation("beach.jpg", OnlyZero, C, 0.01, 6.0, false),
        //new TestColorSegmentation("beach.jpg", TextureAddZero, C, 0.01, 6.0, false),
        //new TestColorSegmentation("beach.jpg", OnlyZero, E, 0.01, 6.0, false),
        new TestColorSegmentation("beach.jpg", Array::TextureAddZero,
                                  Array::E, 0.01, 6.0, false),

        // future test?
        //new TestColorSegmentation("mosaic.raw", OnlyZero, C, 0.05, 7.5, true),
        //new TestColorSegmentation("mosaic.raw", TextureAddZero, C, 0.05, 7.5, true),
        //new TestColorSegmentation("giraffe.png", TextureAddZero, E, 0.05, 6.0, false),
        (TestBase*)0
    };

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

#endif

---