/*@C
PetscViewerDrawGetDraw - Returns PetscDraw object from PetscViewer object.
This PetscDraw object may then be used to perform graphics using
PetscDrawXXX() commands.
Collective on PetscViewer
Input Parameters:
+ viewer - the PetscViewer (created with PetscViewerDrawOpen())
- windownumber - indicates which subwindow (usually 0)
Ouput Parameter:
. draw - the draw object
Level: intermediate
Concepts: drawing^accessing PetscDraw context from PetscViewer
Concepts: graphics
.seealso: PetscViewerDrawGetLG(), PetscViewerDrawGetAxis(), PetscViewerDrawOpen()
@*/
PetscErrorCode PetscViewerDrawGetDraw(PetscViewer viewer,PetscInt windownumber,PetscDraw *draw)
{
PetscViewer_Draw *vdraw;
PetscErrorCode ierr;
PetscBool isdraw;
PetscFunctionBegin;
PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
PetscValidLogicalCollectiveInt(viewer,windownumber,2);
if (draw) PetscValidPointer(draw,3);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
if (!isdraw) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be draw type PetscViewer");
if (windownumber < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Window number cannot be negative");
vdraw = (PetscViewer_Draw*)viewer->data;
windownumber += vdraw->draw_base;
if (windownumber >= vdraw->draw_max) {
/* allocate twice as many slots as needed */
PetscInt draw_max = vdraw->draw_max;
PetscDraw *tdraw = vdraw->draw;
PetscDrawLG *drawlg = vdraw->drawlg;
PetscDrawAxis *drawaxis = vdraw->drawaxis;
vdraw->draw_max = 2*windownumber;
ierr = PetscCalloc3(vdraw->draw_max,&vdraw->draw,vdraw->draw_max,&vdraw->drawlg,vdraw->draw_max,&vdraw->drawaxis);CHKERRQ(ierr);
ierr = PetscMemcpy(vdraw->draw,tdraw,draw_max*sizeof(PetscDraw));CHKERRQ(ierr);
ierr = PetscMemcpy(vdraw->drawlg,drawlg,draw_max*sizeof(PetscDrawLG));CHKERRQ(ierr);
ierr = PetscMemcpy(vdraw->drawaxis,drawaxis,draw_max*sizeof(PetscDrawAxis));CHKERRQ(ierr);
ierr = PetscFree3(tdraw,drawlg,drawaxis);CHKERRQ(ierr);
}
if (!vdraw->draw[windownumber]) {
char *title = vdraw->title, tmp_str[128];
if (windownumber) {
ierr = PetscSNPrintf(tmp_str,sizeof(tmp_str),"%s:%d",vdraw->title?vdraw->title:"",windownumber);CHKERRQ(ierr);
title = tmp_str;
}
ierr = PetscDrawCreate(PetscObjectComm((PetscObject)viewer),vdraw->display,title,PETSC_DECIDE,PETSC_DECIDE,vdraw->w,vdraw->h,&vdraw->draw[windownumber]);CHKERRQ(ierr);
ierr = PetscLogObjectParent((PetscObject)viewer,(PetscObject)vdraw->draw[windownumber]);CHKERRQ(ierr);
if (vdraw->drawtype) {
ierr = PetscDrawSetType(vdraw->draw[windownumber],vdraw->drawtype);CHKERRQ(ierr);
}
ierr = PetscDrawSetPause(vdraw->draw[windownumber],vdraw->pause);CHKERRQ(ierr);
ierr = PetscDrawSetOptionsPrefix(vdraw->draw[windownumber],((PetscObject)viewer)->prefix);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(vdraw->draw[windownumber]);CHKERRQ(ierr);
}
if (draw) *draw = vdraw->draw[windownumber];
if (draw) PetscValidHeaderSpecific(*draw,PETSC_DRAW_CLASSID,-1);
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscErrorCode ierr;
int x = 0,y = 0,width = 300,height = 300;
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,2.0);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawSetViewPort(draw,.25,.25,.75,.75);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0.0,0.0,1.0,1.0,PETSC_DRAW_BLACK);CHKERRQ(ierr);
ierr = PetscDrawString(draw,.2,.2,PETSC_DRAW_RED,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawString(draw,.5,.5,PETSC_DRAW_GREEN,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawString(draw,.2,.8,PETSC_DRAW_BLUE,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/*ierr = PetscDrawStringSetSize(draw,.5,.5);CHKERRQ(ierr);*/
ierr = PetscDrawString(draw,.2,.2,PETSC_DRAW_RED,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawString(draw,.5,.5,PETSC_DRAW_GREEN,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawString(draw,.2,.8,PETSC_DRAW_BLUE,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawResizeWindow(draw,600,600);CHKERRQ(ierr);
ierr = PetscDrawGetWindowSize(draw,&width,&height);CHKERRQ(ierr);
ierr = PetscDrawSetViewPort(draw,0,0,1,1);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr);
/*ierr = PetscDrawLine(draw,0.0,0.0,1.0,1.0,PETSC_DRAW_RED);CHKERRQ(ierr);*/
/*ierr = PetscDrawLine(draw,0.0,1.0,1.0,0.0,PETSC_DRAW_BLUE);CHKERRQ(ierr);*/
ierr = PetscDrawString(draw,.2,.2,PETSC_DRAW_RED,"Some Text\n Some Other Text");CHKERRQ(ierr);
ierr = PetscDrawString(draw,.5,.5,PETSC_DRAW_RED,"ABCygj\n()[]F$");CHKERRQ(ierr);
ierr = PetscDrawString(draw,0,0,PETSC_DRAW_RED,"Horizontal Text (ABCygj)");CHKERRQ(ierr);
ierr = PetscDrawStringVertical(draw,0,1,PETSC_DRAW_RED,"Vertical Text");CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
ierr = PetscDrawSave(draw);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
/*@
PetscViewerDrawSetPause - Sets a pause for each PetscDraw in the viewer
Not Collective
Input Parameters:
+ viewer - the PetscViewer
- pause - the pause value
Level: intermediate
.seealso: PetscViewerDrawOpen(), PetscViewerDrawGetDraw(),
@*/
PetscErrorCode PetscViewerDrawSetPause(PetscViewer viewer,PetscReal pause)
{
PetscErrorCode ierr;
PetscInt i;
PetscBool isdraw;
PetscViewer_Draw *vdraw;
PetscFunctionBegin;
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
if (isdraw) {
vdraw = (PetscViewer_Draw*)viewer->data;
for (i=0; i<vdraw->draw_max; i++) {
if (vdraw->draw[i]) {ierr = PetscDrawSetPause(vdraw->draw[i],pause);CHKERRQ(ierr);}
}
}
PetscFunctionReturn(0);
}
/*@
PetscViewerDrawSetPause - Sets a pause for each PetscDraw in the viewer
Not Collective
Input Parameters:
+ viewer - the PetscViewer
- pause - the pause value
Level: intermediate
.seealso: PetscViewerDrawOpen(), PetscViewerDrawGetDraw(),
@*/
PetscErrorCode PetscViewerDrawSetPause(PetscViewer viewer,PetscReal pause)
{
PetscErrorCode ierr;
PetscViewer_Draw *vdraw;
PetscBool isdraw;
PetscInt i;
PetscFunctionBegin;
PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
if (!isdraw) PetscFunctionReturn(0);
vdraw = (PetscViewer_Draw*)viewer->data;
vdraw->pause = pause;
for (i=0; i<vdraw->draw_max; i++) {
if (vdraw->draw[i]) {ierr = PetscDrawSetPause(vdraw->draw[i],pause);CHKERRQ(ierr);}
}
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDrawLG lg;
PetscErrorCode ierr;
PetscInt Mx = 100,i;
PetscReal x,hx = .1/Mx,pause,xx[3],yy[3];
PetscDraw draw;
const char *const legend[] = {"(1 - u^2)^2","1 - u^2","-(1 - u)log(1 - u)"};
PetscDrawAxis axis;
static PetscDrawViewPorts *ports = 0;
PetscFunctionBegin;
PetscInitialize(&argc,&argv,0,help);
ierr = PetscViewerDrawResize(PETSC_VIEWER_DRAW_(PETSC_COMM_WORLD),1200,800);CHKERRQ(ierr);
ierr = PetscViewerDrawGetDrawLG(PETSC_VIEWER_DRAW_(PETSC_COMM_WORLD),0,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
if (!ports) {
ierr = PetscDrawViewPortsCreateRect(draw,1,2,&ports);CHKERRQ(ierr);
}
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
/*
Plot the energies
*/
ierr = PetscDrawLGSetDimension(lg,3);CHKERRQ(ierr);
ierr = PetscDrawViewPortsSet(ports,1);CHKERRQ(ierr);
x = .9;
for (i=0; i<Mx; i++) {
xx[0] = xx[1] = xx[2] = x;
yy[0] = (1.-x*x)*(1. - x*x);
yy[1] = (1. - x*x);
yy[2] = -(1.-x)*PetscLogScalar(1.-x);
ierr = PetscDrawLGAddPoint(lg,xx,yy);CHKERRQ(ierr);
x += hx;
}
ierr = PetscDrawGetPause(draw,&pause);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,0.0);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Energy","","");CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,legend);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
/*
Plot the forces
*/
ierr = PetscDrawViewPortsSet(ports,0);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
x = .9;
for (i=0; i<Mx; i++) {
xx[0] = xx[1] = xx[2] = x;
yy[0] = x*x*x - x;
yy[1] = -x;
yy[2] = 1.0 + PetscLogScalar(1. - x);
ierr = PetscDrawLGAddPoint(lg,xx,yy);CHKERRQ(ierr);
x += hx;
}
ierr = PetscDrawAxisSetLabels(axis,"Derivative","","");CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,PETSC_NULL);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,pause);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode TSMonitorSPEig(TS ts,PetscInt step,PetscReal ptime,Vec v,void *monctx)
{
TSMonitorSPEigCtx ctx = (TSMonitorSPEigCtx) monctx;
PetscErrorCode ierr;
KSP ksp = ctx->ksp;
PetscInt n,N,nits,neig,i,its = 200;
PetscReal *r,*c,time_step_save;
PetscDrawSP drawsp = ctx->drawsp;
Mat A,B;
Vec xdot;
SNES snes;
PetscFunctionBegin;
if (!step) PetscFunctionReturn(0);
if (((ctx->howoften > 0) && (!(step % ctx->howoften))) || ((ctx->howoften == -1) && ts->reason)) {
ierr = VecDuplicate(v,&xdot);CHKERRQ(ierr);
ierr = TSGetSNES(ts,&snes);CHKERRQ(ierr);
ierr = SNESGetJacobian(snes,&A,&B,NULL,NULL);CHKERRQ(ierr);
ierr = MatDuplicate(A,MAT_DO_NOT_COPY_VALUES,&B);CHKERRQ(ierr);
/*
This doesn't work because methods keep and use internal information about the shift so it
seems we would need code for each method to trick the correct Jacobian in being computed.
*/
time_step_save = ts->time_step;
ts->time_step = PETSC_MAX_REAL;
ierr = SNESComputeJacobian(snes,v,A,B);CHKERRQ(ierr);
ts->time_step = time_step_save;
ierr = KSPSetOperators(ksp,B,B);CHKERRQ(ierr);
ierr = VecGetSize(v,&n);CHKERRQ(ierr);
if (n < 200) its = n;
ierr = KSPSetTolerances(ksp,1.e-10,PETSC_DEFAULT,PETSC_DEFAULT,its);CHKERRQ(ierr);
ierr = VecSetRandom(xdot,ctx->rand);CHKERRQ(ierr);
ierr = KSPSolve(ksp,xdot,xdot);CHKERRQ(ierr);
ierr = VecDestroy(&xdot);CHKERRQ(ierr);
ierr = KSPGetIterationNumber(ksp,&nits);CHKERRQ(ierr);
N = nits+2;
if (nits) {
PetscDraw draw;
PetscReal pause;
PetscDrawAxis axis;
PetscReal xmin,xmax,ymin,ymax;
ierr = PetscDrawSPReset(drawsp);CHKERRQ(ierr);
ierr = PetscDrawSPSetLimits(drawsp,ctx->xmin,ctx->xmax,ctx->ymin,ctx->ymax);CHKERRQ(ierr);
ierr = PetscMalloc2(PetscMax(n,N),&r,PetscMax(n,N),&c);CHKERRQ(ierr);
if (ctx->computeexplicitly) {
ierr = KSPComputeEigenvaluesExplicitly(ksp,n,r,c);CHKERRQ(ierr);
neig = n;
} else {
ierr = KSPComputeEigenvalues(ksp,N,r,c,&neig);CHKERRQ(ierr);
}
/* We used the positive operator to be able to reuse KSPs that require positive definiteness, now flip the spectrum as is conventional for ODEs */
for (i=0; i<neig; i++) r[i] = -r[i];
for (i=0; i<neig; i++) {
if (ts->ops->linearstability) {
PetscReal fr,fi;
ierr = TSComputeLinearStability(ts,r[i],c[i],&fr,&fi);CHKERRQ(ierr);
if ((fr*fr + fi*fi) > 1.0) {
ierr = PetscPrintf(ctx->comm,"Linearized Eigenvalue %g + %g i linear stability function %g norm indicates unstable scheme \n",(double)r[i],(double)c[i],(double)(fr*fr + fi*fi));CHKERRQ(ierr);
}
}
ierr = PetscDrawSPAddPoint(drawsp,r+i,c+i);CHKERRQ(ierr);
}
ierr = PetscFree2(r,c);CHKERRQ(ierr);
ierr = PetscDrawSPGetDraw(drawsp,&draw);CHKERRQ(ierr);
ierr = PetscDrawGetPause(draw,&pause);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,0.0);CHKERRQ(ierr);
ierr = PetscDrawSPDraw(drawsp,PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,pause);CHKERRQ(ierr);
if (ts->ops->linearstability) {
ierr = PetscDrawSPGetAxis(drawsp,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisGetLimits(axis,&xmin,&xmax,&ymin,&ymax);CHKERRQ(ierr);
ierr = PetscDrawIndicatorFunction(draw,xmin,xmax,ymin,ymax,PETSC_DRAW_CYAN,(PetscErrorCode (*)(void*,PetscReal,PetscReal,PetscBool*))TSLinearStabilityIndicator,ts);CHKERRQ(ierr);
ierr = PetscDrawSPDraw(drawsp,PETSC_FALSE);CHKERRQ(ierr);
}
}
ierr = MatDestroy(&B);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
/*
This routine is not parallel
*/
PetscErrorCode MyMonitor(TS ts,PetscInt step,PetscReal time,Vec U,void *ptr)
{
UserCtx *ctx = (UserCtx*)ptr;
PetscDrawLG lg;
PetscErrorCode ierr;
PetscScalar *u;
PetscInt Mx,i,xs,xm,cnt;
PetscReal x,y,hx,pause,sx,len,max,xx[2],yy[2];
PetscDraw draw;
Vec localU;
DM da;
int colors[] = {PETSC_DRAW_YELLOW,PETSC_DRAW_RED,PETSC_DRAW_BLUE};
const char*const legend[] = {"-kappa (\\grad u,\\grad u)","(1 - u^2)^2"};
PetscDrawAxis axis;
PetscDrawViewPorts *ports;
PetscReal vbounds[] = {-1.1,1.1};
PetscFunctionBegin;
ierr = PetscViewerDrawSetBounds(PETSC_VIEWER_DRAW_(PETSC_COMM_WORLD),1,vbounds);CHKERRQ(ierr);
ierr = PetscViewerDrawResize(PETSC_VIEWER_DRAW_(PETSC_COMM_WORLD),1200,800);CHKERRQ(ierr);
ierr = TSGetDM(ts,&da);CHKERRQ(ierr);
ierr = DMGetLocalVector(da,&localU);CHKERRQ(ierr);
ierr = DMDAGetInfo(da,PETSC_IGNORE,&Mx,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE,PETSC_IGNORE);CHKERRQ(ierr);
ierr = DMDAGetCorners(da,&xs,NULL,NULL,&xm,NULL,NULL);CHKERRQ(ierr);
hx = 1.0/(PetscReal)Mx; sx = 1.0/(hx*hx);
ierr = DMGlobalToLocalBegin(da,U,INSERT_VALUES,localU);CHKERRQ(ierr);
ierr = DMGlobalToLocalEnd(da,U,INSERT_VALUES,localU);CHKERRQ(ierr);
ierr = DMDAVecGetArrayRead(da,localU,&u);CHKERRQ(ierr);
ierr = PetscViewerDrawGetDrawLG(PETSC_VIEWER_DRAW_(PETSC_COMM_WORLD),1,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
if (!ctx->ports) {
ierr = PetscDrawViewPortsCreateRect(draw,1,3,&ctx->ports);CHKERRQ(ierr);
}
ports = ctx->ports;
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
xx[0] = 0.0; xx[1] = 1.0; cnt = 2;
ierr = PetscOptionsGetRealArray(NULL,NULL,"-zoom",xx,&cnt,NULL);CHKERRQ(ierr);
xs = xx[0]/hx; xm = (xx[1] - xx[0])/hx;
/*
Plot the energies
*/
ierr = PetscDrawLGSetDimension(lg,1 + (ctx->allencahn ? 1 : 0));CHKERRQ(ierr);
ierr = PetscDrawLGSetColors(lg,colors+1);CHKERRQ(ierr);
ierr = PetscDrawViewPortsSet(ports,2);CHKERRQ(ierr);
x = hx*xs;
for (i=xs; i<xs+xm; i++) {
xx[0] = xx[1] = x;
yy[0] = PetscRealPart(.25*ctx->kappa*(u[i-1] - u[i+1])*(u[i-1] - u[i+1])*sx);
if (ctx->allencahn) yy[1] = .25*PetscRealPart((1. - u[i]*u[i])*(1. - u[i]*u[i]));
ierr = PetscDrawLGAddPoint(lg,xx,yy);CHKERRQ(ierr);
x += hx;
}
ierr = PetscDrawGetPause(draw,&pause);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,0.0);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Energy","","");CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,legend);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
/*
Plot the forces
*/
ierr = PetscDrawViewPortsSet(ports,1);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
x = xs*hx;;
max = 0.;
for (i=xs; i<xs+xm; i++) {
xx[0] = xx[1] = x;
yy[0] = PetscRealPart(ctx->kappa*(u[i-1] + u[i+1] - 2.0*u[i])*sx);
max = PetscMax(max,PetscAbs(yy[0]));
if (ctx->allencahn) {
yy[1] = PetscRealPart(u[i] - u[i]*u[i]*u[i]);
max = PetscMax(max,PetscAbs(yy[1]));
}
ierr = PetscDrawLGAddPoint(lg,xx,yy);CHKERRQ(ierr);
x += hx;
}
ierr = PetscDrawAxisSetLabels(axis,"Right hand side","","");CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,NULL);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
/*
Plot the solution
*/
ierr = PetscDrawLGSetDimension(lg,1);CHKERRQ(ierr);
ierr = PetscDrawViewPortsSet(ports,0);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
x = hx*xs;
ierr = PetscDrawLGSetLimits(lg,x,x+(xm-1)*hx,-1.1,1.1);CHKERRQ(ierr);
ierr = PetscDrawLGSetColors(lg,colors);CHKERRQ(ierr);
for (i=xs; i<xs+xm; i++) {
xx[0] = x;
yy[0] = PetscRealPart(u[i]);
ierr = PetscDrawLGAddPoint(lg,xx,yy);CHKERRQ(ierr);
x += hx;
}
ierr = PetscDrawAxisSetLabels(axis,"Solution","","");CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
/*
Print the forces as arrows on the solution
*/
x = hx*xs;
cnt = xm/60;
cnt = (!cnt) ? 1 : cnt;
for (i=xs; i<xs+xm; i += cnt) {
y = PetscRealPart(u[i]);
len = .5*PetscRealPart(ctx->kappa*(u[i-1] + u[i+1] - 2.0*u[i])*sx)/max;
ierr = PetscDrawArrow(draw,x,y,x,y+len,PETSC_DRAW_RED);CHKERRQ(ierr);
if (ctx->allencahn) {
len = .5*PetscRealPart(u[i] - u[i]*u[i]*u[i])/max;
ierr = PetscDrawArrow(draw,x,y,x,y+len,PETSC_DRAW_BLUE);CHKERRQ(ierr);
}
x += cnt*hx;
}
ierr = DMDAVecRestoreArrayRead(da,localU,&x);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(da,&localU);CHKERRQ(ierr);
ierr = PetscDrawStringSetSize(draw,.2,.2);CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawSetPause(draw,pause);CHKERRQ(ierr);
ierr = PetscDrawPause(draw);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
void PETSC_STDCALL petscdrawsetpause_(PetscDraw draw,PetscReal *lpause, int *__ierr ){
*__ierr = PetscDrawSetPause(
(PetscDraw)PetscToPointer((draw) ),*lpause);
}
