/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

/* 
 * File:   Layer.cpp
 * Author: jens
 * 
 * Created on 25. Oktober 2019, 08:13
 */

#include "RbmLayer.hpp"
using namespace std;

RbmLayer::RbmLayer(const string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
: Rbm(numVisibleX*numVisibleY, numHidden)
, upper(nullptr)	
, lower(nullptr)	
, m_name(name)
, m_id(id)
, m_numVisibleX(numVisibleX)
, m_numVisibleY(numVisibleY)
{
	cout << "Create Layer " << m_name << "." << to_string((int)m_id) << endl;
	m_weightsFile = m_name + "." + to_string((int)m_id) + string(".weights.dat");
}

RbmLayer::RbmLayer(const RbmLayer& orig)
: Rbm(orig.bv().n_elem, orig.bh().n_elem)
, upper(nullptr)	
, lower(nullptr)	
, m_name(orig.m_name)
, m_weightsFile(orig.m_weightsFile)
, m_id(orig.m_id)
, m_numVisibleX(orig.m_numVisibleX)
, m_numVisibleY(orig.m_numVisibleY)
{
}

RbmLayer::~RbmLayer()
{
}

bool RbmLayer::loadWeights(const string &prjname)
{
	int numVisibleX;
	int numVisibleY;
	int numHidden;

	string filename = m_weightsFile;
	if (prjname.size() > 0)
	{
		filename = prjname + string(".") + m_weightsFile;
	}

	FILE *pFile = fopen(filename.c_str(),"r");
	if (!pFile)
	{
		std::cout << "loadWeights(): Could not open " << filename << " for reading!" << std::endl;
		return false;
	}

	int result = fscanf(pFile, "%d %d %d\n", &numVisibleX, &numVisibleY, &numHidden);
	if (result < 0)
	{
		return false;
	}

	size_t numVisible = numVisibleX*numVisibleY;
	
	int i, j;
	float v;
	for (i=0; i < numVisible; i++)
	{
		result = fscanf(pFile, "%f", &v);
		if (result > 0)
		{
			m_bv(i) = v;
		}
	}
	for (i=0; i < numHidden; i++)
	{
		result = fscanf(pFile, "%f", &v);
		if (result > 0)
		{
			m_bh(i) = v;
		}
	}
	for (i=0; i < numVisible; i++)
	{
		for (j=0; j < numHidden; j++)
		{

			result = fscanf(pFile, "%f", &v);
			if (result > 0)
			{
				m_w(i, j) = v;
			}
		}
	}
	fclose(pFile);
	
	return true;
}
	
bool RbmLayer::saveWeights(const string &prjname)
{
	string filename = m_weightsFile;
	if (prjname.size() > 0)
	{
		filename = prjname + string(".") + m_weightsFile;
	}

	FILE *pFile = fopen(filename.c_str(),"w");
	if (!pFile)
	{
		std::cout << "saveWeights(): Could not open " << filename << " for writing!" << std::endl;
		return false;
	}

	int numHidden = m_bh.n_elem;
	int numVisible = m_bv.n_elem;
	fprintf(pFile, "%d %d %d\n", (int)m_numVisibleX, (int)m_numVisibleY, numHidden);

	int i, j;

	for (i=0; i < numVisible; i++)
	{
		fprintf(pFile, "%3.6f\n", m_bv(i));
	}
	for (i=0; i < numHidden; i++)
	{
		fprintf(pFile, "%3.6f\n", m_bh(i));
	}
	for (i=0; i < numVisible; i++)
	{
		for (j=0; j < numHidden; j++)
		{
			fprintf(pFile, "%3.6f ", m_w(i,j));
		}
		fprintf(pFile, "\n");
	}
	fclose(pFile);

	return true;
}

Json::Value RbmLayer::toJson() const
{
	std::cout << "Exporting Layer " << to_string((int)m_id) << std::endl;
	Json::Value layer;
	layer["name"] = m_name;
	layer["id"] = (int)m_id;
	layer["weights_file"] = m_weightsFile;
	layer["numVisibleX"] = (int)m_numVisibleX;
	layer["numVisibleY"] = (int)m_numVisibleY;
	layer["numHidden"] = (int)m_bh.n_elem;
	layer["rbm"] = Rbm::toJson();
	
	return layer;

}

arma::mat RbmLayer::up_pass(const arma::mat &hidden)
{
	arma::mat reconstruction = toVisibleProbs(hidden);
	if (upper)
	{
		return upper->up_pass(reconstruction);
	}
	return toHiddenProbs(reconstruction);
}

arma::mat RbmLayer::down_pass(const arma::mat &visible)
{
	arma::mat hidden = toHiddenProbs(visible);
	if (lower)
	{
		return lower->down_pass(hidden);
	}
	return toVisibleProbs(hidden);
}