Adjustments for CRAN upload
[valse.git] / pkg / src / a.EMGrank.c
CommitLineData
3453829e
BA
1#include <R.h>
2#include <Rdefines.h>
3#include "EMGrank.h"
4
5// See comments in src/sources/EMGrank.c and R/EMGrank.R (wrapper)
6SEXP EMGrank(
7 SEXP Pi_,
8 SEXP Rho_,
9 SEXP mini_,
10 SEXP maxi_,
11 SEXP X_,
12 SEXP Y_,
13 SEXP eps_,
14 SEXP rank_
15) {
16 // Get matrices dimensions
17 SEXP dimX = getAttrib(X_, R_DimSymbol);
18 int n = INTEGER(dimX)[0];
19 int p = INTEGER(dimX)[1];
20 SEXP dimRho = getAttrib(Rho_, R_DimSymbol);
21 int m = INTEGER(dimRho)[0];
22 int k = INTEGER(dimRho)[2];
23
24 ////////////
25 // INPUTS //
26 ////////////
27
28 // get scalar parameters
29 int mini = INTEGER_VALUE(mini_);
30 int maxi = INTEGER_VALUE(maxi_);
31 double eps = NUMERIC_VALUE(eps_);
32
33 // Get pointers from SEXP arrays ; WARNING: by columns !
34 double* Pi = REAL(Pi_);
35 double* Rho = REAL(Rho_);
36 double* X = REAL(X_);
37 double* Y = REAL(Y_);
38 int* rank = INTEGER(rank_);
39
40 /////////////
41 // OUTPUTS //
42 /////////////
43
44 SEXP phi, LLF, dimPhi;
45 PROTECT(dimPhi = allocVector(INTSXP, 3));
46 int* pDimPhi = INTEGER(dimPhi);
47 pDimPhi[0] = p; pDimPhi[1] = m; pDimPhi[2] = k;
48 PROTECT(phi = allocArray(REALSXP, dimPhi));
49 PROTECT(LLF = allocVector(REALSXP, 1));
50 double *pPhi=REAL(phi), *pLLF=REAL(LLF);
51
52 /////////////////////
53 // Call to EMGrank //
54 /////////////////////
55
56 EMGrank_core(Pi, Rho, mini, maxi, X, Y, eps, rank,
57 pPhi,pLLF,
58 n,p,m,k);
59
60 // Build list from OUT params and return it
61 SEXP listParams, listNames;
62 PROTECT(listParams = allocVector(VECSXP, 2));
63 char* lnames[2] = {"phi", "LLF"}; //lists labels
64 PROTECT(listNames = allocVector(STRSXP,2));
65 for (int i=0; i<2; i++)
66 SET_STRING_ELT(listNames,i,mkChar(lnames[i]));
67 setAttrib(listParams, R_NamesSymbol, listNames);
68 SET_VECTOR_ELT(listParams, 0, phi);
69 SET_VECTOR_ELT(listParams, 1, LLF);
70
71 UNPROTECT(5);
72 return listParams;
73}