1 2 #include <petsc-private/petscimpl.h> /*I "petscsys.h" I*/ 3 #include <petscviewersaws.h> 4 #include <petscsys.h> 5 6 #undef __FUNCT__ 7 #define __FUNCT__ "PetscObjectSAWsTakeAccess" 8 /*@C 9 PetscObjectSAWsTakeAccess - Take access of the data fields that have been published to SAWs so they may be changed locally 10 11 Collective on PetscObject 12 13 Input Parameters: 14 . obj - the Petsc variable 15 Thus must be cast with a (PetscObject), for example, 16 PetscObjectSetName((PetscObject)mat,name); 17 18 Level: advanced 19 20 Concepts: publishing object 21 22 .seealso: PetscObjectSetName(), PetscObjectSAWsViewOff(), PetscObjectSAWsGrantAccess() 23 24 @*/ 25 PetscErrorCode PetscObjectSAWsTakeAccess(PetscObject obj) 26 { 27 if (obj->amsmem) { 28 /* cannot wrap with PetscPushStack() because that also deals with the locks */ 29 SAWs_Lock(); 30 } 31 return 0; 32 } 33 34 #undef __FUNCT__ 35 #define __FUNCT__ "PetscObjectSAWsGrantAccess" 36 /*@C 37 PetscObjectSAWsGrantAccess - Grants access of the data fields that have been published to SAWs to the memory snooper to change 38 39 Collective on PetscObject 40 41 Input Parameters: 42 . obj - the Petsc variable 43 Thus must be cast with a (PetscObject), for example, 44 PetscObjectSetName((PetscObject)mat,name); 45 46 Level: advanced 47 48 Concepts: publishing object 49 50 .seealso: PetscObjectSetName(), PetscObjectSAWsViewOff(), PetscObjectSAWsTakeAccess() 51 52 @*/ 53 PetscErrorCode PetscObjectSAWsGrantAccess(PetscObject obj) 54 { 55 if (obj->amsmem) { 56 /* cannot wrap with PetscPushStack() because that also deals with the locks */ 57 SAWs_Unlock(); 58 } 59 return 0; 60 } 61 62 #undef __FUNCT__ 63 #define __FUNCT__ "PetscObjectSAWsBlock" 64 /*@C 65 PetscObjectSAWsBlock - Blocks the object if PetscObjectSAWsSetBlock() has been called 66 67 Collective on PetscObject 68 69 Input Parameters: 70 . obj - the Petsc variable 71 Thus must be cast with a (PetscObject), for example, 72 PetscObjectSetName((PetscObject)mat,name); 73 74 75 Level: advanced 76 77 Concepts: publishing object 78 79 .seealso: PetscObjectSetName(), PetscObjectSAWsViewOff(), PetscObjectSAWsSetBlock() 80 81 @*/ 82 PetscErrorCode PetscObjectSAWsBlock(PetscObject obj) 83 { 84 PetscErrorCode ierr; 85 86 PetscFunctionBegin; 87 PetscValidHeader(obj,1); 88 89 if (!obj->amspublishblock || !obj->amsmem) PetscFunctionReturn(0); 90 ierr = PetscObjectSAWsTakeAccess(obj);CHKERRQ(ierr); 91 while (obj->amsblock) { 92 ierr = PetscObjectSAWsGrantAccess(obj);CHKERRQ(ierr); 93 ierr = PetscInfo(NULL,"Blocking on AMS\n"); 94 ierr = PetscSleep(2.0);CHKERRQ(ierr); 95 ierr = PetscObjectSAWsTakeAccess(obj);CHKERRQ(ierr); 96 } 97 obj->amsblock = PETSC_TRUE; 98 ierr = PetscObjectSAWsGrantAccess(obj);CHKERRQ(ierr); 99 ierr = PetscInfo(NULL,"Out of SAWs block\n"); 100 PetscFunctionReturn(0); 101 } 102 103 #undef __FUNCT__ 104 #define __FUNCT__ "PetscObjectSAWsSetBlock" 105 /*@C 106 PetscObjectSAWsSetBlock - Sets whether an object will block at PetscObjectSAWsBlock() 107 108 Collective on PetscObject 109 110 Input Parameters: 111 + obj - the Petsc variable 112 Thus must be cast with a (PetscObject), for example, 113 PetscObjectSetName((PetscObject)mat,name); 114 - flg - whether it should block 115 116 Level: advanced 117 118 Concepts: publishing object 119 120 .seealso: PetscObjectSetName(), PetscObjectSAWsViewOff(), PetscObjectSAWsBlock() 121 122 @*/ 123 PetscErrorCode PetscObjectSAWsSetBlock(PetscObject obj,PetscBool flg) 124 { 125 PetscFunctionBegin; 126 PetscValidHeader(obj,1); 127 obj->amspublishblock = flg; 128 obj->amsblock = flg; 129 PetscFunctionReturn(0); 130 } 131 132 #undef __FUNCT__ 133 #define __FUNCT__ "PetscObjectSAWsViewOff" 134 PetscErrorCode PetscObjectSAWsViewOff(PetscObject obj) 135 { 136 char dir[1024]; 137 PetscErrorCode ierr; 138 139 PetscFunctionBegin; 140 if (obj->classid == PETSC_VIEWER_CLASSID) PetscFunctionReturn(0); 141 if (!obj->amsmem) PetscFunctionReturn(0); 142 ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s",obj->name);CHKERRQ(ierr); 143 PetscStackCallSAWs(SAWs_Delete,(dir)); 144 PetscFunctionReturn(0); 145 } 146 147