134 lines
3.8 KiB
C
134 lines
3.8 KiB
C
|
/************************************************************************
|
||
|
|
* mexschurfun(X,Y,options)
|
||
|
|
* options = 1, add Y to the diagonal of X (a square matrix)
|
||
|
|
* options = 2, add Y to X
|
||
|
|
************************************************************************/
|
||
|
|
|
||
|
|
#include "mex.h"
|
||
|
|
#include <math.h>
|
||
|
|
#include <matrix.h>
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
PROCEDURE mexFunction - Entry for Matlab
|
||
|
|
*********************************************************************/
|
||
|
|
void mexFunction(const int nlhs, mxArray *plhs[],
|
||
|
|
const int nrhs, const mxArray *prhs[])
|
||
|
|
{
|
||
|
|
double *X, *Y, *Ytmp;
|
||
|
|
mwIndex *irX, *jcX, *irY, *jcY;
|
||
|
|
int n, isspX, isspY, j, jn, k, kstart, kend, r, kstart2, kend2;
|
||
|
|
int options, scalarY;
|
||
|
|
double tmp, tmp2, alpha;
|
||
|
|
|
||
|
|
if(nrhs < 2)
|
||
|
|
mexErrMsgTxt("mexschurfun: requires at least 2 input arguments.");
|
||
|
|
if(nlhs > 0)
|
||
|
|
mexErrMsgTxt("mexschurfun: requires no output argument.");
|
||
|
|
|
||
|
|
X = mxGetPr(prhs[0]);
|
||
|
|
isspX = mxIsSparse(prhs[0]);
|
||
|
|
if (isspX) {
|
||
|
|
irX = mxGetIr(prhs[0]);
|
||
|
|
jcX = mxGetJc(prhs[0]);
|
||
|
|
}
|
||
|
|
n = mxGetM(prhs[0]);
|
||
|
|
if (n != mxGetN(prhs[0])) {
|
||
|
|
mexErrMsgTxt("X should be square."); }
|
||
|
|
isspY = mxIsSparse(prhs[1]);
|
||
|
|
Y = mxGetPr(prhs[1]);
|
||
|
|
if (isspY) {
|
||
|
|
irY = mxGetIr(prhs[1]);
|
||
|
|
jcY = mxGetJc(prhs[1]);
|
||
|
|
}
|
||
|
|
if (nrhs == 2) {
|
||
|
|
if ((mxGetM(prhs[1]) == n) & (mxGetN(prhs[1]) == n)) {
|
||
|
|
options = 2;
|
||
|
|
} else {
|
||
|
|
options = 1;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
options = (int) (*mxGetPr(prhs[2]));
|
||
|
|
}
|
||
|
|
if (options == 1) {
|
||
|
|
if ((mxGetN(prhs[1]) != 1) & (mxGetM(prhs[1]) != 1)) {
|
||
|
|
mexErrMsgTxt("mexschurfun: Y should be a vector."); }
|
||
|
|
} else {
|
||
|
|
if ((mxGetN(prhs[1]) == 1) & (mxGetM(prhs[1]) == 1)) {
|
||
|
|
scalarY = 1;
|
||
|
|
alpha = Y[0];
|
||
|
|
} else {
|
||
|
|
scalarY = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
/********************************************************/
|
||
|
|
Ytmp = mxCalloc(n,sizeof(double));
|
||
|
|
if (options==1) {
|
||
|
|
if (isspY) {
|
||
|
|
for (k=0; k<jcY[1]; ++k) { r=irY[k]; Ytmp[r]=Y[k]; }
|
||
|
|
} else {
|
||
|
|
for (k=0; k<n; ++k) { Ytmp[k]=Y[k]; }
|
||
|
|
}
|
||
|
|
if (isspX) {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
kstart = jcX[j]; kend = jcX[j+1];
|
||
|
|
for (k=kstart; k<kend; k++) {
|
||
|
|
r = irX[k];
|
||
|
|
if (r==j) { X[k] += Ytmp[j]; break; }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
for (j=0; j<n; j++) { jn = j*n; X[j+jn] += Ytmp[j]; }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (scalarY==1) {
|
||
|
|
if (isspX) {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
kstart = jcX[j]; kend = jcX[j+1];
|
||
|
|
for (k=kstart; k<kend; k++) {
|
||
|
|
r = irX[k];
|
||
|
|
X[k] += alpha; }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
jn = j*n;
|
||
|
|
for (k=0; k<n; k++) { X[k+jn] += alpha; }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (isspX & !isspY) {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
kstart = jcX[j]; kend = jcX[j+1]; jn = j*n;
|
||
|
|
for (k=kstart; k<kend; k++) {
|
||
|
|
r = irX[k];
|
||
|
|
X[k] += Y[r+jn]; }
|
||
|
|
}
|
||
|
|
} else if (!isspX & !isspY) {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
jn = j*n;
|
||
|
|
for (k=0; k<n; k++) { X[k+jn] += Y[k+jn]; }
|
||
|
|
}
|
||
|
|
} else if (!isspX & isspY) {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
kstart = jcY[j]; kend = jcY[j+1]; jn = j*n;
|
||
|
|
for (k=kstart; k<kend; k++) {
|
||
|
|
r = irY[k];
|
||
|
|
X[r+jn] += Y[k]; }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
for (j=0; j<n; j++) {
|
||
|
|
kstart2 = jcY[j]; kend2 = jcY[j+1];
|
||
|
|
for (k=kstart2; k<kend2; ++k) { r=irY[k]; Ytmp[r]=Y[k]; }
|
||
|
|
kstart = jcX[j]; kend = jcX[j+1];
|
||
|
|
for (k=kstart; k<kend; k++) {
|
||
|
|
r = irX[k];
|
||
|
|
X[k] += Ytmp[r]; }
|
||
|
|
for (k=kstart2; k<kend2; ++k) { r=irY[k]; Ytmp[r]=0.0; }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
mxFree(Ytmp);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
/************************************************************************/
|