fastica_nonlinearity.hh

Go to the documentation of this file.

/* -*- mia-c++  -*-
 *
 * This file is part of MIA - a toolbox for medical image analysis 
 * Copyright (c) Leipzig, Madrid 1999-2017 Gert Wollny
 *
 * MIA 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.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MIA; if not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef mia_core_fastica_nonlinearity_hh
#define mia_core_fastica_nonlinearity_hh

#include <mia/core/gsl_vector.hh>
#include <mia/core/gsl_matrix.hh>

#include <mia/core/factory.hh>
namespace mia {


class EXPORT_CORE CFastICANonlinearityBase : public CProductBase {
public: 

        typedef CFastICANonlinearityBase plugin_data; 

        static const char *data_descr; 

        CFastICANonlinearityBase(); 

        void set_signal(const gsl::Matrix *signal); 
        void set_mu(double m); 

protected: 
        double get_sample_scale() const {return m_sample_scale;}
        double get_mu() const { return m_mu;}; 
        const gsl::Matrix& get_signal() const; 

private: 
        virtual void post_set_signal() = 0;
        double m_mu;
        double m_sample_scale;
        const gsl::Matrix *m_signal; 
}; 

class EXPORT_CORE CFastICADeflNonlinearity : public CFastICANonlinearityBase {
public: 

        typedef CFastICADeflNonlinearity plugin_type; 

        static const char *type_descr; 


        void apply(gsl::Vector& w); 
        void apply(gsl::Matrix& W);

        std::vector<double> get_saddle_test_table(const gsl::Matrix& ics) const;  
        double get_saddle_test_value(const gsl::Vector& ic) const;

protected: 
        virtual void post_set_signal();
private: 
        virtual double get_correction_and_scale(gsl::Vector& XTw, gsl::Vector& correction) = 0; 
        
        virtual double do_get_saddle_test_value(const gsl::Vector& ic) const = 0;
        void sum_final(gsl::Vector& w, double scale); 
        void sum_final_stabelized(gsl::Vector& w, double scale); 

        gsl::Vector m_XTw;
        gsl::Vector m_workspace; 
}; 


typedef std::shared_ptr<CFastICADeflNonlinearity> PFastICADeflNonlinearity; 


EXPORT_CORE PFastICADeflNonlinearity produce_fastica_nonlinearity(const std::string& descr); 


typedef TFactory<CFastICADeflNonlinearity> CFastICADeflNonlinearityPlugin; 

typedef THandlerSingleton<TFactoryPluginHandler<CFastICADeflNonlinearityPlugin> > CFastICADeflNonlinearityPluginHandler;

}

#endif