#include <cstdio>
#include <cmath>
#include <armadillo>
#include "Rbm.hpp"

int main()
{
	printf("Hallo, Welt!\n");

	// Position of a particle                     //    |
	arma::vec Pos = {{0},                         //    | (0,1)
				   {1}};                        //    +---x-->

	// Rotation matrix 
	double phi = -3.1416/2; 
	arma::mat RotM = {{+cos(phi), -sin(phi)},
					{+sin(phi), +cos(phi)}};

	Pos.print("Current position of the particle:");
	std::cout << "Rotating the point " << phi*180/3.1416 << " deg" << std::endl;

	Pos = RotM*Pos;

	Pos.print("New position of the particle:");   //    ^
												//    x (1,0)
												//    | 
	arma::mat Z = arma::eye<arma::mat>(4000,4000);
//	Z.print("Z:");
	printf("Z.n_rows = %d\n", (int)Z.n_rows);
	printf("Z.n_cols = %d\n", (int)Z.n_cols);
	printf("Z.n_elem = %d\n", (int)Z.n_elem);
	//    +------>

	Rbm::Params params;
	Rbm rbm(28*28, 64);
	arma::mat batch = arma::randu(1000, 28*28);
	rbm.train(batch, 1000, 100, params, nullptr);
	
//	arma::mat v = arma::randu(28*28, 1);
//	arma::mat h = rbm.toHidden(v);
//	arma::mat r = rbm.toVisible(h);
	return 0;
}