/*
 * 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 RBMLAYER_HPP
#define RBMLAYER_HPP

#include <string>

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

class Layer : public Rbm
{
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;
	bool loadWeights(const std::string &prjname="");
	bool saveWeights(const std::string &prjname="");

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

	size_t id()
	{
		return m_id;
	}
	
	int numVisibleX()
	{
		return m_numVisibleX;
	}

	int numVisibleY()
	{
		return m_numVisibleY;
	}
	
	int numHidden()
	{
		return m_bh.n_elem;
	}
private:
	std::string m_name;
	std::string m_weightsFile;
	size_t m_id;
	size_t m_numVisibleX;
	size_t m_numVisibleY;

};

#endif /* RBMLAYER_HPP */