PetscErrorCode VecView_Seq_Draw_LG(Vec xin,PetscViewer v)
{
PetscDraw draw;
PetscBool isnull;
PetscDrawLG lg;
PetscErrorCode ierr;
PetscInt i,c,bs = PetscAbs(xin->map->bs),n = xin->map->n/bs;
const PetscScalar *xv;
PetscReal *xx,*yy;
int colors[] = {PETSC_DRAW_RED};
PetscFunctionBegin;
ierr = PetscViewerDrawGetDraw(v,0,&draw);CHKERRQ(ierr);
ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
if (isnull) PetscFunctionReturn(0);
ierr = PetscMalloc2(n,&xx,n,&yy);CHKERRQ(ierr);
ierr = VecGetArrayRead(xin,&xv);CHKERRQ(ierr);
for (c=0; c<bs; c++) {
ierr = PetscViewerDrawGetDrawLG(v,c,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
ierr = PetscDrawLGSetDimension(lg,1);CHKERRQ(ierr);
ierr = PetscDrawLGSetColors(lg,colors);CHKERRQ(ierr);
for (i=0; i<n; i++) {
xx[i] = (PetscReal)i;
yy[i] = PetscRealPart(xv[c + i*bs]);
}
ierr = PetscDrawLGAddPoints(lg,n,&xx,&yy);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
}
ierr = VecRestoreArrayRead(xin,&xv);CHKERRQ(ierr);
ierr = PetscFree2(xx,yy);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode VecView_Seq_Draw_LG(Vec xin,PetscViewer v)
{
PetscErrorCode ierr;
PetscInt i,c,bs = PetscAbs(xin->map->bs),n = xin->map->n/bs;
PetscDraw win;
PetscReal *xx;
PetscDrawLG lg;
const PetscScalar *xv;
PetscReal *yy;
PetscFunctionBegin;
ierr = PetscMalloc1(n,&xx);CHKERRQ(ierr);
ierr = PetscMalloc1(n,&yy);CHKERRQ(ierr);
ierr = VecGetArrayRead(xin,&xv);CHKERRQ(ierr);
for (c=0; c<bs; c++) {
ierr = PetscViewerDrawGetDrawLG(v,c,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetDraw(lg,&win);CHKERRQ(ierr);
ierr = PetscDrawCheckResizedWindow(win);CHKERRQ(ierr);
ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);
for (i=0; i<n; i++) {
xx[i] = (PetscReal) i;
yy[i] = PetscRealPart(xv[c + i*bs]);
}
ierr = PetscDrawLGAddPoints(lg,n,&xx,&yy);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = PetscDrawSynchronizedFlush(win);CHKERRQ(ierr);
}
ierr = VecRestoreArrayRead(xin,&xv);CHKERRQ(ierr);
ierr = PetscFree(yy);CHKERRQ(ierr);
ierr = PetscFree(xx);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@C
DMNetworkMonitorAdd - Adds a new viewer to monitor
Collective on DMNetworkMonitor
Input Parameters:
+ monitor - the monitor
. name - name of viewer
. element - vertex / edge number
. nodes - number of nodes
. start - variable starting offset
. blocksize - variable blocksize
. xmin - xmin (or PETSC_DECIDE) for viewer
. xmax - xmax (or PETSC_DECIDE) for viewer
. ymin - ymin for viewer
. ymax - ymax for viewer
- hold - determines if plot limits should be held
Level: intermediate
Notes:
This is written to be independent of the semantics associated to the variables
at a given network vertex / edge.
Precisely, the parameters nodes, start and blocksize allow you to select a general
strided subarray of the variables to monitor.
.seealso: DMNetworkMonitorCreate(), DMNetworkMonitorDestroy()
@*/
PetscErrorCode DMNetworkMonitorAdd(DMNetworkMonitor monitor,const char *name,PetscInt element,PetscInt nodes,PetscInt start,PetscInt blocksize,PetscReal xmin,PetscReal xmax,PetscReal ymin,PetscReal ymax,PetscBool hold)
{
PetscErrorCode ierr;
PetscDrawLG drawlg;
PetscDrawAxis axis;
PetscMPIInt rank, size;
DMNetworkMonitorList node;
char titleBuffer[64];
PetscInt vStart,vEnd,eStart,eEnd;
PetscFunctionBegin;
ierr = MPI_Comm_rank(monitor->comm, &rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(monitor->comm, &size);CHKERRQ(ierr);
ierr = DMNetworkGetVertexRange(monitor->network, &vStart, &vEnd);CHKERRQ(ierr);
ierr = DMNetworkGetEdgeRange(monitor->network, &eStart, &eEnd);CHKERRQ(ierr);
/* Make window title */
if (vStart <= element && element < vEnd) {
ierr = PetscSNPrintf(titleBuffer, 64, "%s @ vertex %d [%d / %d]", name, element - vStart, rank, size-1);CHKERRQ(ierr);
} else if (eStart <= element && element < eEnd) {
ierr = PetscSNPrintf(titleBuffer, 64, "%s @ edge %d [%d / %d]", name, element - eStart, rank, size-1);CHKERRQ(ierr);
} else {
/* vertex / edge is not on local machine, so skip! */
PetscFunctionReturn(0);
}
ierr = PetscMalloc1(1, &node);CHKERRQ(ierr);
/* Setup viewer. */
ierr = PetscViewerDrawOpen(monitor->comm, NULL, titleBuffer, PETSC_DECIDE, PETSC_DECIDE, PETSC_DRAW_QUARTER_SIZE, PETSC_DRAW_QUARTER_SIZE, &(node->viewer));CHKERRQ(ierr);
ierr = PetscViewerPushFormat(node->viewer, PETSC_VIEWER_DRAW_LG_XRANGE);CHKERRQ(ierr);
ierr = PetscViewerDrawGetDrawLG(node->viewer, 0, &drawlg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(drawlg, &axis);CHKERRQ(ierr);
if (xmin != PETSC_DECIDE && xmax != PETSC_DECIDE) {
ierr = PetscDrawAxisSetLimits(axis, xmin, xmax, ymin, ymax);CHKERRQ(ierr);
} else {
ierr = PetscDrawAxisSetLimits(axis, 0, nodes-1, ymin, ymax);CHKERRQ(ierr);
}
ierr = PetscDrawAxisSetHoldLimits(axis, hold);CHKERRQ(ierr);
/* Setup vector storage for drawing. */
ierr = VecCreateSeq(PETSC_COMM_SELF, nodes, &(node->v));CHKERRQ(ierr);
node->element = element;
node->nodes = nodes;
node->start = start;
node->blocksize = blocksize;
node->next = monitor->firstnode;
monitor->firstnode = node;
PetscFunctionReturn(0);
}
Draw( unsigned dim_, const MPI_Comm comm ) : dim( dim_ )
{
// static unsigned w{0};
window = _draw_window_num_;
_draw_window_num_ += 1;
viewer = PETSC_VIEWER_DRAW_( comm );
PetscViewerDrawGetDraw( viewer, static_cast<int>( window ), &draw );
PetscViewerDrawGetDrawLG( viewer, static_cast<int>( window ), &lg );
PetscDrawSetDoubleBuffer( draw );
PetscDrawLGSetDimension( lg, static_cast<int>( dim ) );
PetscDrawLGGetAxis( lg, &axis );
}
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 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);
}
/*
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);
}
PETSC_EXTERN void PETSC_STDCALL petscviewerdrawgetdrawlg_(PetscViewer *vin,int *win,PetscDrawLG *drawlg,PetscErrorCode *ierr)
{
PetscViewer v;
PetscPatchDefaultViewers_Fortran(vin,v);
*ierr = PetscViewerDrawGetDrawLG(v,*win,drawlg);
}
