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

  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.

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

//[Headers] You can add your own extra header files here...
//[/Headers]

#include <unistd.h>
#include <sys/types.h>
#include <pwd.h>
#include "MainComponent.hpp"


//[MiscUserDefs] You can add your own user definitions and misc code here...
void mylog(const char* format, ...)
{
	char log_buf[257];
    va_list argptr;
    va_start(argptr, format);
    vprintf(format, argptr);
    va_end(argptr);
}

//[/MiscUserDefs]

//==============================================================================
MainComponent::MainComponent (const String prjname)
: Thread("RBM")
, m_stack(nullptr)
, m_pLayer(nullptr)
, m_trainingIndex(0)
, m_weightIndex(0)
, m_toolTipWindow(this)
{
    addAndMakeVisible (trainButton = new TextButton ("Train button"));
    trainButton->setButtonText (TRANS("Train"));
    trainButton->addListener (this);

    addAndMakeVisible (addButton = new TextButton ("Add button"));
    addButton->setButtonText (TRANS("Add"));
    addButton->addListener (this);

    addAndMakeVisible (patterSlider = new Slider ("Pattern slider"));
    patterSlider->setTooltip (TRANS("Training index"));
    patterSlider->setRange (0, 0, 1);
    patterSlider->setSliderStyle (Slider::LinearBar);
    patterSlider->setTextBoxStyle (Slider::TextBoxLeft, false, 80, 20);
    patterSlider->addListener (this);

    addAndMakeVisible (ShakeButton = new TextButton ("Shake button"));
    ShakeButton->setButtonText (TRANS("Shake"));
    ShakeButton->addListener (this);

    addAndMakeVisible (WeightsSlider = new Slider ("Weights slider"));
    WeightsSlider->setTooltip (TRANS("Weight index"));
    WeightsSlider->setRange (0, 0, 1);
    WeightsSlider->setSliderStyle (Slider::LinearBar);
    WeightsSlider->setTextBoxStyle (Slider::TextBoxLeft, false, 80, 20);
    WeightsSlider->addListener (this);

    addAndMakeVisible (numEpochslabel = new Label ("Num Epochs label",
                                                   TRANS("1000")));
    numEpochslabel->setFont (Font (15.00f, Font::plain));
    numEpochslabel->setJustificationType (Justification::centred);
    numEpochslabel->setEditable (true, true, false);
    numEpochslabel->setColour (TextEditor::textColourId, Colours::black);
    numEpochslabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    numEpochslabel->addListener (this);

    addAndMakeVisible (learningRateLabel = new Label ("Learning Rate label",
                                                      TRANS("0.01")));
    learningRateLabel->setFont (Font (15.00f, Font::plain));
    learningRateLabel->setJustificationType (Justification::centred);
    learningRateLabel->setEditable (true, true, false);
    learningRateLabel->setColour (TextEditor::textColourId, Colours::black);
    learningRateLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    learningRateLabel->addListener (this);

    addAndMakeVisible (numVisibleLabel = new Label ("Num Visible label",
                                                    TRANS("16")));
    numVisibleLabel->setTooltip (TRANS("Number of visible units X"));
    numVisibleLabel->setFont (Font (15.00f, Font::plain));
    numVisibleLabel->setJustificationType (Justification::centred);
    numVisibleLabel->setEditable (true, true, false);
    numVisibleLabel->setColour (TextEditor::textColourId, Colours::black);
    numVisibleLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    numVisibleLabel->addListener (this);

    addAndMakeVisible (numHiddenLabel = new Label ("Num Hidden label",
                                                   TRANS("64")));
    numHiddenLabel->setTooltip (TRANS("Number of hidden units"));
    numHiddenLabel->setFont (Font (15.00f, Font::plain));
    numHiddenLabel->setJustificationType (Justification::centred);
    numHiddenLabel->setEditable (true, true, false);
    numHiddenLabel->setColour (TextEditor::textColourId, Colours::black);
    numHiddenLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    numHiddenLabel->addListener (this);

    addAndMakeVisible (createButton = new TextButton ("Create button"));
    createButton->setButtonText (TRANS("Create"));
    createButton->addListener (this);

    addAndMakeVisible (projectNameLabel = new Label ("Project Name label",
                                                     prjname));
    projectNameLabel->setFont (Font (15.00f, Font::plain));
    projectNameLabel->setJustificationType (Justification::centred);
    projectNameLabel->setEditable (true, true, false);
    projectNameLabel->setColour (TextEditor::textColourId, Colours::black);
    projectNameLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    projectNameLabel->addListener (this);

    addAndMakeVisible (loadButton = new TextButton ("Load button"));
    loadButton->setButtonText (TRANS("Load"));
    loadButton->addListener (this);

    addAndMakeVisible (saveButton = new TextButton ("Save button"));
    saveButton->setButtonText (TRANS("Save"));
    saveButton->addListener (this);

    addAndMakeVisible (numVisibleYLabel = new Label ("Num Visible Y label",
                                                     TRANS("16")));
    numVisibleYLabel->setTooltip (TRANS("Number of visible units Y"));
    numVisibleYLabel->setFont (Font (15.00f, Font::plain));
    numVisibleYLabel->setJustificationType (Justification::centred);
    numVisibleYLabel->setEditable (true, true, false);
    numVisibleYLabel->setColour (TextEditor::textColourId, Colours::black);
    numVisibleYLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    numVisibleYLabel->addListener (this);

    addAndMakeVisible (loadTrainingButton = new TextButton ("Load Training button"));
    loadTrainingButton->setButtonText (TRANS("Load T"));
    loadTrainingButton->addListener (this);

    addAndMakeVisible (saveTrainingButton = new TextButton ("Save Training button"));
    saveTrainingButton->setButtonText (TRANS("Save T"));
    saveTrainingButton->addListener (this);

    addAndMakeVisible (clearTrainingButton = new TextButton ("Clear Training button"));
    clearTrainingButton->setButtonText (TRANS("Clear T"));
    clearTrainingButton->addListener (this);

    addAndMakeVisible (removeTrainingButton = new TextButton ("Remove Training button"));
    removeTrainingButton->setButtonText (TRANS("Remove T"));
    removeTrainingButton->addListener (this);

    addAndMakeVisible (numGibbsSlider = new Slider ("Num. Gibbs slider"));
    numGibbsSlider->setTooltip (TRANS("Number of gibbs samples"));
    numGibbsSlider->setRange (1, 101, 1);
    numGibbsSlider->setSliderStyle (Slider::LinearBar);
    numGibbsSlider->setTextBoxStyle (Slider::TextBoxLeft, false, 80, 20);
    numGibbsSlider->addListener (this);

    addAndMakeVisible (rbmDoRaoBlackwellToggleButton = new ToggleButton ("rbmDoRaoBlackwell toggle button"));
    rbmDoRaoBlackwellToggleButton->setTooltip (TRANS("Don\'t sample hidden during learning"));
    rbmDoRaoBlackwellToggleButton->setButtonText (TRANS("Rao-Blackwell"));
    rbmDoRaoBlackwellToggleButton->addListener (this);

    addAndMakeVisible (rbmDoSampleVisibleToggleButton = new ToggleButton ("rbmDoSampleVisible toggle button"));
    rbmDoSampleVisibleToggleButton->setTooltip (TRANS("Sample visible during learning\n"));
    rbmDoSampleVisibleToggleButton->setButtonText (TRANS("Sample Visible"));
    rbmDoSampleVisibleToggleButton->addListener (this);

    addAndMakeVisible (lambdaLabel = new Label ("Lambda label",
                                                TRANS("1.0")));
    lambdaLabel->setFont (Font (15.00f, Font::plain));
    lambdaLabel->setJustificationType (Justification::centred);
    lambdaLabel->setEditable (true, true, false);
    lambdaLabel->setColour (TextEditor::textColourId, Colours::black);
    lambdaLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    lambdaLabel->addListener (this);

    addAndMakeVisible (sigmaLabel = new Label ("Sigma label",
                                               TRANS("1.0")));
    sigmaLabel->setFont (Font (15.00f, Font::plain));
    sigmaLabel->setJustificationType (Justification::centred);
    sigmaLabel->setEditable (true, true, false);
    sigmaLabel->setColour (TextEditor::textColourId, Colours::black);
    sigmaLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    sigmaLabel->addListener (this);

    addAndMakeVisible (rbmUseVisibleGaussianToggleButton = new ToggleButton ("rbmUseVisibleGaussian toggle button"));
    rbmUseVisibleGaussianToggleButton->setButtonText (TRANS("Use gaussian visible"));
    rbmUseVisibleGaussianToggleButton->addListener (this);

    addAndMakeVisible (rbmDoSparseToggleButton = new ToggleButton ("rbmDoSparse toggle button"));
    rbmDoSparseToggleButton->setButtonText (TRANS("Do sparse"));
    rbmDoSparseToggleButton->addListener (this);

    addAndMakeVisible (sparsityLabel = new Label ("Sparsity label",
                                                  TRANS("0.05")));
    sparsityLabel->setFont (Font (15.00f, Font::plain));
    sparsityLabel->setJustificationType (Justification::centred);
    sparsityLabel->setEditable (true, true, false);
    sparsityLabel->setColour (TextEditor::textColourId, Colours::black);
    sparsityLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    sparsityLabel->addListener (this);

    addAndMakeVisible (sigmaDecayLabel = new Label ("SigmaDecay label",
                                                    TRANS("1.0")));
    sigmaDecayLabel->setFont (Font (15.00f, Font::plain));
    sigmaDecayLabel->setJustificationType (Justification::centred);
    sigmaDecayLabel->setEditable (true, true, false);
    sigmaDecayLabel->setColour (TextEditor::textColourId, Colours::black);
    sigmaDecayLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    sigmaDecayLabel->addListener (this);

    addAndMakeVisible (weightDecayLabel = new Label ("WeightDecay label",
                                                     TRANS("0")));
    weightDecayLabel->setFont (Font (15.00f, Font::plain));
    weightDecayLabel->setJustificationType (Justification::centred);
    weightDecayLabel->setEditable (true, true, false);
    weightDecayLabel->setColour (TextEditor::textColourId, Colours::black);
    weightDecayLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    weightDecayLabel->addListener (this);

    addAndMakeVisible (m_progressBarSlider = new Slider ("progressBar slider"));
    m_progressBarSlider->setTooltip (TRANS("Progress"));
    m_progressBarSlider->setRange (0, 100, 1);
    m_progressBarSlider->setSliderStyle (Slider::LinearBar);
    m_progressBarSlider->setTextBoxStyle (Slider::TextBoxLeft, true, 80, 20);
    m_progressBarSlider->addListener (this);

    addAndMakeVisible (momentumLabel = new Label ("Momentum label",
                                                  TRANS("0.5")));
    momentumLabel->setFont (Font (15.00f, Font::plain));
    momentumLabel->setJustificationType (Justification::centred);
    momentumLabel->setEditable (true, true, false);
    momentumLabel->setColour (TextEditor::textColourId, Colours::black);
    momentumLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    momentumLabel->addListener (this);

    addAndMakeVisible (sparsityLearningRateLabel = new Label ("SparsityLearn label",
                                                              TRANS("0.01")));
    sparsityLearningRateLabel->setFont (Font (15.00f, Font::plain));
    sparsityLearningRateLabel->setJustificationType (Justification::centred);
    sparsityLearningRateLabel->setEditable (true, true, false);
    sparsityLearningRateLabel->setColour (TextEditor::textColourId, Colours::black);
    sparsityLearningRateLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    sparsityLearningRateLabel->addListener (this);

    addAndMakeVisible (weightInitLabel = new Label ("WeightInit label",
                                                    TRANS("0.01")));
    weightInitLabel->setFont (Font (15.00f, Font::plain));
    weightInitLabel->setJustificationType (Justification::centred);
    weightInitLabel->setEditable (true, true, false);
    weightInitLabel->setColour (TextEditor::textColourId, Colours::black);
    weightInitLabel->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    weightInitLabel->addListener (this);

    addAndMakeVisible (rbmLearnVarianceButton = new ToggleButton ("rbmLearnVariance button"));
    rbmLearnVarianceButton->setButtonText (TRANS("Learn Variance"));
    rbmLearnVarianceButton->addListener (this);

    addAndMakeVisible (rbmNormalizeDataToggleButton = new ToggleButton ("rbmNormalizeData toggle button"));
    rbmNormalizeDataToggleButton->setButtonText (TRANS("Normalize data"));
    rbmNormalizeDataToggleButton->addListener (this);

    addAndMakeVisible (m_rbmSelect = new ComboBox ("RBM Selector"));
    m_rbmSelect->setEditableText (false);
    m_rbmSelect->setJustificationType (Justification::centredLeft);
    m_rbmSelect->setTextWhenNothingSelected (String::empty);
    m_rbmSelect->setTextWhenNoChoicesAvailable (TRANS("(no choices)"));
    m_rbmSelect->addListener (this);

    addAndMakeVisible (rbmDoSampleBatch = new ToggleButton ("rbmDoSampleBatch button"));
    rbmDoSampleBatch->setTooltip (TRANS("Sample training data during learning\n"));
    rbmDoSampleBatch->setButtonText (TRANS("Sample training"));
    rbmDoSampleBatch->addListener (this);

    addAndMakeVisible (sizeMiniBatch = new Label ("sizeMiniBatch",
                                                  TRANS("100")));
    sizeMiniBatch->setTooltip (TRANS("Mini batch size"));
    sizeMiniBatch->setFont (Font (15.00f, Font::plain));
    sizeMiniBatch->setJustificationType (Justification::centred);
    sizeMiniBatch->setEditable (true, true, false);
    sizeMiniBatch->setColour (TextEditor::textColourId, Colours::black);
    sizeMiniBatch->setColour (TextEditor::backgroundColourId, Colour (0x00000000));
    sizeMiniBatch->addListener (this);

    addAndMakeVisible (rbmUseHiddenGaussianToggleButton = new ToggleButton ("rbmUseHiddenGaussian toggle button"));
    rbmUseHiddenGaussianToggleButton->setButtonText (TRANS("Use gaussian hidden"));
    rbmUseHiddenGaussianToggleButton->addListener (this);


    //[UserPreSize]
    m_progressBarSlider->setValue(100);
    //[/UserPreSize]

    setSize (1200, 600);


    //[Constructor] You can add your own custom stuff here..
    //[/Constructor]
}

