Subset.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) 2011-2012 Heiko Strathmann
 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
 */

#ifndef __SUBSET_H_
#define __SUBSET_H_

#include <shogun/base/SGObject.h>
#include <shogun/mathematics/Math.h>

namespace shogun
{

class CSubset: public CSGObject
{
    friend class CSubsetStack;

public:
    CSubset();

    CSubset(SGVector<index_t> subset_idx);

    virtual ~CSubset();

    index_t get_size() const { return m_subset_idx.vlen; }

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

    SGVector<index_t> get_subset_idx() const { return m_subset_idx; }

private:
    void init();

private:
    SGVector<index_t> m_subset_idx;
};

}
#endif /* __SUBSET_H_ */