TOPFeatures.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) 1999-2009 Soeren Sonnenburg
 * Written (W) 1999-2008 Gunnar Raetsch
 * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */

#ifndef _CTOPFEATURES__H__
#define _CTOPFEATURES__H__

#include <shogun/lib/config.h>

#include <shogun/features/DenseFeatures.h>
#include <shogun/distributions/HMM.h>

namespace shogun
{
template <class T> class CDenseFeatures;
class CHMM;

#ifndef DOXYGEN_SHOULD_SKIP_THIS

struct T_HMM_INDIZES
{
    int32_t* idx_p;
    int32_t* idx_q;
    int32_t* idx_a_rows;
    int32_t* idx_a_cols;
    int32_t* idx_b_rows;
    int32_t* idx_b_cols;

    int32_t num_p;
    int32_t num_q;
    int32_t num_a;
    int32_t num_b;
};
#endif // DOXYGEN_SHOULD_SKIP_THIS

class CTOPFeatures : public CDenseFeatures<float64_t>
{
    public:
        CTOPFeatures();

        CTOPFeatures(int32_t size, CHMM* p, CHMM* n, bool neglin, bool poslin);

        CTOPFeatures(const CTOPFeatures &orig);

        virtual ~CTOPFeatures();

        void set_models(CHMM* p, CHMM* n);

        virtual float64_t* set_feature_matrix();

        int32_t compute_num_features();

        bool compute_relevant_indizes(CHMM* hmm, T_HMM_INDIZES* hmm_idx);

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

    protected:
        virtual float64_t* compute_feature_vector(
            int32_t num, int32_t& len, float64_t* target=NULL);

        void compute_feature_vector(float64_t* addr, int32_t num, int32_t& len);

    private:
        void init();

    protected:
        CHMM* pos;
        CHMM* neg;
        bool neglinear;
        bool poslinear;

        T_HMM_INDIZES pos_relevant_indizes;
        T_HMM_INDIZES neg_relevant_indizes;
};
}
#endif