#include "EMGLLF.h"
#include "test_utils.h"
#include <stdlib.h>

int main(int argc, char** argv)
{
	int* dimensions = readArray_int("dimensions");
	int n = dimensions[0];
	int p = dimensions[1];
	int m = dimensions[2];
	int k = dimensions[3];
	free(dimensions);

	////////////
	// INPUTS //
	Real* phiInit = readArray_real("phiInit");
	Real* rhoInit = readArray_real("rhoInit");
	Real* piInit = readArray_real("piInit");
	Real* gamInit = readArray_real("gamInit");
	int mini = read_int("mini");
	int maxi = read_int("maxi");
	Real gamma = read_real("gamma");
	Real lambda = read_real("lambda");
	Real* X = readArray_real("X");
	Real* Y = readArray_real("Y");
	Real eps = read_real("eps");
	////////////

	/////////////
	// OUTPUTS //
	Real* phi = (Real*)malloc(p*m*k*sizeof(Real));
	Real* rho = (Real*)malloc(m*m*k*sizeof(Real));
	Real* pi = (Real*)malloc(k*sizeof(Real));
	Real llh;
	Real* S = (Real*)malloc(p*m*k*sizeof(Real));
	int* affec = (int*)malloc(n*sizeof(int));
	/////////////

	////////////////////
	// Call to EMGLLF //
	EMGLLF_core(phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,lambda,X,Y,eps,
		phi,rho,pi,&llh,S,affec,
		n,p,m,k);
	////////////////////

	free(phiInit);
	free(rhoInit);
	free(piInit);
	free(gamInit);
	free(X);
	free(Y);

	// Compare to reference outputs
	Real* ref_phi = readArray_real("phi");
	compareArray_real("phi", phi, ref_phi, p*m*k);
	free(phi);
	free(ref_phi);

	Real* ref_rho = readArray_real("rho");
	compareArray_real("rho", rho, ref_rho, m*m*k);
	free(rho);
	free(ref_rho);

	Real* ref_pi = readArray_real("pi");
	compareArray_real("pi", pi, ref_pi, k);
	free(pi);
	free(ref_pi);

	Real ref_llh = read_real("llh");
	compareArray_real("llh", &llh, &ref_llh, 1);

	Real* ref_S = readArray_real("S");
	compareArray_real("S", S, ref_S, p*m*k);
	free(S);
	free(ref_S);

	int* ref_affec = readArray_int("affec");
	compareArray_int("affec", affec, ref_affec, n);
	free(affec);
	free(ref_affec);

	return 0;
}