/*@C
PetscDrawBoxedString - Draws a string with a box around it
Not Collective
Input Parameters:
+ draw - the drawing context
. sxl - the coordinates of center of the box
. syl - the coordinates of top line of box
. sc - the color of the text
. bc - the color of the bounding box
- text - the text to draw
Output Parameter:
. w,h - width and height of resulting box (optional)
Level: beginner
Concepts: drawing^string
Concepts: string^drawing
.seealso: PetscDrawStringVertical(), PetscDrawBoxedStringSize()
@*/
PetscErrorCode PetscDrawBoxedString(PetscDraw draw,PetscReal sxl,PetscReal syl,int sc,int bc,const char text[],PetscReal *w,PetscReal *h)
{
PetscErrorCode ierr ;
PetscBool isnull;
PetscReal top,left,right,bottom,tw,th;
size_t len,mlen = 0;
char **array;
int cnt,i;
PetscFunctionBegin;
PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
PetscValidCharPointer(text,5);
ierr = PetscObjectTypeCompare((PetscObject)draw,PETSC_DRAW_NULL,&isnull);CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
if (draw->ops->boxedstring) {
ierr = (*draw->ops->boxedstring)(draw,sxl,syl,sc,bc,text,w,h);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
ierr = PetscStrToArray(text,'\n',&cnt,&array);CHKERRQ(ierr);
for (i=0; i<cnt; i++) {
ierr = PetscStrlen(array[i],&len);CHKERRQ(ierr);
mlen = PetscMax(mlen,len);
}
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
top = syl;
left = sxl - .5*(mlen + 2)*tw;
right = sxl + .5*(mlen + 2)*tw;
bottom = syl - (1.0 + cnt)*th;
if (w) *w = right - left;
if (h) *h = top - bottom;
/* compute new bounding box */
draw->boundbox_xl = PetscMin(draw->boundbox_xl,left);
draw->boundbox_xr = PetscMax(draw->boundbox_xr,right);
draw->boundbox_yl = PetscMin(draw->boundbox_yl,bottom);
draw->boundbox_yr = PetscMax(draw->boundbox_yr,top);
/* top, left, bottom, right lines */
ierr = PetscDrawLine(draw,left,top,right,top,bc);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,left,bottom,left,top,bc);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,right,bottom,right,top,bc);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,left,bottom,right,bottom,bc);CHKERRQ(ierr);
for (i=0; i<cnt; i++) {
ierr = PetscDrawString(draw,left + tw,top - (1.5 + i)*th,sc,array[i]);CHKERRQ(ierr);
}
ierr = PetscStrToArrayDestroy(cnt,array);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
PetscDrawLGDraw - Redraws a line graph.
Not Collective,but ignored by all processors except processor 0 in PetscDrawLG
Input Parameter:
. lg - the line graph context
Level: intermediate
.seealso: PetscDrawSPDraw(), PetscDrawLGSPDraw()
@*/
PetscErrorCode PetscDrawLGDraw(PetscDrawLG lg)
{
PetscReal xmin=lg->xmin,xmax=lg->xmax,ymin=lg->ymin,ymax=lg->ymax;
PetscErrorCode ierr;
int i,j,dim = lg->dim,nopts = lg->nopts,rank,cl;
PetscDraw draw = lg->win;
PetscFunctionBegin;
if (lg && ((PetscObject)lg)->classid == PETSC_DRAW_CLASSID) PetscFunctionReturn(0);
PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr);
ierr = MPI_Comm_rank(((PetscObject)lg)->comm,&rank);CHKERRQ(ierr);
if (!rank) {
for (i=0; i<dim; i++) {
for (j=1; j<nopts; j++) {
if (lg->colors) cl = lg->colors[i];
else cl = PETSC_DRAW_BLACK+i;
ierr = PetscDrawLine(draw,lg->x[(j-1)*dim+i],lg->y[(j-1)*dim+i],lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);
if (lg->use_dots) {
ierr = PetscDrawString(draw,lg->x[j*dim+i],lg->y[j*dim+i],PETSC_DRAW_RED,"x");CHKERRQ(ierr);
}
}
}
}
if (lg->legend) {
PetscReal xl,yl,xr,yr,tw,th;
size_t len,mlen = 0;
int cl;
ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
ierr = PetscStrlen(lg->legend[i],&len);CHKERRQ(ierr);
mlen = PetscMax(mlen,len);
}
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - 3*th,xr - 2*tw,yr - 3*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - 3*th,xr - (mlen + 8)*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
cl = (lg->colors ? lg->colors[i] : i + 1);
ierr = PetscDrawLine(draw,xr - (mlen + 6.7)*tw,yr - (4 + i)*th,xr - (mlen + 3.2)*tw,yr - (4 + i)*th,cl);CHKERRQ(ierr);
ierr = PetscDrawString(draw,xr - (mlen + 3)*tw,yr - (4.5 + i)*th,PETSC_DRAW_BLACK,lg->legend[i]);CHKERRQ(ierr);
}
ierr = PetscDrawLine(draw,xr - 2*tw,yr - 3*th,xr - 2*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - (4+lg->dim)*th,xr - 2*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
ierr = PetscDrawFlush(lg->win);CHKERRQ(ierr);
ierr = PetscDrawPause(lg->win);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode PetscDrawStringVertical_TikZ(PetscDraw draw,PetscReal xl,PetscReal yl,int cl,const char text[])
{
PetscDraw_TikZ *win = (PetscDraw_TikZ*)draw->data;
PetscErrorCode ierr;
size_t len;
PetscReal width;
PetscFunctionBegin;
win->written = PETSC_TRUE;
ierr = PetscStrlen(text,&len);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&width,NULL);CHKERRQ(ierr);
yl = yl - len*width*(draw->coor_yr - draw->coor_yl)/(draw->coor_xr - draw->coor_xl);
ierr = PetscFPrintf(PetscObjectComm((PetscObject)draw),win->fd,"\\node [rotate=90, %s] at (%g,%g) {%s};\n",TikZColorMap(cl),XTRANS(draw,xl),YTRANS(draw,yl),text);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode PCView_MG(PC pc,PetscViewer viewer)
{
PC_MG *mg = (PC_MG*)pc->data;
PC_MG_Levels **mglevels = mg->levels;
PetscErrorCode ierr;
PetscInt levels = mglevels ? mglevels[0]->levels : 0,i;
PetscBool iascii,isbinary,isdraw;
PetscFunctionBegin;
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
if (iascii) {
const char *cyclename = levels ? (mglevels[0]->cycles == PC_MG_CYCLE_V ? "v" : "w") : "unknown";
ierr = PetscViewerASCIIPrintf(viewer," MG: type is %s, levels=%D cycles=%s\n", PCMGTypes[mg->am],levels,cyclename);CHKERRQ(ierr);
if (mg->am == PC_MG_MULTIPLICATIVE) {
ierr = PetscViewerASCIIPrintf(viewer," Cycles per PCApply=%d\n",mg->cyclesperpcapply);CHKERRQ(ierr);
}
if (mg->galerkin) {
ierr = PetscViewerASCIIPrintf(viewer," Using Galerkin computed coarse grid matrices\n");CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer," Not using Galerkin computed coarse grid matrices\n");CHKERRQ(ierr);
}
if (mg->view){
ierr = (*mg->view)(pc,viewer);CHKERRQ(ierr);
}
for (i=0; i<levels; i++) {
if (!i) {
ierr = PetscViewerASCIIPrintf(viewer,"Coarse grid solver -- level -------------------------------\n",i);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer,"Down solver (pre-smoother) on level %D -------------------------------\n",i);CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
ierr = KSPView(mglevels[i]->smoothd,viewer);CHKERRQ(ierr);
ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
if (i && mglevels[i]->smoothd == mglevels[i]->smoothu) {
ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) same as down solver (pre-smoother)\n");CHKERRQ(ierr);
} else if (i) {
ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) on level %D -------------------------------\n",i);CHKERRQ(ierr);
ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
ierr = KSPView(mglevels[i]->smoothu,viewer);CHKERRQ(ierr);
ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
}
}
} else if (isbinary) {
for (i=levels-1; i>=0; i--) {
ierr = KSPView(mglevels[i]->smoothd,viewer);CHKERRQ(ierr);
if (i && mglevels[i]->smoothd != mglevels[i]->smoothu) {
ierr = KSPView(mglevels[i]->smoothu,viewer);CHKERRQ(ierr);
}
}
} else if (isdraw) {
PetscDraw draw;
PetscReal x,w,y,bottom,th;
ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,NULL,&th);CHKERRQ(ierr);
bottom = y - th;
for (i=levels-1; i>=0; i--) {
if (!mglevels[i]->smoothu || (mglevels[i]->smoothu == mglevels[i]->smoothd)) {
ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr);
ierr = KSPView(mglevels[i]->smoothd,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
} else {
w = 0.5*PetscMin(1.0-x,x);
ierr = PetscDrawPushCurrentPoint(draw,x+w,bottom);CHKERRQ(ierr);
ierr = KSPView(mglevels[i]->smoothd,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
ierr = PetscDrawPushCurrentPoint(draw,x-w,bottom);CHKERRQ(ierr);
ierr = KSPView(mglevels[i]->smoothu,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
}
ierr = PetscDrawGetBoundingBox(draw,NULL,&bottom,NULL,NULL);CHKERRQ(ierr);
bottom -= th;
}
}
PetscFunctionReturn(0);
}
PetscErrorCode SNESView_FAS(SNES snes, PetscViewer viewer)
{
SNES_FAS *fas = (SNES_FAS*) snes->data;
PetscBool isFine,iascii,isdraw;
PetscInt i;
PetscErrorCode ierr;
SNES smoothu, smoothd, levelsnes;
PetscFunctionBegin;
ierr = SNESFASCycleIsFine(snes, &isFine);CHKERRQ(ierr);
if (isFine) {
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
if (iascii) {
ierr = PetscViewerASCIIPrintf(viewer, "FAS: type is %s, levels=%D, cycles=%D\n", SNESFASTypes[fas->fastype], fas->levels, fas->n_cycles);CHKERRQ(ierr);
if (fas->galerkin) {
ierr = PetscViewerASCIIPrintf(viewer," Using Galerkin computed coarse grid function evaluation\n");CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer," Not using Galerkin computed coarse grid function evaluation\n");CHKERRQ(ierr);
}
for (i=0; i<fas->levels; i++) {
ierr = SNESFASGetCycleSNES(snes, i, &levelsnes);CHKERRQ(ierr);
ierr = SNESFASCycleGetSmootherUp(levelsnes, &smoothu);CHKERRQ(ierr);
ierr = SNESFASCycleGetSmootherDown(levelsnes, &smoothd);CHKERRQ(ierr);
if (!i) {
ierr = PetscViewerASCIIPrintf(viewer,"Coarse grid solver -- level %D -------------------------------\n",i);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer,"Down solver (pre-smoother) on level %D -------------------------------\n",i);CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
if (smoothd) {
ierr = SNESView(smoothd,viewer);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer,"Not yet available\n");CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
if (i && (smoothd == smoothu)) {
ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) same as down solver (pre-smoother)\n");CHKERRQ(ierr);
} else if (i) {
ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) on level %D -------------------------------\n",i);CHKERRQ(ierr);
ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
if (smoothu) {
ierr = SNESView(smoothu,viewer);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer,"Not yet available\n");CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
}
}
} else if (isdraw) {
PetscDraw draw;
PetscReal x,w,y,bottom,th,wth;
SNES_FAS *curfas = fas;
ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&wth,&th);CHKERRQ(ierr);
bottom = y - th;
while (curfas) {
if (!curfas->smoothu) {
ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr);
if (curfas->smoothd) ierr = SNESView(curfas->smoothd,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
} else {
w = 0.5*PetscMin(1.0-x,x);
ierr = PetscDrawPushCurrentPoint(draw,x-w,bottom);CHKERRQ(ierr);
if (curfas->smoothd) ierr = SNESView(curfas->smoothd,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
ierr = PetscDrawPushCurrentPoint(draw,x+w,bottom);CHKERRQ(ierr);
if (curfas->smoothu) ierr = SNESView(curfas->smoothu,viewer);CHKERRQ(ierr);
ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
}
/* this is totally bogus but we have no way of knowing how low the previous one was draw to */
bottom -= 5*th;
if (curfas->next) curfas = (SNES_FAS*)curfas->next->data;
else curfas = NULL;
}
}
}
PetscFunctionReturn(0);
}
/*@
PetscDrawBarDraw - Redraws a bar graph.
Collective on PetscDrawBar
Input Parameter:
. bar - The bar graph context
Level: intermediate
.seealso: PetscDrawBar, PetscDrawBarCreate(), PetscDrawBarSetData()
@*/
PetscErrorCode PetscDrawBarDraw(PetscDrawBar bar)
{
PetscDraw draw;
PetscBool isnull;
PetscReal xmin,xmax,ymin,ymax,*values,binLeft,binRight;
PetscInt numValues,i,bcolor,color,idx,*perm,nplot;
PetscMPIInt rank;
PetscErrorCode ierr;
char **labels;
PetscFunctionBegin;
PetscValidHeaderSpecific(bar,PETSC_DRAWBAR_CLASSID,1);
ierr = PetscDrawIsNull(bar->win,&isnull);CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)bar),&rank);CHKERRQ(ierr);
if (bar->numBins < 1) PetscFunctionReturn(0);
color = bar->color;
if (color == PETSC_DRAW_ROTATE) bcolor = PETSC_DRAW_BLACK+1;
else bcolor = color;
numValues = bar->numBins;
values = bar->values;
if (bar->ymin == bar->ymax) {
/* user has not set bounds on bars so set them based on the data */
ymin = PETSC_MAX_REAL;
ymax = PETSC_MIN_REAL;
for (i=0; i<numValues; i++) {
ymin = PetscMin(ymin,values[i]);
ymax = PetscMax(ymax,values[i]);
}
} else {
ymin = bar->ymin;
ymax = bar->ymax;
}
nplot = numValues; /* number of points to actually plot; if some are lower than requested tolerance */
xmin = 0.0;
xmax = nplot;
labels = bar->labels;
if (bar->sort) {
ierr = PetscMalloc1(numValues,&perm);CHKERRQ(ierr);
for (i=0; i<numValues;i++) perm[i] = i;
ierr = PetscSortRealWithPermutation(numValues,values,perm);CHKERRQ(ierr);
if (bar->sorttolerance) {
for (i=0; i<numValues;i++) {
if (values[perm[numValues - i - 1]] < bar->sorttolerance) {
nplot = i;
break;
}
}
}
}
draw = bar->win;
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLimits(bar->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
ierr = PetscDrawAxisDraw(bar->axis);CHKERRQ(ierr);
ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
if (!rank) { /* Draw bins */
for (i=0; i<nplot; i++) {
idx = (bar->sort ? perm[numValues - i - 1] : i);
binLeft = xmin + i;
binRight = xmin + i + 1;
ierr = PetscDrawRectangle(draw,binLeft,ymin,binRight,values[idx],bcolor,bcolor,bcolor,bcolor);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,binLeft,ymin,binLeft,values[idx],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,binRight,ymin,binRight,values[idx],PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,binLeft,values[idx],binRight,values[idx],PETSC_DRAW_BLACK);CHKERRQ(ierr);
if (labels) {
PetscReal h;
ierr = PetscDrawStringGetSize(draw,NULL,&h);CHKERRQ(ierr);
ierr = PetscDrawStringCentered(draw,.5*(binLeft+binRight),ymin - 1.5*h,bcolor,labels[idx]);CHKERRQ(ierr);
}
if (color == PETSC_DRAW_ROTATE) bcolor++;
if (bcolor > PETSC_DRAW_BASIC_COLORS-1) bcolor = PETSC_DRAW_BLACK+1;
}
}
ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
if (bar->sort) {ierr = PetscFree(perm);CHKERRQ(ierr);}
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
PetscDrawLGDraw - Redraws a line graph.
Not Collective,but ignored by all processors except processor 0 in PetscDrawLG
Input Parameter:
. lg - the line graph context
Level: intermediate
.seealso: PetscDrawSPDraw(), PetscDrawLGSPDraw()
@*/
PetscErrorCode PetscDrawLGDraw(PetscDrawLG lg)
{
PetscReal xmin=lg->xmin,xmax=lg->xmax,ymin=lg->ymin,ymax=lg->ymax;
PetscErrorCode ierr;
int i,j,dim = lg->dim,nopts = lg->nopts,rank,cl;
PetscDraw draw = lg->win;
PetscBool isnull;
PetscFunctionBegin;
if (lg && ((PetscObject)lg)->classid == PETSC_DRAW_CLASSID) PetscFunctionReturn(0);
PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
#if defined(PETSC_HAVE_SETJMP_H) && defined(PETSC_HAVE_X)
if (!setjmp(PetscXIOErrorJumpBuf)) XSetIOErrorHandler((XIOErrorHandler)PetscXIOHandler);
else {
XSetIOErrorHandler(PETSC_NULL);
ierr = PetscDrawSetType(draw,PETSC_DRAW_NULL);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
#endif
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr);
ierr = MPI_Comm_rank(((PetscObject)lg)->comm,&rank);CHKERRQ(ierr);
if (!rank) {
for (i=0; i<dim; i++) {
for (j=1; j<nopts; j++) {
if (lg->colors) cl = lg->colors[i];
else cl = PETSC_DRAW_BLACK+i;
ierr = PetscDrawLine(draw,lg->x[(j-1)*dim+i],lg->y[(j-1)*dim+i],lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);
if (lg->use_dots) {
ierr = PetscDrawString(draw,lg->x[j*dim+i],lg->y[j*dim+i],cl,"x");CHKERRQ(ierr);
}
}
}
}
if (lg->legend) {
PetscReal xl,yl,xr,yr,tw,th;
size_t len,mlen = 0;
int cl;
ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
ierr = PetscStrlen(lg->legend[i],&len);CHKERRQ(ierr);
mlen = PetscMax(mlen,len);
}
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - 3*th,xr - 2*tw,yr - 3*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - 3*th,xr - (mlen + 8)*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
cl = (lg->colors ? lg->colors[i] : i + 1);
ierr = PetscDrawLine(draw,xr - (mlen + 6.7)*tw,yr - (4 + i)*th,xr - (mlen + 3.2)*tw,yr - (4 + i)*th,cl);CHKERRQ(ierr);
ierr = PetscDrawString(draw,xr - (mlen + 3)*tw,yr - (4.5 + i)*th,PETSC_DRAW_BLACK,lg->legend[i]);CHKERRQ(ierr);
}
ierr = PetscDrawLine(draw,xr - 2*tw,yr - 3*th,xr - 2*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr - (mlen + 8)*tw,yr - (4+lg->dim)*th,xr - 2*tw,yr - (4+lg->dim)*th,PETSC_DRAW_BLACK);CHKERRQ(ierr);
}
ierr = PetscDrawFlush(lg->win);CHKERRQ(ierr);
ierr = PetscDrawPause(lg->win);CHKERRQ(ierr);
#if defined(PETSC_HAVE_SETJMP_H) && defined(PETSC_HAVE_X)
XSetIOErrorHandler(PETSC_NULL);
#endif
PetscFunctionReturn(0);
}
PETSC_EXTERN void PETSC_STDCALL petscdrawstringgetsize_(PetscDraw draw,PetscReal *width,PetscReal *height, int *__ierr ) {
*__ierr = PetscDrawStringGetSize(
(PetscDraw)PetscToPointer((draw) ),width,height);
}
/*@
PetscDrawLGDraw - Redraws a line graph.
Collective on PetscDrawLG
Input Parameter:
. lg - the line graph context
Level: intermediate
.seealso: PetscDrawSPDraw(), PetscDrawLGSPDraw(), PetscDrawLGReset()
@*/
PetscErrorCode PetscDrawLGDraw(PetscDrawLG lg)
{
PetscReal xmin,xmax,ymin,ymax;
PetscErrorCode ierr;
PetscMPIInt rank;
PetscDraw draw;
PetscBool isnull;
PetscFunctionBegin;
PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)lg),&rank);CHKERRQ(ierr);
draw = lg->win;
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
xmin = lg->xmin; xmax = lg->xmax; ymin = lg->ymin; ymax = lg->ymax;
ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr);
ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
if (!rank) {
int i,j,dim=lg->dim,nopts=lg->nopts,cl;
for (i=0; i<dim; i++) {
for (j=1; j<nopts; j++) {
cl = lg->colors ? lg->colors[i] : (PETSC_DRAW_BLACK + i);
ierr = PetscDrawLine(draw,lg->x[(j-1)*dim+i],lg->y[(j-1)*dim+i],lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);
if (lg->use_markers) {ierr = PetscDrawMarker(draw,lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);}
}
}
}
if (!rank && lg->legend) {
int i,dim=lg->dim,cl;
PetscReal xl,yl,xr,yr,tw,th;
size_t slen,len=0;
ierr = PetscDrawAxisGetLimits(lg->axis,&xl,&xr,&yl,&yr);CHKERRQ(ierr);
ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
ierr = PetscStrlen(lg->legend[i],&slen);CHKERRQ(ierr);
len = PetscMax(len,slen);
}
xr = xr - 1.5*tw; xl = xr - (len + 7)*tw;
yr = yr - 1.0*th; yl = yr - (dim + 1)*th;
ierr = PetscDrawLine(draw,xl,yl,xr,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr,yl,xr,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xr,yr,xl,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,xl,yr,xl,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr);
for (i=0; i<dim; i++) {
cl = lg->colors ? lg->colors[i] : (PETSC_DRAW_BLACK + i);
ierr = PetscDrawLine(draw,xl + 1*tw,yr - (i + 1)*th,xl + 5*tw,yr - (i + 1)*th,cl);CHKERRQ(ierr);
ierr = PetscDrawString(draw,xl + 6*tw,yr - (i + 1.5)*th,PETSC_DRAW_BLACK,lg->legend[i]);CHKERRQ(ierr);
}
}
ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
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);
}
