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

#ifndef __SUBSET_H_
#define __SUBSET_H_

#include <shogun/base/SGObject.h>

namespace shogun
{

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

    CSubset(SGVector<index_t> subset_idx);

    virtual ~CSubset();

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

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

    CSubset* duplicate();

    inline index_t subset_idx_conversion(index_t idx) const
    {
        return m_subset_idx.vector ? m_subset_idx.vector[idx] : idx;
    }
private:
    void init();

private:
    const SGVector<index_t> m_subset_idx;
};

}
#endif /* __SUBSET_H_ */