1e884886fSBarry Smith 2c6db04a5SJed Brown #include <private/matimpl.h> 3c6db04a5SJed Brown #include <../src/mat/impls/mffd/mffdimpl.h> /*I "petscmat.h" I*/ 4e884886fSBarry Smith 5b022a5c1SBarry Smith PetscFList MatMFFDList = 0; 6ace3abfcSBarry Smith PetscBool MatMFFDRegisterAllCalled = PETSC_FALSE; 7e884886fSBarry Smith 87087cfbeSBarry Smith PetscClassId MATMFFD_CLASSID; 9166c7f25SBarry Smith PetscLogEvent MATMFFD_Mult; 10e884886fSBarry Smith 11ace3abfcSBarry Smith static PetscBool MatMFFDPackageInitialized = PETSC_FALSE; 12b022a5c1SBarry Smith #undef __FUNCT__ 13b022a5c1SBarry Smith #define __FUNCT__ "MatMFFDFinalizePackage" 14b022a5c1SBarry Smith /*@C 152390153bSJed Brown MatMFFDFinalizePackage - This function destroys everything in the MatMFFD package. It is 16b022a5c1SBarry Smith called from PetscFinalize(). 17b022a5c1SBarry Smith 18b022a5c1SBarry Smith Level: developer 19b022a5c1SBarry Smith 202390153bSJed Brown .keywords: Petsc, destroy, package 21b022a5c1SBarry Smith .seealso: PetscFinalize() 22b022a5c1SBarry Smith @*/ 237087cfbeSBarry Smith PetscErrorCode MatMFFDFinalizePackage(void) 24b022a5c1SBarry Smith { 25b022a5c1SBarry Smith PetscFunctionBegin; 26b022a5c1SBarry Smith MatMFFDPackageInitialized = PETSC_FALSE; 27b022a5c1SBarry Smith MatMFFDRegisterAllCalled = PETSC_FALSE; 28b022a5c1SBarry Smith MatMFFDList = PETSC_NULL; 29b022a5c1SBarry Smith PetscFunctionReturn(0); 30b022a5c1SBarry Smith } 31b022a5c1SBarry Smith 32e884886fSBarry Smith #undef __FUNCT__ 333ec795f1SBarry Smith #define __FUNCT__ "MatMFFDInitializePackage" 343ec795f1SBarry Smith /*@C 353ec795f1SBarry Smith MatMFFDInitializePackage - This function initializes everything in the MatMFFD package. It is called 363ec795f1SBarry Smith from PetscDLLibraryRegister() when using dynamic libraries, and on the first call to MatCreate_MFFD() 373ec795f1SBarry Smith when using static libraries. 383ec795f1SBarry Smith 393ec795f1SBarry Smith Input Parameter: 403ec795f1SBarry Smith . path - The dynamic library path, or PETSC_NULL 413ec795f1SBarry Smith 423ec795f1SBarry Smith Level: developer 433ec795f1SBarry Smith 443ec795f1SBarry Smith .keywords: Vec, initialize, package 453ec795f1SBarry Smith .seealso: PetscInitialize() 463ec795f1SBarry Smith @*/ 477087cfbeSBarry Smith PetscErrorCode MatMFFDInitializePackage(const char path[]) 483ec795f1SBarry Smith { 493ec795f1SBarry Smith char logList[256]; 503ec795f1SBarry Smith char *className; 51ace3abfcSBarry Smith PetscBool opt; 523ec795f1SBarry Smith PetscErrorCode ierr; 533ec795f1SBarry Smith 543ec795f1SBarry Smith PetscFunctionBegin; 55b022a5c1SBarry Smith if (MatMFFDPackageInitialized) PetscFunctionReturn(0); 56b022a5c1SBarry Smith MatMFFDPackageInitialized = PETSC_TRUE; 573ec795f1SBarry Smith /* Register Classes */ 580700a824SBarry Smith ierr = PetscClassIdRegister("MatMFFD",&MATMFFD_CLASSID);CHKERRQ(ierr); 593ec795f1SBarry Smith /* Register Constructors */ 603ec795f1SBarry Smith ierr = MatMFFDRegisterAll(path);CHKERRQ(ierr); 613ec795f1SBarry Smith /* Register Events */ 620700a824SBarry Smith ierr = PetscLogEventRegister("MatMult MF", MATMFFD_CLASSID,&MATMFFD_Mult);CHKERRQ(ierr); 633ec795f1SBarry Smith 643ec795f1SBarry Smith /* Process info exclusions */ 653ec795f1SBarry Smith ierr = PetscOptionsGetString(PETSC_NULL, "-info_exclude", logList, 256, &opt);CHKERRQ(ierr); 663ec795f1SBarry Smith if (opt) { 673ec795f1SBarry Smith ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr); 683ec795f1SBarry Smith if (className) { 690700a824SBarry Smith ierr = PetscInfoDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr); 703ec795f1SBarry Smith } 713ec795f1SBarry Smith } 723ec795f1SBarry Smith /* Process summary exclusions */ 733ec795f1SBarry Smith ierr = PetscOptionsGetString(PETSC_NULL, "-log_summary_exclude", logList, 256, &opt);CHKERRQ(ierr); 743ec795f1SBarry Smith if (opt) { 753ec795f1SBarry Smith ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr); 763ec795f1SBarry Smith if (className) { 770700a824SBarry Smith ierr = PetscLogEventDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr); 783ec795f1SBarry Smith } 793ec795f1SBarry Smith } 80b022a5c1SBarry Smith ierr = PetscRegisterFinalize(MatMFFDFinalizePackage);CHKERRQ(ierr); 813ec795f1SBarry Smith PetscFunctionReturn(0); 823ec795f1SBarry Smith } 833ec795f1SBarry Smith 843ec795f1SBarry Smith #undef __FUNCT__ 85e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetType" 86e884886fSBarry Smith /*@C 87e884886fSBarry Smith MatMFFDSetType - Sets the method that is used to compute the 88e884886fSBarry Smith differencing parameter for finite differene matrix-free formulations. 89e884886fSBarry Smith 90e884886fSBarry Smith Input Parameters: 91e884886fSBarry Smith + mat - the "matrix-free" matrix created via MatCreateSNESMF(), or MatCreateMFFD() 92e884886fSBarry Smith or MatSetType(mat,MATMFFD); 93e884886fSBarry Smith - ftype - the type requested, either MATMFFD_WP or MATMFFD_DS 94e884886fSBarry Smith 95e884886fSBarry Smith Level: advanced 96e884886fSBarry Smith 97e884886fSBarry Smith Notes: 98e884886fSBarry Smith For example, such routines can compute h for use in 99e884886fSBarry Smith Jacobian-vector products of the form 100e884886fSBarry Smith 101e884886fSBarry Smith F(x+ha) - F(x) 102e884886fSBarry Smith F'(u)a ~= ---------------- 103e884886fSBarry Smith h 104e884886fSBarry Smith 105d2d6cebeSBarry Smith .seealso: MatCreateSNESMF(), MatMFFDRegisterDynamic(), MatMFFDSetFunction() 106e884886fSBarry Smith @*/ 1077087cfbeSBarry Smith PetscErrorCode MatMFFDSetType(Mat mat,const MatMFFDType ftype) 108e884886fSBarry Smith { 109e884886fSBarry Smith PetscErrorCode ierr,(*r)(MatMFFD); 110e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 111ace3abfcSBarry Smith PetscBool match; 112e884886fSBarry Smith 113e884886fSBarry Smith PetscFunctionBegin; 1140700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 115e884886fSBarry Smith PetscValidCharPointer(ftype,2); 116e884886fSBarry Smith 1179a13eae7SJed Brown ierr = PetscTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr); 1189a13eae7SJed Brown if (!match) PetscFunctionReturn(0); 1199a13eae7SJed Brown 120e884886fSBarry Smith /* already set, so just return */ 121e884886fSBarry Smith ierr = PetscTypeCompare((PetscObject)ctx,ftype,&match);CHKERRQ(ierr); 122e884886fSBarry Smith if (match) PetscFunctionReturn(0); 123e884886fSBarry Smith 124e884886fSBarry Smith /* destroy the old one if it exists */ 125e884886fSBarry Smith if (ctx->ops->destroy) { 126e884886fSBarry Smith ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr); 127e884886fSBarry Smith } 128e884886fSBarry Smith 1294b91b6eaSBarry Smith ierr = PetscFListFind(MatMFFDList,((PetscObject)ctx)->comm,ftype,PETSC_TRUE,(void (**)(void)) &r);CHKERRQ(ierr); 130e32f2f54SBarry Smith if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown MatMFFD type %s given",ftype); 131e884886fSBarry Smith ierr = (*r)(ctx);CHKERRQ(ierr); 132e884886fSBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)ctx,ftype);CHKERRQ(ierr); 133e884886fSBarry Smith PetscFunctionReturn(0); 134e884886fSBarry Smith } 135e884886fSBarry Smith 136e884886fSBarry Smith typedef PetscErrorCode (*FCN1)(void*,Vec); /* force argument to next function to not be extern C*/ 137e884886fSBarry Smith EXTERN_C_BEGIN 138e884886fSBarry Smith #undef __FUNCT__ 139c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase_MFFD" 1407087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunctioniBase_MFFD(Mat mat,FCN1 func) 141e884886fSBarry Smith { 142e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 143e884886fSBarry Smith 144e884886fSBarry Smith PetscFunctionBegin; 145e884886fSBarry Smith ctx->funcisetbase = func; 146e884886fSBarry Smith PetscFunctionReturn(0); 147e884886fSBarry Smith } 148e884886fSBarry Smith EXTERN_C_END 149e884886fSBarry Smith 150e884886fSBarry Smith typedef PetscErrorCode (*FCN2)(void*,PetscInt,Vec,PetscScalar*); /* force argument to next function to not be extern C*/ 151e884886fSBarry Smith EXTERN_C_BEGIN 152e884886fSBarry Smith #undef __FUNCT__ 153c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni_MFFD" 1547087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunctioni_MFFD(Mat mat,FCN2 funci) 155e884886fSBarry Smith { 156e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 157e884886fSBarry Smith 158e884886fSBarry Smith PetscFunctionBegin; 159e884886fSBarry Smith ctx->funci = funci; 160e884886fSBarry Smith PetscFunctionReturn(0); 161e884886fSBarry Smith } 162e884886fSBarry Smith EXTERN_C_END 163e884886fSBarry Smith 164bc0f08ceSBarry Smith EXTERN_C_BEGIN 165bc0f08ceSBarry Smith #undef __FUNCT__ 166bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory_MFFD" 167bc0f08ceSBarry Smith PetscErrorCode MatMFFDResetHHistory_MFFD(Mat J) 168bc0f08ceSBarry Smith { 169bc0f08ceSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 170bc0f08ceSBarry Smith 171bc0f08ceSBarry Smith PetscFunctionBegin; 172bc0f08ceSBarry Smith ctx->ncurrenth = 0; 173bc0f08ceSBarry Smith PetscFunctionReturn(0); 174bc0f08ceSBarry Smith } 175bc0f08ceSBarry Smith EXTERN_C_END 176e884886fSBarry Smith 177e884886fSBarry Smith #undef __FUNCT__ 178e884886fSBarry Smith #define __FUNCT__ "MatMFFDRegister" 1797087cfbeSBarry Smith PetscErrorCode MatMFFDRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(MatMFFD)) 180e884886fSBarry Smith { 181e884886fSBarry Smith PetscErrorCode ierr; 182e884886fSBarry Smith char fullname[PETSC_MAX_PATH_LEN]; 183e884886fSBarry Smith 184e884886fSBarry Smith PetscFunctionBegin; 185e884886fSBarry Smith ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr); 186b022a5c1SBarry Smith ierr = PetscFListAdd(&MatMFFDList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr); 187e884886fSBarry Smith PetscFunctionReturn(0); 188e884886fSBarry Smith } 189e884886fSBarry Smith 190e884886fSBarry Smith 191e884886fSBarry Smith #undef __FUNCT__ 192e884886fSBarry Smith #define __FUNCT__ "MatMFFDRegisterDestroy" 193e884886fSBarry Smith /*@C 194e884886fSBarry Smith MatMFFDRegisterDestroy - Frees the list of MatMFFD methods that were 195e884886fSBarry Smith registered by MatMFFDRegisterDynamic). 196e884886fSBarry Smith 197e884886fSBarry Smith Not Collective 198e884886fSBarry Smith 199e884886fSBarry Smith Level: developer 200e884886fSBarry Smith 201e884886fSBarry Smith .keywords: MatMFFD, register, destroy 202e884886fSBarry Smith 203e884886fSBarry Smith .seealso: MatMFFDRegisterDynamic), MatMFFDRegisterAll() 204e884886fSBarry Smith @*/ 2057087cfbeSBarry Smith PetscErrorCode MatMFFDRegisterDestroy(void) 206e884886fSBarry Smith { 207e884886fSBarry Smith PetscErrorCode ierr; 208e884886fSBarry Smith 209e884886fSBarry Smith PetscFunctionBegin; 210b022a5c1SBarry Smith ierr = PetscFListDestroy(&MatMFFDList);CHKERRQ(ierr); 211e884886fSBarry Smith MatMFFDRegisterAllCalled = PETSC_FALSE; 212e884886fSBarry Smith PetscFunctionReturn(0); 213e884886fSBarry Smith } 214e884886fSBarry Smith 215bc0f08ceSBarry Smith EXTERN_C_BEGIN 216bc0f08ceSBarry Smith #undef __FUNCT__ 217bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace_MFFD" 218bc0f08ceSBarry Smith PetscErrorCode MatMFFDAddNullSpace_MFFD(Mat J,MatNullSpace nullsp) 219bc0f08ceSBarry Smith { 220bc0f08ceSBarry Smith PetscErrorCode ierr; 221bc0f08ceSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 222bc0f08ceSBarry Smith 223bc0f08ceSBarry Smith PetscFunctionBegin; 224bc0f08ceSBarry Smith ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr); 225bc0f08ceSBarry Smith if (ctx->sp) { ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr); } 226bc0f08ceSBarry Smith ctx->sp = nullsp; 227bc0f08ceSBarry Smith PetscFunctionReturn(0); 228bc0f08ceSBarry Smith } 229bc0f08ceSBarry Smith EXTERN_C_END 230bc0f08ceSBarry Smith 231e884886fSBarry Smith /* ----------------------------------------------------------------------------------------*/ 232e884886fSBarry Smith #undef __FUNCT__ 233e884886fSBarry Smith #define __FUNCT__ "MatDestroy_MFFD" 234e884886fSBarry Smith PetscErrorCode MatDestroy_MFFD(Mat mat) 235e884886fSBarry Smith { 236e884886fSBarry Smith PetscErrorCode ierr; 237e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 238e884886fSBarry Smith 239e884886fSBarry Smith PetscFunctionBegin; 2406bf464f9SBarry Smith ierr = VecDestroy(&ctx->w);CHKERRQ(ierr); 2416bf464f9SBarry Smith ierr = VecDestroy(&ctx->drscale);CHKERRQ(ierr); 2426bf464f9SBarry Smith ierr = VecDestroy(&ctx->dlscale);CHKERRQ(ierr); 2436bf464f9SBarry Smith ierr = VecDestroy(&ctx->dshift);CHKERRQ(ierr); 244cfe22f5eSBarry Smith if (ctx->current_f_allocated) { 2456bf464f9SBarry Smith ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr); 246cfe22f5eSBarry Smith } 247e884886fSBarry Smith if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);} 2486bf464f9SBarry Smith ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr); 2496bf464f9SBarry Smith ierr = PetscHeaderDestroy(&ctx);CHKERRQ(ierr); 250bf0cc555SLisandro Dalcin mat->data = 0; 251e884886fSBarry Smith 252e884886fSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetBase_C","",PETSC_NULL);CHKERRQ(ierr); 253e884886fSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioniBase_C","",PETSC_NULL);CHKERRQ(ierr); 254e884886fSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioni_C","",PETSC_NULL);CHKERRQ(ierr); 255bc0f08ceSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunction_C","",PETSC_NULL);CHKERRQ(ierr); 256bc0f08ceSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctionError_C","",PETSC_NULL);CHKERRQ(ierr); 257e884886fSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetCheckh_C","",PETSC_NULL);CHKERRQ(ierr); 258bc0f08ceSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetPeriod_C","",PETSC_NULL);CHKERRQ(ierr); 259bc0f08ceSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDResetHHistory_C","",PETSC_NULL);CHKERRQ(ierr); 260bc0f08ceSBarry Smith ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDAddNullSpace_C","",PETSC_NULL);CHKERRQ(ierr); 261e884886fSBarry Smith 262e884886fSBarry Smith PetscFunctionReturn(0); 263e884886fSBarry Smith } 264e884886fSBarry Smith 265e884886fSBarry Smith #undef __FUNCT__ 266e884886fSBarry Smith #define __FUNCT__ "MatView_MFFD" 267e884886fSBarry Smith /* 268e884886fSBarry Smith MatMFFDView_MFFD - Views matrix-free parameters. 269e884886fSBarry Smith 270e884886fSBarry Smith */ 271e884886fSBarry Smith PetscErrorCode MatView_MFFD(Mat J,PetscViewer viewer) 272e884886fSBarry Smith { 273e884886fSBarry Smith PetscErrorCode ierr; 274e884886fSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 275ace3abfcSBarry Smith PetscBool iascii; 276e884886fSBarry Smith 277e884886fSBarry Smith PetscFunctionBegin; 2782692d6eeSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 279e884886fSBarry Smith if (iascii) { 280e884886fSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," matrix-free approximation:\n");CHKERRQ(ierr); 281e884886fSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," err=%G (relative error in function evaluation)\n",ctx->error_rel);CHKERRQ(ierr); 2827adad957SLisandro Dalcin if (!((PetscObject)ctx)->type_name) { 283e884886fSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," The compute h routine has not yet been set\n");CHKERRQ(ierr); 284e884886fSBarry Smith } else { 2857adad957SLisandro Dalcin ierr = PetscViewerASCIIPrintf(viewer," Using %s compute h routine\n",((PetscObject)ctx)->type_name);CHKERRQ(ierr); 286e884886fSBarry Smith } 287e884886fSBarry Smith if (ctx->ops->view) { 288e884886fSBarry Smith ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr); 289e884886fSBarry Smith } 290*a8248277SBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for matrix-free matrix",((PetscObject)viewer)->type_name); 291e884886fSBarry Smith PetscFunctionReturn(0); 292e884886fSBarry Smith } 293e884886fSBarry Smith 294e884886fSBarry Smith #undef __FUNCT__ 295e884886fSBarry Smith #define __FUNCT__ "MatAssemblyEnd_MFFD" 296e884886fSBarry Smith /* 297e884886fSBarry Smith MatAssemblyEnd_MFFD - Resets the ctx->ncurrenth to zero. This 298e884886fSBarry Smith allows the user to indicate the beginning of a new linear solve by calling 299e884886fSBarry Smith MatAssemblyXXX() on the matrix free matrix. This then allows the 3001d0fab5eSBarry Smith MatCreateMFFD_WP() to properly compute ||U|| only the first time 301e884886fSBarry Smith in the linear solver rather than every time. 302e884886fSBarry Smith */ 303e884886fSBarry Smith PetscErrorCode MatAssemblyEnd_MFFD(Mat J,MatAssemblyType mt) 304e884886fSBarry Smith { 305e884886fSBarry Smith PetscErrorCode ierr; 306e884886fSBarry Smith MatMFFD j = (MatMFFD)J->data; 307e884886fSBarry Smith 308e884886fSBarry Smith PetscFunctionBegin; 309e884886fSBarry Smith ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr); 310e884886fSBarry Smith j->vshift = 0.0; 311e884886fSBarry Smith j->vscale = 1.0; 312e884886fSBarry Smith PetscFunctionReturn(0); 313e884886fSBarry Smith } 314e884886fSBarry Smith 315e884886fSBarry Smith #undef __FUNCT__ 316e884886fSBarry Smith #define __FUNCT__ "MatMult_MFFD" 317e884886fSBarry Smith /* 318e884886fSBarry Smith MatMult_MFFD - Default matrix-free form for Jacobian-vector product, y = F'(u)*a: 319e884886fSBarry Smith 320e884886fSBarry Smith y ~= (F(u + ha) - F(u))/h, 321e884886fSBarry Smith where F = nonlinear function, as set by SNESSetFunction() 322e884886fSBarry Smith u = current iterate 323e884886fSBarry Smith h = difference interval 324e884886fSBarry Smith */ 325e884886fSBarry Smith PetscErrorCode MatMult_MFFD(Mat mat,Vec a,Vec y) 326e884886fSBarry Smith { 327e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 328e884886fSBarry Smith PetscScalar h; 329e884886fSBarry Smith Vec w,U,F; 330e884886fSBarry Smith PetscErrorCode ierr; 331ace3abfcSBarry Smith PetscBool zeroa; 332e884886fSBarry Smith 333e884886fSBarry Smith PetscFunctionBegin; 33484da1bf7SJed Brown if (!ctx->current_u) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONGSTATE,"MatMFFDSetBase() has not been called, this is often caused by forgetting to call \n\t\tMatAssemblyBegin/End on the first Mat in the SNES compute function"); 335e884886fSBarry Smith /* We log matrix-free matrix-vector products separately, so that we can 336e884886fSBarry Smith separate the performance monitoring from the cases that use conventional 337e884886fSBarry Smith storage. We may eventually modify event logging to associate events 338e884886fSBarry Smith with particular objects, hence alleviating the more general problem. */ 339e884886fSBarry Smith ierr = PetscLogEventBegin(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr); 340e884886fSBarry Smith 341e884886fSBarry Smith w = ctx->w; 342e884886fSBarry Smith U = ctx->current_u; 3433ec795f1SBarry Smith F = ctx->current_f; 344e884886fSBarry Smith /* 345e884886fSBarry Smith Compute differencing parameter 346e884886fSBarry Smith */ 347e884886fSBarry Smith if (!ctx->ops->compute) { 348e884886fSBarry Smith ierr = MatMFFDSetType(mat,MATMFFD_WP);CHKERRQ(ierr); 3499c6ac3b3SBarry Smith ierr = MatSetFromOptions(mat);CHKERRQ(ierr); 350e884886fSBarry Smith } 351e884886fSBarry Smith ierr = (*ctx->ops->compute)(ctx,U,a,&h,&zeroa);CHKERRQ(ierr); 352e884886fSBarry Smith if (zeroa) { 353e884886fSBarry Smith ierr = VecSet(y,0.0);CHKERRQ(ierr); 354e884886fSBarry Smith PetscFunctionReturn(0); 355e884886fSBarry Smith } 356e884886fSBarry Smith 357e32f2f54SBarry Smith if (PetscIsInfOrNanScalar(h)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Computed Nan differencing parameter h"); 358e884886fSBarry Smith if (ctx->checkh) { 359e884886fSBarry Smith ierr = (*ctx->checkh)(ctx->checkhctx,U,a,&h);CHKERRQ(ierr); 360e884886fSBarry Smith } 361e884886fSBarry Smith 362e884886fSBarry Smith /* keep a record of the current differencing parameter h */ 363e884886fSBarry Smith ctx->currenth = h; 364e884886fSBarry Smith #if defined(PETSC_USE_COMPLEX) 365e884886fSBarry Smith ierr = PetscInfo2(mat,"Current differencing parameter: %G + %G i\n",PetscRealPart(h),PetscImaginaryPart(h));CHKERRQ(ierr); 366e884886fSBarry Smith #else 367e884886fSBarry Smith ierr = PetscInfo1(mat,"Current differencing parameter: %15.12e\n",h);CHKERRQ(ierr); 368e884886fSBarry Smith #endif 369e884886fSBarry Smith if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) { 370e884886fSBarry Smith ctx->historyh[ctx->ncurrenth] = h; 371e884886fSBarry Smith } 372e884886fSBarry Smith ctx->ncurrenth++; 373e884886fSBarry Smith 374e884886fSBarry Smith /* w = u + ha */ 375c3bb7e23SBarry Smith if (ctx->drscale) { 376c3bb7e23SBarry Smith ierr = VecPointwiseMult(ctx->drscale,a,U);CHKERRQ(ierr); 377c3bb7e23SBarry Smith ierr = VecAYPX(U,h,w);CHKERRQ(ierr); 378c3bb7e23SBarry Smith } else { 379e884886fSBarry Smith ierr = VecWAXPY(w,h,a,U);CHKERRQ(ierr); 380c3bb7e23SBarry Smith } 381e884886fSBarry Smith 382bcddec3dSBarry Smith /* compute func(U) as base for differencing; only needed first time in and not when provided by user */ 383bcddec3dSBarry Smith if (ctx->ncurrenth == 1 && ctx->current_f_allocated) { 384e884886fSBarry Smith ierr = (*ctx->func)(ctx->funcctx,U,F);CHKERRQ(ierr); 38552121784SBarry Smith } 386e884886fSBarry Smith ierr = (*ctx->func)(ctx->funcctx,w,y);CHKERRQ(ierr); 387e884886fSBarry Smith 388e884886fSBarry Smith ierr = VecAXPY(y,-1.0,F);CHKERRQ(ierr); 389e884886fSBarry Smith ierr = VecScale(y,1.0/h);CHKERRQ(ierr); 390e884886fSBarry Smith 391c35ec66cSMatthew G Knepley if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) { 392e884886fSBarry Smith ierr = VecAXPBY(y,ctx->vshift,ctx->vscale,a);CHKERRQ(ierr); 393c3bb7e23SBarry Smith } 394c3bb7e23SBarry Smith if (ctx->dlscale) { 395c3bb7e23SBarry Smith ierr = VecPointwiseMult(y,ctx->dlscale,y);CHKERRQ(ierr); 396c3bb7e23SBarry Smith } 397c3bb7e23SBarry Smith if (ctx->dshift) { 398c3bb7e23SBarry Smith ierr = VecPointwiseMult(ctx->dshift,a,U);CHKERRQ(ierr); 399c3bb7e23SBarry Smith ierr = VecAXPY(y,1.0,U);CHKERRQ(ierr); 400c3bb7e23SBarry Smith } 401e884886fSBarry Smith 402e884886fSBarry Smith if (ctx->sp) {ierr = MatNullSpaceRemove(ctx->sp,y,PETSC_NULL);CHKERRQ(ierr);} 403e884886fSBarry Smith 404e884886fSBarry Smith ierr = PetscLogEventEnd(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr); 405e884886fSBarry Smith PetscFunctionReturn(0); 406e884886fSBarry Smith } 407e884886fSBarry Smith 408e884886fSBarry Smith #undef __FUNCT__ 409e884886fSBarry Smith #define __FUNCT__ "MatGetDiagonal_MFFD" 410e884886fSBarry Smith /* 411e884886fSBarry Smith MatGetDiagonal_MFFD - Gets the diagonal for a matrix free matrix 412e884886fSBarry Smith 413e884886fSBarry Smith y ~= (F(u + ha) - F(u))/h, 414e884886fSBarry Smith where F = nonlinear function, as set by SNESSetFunction() 415e884886fSBarry Smith u = current iterate 416e884886fSBarry Smith h = difference interval 417e884886fSBarry Smith */ 418e884886fSBarry Smith PetscErrorCode MatGetDiagonal_MFFD(Mat mat,Vec a) 419e884886fSBarry Smith { 420e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 421e884886fSBarry Smith PetscScalar h,*aa,*ww,v; 422e884886fSBarry Smith PetscReal epsilon = PETSC_SQRT_MACHINE_EPSILON,umin = 100.0*PETSC_SQRT_MACHINE_EPSILON; 423e884886fSBarry Smith Vec w,U; 424e884886fSBarry Smith PetscErrorCode ierr; 425e884886fSBarry Smith PetscInt i,rstart,rend; 426e884886fSBarry Smith 427e884886fSBarry Smith PetscFunctionBegin; 428e7e72b3dSBarry Smith if (!ctx->funci) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Requires calling MatMFFDSetFunctioni() first"); 429e884886fSBarry Smith 430e884886fSBarry Smith w = ctx->w; 431e884886fSBarry Smith U = ctx->current_u; 432e884886fSBarry Smith ierr = (*ctx->func)(ctx->funcctx,U,a);CHKERRQ(ierr); 433e884886fSBarry Smith ierr = (*ctx->funcisetbase)(ctx->funcctx,U);CHKERRQ(ierr); 434e884886fSBarry Smith ierr = VecCopy(U,w);CHKERRQ(ierr); 435e884886fSBarry Smith 436e884886fSBarry Smith ierr = VecGetOwnershipRange(a,&rstart,&rend);CHKERRQ(ierr); 437e884886fSBarry Smith ierr = VecGetArray(a,&aa);CHKERRQ(ierr); 438e884886fSBarry Smith for (i=rstart; i<rend; i++) { 439e884886fSBarry Smith ierr = VecGetArray(w,&ww);CHKERRQ(ierr); 440e884886fSBarry Smith h = ww[i-rstart]; 441e884886fSBarry Smith if (h == 0.0) h = 1.0; 442e884886fSBarry Smith #if !defined(PETSC_USE_COMPLEX) 443e884886fSBarry Smith if (h < umin && h >= 0.0) h = umin; 444e884886fSBarry Smith else if (h < 0.0 && h > -umin) h = -umin; 445e884886fSBarry Smith #else 446e884886fSBarry Smith if (PetscAbsScalar(h) < umin && PetscRealPart(h) >= 0.0) h = umin; 447e884886fSBarry Smith else if (PetscRealPart(h) < 0.0 && PetscAbsScalar(h) < umin) h = -umin; 448e884886fSBarry Smith #endif 449e884886fSBarry Smith h *= epsilon; 450e884886fSBarry Smith 451e884886fSBarry Smith ww[i-rstart] += h; 452e884886fSBarry Smith ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr); 453e884886fSBarry Smith ierr = (*ctx->funci)(ctx->funcctx,i,w,&v);CHKERRQ(ierr); 454e884886fSBarry Smith aa[i-rstart] = (v - aa[i-rstart])/h; 455e884886fSBarry Smith 456e884886fSBarry Smith /* possibly shift and scale result */ 457c35ec66cSMatthew G Knepley if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) { 458e884886fSBarry Smith aa[i - rstart] = ctx->vshift + ctx->vscale*aa[i-rstart]; 459c3bb7e23SBarry Smith } 460e884886fSBarry Smith 461e884886fSBarry Smith ierr = VecGetArray(w,&ww);CHKERRQ(ierr); 462e884886fSBarry Smith ww[i-rstart] -= h; 463e884886fSBarry Smith ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr); 464e884886fSBarry Smith } 465e884886fSBarry Smith ierr = VecRestoreArray(a,&aa);CHKERRQ(ierr); 466e884886fSBarry Smith PetscFunctionReturn(0); 467e884886fSBarry Smith } 468e884886fSBarry Smith 469e884886fSBarry Smith #undef __FUNCT__ 470c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalScale_MFFD" 471c3bb7e23SBarry Smith PetscErrorCode MatDiagonalScale_MFFD(Mat mat,Vec ll,Vec rr) 472c3bb7e23SBarry Smith { 473c3bb7e23SBarry Smith MatMFFD aij = (MatMFFD)mat->data; 474c3bb7e23SBarry Smith PetscErrorCode ierr; 475c3bb7e23SBarry Smith 476c3bb7e23SBarry Smith PetscFunctionBegin; 477c3bb7e23SBarry Smith if (ll && !aij->dlscale) { 478c3bb7e23SBarry Smith ierr = VecDuplicate(ll,&aij->dlscale);CHKERRQ(ierr); 479c3bb7e23SBarry Smith } 480c3bb7e23SBarry Smith if (rr && !aij->drscale) { 481c3bb7e23SBarry Smith ierr = VecDuplicate(rr,&aij->drscale);CHKERRQ(ierr); 482c3bb7e23SBarry Smith } 483c3bb7e23SBarry Smith if (ll) { 484c3bb7e23SBarry Smith ierr = VecCopy(ll,aij->dlscale);CHKERRQ(ierr); 485c3bb7e23SBarry Smith } 486c3bb7e23SBarry Smith if (rr) { 487c3bb7e23SBarry Smith ierr = VecCopy(rr,aij->drscale);CHKERRQ(ierr); 488c3bb7e23SBarry Smith } 489c3bb7e23SBarry Smith PetscFunctionReturn(0); 490c3bb7e23SBarry Smith } 491c3bb7e23SBarry Smith 492c3bb7e23SBarry Smith #undef __FUNCT__ 493c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalSet_MFFD" 494c3bb7e23SBarry Smith PetscErrorCode MatDiagonalSet_MFFD(Mat mat,Vec ll,InsertMode mode) 495c3bb7e23SBarry Smith { 496c3bb7e23SBarry Smith MatMFFD aij = (MatMFFD)mat->data; 497c3bb7e23SBarry Smith PetscErrorCode ierr; 498c3bb7e23SBarry Smith 499c3bb7e23SBarry Smith PetscFunctionBegin; 500c3bb7e23SBarry Smith if (mode == INSERT_VALUES) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_SUP,"No diagonal set with INSERT_VALUES"); 501c3bb7e23SBarry Smith if (!aij->dshift) { 502c3bb7e23SBarry Smith ierr = VecDuplicate(ll,&aij->dshift);CHKERRQ(ierr); 503c3bb7e23SBarry Smith } 504c3bb7e23SBarry Smith ierr = VecAXPY(aij->dshift,1.0,ll);CHKERRQ(ierr); 505c3bb7e23SBarry Smith PetscFunctionReturn(0); 506c3bb7e23SBarry Smith } 507c3bb7e23SBarry Smith 508c3bb7e23SBarry Smith #undef __FUNCT__ 509e884886fSBarry Smith #define __FUNCT__ "MatShift_MFFD" 510e884886fSBarry Smith PetscErrorCode MatShift_MFFD(Mat Y,PetscScalar a) 511e884886fSBarry Smith { 512e884886fSBarry Smith MatMFFD shell = (MatMFFD)Y->data; 513e884886fSBarry Smith PetscFunctionBegin; 514e884886fSBarry Smith shell->vshift += a; 515e884886fSBarry Smith PetscFunctionReturn(0); 516e884886fSBarry Smith } 517e884886fSBarry Smith 518e884886fSBarry Smith #undef __FUNCT__ 519e884886fSBarry Smith #define __FUNCT__ "MatScale_MFFD" 520e884886fSBarry Smith PetscErrorCode MatScale_MFFD(Mat Y,PetscScalar a) 521e884886fSBarry Smith { 522e884886fSBarry Smith MatMFFD shell = (MatMFFD)Y->data; 523e884886fSBarry Smith PetscFunctionBegin; 524e884886fSBarry Smith shell->vscale *= a; 525e884886fSBarry Smith PetscFunctionReturn(0); 526e884886fSBarry Smith } 527e884886fSBarry Smith 528e884886fSBarry Smith EXTERN_C_BEGIN 529e884886fSBarry Smith #undef __FUNCT__ 530c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetBase_MFFD" 5317087cfbeSBarry Smith PetscErrorCode MatMFFDSetBase_MFFD(Mat J,Vec U,Vec F) 532e884886fSBarry Smith { 533e884886fSBarry Smith PetscErrorCode ierr; 534e884886fSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 535e884886fSBarry Smith 536e884886fSBarry Smith PetscFunctionBegin; 537e884886fSBarry Smith ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr); 538e884886fSBarry Smith ctx->current_u = U; 53952121784SBarry Smith if (F) { 5406bf464f9SBarry Smith if (ctx->current_f_allocated) {ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);} 5413ec795f1SBarry Smith ctx->current_f = F; 542cfe22f5eSBarry Smith ctx->current_f_allocated = PETSC_FALSE; 543cfe22f5eSBarry Smith } else if (!ctx->current_f_allocated) { 54452121784SBarry Smith ierr = VecDuplicate(ctx->current_u, &ctx->current_f);CHKERRQ(ierr); 545cfe22f5eSBarry Smith ctx->current_f_allocated = PETSC_TRUE; 54652121784SBarry Smith } 547e884886fSBarry Smith if (!ctx->w) { 548e884886fSBarry Smith ierr = VecDuplicate(ctx->current_u, &ctx->w);CHKERRQ(ierr); 549e884886fSBarry Smith } 550e884886fSBarry Smith J->assembled = PETSC_TRUE; 551e884886fSBarry Smith PetscFunctionReturn(0); 552e884886fSBarry Smith } 553e884886fSBarry Smith EXTERN_C_END 554e884886fSBarry Smith typedef PetscErrorCode (*FCN3)(void*,Vec,Vec,PetscScalar*); /* force argument to next function to not be extern C*/ 555e884886fSBarry Smith EXTERN_C_BEGIN 556e884886fSBarry Smith #undef __FUNCT__ 557c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh_MFFD" 5587087cfbeSBarry Smith PetscErrorCode MatMFFDSetCheckh_MFFD(Mat J,FCN3 fun,void*ectx) 559e884886fSBarry Smith { 560e884886fSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 561e884886fSBarry Smith 562e884886fSBarry Smith PetscFunctionBegin; 563e884886fSBarry Smith ctx->checkh = fun; 564e884886fSBarry Smith ctx->checkhctx = ectx; 565e884886fSBarry Smith PetscFunctionReturn(0); 566e884886fSBarry Smith } 567e884886fSBarry Smith EXTERN_C_END 568e884886fSBarry Smith 569e884886fSBarry Smith #undef __FUNCT__ 5706aa9148fSLisandro Dalcin #define __FUNCT__ "MatMFFDSetOptionsPrefix" 5716aa9148fSLisandro Dalcin /*@C 5726aa9148fSLisandro Dalcin MatMFFDSetOptionsPrefix - Sets the prefix used for searching for all 5736aa9148fSLisandro Dalcin MatMFFD options in the database. 5746aa9148fSLisandro Dalcin 5756aa9148fSLisandro Dalcin Collective on Mat 5766aa9148fSLisandro Dalcin 5776aa9148fSLisandro Dalcin Input Parameter: 5786aa9148fSLisandro Dalcin + A - the Mat context 5796aa9148fSLisandro Dalcin - prefix - the prefix to prepend to all option names 5806aa9148fSLisandro Dalcin 5816aa9148fSLisandro Dalcin Notes: 5826aa9148fSLisandro Dalcin A hyphen (-) must NOT be given at the beginning of the prefix name. 5836aa9148fSLisandro Dalcin The first character of all runtime options is AUTOMATICALLY the hyphen. 5846aa9148fSLisandro Dalcin 5856aa9148fSLisandro Dalcin Level: advanced 5866aa9148fSLisandro Dalcin 5876aa9148fSLisandro Dalcin .keywords: SNES, matrix-free, parameters 5886aa9148fSLisandro Dalcin 5899c6ac3b3SBarry Smith .seealso: MatSetFromOptions(), MatCreateSNESMF() 5906aa9148fSLisandro Dalcin @*/ 5917087cfbeSBarry Smith PetscErrorCode MatMFFDSetOptionsPrefix(Mat mat,const char prefix[]) 5926aa9148fSLisandro Dalcin 5936aa9148fSLisandro Dalcin { 5946aa9148fSLisandro Dalcin MatMFFD mfctx = mat ? (MatMFFD)mat->data : PETSC_NULL; 5956aa9148fSLisandro Dalcin PetscErrorCode ierr; 5966aa9148fSLisandro Dalcin PetscFunctionBegin; 5970700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 5980700a824SBarry Smith PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1); 5996aa9148fSLisandro Dalcin ierr = PetscObjectSetOptionsPrefix((PetscObject)mfctx,prefix);CHKERRQ(ierr); 6006aa9148fSLisandro Dalcin PetscFunctionReturn(0); 6016aa9148fSLisandro Dalcin } 6026aa9148fSLisandro Dalcin 6036aa9148fSLisandro Dalcin #undef __FUNCT__ 6049c6ac3b3SBarry Smith #define __FUNCT__ "MatSetFromOptions_MFFD" 6059c6ac3b3SBarry Smith PetscErrorCode MatSetFromOptions_MFFD(Mat mat) 606e884886fSBarry Smith { 60771f08665SBarry Smith MatMFFD mfctx = (MatMFFD)mat->data; 608e884886fSBarry Smith PetscErrorCode ierr; 609ace3abfcSBarry Smith PetscBool flg; 610e884886fSBarry Smith char ftype[256]; 611e884886fSBarry Smith 612e884886fSBarry Smith PetscFunctionBegin; 6130700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 6140700a824SBarry Smith PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1); 6157adad957SLisandro Dalcin ierr = PetscOptionsBegin(((PetscObject)mfctx)->comm,((PetscObject)mfctx)->prefix,"Set matrix free computation parameters","MatMFFD");CHKERRQ(ierr); 616b022a5c1SBarry Smith ierr = PetscOptionsList("-mat_mffd_type","Matrix free type","MatMFFDSetType",MatMFFDList,((PetscObject)mfctx)->type_name,ftype,256,&flg);CHKERRQ(ierr); 617e884886fSBarry Smith if (flg) { 618e884886fSBarry Smith ierr = MatMFFDSetType(mat,ftype);CHKERRQ(ierr); 619e884886fSBarry Smith } 620e884886fSBarry Smith 621e884886fSBarry Smith ierr = PetscOptionsReal("-mat_mffd_err","set sqrt relative error in function","MatMFFDSetFunctionError",mfctx->error_rel,&mfctx->error_rel,0);CHKERRQ(ierr); 622e884886fSBarry Smith ierr = PetscOptionsInt("-mat_mffd_period","how often h is recomputed","MatMFFDSetPeriod",mfctx->recomputeperiod,&mfctx->recomputeperiod,0);CHKERRQ(ierr); 623e884886fSBarry Smith 62490d69ab7SBarry Smith flg = PETSC_FALSE; 625acfcf0e5SJed Brown ierr = PetscOptionsBool("-mat_mffd_check_positivity","Insure that U + h*a is nonnegative","MatMFFDSetCheckh",flg,&flg,PETSC_NULL);CHKERRQ(ierr); 626e884886fSBarry Smith if (flg) { 627e884886fSBarry Smith ierr = MatMFFDSetCheckh(mat,MatMFFDCheckPositivity,0);CHKERRQ(ierr); 628e884886fSBarry Smith } 629e884886fSBarry Smith if (mfctx->ops->setfromoptions) { 630e884886fSBarry Smith ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr); 631e884886fSBarry Smith } 632e884886fSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 633e884886fSBarry Smith PetscFunctionReturn(0); 634e884886fSBarry Smith } 635e884886fSBarry Smith 636bc0f08ceSBarry Smith EXTERN_C_BEGIN 637bc0f08ceSBarry Smith #undef __FUNCT__ 638bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod_MFFD" 639bc0f08ceSBarry Smith PetscErrorCode MatMFFDSetPeriod_MFFD(Mat mat,PetscInt period) 640bc0f08ceSBarry Smith { 641bc0f08ceSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 642bc0f08ceSBarry Smith 643bc0f08ceSBarry Smith PetscFunctionBegin; 644bc0f08ceSBarry Smith PetscValidLogicalCollectiveInt(mat,period,2); 645bc0f08ceSBarry Smith ctx->recomputeperiod = period; 646bc0f08ceSBarry Smith PetscFunctionReturn(0); 647bc0f08ceSBarry Smith } 648bc0f08ceSBarry Smith EXTERN_C_END 649bc0f08ceSBarry Smith 650bc0f08ceSBarry Smith EXTERN_C_BEGIN 651bc0f08ceSBarry Smith #undef __FUNCT__ 652bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunction_MFFD" 653bc0f08ceSBarry Smith PetscErrorCode MatMFFDSetFunction_MFFD(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx) 654bc0f08ceSBarry Smith { 655bc0f08ceSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 656bc0f08ceSBarry Smith 657bc0f08ceSBarry Smith PetscFunctionBegin; 658bc0f08ceSBarry Smith ctx->func = func; 659bc0f08ceSBarry Smith ctx->funcctx = funcctx; 660bc0f08ceSBarry Smith PetscFunctionReturn(0); 661bc0f08ceSBarry Smith } 662bc0f08ceSBarry Smith EXTERN_C_END 663bc0f08ceSBarry Smith 664bc0f08ceSBarry Smith EXTERN_C_BEGIN 665bc0f08ceSBarry Smith #undef __FUNCT__ 666bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError_MFFD" 667bc0f08ceSBarry Smith PetscErrorCode MatMFFDSetFunctionError_MFFD(Mat mat,PetscReal error) 668bc0f08ceSBarry Smith { 669bc0f08ceSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 670bc0f08ceSBarry Smith 671bc0f08ceSBarry Smith PetscFunctionBegin; 672bc0f08ceSBarry Smith PetscValidLogicalCollectiveReal(mat,error,2); 673bc0f08ceSBarry Smith if (error != PETSC_DEFAULT) ctx->error_rel = error; 674bc0f08ceSBarry Smith PetscFunctionReturn(0); 675bc0f08ceSBarry Smith } 676bc0f08ceSBarry Smith EXTERN_C_END 677bc0f08ceSBarry Smith 678e884886fSBarry Smith /*MC 679e884886fSBarry Smith MATMFFD - MATMFFD = "mffd" - A matrix free matrix type. 680e884886fSBarry Smith 681e884886fSBarry Smith Level: advanced 682e884886fSBarry Smith 683d2d6cebeSBarry Smith .seealso: MatCreateMFFD(), MatCreateSNESMF(), MatMFFDSetFunction() 684e884886fSBarry Smith M*/ 685e884886fSBarry Smith EXTERN_C_BEGIN 686e884886fSBarry Smith #undef __FUNCT__ 687e884886fSBarry Smith #define __FUNCT__ "MatCreate_MFFD" 6887087cfbeSBarry Smith PetscErrorCode MatCreate_MFFD(Mat A) 689e884886fSBarry Smith { 690e884886fSBarry Smith MatMFFD mfctx; 691e884886fSBarry Smith PetscErrorCode ierr; 692e884886fSBarry Smith 693e884886fSBarry Smith PetscFunctionBegin; 6943ec795f1SBarry Smith #ifndef PETSC_USE_DYNAMIC_LIBRARIES 6953ec795f1SBarry Smith ierr = MatMFFDInitializePackage(PETSC_NULL);CHKERRQ(ierr); 6963ec795f1SBarry Smith #endif 697e884886fSBarry Smith 6980700a824SBarry Smith ierr = PetscHeaderCreate(mfctx,_p_MatMFFD,struct _MFOps,MATMFFD_CLASSID,0,"MatMFFD",((PetscObject)A)->comm,MatDestroy_MFFD,MatView_MFFD);CHKERRQ(ierr); 699e884886fSBarry Smith mfctx->sp = 0; 700e884886fSBarry Smith mfctx->error_rel = PETSC_SQRT_MACHINE_EPSILON; 701e884886fSBarry Smith mfctx->recomputeperiod = 1; 702e884886fSBarry Smith mfctx->count = 0; 703e884886fSBarry Smith mfctx->currenth = 0.0; 704e884886fSBarry Smith mfctx->historyh = PETSC_NULL; 705e884886fSBarry Smith mfctx->ncurrenth = 0; 706e884886fSBarry Smith mfctx->maxcurrenth = 0; 7077adad957SLisandro Dalcin ((PetscObject)mfctx)->type_name = 0; 708e884886fSBarry Smith 709e884886fSBarry Smith mfctx->vshift = 0.0; 710e884886fSBarry Smith mfctx->vscale = 1.0; 711e884886fSBarry Smith 712e884886fSBarry Smith /* 713e884886fSBarry Smith Create the empty data structure to contain compute-h routines. 714e884886fSBarry Smith These will be filled in below from the command line options or 715e884886fSBarry Smith a later call with MatMFFDSetType() or if that is not called 716e884886fSBarry Smith then it will default in the first use of MatMult_MFFD() 717e884886fSBarry Smith */ 718e884886fSBarry Smith mfctx->ops->compute = 0; 719e884886fSBarry Smith mfctx->ops->destroy = 0; 720e884886fSBarry Smith mfctx->ops->view = 0; 721e884886fSBarry Smith mfctx->ops->setfromoptions = 0; 722e884886fSBarry Smith mfctx->hctx = 0; 723e884886fSBarry Smith 724e884886fSBarry Smith mfctx->func = 0; 725e884886fSBarry Smith mfctx->funcctx = 0; 726e884886fSBarry Smith mfctx->w = PETSC_NULL; 727e884886fSBarry Smith 728e884886fSBarry Smith A->data = mfctx; 729e884886fSBarry Smith 730e884886fSBarry Smith A->ops->mult = MatMult_MFFD; 731e884886fSBarry Smith A->ops->destroy = MatDestroy_MFFD; 732e884886fSBarry Smith A->ops->view = MatView_MFFD; 733e884886fSBarry Smith A->ops->assemblyend = MatAssemblyEnd_MFFD; 734e884886fSBarry Smith A->ops->getdiagonal = MatGetDiagonal_MFFD; 735e884886fSBarry Smith A->ops->scale = MatScale_MFFD; 736e884886fSBarry Smith A->ops->shift = MatShift_MFFD; 737c3bb7e23SBarry Smith A->ops->diagonalscale = MatDiagonalScale_MFFD; 738c3bb7e23SBarry Smith A->ops->diagonalset = MatDiagonalSet_MFFD; 7399c6ac3b3SBarry Smith A->ops->setfromoptions = MatSetFromOptions_MFFD; 740e884886fSBarry Smith A->assembled = PETSC_TRUE; 741e884886fSBarry Smith 74226283091SBarry Smith ierr = PetscLayoutSetBlockSize(A->rmap,1);CHKERRQ(ierr); 74326283091SBarry Smith ierr = PetscLayoutSetBlockSize(A->cmap,1);CHKERRQ(ierr); 74426283091SBarry Smith ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 74526283091SBarry Smith ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 746ee1cef2cSJed Brown 747c879e56bSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetBase_C","MatMFFDSetBase_MFFD",MatMFFDSetBase_MFFD);CHKERRQ(ierr); 748c879e56bSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioniBase_C","MatMFFDSetFunctioniBase_MFFD",MatMFFDSetFunctioniBase_MFFD);CHKERRQ(ierr); 749c879e56bSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioni_C","MatMFFDSetFunctioni_MFFD",MatMFFDSetFunctioni_MFFD);CHKERRQ(ierr); 750bc0f08ceSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunction_C","MatMFFDSetFunction_MFFD",MatMFFDSetFunction_MFFD);CHKERRQ(ierr); 751c879e56bSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetCheckh_C","MatMFFDSetCheckh_MFFD",MatMFFDSetCheckh_MFFD);CHKERRQ(ierr); 752bc0f08ceSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetPeriod_C","MatMFFDSetPeriod_MFFD",MatMFFDSetPeriod_MFFD);CHKERRQ(ierr); 753bc0f08ceSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctionError_C","MatMFFDSetFunctionError_MFFD",MatMFFDSetFunctionError_MFFD);CHKERRQ(ierr); 754bc0f08ceSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDResetHHistory_C","MatMFFDResetHHistory_MFFD",MatMFFDResetHHistory_MFFD);CHKERRQ(ierr); 755bc0f08ceSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDAddNullSpace_C","MatMFFDAddNullSpace_MFFD",MatMFFDAddNullSpace_MFFD);CHKERRQ(ierr); 756e884886fSBarry Smith mfctx->mat = A; 757e884886fSBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)A,MATMFFD);CHKERRQ(ierr); 758e884886fSBarry Smith PetscFunctionReturn(0); 759e884886fSBarry Smith } 760e884886fSBarry Smith EXTERN_C_END 761e884886fSBarry Smith 762e884886fSBarry Smith #undef __FUNCT__ 763e884886fSBarry Smith #define __FUNCT__ "MatCreateMFFD" 764e884886fSBarry Smith /*@ 765e884886fSBarry Smith MatCreateMFFD - Creates a matrix-free matrix. See also MatCreateSNESMF() 766e884886fSBarry Smith 767e884886fSBarry Smith Collective on Vec 768e884886fSBarry Smith 769e884886fSBarry Smith Input Parameters: 770fef1beadSBarry Smith + comm - MPI communicator 771fef1beadSBarry Smith . m - number of local rows (or PETSC_DECIDE to have calculated if M is given) 772fef1beadSBarry Smith This value should be the same as the local size used in creating the 773fef1beadSBarry Smith y vector for the matrix-vector product y = Ax. 774fef1beadSBarry Smith . n - This value should be the same as the local size used in creating the 775fef1beadSBarry Smith x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have 776fef1beadSBarry Smith calculated if N is given) For square matrices n is almost always m. 777fef1beadSBarry Smith . M - number of global rows (or PETSC_DETERMINE to have calculated if m is given) 778fef1beadSBarry Smith - N - number of global columns (or PETSC_DETERMINE to have calculated if n is given) 779fef1beadSBarry Smith 780e884886fSBarry Smith 781e884886fSBarry Smith Output Parameter: 782e884886fSBarry Smith . J - the matrix-free matrix 783e884886fSBarry Smith 7849c6ac3b3SBarry Smith Options Database Keys: call MatSetFromOptions() to trigger these 7859c6ac3b3SBarry Smith + -mat_mffd_type - wp or ds (see MATMFFD_WP or MATMFFD_DS) 7869c6ac3b3SBarry Smith - -mat_mffd_err - square root of estimated relative error in function evaluation 7879c6ac3b3SBarry Smith - -mat_mffd_period - how often h is recomputed, defaults to 1, everytime 7889c6ac3b3SBarry Smith 7899c6ac3b3SBarry Smith 790e884886fSBarry Smith Level: advanced 791e884886fSBarry Smith 792e884886fSBarry Smith Notes: 793e884886fSBarry Smith The matrix-free matrix context merely contains the function pointers 794e884886fSBarry Smith and work space for performing finite difference approximations of 795e884886fSBarry Smith Jacobian-vector products, F'(u)*a, 796e884886fSBarry Smith 797e884886fSBarry Smith The default code uses the following approach to compute h 798e884886fSBarry Smith 799e884886fSBarry Smith .vb 800e884886fSBarry Smith F'(u)*a = [F(u+h*a) - F(u)]/h where 801e884886fSBarry Smith h = error_rel*u'a/||a||^2 if |u'a| > umin*||a||_{1} 802e884886fSBarry Smith = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2 otherwise 803e884886fSBarry Smith where 804e884886fSBarry Smith error_rel = square root of relative error in function evaluation 805e884886fSBarry Smith umin = minimum iterate parameter 806e884886fSBarry Smith .ve 807e884886fSBarry Smith 808e884886fSBarry Smith The user can set the error_rel via MatMFFDSetFunctionError() and 8090598bfebSBarry Smith umin via MatMFFDDSSetUmin(); see the <A href="../../docs/manual.pdf#nameddest=ch_snes">SNES chapter of the users manual</A> for details. 810e884886fSBarry Smith 811e884886fSBarry Smith The user should call MatDestroy() when finished with the matrix-free 812e884886fSBarry Smith matrix context. 813e884886fSBarry Smith 814e884886fSBarry Smith Options Database Keys: 815e884886fSBarry Smith + -mat_mffd_err <error_rel> - Sets error_rel 816e884886fSBarry Smith . -mat_mffd_unim <umin> - Sets umin (for default PETSc routine that computes h only) 817e884886fSBarry Smith - -mat_mffd_check_positivity 818e884886fSBarry Smith 819e884886fSBarry Smith .keywords: default, matrix-free, create, matrix 820e884886fSBarry Smith 8211d0fab5eSBarry Smith .seealso: MatDestroy(), MatMFFDSetFunctionError(), MatMFFDDSSetUmin(), MatMFFDSetFunction() 822e884886fSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory(), MatCreateSNESMF(), 8231d0fab5eSBarry Smith MatMFFDGetH(), MatMFFDRegisterDynamic), MatMFFDComputeJacobian() 824e884886fSBarry Smith 825e884886fSBarry Smith @*/ 8267087cfbeSBarry Smith PetscErrorCode MatCreateMFFD(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,Mat *J) 827e884886fSBarry Smith { 828e884886fSBarry Smith PetscErrorCode ierr; 829e884886fSBarry Smith 830e884886fSBarry Smith PetscFunctionBegin; 831e884886fSBarry Smith ierr = MatCreate(comm,J);CHKERRQ(ierr); 832fef1beadSBarry Smith ierr = MatSetSizes(*J,m,n,M,N);CHKERRQ(ierr); 833e884886fSBarry Smith ierr = MatSetType(*J,MATMFFD);CHKERRQ(ierr); 834e884886fSBarry Smith PetscFunctionReturn(0); 835e884886fSBarry Smith } 836e884886fSBarry Smith 837e884886fSBarry Smith 838e884886fSBarry Smith #undef __FUNCT__ 839e884886fSBarry Smith #define __FUNCT__ "MatMFFDGetH" 840e884886fSBarry Smith /*@ 841e884886fSBarry Smith MatMFFDGetH - Gets the last value that was used as the differencing 842e884886fSBarry Smith parameter. 843e884886fSBarry Smith 844e884886fSBarry Smith Not Collective 845e884886fSBarry Smith 846e884886fSBarry Smith Input Parameters: 847e884886fSBarry Smith . mat - the matrix obtained with MatCreateSNESMF() 848e884886fSBarry Smith 849e884886fSBarry Smith Output Paramter: 850e884886fSBarry Smith . h - the differencing step size 851e884886fSBarry Smith 852e884886fSBarry Smith Level: advanced 853e884886fSBarry Smith 854e884886fSBarry Smith .keywords: SNES, matrix-free, parameters 855e884886fSBarry Smith 8561d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDSetHHistory(), MatCreateMFFD(), MATMFFD, MatMFFDResetHHistory() 857e884886fSBarry Smith @*/ 8587087cfbeSBarry Smith PetscErrorCode MatMFFDGetH(Mat mat,PetscScalar *h) 859e884886fSBarry Smith { 860e884886fSBarry Smith MatMFFD ctx = (MatMFFD)mat->data; 861bc0f08ceSBarry Smith PetscErrorCode ierr; 862bc0f08ceSBarry Smith PetscBool match; 863e884886fSBarry Smith 864e884886fSBarry Smith PetscFunctionBegin; 865bc0f08ceSBarry Smith ierr = PetscTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr); 866bc0f08ceSBarry Smith if (!match) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix"); 867bc0f08ceSBarry Smith 868e884886fSBarry Smith *h = ctx->currenth; 869e884886fSBarry Smith PetscFunctionReturn(0); 870e884886fSBarry Smith } 871e884886fSBarry Smith 872e884886fSBarry Smith #undef __FUNCT__ 873e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunction" 874e884886fSBarry Smith /*@C 875e884886fSBarry Smith MatMFFDSetFunction - Sets the function used in applying the matrix free. 876e884886fSBarry Smith 8773f9fe445SBarry Smith Logically Collective on Mat 878e884886fSBarry Smith 879e884886fSBarry Smith Input Parameters: 880e884886fSBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 881e884886fSBarry Smith . func - the function to use 882e884886fSBarry Smith - funcctx - optional function context passed to function 883e884886fSBarry Smith 884e884886fSBarry Smith Level: advanced 885e884886fSBarry Smith 886e884886fSBarry Smith Notes: 887e884886fSBarry Smith If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free 888e884886fSBarry Smith matrix inside your compute Jacobian routine 889e884886fSBarry Smith 8903ec795f1SBarry Smith If this is not set then it will use the function set with SNESSetFunction() if MatCreateSNESMF() was used. 891e884886fSBarry Smith 892e884886fSBarry Smith .keywords: SNES, matrix-free, function 893e884886fSBarry Smith 8941d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD, 8951d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction() 896e884886fSBarry Smith @*/ 8977087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunction(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx) 898e884886fSBarry Smith { 899bc0f08ceSBarry Smith PetscErrorCode ierr; 900e884886fSBarry Smith 901e884886fSBarry Smith PetscFunctionBegin; 902bc0f08ceSBarry Smith ierr = PetscTryMethod(mat,"MatMFFDSetFunction_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec),void*),(mat,func,funcctx));CHKERRQ(ierr); 903e884886fSBarry Smith PetscFunctionReturn(0); 904e884886fSBarry Smith } 905e884886fSBarry Smith 906e884886fSBarry Smith #undef __FUNCT__ 907e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni" 908e884886fSBarry Smith /*@C 909e884886fSBarry Smith MatMFFDSetFunctioni - Sets the function for a single component 910e884886fSBarry Smith 9113f9fe445SBarry Smith Logically Collective on Mat 912e884886fSBarry Smith 913e884886fSBarry Smith Input Parameters: 914e884886fSBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 915e884886fSBarry Smith - funci - the function to use 916e884886fSBarry Smith 917e884886fSBarry Smith Level: advanced 918e884886fSBarry Smith 919e884886fSBarry Smith Notes: 920e884886fSBarry Smith If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free 921e884886fSBarry Smith matrix inside your compute Jacobian routine 922e884886fSBarry Smith 923e884886fSBarry Smith 924e884886fSBarry Smith .keywords: SNES, matrix-free, function 925e884886fSBarry Smith 9261d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction() 9271d0fab5eSBarry Smith 928e884886fSBarry Smith @*/ 9297087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunctioni(Mat mat,PetscErrorCode (*funci)(void*,PetscInt,Vec,PetscScalar*)) 930e884886fSBarry Smith { 9314ac538c5SBarry Smith PetscErrorCode ierr; 932e884886fSBarry Smith 933e884886fSBarry Smith PetscFunctionBegin; 9340700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 9354ac538c5SBarry Smith ierr = PetscTryMethod(mat,"MatMFFDSetFunctioni_C",(Mat,PetscErrorCode (*)(void*,PetscInt,Vec,PetscScalar*)),(mat,funci));CHKERRQ(ierr); 936e884886fSBarry Smith PetscFunctionReturn(0); 937e884886fSBarry Smith } 938e884886fSBarry Smith 939e884886fSBarry Smith 940e884886fSBarry Smith #undef __FUNCT__ 941e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase" 942e884886fSBarry Smith /*@C 943e884886fSBarry Smith MatMFFDSetFunctioniBase - Sets the base vector for a single component function evaluation 944e884886fSBarry Smith 9453f9fe445SBarry Smith Logically Collective on Mat 946e884886fSBarry Smith 947e884886fSBarry Smith Input Parameters: 948e884886fSBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 949e884886fSBarry Smith - func - the function to use 950e884886fSBarry Smith 951e884886fSBarry Smith Level: advanced 952e884886fSBarry Smith 953e884886fSBarry Smith Notes: 954e884886fSBarry Smith If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free 955e884886fSBarry Smith matrix inside your compute Jacobian routine 956e884886fSBarry Smith 957e884886fSBarry Smith 958e884886fSBarry Smith .keywords: SNES, matrix-free, function 959e884886fSBarry Smith 960e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD 9611d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction() 962e884886fSBarry Smith @*/ 9637087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunctioniBase(Mat mat,PetscErrorCode (*func)(void*,Vec)) 964e884886fSBarry Smith { 9654ac538c5SBarry Smith PetscErrorCode ierr; 966e884886fSBarry Smith 967e884886fSBarry Smith PetscFunctionBegin; 9680700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 9694ac538c5SBarry Smith ierr = PetscTryMethod(mat,"MatMFFDSetFunctioniBase_C",(Mat,PetscErrorCode (*)(void*,Vec)),(mat,func));CHKERRQ(ierr); 970e884886fSBarry Smith PetscFunctionReturn(0); 971e884886fSBarry Smith } 972e884886fSBarry Smith 973e884886fSBarry Smith #undef __FUNCT__ 974e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod" 975e884886fSBarry Smith /*@ 976e884886fSBarry Smith MatMFFDSetPeriod - Sets how often h is recomputed, by default it is everytime 977e884886fSBarry Smith 9783f9fe445SBarry Smith Logically Collective on Mat 979e884886fSBarry Smith 980e884886fSBarry Smith Input Parameters: 981e884886fSBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 982e884886fSBarry Smith - period - 1 for everytime, 2 for every second etc 983e884886fSBarry Smith 984e884886fSBarry Smith Options Database Keys: 985e884886fSBarry Smith + -mat_mffd_period <period> 986e884886fSBarry Smith 987e884886fSBarry Smith Level: advanced 988e884886fSBarry Smith 989e884886fSBarry Smith 990e884886fSBarry Smith .keywords: SNES, matrix-free, parameters 991e884886fSBarry Smith 992e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), 9931d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory() 994e884886fSBarry Smith @*/ 9957087cfbeSBarry Smith PetscErrorCode MatMFFDSetPeriod(Mat mat,PetscInt period) 996e884886fSBarry Smith { 997bc0f08ceSBarry Smith PetscErrorCode ierr; 998e884886fSBarry Smith 999e884886fSBarry Smith PetscFunctionBegin; 1000bc0f08ceSBarry Smith ierr = PetscTryMethod(mat,"MatMFFDSetPeriod_C",(Mat,PetscInt),(mat,period));CHKERRQ(ierr); 1001e884886fSBarry Smith PetscFunctionReturn(0); 1002e884886fSBarry Smith } 1003e884886fSBarry Smith 1004e884886fSBarry Smith #undef __FUNCT__ 1005e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError" 1006e884886fSBarry Smith /*@ 1007e884886fSBarry Smith MatMFFDSetFunctionError - Sets the error_rel for the approximation of 1008e884886fSBarry Smith matrix-vector products using finite differences. 1009e884886fSBarry Smith 10103f9fe445SBarry Smith Logically Collective on Mat 1011e884886fSBarry Smith 1012e884886fSBarry Smith Input Parameters: 1013e884886fSBarry Smith + mat - the matrix free matrix created via MatCreateMFFD() or MatCreateSNESMF() 1014e884886fSBarry Smith - error_rel - relative error (should be set to the square root of 1015e884886fSBarry Smith the relative error in the function evaluations) 1016e884886fSBarry Smith 1017e884886fSBarry Smith Options Database Keys: 1018e884886fSBarry Smith + -mat_mffd_err <error_rel> - Sets error_rel 1019e884886fSBarry Smith 1020e884886fSBarry Smith Level: advanced 1021e884886fSBarry Smith 1022e884886fSBarry Smith Notes: 1023e884886fSBarry Smith The default matrix-free matrix-vector product routine computes 1024e884886fSBarry Smith .vb 1025e884886fSBarry Smith F'(u)*a = [F(u+h*a) - F(u)]/h where 1026e884886fSBarry Smith h = error_rel*u'a/||a||^2 if |u'a| > umin*||a||_{1} 1027e884886fSBarry Smith = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2 else 1028e884886fSBarry Smith .ve 1029e884886fSBarry Smith 1030e884886fSBarry Smith .keywords: SNES, matrix-free, parameters 1031e884886fSBarry Smith 1032e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD 10331d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory() 1034e884886fSBarry Smith @*/ 10357087cfbeSBarry Smith PetscErrorCode MatMFFDSetFunctionError(Mat mat,PetscReal error) 1036e884886fSBarry Smith { 1037bc0f08ceSBarry Smith PetscErrorCode ierr; 1038e884886fSBarry Smith 1039e884886fSBarry Smith PetscFunctionBegin; 1040bc0f08ceSBarry Smith ierr = PetscTryMethod(mat,"MatMFFDSetFunctionError_C",(Mat,PetscReal),(mat,error));CHKERRQ(ierr); 1041e884886fSBarry Smith PetscFunctionReturn(0); 1042e884886fSBarry Smith } 1043e884886fSBarry Smith 1044e884886fSBarry Smith #undef __FUNCT__ 1045e884886fSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace" 1046e884886fSBarry Smith /*@ 1047e884886fSBarry Smith MatMFFDAddNullSpace - Provides a null space that an operator is 1048e884886fSBarry Smith supposed to have. Since roundoff will create a small component in 1049e884886fSBarry Smith the null space, if you know the null space you may have it 1050e884886fSBarry Smith automatically removed. 1051e884886fSBarry Smith 10523f9fe445SBarry Smith Logically Collective on Mat 1053e884886fSBarry Smith 1054e884886fSBarry Smith Input Parameters: 1055e884886fSBarry Smith + J - the matrix-free matrix context 1056e884886fSBarry Smith - nullsp - object created with MatNullSpaceCreate() 1057e884886fSBarry Smith 1058e884886fSBarry Smith Level: advanced 1059e884886fSBarry Smith 1060e884886fSBarry Smith .keywords: SNES, matrix-free, null space 1061e884886fSBarry Smith 1062e884886fSBarry Smith .seealso: MatNullSpaceCreate(), MatMFFDGetH(), MatCreateSNESMF(), MatCreateMFFD(), MATMFFD 10631d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDResetHHistory() 1064e884886fSBarry Smith @*/ 10657087cfbeSBarry Smith PetscErrorCode MatMFFDAddNullSpace(Mat J,MatNullSpace nullsp) 1066e884886fSBarry Smith { 1067e884886fSBarry Smith PetscErrorCode ierr; 1068e884886fSBarry Smith 1069e884886fSBarry Smith PetscFunctionBegin; 1070bc0f08ceSBarry Smith ierr = PetscTryMethod(J,"MatMFFDAddNullSpace_C",(Mat,MatNullSpace),(J,nullsp));CHKERRQ(ierr); 1071e884886fSBarry Smith PetscFunctionReturn(0); 1072e884886fSBarry Smith } 1073e884886fSBarry Smith 1074e884886fSBarry Smith #undef __FUNCT__ 1075e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetHHistory" 1076e884886fSBarry Smith /*@ 1077e884886fSBarry Smith MatMFFDSetHHistory - Sets an array to collect a history of the 1078e884886fSBarry Smith differencing values (h) computed for the matrix-free product. 1079e884886fSBarry Smith 10803f9fe445SBarry Smith Logically Collective on Mat 1081e884886fSBarry Smith 1082e884886fSBarry Smith Input Parameters: 1083e884886fSBarry Smith + J - the matrix-free matrix context 1084e884886fSBarry Smith . histroy - space to hold the history 1085e884886fSBarry Smith - nhistory - number of entries in history, if more entries are generated than 1086e884886fSBarry Smith nhistory, then the later ones are discarded 1087e884886fSBarry Smith 1088e884886fSBarry Smith Level: advanced 1089e884886fSBarry Smith 1090e884886fSBarry Smith Notes: 1091e884886fSBarry Smith Use MatMFFDResetHHistory() to reset the history counter and collect 1092e884886fSBarry Smith a new batch of differencing parameters, h. 1093e884886fSBarry Smith 1094e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history 1095e884886fSBarry Smith 1096e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(), 10971d0fab5eSBarry Smith MatMFFDResetHHistory(), MatMFFDSetFunctionError() 1098e884886fSBarry Smith 1099e884886fSBarry Smith @*/ 11007087cfbeSBarry Smith PetscErrorCode MatMFFDSetHHistory(Mat J,PetscScalar history[],PetscInt nhistory) 1101e884886fSBarry Smith { 1102e884886fSBarry Smith MatMFFD ctx = (MatMFFD)J->data; 1103bc0f08ceSBarry Smith PetscErrorCode ierr; 1104bc0f08ceSBarry Smith PetscBool match; 1105e884886fSBarry Smith 1106e884886fSBarry Smith PetscFunctionBegin; 1107bc0f08ceSBarry Smith ierr = PetscTypeCompare((PetscObject)J,MATMFFD,&match);CHKERRQ(ierr); 1108bc0f08ceSBarry Smith if (!match) SETERRQ(((PetscObject)J)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix"); 1109e884886fSBarry Smith ctx->historyh = history; 1110e884886fSBarry Smith ctx->maxcurrenth = nhistory; 111175567043SBarry Smith ctx->currenth = 0.; 1112e884886fSBarry Smith PetscFunctionReturn(0); 1113e884886fSBarry Smith } 1114e884886fSBarry Smith 1115bc0f08ceSBarry Smith 1116e884886fSBarry Smith #undef __FUNCT__ 1117e884886fSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory" 1118e884886fSBarry Smith /*@ 1119e884886fSBarry Smith MatMFFDResetHHistory - Resets the counter to zero to begin 1120e884886fSBarry Smith collecting a new set of differencing histories. 1121e884886fSBarry Smith 11223f9fe445SBarry Smith Logically Collective on Mat 1123e884886fSBarry Smith 1124e884886fSBarry Smith Input Parameters: 1125e884886fSBarry Smith . J - the matrix-free matrix context 1126e884886fSBarry Smith 1127e884886fSBarry Smith Level: advanced 1128e884886fSBarry Smith 1129e884886fSBarry Smith Notes: 1130e884886fSBarry Smith Use MatMFFDSetHHistory() to create the original history counter. 1131e884886fSBarry Smith 1132e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history 1133e884886fSBarry Smith 1134e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(), 11351d0fab5eSBarry Smith MatMFFDSetHHistory(), MatMFFDSetFunctionError() 1136e884886fSBarry Smith 1137e884886fSBarry Smith @*/ 11387087cfbeSBarry Smith PetscErrorCode MatMFFDResetHHistory(Mat J) 1139e884886fSBarry Smith { 1140bc0f08ceSBarry Smith PetscErrorCode ierr; 1141e884886fSBarry Smith 1142e884886fSBarry Smith PetscFunctionBegin; 1143bc0f08ceSBarry Smith ierr = PetscTryMethod(J,"MatMFFDResetHHistory_C",(Mat),(J));CHKERRQ(ierr); 1144e884886fSBarry Smith PetscFunctionReturn(0); 1145e884886fSBarry Smith } 1146e884886fSBarry Smith 1147e884886fSBarry Smith 1148e884886fSBarry Smith #undef __FUNCT__ 1149e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetBase" 1150e884886fSBarry Smith /*@ 1151e884886fSBarry Smith MatMFFDSetBase - Sets the vector U at which matrix vector products of the 1152e884886fSBarry Smith Jacobian are computed 1153e884886fSBarry Smith 11543f9fe445SBarry Smith Logically Collective on Mat 1155e884886fSBarry Smith 1156e884886fSBarry Smith Input Parameters: 1157e884886fSBarry Smith + J - the MatMFFD matrix 11583ec795f1SBarry Smith . U - the vector 1159bcddec3dSBarry Smith - F - (optional) vector that contains F(u) if it has been already computed 1160e884886fSBarry Smith 1161e884886fSBarry Smith Notes: This is rarely used directly 1162e884886fSBarry Smith 11638af5ae88SBarry Smith If F is provided then it is not recomputed. Otherwise the function is evaluated at the base 11648af5ae88SBarry Smith point during the first MatMult() after each call to MatMFFDSetBase(). 1165dff2f722SBarry Smith 1166e884886fSBarry Smith Level: advanced 1167e884886fSBarry Smith 1168e884886fSBarry Smith @*/ 11697087cfbeSBarry Smith PetscErrorCode MatMFFDSetBase(Mat J,Vec U,Vec F) 1170e884886fSBarry Smith { 11714ac538c5SBarry Smith PetscErrorCode ierr; 1172e884886fSBarry Smith 1173e884886fSBarry Smith PetscFunctionBegin; 11740700a824SBarry Smith PetscValidHeaderSpecific(J,MAT_CLASSID,1); 11750700a824SBarry Smith PetscValidHeaderSpecific(U,VEC_CLASSID,2); 11760700a824SBarry Smith if (F) PetscValidHeaderSpecific(F,VEC_CLASSID,3); 11774ac538c5SBarry Smith ierr = PetscTryMethod(J,"MatMFFDSetBase_C",(Mat,Vec,Vec),(J,U,F));CHKERRQ(ierr); 1178e884886fSBarry Smith PetscFunctionReturn(0); 1179e884886fSBarry Smith } 1180e884886fSBarry Smith 1181e884886fSBarry Smith #undef __FUNCT__ 1182e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh" 1183e884886fSBarry Smith /*@C 1184e884886fSBarry Smith MatMFFDSetCheckh - Sets a function that checks the computed h and adjusts 1185e884886fSBarry Smith it to satisfy some criteria 1186e884886fSBarry Smith 11873f9fe445SBarry Smith Logically Collective on Mat 1188e884886fSBarry Smith 1189e884886fSBarry Smith Input Parameters: 1190e884886fSBarry Smith + J - the MatMFFD matrix 1191e884886fSBarry Smith . fun - the function that checks h 1192e884886fSBarry Smith - ctx - any context needed by the function 1193e884886fSBarry Smith 1194e884886fSBarry Smith Options Database Keys: 1195e884886fSBarry Smith . -mat_mffd_check_positivity 1196e884886fSBarry Smith 1197e884886fSBarry Smith Level: advanced 1198e884886fSBarry Smith 1199e884886fSBarry Smith Notes: For example, MatMFFDSetCheckPositivity() insures that all entries 1200e884886fSBarry Smith of U + h*a are non-negative 1201e884886fSBarry Smith 1202e884886fSBarry Smith .seealso: MatMFFDSetCheckPositivity() 1203e884886fSBarry Smith @*/ 12047087cfbeSBarry Smith PetscErrorCode MatMFFDSetCheckh(Mat J,PetscErrorCode (*fun)(void*,Vec,Vec,PetscScalar*),void* ctx) 1205e884886fSBarry Smith { 12064ac538c5SBarry Smith PetscErrorCode ierr; 1207e884886fSBarry Smith 1208e884886fSBarry Smith PetscFunctionBegin; 12090700a824SBarry Smith PetscValidHeaderSpecific(J,MAT_CLASSID,1); 12104ac538c5SBarry Smith ierr = PetscTryMethod(J,"MatMFFDSetCheckh_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec,PetscScalar*),void*),(J,fun,ctx));CHKERRQ(ierr); 1211e884886fSBarry Smith PetscFunctionReturn(0); 1212e884886fSBarry Smith } 1213e884886fSBarry Smith 1214e884886fSBarry Smith #undef __FUNCT__ 1215e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckPositivity" 1216e884886fSBarry Smith /*@ 1217e884886fSBarry Smith MatMFFDCheckPositivity - Checks that all entries in U + h*a are positive or 1218e884886fSBarry Smith zero, decreases h until this is satisfied. 1219e884886fSBarry Smith 12203f9fe445SBarry Smith Logically Collective on Vec 1221e884886fSBarry Smith 1222e884886fSBarry Smith Input Parameters: 1223e884886fSBarry Smith + U - base vector that is added to 1224e884886fSBarry Smith . a - vector that is added 1225e884886fSBarry Smith . h - scaling factor on a 1226e884886fSBarry Smith - dummy - context variable (unused) 1227e884886fSBarry Smith 1228e884886fSBarry Smith Options Database Keys: 1229e884886fSBarry Smith . -mat_mffd_check_positivity 1230e884886fSBarry Smith 1231e884886fSBarry Smith Level: advanced 1232e884886fSBarry Smith 1233e884886fSBarry Smith Notes: This is rarely used directly, rather it is passed as an argument to 1234e884886fSBarry Smith MatMFFDSetCheckh() 1235e884886fSBarry Smith 1236e884886fSBarry Smith .seealso: MatMFFDSetCheckh() 1237e884886fSBarry Smith @*/ 12387087cfbeSBarry Smith PetscErrorCode MatMFFDCheckPositivity(void* dummy,Vec U,Vec a,PetscScalar *h) 1239e884886fSBarry Smith { 1240e884886fSBarry Smith PetscReal val, minval; 1241e884886fSBarry Smith PetscScalar *u_vec, *a_vec; 1242e884886fSBarry Smith PetscErrorCode ierr; 1243e884886fSBarry Smith PetscInt i,n; 1244e884886fSBarry Smith MPI_Comm comm; 1245e884886fSBarry Smith 1246e884886fSBarry Smith PetscFunctionBegin; 1247e884886fSBarry Smith ierr = PetscObjectGetComm((PetscObject)U,&comm);CHKERRQ(ierr); 1248e884886fSBarry Smith ierr = VecGetArray(U,&u_vec);CHKERRQ(ierr); 1249e884886fSBarry Smith ierr = VecGetArray(a,&a_vec);CHKERRQ(ierr); 1250e884886fSBarry Smith ierr = VecGetLocalSize(U,&n);CHKERRQ(ierr); 1251e884886fSBarry Smith minval = PetscAbsScalar(*h*1.01); 1252e884886fSBarry Smith for(i=0;i<n;i++) { 1253e884886fSBarry Smith if (PetscRealPart(u_vec[i] + *h*a_vec[i]) <= 0.0) { 1254e884886fSBarry Smith val = PetscAbsScalar(u_vec[i]/a_vec[i]); 1255e884886fSBarry Smith if (val < minval) minval = val; 1256e884886fSBarry Smith } 1257e884886fSBarry Smith } 1258e884886fSBarry Smith ierr = VecRestoreArray(U,&u_vec);CHKERRQ(ierr); 1259e884886fSBarry Smith ierr = VecRestoreArray(a,&a_vec);CHKERRQ(ierr); 1260d9822059SBarry Smith ierr = MPI_Allreduce(&minval,&val,1,MPIU_REAL,MPIU_MIN,comm);CHKERRQ(ierr); 1261e884886fSBarry Smith if (val <= PetscAbsScalar(*h)) { 1262e884886fSBarry Smith ierr = PetscInfo2(U,"Scaling back h from %G to %G\n",PetscRealPart(*h),.99*val);CHKERRQ(ierr); 1263e884886fSBarry Smith if (PetscRealPart(*h) > 0.0) *h = 0.99*val; 1264e884886fSBarry Smith else *h = -0.99*val; 1265e884886fSBarry Smith } 1266e884886fSBarry Smith PetscFunctionReturn(0); 1267e884886fSBarry Smith } 1268e884886fSBarry Smith 1269e884886fSBarry Smith 1270e884886fSBarry Smith 1271e884886fSBarry Smith 1272e884886fSBarry Smith 1273e884886fSBarry Smith 1274e884886fSBarry Smith 1275