1 /* 2 Defines the operations for the X PetscDraw implementation. 3 */ 4 5 #include <../src/sys/classes/draw/impls/x/ximpl.h> /*I "petscsys.h" I*/ 6 7 /* 8 These macros transform from the users coordinates to the X-window pixel coordinates. 9 */ 10 #define XTRANS(draw, xwin, x) ((int)(((xwin)->w - 1) * ((draw)->port_xl + (((x - (draw)->coor_xl) * ((draw)->port_xr - (draw)->port_xl)) / ((draw)->coor_xr - (draw)->coor_xl))))) 11 #define YTRANS(draw, xwin, y) (((xwin)->h - 1) - (int)(((xwin)->h - 1) * ((draw)->port_yl + (((y - (draw)->coor_yl) * ((draw)->port_yr - (draw)->port_yl)) / ((draw)->coor_yr - (draw)->coor_yl))))) 12 13 #define ITRANS(draw, xwin, i) ((draw)->coor_xl + (((PetscReal)(i)) * ((draw)->coor_xr - (draw)->coor_xl) / ((xwin)->w - 1) - (draw)->port_xl) / ((draw)->port_xr - (draw)->port_xl)) 14 #define JTRANS(draw, xwin, j) ((draw)->coor_yl + (((PetscReal)(j)) / ((xwin)->h - 1) + (draw)->port_yl - 1) * ((draw)->coor_yr - (draw)->coor_yl) / ((draw)->port_yl - (draw)->port_yr)) 15 16 static PetscErrorCode PetscDrawSetViewport_X(PetscDraw draw, PetscReal xl, PetscReal yl, PetscReal xr, PetscReal yr) 17 { 18 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 19 int xa, ya, xb, yb, xmax = XiWin->w - 1, ymax = XiWin->h - 1; 20 XRectangle box; 21 22 PetscFunctionBegin; 23 xa = (int)(xl * xmax); 24 ya = ymax - (int)(yr * ymax); 25 xb = (int)(xr * xmax); 26 yb = ymax - (int)(yl * ymax); 27 PetscDrawCollectiveBegin(draw); 28 box.x = (short)xa; 29 box.width = (unsigned short)(xb + 1 - xa); 30 box.y = (short)ya; 31 box.height = (unsigned short)(yb + 1 - ya); 32 XSetClipRectangles(XiWin->disp, XiWin->gc.set, 0, 0, &box, 1, Unsorted); 33 PetscDrawCollectiveEnd(draw); 34 PetscFunctionReturn(PETSC_SUCCESS); 35 } 36 37 static PetscErrorCode PetscDrawCoordinateToPixel_X(PetscDraw draw, PetscReal x, PetscReal y, int *i, int *j) 38 { 39 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 40 41 PetscFunctionBegin; 42 *i = XTRANS(draw, XiWin, x); 43 *j = YTRANS(draw, XiWin, y); 44 PetscFunctionReturn(PETSC_SUCCESS); 45 } 46 47 static PetscErrorCode PetscDrawPixelToCoordinate_X(PetscDraw draw, int i, int j, PetscReal *x, PetscReal *y) 48 { 49 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 50 51 PetscFunctionBegin; 52 *x = ITRANS(draw, XiWin, i); 53 *y = JTRANS(draw, XiWin, j); 54 PetscFunctionReturn(PETSC_SUCCESS); 55 } 56 57 static PetscErrorCode PetscDrawPoint_X(PetscDraw draw, PetscReal x, PetscReal y, int c) 58 { 59 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 60 int xx, yy, i, j; 61 62 PetscFunctionBegin; 63 xx = XTRANS(draw, XiWin, x); 64 yy = YTRANS(draw, XiWin, y); 65 PetscDrawXiSetColor(XiWin, c); 66 for (i = -1; i < 2; i++) { 67 for (j = -1; j < 2; j++) XDrawPoint(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, xx + i, yy + j); 68 } 69 PetscFunctionReturn(PETSC_SUCCESS); 70 } 71 72 static PetscErrorCode PetscDrawPointPixel_X(PetscDraw draw, int x, int y, int c) 73 { 74 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 75 76 PetscFunctionBegin; 77 PetscDrawXiSetColor(XiWin, c); 78 XDrawPoint(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x, y); 79 PetscFunctionReturn(PETSC_SUCCESS); 80 } 81 82 static PetscErrorCode PetscDrawLine_X(PetscDraw draw, PetscReal xl, PetscReal yl, PetscReal xr, PetscReal yr, int cl) 83 { 84 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 85 int x_1, y_1, x_2, y_2; 86 87 PetscFunctionBegin; 88 PetscDrawXiSetColor(XiWin, cl); 89 x_1 = XTRANS(draw, XiWin, xl); 90 x_2 = XTRANS(draw, XiWin, xr); 91 y_1 = YTRANS(draw, XiWin, yl); 92 y_2 = YTRANS(draw, XiWin, yr); 93 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_1, y_1, x_2, y_2); 94 PetscFunctionReturn(PETSC_SUCCESS); 95 } 96 97 static PetscErrorCode PetscDrawArrow_X(PetscDraw draw, PetscReal xl, PetscReal yl, PetscReal xr, PetscReal yr, int cl) 98 { 99 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 100 int x_1, y_1, x_2, y_2; 101 102 PetscFunctionBegin; 103 PetscDrawXiSetColor(XiWin, cl); 104 x_1 = XTRANS(draw, XiWin, xl); 105 x_2 = XTRANS(draw, XiWin, xr); 106 y_1 = YTRANS(draw, XiWin, yl); 107 y_2 = YTRANS(draw, XiWin, yr); 108 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_1, y_1, x_2, y_2); 109 if (x_1 == x_2 && y_1 == y_2) PetscFunctionReturn(PETSC_SUCCESS); 110 if (x_1 == x_2 && PetscAbs(y_1 - y_2) > 7) { 111 if (y_2 > y_1) { 112 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 - 3, y_2 - 3); 113 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 + 3, y_2 - 3); 114 } else { 115 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 - 3, y_2 + 3); 116 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 + 3, y_2 + 3); 117 } 118 } 119 if (y_1 == y_2 && PetscAbs(x_1 - x_2) > 7) { 120 if (x_2 > x_1) { 121 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2 - 3, y_2 - 3, x_2, y_2); 122 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2 - 3, y_2 + 3, x_2, y_2); 123 } else { 124 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 + 3, y_2 - 3); 125 XDrawLine(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x_2, y_2, x_2 + 3, y_2 + 3); 126 } 127 } 128 PetscFunctionReturn(PETSC_SUCCESS); 129 } 130 131 static PetscErrorCode PetscDrawRectangle_X(PetscDraw draw, PetscReal xl, PetscReal yl, PetscReal xr, PetscReal yr, int c1, int c2, int c3, int c4) 132 { 133 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 134 int x, y, w, h, c = (c1 + c2 + c3 + c4) / 4; 135 136 PetscFunctionBegin; 137 PetscDrawXiSetColor(XiWin, c); 138 x = XTRANS(draw, XiWin, xl); 139 w = XTRANS(draw, XiWin, xr) + 1 - x; 140 if (w <= 0) w = 1; 141 y = YTRANS(draw, XiWin, yr); 142 h = YTRANS(draw, XiWin, yl) + 1 - y; 143 if (h <= 0) h = 1; 144 XFillRectangle(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, x, y, w, h); 145 PetscFunctionReturn(PETSC_SUCCESS); 146 } 147 148 static PetscErrorCode PetscDrawEllipse_X(PetscDraw draw, PetscReal x, PetscReal y, PetscReal a, PetscReal b, int c) 149 { 150 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 151 int xA, yA, w, h; 152 153 PetscFunctionBegin; 154 PetscDrawXiSetColor(XiWin, c); 155 xA = XTRANS(draw, XiWin, x - a / 2); 156 w = XTRANS(draw, XiWin, x + a / 2) + 1 - xA; 157 w = PetscAbs(w); 158 yA = YTRANS(draw, XiWin, y + b / 2); 159 h = YTRANS(draw, XiWin, y - b / 2) + 1 - yA; 160 h = PetscAbs(h); 161 XFillArc(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, xA, yA, w, h, 0, 360 * 64); 162 PetscFunctionReturn(PETSC_SUCCESS); 163 } 164 165 PETSC_INTERN PetscErrorCode PetscDrawInterpolatedTriangle_X(PetscDraw_X *, int, int, int, int, int, int, int, int, int); 166 167 static PetscErrorCode PetscDrawTriangle_X(PetscDraw draw, PetscReal X1, PetscReal Y_1, PetscReal X2, PetscReal Y2, PetscReal X3, PetscReal Y3, int c1, int c2, int c3) 168 { 169 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 170 171 PetscFunctionBegin; 172 if (c1 == c2 && c2 == c3) { 173 XPoint pt[3]; 174 PetscDrawXiSetColor(XiWin, c1); 175 pt[0].x = XTRANS(draw, XiWin, X1); 176 pt[0].y = YTRANS(draw, XiWin, Y_1); 177 pt[1].x = XTRANS(draw, XiWin, X2); 178 pt[1].y = YTRANS(draw, XiWin, Y2); 179 pt[2].x = XTRANS(draw, XiWin, X3); 180 pt[2].y = YTRANS(draw, XiWin, Y3); 181 XFillPolygon(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, pt, 3, Convex, CoordModeOrigin); 182 } else { 183 int x1, y_1, x2, y2, x3, y3; 184 x1 = XTRANS(draw, XiWin, X1); 185 y_1 = YTRANS(draw, XiWin, Y_1); 186 x2 = XTRANS(draw, XiWin, X2); 187 y2 = YTRANS(draw, XiWin, Y2); 188 x3 = XTRANS(draw, XiWin, X3); 189 y3 = YTRANS(draw, XiWin, Y3); 190 PetscCall(PetscDrawInterpolatedTriangle_X(XiWin, x1, y_1, c1, x2, y2, c2, x3, y3, c3)); 191 } 192 PetscFunctionReturn(PETSC_SUCCESS); 193 } 194 195 static PetscErrorCode PetscDrawStringSetSize_X(PetscDraw draw, PetscReal x, PetscReal y) 196 { 197 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 198 int w, h; 199 200 PetscFunctionBegin; 201 w = (int)((XiWin->w) * x * (draw->port_xr - draw->port_xl) / (draw->coor_xr - draw->coor_xl)); 202 h = (int)((XiWin->h) * y * (draw->port_yr - draw->port_yl) / (draw->coor_yr - draw->coor_yl)); 203 PetscCall(PetscFree(XiWin->font)); 204 PetscCall(PetscDrawXiFontFixed(XiWin, w, h, &XiWin->font)); 205 PetscFunctionReturn(PETSC_SUCCESS); 206 } 207 208 static PetscErrorCode PetscDrawStringGetSize_X(PetscDraw draw, PetscReal *x, PetscReal *y) 209 { 210 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 211 PetscReal w, h; 212 213 PetscFunctionBegin; 214 w = XiWin->font->font_w; 215 h = XiWin->font->font_h; 216 if (x) *x = w * (draw->coor_xr - draw->coor_xl) / ((XiWin->w) * (draw->port_xr - draw->port_xl)); 217 if (y) *y = h * (draw->coor_yr - draw->coor_yl) / ((XiWin->h) * (draw->port_yr - draw->port_yl)); 218 PetscFunctionReturn(PETSC_SUCCESS); 219 } 220 221 static PetscErrorCode PetscDrawString_X(PetscDraw draw, PetscReal x, PetscReal y, int c, const char chrs[]) 222 { 223 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 224 int xx, yy, descent = XiWin->font->font_descent; 225 size_t len; 226 char *substr; 227 PetscToken token; 228 229 PetscFunctionBegin; 230 xx = XTRANS(draw, XiWin, x); 231 yy = YTRANS(draw, XiWin, y); 232 PetscDrawXiSetColor(XiWin, c); 233 234 PetscCall(PetscTokenCreate(chrs, '\n', &token)); 235 PetscCall(PetscTokenFind(token, &substr)); 236 while (substr) { 237 PetscCall(PetscStrlen(substr, &len)); 238 XDrawString(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, xx, yy - descent, substr, len); 239 yy += XiWin->font->font_h; 240 PetscCall(PetscTokenFind(token, &substr)); 241 } 242 PetscCall(PetscTokenDestroy(&token)); 243 PetscFunctionReturn(PETSC_SUCCESS); 244 } 245 246 static PetscErrorCode PetscDrawStringVertical_X(PetscDraw draw, PetscReal x, PetscReal y, int c, const char text[]) 247 { 248 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 249 int xx, yy, offset = XiWin->font->font_h - XiWin->font->font_descent; 250 char chr[2] = {0, 0}; 251 252 PetscFunctionBegin; 253 xx = XTRANS(draw, XiWin, x); 254 yy = YTRANS(draw, XiWin, y); 255 PetscDrawXiSetColor(XiWin, c); 256 while ((chr[0] = *text++)) { 257 XDrawString(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, xx, yy + offset, chr, 1); 258 yy += XiWin->font->font_h; 259 } 260 PetscFunctionReturn(PETSC_SUCCESS); 261 } 262 263 static PetscErrorCode PetscDrawFlush_X(PetscDraw draw) 264 { 265 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 266 PetscMPIInt rank; 267 268 PetscFunctionBegin; 269 /* make sure the X server processed requests from all processes */ 270 PetscDrawCollectiveBegin(draw); 271 XSync(XiWin->disp, False); 272 PetscDrawCollectiveEnd(draw); 273 PetscCallMPI(MPI_Barrier(PetscObjectComm((PetscObject)draw))); 274 275 /* transfer pixmap contents to window (only the first process does this) */ 276 if (XiWin->drw && XiWin->win) { 277 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 278 PetscDrawCollectiveBegin(draw); 279 if (rank == 0) XCopyArea(XiWin->disp, XiWin->drw, XiWin->win, XiWin->gc.set, 0, 0, XiWin->w, XiWin->h, 0, 0); 280 if (rank == 0) XSync(XiWin->disp, False); 281 PetscDrawCollectiveEnd(draw); 282 PetscCallMPI(MPI_Barrier(PetscObjectComm((PetscObject)draw))); 283 } 284 PetscFunctionReturn(PETSC_SUCCESS); 285 } 286 287 static PetscErrorCode PetscDrawClear_X(PetscDraw draw) 288 { 289 PetscDraw_X *XiWin = (PetscDraw_X *)draw->data; 290 int xmax = XiWin->w - 1, ymax = XiWin->h - 1; 291 PetscReal xl = draw->port_xl, yl = draw->port_yl; 292 PetscReal xr = draw->port_xr, yr = draw->port_yr; 293 PetscMPIInt rank; 294 295 PetscFunctionBegin; 296 /* make sure the X server processed requests from all processes */ 297 PetscDrawCollectiveBegin(draw); 298 XSync(XiWin->disp, False); 299 PetscDrawCollectiveEnd(draw); 300 PetscCallMPI(MPI_Barrier(PetscObjectComm((PetscObject)draw))); 301 302 /* only the first process handles the clearing business */ 303 PetscDrawCollectiveBegin(draw); 304 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 305 if (rank == 0) { 306 int xa = (int)(xl * xmax), ya = ymax - (int)(yr * ymax); 307 int xb = (int)(xr * xmax), yb = ymax - (int)(yl * ymax); 308 unsigned int w = (unsigned int)(xb + 1 - xa); 309 unsigned int h = (unsigned int)(yb + 1 - ya); 310 PetscDrawXiSetPixVal(XiWin, XiWin->background); 311 XFillRectangle(XiWin->disp, PetscDrawXiDrawable(XiWin), XiWin->gc.set, xa, ya, w, h); 312 XSync(XiWin->disp, False); 313 } 314 PetscDrawCollectiveEnd(draw); 315 PetscCallMPI(MPI_Barrier(PetscObjectComm((PetscObject)draw))); 316 PetscFunctionReturn(PETSC_SUCCESS); 317 } 318 319 static PetscErrorCode PetscDrawSetDoubleBuffer_X(PetscDraw draw) 320 { 321 PetscDraw_X *win = (PetscDraw_X *)draw->data; 322 PetscMPIInt rank; 323 324 PetscFunctionBegin; 325 if (win->drw) PetscFunctionReturn(PETSC_SUCCESS); 326 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 327 328 PetscDrawCollectiveBegin(draw); 329 if (rank == 0) PetscCall(PetscDrawXiQuickPixmap(win)); 330 PetscDrawCollectiveEnd(draw); 331 PetscCallMPI(MPI_Bcast(&win->drw, 1, MPI_UNSIGNED_LONG, 0, PetscObjectComm((PetscObject)draw))); 332 PetscFunctionReturn(PETSC_SUCCESS); 333 } 334 335 static PetscErrorCode PetscDrawGetPopup_X(PetscDraw draw, PetscDraw *popup) 336 { 337 PetscDraw_X *win = (PetscDraw_X *)draw->data; 338 PetscBool flg = PETSC_TRUE; 339 340 PetscFunctionBegin; 341 PetscCall(PetscOptionsGetBool(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-draw_popup", &flg, NULL)); 342 if (!flg || !win->win) { 343 *popup = NULL; 344 PetscFunctionReturn(PETSC_SUCCESS); 345 } 346 347 PetscCall(PetscDrawCreate(PetscObjectComm((PetscObject)draw), draw->display, NULL, win->x, win->y + win->h + 10, 220, 220, popup)); 348 PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*popup, "popup_")); 349 PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)*popup, ((PetscObject)draw)->prefix)); 350 PetscCall(PetscDrawSetType(*popup, PETSC_DRAW_X)); 351 draw->popup = *popup; 352 PetscFunctionReturn(PETSC_SUCCESS); 353 } 354 355 static PetscErrorCode PetscDrawSetTitle_X(PetscDraw draw, const char title[]) 356 { 357 PetscDraw_X *win = (PetscDraw_X *)draw->data; 358 PetscMPIInt rank; 359 360 PetscFunctionBegin; 361 if (!win->win) PetscFunctionReturn(PETSC_SUCCESS); 362 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 363 PetscDrawCollectiveBegin(draw); 364 if (rank == 0) { 365 size_t len; 366 XTextProperty prop; 367 PetscCall(PetscStrlen(title, &len)); 368 XGetWMName(win->disp, win->win, &prop); 369 XFree((void *)prop.value); 370 prop.value = (unsigned char *)title; 371 prop.nitems = (long)len; 372 XSetWMName(win->disp, win->win, &prop); 373 } 374 PetscDrawCollectiveEnd(draw); 375 PetscFunctionReturn(PETSC_SUCCESS); 376 } 377 378 static PetscErrorCode PetscDrawCheckResizedWindow_X(PetscDraw draw) 379 { 380 PetscDraw_X *win = (PetscDraw_X *)draw->data; 381 int xywh[4]; 382 PetscMPIInt rank; 383 384 PetscFunctionBegin; 385 if (!win->win) PetscFunctionReturn(PETSC_SUCCESS); 386 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 387 388 PetscDrawCollectiveBegin(draw); 389 if (rank == 0) PetscCall(PetscDrawXiGetGeometry(win, xywh, xywh + 1, xywh + 2, xywh + 3)); 390 PetscDrawCollectiveEnd(draw); 391 PetscCallMPI(MPI_Bcast(xywh, 4, MPI_INT, 0, PetscObjectComm((PetscObject)draw))); 392 393 /* record new window position */ 394 draw->x = win->x = xywh[0]; 395 draw->y = win->y = xywh[1]; 396 if (xywh[2] == win->w && xywh[3] == win->h) PetscFunctionReturn(PETSC_SUCCESS); 397 /* record new window sizes */ 398 draw->w = win->w = xywh[2]; 399 draw->h = win->h = xywh[3]; 400 401 /* recreate pixmap (only first processor does this) */ 402 PetscDrawCollectiveBegin(draw); 403 if (rank == 0 && win->drw) PetscCall(PetscDrawXiQuickPixmap(win)); 404 PetscDrawCollectiveEnd(draw); 405 PetscCallMPI(MPI_Bcast(&win->drw, 1, MPI_UNSIGNED_LONG, 0, PetscObjectComm((PetscObject)draw))); 406 /* reset the clipping */ 407 PetscCall(PetscDrawSetViewport_X(draw, draw->port_xl, draw->port_yl, draw->port_xr, draw->port_yr)); 408 PetscFunctionReturn(PETSC_SUCCESS); 409 } 410 411 static PetscErrorCode PetscDrawResizeWindow_X(PetscDraw draw, int w, int h) 412 { 413 PetscDraw_X *win = (PetscDraw_X *)draw->data; 414 PetscMPIInt rank; 415 416 PetscFunctionBegin; 417 if (w == win->w && h == win->h) PetscFunctionReturn(PETSC_SUCCESS); 418 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 419 420 if (win->win) { 421 PetscDrawCollectiveBegin(draw); 422 if (rank == 0) PetscCall(PetscDrawXiResizeWindow(win, w, h)); 423 PetscDrawCollectiveEnd(draw); 424 PetscCall(PetscDrawCheckResizedWindow_X(draw)); 425 } else if (win->drw) { 426 draw->w = win->w = w; 427 draw->h = win->h = h; 428 /* recreate pixmap (only first processor does this) */ 429 PetscDrawCollectiveBegin(draw); 430 if (rank == 0) PetscCall(PetscDrawXiQuickPixmap(win)); 431 PetscCallMPI(MPI_Bcast(&win->drw, 1, MPI_UNSIGNED_LONG, 0, PetscObjectComm((PetscObject)draw))); 432 /* reset the clipping */ 433 PetscDrawCollectiveEnd(draw); 434 PetscCall(PetscDrawSetViewport_X(draw, draw->port_xl, draw->port_yl, draw->port_xr, draw->port_yr)); 435 } 436 PetscFunctionReturn(PETSC_SUCCESS); 437 } 438 439 #include <X11/cursorfont.h> 440 441 static PetscErrorCode PetscDrawGetMouseButton_X(PetscDraw draw, PetscDrawButton *button, PetscReal *x_user, PetscReal *y_user, PetscReal *x_phys, PetscReal *y_phys) 442 { 443 PetscDraw_X *win = (PetscDraw_X *)draw->data; 444 Cursor cursor; 445 XEvent report; 446 Window root, child; 447 int root_x, root_y, px = 0, py = 0; 448 unsigned int w, h, border, depth; 449 unsigned int keys_button; 450 PetscMPIInt rank; 451 PetscReal xx, yy; 452 453 PetscFunctionBegin; 454 *button = PETSC_BUTTON_NONE; 455 if (!win->win) PetscFunctionReturn(PETSC_SUCCESS); 456 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 457 458 PetscDrawCollectiveBegin(draw); 459 if (rank) goto finally; 460 461 /* change cursor to indicate input */ 462 cursor = XCreateFontCursor(win->disp, XC_hand2); 463 PetscCheck(cursor, PETSC_COMM_SELF, PETSC_ERR_LIB, "Unable to create X cursor"); 464 XDefineCursor(win->disp, win->win, cursor); 465 /* wait for mouse button events */ 466 XSelectInput(win->disp, win->win, ButtonPressMask | ButtonReleaseMask); 467 while (XCheckTypedEvent(win->disp, ButtonPress, &report)) 468 ; 469 XMaskEvent(win->disp, ButtonReleaseMask, &report); 470 /* get mouse pointer coordinates */ 471 XQueryPointer(win->disp, report.xmotion.window, &root, &child, &root_x, &root_y, &px, &py, &keys_button); 472 /* the user may resize the window before pressing the mouse button */ 473 XGetGeometry(win->disp, win->win, &root, &root_x, &root_y, &w, &h, &border, &depth); 474 /* cleanup input event handler and cursor */ 475 XSelectInput(win->disp, win->win, NoEventMask); 476 XUndefineCursor(win->disp, win->win); 477 XFreeCursor(win->disp, cursor); 478 XSync(win->disp, False); 479 480 switch (report.xbutton.button) { 481 case Button1: 482 *button = PETSC_BUTTON_LEFT; 483 break; 484 case Button2: 485 *button = PETSC_BUTTON_CENTER; 486 break; 487 case Button3: 488 *button = PETSC_BUTTON_RIGHT; 489 break; 490 case Button4: 491 *button = PETSC_BUTTON_WHEEL_UP; 492 break; 493 case Button5: 494 *button = PETSC_BUTTON_WHEEL_DOWN; 495 break; 496 } 497 if (report.xbutton.state & ShiftMask) { 498 switch (report.xbutton.button) { 499 case Button1: 500 *button = PETSC_BUTTON_LEFT_SHIFT; 501 break; 502 case Button2: 503 *button = PETSC_BUTTON_CENTER_SHIFT; 504 break; 505 case Button3: 506 *button = PETSC_BUTTON_RIGHT_SHIFT; 507 break; 508 } 509 } 510 xx = ((PetscReal)px) / w; 511 yy = 1 - ((PetscReal)py) / h; 512 if (x_user) *x_user = draw->coor_xl + (xx - draw->port_xl) * (draw->coor_xr - draw->coor_xl) / (draw->port_xr - draw->port_xl); 513 if (y_user) *y_user = draw->coor_yl + (yy - draw->port_yl) * (draw->coor_yr - draw->coor_yl) / (draw->port_yr - draw->port_yl); 514 if (x_phys) *x_phys = xx; 515 if (y_phys) *y_phys = yy; 516 517 finally: 518 PetscDrawCollectiveEnd(draw); 519 PetscCall(PetscDrawCheckResizedWindow_X(draw)); 520 PetscFunctionReturn(PETSC_SUCCESS); 521 } 522 523 static PetscErrorCode PetscDrawSetVisible_X(PetscDraw draw, PetscBool visible) 524 { 525 PetscDraw_X *Xwin = (PetscDraw_X *)draw->data; 526 PetscMPIInt rank; 527 528 PetscFunctionBegin; 529 PetscDrawCollectiveBegin(draw); 530 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 531 if (rank == 0) { 532 if (Xwin->win) { 533 if (visible) { 534 XMapWindow(Xwin->disp, Xwin->win); 535 } else { 536 XUnmapWindow(Xwin->disp, Xwin->win); 537 XFlush(Xwin->disp); 538 } 539 } 540 } 541 PetscDrawCollectiveEnd(draw); 542 PetscFunctionReturn(PETSC_SUCCESS); 543 } 544 545 static PetscErrorCode PetscDrawPause_X(PetscDraw draw) 546 { 547 PetscDraw_X *win = (PetscDraw_X *)draw->data; 548 549 PetscFunctionBegin; 550 if (!win->win) PetscFunctionReturn(PETSC_SUCCESS); 551 if (draw->pause > 0) PetscCall(PetscSleep(draw->pause)); 552 else if (draw->pause == -1) { 553 PetscDrawButton button = PETSC_BUTTON_NONE; 554 PetscCall(PetscDrawGetMouseButton(draw, &button, NULL, NULL, NULL, NULL)); 555 if (button == PETSC_BUTTON_CENTER) draw->pause = 0; 556 } 557 PetscFunctionReturn(PETSC_SUCCESS); 558 } 559 560 static PetscErrorCode PetscDrawDestroy_X(PetscDraw draw) 561 { 562 PetscDraw_X *win = (PetscDraw_X *)draw->data; 563 564 PetscFunctionBegin; 565 PetscCall(PetscDrawDestroy(&draw->popup)); 566 PetscCall(PetscDrawXiClose(win)); 567 PetscCall(PetscFree(draw->data)); 568 PetscFunctionReturn(PETSC_SUCCESS); 569 } 570 571 static PetscErrorCode PetscDrawGetSingleton_X(PetscDraw, PetscDraw *); 572 static PetscErrorCode PetscDrawRestoreSingleton_X(PetscDraw, PetscDraw *); 573 PETSC_INTERN PetscErrorCode PetscDrawGetImage_X(PetscDraw, unsigned char[][3], unsigned int *, unsigned int *, unsigned char *[]); 574 575 static struct _PetscDrawOps DvOps = {PetscDrawSetDoubleBuffer_X, PetscDrawFlush_X, PetscDrawLine_X, NULL, NULL, PetscDrawPoint_X, NULL, PetscDrawString_X, PetscDrawStringVertical_X, PetscDrawStringSetSize_X, PetscDrawStringGetSize_X, PetscDrawSetViewport_X, PetscDrawClear_X, PetscDrawRectangle_X, PetscDrawTriangle_X, PetscDrawEllipse_X, PetscDrawGetMouseButton_X, PetscDrawPause_X, NULL, NULL, PetscDrawGetPopup_X, PetscDrawSetTitle_X, PetscDrawCheckResizedWindow_X, PetscDrawResizeWindow_X, PetscDrawDestroy_X, NULL, PetscDrawGetSingleton_X, PetscDrawRestoreSingleton_X, NULL, PetscDrawGetImage_X, NULL, PetscDrawArrow_X, PetscDrawCoordinateToPixel_X, PetscDrawPixelToCoordinate_X, PetscDrawPointPixel_X, NULL, PetscDrawSetVisible_X}; 576 577 static PetscErrorCode PetscDrawGetSingleton_X(PetscDraw draw, PetscDraw *sdraw) 578 { 579 PetscDraw_X *Xwin = (PetscDraw_X *)draw->data, *sXwin; 580 581 PetscFunctionBegin; 582 PetscCall(PetscDrawCreate(PETSC_COMM_SELF, draw->display, draw->title, draw->x, draw->y, draw->w, draw->h, sdraw)); 583 PetscCall(PetscObjectChangeTypeName((PetscObject)*sdraw, PETSC_DRAW_X)); 584 PetscCall(PetscMemcpy((*sdraw)->ops, &DvOps, sizeof(DvOps))); 585 586 if (draw->popup) PetscCall(PetscDrawGetSingleton(draw->popup, &(*sdraw)->popup)); 587 (*sdraw)->pause = draw->pause; 588 (*sdraw)->coor_xl = draw->coor_xl; 589 (*sdraw)->coor_xr = draw->coor_xr; 590 (*sdraw)->coor_yl = draw->coor_yl; 591 (*sdraw)->coor_yr = draw->coor_yr; 592 (*sdraw)->port_xl = draw->port_xl; 593 (*sdraw)->port_xr = draw->port_xr; 594 (*sdraw)->port_yl = draw->port_yl; 595 (*sdraw)->port_yr = draw->port_yr; 596 597 /* share drawables (windows and/or pixmap) from the parent draw */ 598 PetscCall(PetscNew(&sXwin)); 599 (*sdraw)->data = (void *)sXwin; 600 PetscCall(PetscDrawXiInit(sXwin, draw->display)); 601 if (Xwin->win) { 602 PetscCall(PetscDrawXiQuickWindowFromWindow(sXwin, Xwin->win)); 603 sXwin->drw = Xwin->drw; /* XXX If the window is ever resized, this is wrong! */ 604 } else if (Xwin->drw) { 605 PetscCall(PetscDrawXiColormap(sXwin)); 606 sXwin->drw = Xwin->drw; 607 } 608 PetscCall(PetscDrawXiGetGeometry(sXwin, &sXwin->x, &sXwin->y, &sXwin->w, &sXwin->h)); 609 (*sdraw)->x = sXwin->x; 610 (*sdraw)->y = sXwin->y; 611 (*sdraw)->w = sXwin->w; 612 (*sdraw)->h = sXwin->h; 613 PetscFunctionReturn(PETSC_SUCCESS); 614 } 615 616 static PetscErrorCode PetscDrawRestoreSingleton_X(PetscDraw draw, PetscDraw *sdraw) 617 { 618 PetscFunctionBegin; 619 if (draw->popup && (*sdraw)->popup) { 620 PetscBool isdrawx; 621 PetscDraw_X *pXwin = (PetscDraw_X *)draw->popup->data; 622 PetscDraw_X *sXwin = (PetscDraw_X *)(*sdraw)->popup->data; 623 PetscCall(PetscObjectTypeCompare((PetscObject)draw->popup, PETSC_DRAW_X, &isdrawx)); 624 if (!isdrawx) goto finally; 625 PetscCall(PetscObjectTypeCompare((PetscObject)(*sdraw)->popup, PETSC_DRAW_X, &isdrawx)); 626 if (!isdrawx) goto finally; 627 if (sXwin->win == pXwin->win) PetscCall(PetscDrawRestoreSingleton(draw->popup, &(*sdraw)->popup)); 628 } 629 finally: 630 PetscCall(PetscDrawDestroy(sdraw)); 631 PetscFunctionReturn(PETSC_SUCCESS); 632 } 633 634 static PetscErrorCode PetscDrawXGetDisplaySize_Private(const char name[], int *width, int *height, PetscBool *has_display) 635 { 636 Display *display; 637 638 PetscFunctionBegin; 639 display = XOpenDisplay(name); 640 if (display) { 641 *has_display = PETSC_TRUE; 642 *width = (int)DisplayWidth(display, DefaultScreen(display)); 643 *height = (int)DisplayHeight(display, DefaultScreen(display)); 644 XCloseDisplay(display); 645 } else { 646 *has_display = PETSC_FALSE; 647 *width = 0; 648 *height = 0; 649 PetscCall((*PetscErrorPrintf)("Unable to open display on %s\n\ 650 Make sure your COMPUTE NODES are authorized to connect\n\ 651 to this X server and either your DISPLAY variable\n\ 652 is set or you use the -display name option\n", 653 name)); 654 } 655 PetscFunctionReturn(PETSC_SUCCESS); 656 } 657 658 /*MC 659 PETSC_DRAW_X - PETSc graphics device that uses either X windows or its virtual version Xvfb 660 661 Options Database Keys: 662 + -display <display> - sets the display to use 663 . -x_virtual - forces use of a X virtual display Xvfb that will not display anything but -draw_save will still work. 664 Xvfb is automatically started up in PetscSetDisplay() with this option 665 . -draw_size w,h - percentage of screen (either 1, .5, .3, .25), or size in pixels 666 . -geometry x,y,w,h - set location and size in pixels 667 . -draw_virtual - do not open a window (draw on a pixmap), -draw_save will still work 668 - -draw_double_buffer - avoid window flickering (draw on pixmap and flush to window) 669 670 Level: beginner 671 672 .seealso: `PetscDraw`, `PetscDrawOpenX()`, `PetscDrawSetDisplay()`, `PetscDrawSetFromOptions()` 673 M*/ 674 675 PETSC_EXTERN PetscErrorCode PetscDrawCreate_X(PetscDraw draw) 676 { 677 PetscDraw_X *Xwin; 678 PetscMPIInt rank; 679 int x = draw->x, y = draw->y, w = draw->w, h = draw->h; 680 static int xavailable = 0, yavailable = 0, ybottom = 0, xmax = 0, ymax = 0; 681 PetscBool set, dvirtual = PETSC_FALSE, doublebuffer = PETSC_TRUE, has_display; 682 PetscInt xywh[4], osize = 4, nsizes = 2; 683 PetscReal sizes[2] = {.3, .3}; 684 static size_t DISPLAY_LENGTH = 265; 685 686 PetscFunctionBegin; 687 /* get the display variable */ 688 if (!draw->display) { 689 PetscCall(PetscMalloc1(DISPLAY_LENGTH, &draw->display)); 690 PetscCall(PetscGetDisplay(draw->display, DISPLAY_LENGTH)); 691 } 692 693 /* initialize the display size */ 694 if (!xmax) { 695 PetscCall(PetscDrawXGetDisplaySize_Private(draw->display, &xmax, &ymax, &has_display)); 696 /* if some processors fail on this and others succeed then this is a problem ! */ 697 if (!has_display) { 698 PetscCall((*PetscErrorPrintf)("PETSc unable to use X windows\nproceeding without graphics\n")); 699 PetscCall(PetscDrawSetType(draw, PETSC_DRAW_NULL)); 700 PetscFunctionReturn(PETSC_SUCCESS); 701 } 702 } 703 704 /* allow user to set size of drawable */ 705 PetscCall(PetscOptionsGetRealArray(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-draw_size", sizes, &nsizes, &set)); 706 if (set && nsizes == 1 && sizes[0] > 1.0) sizes[1] = sizes[0]; 707 if (set) { 708 if (sizes[0] > 1.0) w = (int)sizes[0]; 709 else if (sizes[0] == 1.0) w = PETSC_DRAW_FULL_SIZE; 710 else if (sizes[0] == .5) w = PETSC_DRAW_HALF_SIZE; 711 else if (PetscEqualReal(sizes[0], .3)) w = PETSC_DRAW_THIRD_SIZE; /* sizes == 0.3 will never be true */ 712 else if (sizes[0] == .25) w = PETSC_DRAW_QUARTER_SIZE; 713 if (sizes[1] > 1.0) h = (int)sizes[1]; 714 else if (sizes[1] == 1.0) h = PETSC_DRAW_FULL_SIZE; 715 else if (sizes[1] == .5) h = PETSC_DRAW_HALF_SIZE; 716 else if (PetscEqualReal(sizes[1], .3)) h = PETSC_DRAW_THIRD_SIZE; /* sizes == 0.3 will never be true */ 717 else if (sizes[1] == .25) h = PETSC_DRAW_QUARTER_SIZE; 718 } 719 if (w == PETSC_DECIDE || w == PETSC_DEFAULT) w = draw->w = 300; 720 if (h == PETSC_DECIDE || h == PETSC_DEFAULT) h = draw->h = 300; 721 switch (w) { 722 case PETSC_DRAW_FULL_SIZE: 723 w = draw->w = (xmax - 10); 724 break; 725 case PETSC_DRAW_HALF_SIZE: 726 w = draw->w = (xmax - 20) / 2; 727 break; 728 case PETSC_DRAW_THIRD_SIZE: 729 w = draw->w = (xmax - 30) / 3; 730 break; 731 case PETSC_DRAW_QUARTER_SIZE: 732 w = draw->w = (xmax - 40) / 4; 733 break; 734 } 735 switch (h) { 736 case PETSC_DRAW_FULL_SIZE: 737 h = draw->h = (ymax - 10); 738 break; 739 case PETSC_DRAW_HALF_SIZE: 740 h = draw->h = (ymax - 20) / 2; 741 break; 742 case PETSC_DRAW_THIRD_SIZE: 743 h = draw->h = (ymax - 30) / 3; 744 break; 745 case PETSC_DRAW_QUARTER_SIZE: 746 h = draw->h = (ymax - 40) / 4; 747 break; 748 } 749 750 PetscCall(PetscOptionsGetBool(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-draw_virtual", &dvirtual, NULL)); 751 752 if (!dvirtual) { 753 /* allow user to set location and size of window */ 754 xywh[0] = x; 755 xywh[1] = y; 756 xywh[2] = w; 757 xywh[3] = h; 758 PetscCall(PetscOptionsGetIntArray(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-geometry", xywh, &osize, NULL)); 759 x = (int)xywh[0]; 760 y = (int)xywh[1]; 761 w = (int)xywh[2]; 762 h = (int)xywh[3]; 763 if (w == PETSC_DECIDE || w == PETSC_DEFAULT) w = 300; 764 if (h == PETSC_DECIDE || h == PETSC_DEFAULT) h = 300; 765 draw->x = x; 766 draw->y = y; 767 draw->w = w; 768 draw->h = h; 769 770 if (draw->x == PETSC_DECIDE || draw->y == PETSC_DECIDE) { 771 /* 772 PETSc tries to place windows starting in the upper left corner 773 and moving across to the right. 774 775 +0,0-------------------------------------------+ 776 | Region used so far +xavailable,yavailable | 777 | | | 778 | | | 779 +--------------------- +ybottom | 780 | | 781 | | 782 +----------------------------------------------+xmax,ymax 783 784 */ 785 /* First: can we add it to the right? */ 786 if (xavailable + w + 10 <= xmax) { 787 x = xavailable; 788 y = yavailable; 789 ybottom = PetscMax(ybottom, y + h + 30); 790 } else { 791 /* No, so add it below on the left */ 792 xavailable = x = 0; 793 yavailable = y = ybottom; 794 ybottom = ybottom + h + 30; 795 } 796 } 797 /* update available region */ 798 xavailable = PetscMax(xavailable, x + w + 10); 799 if (xavailable >= xmax) { 800 xavailable = 0; 801 yavailable = yavailable + h + 30; 802 ybottom = yavailable; 803 } 804 if (yavailable >= ymax) { 805 y = 0; 806 yavailable = 0; 807 ybottom = 0; 808 } 809 810 } /* endif (!dvirtual) */ 811 812 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)draw), &rank)); 813 PetscCheck(rank != 0 || (w > 0 && h > 0), PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative window width or height"); 814 815 PetscCall(PetscNew(&Xwin)); 816 PetscCall(PetscMemcpy(draw->ops, &DvOps, sizeof(DvOps))); 817 draw->data = (void *)Xwin; 818 819 PetscCall(PetscDrawXiInit(Xwin, draw->display)); 820 if (!dvirtual) { 821 Xwin->x = x; 822 Xwin->y = y; 823 Xwin->w = w; 824 Xwin->h = h; 825 if (rank == 0) PetscCall(PetscDrawXiQuickWindow(Xwin, draw->title, x, y, w, h)); 826 PetscCallMPI(MPI_Bcast(&Xwin->win, 1, MPI_UNSIGNED_LONG, 0, PetscObjectComm((PetscObject)draw))); 827 if (rank) PetscCall(PetscDrawXiQuickWindowFromWindow(Xwin, Xwin->win)); 828 } else { 829 Xwin->x = 0; 830 Xwin->y = 0; 831 Xwin->w = w; 832 Xwin->h = h; 833 PetscCall(PetscDrawXiColormap(Xwin)); 834 if (rank == 0) PetscCall(PetscDrawXiQuickPixmap(Xwin)); 835 PetscCallMPI(MPI_Bcast(&Xwin->drw, 1, MPI_UNSIGNED_LONG, 0, PetscObjectComm((PetscObject)draw))); 836 } 837 PetscCall(PetscDrawXiGetGeometry(Xwin, &Xwin->x, &Xwin->y, &Xwin->w, &Xwin->h)); 838 draw->x = Xwin->x; 839 draw->y = Xwin->y; 840 draw->w = Xwin->w; 841 draw->h = Xwin->h; 842 843 PetscCall(PetscOptionsGetBool(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-draw_double_buffer", &doublebuffer, NULL)); 844 if (doublebuffer) PetscCall(PetscDrawSetDoubleBuffer(draw)); 845 PetscFunctionReturn(PETSC_SUCCESS); 846 } 847