MainComponent::~MainComponent()
{
    //[Destructor_pre]. You can add your own custom destruction code here..
    //[/Destructor_pre]

    trainButton = nullptr;
    addButton = nullptr;
    patterSlider = nullptr;
    ShakeButton = nullptr;
    WeightsSlider = nullptr;
    numEpochslabel = nullptr;
    learningRateLabel = nullptr;
    numVisibleLabel = nullptr;
    numHiddenLabel = nullptr;
    createButton = nullptr;
    projectNameLabel = nullptr;
    loadButton = nullptr;
    saveButton = nullptr;
    numVisibleYLabel = nullptr;
    loadTrainingButton = nullptr;
    saveTrainingButton = nullptr;
    clearTrainingButton = nullptr;
    removeTrainingButton = nullptr;
    numGibbsSlider = nullptr;
    rbmDoRaoBlackwellToggleButton = nullptr;
    rbmDoSampleVisibleToggleButton = nullptr;
    lambdaLabel = nullptr;
    sigmaLabel = nullptr;
    rbmUseVisibleGaussianToggleButton = nullptr;
    rbmDoSparseToggleButton = nullptr;
    sparsityLabel = nullptr;
    sigmaDecayLabel = nullptr;
    weightDecayLabel = nullptr;
    m_progressBarSlider = nullptr;
    momentumLabel = nullptr;
    sparsityLearningRateLabel = nullptr;
    weightInitLabel = nullptr;
    rbmLearnVarianceButton = nullptr;
    rbmNormalizeDataToggleButton = nullptr;
    m_rbmSelect = nullptr;
    rbmDoSampleBatch = nullptr;
    sizeMiniBatch = nullptr;
    rbmUseHiddenGaussianToggleButton = nullptr;


    //[Destructor]. You can add your own custom destruction code here..

    //[/Destructor]
}

