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

---

ReadCodebook.h

Go to the documentation of this file.

#ifndef Impala_Core_Feature_ReadCodebook_h
#define Impala_Core_Feature_ReadCodebook_h

#include "Core/Feature/FeatureTable.h"
#include "Core/Feature/PointDescriptorIO.h"
#include "Core/Matrix/Mat.h"

namespace Impala
{
namespace Core
{
namespace Feature
{


FeatureTable*
ReadCodebookFromBuffer(Util::IOBuffer* buf)
{
    ILOG_VAR(Impala.Core.Feature.ReadCodebookFromBuffer);

    using namespace Matrix;

    FeatureTable* result = 0;
    buf->Seek(0, SEEK_SET);
    char buf2[2049];
    buf2[2048] = '\0';
    buf->Read(buf2, 2048);
    buf->Seek(0, SEEK_SET);
    String recognizer(buf2);
    if(recognizer.find("nr strings") != String::npos)
    {   // nr strings : then FeatureTableOld
        typedef Table::TableTem<Column::ColumnTem<String>,
                        Vector::VectorSet<Array::Array2dScalarReal64> >
                        FeatureTableOldType;
        FeatureTableOldType* oldCodebook = new FeatureTableOldType(
                 Column::ColumnTem<String>(0),
                 Vector::VectorSet<Array::Array2dScalarReal64>(true, 1, 0));
        Read(oldCodebook, buf);
        
        int d = oldCodebook->GetColumn2()->GetVectorLength(0);
        // need to copy oldCodebook into result
        result = new FeatureTable(FeatureDefinition(""), 
                               oldCodebook->Size(), 
                               d);
        Mat* matrixCodebook = oldCodebook->GetColumn2()->GetStorage();
        for(int i = 0; i < oldCodebook->Size(); i++)
        {
            Quid q = MakeQuidCodebook(0, 0, i);
            Core::Vector::VectorTem<Real64>
                v(d);
            for(int j = 0; j < d; j++)
            {
                v[j] = *MatE(matrixCodebook, i, j);
            }
            result->Add(q, v);
        }
        ILOG_WARN("Codebook is in an old format. Future versions might not work.");
    }
    else if(recognizer.substr(0,5) == String("KOEN1") || 
            recognizer.substr(0,8) == String("BINDESC1"))
    {
        // other format
        buf->Seek(0, SEEK_SET);
        Mat* ignored = 0;
        Mat* descriptors = 0;
        ReadBINDESC1FromBuffer(ignored, descriptors, buf, false, true);
        result = new FeatureTable(FeatureDefinition(""),
                                  MatNrRow(descriptors), 
                                  MatNrCol(descriptors));
        for(int i = 0; i < MatNrRow(descriptors); i++)
        {
            Vector::VectorTem<Real64> v(MatNrCol(descriptors), 
                                        MatE(descriptors, i, 0),
                                        true);
            result->Add(i, v);
        }
        delete descriptors;
    }
    else
    {
        result = new FeatureTable(FeatureDefinition(""));
        Read(result, buf);
    }
    return result;
}


} // namespace Feature
} // namespace Core
} // namespace Impala

#endif

---