C:/Users/albertlr/projects/TarzaNN/TarzaNN/SOMFeaturePlane.h

Go to the documentation of this file.

#ifndef SOMFeaturePlane_H_INCLUDED
#define SOMFeaturePlane_H_INCLUDED

#include  <qimage.h>

#include "nowarn.h"
#include "common.h"
#include <stdlib.h>

#include "OutputImage.h"
#include "LearningFeaturePlane.h"
#include "FilterBank.h"
#include "PlaneInputWindowSet.h"
#include "MatrixOfMatrix.h"

class SOMFeaturePlane : public LearningFeaturePlane {
        friend class GatingUnits;
        friend class FeaturePlaneFactory;
public:
    ~SOMFeaturePlane();
        void read(QDataStream* inStream, QProgressDialog* progress, int* index);

        void save(QDataStream* outStream, QProgressDialog* progress, int* index);
        bool isLearning();
        
protected:
        
        SOMFeaturePlane( Network* net, int l, QString* name, int w, int h, int angles, int speeds, int alpha, int speed, bool isWTA, bool visible, Neuron* neuron, NotifyStrategyAbstract* notify, bool learning);
        void step();
        
        float nonlinearity(Matrix* totalActivation, OutputImage* workOutput);


private:
        FilterBank* lgnONFilters;
        FilterBank* lgnOFFFilters;
        FilterBank* excitatoryFilters;
        FilterBank* inhibitoryFilters;
        
        void addLateralContribution(FilterBank* excitatoryFilters, OutputImage* lastTimestep, float factor);
        void learn();
        void updateWeights(FilterBank* filters, Matrix* input, float alpha, bool fromInput);

};
#endif //SOMFeaturePlane_H_INCLUDED