//==============================================================================
void MainComponent::paint (Graphics& g)
{
    //[UserPrePaint] Add your own custom painting code here..
    //[/UserPrePaint]

    g.fillAll (Colours::white);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Sigma"),
                560, 138, 80, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Lambda"),
                648, 138, 80, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Num. epochs"),
                551, 186, 91, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Alpha"),
                648, 186, 80, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Sparsity"),
                736, 138, 80, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Sigma decay"),
                824, 138, 96, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Weight decay"),
                824, 186, 96, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Momentum"),
                736, 186, 80, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Sparsity Alpha"),
                936, 138, 96, 14,
                Justification::centred, true);

    g.setColour (Colours::black);
    g.setFont (Font (15.00f, Font::plain));
    g.drawText (TRANS("Weight init"),
                936, 186, 96, 14,
                Justification::centred, true);

    //[UserPaint] Add your own custom painting code here..
    //[/UserPaint]
}

void MainComponent::resized()
{
    trainButton->setBounds (648, 20, 72, 24);
    addButton->setBounds (556, 20, 72, 24);
    patterSlider->setBounds (896, 280, 220, 24);
    ShakeButton->setBounds (740, 20, 72, 24);
    WeightsSlider->setBounds (896, 320, 220, 24);
    numEpochslabel->setBounds (564, 200, 72, 24);
    learningRateLabel->setBounds (652, 200, 72, 24);
    numVisibleLabel->setBounds (656, 248, 48, 24);
    numHiddenLabel->setBounds (780, 248, 48, 24);
    createButton->setBounds (728, 280, 72, 24);
    projectNameLabel->setBounds (552, 248, 96, 24);
    loadButton->setBounds (564, 280, 72, 24);
    saveButton->setBounds (644, 280, 72, 24);
    numVisibleYLabel->setBounds (716, 248, 48, 24);
    loadTrainingButton->setBounds (564, 316, 72, 24);
    saveTrainingButton->setBounds (644, 316, 72, 24);
    clearTrainingButton->setBounds (804, 316, 72, 24);
    removeTrainingButton->setBounds (724, 316, 72, 24);
    numGibbsSlider->setBounds (896, 240, 220, 24);
    rbmDoRaoBlackwellToggleButton->setBounds (556, 60, 112, 24);
    rbmDoSampleVisibleToggleButton->setBounds (680, 60, 128, 24);
    lambdaLabel->setBounds (652, 152, 72, 24);
    sigmaLabel->setBounds (564, 152, 72, 24);
    rbmUseVisibleGaussianToggleButton->setBounds (680, 92, 156, 24);
    rbmDoSparseToggleButton->setBounds (556, 92, 96, 24);
    sparsityLabel->setBounds (740, 152, 72, 24);
    sigmaDecayLabel->setBounds (836, 152, 72, 24);
    weightDecayLabel->setBounds (836, 200, 72, 24);
    m_progressBarSlider->setBounds (828, 20, 220, 24);
    momentumLabel->setBounds (740, 200, 72, 24);
    sparsityLearningRateLabel->setBounds (948, 152, 72, 24);
    weightInitLabel->setBounds (948, 200, 72, 24);
    rbmLearnVarianceButton->setBounds (1004, 92, 128, 24);
    rbmNormalizeDataToggleButton->setBounds (1004, 60, 124, 24);
    m_rbmSelect->setBounds (812, 280, 62, 24);
    rbmDoSampleBatch->setBounds (844, 60, 128, 24);
    sizeMiniBatch->setBounds (1064, 20, 48, 24);
    rbmUseHiddenGaussianToggleButton->setBounds (840, 92, 156, 24);
    //[UserResized] Add your own custom resize handling here..
    //[/UserResized]
}

