AttributeFeatures.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2009 Soeren Sonnenburg
 * Copyright (C) 2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */
#ifndef _CATTRIBUTE_FEATURES__H__
#define _CATTRIBUTE_FEATURES__H__

#include <string.h>

#include <shogun/features/Features.h>
#include <shogun/base/DynArray.h>

namespace shogun
{

#ifndef DOXYGEN_SHOULD_SKIP_THIS

struct T_ATTRIBUTE
{
    char* attr_name;
    CFeatures* attr_obj;
};
#endif // DOXYGEN_SHOULD_SKIP_THIS

class CAttributeFeatures : public CFeatures
{

public:
    CAttributeFeatures();

    virtual ~CAttributeFeatures();

    CFeatures* get_attribute(char* attr_name)
    {
        int32_t idx=find_attr_index(attr_name);
        if (idx>=0)
        {
            CFeatures* f=features[idx].attr_obj;
            SG_REF(f);
            return f;
        }

        return NULL;
    }

    inline void get_attribute_by_index(int idx, const char* &attr_name, CFeatures* &attr_obj)
    {
        T_ATTRIBUTE a= features.get_element_safe(idx);
        attr_name= a.attr_name;
        attr_obj= a.attr_obj;
        SG_REF(a.attr_obj);
    }

    inline bool set_attribute(char* attr_name, CFeatures* attr_obj)
    {
        int32_t idx=find_attr_index(attr_name);
        if (idx==-1)
            idx=features.get_num_elements();

        T_ATTRIBUTE a;
        a.attr_name=strdup(attr_name);
        a.attr_obj=attr_obj;

        SG_REF(attr_obj);

        return features.set_element(a, idx);
    }

    inline bool del_attribute(char* attr_name)
    {
        int32_t idx=find_attr_index(attr_name);

        if (idx>=0)
        {
            T_ATTRIBUTE a= features[idx];
            SG_FREE(a.attr_name);
            SG_UNREF(a.attr_obj);
            return true;
        }
        return false;
    }


    inline int32_t get_num_attributes()
    {
        return features.get_num_elements();
    }

    inline virtual const char* get_name() const { return "AttributeFeatures"; }

    virtual CFeatures* duplicate() const=0;

    virtual EFeatureType get_feature_type()=0;

    virtual EFeatureClass get_feature_class()=0;

    virtual int32_t get_num_vectors() const=0 ;

    virtual int32_t get_size()=0;

protected:
    inline int32_t find_attr_index(char* attr_name)
    {
        int32_t n=features.get_num_elements();
        for (int32_t i=0; i<n; i++)
        {
            if (!strcmp(features[n].attr_name, attr_name))
                return i;
        }

        return -1;
    }


protected:
    DynArray<T_ATTRIBUTE> features;
};
}
#endif