SubsetStack.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) 2012 Heiko Strathmann
 */

#ifndef __SUBSETSTACK_H_
#define __SUBSETSTACK_H_

#include <shogun/base/SGObject.h>
#include <shogun/mathematics/Math.h>
#include <shogun/lib/DynamicObjectArray.h>
#include <shogun/features/Subset.h>


namespace shogun
{

class CSubsetStack: public CSGObject
{
public:
    CSubsetStack();

    CSubsetStack(const CSubsetStack& other);

    virtual ~CSubsetStack();

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

    virtual void add_subset(SGVector<index_t> subset);

    virtual void remove_subset();

    virtual void remove_all_subsets();

    inline index_t get_size() {
        if (!has_subsets())
            SG_WARNING("CSubsetStack::get_size(): No subset in stack!\n");

        return has_subsets() ? m_active_subset->get_size() : -1;
    }

    virtual bool has_subsets() const
    {
        return m_active_subsets_stack->get_num_elements();
    }

    CSubset* get_last_subset() const { return m_active_subset; }

    inline index_t subset_idx_conversion(index_t idx) const
    {
        return m_active_subset ? m_active_subset->m_subset_idx.vector[idx] : idx;
    }

private:
    void init();

private:
    CDynamicObjectArray* m_active_subsets_stack;

    CSubset* m_active_subset;
};

}
#endif /* __SUBSETSTACK_H_ */