void MainComponent::buttonClicked (Button* buttonThatWasClicked)
{
    //[UserbuttonClicked_Pre]
    //[/UserbuttonClicked_Pre]

    if (buttonThatWasClicked == trainButton)
    {
        //[UserButtonCode_trainButton] -- add your button handler code here..
		if (isThreadRunning())
		{
			m_doStop = true;
		}
		else
		{
			m_doStop = false;
			startThread();
		}
        //[/UserButtonCode_trainButton]
    }
    else if (buttonThatWasClicked == addButton)
    {
        //[UserButtonCode_addButton] -- add your button handler code here..
		RbmComponent *pComp = static_cast<RbmComponent*>(m_stack->getLayer(0));
		addTraining(pComp->DrawTraining->getData());
        //[/UserButtonCode_addButton]
    }
    else if (buttonThatWasClicked == ShakeButton)
    {
        //[UserButtonCode_ShakeButton] -- add your button handler code here..
    	m_pLayer->weightsInit(weightInitLabel->getText().getFloatValue());
		m_pLayer->redrawWeights();
    	m_pLayer->redrawReconstruction();
        //[/UserButtonCode_ShakeButton]
    }
    else if (buttonThatWasClicked == createButton)
    {
        //[UserButtonCode_createButton] -- add your button handler code here..
		int next_index = m_rbmSelect->getNumItems();
		int numVisX = numVisibleLabel->getText().getIntValue();
		int numVisY = numVisibleYLabel->getText().getIntValue();
		int numHid = numHiddenLabel->getText().getIntValue();
		if (!m_stack)
		{
			m_stack = new Stack(std::string(projectNameLabel->getText().getCharPointer()));
		}
		m_stack->addLayer(onConstruct("Layer", next_index, numVisX, numVisY, numHid));
		m_rbmSelect->addItem(String(next_index), next_index+1);
		m_rbmSelect->setSelectedId(next_index+1, sendNotification);
        //[/UserButtonCode_createButton]
    }
    else if (buttonThatWasClicked == loadButton)
    {
        //[UserButtonCode_loadButton] -- add your button handler code here..
		m_stack = new Stack(std::string(projectNameLabel->getText().getCharPointer()));
		m_stack->load(this);
		m_stack->loadWeights();
		m_rbmSelect->clear(dontSendNotification);
		
		for (int id=0; id < m_stack->numLayers(); id++)
		{
			m_rbmSelect->addItem(String(id), id+1);
		}		
		m_rbmSelect->setSelectedId(1, sendNotification);
		loadTraining();
        //[/UserButtonCode_loadButton]
    }
    else if (buttonThatWasClicked == saveButton)
    {
        //[UserButtonCode_saveButton] -- add your button handler code here..
		save();
        //[/UserButtonCode_saveButton]
    }
    else if (buttonThatWasClicked == loadTrainingButton)
    {
        //[UserButtonCode_loadTrainingButton] -- add your button handler code here..
		loadTraining();
        //[/UserButtonCode_loadTrainingButton]
    }
    else if (buttonThatWasClicked == saveTrainingButton)
    {
        //[UserButtonCode_saveTrainingButton] -- add your button handler code here..
		saveTraining();
        //[/UserButtonCode_saveTrainingButton]
    }
    else if (buttonThatWasClicked == clearTrainingButton)
    {
        //[UserButtonCode_clearTrainingButton] -- add your button handler code here..
		clearTraining();
        //[/UserButtonCode_clearTrainingButton]
    }
    else if (buttonThatWasClicked == removeTrainingButton)
    {
        //[UserButtonCode_removeTrainingButton] -- add your button handler code here..
		removeTrainingAt((int)patterSlider->getValue());
        //[/UserButtonCode_removeTrainingButton]
    }
    else if (buttonThatWasClicked == rbmDoRaoBlackwellToggleButton)
    {
        //[UserButtonCode_rbmDoRaoBlackwellToggleButton] -- add your button handler code here..
    	m_pLayer->params().doRaoBlackwell = buttonThatWasClicked->getToggleState();
        //[/UserButtonCode_rbmDoRaoBlackwellToggleButton]
    }
    else if (buttonThatWasClicked == rbmDoSampleVisibleToggleButton)
    {
        //[UserButtonCode_rbmDoSampleVisibleToggleButton] -- add your button handler code here..
		m_pLayer->params().gibbsDoSampleVisible = buttonThatWasClicked->getToggleState();
        //[/UserButtonCode_rbmDoSampleVisibleToggleButton]
    }
    else if (buttonThatWasClicked == rbmUseVisibleGaussianToggleButton)
    {
        //[UserButtonCode_rbmUseVisibleGaussianToggleButton] -- add your button handler code here..
        //[/UserButtonCode_rbmUseVisibleGaussianToggleButton]
    }
    else if (buttonThatWasClicked == rbmDoSparseToggleButton)
    {
        //[UserButtonCode_rbmDoSparseToggleButton] -- add your button handler code here..
        //[/UserButtonCode_rbmDoSparseToggleButton]
    }
    else if (buttonThatWasClicked == rbmLearnVarianceButton)
    {
        //[UserButtonCode_rbmLearnVarianceButton] -- add your button handler code here..
        //[/UserButtonCode_rbmLearnVarianceButton]
    }
    else if (buttonThatWasClicked == rbmNormalizeDataToggleButton)
    {
        //[UserButtonCode_rbmNormalizeDataToggleButton] -- add your button handler code here..
        //[/UserButtonCode_rbmNormalizeDataToggleButton]
    }
    else if (buttonThatWasClicked == rbmDoSampleBatch)
    {
        //[UserButtonCode_rbmDoSampleBatch] -- add your button handler code here..
		m_pLayer->params().doSampleBatch = buttonThatWasClicked->getToggleState();
        //[/UserButtonCode_rbmDoSampleBatch]
    }
    else if (buttonThatWasClicked == rbmUseHiddenGaussianToggleButton)
    {
        //[UserButtonCode_rbmUseHiddenGaussianToggleButton] -- add your button handler code here..
       //[/UserButtonCode_rbmUseHiddenGaussianToggleButton]
    }

    //[UserbuttonClicked_Post]
    //[/UserbuttonClicked_Post]
}

