static PyObject* symbolic(PyObject *self, PyObject *args)
{
spmatrix *A;
double info[UMFPACK_INFO];
void *symbolic;
if (!PyArg_ParseTuple(args, "O", &A)) return NULL;
if (!SpMatrix_Check(A)) PY_ERR_TYPE("A must be a sparse matrix");
if (SP_NCOLS(A) == 0 || SP_NROWS(A) == 0) {
PyErr_SetString(PyExc_ValueError, "A must have at least one "
"row and column");
return NULL;
}
switch (SP_ID(A)){
case DOUBLE:
UMFD(symbolic)(SP_NROWS(A), SP_NCOLS(A), SP_COL(A),
SP_ROW(A), SP_VAL(A), &symbolic, NULL, info);
if (info[UMFPACK_STATUS] == UMFPACK_OK)
#if PY_MAJOR_VERSION >= 3
return (PyObject *) PyCapsule_New( (void *) symbolic,
"UMFPACK SYM D FACTOR",
(PyCapsule_Destructor) &free_umfpack_d_symbolic);
#else
return (PyObject *) PyCObject_FromVoidPtrAndDesc(
(void *) symbolic, "UMFPACK SYM D FACTOR",
free_umfpack_d_symbolic);
#endif
else
UMFD(free_symbolic)(&symbolic);
break;
case COMPLEX:
UMFZ(symbolic)(SP_NROWS(A), SP_NCOLS(A), SP_COL(A),
SP_ROW(A), SP_VAL(A), NULL, &symbolic, NULL, info);
if (info[UMFPACK_STATUS] == UMFPACK_OK)
#if PY_MAJOR_VERSION >= 3
return (PyObject *) PyCapsule_New(
(void *) symbolic, "UMFPACK SYM Z FACTOR",
(PyCapsule_Destructor) &free_umfpack_z_symbolic);
#else
return (PyObject *) PyCObject_FromVoidPtrAndDesc(
(void *) symbolic, "UMFPACK SYM Z FACTOR",
free_umfpack_z_symbolic);
#endif
else
static PyObject* getfactor(PyObject *self, PyObject *args)
{
PyObject *F;
cholmod_factor *Lf;
cholmod_sparse *Ls;
#if PY_MAJOR_VERSION >= 3
const char *descr;
#else
char *descr;
#endif
if (!set_options()) return NULL;
if (!PyArg_ParseTuple(args, "O", &F)) return NULL;
#if PY_MAJOR_VERSION >= 3
if (!PyCapsule_CheckExact(F) || !(descr = PyCapsule_GetName(F)))
err_CO("F");
if (strncmp(descr, "CHOLMOD FACTOR", 14))
PY_ERR_TYPE("F is not a CHOLMOD factor");
Lf = (cholmod_factor *) PyCapsule_GetPointer(F, descr);
#else
if (!PyCObject_Check(F)) err_CO("F");
descr = PyCObject_GetDesc(F);
if (!descr || strncmp(descr, "CHOLMOD FACTOR", 14))
PY_ERR_TYPE("F is not a CHOLMOD factor");
Lf = (cholmod_factor *) PyCObject_AsVoidPtr(F);
#endif
/* Check factorization */
if (Lf->xtype == CHOLMOD_PATTERN)
PY_ERR(PyExc_ValueError, "F must be a numeric Cholesky factor");
if (!(Ls = CHOL(factor_to_sparse)(Lf, &Common)))
return PyErr_NoMemory();
spmatrix *ret = SpMatrix_New(Ls->nrow, Ls->ncol, Ls->nzmax,
(Ls->xtype == CHOLMOD_REAL ? DOUBLE : COMPLEX));
if (!ret) {
CHOL(free_sparse)(&Ls, &Common);
return PyErr_NoMemory();
}
memcpy(SP_COL(ret), Ls->p, (Ls->ncol+1)*sizeof(int_t));
memcpy(SP_ROW(ret), Ls->i, (Ls->nzmax)*sizeof(int_t));
memcpy(SP_VAL(ret), Ls->x, (Ls->nzmax)*E_SIZE[SP_ID(ret)]);
CHOL(free_sparse)(&Ls, &Common);
return (PyObject *)ret;
}
static cholmod_sparse * create_matrix(spmatrix *A)
{
cholmod_sparse *B;
if (!(B = CHOL(allocate_sparse)(SP_NROWS(A), SP_NCOLS(A), 0,
1, 0, 0, (SP_ID(A) == DOUBLE ? CHOLMOD_REAL : CHOLMOD_COMPLEX),
&Common))) return NULL;
int i;
for (i=0; i<SP_NCOLS(A); i++)
((int_t *)B->nz)[i] = SP_COL(A)[i+1] - SP_COL(A)[i];
B->x = SP_VAL(A);
B->i = SP_ROW(A);
B->nzmax = SP_NNZ(A);
memcpy(B->p, SP_COL(A), (SP_NCOLS(A)+1)*sizeof(int_t));
return B;
}
static PyObject* splinsolve(PyObject *self, PyObject *args,
PyObject *kwrds)
{
spmatrix *A, *B, *X;
matrix *P=NULL;
int n, nnz;
cholmod_sparse *Ac=NULL, *Bc=NULL, *Xc=NULL;
cholmod_factor *L=NULL;
#if PY_MAJOR_VERSION >= 3
int uplo_='L';
#endif
char uplo='L';
char *kwlist[] = {"A", "B", "p", "uplo", NULL};
if (!set_options()) return NULL;
#if PY_MAJOR_VERSION >= 3
if (!PyArg_ParseTupleAndKeywords(args, kwrds, "OO|OC", kwlist, &A,
&B, &P, &uplo_)) return NULL;
uplo = (char) uplo_;
#else
if (!PyArg_ParseTupleAndKeywords(args, kwrds, "OO|Oc", kwlist, &A,
&B, &P, &uplo)) return NULL;
#endif
if (!SpMatrix_Check(A) || SP_NROWS(A) != SP_NCOLS(A))
PY_ERR_TYPE("A is not a square sparse matrix");
n = SP_NROWS(A);
nnz = SP_NNZ(A);
if (!SpMatrix_Check(B) || SP_ID(A) != SP_ID(B))
PY_ERR_TYPE("B must be a sparse matrix of the same type as A");
if (SP_NROWS(B) != n)
PY_ERR(PyExc_ValueError, "incompatible dimensions for B");
if (P) {
if (!Matrix_Check(P) || MAT_ID(P) != INT) err_int_mtrx("p");
if (MAT_LGT(P) != n) err_buf_len("p");
if (!CHOL(check_perm)(P->buffer, n, n, &Common))
PY_ERR(PyExc_ValueError, "not a valid permutation");
}
if (uplo != 'U' && uplo != 'L') err_char("uplo", "'L', 'U'");
if (!(Ac = pack(A, uplo))) return PyErr_NoMemory();
L = CHOL(analyze_p) (Ac, P ? MAT_BUFI(P): NULL, NULL, 0, &Common);
if (Common.status != CHOLMOD_OK){
CHOL(free_factor)(&L, &Common);
CHOL(free_sparse)(&Ac, &Common);
if (Common.status == CHOLMOD_OUT_OF_MEMORY)
return PyErr_NoMemory();
else {
PyErr_SetString(PyExc_ValueError, "symbolic factorization "
"failed");
return NULL;
}
}
CHOL(factorize) (Ac, L, &Common);
CHOL(free_sparse)(&Ac, &Common);
if (Common.status > 0) switch (Common.status) {
case CHOLMOD_NOT_POSDEF:
PyErr_SetObject(PyExc_ArithmeticError, Py_BuildValue("i",
L->minor));
CHOL(free_factor)(&L, &Common);
return NULL;
break;
case CHOLMOD_DSMALL:
/* This never happens unless we change the default value
* of Common.dbound (0.0). */
if (L->is_ll)
PyErr_Warn(PyExc_RuntimeWarning, "tiny diagonal "
"elements in L");
else
PyErr_Warn(PyExc_RuntimeWarning, "tiny diagonal "
"elements in D");
break;
default:
PyErr_Warn(PyExc_UserWarning, "");
}
if (L->minor<n) {
CHOL(free_factor)(&L, &Common);
PY_ERR(PyExc_ArithmeticError, "singular matrix");
}
if (!(Bc = create_matrix(B))) {
CHOL(free_factor)(&L, &Common);
return PyErr_NoMemory();
}
Xc = CHOL(spsolve)(0, L, Bc, &Common);
free_matrix(Bc);
CHOL(free_factor)(&L, &Common);
if (Common.status != CHOLMOD_OK){
CHOL(free_sparse)(&Xc, &Common);
if (Common.status == CHOLMOD_OUT_OF_MEMORY)
return PyErr_NoMemory();
else
PY_ERR(PyExc_ValueError, "solve step failed");
}
if (!(X = SpMatrix_New(Xc->nrow, Xc->ncol,
((int_t*)Xc->p)[Xc->ncol], SP_ID(A)))) {
CHOL(free_sparse)(&Xc, &Common);
return PyErr_NoMemory();
}
memcpy(SP_COL(X), (int_t *) Xc->p, (Xc->ncol+1)*sizeof(int_t));
memcpy(SP_ROW(X), (int_t *) Xc->i,
((int_t *) Xc->p)[Xc->ncol]*sizeof(int_t));
memcpy(SP_VAL(X), (double *) Xc->x,
((int_t *) Xc->p)[Xc->ncol]*E_SIZE[SP_ID(X)]);
CHOL(free_sparse)(&Xc, &Common);
return (PyObject *) X;
}
static PyObject* spsolve(PyObject *self, PyObject *args,
PyObject *kwrds)
{
spmatrix *B, *X=NULL;
cholmod_sparse *Bc=NULL, *Xc=NULL;
PyObject *F;
cholmod_factor *L;
int n, sys=0;
#if PY_MAJOR_VERSION >= 3
const char *descr;
#else
char *descr;
#endif
char *kwlist[] = {"F", "B", "sys", NULL};
int sysvalues[] = {CHOLMOD_A, CHOLMOD_LDLt, CHOLMOD_LD,
CHOLMOD_DLt, CHOLMOD_L, CHOLMOD_Lt, CHOLMOD_D, CHOLMOD_P,
CHOLMOD_Pt };
if (!set_options()) return NULL;
if (!PyArg_ParseTupleAndKeywords(args, kwrds, "OO|i", kwlist, &F,
&B, &sys)) return NULL;
#if PY_MAJOR_VERSION >= 3
if (!PyCapsule_CheckExact(F) || !(descr = PyCapsule_GetName(F)))
err_CO("F");
if (strncmp(descr, "CHOLMOD FACTOR", 14))
PY_ERR_TYPE("F is not a CHOLMOD factor");
L = (cholmod_factor *) PyCapsule_GetPointer(F, descr);
#else
if (!PyCObject_Check(F)) err_CO("F");
descr = PyCObject_GetDesc(F);
if (!descr || strncmp(descr, "CHOLMOD FACTOR", 14))
PY_ERR_TYPE("F is not a CHOLMOD factor");
L = (cholmod_factor *) PyCObject_AsVoidPtr(F);
#endif
if (L->xtype == CHOLMOD_PATTERN)
PY_ERR(PyExc_ValueError, "called with symbolic factor");
n = L->n;
if (L->minor<n) PY_ERR(PyExc_ArithmeticError, "singular matrix");
if (sys < 0 || sys > 8)
PY_ERR(PyExc_ValueError, "invalid value for sys");
if (!SpMatrix_Check(B) ||
(SP_ID(B) == DOUBLE && L->xtype == CHOLMOD_COMPLEX) ||
(SP_ID(B) == COMPLEX && L->xtype == CHOLMOD_REAL))
PY_ERR_TYPE("B must a sparse matrix of the same "
"numerical type as F");
if (SP_NROWS(B) != n)
PY_ERR(PyExc_ValueError, "incompatible dimensions for B");
if (!(Bc = create_matrix(B))) return PyErr_NoMemory();
Xc = CHOL(spsolve)(sysvalues[sys], L, Bc, &Common);
free_matrix(Bc);
if (Common.status == CHOLMOD_OUT_OF_MEMORY) return PyErr_NoMemory();
if (Common.status != CHOLMOD_OK)
PY_ERR(PyExc_ValueError, "solve step failed");
if (!(X = SpMatrix_New(Xc->nrow, Xc->ncol,
((int_t*)Xc->p)[Xc->ncol], (L->xtype == CHOLMOD_REAL ? DOUBLE :
COMPLEX)))) {
CHOL(free_sparse)(&Xc, &Common);
return PyErr_NoMemory();
}
memcpy(SP_COL(X), Xc->p, (Xc->ncol+1)*sizeof(int_t));
memcpy(SP_ROW(X), Xc->i, ((int_t *)Xc->p)[Xc->ncol]*sizeof(int_t));
memcpy(SP_VAL(X), Xc->x,
((int_t *) Xc->p)[Xc->ncol]*E_SIZE[SP_ID(X)]);
CHOL(free_sparse)(&Xc, &Common);
return (PyObject *) X;
}
static PyObject* linsolve(PyObject *self, PyObject *args,
PyObject *kwrds)
{
spmatrix *A;
matrix *B;
#if PY_MAJOR_VERSION >= 3
int trans_ = 'N';
#endif
char trans='N';
double info[UMFPACK_INFO];
int oB=0, n, nrhs=-1, ldB=0, k;
void *symbolic, *numeric, *x;
char *kwlist[] = {"A", "B", "trans", "nrhs", "ldB", "offsetB",
NULL};
#if PY_MAJOR_VERSION >= 3
if (!PyArg_ParseTupleAndKeywords(args, kwrds, "OO|Ciii", kwlist,
&A, &B, &trans_, &nrhs, &ldB, &oB)) return NULL;
trans = (char) trans_;
#else
if (!PyArg_ParseTupleAndKeywords(args, kwrds, "OO|ciii", kwlist,
&A, &B, &trans, &nrhs, &ldB, &oB)) return NULL;
#endif
if (!SpMatrix_Check(A) || SP_NROWS(A) != SP_NCOLS(A))
PY_ERR_TYPE("A must be a square sparse matrix");
n = SP_NROWS(A);
if (!Matrix_Check(B) || MAT_ID(B) != SP_ID(A))
PY_ERR_TYPE("B must a dense matrix of the same numeric type "
"as A");
if (nrhs < 0) nrhs = B->ncols;
if (n == 0 || nrhs == 0) return Py_BuildValue("i", 0);
if (ldB == 0) ldB = MAX(1,B->nrows);
if (ldB < MAX(1,n)) err_ld("ldB");
if (oB < 0) err_nn_int("offsetB");
if (oB + (nrhs-1)*ldB + n > MAT_LGT(B)) err_buf_len("B");
if (trans != 'N' && trans != 'T' && trans != 'C')
err_char("trans", "'N', 'T', 'C'");
if (SP_ID(A) == DOUBLE)
UMFD(symbolic)(n, n, SP_COL(A), SP_ROW(A), SP_VAL(A), &symbolic,
NULL, info);
else
UMFZ(symbolic)(n, n, SP_COL(A), SP_ROW(A), SP_VAL(A), NULL,
&symbolic, NULL, info);
if (info[UMFPACK_STATUS] != UMFPACK_OK){
if (SP_ID(A) == DOUBLE)
UMFD(free_symbolic)(&symbolic);
else
UMFZ(free_symbolic)(&symbolic);
if (info[UMFPACK_STATUS] == UMFPACK_ERROR_out_of_memory)
return PyErr_NoMemory();
else {
snprintf(umfpack_error,20,"%s %i","UMFPACK ERROR",
(int) info[UMFPACK_STATUS]);
PyErr_SetString(PyExc_ValueError, umfpack_error);
return NULL;
}
}
if (SP_ID(A) == DOUBLE) {
UMFD(numeric)(SP_COL(A), SP_ROW(A), SP_VAL(A), symbolic,
&numeric, NULL, info);
UMFD(free_symbolic)(&symbolic);
} else {
UMFZ(numeric)(SP_COL(A), SP_ROW(A), SP_VAL(A), NULL, symbolic,
&numeric, NULL, info);
UMFZ(free_symbolic)(&symbolic);
}
if (info[UMFPACK_STATUS] != UMFPACK_OK){
if (SP_ID(A) == DOUBLE)
UMFD(free_numeric)(&numeric);
else
UMFZ(free_numeric)(&numeric);
if (info[UMFPACK_STATUS] == UMFPACK_ERROR_out_of_memory)
return PyErr_NoMemory();
else {
if (info[UMFPACK_STATUS] == UMFPACK_WARNING_singular_matrix)
PyErr_SetString(PyExc_ArithmeticError, "singular "
"matrix");
else {
snprintf(umfpack_error,20,"%s %i","UMFPACK ERROR",
(int) info[UMFPACK_STATUS]);
PyErr_SetString(PyExc_ValueError, umfpack_error);
}
return NULL;
}
}
if (!(x = malloc(n*E_SIZE[SP_ID(A)]))) {
if (SP_ID(A) == DOUBLE)
UMFD(free_numeric)(&numeric);
else
UMFZ(free_numeric)(&numeric);
return PyErr_NoMemory();
}
for (k=0; k<nrhs; k++){
if (SP_ID(A) == DOUBLE)
UMFD(solve)(trans == 'N' ? UMFPACK_A: UMFPACK_Aat,
SP_COL(A), SP_ROW(A), SP_VAL(A), x,
MAT_BUFD(B) + k*ldB + oB, numeric, NULL, info);
else
UMFZ(solve)(trans == 'N' ? UMFPACK_A: trans == 'C' ?
UMFPACK_At : UMFPACK_Aat, SP_COL(A), SP_ROW(A),
SP_VAL(A), NULL, x, NULL,
(double *)(MAT_BUFZ(B) + k*ldB + oB), NULL, numeric,
NULL, info);
if (info[UMFPACK_STATUS] == UMFPACK_OK)
memcpy(B->buffer + (k*ldB + oB)*E_SIZE[SP_ID(A)], x,
n*E_SIZE[SP_ID(A)]);
else
break;
}
free(x);
if (SP_ID(A) == DOUBLE)
UMFD(free_numeric)(&numeric);
else
UMFZ(free_numeric)(&numeric);
if (info[UMFPACK_STATUS] != UMFPACK_OK){
if (info[UMFPACK_STATUS] == UMFPACK_ERROR_out_of_memory)
return PyErr_NoMemory();
else {
if (info[UMFPACK_STATUS] == UMFPACK_WARNING_singular_matrix)
PyErr_SetString(PyExc_ArithmeticError, "singular "
"matrix");
else {
snprintf(umfpack_error,20,"%s %i","UMFPACK ERROR",
(int) info[UMFPACK_STATUS]);
PyErr_SetString(PyExc_ValueError, umfpack_error);
}
return NULL;
}
}
return Py_BuildValue("");
}
