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 <shogun/lib/config.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);

    void get_attribute_by_index(int idx, const char* &attr_name, CFeatures* &attr_obj);

    bool set_attribute(char* attr_name, CFeatures* attr_obj);

    bool del_attribute(char* attr_name);

    int32_t get_num_attributes();

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

    virtual CFeatures* duplicate() const=0;

    virtual EFeatureType get_feature_type() const=0;

    virtual EFeatureClass get_feature_class() const=0;

    virtual int32_t get_num_vectors() const=0;

protected:
    inline int32_t find_attr_index(char* attr_name);

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