void MainComponent::sliderValueChanged (Slider* sliderThatWasMoved)
{
    //[UsersliderValueChanged_Pre]
    //[/UsersliderValueChanged_Pre]

    if (sliderThatWasMoved == patterSlider)
    {
        //[UserSliderCode_patterSlider] -- add your slider handling code here..
		m_trainingIndex = (int)sliderThatWasMoved->getValue();
		m_pLayer->setTrainingData(m_stack->trainingData().row(m_trainingIndex));
        //[/UserSliderCode_patterSlider]
    }
    else if (sliderThatWasMoved == WeightsSlider)
    {
        //[UserSliderCode_WeightsSlider] -- add your slider handling code here..
		int weightIndex = (int)sliderThatWasMoved->getValue();
		m_pLayer->redrawWeights(weightIndex);
        //[/UserSliderCode_WeightsSlider]
    }
    else if (sliderThatWasMoved == numGibbsSlider)
    {
        //[UserSliderCode_numGibbsSlider] -- add your slider handling code here..
    	m_pLayer->params().numGibbs = sliderThatWasMoved->getValue();
		m_pLayer->redrawReconstruction();
        //[/UserSliderCode_numGibbsSlider]
    }
    else if (sliderThatWasMoved == m_progressBarSlider)
    {
        //[UserSliderCode_m_progressBarSlider] -- add your slider handling code here..
        //[/UserSliderCode_m_progressBarSlider]
    }

    //[UsersliderValueChanged_Post]
    //[/UsersliderValueChanged_Post]
}

