1 #include <petsc/private/drawimpl.h> /*I "petscdraw.h" I*/ 2 3 /*@ 4 PetscDrawLGAddCommonPoint - Adds another point to each of the line graphs. All the points share 5 the same new X coordinate. The new point must have an X coordinate larger than the old points. 6 7 Logically Collective 8 9 Input Parameters: 10 + lg - the line graph context 11 . x - the common x coordinate point 12 - y - the new y coordinate point for each curve. 13 14 Level: intermediate 15 16 Notes: 17 You must call `PetscDrawLGDraw()` to display any added points 18 19 Call `PetscDrawLGReset()` to remove all points 20 21 .seealso: `PetscDrawLG`, `PetscDrawLGCreate()`, `PetscDrawLGAddPoints()`, `PetscDrawLGAddPoint()`, `PetscDrawLGReset()`, `PetscDrawLGDraw()` 22 @*/ 23 PetscErrorCode PetscDrawLGAddCommonPoint(PetscDrawLG lg, const PetscReal x, const PetscReal *y) 24 { 25 PetscInt i; 26 27 PetscFunctionBegin; 28 PetscValidHeaderSpecific(lg, PETSC_DRAWLG_CLASSID, 1); 29 30 if (lg->loc + lg->dim >= lg->len) { /* allocate more space */ 31 PetscReal *tmpx, *tmpy; 32 PetscCall(PetscMalloc2(lg->len + lg->dim * PETSC_DRAW_LG_CHUNK_SIZE, &tmpx, lg->len + lg->dim * PETSC_DRAW_LG_CHUNK_SIZE, &tmpy)); 33 PetscCall(PetscArraycpy(tmpx, lg->x, lg->len)); 34 PetscCall(PetscArraycpy(tmpy, lg->y, lg->len)); 35 PetscCall(PetscFree2(lg->x, lg->y)); 36 lg->x = tmpx; 37 lg->y = tmpy; 38 lg->len += lg->dim * PETSC_DRAW_LG_CHUNK_SIZE; 39 } 40 for (i = 0; i < lg->dim; i++) { 41 if (x > lg->xmax) lg->xmax = x; 42 if (x < lg->xmin) lg->xmin = x; 43 if (y[i] > lg->ymax) lg->ymax = y[i]; 44 if (y[i] < lg->ymin) lg->ymin = y[i]; 45 46 lg->x[lg->loc] = x; 47 lg->y[lg->loc++] = y[i]; 48 } 49 lg->nopts++; 50 PetscFunctionReturn(PETSC_SUCCESS); 51 } 52 53 /*@ 54 PetscDrawLGAddPoint - Adds another point to each of the line graphs. 55 The new point must have an X coordinate larger than the old points. 56 57 Logically Collective 58 59 Input Parameters: 60 + lg - the line graph context 61 . x - array containing the x coordinate for the point on each curve 62 - y - array containing the y coordinate for the point on each curve 63 64 Level: intermediate 65 66 Notes: 67 You must call `PetscDrawLGDraw()` to display any added points 68 69 Call `PetscDrawLGReset()` to remove all points 70 71 .seealso: `PetscDrawLG`, `PetscDrawLGCreate()`, `PetscDrawLGAddPoints()`, `PetscDrawLGAddCommonPoint()`, `PetscDrawLGReset()`, `PetscDrawLGDraw()` 72 @*/ 73 PetscErrorCode PetscDrawLGAddPoint(PetscDrawLG lg, const PetscReal *x, const PetscReal *y) 74 { 75 PetscInt i; 76 PetscReal xx; 77 78 PetscFunctionBegin; 79 PetscValidHeaderSpecific(lg, PETSC_DRAWLG_CLASSID, 1); 80 81 if (lg->loc + lg->dim >= lg->len) { /* allocate more space */ 82 PetscReal *tmpx, *tmpy; 83 PetscCall(PetscMalloc2(lg->len + lg->dim * PETSC_DRAW_LG_CHUNK_SIZE, &tmpx, lg->len + lg->dim * PETSC_DRAW_LG_CHUNK_SIZE, &tmpy)); 84 PetscCall(PetscArraycpy(tmpx, lg->x, lg->len)); 85 PetscCall(PetscArraycpy(tmpy, lg->y, lg->len)); 86 PetscCall(PetscFree2(lg->x, lg->y)); 87 lg->x = tmpx; 88 lg->y = tmpy; 89 lg->len += lg->dim * PETSC_DRAW_LG_CHUNK_SIZE; 90 } 91 for (i = 0; i < lg->dim; i++) { 92 if (!x) { 93 xx = lg->nopts; 94 } else { 95 xx = x[i]; 96 } 97 if (xx > lg->xmax) lg->xmax = xx; 98 if (xx < lg->xmin) lg->xmin = xx; 99 if (y[i] > lg->ymax) lg->ymax = y[i]; 100 if (y[i] < lg->ymin) lg->ymin = y[i]; 101 102 lg->x[lg->loc] = xx; 103 lg->y[lg->loc++] = y[i]; 104 } 105 lg->nopts++; 106 PetscFunctionReturn(PETSC_SUCCESS); 107 } 108 109 /*@C 110 PetscDrawLGAddPoints - Adds several points to each of the line graphs. 111 The new points must have an X coordinate larger than the old points. 112 113 Logically Collective 114 115 Input Parameters: 116 + lg - the line graph context 117 . xx - array of pointers that point to arrays containing the new x coordinates for each curve. 118 . yy - array of pointers that point to arrays containing the new y points for each curve. 119 - n - number of points being added 120 121 Level: intermediate 122 123 Notes: 124 You must call `PetscDrawLGDraw()` to display any added points 125 126 Call `PetscDrawLGReset()` to remove all points 127 128 .seealso: `PetscDrawLG`, `PetscDrawLGCreate()`, `PetscDrawLGAddPoint()`, `PetscDrawLGAddCommonPoint()`, `PetscDrawLGReset()`, `PetscDrawLGDraw()` 129 @*/ 130 PetscErrorCode PetscDrawLGAddPoints(PetscDrawLG lg, PetscInt n, PetscReal **xx, PetscReal **yy) 131 { 132 PetscInt i, j, k; 133 PetscReal *x, *y; 134 135 PetscFunctionBegin; 136 PetscValidHeaderSpecific(lg, PETSC_DRAWLG_CLASSID, 1); 137 138 if (lg->loc + n * lg->dim >= lg->len) { /* allocate more space */ 139 PetscReal *tmpx, *tmpy; 140 PetscInt chunk = PETSC_DRAW_LG_CHUNK_SIZE; 141 142 if (n > chunk) chunk = n; 143 PetscCall(PetscMalloc2(lg->len + lg->dim * chunk, &tmpx, lg->len + lg->dim * chunk, &tmpy)); 144 PetscCall(PetscArraycpy(tmpx, lg->x, lg->len)); 145 PetscCall(PetscArraycpy(tmpy, lg->y, lg->len)); 146 PetscCall(PetscFree2(lg->x, lg->y)); 147 lg->x = tmpx; 148 lg->y = tmpy; 149 lg->len += lg->dim * chunk; 150 } 151 for (j = 0; j < lg->dim; j++) { 152 x = xx[j]; 153 y = yy[j]; 154 k = lg->loc + j; 155 for (i = 0; i < n; i++) { 156 if (x[i] > lg->xmax) lg->xmax = x[i]; 157 if (x[i] < lg->xmin) lg->xmin = x[i]; 158 if (y[i] > lg->ymax) lg->ymax = y[i]; 159 if (y[i] < lg->ymin) lg->ymin = y[i]; 160 161 lg->x[k] = x[i]; 162 lg->y[k] = y[i]; 163 k += lg->dim; 164 } 165 } 166 lg->loc += n * lg->dim; 167 lg->nopts += n; 168 PetscFunctionReturn(PETSC_SUCCESS); 169 } 170