/*
 * 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.hpp
 * Author: jens
 *
 * Created on 25. Oktober 2019, 08:13
 */

#ifndef LAYER_HPP
#define LAYER_HPP

#include <string>

#include <armadillo>
#include <jsoncpp/json/json.h>
#include "Rbm.hpp"
#include "ILayer.hpp"

class Layer
{
public:
	Layer *upper;
	Layer *lower;

	Layer(const std::string &prjname, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden);
	Layer(const Layer& orig);
	virtual ~Layer();

	Json::Value toJson() const;
	Rbm& rbm();
	void saveWeights();

	arma::mat up_pass(const arma::mat& hidden);
	arma::mat down_pass(const arma::mat& visible);

	size_t id()
	{
		return m_id;
	}
	
private:
	const std::string &m_prjname;
	std::string m_weightsFile;
	size_t m_id;
	size_t m_numVisibleX;
	size_t m_numVisibleY;
	Rbm::Params m_rbm_params;
	Rbm m_rbm;

};

#endif /* LAYER_HPP */