void MainComponent::labelTextChanged (Label* labelThatHasChanged)
{
    //[UserlabelTextChanged_Pre]
    //[/UserlabelTextChanged_Pre]

    if (labelThatHasChanged == numEpochslabel)
    {
        //[UserLabelCode_numEpochslabel] -- add your label text handling code here..
		m_pLayer->params().numEpochs = labelThatHasChanged->getText().getFloatValue();
        //[/UserLabelCode_numEpochslabel]
    }
    else if (labelThatHasChanged == learningRateLabel)
    {
        //[UserLabelCode_learningRateLabel] -- add your label text handling code here..
    	m_pLayer->params().learningRate = labelThatHasChanged->getText().getFloatValue();
        //[/UserLabelCode_learningRateLabel]
    }
    else if (labelThatHasChanged == numVisibleLabel)
    {
        //[UserLabelCode_numVisibleLabel] -- add your label text handling code here..
        //[/UserLabelCode_numVisibleLabel]
    }
    else if (labelThatHasChanged == numHiddenLabel)
    {
        //[UserLabelCode_numHiddenLabel] -- add your label text handling code here..
        //[/UserLabelCode_numHiddenLabel]
    }
    else if (labelThatHasChanged == projectNameLabel)
    {
        //[UserLabelCode_projectNameLabel] -- add your label text handling code here..
        //[/UserLabelCode_projectNameLabel]
    }
    else if (labelThatHasChanged == numVisibleYLabel)
    {
        //[UserLabelCode_numVisibleYLabel] -- add your label text handling code here..
        //[/UserLabelCode_numVisibleYLabel]
    }
    else if (labelThatHasChanged == lambdaLabel)
    {
        //[UserLabelCode_lambdaLabel] -- add your label text handling code here..
        //[/UserLabelCode_lambdaLabel]
    }
    else if (labelThatHasChanged == sigmaLabel)
    {
        //[UserLabelCode_sigmaLabel] -- add your label text handling code here..
        //[/UserLabelCode_sigmaLabel]
    }
    else if (labelThatHasChanged == sparsityLabel)
    {
        //[UserLabelCode_sparsityLabel] -- add your label text handling code here..
        //[/UserLabelCode_sparsityLabel]
    }
    else if (labelThatHasChanged == sigmaDecayLabel)
    {
        //[UserLabelCode_sigmaDecayLabel] -- add your label text handling code here..
         //[/UserLabelCode_sigmaDecayLabel]
    }
    else if (labelThatHasChanged == weightDecayLabel)
    {
        //[UserLabelCode_weightDecayLabel] -- add your label text handling code here..
    	m_pLayer->params().weightDecay = labelThatHasChanged->getText().getFloatValue();
        //[/UserLabelCode_weightDecayLabel]
    }
    else if (labelThatHasChanged == momentumLabel)
    {
        //[UserLabelCode_momentumLabel] -- add your label text handling code here..
    	m_pLayer->params().momentum = labelThatHasChanged->getText().getFloatValue();
        //[/UserLabelCode_momentumLabel]
    }
    else if (labelThatHasChanged == sparsityLearningRateLabel)
    {
        //[UserLabelCode_sparsityLearningRateLabel] -- add your label text handling code here..
        //[/UserLabelCode_sparsityLearningRateLabel]
    }
    else if (labelThatHasChanged == weightInitLabel)
    {
        //[UserLabelCode_weightInitLabel] -- add your label text handling code here..
        //[/UserLabelCode_weightInitLabel]
    }
    else if (labelThatHasChanged == sizeMiniBatch)
    {
        //[UserLabelCode_sizeMiniBatch] -- add your label text handling code here..
		m_pLayer->params().miniBatchSize = labelThatHasChanged->getText().getFloatValue();
        //[/UserLabelCode_sizeMiniBatch]
    }

    //[UserlabelTextChanged_Post]
    //[/UserlabelTextChanged_Post]
}

