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

---

ClusterFeatures.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_ClusterFeatures_h
#define Impala_Core_ImageSet_ClusterFeatures_h

#include "Core/Feature/ClusterorFactory.h"
#include "Core/Feature/Computor.h"
#include "Core/ImageSet/Reporter.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class ClusterFeatures : public Listener
{
public:

    typedef Feature::Computor Computor;
    typedef Feature::Clusteror Clusteror;
    typedef Feature::FeatureTable FeatureTable;

    ClusterFeatures(Computor* computor, String clusteror, Reporter* reporter,
                    CmdOptions& options)
    {
        mReporter = reporter;
        mComputor = computor;

        if (options.GetNrArg() < 7)
            ILOG_ERROR("missing argument");
        String mode = options.GetArg(4);
        int nrWantedClusters = atoi(options.GetArg(5));
        int nrSamplesAtime = atoi(options.GetArg(6));
        ILOG_INFO("mode = " << mode << ", wanted = " << nrWantedClusters <<
                  ", samplesAtime = " << nrSamplesAtime);

        mComputor->MakeRegionFeatureTableSet();

        for (int p=0 ; p<mComputor->GetNrPixelFeatureSets() ; p++)
        {
            for (int r=0 ; r<mComputor->GetNrRegionFeatureSets() ; r++)
            {
                //int minElemsInCluster = mComputor->GetRegionFeatureSetSize(r)+1;
                int minElemsInCluster = 2;
                Feature::ClusterorFactory::Instance().Construct
                    (mClusterors, p, r, nrWantedClusters, nrSamplesAtime,
                     minElemsInCluster, clusteror, options);
            }
        }

        /* todo ??
        if ((mode == "next") || (mode == "count"))
        {
            mComputor->ReadProtoFeatureTableSet("all");
            mComputor->GetProtoFeatureTableSet()->Dump(false);
            for (int i=0 ; i<mClusterors.size() ; i++)
            {
                Clusteror* c = mClusterors[i];
                FeatureTable* tab = mComputor->GetProtoFeatureTable
                    (c->GetPixelFeatureSet(), c->GetRegionFeatureSet(),
                     c->GetClusterType(), c->GetClusterVal());
                c->InitClusterVec(tab, mode, nrWantedClusters);
            }
        }
        */
    }

    virtual void
    HandleNewFile(ImageSet* is, int fileId, Array::Array2dVec3UInt8* im)
    {
        ILOG_DEBUG("new file " << fileId);
        Quid quid = is->GetQuidImage(fileId);

        mComputor->ComputeRegionFeatures(im, quid);

        for (int p=0 ; p<mComputor->GetNrPixelFeatureSets() ; p++)
        {
            for (int r=0 ; r<mComputor->GetNrRegionFeatureSets() ; r++)
            {
                FeatureTable* tab = mComputor->GetRegionFeatureTable(p, r);
                for (int clustID=0 ; clustID<mClusterors.size() ; clustID++)
                {
                    if (mClusterors[clustID]->GetPixelFeatureSet() == p
                        && mClusterors[clustID]->GetRegionFeatureSet() == r)
                    {
                        mClusterors[clustID]->Cluster(tab);
                    }
                }
            }
        }

        mComputor->GetRegionFeatureTableSet()->SetEmpty();

        // experimental
        bool allDone = true;
        for (int i=0 ; i<mClusterors.size() ; i++)
        {
            if (mReporter->DoPrint())
                mClusterors[i]->PrintStatus();
            allDone = allDone && mClusterors[i]->HadEnough();
        }
        if (allDone)
        {
            ILOG_INFO("all clusterors had enough");
            HandleDoneWalk(is);
            ILOG_INFO("had enough, bye");
            exit(0);
        }
        ILOG_DEBUG("done file " << fileId);
    }

    virtual void
    HandleDoneWalk(ImageSet* is)
    {
        ILOG_INFO("Done Walk");
        for (int i=0 ; i<mClusterors.size() ; i++)
            mClusterors[i]->FinishBuffer();
        std::cout << std::endl;

        for (int i=0 ; i<mClusterors.size() ; i++)
        {
            Clusteror* c = mClusterors[i];
            FeatureTable* tab = mComputor->MakeProtoFeatureTable
                (c->GetPixelFeatureSet(), c->GetRegionFeatureSet(),
                 c->GetClusterType(), c->GetClusterVal());
            c->Clusters2FeatureTable(tab);
        }

        mComputor->WriteProtoFeatureTableSet(is);
    }

private:

    Reporter*               mReporter;
    Feature::Computor*      mComputor;
    std::vector<Clusteror*> mClusterors;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(ClusterFeatures, Impala.Core.ImageSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---