| 1 | #include "EMGrank.h" |
| 2 | #include "ioutils.h" |
| 3 | |
| 4 | int main(int argc, char** argv) |
| 5 | { |
| 6 | // read dimensions |
| 7 | const int nbDims = 4; |
| 8 | int* dimensions = readArray_int("dimensions",&nbDims,1); |
| 9 | int n = dimensions[0]; |
| 10 | int p = dimensions[1]; |
| 11 | int m = dimensions[2]; |
| 12 | int k = dimensions[3]; |
| 13 | free(dimensions); |
| 14 | int lengthOne = 1; |
| 15 | |
| 16 | //////////// |
| 17 | // INPUTS // |
| 18 | //////////// |
| 19 | |
| 20 | // Rho |
| 21 | const int dimRho[] = {m, m, k}; |
| 22 | float* Rho = readArray_real("Rho",dimRho,3); |
| 23 | |
| 24 | // Pi |
| 25 | float* Pi = readArray_real("Pi",&k,1); |
| 26 | |
| 27 | // min number of iterations |
| 28 | int* pmini = readArray_int("mini",&lengthOne,1); |
| 29 | int mini = *pmini; |
| 30 | free(pmini); |
| 31 | |
| 32 | // max number of iterations |
| 33 | int* pmaxi = readArray_int("maxi",&lengthOne,1); |
| 34 | int maxi = *pmaxi; |
| 35 | free(pmaxi); |
| 36 | |
| 37 | // X |
| 38 | const int dimX[] = {n, p}; |
| 39 | float* X = readArray_real("X",dimX, 2); |
| 40 | |
| 41 | // Y |
| 42 | const int dimY[] = {n, m}; |
| 43 | float* Y = readArray_real("Y",dimY, 2); |
| 44 | |
| 45 | // tau |
| 46 | float* ptau = readArray_real("tau",&lengthOne,1); |
| 47 | float tau = *ptau; |
| 48 | free(ptau); |
| 49 | |
| 50 | // tau |
| 51 | int* rank = readArray_int("rank",&k,1); |
| 52 | |
| 53 | ///////////// |
| 54 | // OUTPUTS // |
| 55 | ///////////// |
| 56 | |
| 57 | // phi |
| 58 | const int dimPhi[] = {p, m, k}; |
| 59 | float* phi = (float*)malloc(dimPhi[0]*dimPhi[1]*dimPhi[2]*sizeof(float)); |
| 60 | |
| 61 | // LLF |
| 62 | float* LLF = (float*)malloc(1*sizeof(float)); |
| 63 | |
| 64 | ////////////////////////// |
| 65 | // Main call to EMGrank // |
| 66 | ////////////////////////// |
| 67 | |
| 68 | EMGrank(Pi,Rho,mini,maxi,X,Y,tau,rank, |
| 69 | phi,LLF, |
| 70 | n,p,m,k); |
| 71 | |
| 72 | // free input pointers |
| 73 | free(Rho); |
| 74 | free(Pi); |
| 75 | free(X); |
| 76 | free(Y); |
| 77 | free(rank); |
| 78 | |
| 79 | // Compare to reference outputs |
| 80 | float* ref_phi = readArray_real("phi",dimPhi, 3); |
| 81 | compareArray_real("phi", phi, ref_phi, dimPhi[0]*dimPhi[1]*dimPhi[2]); |
| 82 | free(phi); |
| 83 | free(ref_phi); |
| 84 | |
| 85 | // LLF |
| 86 | float* ref_LLF = readArray_real("LLF",&lengthOne,1); |
| 87 | compareArray_real("LLF", LLF, ref_LLF, 1); |
| 88 | free(LLF); |
| 89 | free(ref_LLF); |
| 90 | |
| 91 | return 0; |
| 92 | } |