void MainComponent::comboBoxChanged (ComboBox* comboBoxThatHasChanged)
{
    //[UsercomboBoxChanged_Pre]
    //[/UsercomboBoxChanged_Pre]

    if (comboBoxThatHasChanged == m_rbmSelect)
    {
		int index = m_rbmSelect->getSelectedItemIndex();
		m_pLayer = static_cast<RbmComponent*>(m_stack->getLayer(index));
		updateControls();
        //[/UserComboBoxCode_m_rbmSelect]
    }

    //[UsercomboBoxChanged_Post]
    //[/UsercomboBoxChanged_Post]
}

void MainComponent::mouseMove (const MouseEvent& e)
{
    //[UserCode_mouseMove] -- Add your code here...
    //[/UserCode_mouseMove]
}

void MainComponent::mouseEnter (const MouseEvent& e)
{
    //[UserCode_mouseEnter] -- Add your code here...
    //[/UserCode_mouseEnter]
}

void MainComponent::mouseExit (const MouseEvent& e)
{
    //[UserCode_mouseExit] -- Add your code here...
    //[/UserCode_mouseExit]
}

void MainComponent::mouseDown (const MouseEvent& e)
{
    //[UserCode_mouseDown] -- Add your code here...
    //[/UserCode_mouseDown]
}

