1 2 #include <petscsys.h> 3 #include <petsc/private/viewerimpl.h> 4 5 struct _n_PetscViewers { 6 MPI_Comm comm; 7 PetscViewer *viewer; 8 int n; 9 }; 10 11 /*@C 12 PetscViewersDestroy - Destroys a set of PetscViewers created with PetscViewersCreate(). 13 14 Collective on PetscViewers 15 16 Input Parameters: 17 . v - the PetscViewers to be destroyed. 18 19 Level: intermediate 20 21 .seealso: PetscViewerSocketOpen(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDrawOpen(), PetscViewersCreate() 22 23 @*/ 24 PetscErrorCode PetscViewersDestroy(PetscViewers *v) 25 { 26 int i; 27 PetscErrorCode ierr; 28 29 PetscFunctionBegin; 30 if (!*v) PetscFunctionReturn(0); 31 for (i=0; i<(*v)->n; i++) { 32 ierr = PetscViewerDestroy(&(*v)->viewer[i]);CHKERRQ(ierr); 33 } 34 ierr = PetscFree((*v)->viewer);CHKERRQ(ierr); 35 ierr = PetscFree(*v);CHKERRQ(ierr); 36 PetscFunctionReturn(0); 37 } 38 39 /*@C 40 PetscViewersCreate - Creates a container to hold a set of PetscViewers. 41 42 Collective on MPI_Comm 43 44 Input Parameter: 45 . comm - the MPI communicator 46 47 Output Parameter: 48 . v - the collection of PetscViewers 49 50 Level: intermediate 51 52 .seealso: PetscViewerCreate(), PetscViewersDestroy() 53 54 @*/ 55 PetscErrorCode PetscViewersCreate(MPI_Comm comm,PetscViewers *v) 56 { 57 PetscErrorCode ierr; 58 59 PetscFunctionBegin; 60 ierr = PetscNew(v);CHKERRQ(ierr); 61 (*v)->n = 64; 62 (*v)->comm = comm; 63 64 ierr = PetscCalloc1(64,&(*v)->viewer);CHKERRQ(ierr); 65 PetscFunctionReturn(0); 66 } 67 68 /*@C 69 PetscViewersGetViewer - Gets a PetscViewer from a PetscViewer collection 70 71 Not Collective, but PetscViewer will be collective object on PetscViewers 72 73 Input Parameter: 74 + viewers - object created with PetscViewersCreate() 75 - n - number of PetscViewer you want 76 77 Output Parameter: 78 . viewer - the PetscViewer 79 80 Level: intermediate 81 82 .seealso: PetscViewersCreate(), PetscViewersDestroy() 83 84 @*/ 85 PetscErrorCode PetscViewersGetViewer(PetscViewers viewers,PetscInt n,PetscViewer *viewer) 86 { 87 PetscErrorCode ierr; 88 89 PetscFunctionBegin; 90 if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Cannot access using a negative index - %d\n",n); 91 if (n >= viewers->n) { 92 PetscViewer *v; 93 int newn = n + 64; /* add 64 new ones at a time */ 94 95 ierr = PetscCalloc1(newn,&v);CHKERRQ(ierr); 96 ierr = PetscMemcpy(v,viewers->viewer,viewers->n*sizeof(PetscViewer));CHKERRQ(ierr); 97 ierr = PetscFree(viewers->viewer);CHKERRQ(ierr); 98 99 viewers->viewer = v; 100 } 101 if (!viewers->viewer[n]) { 102 ierr = PetscViewerCreate(viewers->comm,&viewers->viewer[n]);CHKERRQ(ierr); 103 } 104 *viewer = viewers->viewer[n]; 105 PetscFunctionReturn(0); 106 } 107 108 109 110 111 112 113