关联用户自定义预条件回调函数,如不使用自定义的预条件,该接口可以不使用。
C Interface:
int KmlIssCgSetUserPreconditionerSI(KmlSolverTask **handle, void *ustruct, int (*fptr)(void *ustruct, float *x));
int KmlIssCgSetUserPreconditionerDI(KmlSolverTask **handle, void *ustruct, int (*fptr)(void *ustruct, double *x));
参数名 |
类型 |
描述 |
输入/输出 |
---|---|---|---|
handle |
KmlSolverTask ** |
求解器句柄,传入之前步骤的变量。 |
输入/输出 |
ustruct |
void * |
指向用户数据的指针,每次调用预条件子时作为第一个参数传递。 |
输入 |
(*fptr) |
int |
用户自定义的函数指针,指向回调函数,该回调函数进行预条件子与其第二参数间的计算。 |
输入 |
返回值 |
类型 |
描述 |
---|---|---|
NO_ERROR |
int |
正常执行。 |
KML_SOLVER_INVALID_ARGUMENT |
int |
无效的参数。 |
#include "kml_iss.h"
class pc { int n; double *b; }; int mut(void *unstruct,double *x) { /* Compute the product of the precondition matrix and the vector x, the precondition matrix is a diagonal matrix */ int n; pc *a; a = (pc *)unstruct; n = a->n; for(int i=0; i<n; i++){ x[i] = x[i] * a->b[i]; } return 0; } KmlSolverTask *handle int n = 8; double a[17] = { 1.0,1.0,2.0,9.0,2.0,1.0,-3.0,3.0,2.0,9.0,-5.0,6.0,1.0,4.0,1.0,7.0,2.0 }; int ja[17] = { 0,3,4,1,2,3,5,2,7,3,6,4,5,5,7,6,7 }; int ia[9] = {0, 3, 7, 9, 11, 13, 15, 16, 17}; int ierr; pc *unstruct = new unstr; unstruct->b = new double [n]; for(int i=0; i<n; i++){ unstruct->b[i] = 2; } ierr = KmlIssCgInitDI(&handle, n, a, ja, ia); ierr = KmlIssCgSetUserPreconditionerDI(&handle, ustruct, mut);