projects
/
valse.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
cosmetics (rename tau to eps everywhere)
[valse.git]
/
pkg
/
src
/
adapters
/
a.EMGLLF.c
diff --git
a/pkg/src/adapters/a.EMGLLF.c
b/pkg/src/adapters/a.EMGLLF.c
index
0df00bd
..
1503315
100644
(file)
--- a/
pkg/src/adapters/a.EMGLLF.c
+++ b/
pkg/src/adapters/a.EMGLLF.c
@@
-2,6
+2,7
@@
#include <Rdefines.h>
#include "EMGLLF.h"
#include <Rdefines.h>
#include "EMGLLF.h"
+// See comments in src/sources/EMGLLF.c and R/EMGLLF.R (wrapper)
SEXP EMGLLF(
SEXP phiInit_,
SEXP rhoInit_,
SEXP EMGLLF(
SEXP phiInit_,
SEXP rhoInit_,
@@
-13,7
+14,7
@@
SEXP EMGLLF(
SEXP lambda_,
SEXP X_,
SEXP Y_,
SEXP lambda_,
SEXP X_,
SEXP Y_,
- SEXP
tau
_
+ SEXP
eps
_
) {
// Get matrices dimensions
int n = INTEGER(getAttrib(X_, R_DimSymbol))[0];
) {
// Get matrices dimensions
int n = INTEGER(getAttrib(X_, R_DimSymbol))[0];
@@
-31,7
+32,7
@@
SEXP EMGLLF(
int maxi = INTEGER_VALUE(maxi_);
double gamma = NUMERIC_VALUE(gamma_);
double lambda = NUMERIC_VALUE(lambda_);
int maxi = INTEGER_VALUE(maxi_);
double gamma = NUMERIC_VALUE(gamma_);
double lambda = NUMERIC_VALUE(lambda_);
- double
tau = NUMERIC_VALUE(tau
_);
+ double
eps = NUMERIC_VALUE(eps
_);
// Get pointers from SEXP arrays ; WARNING: by columns !
double* phiInit = REAL(phiInit_);
// Get pointers from SEXP arrays ; WARNING: by columns !
double* phiInit = REAL(phiInit_);
@@
-45,7
+46,7
@@
SEXP EMGLLF(
// OUTPUTS //
/////////////
// OUTPUTS //
/////////////
- SEXP phi, rho, pi, LLF, S, dimPhiS, dimRho;
+ SEXP phi, rho, pi, LLF, S,
affec,
dimPhiS, dimRho;
PROTECT(dimPhiS = allocVector(INTSXP, 3));
int* pDimPhiS = INTEGER(dimPhiS);
pDimPhiS[0] = p; pDimPhiS[1] = m; pDimPhiS[2] = k;
PROTECT(dimPhiS = allocVector(INTSXP, 3));
int* pDimPhiS = INTEGER(dimPhiS);
pDimPhiS[0] = p; pDimPhiS[1] = m; pDimPhiS[2] = k;
@@
-57,22
+58,25
@@
SEXP EMGLLF(
PROTECT(pi = allocVector(REALSXP, k));
PROTECT(LLF = allocVector(REALSXP, maxi-mini+1));
PROTECT(S = allocArray(REALSXP, dimPhiS));
PROTECT(pi = allocVector(REALSXP, k));
PROTECT(LLF = allocVector(REALSXP, maxi-mini+1));
PROTECT(S = allocArray(REALSXP, dimPhiS));
+ PROTECT(affec = allocVector(INTSXP, n));
double *pPhi=REAL(phi), *pRho=REAL(rho), *pPi=REAL(pi), *pLLF=REAL(LLF), *pS=REAL(S);
double *pPhi=REAL(phi), *pRho=REAL(rho), *pPi=REAL(pi), *pLLF=REAL(LLF), *pS=REAL(S);
+ int *pAffec=INTEGER(affec);
////////////////////
// Call to EMGLLF //
////////////////////
////////////////////
// Call to EMGLLF //
////////////////////
- EMGLLF_core(phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,lambda,X,Y,
tau
,
- pPhi,pRho,pPi,pLLF,pS,
+ EMGLLF_core(phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,lambda,X,Y,
eps
,
+ pPhi,pRho,pPi,pLLF,pS,
pAffec,
n,p,m,k);
// Build list from OUT params and return it
SEXP listParams, listNames;
n,p,m,k);
// Build list from OUT params and return it
SEXP listParams, listNames;
- PROTECT(listParams = allocVector(VECSXP, 5));
- char* lnames[5] = {"phi", "rho", "pi", "LLF", "S"}; //lists labels
- PROTECT(listNames = allocVector(STRSXP,5));
- for (int i=0; i<5; i++)
+ int nouts = 6;
+ PROTECT(listParams = allocVector(VECSXP, nouts));
+ char* lnames[6] = {"phi", "rho", "pi", "LLF", "S", "affec"}; //lists labels
+ PROTECT(listNames = allocVector(STRSXP,nouts));
+ for (int i=0; i<nouts; i++)
SET_STRING_ELT(listNames,i,mkChar(lnames[i]));
setAttrib(listParams, R_NamesSymbol, listNames);
SET_VECTOR_ELT(listParams, 0, phi);
SET_STRING_ELT(listNames,i,mkChar(lnames[i]));
setAttrib(listParams, R_NamesSymbol, listNames);
SET_VECTOR_ELT(listParams, 0, phi);
@@
-80,7
+84,8
@@
SEXP EMGLLF(
SET_VECTOR_ELT(listParams, 2, pi);
SET_VECTOR_ELT(listParams, 3, LLF);
SET_VECTOR_ELT(listParams, 4, S);
SET_VECTOR_ELT(listParams, 2, pi);
SET_VECTOR_ELT(listParams, 3, LLF);
SET_VECTOR_ELT(listParams, 4, S);
+ SET_VECTOR_ELT(listParams, 5, affec);
- UNPROTECT(
9
);
+ UNPROTECT(
10
);
return listParams;
}
return listParams;
}