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

  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.

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


#include "RbmComponent.hpp"



//==============================================================================
RbmComponent::RbmComponent (const std::string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
	: Layer(name, id, numVisibleX, numVisibleY, numHidden)
	, m_currWeightIndexToDraw(0)
	, DrawTraining(nullptr)
	, DrawReconstruction(nullptr)
	, DrawWeights(nullptr)
	, DrawHidden(nullptr)
{
	addAndMakeVisible (DrawTraining = new DrawComponent (numVisibleX, numVisibleY));
	DrawTraining->setListener(this);

    addAndMakeVisible (DrawReconstruction = new DrawComponent (numVisibleX, numVisibleY));

	
    addAndMakeVisible (DrawWeights = new DrawComponent (numVisibleX, numVisibleY, 0.5, 0.5));
    addAndMakeVisible (DrawHidden = new DrawComponent (numHidden, 1));
	DrawHidden->setListener(this);
	redrawWeights();
	redrawReconstruction();

    setSize (430, 130);
	resized();
	setBounds (16, 140*id+16, 430, 130);
}

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

    //[Destructor]. You can add your own custom destruction code here..
    DrawTraining = nullptr;
    DrawReconstruction = nullptr;
    DrawWeights = nullptr;
    DrawHidden = nullptr;
    //[/Destructor]
}

//==============================================================================
void RbmComponent::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"),
                32, 306, 80, 14,
                Justification::centred, true);

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

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

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

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

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

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

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

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

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

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

void RbmComponent::resized()
{
	DrawTraining->setBoundsRelative(0, 0, 100./430, 100./130);
	DrawReconstruction->setBoundsRelative(110./430, 0, 100./430, 100./130);
    DrawWeights->setBoundsRelative (220./430, 0, 100./430, 100./130);
    DrawHidden->setBoundsRelative (0, 110./130, 430./430, 20./130);
}

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

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

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

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

void RbmComponent::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 RbmComponent::mouseUp (const MouseEvent& e)
{
    //[UserCode_mouseUp] -- Add your code here...
    //[/UserCode_mouseUp]
}

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

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

void RbmComponent::onDraw(DrawComponent &obj)
{
	if (&obj == DrawHidden)
	{
		downPass(toVisibleProbs(obj.getData()));
	}
	if (&obj == DrawTraining)
	{
    	redrawReconstruction();
	}
}

void RbmComponent::redrawReconstruction()
{
	uint32_t i;

	for (i=0; i < params().numGibbs; i++)
	{
		downPass(DrawReconstruction->getData());
	}
	DrawReconstruction->DrawData();
}

void RbmComponent::upPass(const arma::mat& v)
{
	DrawReconstruction->getData() = v;
	DrawReconstruction->DrawData();
	if (next)
	{
		RbmComponent *pComp = static_cast<RbmComponent*> (next);
		pComp->upPass(toHiddenProbs(v));
	}
}

void RbmComponent::downPass(const arma::mat& v)
{
	DrawReconstruction->getData() = v;
	DrawReconstruction->DrawData();
	if (prev)
	{
		RbmComponent *pComp = static_cast<RbmComponent*> (prev);
		pComp->downPass(prev->toVisibleProbs(v));
	}
}

arma::mat RbmComponent::getConvolutedWeight(arma::mat const &w)
{
//	if (next)
//	{
//		arma::mat wc = w * next->w().t(); // * 1.0/sqrt((double)w.cols());
//		return static_cast<RbmComponent*>(next)->getConvolutedWeight(wc);
//	}
//	else
	{
		return w;
	}
	
}
void RbmComponent::redrawWeights()
{
	DrawWeights->getData() = getConvolutedWeight(w().col(m_currWeightIndexToDraw));
	DrawWeights->DrawData();
}

void RbmComponent::redrawWeights(size_t index)
{
	m_currWeightIndexToDraw = index;
	redrawWeights();
}

void RbmComponent::setTrainingData(arma::mat const& batch)
{
	
	DrawTraining->getData() = trainingData(batch);
	DrawTraining->DrawData();
		
	redrawReconstruction();
	upPass(DrawTraining->getData());
}
//[/MiscUserCode]


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