void MainComponent::mouseDrag (const MouseEvent& e)
{
    //[UserCode_mouseDrag] -- Add your code here...
//	e.eventComponent->setCentrePosition(e.getPosition().x, e.getPosition().y);
	std::cout << "Mouse = " << e.x << "," << e.y << std::endl;
    //[/UserCode_mouseDrag]
}

void MainComponent::mouseUp (const MouseEvent& e)
{
    //[UserCode_mouseUp] -- Add your code here...
    //[/UserCode_mouseUp]
}

void MainComponent::mouseDoubleClick (const MouseEvent& e)
{
    //[UserCode_mouseDoubleClick] -- Add your code here...
    //[/UserCode_mouseDoubleClick]
}

void MainComponent::mouseWheelMove (const MouseEvent& e, const MouseWheelDetails& wheel)
{
    //[UserCode_mouseWheelMove] -- Add your code here...
    //[/UserCode_mouseWheelMove]
}

//[MiscUserCode] You can add your own definitions of your custom methods or any other code here...
void MainComponent::save ()
{
	m_stack->saveTraining();
	m_stack->saveWeights();
	m_stack->save();
}

const juce::String& MainComponent::getBaseDir()
{
	struct passwd *pw = getpwuid(getuid());

	const char *homedir = pw->pw_dir;


	m_baseDir = String(homedir) + String("/.rbm/") + projectNameLabel->getText();

	return m_baseDir;

}

void MainComponent::run()
{
	m_pLayer->train(m_stack->trainingData(), this);
}

bool MainComponent::onProgress(Rbm *pRbm, const Rbm::Status &status)
{
	RbmComponent *pComp = static_cast<RbmComponent*>(pRbm);
	pComp->upPass(pComp->DrawTraining->getData());
	pComp->redrawReconstruction();
	pComp->redrawWeights();

	m_progressBarSlider->setValue(status.progress);
	return !m_doStop;
}

void MainComponent::updateControls()
{
	rbmDoRaoBlackwellToggleButton->setToggleState(m_pLayer->params().doRaoBlackwell, dontSendNotification);
	rbmDoSampleVisibleToggleButton->setToggleState(m_pLayer->params().gibbsDoSampleVisible, dontSendNotification);
	rbmDoSampleBatch->setToggleState(m_pLayer->params().doSampleBatch, dontSendNotification);
	
	weightDecayLabel->setText(String(m_pLayer->params().weightDecay), dontSendNotification);
	learningRateLabel->setText(String(m_pLayer->params().learningRate), dontSendNotification);
	momentumLabel->setText(String(m_pLayer->params().momentum), dontSendNotification);
	numVisibleLabel->setText(String(m_pLayer->numVisibleX()), dontSendNotification );
	numVisibleYLabel->setText(String(m_pLayer->numVisibleY()), dontSendNotification );
	numHiddenLabel->setText(String(m_pLayer->numHidden()), dontSendNotification );
	numEpochslabel->setText(String((int)m_pLayer->params().numEpochs), dontSendNotification);
	sizeMiniBatch->setText(String((int)m_pLayer->params().miniBatchSize), dontSendNotification);

	numGibbsSlider->setValue(m_pLayer->params().numGibbs, dontSendNotification);
	WeightsSlider->setRange(0, m_pLayer->numHidden()-1, 1);
	numGibbsSlider->setValue(m_pLayer->params().numGibbs);
   	WeightsSlider->setValue(m_weightIndex);
	patterSlider->setValue(m_trainingIndex);

}
