/*@
PetscDrawScalePopup - PetscDraws 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
@*/
PetscErrorCode PetscDrawScalePopup(PetscDraw popup,PetscReal min,PetscReal max)
{
PetscReal xl = 0.0,yl = 0.0,xr = 2.0,yr = 1.0,value;
PetscErrorCode ierr;
int i,c = PETSC_DRAW_BASIC_COLORS,rank;
char string[32];
MPI_Comm comm;
PetscFunctionBegin;
ierr = PetscDrawCheckResizedWindow(popup);CHKERRQ(ierr);
ierr = PetscObjectGetComm((PetscObject)popup,&comm);CHKERRQ(ierr);
ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
if (rank) PetscFunctionReturn(0);
for (i=0; i<10; i++) {
ierr = PetscDrawRectangle(popup,xl,yl,xr,yr,c,c,c,c);CHKERRQ(ierr);
yl += .1; yr += .1; c = (int)((double)c + (245. - PETSC_DRAW_BASIC_COLORS)/9.);
}
for (i=0; i<10; i++) {
value = min + i*(max-min)/9.0;
/* 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;
sprintf(string,"%18.16e",(double)value);
ierr = PetscDrawString(popup,.2,.02 + i/10.0,PETSC_DRAW_BLACK,string);CHKERRQ(ierr);
}
ierr = PetscDrawSetTitle(popup,"Contour Scale");CHKERRQ(ierr);
ierr = PetscDrawFlush(popup);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 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 KSPMonitorLGRange(KSP ksp,PetscInt n,PetscReal rnorm,void *monctx)
{
PetscDrawLG lg;
PetscErrorCode ierr;
PetscReal x,y,per;
PetscViewer v = (PetscViewer)monctx;
static PetscReal prev; /* should be in the context */
PetscDraw draw;
PetscFunctionBegin;
ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
x = (PetscReal) n;
if (rnorm > 0.0) y = PetscLog10Real(rnorm);
else y = -15.0;
ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
if (n < 20 || !(n % 5)) {
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
}
ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
ierr = KSPMonitorRange_Private(ksp,n,&per);CHKERRQ(ierr);
if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
x = (PetscReal) n;
y = 100.0*per;
ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
if (n < 20 || !(n % 5)) {
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
}
ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
x = (PetscReal) n;
y = (prev - rnorm)/prev;
ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
if (n < 20 || !(n % 5)) {
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
}
ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
x = (PetscReal) n;
y = (prev - rnorm)/(prev*per);
if (n > 5) {
ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
}
if (n < 20 || !(n % 5)) {
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
}
prev = rnorm;
PetscFunctionReturn(0);
}
