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

---

CombineConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_CombineConcepts_h
#define Impala_Core_ImageSet_CombineConcepts_h

#include "Core/Column/AddAssign.h"
#include "Core/Column/MulAssign.h"
#include "Core/Column/DivAssign.h"
#include "Core/Table/SimilarityTableSet.h"
#include "Core/ImageSet/Reporter.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class CombineConcepts : public Listener
{
public:
    typedef Feature::FeatureDefinition FeatureDefinition;
    typedef Table::SimilarityTableSet SimilarityTableSet;

    CombineConcepts(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        if (options.GetNrArg() < 7)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mConceptFileName = options.GetArg(2);
        mModel = options.GetArg(3);
        mResultName = options.GetArg(4);
        mCombi = options.GetArg(5);
        if ((mCombi != "mul") && (mCombi != "avg"))
        {
            ILOG_ERROR("Unknown combination method : " << mCombi);
            return;
        }
        for (int i=6 ; i<options.GetNrArg() ; i++)
            mFeatureDefs.push_back(options.GetArg(i));
    }

    virtual void
    HandleNewWalk(ImageSet* is, String walkType)
    {
        mWalkType = walkType;
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        SimilarityTableSet* combinedSet = 0;
        for (int i=0 ; i<mFeatureDefs.size() ; i++)
        {
            if (combinedSet == 0)
            {
                combinedSet = SimilarityTableSet::MakeFromFile
                        (is, mWalkType, mConceptFileName, mModel, 
                         mFeatureDefs[i], dirId);
             }
            else
            {
                SimilarityTableSet* simSet = SimilarityTableSet::MakeFromFile
                    (is, mWalkType, mConceptFileName, mModel, mFeatureDefs[i], 
                     dirId);
                for (int t=0 ; t<simSet->NrTables() ; t++)
                {
                    if (mCombi == "mul")
                        MulAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                                  simSet->GetSimTable(t)->GetColumn1(), 0, -1);
                    else
                        AddAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                                  simSet->GetSimTable(t)->GetColumn1(), 0, -1);
                }
                delete simSet;
            }
        }
        if (mCombi == "avg")
            for (int t=0 ; t<combinedSet->NrTables() ; t++)
                DivAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                          mFeatureDefs.size(), 0, -1);
        combinedSet->ComputeRanks(true);

        combinedSet->Save(is, mWalkType, mConceptFileName, mModel, mResultName, 
                          dirId, true);
        delete combinedSet;
    }

private:

    Reporter*           mReporter;
    String              mConceptFileName;
    String              mModel;
    String              mResultName;
    String              mCombi;
    std::vector<String> mFeatureDefs;
    String              mWalkType;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(CombineConcepts, Impala.Core.ImageSet);

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

#endif

---