#include <iostream>
#include <gsl/gsl_rng.h>
#include "tref.hpp"
#include "trefsim.hpp"

int main()
{
	long unsigned int i, l;
	tref::smalltref sys(10,5,50,0.5,0,1e-12,true);
	unsigned int M = sys.objects(), N = sys.users();
	gsl_rng *r = gsl_rng_alloc(gsl_rng_mt19937);
	for(i=0;i<N;++i)
		for(l=0;l<M;++l)
			sys.addRating(l,i,gsl_rng_uniform_pos(r));
	std::cout << "Raw ratings: " << std::endl;
	sys.printRatings();
	std::cout << std::endl;
	i = sys.iterationCycle();
	std::cout << "Calculated optimum quality in " << i << " iterations..." << std::endl;
	for(l=0;l<M;++l)
		std::cout << "Object " << l << " weighted quality estimate: " << sys.objectQuality(l) << std::endl;
	std::cout << std::endl;

	gsl_rng_set(r,1001);
	tref::multisim<tref::sim,tref::smalltref>
		sim(/* sims */ 100,
		    /*objects*/ 1000, /*users*/ 1000,
		    /*sparsity*/ 1, /*beta*/ 0.8, /*epsilon*/ 1e-36, /*convergence*/ 1e-12,
		    /*qualityScale*/ 10, /*abilityScale*/ 1, /*ratingScale*/ 1,
		    /*rng*/ r, /*verbose*/ true);
	std::cout << "Delta = " << sim.qualityDeltaMean() << " +/- " << sim.qualityDeltaSD();
	std::cout << " calculated in " << sim.iterationsMean() << " +/- " << sim.iterationsSD() << " iterations." << std::endl;
	gsl_rng_free(r);
	return 0;
}