/*
  ==============================================================================

  This is an automatically generated GUI class created by the Introjucer!

  Be careful when adding custom code to these files, as only the code within
  the "//[xyz]" and "//[/xyz]" sections will be retained when the file is loaded
  and re-saved.

  Created with Introjucer version: 3.1.0

  ------------------------------------------------------------------------------

  The Introjucer is part of the JUCE library - "Jules' Utility Class Extensions"
  Copyright 2004-13 by Raw Material Software Ltd.

  ==============================================================================
*/

#ifndef __RBM_COMPONENT__
#define __RBM_COMPONENT__

//[Headers]     -- You can add your own extra header files here --
#include "JuceHeader.h"
#include "DrawComponent.h"
#include "LayerArray.hpp"
#include "Rbm.hpp"
//[/Headers]

class RbmComponentListener
{
public:
	RbmComponentListener() {}
	virtual ~RbmComponentListener()
	{
	}
	virtual void onLayerSizeChanged(size_t size) = 0;
	virtual void onRbmEpochTrained(size_t progressPercent) = 0;
};


//==============================================================================
/**
                                                                    //[Comments]
    An auto-generated component, created by the Introjucer.

    Describe your class and how it works here!
                                                                    //[/Comments]
*/
class RbmComponent  : public Component,
                       public LayerArrayListener,
                       public RbmListener,
                       public DrawListener
{
public:
    //==============================================================================
    RbmComponent (Weights &weights, RbmComponentListener &listener);
    ~RbmComponent();

    //==============================================================================
    //[UserMethods]     -- You can add your own custom methods in this section.
    //[/UserMethods]

    void paint (Graphics& g);
    void resized() override;
    void mouseMove (const MouseEvent& e) override;
    void mouseEnter (const MouseEvent& e) override;
    void mouseExit (const MouseEvent& e) override;
    void mouseDown (const MouseEvent& e) override;
    void mouseDrag (const MouseEvent& e) override;
    void mouseUp (const MouseEvent& e) override;
    void mouseDoubleClick (const MouseEvent& e) override;
    void mouseWheelMove (const MouseEvent& e, const MouseWheelDetails& wheel) override;

   	void setDoLearnVariance(bool value);
	void setDoRaoBlackwell(bool enable);
	void setUseProbsForHiddenReconstruction(bool enable);
	void setUseVisibleGaussian(bool enable);
	void setDoSparse(bool enable);
	void setNormalizeData(bool enable)
	{
		m_pRbm->setNormalizeData(enable);
	}

	void setLambda(double value);
	void setSigmaDecay(double value);
	void setWeightDecay(double value);
	void setSparsity(double value);
	void setNumGibbs(size_t value);
	void setSigma(double value);

	void setMuWeights(double value)
	{
		m_pRbm->setMuWeights(value);
	}

	void setMuSparsity(double value)
	{
		m_pRbm->setMuSparsity(value);
	}

	void setMomentum(double value)
	{
		m_pRbm->setMomentum(value);
	}
	
	void addFromTraining()
	{
		DrawReconstruction->getData() = DrawTraining->getData();
		DrawReconstruction->DrawData();
		
		m_layers.add(DrawTraining->getData(), m_weights.getNumVisible());
		m_listener.onLayerSizeChanged(m_layers.getSize());
	}

	void selectWeights(size_t index)
	{
		m_currWeightIndexToDraw = index;
		redrawWeights();
	}
	
	void selectTraining(size_t index)
	{
		if (m_layers.getSize() > 0)
		{
			m_currTrainingIndexToDraw = index;
			RowVectorXd t = m_layers.getAt(index);

			DrawTraining->getData() = t;
			DrawTraining->DrawData();
			
			redrawReconstruction();
		}
	}

	void copyReconstructionToTraining()
	{
		DrawTraining->getData() = DrawReconstruction->getData();
		DrawTraining->DrawData();
	}
	
	void clearTraining()
	{
		m_layers.clear();
	}
	
	void loadTraining(const char *pFilename)
	{
		m_layers.clear();
		m_layers.load(pFilename);
	}
	
	void saveTraining(const char *pFilename)
	{
		m_layers.save(pFilename);
	}
	
	void removeTrainingAt(size_t index)
	{
		m_layers.removeAt(index);
	}

	void redrawWeights();
	void redrawReconstruction();
	void train(size_t numEpochs)
	{
		m_pRbm->train(numEpochs);
	}
	
private:
    //[UserVariables]   -- You can add your own custom variables in this section.
	ScopedPointer<Rbm> m_pRbm;
    Weights &m_weights;
	RbmComponentListener &m_listener;
    ScopedPointer<DrawComponent> DrawTraining;
    ScopedPointer<DrawComponent> DrawReconstruction;
    ScopedPointer<DrawComponent> DrawWeights;
    ScopedPointer<DrawComponent> DrawVars;
    ScopedPointer<DrawComponent> DrawHidden;
	LayerArray m_layers;
	size_t m_currWeightIndexToDraw;
	size_t m_currTrainingIndexToDraw;
	void onChanged(const LayerArray &obj) override;
	void onEpochTrained(const Rbm &obj) override;
	void onDraw(DrawComponent &obj) override;
    //[/UserVariables]

    //==============================================================================

    //==============================================================================
    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (RbmComponent)
};

//[EndFile] You can add extra defines here...
//[/EndFile]

#endif   // __RBM_COMPONENT__