/*@C
PetscDrawViewPortsCreate - Splits a window into smaller view ports. Each processor shares all the viewports.
Collective on PetscDraw
Input Parameters:
+ draw - the drawing context
- nports - the number of ports
Output Parameter:
. ports - a PetscDrawViewPorts context (C structure)
Options Database:
. -draw_ports - display multiple fields in the same window with PetscDrawPorts instead of in seperate windows
Level: advanced
Concepts: drawing^in subset of window
.seealso: PetscDrawSplitViewPort(), PetscDrawSetViewPort(), PetscDrawViewPortsSet(), PetscDrawViewPortsDestroy()
@*/
PetscErrorCode PetscDrawViewPortsCreate(PetscDraw draw,PetscInt nports,PetscDrawViewPorts **newports)
{
PetscDrawViewPorts *ports;
PetscInt i,n;
PetscBool isnull;
PetscMPIInt rank;
PetscReal *xl,*xr,*yl,*yr,h;
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
if (nports < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Number of divisions must be positive: %d", nports);
PetscValidPointer(newports,3);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
if (isnull) {*newports = NULL; PetscFunctionReturn(0);}
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)draw),&rank);CHKERRQ(ierr);
ierr = PetscNew(&ports);CHKERRQ(ierr); *newports = ports;
ports->draw = draw;
ports->nports = nports;
ierr = PetscObjectReference((PetscObject)draw);CHKERRQ(ierr);
/* save previous drawport of window */
ierr = PetscDrawGetViewPort(draw,&ports->port_xl,&ports->port_yl,&ports->port_xr,&ports->port_yr);CHKERRQ(ierr);
n = (PetscInt)(.1 + PetscSqrtReal((PetscReal)nports));
while (n*n < nports) n++;
h = 1.0/n;
ierr = PetscMalloc4(n*n,&xl,n*n,&xr,n*n,&yl,n*n,&yr);CHKERRQ(ierr);
ports->xl = xl;
ports->xr = xr;
ports->yl = yl;
ports->yr = yr;
ierr = PetscDrawSetCoordinates(draw,0.0,0.0,1.0,1.0);CHKERRQ(ierr);
ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
for (i=0; i<n*n; i++) {
xl[i] = (i % n)*h;
xr[i] = xl[i] + h;
yl[i] = (i / n)*h;
yr[i] = yl[i] + h;
if (!rank) {
ierr = PetscDrawLine(draw,xl[i],yl[i],xl[i],yr[i],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xl[i],yr[i],xr[i],yr[i],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr[i],yr[i],xr[i],yl[i],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr[i],yl[i],xl[i],yl[i],PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
xl[i] += .05*h;
xr[i] -= .05*h;
yl[i] += .05*h;
yr[i] -= .05*h;
}
ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
PetscDrawScalePopup - draws a contour scale window.
Collective on PetscDraw
Input Parameters:
+ popup - the window (often a window obtained via PetscDrawGetPopup()
. min - minimum value being plotted
- max - maximum value being plotted
Level: intermediate
Notes: All processors that share the draw MUST call this routine
.seealso: PetscDrawGetPopup(), PetscDrawTensorContour()
@*/
PetscErrorCode PetscDrawScalePopup(PetscDraw popup,PetscReal min,PetscReal max)
{
PetscBool isnull;
PetscReal xl = 0.0,yl = 0.0,xr = 1.0,yr = 1.0;
PetscMPIInt rank;
PetscErrorCode ierr;
int i;
char string[32];
PetscFunctionBegin;
if (!popup) PetscFunctionReturn(0);
PetscValidHeaderSpecific(popup,PETSC_DRAW_CLASSID,1);
ierr = PetscDrawIsNull(popup,&isnull);
CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)popup),&rank);
CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(popup);
CHKERRQ(ierr);
ierr = PetscDrawClear(popup);
CHKERRQ(ierr);
ierr = PetscDrawSetTitle(popup,"Contour Scale");
CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(popup,xl,yl,xr,yr);
CHKERRQ(ierr);
ierr = PetscDrawCollectiveBegin(popup);
CHKERRQ(ierr);
if (!rank) {
for (i=0; i<10; i++) {
int c = PetscDrawRealToColor((PetscReal)i/9,0,1);
ierr = PetscDrawRectangle(popup,xl,yl,xr,yr,c,c,c,c);
CHKERRQ(ierr);
yl += 0.1;
}
for (i=0; i<10; i++) {
PetscReal value = min + i*(max-min)/9;
/* look for a value that should be zero, but is not due to round-off */
if (PetscAbsReal(value) < 1.e-10 && max-min > 1.e-6) value = 0.0;
ierr = PetscSNPrintf(string,sizeof(string),"%18.16e",(double)value);
CHKERRQ(ierr);
ierr = PetscDrawString(popup,0.2,0.02+i/10.0,PETSC_DRAW_BLACK,string);
CHKERRQ(ierr);
}
}
ierr = PetscDrawCollectiveEnd(popup);
CHKERRQ(ierr);
ierr = PetscDrawFlush(popup);
CHKERRQ(ierr);
ierr = PetscDrawSave(popup);
CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode DrawFA(FA fa,Vec v)
{
PetscErrorCode ierr;
PetscScalar *va;
ZoomCtx zctx;
PetscReal xmint = 10000.0,xmaxt = -10000.0,ymint = 100000.0,ymaxt = -10000.0;
PetscReal xmin,xmax,ymin,ymax;
PetscInt i,vn,ln,j;
PetscFunctionBeginUser;
ierr = VecGetArray(v,&va);CHKERRQ(ierr);
ierr = VecGetSize(v,&vn);CHKERRQ(ierr);
ierr = VecGetSize(fa->l,&ln);CHKERRQ(ierr);
for (j=0; j<3; j++) {
if (vn == ln) {
zctx.xy[j] = va + 2*fa->offl[j];
zctx.m[j] = fa->ml[j];
zctx.n[j] = fa->nl[j];
} else {
zctx.xy[j] = va + 2*fa->offg[j];
zctx.m[j] = fa->mg[j];
zctx.n[j] = fa->ng[j];
}
for (i=0; i<zctx.m[j]*zctx.n[j]; i++) {
if (zctx.xy[j][2*i] > xmax) xmax = zctx.xy[j][2*i];
if (zctx.xy[j][2*i] < xmin) xmin = zctx.xy[j][2*i];
if (zctx.xy[j][2*i+1] > ymax) ymax = zctx.xy[j][2*i+1];
if (zctx.xy[j][2*i+1] < ymin) ymin = zctx.xy[j][2*i+1];
}
}
ierr = MPI_Allreduce(&xmin,&xmint,1,MPI_DOUBLE,MPI_MIN,PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = MPI_Allreduce(&xmax,&xmaxt,1,MPI_DOUBLE,MPI_MAX,PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = MPI_Allreduce(&ymin,&ymint,1,MPI_DOUBLE,MPI_MIN,PETSC_COMM_WORLD);CHKERRQ(ierr);
ierr = MPI_Allreduce(&ymax,&ymaxt,1,MPI_DOUBLE,MPI_MAX,PETSC_COMM_WORLD);CHKERRQ(ierr);
xmin = xmint - .2*(xmaxt - xmint);
xmax = xmaxt + .2*(xmaxt - xmint);
ymin = ymint - .2*(ymaxt - ymint);
ymax = ymaxt + .2*(ymaxt - ymint);
#if defined(PETSC_HAVE_X) || defined(PETSC_HAVE_OPENGL)
{
PetscDraw draw;
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"meshes",PETSC_DECIDE,PETSC_DECIDE,700,700,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);CHKERRQ(ierr);
ierr = PetscDrawZoom(draw,DrawPatch,&zctx);CHKERRQ(ierr);
ierr = VecRestoreArray(v,&va);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
}
#endif
PetscFunctionReturn(0);
}
static PetscErrorCode MatFDColoringView_Draw(MatFDColoring fd,PetscViewer viewer)
{
PetscErrorCode ierr;
PetscBool isnull;
PetscDraw draw;
PetscReal xr,yr,xl,yl,h,w;
PetscFunctionBegin;
ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
ierr = PetscObjectCompose((PetscObject)fd,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
xr = fd->N; yr = fd->M; h = yr/10.0; w = xr/10.0;
xr += w; yr += h; xl = -w; yl = -h;
ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
ierr = PetscDrawZoom(draw,MatFDColoringView_Draw_Zoom,fd);CHKERRQ(ierr);
ierr = PetscObjectCompose((PetscObject)fd,"Zoomviewer",NULL);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode DMView_DA_1d(DM da,PetscViewer viewer)
{
PetscErrorCode ierr;
PetscMPIInt rank;
PetscBool iascii,isdraw,isbinary;
DM_DA *dd = (DM_DA*)da->data;
#if defined(PETSC_HAVE_MATLAB_ENGINE)
PetscBool ismatlab;
#endif
PetscFunctionBegin;
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
#if defined(PETSC_HAVE_MATLAB_ENGINE)
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERMATLAB,&ismatlab);CHKERRQ(ierr);
#endif
if (iascii) {
PetscViewerFormat format;
ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL) {
DMDALocalInfo info;
ierr = DMDAGetLocalInfo(da,&info);CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Processor [%d] M %D m %D w %D s %D\n",rank,dd->M,dd->m,dd->w,dd->s);CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D\n",info.xs,info.xs+info.xm);CHKERRQ(ierr);
ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr);
} else {
ierr = DMView_DA_VTK(da, viewer);CHKERRQ(ierr);
}
} else if (isdraw) {
PetscDraw draw;
double ymin = -1,ymax = 1,xmin = -1,xmax = dd->M,x;
PetscInt base;
char node[10];
PetscBool isnull;
ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);CHKERRQ(ierr);
ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
/* first processor draws all node lines */
if (!rank) {
PetscInt xmin_tmp;
ymin = 0.0; ymax = 0.3;
for (xmin_tmp=0; xmin_tmp < dd->M; xmin_tmp++) {
ierr = PetscDrawLine(draw,(double)xmin_tmp,ymin,(double)xmin_tmp,ymax,PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
xmin = 0.0; xmax = dd->M - 1;
ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
/* draw my box */
ymin = 0; ymax = 0.3; xmin = dd->xs / dd->w; xmax = (dd->xe / dd->w) - 1;
ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
/* Put in index numbers */
base = dd->base / dd->w;
for (x=xmin; x<=xmax; x++) {
sprintf(node,"%d",(int)base++);
ierr = PetscDrawString(draw,x,ymin,PETSC_DRAW_RED,node);CHKERRQ(ierr);
}
ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
} else if (isbinary) {
ierr = DMView_DA_Binary(da,viewer);CHKERRQ(ierr);
#if defined(PETSC_HAVE_MATLAB_ENGINE)
} else if (ismatlab) {
ierr = DMView_DA_Matlab(da,viewer);CHKERRQ(ierr);
#endif
}
PetscFunctionReturn(0);
}
/*@C
PetscDrawViewPortsCreateRect - Splits a window into smaller
view ports. Each processor shares all the viewports. The number
of views in the x- and y-directions is specified.
Collective on PetscDraw
Input Parameters:
+ draw - the drawing context
. nx - the number of x divisions
- ny - the number of y divisions
Output Parameter:
. ports - a PetscDrawViewPorts context (C structure)
Level: advanced
Concepts: drawing^in subset of window
.seealso: PetscDrawSplitViewPort(), PetscDrawSetViewPort(), PetscDrawViewPortsSet(), PetscDrawViewPortsDestroy()
@*/
PetscErrorCode PetscDrawViewPortsCreateRect(PetscDraw draw,PetscInt nx,PetscInt ny,PetscDrawViewPorts **newports)
{
PetscDrawViewPorts *ports;
PetscReal *xl,*xr,*yl,*yr,hx,hy;
PetscInt i,j,k,n;
PetscBool isnull;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
if ((nx < 1) || (ny < 1)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Number of divisions must be positive: %d x %d", nx, ny);
PetscValidPointer(newports,3);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
if (isnull) {*newports = NULL; PetscFunctionReturn(0);}
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)draw),&rank);CHKERRQ(ierr);
n = nx*ny;
hx = 1.0/nx;
hy = 1.0/ny;
ierr = PetscNew(&ports);CHKERRQ(ierr); *newports = ports;
ports->draw = draw;
ports->nports = n;
ierr = PetscObjectReference((PetscObject) draw);CHKERRQ(ierr);
/* save previous drawport of window */
ierr = PetscDrawGetViewPort(draw,&ports->port_xl,&ports->port_yl,&ports->port_xr,&ports->port_yr);CHKERRQ(ierr);
ierr = PetscMalloc4(n,&xl,n,&xr,n,&yl,n,&yr);CHKERRQ(ierr);
ports->xr = xr;
ports->xl = xl;
ports->yl = yl;
ports->yr = yr;
ierr = PetscDrawSetCoordinates(draw,0.0,0.0,1.0,1.0);CHKERRQ(ierr);
ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
for (i = 0; i < nx; i++) {
for (j = 0; j < ny; j++) {
k = j*nx+i;
xl[k] = i*hx;
xr[k] = xl[k] + hx;
yl[k] = j*hy;
yr[k] = yl[k] + hy;
if (!rank) {
ierr = PetscDrawLine(draw,xl[k],yl[k],xl[k],yr[k],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xl[k],yr[k],xr[k],yr[k],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr[k],yr[k],xr[k],yl[k],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr[k],yl[k],xl[k],yl[k],PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
xl[k] += .05*hx;
xr[k] -= .05*hx;
yl[k] += .05*hy;
yr[k] -= .05*hy;
}
}
ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode VecView_MPI_Draw_DA1d(Vec xin,PetscViewer v)
{
DM da;
PetscErrorCode ierr;
PetscMPIInt rank,size,tag1,tag2;
PetscInt i,n,N,step,istart,isize,j,nbounds;
MPI_Status status;
PetscReal coors[4],ymin,ymax,min,max,xmin = 0.0,xmax = 0.0,tmp = 0.0,xgtmp = 0.0;
const PetscScalar *array,*xg;
PetscDraw draw;
PetscBool isnull,showpoints = PETSC_FALSE;
MPI_Comm comm;
PetscDrawAxis axis;
Vec xcoor;
DMBoundaryType bx;
const PetscReal *bounds;
PetscInt *displayfields;
PetscInt k,ndisplayfields;
PetscBool hold;
PetscFunctionBegin;
ierr = PetscViewerDrawGetDraw(v,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
ierr = PetscViewerDrawGetBounds(v,&nbounds,&bounds);CHKERRQ(ierr);
ierr = VecGetDM(xin,&da);CHKERRQ(ierr);
if (!da) SETERRQ(PetscObjectComm((PetscObject)xin),PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
ierr = PetscOptionsGetBool(NULL,"-draw_vec_mark_points",&showpoints,NULL);CHKERRQ(ierr);
ierr = DMDAGetInfo(da,0,&N,0,0,0,0,0,&step,0,&bx,0,0,0);CHKERRQ(ierr);
ierr = DMDAGetCorners(da,&istart,0,0,&isize,0,0);CHKERRQ(ierr);
ierr = VecGetArrayRead(xin,&array);CHKERRQ(ierr);
ierr = VecGetLocalSize(xin,&n);CHKERRQ(ierr);
n = n/step;
/* get coordinates of nodes */
ierr = DMGetCoordinates(da,&xcoor);CHKERRQ(ierr);
if (!xcoor) {
ierr = DMDASetUniformCoordinates(da,0.0,1.0,0.0,0.0,0.0,0.0);CHKERRQ(ierr);
ierr = DMGetCoordinates(da,&xcoor);CHKERRQ(ierr);
}
ierr = VecGetArrayRead(xcoor,&xg);CHKERRQ(ierr);
ierr = PetscObjectGetComm((PetscObject)xin,&comm);CHKERRQ(ierr);
ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
/*
Determine the min and max x coordinate in plot
*/
if (!rank) {
xmin = PetscRealPart(xg[0]);
}
if (rank == size-1) {
xmax = PetscRealPart(xg[n-1]);
}
ierr = MPI_Bcast(&xmin,1,MPIU_REAL,0,comm);CHKERRQ(ierr);
ierr = MPI_Bcast(&xmax,1,MPIU_REAL,size-1,comm);CHKERRQ(ierr);
ierr = DMDASelectFields(da,&ndisplayfields,&displayfields);CHKERRQ(ierr);
for (k=0; k<ndisplayfields; k++) {
j = displayfields[k];
ierr = PetscViewerDrawGetDraw(v,k,&draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
/*
Determine the min and max y coordinate in plot
*/
min = 1.e20; max = -1.e20;
for (i=0; i<n; i++) {
if (PetscRealPart(array[j+i*step]) < min) min = PetscRealPart(array[j+i*step]);
if (PetscRealPart(array[j+i*step]) > max) max = PetscRealPart(array[j+i*step]);
}
if (min + 1.e-10 > max) {
min -= 1.e-5;
max += 1.e-5;
}
if (j < nbounds) {
min = PetscMin(min,bounds[2*j]);
max = PetscMax(max,bounds[2*j+1]);
}
ierr = MPI_Reduce(&min,&ymin,1,MPIU_REAL,MPIU_MIN,0,comm);CHKERRQ(ierr);
ierr = MPI_Reduce(&max,&ymax,1,MPIU_REAL,MPIU_MAX,0,comm);CHKERRQ(ierr);
ierr = PetscViewerDrawGetHold(v,&hold);CHKERRQ(ierr);
if (!hold) {
ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
}
ierr = PetscViewerDrawGetDrawAxis(v,k,&axis);CHKERRQ(ierr);
ierr = PetscLogObjectParent((PetscObject)draw,(PetscObject)axis);CHKERRQ(ierr);
if (!rank) {
const char *title;
ierr = PetscDrawAxisSetLimits(axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
ierr = PetscDrawAxisDraw(axis);CHKERRQ(ierr);
ierr = PetscDrawGetCoordinates(draw,coors,coors+1,coors+2,coors+3);CHKERRQ(ierr);
ierr = DMDAGetFieldName(da,j,&title);CHKERRQ(ierr);
if (title) {ierr = PetscDrawSetTitle(draw,title);CHKERRQ(ierr);}
}
ierr = MPI_Bcast(coors,4,MPIU_REAL,0,comm);CHKERRQ(ierr);
if (rank) {
ierr = PetscDrawSetCoordinates(draw,coors[0],coors[1],coors[2],coors[3]);CHKERRQ(ierr);
}
/* draw local part of vector */
ierr = PetscObjectGetNewTag((PetscObject)xin,&tag1);CHKERRQ(ierr);
ierr = PetscObjectGetNewTag((PetscObject)xin,&tag2);CHKERRQ(ierr);
if (rank < size-1) { /*send value to right */
ierr = MPI_Send((void*)&array[j+(n-1)*step],1,MPIU_REAL,rank+1,tag1,comm);CHKERRQ(ierr);
ierr = MPI_Send((void*)&xg[n-1],1,MPIU_REAL,rank+1,tag1,comm);CHKERRQ(ierr);
}
if (!rank && bx == DM_BOUNDARY_PERIODIC && size > 1) { /* first processor sends first value to last */
ierr = MPI_Send((void*)&array[j],1,MPIU_REAL,size-1,tag2,comm);CHKERRQ(ierr);
}
for (i=1; i<n; i++) {
ierr = PetscDrawLine(draw,PetscRealPart(xg[i-1]),PetscRealPart(array[j+step*(i-1)]),PetscRealPart(xg[i]),PetscRealPart(array[j+step*i]),PETSC_DRAW_RED);CHKERRQ(ierr);
if (showpoints) {
ierr = PetscDrawPoint(draw,PetscRealPart(xg[i-1]),PetscRealPart(array[j+step*(i-1)]),PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
}
if (rank) { /* receive value from left */
ierr = MPI_Recv(&tmp,1,MPIU_REAL,rank-1,tag1,comm,&status);CHKERRQ(ierr);
ierr = MPI_Recv(&xgtmp,1,MPIU_REAL,rank-1,tag1,comm,&status);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xgtmp,tmp,PetscRealPart(xg[0]),PetscRealPart(array[j]),PETSC_DRAW_RED);CHKERRQ(ierr);
if (showpoints) {
ierr = PetscDrawPoint(draw,xgtmp,tmp,PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
}
if (rank == size-1 && bx == DM_BOUNDARY_PERIODIC && size > 1) {
ierr = MPI_Recv(&tmp,1,MPIU_REAL,0,tag2,comm,&status);CHKERRQ(ierr);
/* If the mesh is not uniform we do not know the mesh spacing between the last point on the right and the first ghost point */
ierr = PetscDrawLine(draw,PetscRealPart(xg[n-1]),PetscRealPart(array[j+step*(n-1)]),PetscRealPart(xg[n-1]+(xg[n-1]-xg[n-2])),tmp,PETSC_DRAW_RED);CHKERRQ(ierr);
if (showpoints) {
ierr = PetscDrawPoint(draw,PetscRealPart(xg[n-2]),PetscRealPart(array[j+step*(n-1)]),PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
}
ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
}
ierr = PetscFree(displayfields);CHKERRQ(ierr);
ierr = VecRestoreArrayRead(xcoor,&xg);CHKERRQ(ierr);
ierr = VecRestoreArrayRead(xin,&array);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode DMView_DA_2d(DM da,PetscViewer viewer)
{
PetscErrorCode ierr;
PetscMPIInt rank;
PetscBool iascii,isdraw,isbinary;
DM_DA *dd = (DM_DA*)da->data;
#if defined(PETSC_HAVE_MATLAB_ENGINE)
PetscBool ismatlab;
#endif
PetscFunctionBegin;
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);
CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);
CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);
CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);
CHKERRQ(ierr);
#if defined(PETSC_HAVE_MATLAB_ENGINE)
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERMATLAB,&ismatlab);
CHKERRQ(ierr);
#endif
if (iascii) {
PetscViewerFormat format;
ierr = PetscViewerGetFormat(viewer, &format);
CHKERRQ(ierr);
if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL) {
DMDALocalInfo info;
ierr = DMDAGetLocalInfo(da,&info);
CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);
CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Processor [%d] M %D N %D m %D n %D w %D s %D\n",rank,dd->M,dd->N,dd->m,dd->n,dd->w,dd->s);
CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D, Y range of indices: %D %D\n",info.xs,info.xs+info.xm,info.ys,info.ys+info.ym);
CHKERRQ(ierr);
ierr = PetscViewerFlush(viewer);
CHKERRQ(ierr);
ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);
CHKERRQ(ierr);
} else {
ierr = DMView_DA_VTK(da,viewer);
CHKERRQ(ierr);
}
} else if (isdraw) {
PetscDraw draw;
double ymin = -1*dd->s-1,ymax = dd->N+dd->s;
double xmin = -1*dd->s-1,xmax = dd->M+dd->s;
double x,y;
PetscInt base,*idx;
char node[10];
PetscBool isnull;
ierr = PetscViewerDrawGetDraw(viewer,0,&draw);
CHKERRQ(ierr);
ierr = PetscDrawIsNull(draw,&isnull);
CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
if (!da->coordinates) {
ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);
CHKERRQ(ierr);
}
ierr = PetscDrawSynchronizedClear(draw);
CHKERRQ(ierr);
/* first processor draw all node lines */
if (!rank) {
ymin = 0.0;
ymax = dd->N - 1;
for (xmin=0; xmin<dd->M; xmin++) {
ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_BLACK);
CHKERRQ(ierr);
}
xmin = 0.0;
xmax = dd->M - 1;
for (ymin=0; ymin<dd->N; ymin++) {
ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);
CHKERRQ(ierr);
}
}
ierr = PetscDrawSynchronizedFlush(draw);
CHKERRQ(ierr);
ierr = PetscDrawPause(draw);
CHKERRQ(ierr);
/* draw my box */
ymin = dd->ys;
ymax = dd->ye - 1;
xmin = dd->xs/dd->w;
xmax =(dd->xe-1)/dd->w;
ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);
CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);
CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);
CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);
CHKERRQ(ierr);
/* put in numbers */
base = (dd->base)/dd->w;
for (y=ymin; y<=ymax; y++) {
for (x=xmin; x<=xmax; x++) {
sprintf(node,"%d",(int)base++);
ierr = PetscDrawString(draw,x,y,PETSC_DRAW_BLACK,node);
CHKERRQ(ierr);
}
}
ierr = PetscDrawSynchronizedFlush(draw);
CHKERRQ(ierr);
ierr = PetscDrawPause(draw);
CHKERRQ(ierr);
/* overlay ghost numbers, useful for error checking */
/* put in numbers */
base = 0;
idx = dd->idx;
ymin = dd->Ys;
ymax = dd->Ye;
xmin = dd->Xs;
xmax = dd->Xe;
for (y=ymin; y<ymax; y++) {
for (x=xmin; x<xmax; x++) {
if ((base % dd->w) == 0) {
sprintf(node,"%d",(int)(idx[base]/dd->w));
ierr = PetscDrawString(draw,x/dd->w,y,PETSC_DRAW_BLUE,node);
CHKERRQ(ierr);
}
base++;
}
}
ierr = PetscDrawSynchronizedFlush(draw);
CHKERRQ(ierr);
ierr = PetscDrawPause(draw);
CHKERRQ(ierr);
} else if (isbinary) {
ierr = DMView_DA_Binary(da,viewer);
CHKERRQ(ierr);
#if defined(PETSC_HAVE_MATLAB_ENGINE)
} else if (ismatlab) {
ierr = DMView_DA_Matlab(da,viewer);
CHKERRQ(ierr);
#endif
}
PetscFunctionReturn(0);
}
/*@
PetscDrawAxisDraw - PetscDraws an axis.
Not Collective (ignored on all processors except processor 0 of PetscDrawAxis)
Input Parameter:
. axis - Axis structure
Level: advanced
Note:
This draws the actual axis. The limits etc have already been set.
By picking special routines for the ticks and labels, special
effects may be generated. These routines are part of the Axis
structure (axis).
@*/
PetscErrorCode PetscDrawAxisDraw(PetscDrawAxis axis)
{
int i,ntick,numx,numy,ac = axis->ac,tc = axis->tc,cc = axis->cc,rank;
size_t len;
PetscReal tickloc[MAXSEGS],sep,h,w,tw,th,xl,xr,yl,yr;
char *p;
PetscDraw draw = axis->win;
PetscErrorCode ierr;
PetscFunctionBegin;
if (!axis) PetscFunctionReturn(0);
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)axis),&rank);CHKERRQ(ierr);
if (rank) PetscFunctionReturn(0);
if (axis->xlow == axis->xhigh) {axis->xlow -= .5; axis->xhigh += .5;}
/* if ((axis->yhigh - axis->ylow) <= 1.e-5*PetscMax(PetscAbsReal(axis->yhigh),PetscAbsReal(axis->ylow))) {
axis->ylow -= 1.e-5*PetscMax(PetscAbsReal(axis->yhigh),PetscAbsReal(axis->ylow));
axis->yhigh += 1.e-5*PetscMax(PetscAbsReal(axis->yhigh),PetscAbsReal(axis->ylow));
} */
if (axis->ylow == axis->yhigh) {axis->ylow -= .5; axis->yhigh += .5;}
xl = axis->xlow; xr = axis->xhigh; yl = axis->ylow; yr = axis->yhigh;
ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
numx = (int)(.15*(xr-xl)/tw); if (numx > 6) numx = 6;if (numx< 2) numx = 2;
numy = (int)(.50*(yr-yl)/th); if (numy > 6) numy = 6;if (numy< 2) numy = 2;
xl -= 11*tw; xr += 2*tw; yl -= 2.5*th; yr += 2*th;
if (axis->xlabel) yl -= 2*th;
if (axis->ylabel) xl -= 2*tw;
ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,axis->xlow,axis->ylow,axis->xhigh,axis->ylow,ac);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,axis->xlow,axis->ylow,axis->xlow,axis->yhigh,ac);CHKERRQ(ierr);
if (axis->toplabel) {
ierr = PetscStrlen(axis->toplabel,&len);CHKERRQ(ierr);
w = xl + .5*(xr - xl) - .5*len*tw;
h = axis->yhigh;
ierr = PetscDrawString(draw,w,h,cc,axis->toplabel);CHKERRQ(ierr);
}
/* PetscDraw the ticks and labels */
if (axis->xticks) {
ierr = (*axis->xticks)(axis->xlow,axis->xhigh,numx,&ntick,tickloc,MAXSEGS);CHKERRQ(ierr);
/* PetscDraw in tick marks */
for (i=0; i<ntick; i++) {
ierr = PetscDrawLine(draw,tickloc[i],axis->ylow-.5*th,tickloc[i],axis->ylow+.5*th,tc);CHKERRQ(ierr);
}
/* label ticks */
for (i=0; i<ntick; i++) {
if (axis->xlabelstr) {
if (i < ntick - 1) sep = tickloc[i+1] - tickloc[i];
else if (i > 0) sep = tickloc[i] - tickloc[i-1];
else sep = 0.0;
ierr = (*axis->xlabelstr)(tickloc[i],sep,&p);CHKERRQ(ierr);
ierr = PetscStrlen(p,&len);CHKERRQ(ierr);
w = .5*len*tw;
ierr = PetscDrawString(draw,tickloc[i]-w,axis->ylow-1.2*th,cc,p);CHKERRQ(ierr);
}
}
}
if (axis->xlabel) {
ierr = PetscStrlen(axis->xlabel,&len);CHKERRQ(ierr);
w = xl + .5*(xr - xl) - .5*len*tw;
h = axis->ylow - 2.5*th;
ierr = PetscDrawString(draw,w,h,cc,axis->xlabel);CHKERRQ(ierr);
}
if (axis->yticks) {
ierr = (*axis->yticks)(axis->ylow,axis->yhigh,numy,&ntick,tickloc,MAXSEGS);CHKERRQ(ierr);
/* PetscDraw in tick marks */
for (i=0; i<ntick; i++) {
ierr = PetscDrawLine(draw,axis->xlow -.5*tw,tickloc[i],axis->xlow+.5*tw,tickloc[i],tc);CHKERRQ(ierr);
}
/* label ticks */
for (i=0; i<ntick; i++) {
if (axis->ylabelstr) {
if (i < ntick - 1) sep = tickloc[i+1] - tickloc[i];
else if (i > 0) sep = tickloc[i] - tickloc[i-1];
else sep = 0.0;
ierr = (*axis->xlabelstr)(tickloc[i],sep,&p);CHKERRQ(ierr);
ierr = PetscStrlen(p,&len);CHKERRQ(ierr);
w = axis->xlow - len * tw - 1.2*tw;
ierr = PetscDrawString(draw,w,tickloc[i]-.5*th,cc,p);CHKERRQ(ierr);
}
}
}
if (axis->ylabel) {
ierr = PetscStrlen(axis->ylabel,&len);CHKERRQ(ierr);
h = yl + .5*(yr - yl) + .5*len*th;
w = xl + 1.5*tw;
ierr = PetscDrawStringVertical(draw,w,h,cc,axis->ylabel);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscErrorCode ierr;
int i,j,w,h;
int k = PETSC_DRAW_BLACK;
int r = PETSC_DRAW_RED;
int g = PETSC_DRAW_GREEN;
int b = PETSC_DRAW_BLUE;
int y = PETSC_DRAW_YELLOW;
int c0 = PETSC_DRAW_BASIC_COLORS;
int c2 = 255;
int c1 = (c0+c2)/2;
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Draw Example",PETSC_DECIDE,PETSC_DECIDE,101,101,&draw);CHKERRQ(ierr);
/*ierr = PetscDrawSetPause(draw,2.0);CHKERRQ(ierr);*/
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawGetWindowSize(draw,&w,&h);CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(draw,0,0,--w,--h);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/* one-pixel lines in the window corners */
ierr = PetscDrawLine(draw,0,0,0,0,r);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,w,0,w,0,r);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0,h,0,h,r);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,w,h,w,h,r);CHKERRQ(ierr);
/* border lines with two pixels from borders */
ierr = PetscDrawLine(draw,0+2,0,w-2,0,k);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0+2,h,w-2,h,k);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0,0+2,0,h-2,k);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,w,0+2,w,h-2,k);CHKERRQ(ierr);
/* oblique lines */
ierr = PetscDrawLine(draw,0+2,h/2,w-2,h-2,b);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0+1,h-1,w-1,0+1,b);CHKERRQ(ierr);
/* vertical up and down arrow, two pixels from borders */
ierr = PetscDrawArrow(draw,1*w/4,0+2,1*w/4,h-2,g);CHKERRQ(ierr);
ierr = PetscDrawArrow(draw,3*w/4,h-2,3*w/4,0+2,g);CHKERRQ(ierr);
/* horizontal right and left arrow, two pixels from borders */
ierr = PetscDrawArrow(draw,0+2,3*h/4,w-2,3*h/4,g);CHKERRQ(ierr);
ierr = PetscDrawArrow(draw,w-2,1*h/4,0+2,1*h/4,g);CHKERRQ(ierr);
/* flush, save, and pause */
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawGetWindowSize(draw,&w,&h);CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(draw,0,0,--w,--h);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/* one-pixel rectangles in the window corners */
ierr = PetscDrawRectangle(draw,0,0,0,0,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,w,0,w,0,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,0,h,0,h,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,w,h,w,h,k,k,k,k);CHKERRQ(ierr);
/* border rectangles with two pixels from borders */
ierr = PetscDrawRectangle(draw,0+2,0,w-2,0,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,0+2,h,w-2,h,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,0,0+2,0,h-2,k,k,k,k);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,w,0+2,w,h-2,k,k,k,k);CHKERRQ(ierr);
/* more rectangles */
ierr = PetscDrawRectangle(draw,0+2,0+2,w/2-1,h/2-1,b,b,b,b);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,0+2,h/2+1,w/2-1,h-2,r,r,r,r);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,w/2+1,h/2+1,w-2,h-2,g,g,g,g);CHKERRQ(ierr);
ierr = PetscDrawRectangle(draw,w/2+1,0+2,w-2,h/2-1,y,y,y,y);CHKERRQ(ierr);
/* flush, save, and pause */
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawGetWindowSize(draw,&w,&h);CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(draw,0,0,--w,--h);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/* interpolated triangles, one pixel from borders */
ierr = PetscDrawTriangle(draw,0+1,0+1,w-1,0+1,w-1,h-1,c0,c1,c2);CHKERRQ(ierr);
ierr = PetscDrawTriangle(draw,0+1,0+1,0+1,h-1,w-1,h-1,c0,c1,c2);CHKERRQ(ierr);
/* interpolated triangle, oblique, inside canvas */
ierr = PetscDrawTriangle(draw,w/4,h/4,w/2,3*h/4,3*w/4,h/2,c2,c1,c0);CHKERRQ(ierr);
/* flush, save, and pause */
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawGetWindowSize(draw,&w,&h);CHKERRQ(ierr);
ierr = PetscDrawSetCoordinates(draw,0,0,--w,--h);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/* circles and ellipses */
ierr = PetscDrawEllipse(draw,w/2,h/2,w-1,h-1,r);CHKERRQ(ierr);
ierr = PetscDrawEllipse(draw,w,h/2,w/2,h,g);CHKERRQ(ierr);
ierr = PetscDrawEllipse(draw,0,0,w,h/2,b);CHKERRQ(ierr);
ierr = PetscDrawEllipse(draw,w/4,3*h/4,w/2,h/4,y);CHKERRQ(ierr);
ierr = PetscDrawCoordinateToPixel(draw,w/2,h/2,&i,&j);CHKERRQ(ierr);
ierr = PetscDrawPointPixel(draw,i,j,k);CHKERRQ(ierr);
/* flush, save, and pause */
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return 0;
}
void PETSC_STDCALL petscdrawsetcoordinates_(PetscDraw draw,PetscReal *xl,PetscReal *yl,PetscReal *xr,PetscReal *yr, int *__ierr ){
*__ierr = PetscDrawSetCoordinates(
(PetscDraw)PetscToPointer((draw) ),*xl,*yl,*xr,*yr);
}
