int main(int argc,char **argv)
{
PetscDraw draw;
PetscErrorCode ierr;
int x = 0,y = 0,width = 300,height = 300;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw);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 = PetscDrawStringSetSize(draw,.5,.5);CHKERRQ(ierr);
ierr = PetscDrawString(draw,.2,.2,PETSC_DRAW_BLUE,"Some Text");CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscSleep(2);CHKERRQ(ierr);
ierr = PetscDrawClear(draw);CHKERRQ(ierr); ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawResizeWindow(draw,600,600);CHKERRQ(ierr);
ierr = PetscSleep(2);CHKERRQ(ierr);
ierr = PetscDrawLine(draw,0.0,1.0,1.0,0.0,PETSC_DRAW_BLUE);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscSleep(2);CHKERRQ(ierr);
ierr = PetscDrawDestroy(draw);CHKERRQ(ierr);
ierr = PetscFinalize();CHKERRQ(ierr);
return 0;
}
/*@C
TSMonitorSPEigCtxCreate - Creates a context for use with TS to monitor the eigenvalues of the linearized operator
Collective on TS
Input Parameters:
+ host - the X display to open, or null for the local machine
. label - the title to put in the title bar
. x, y - the screen coordinates of the upper left coordinate of the window
. m, n - the screen width and height in pixels
- howoften - if positive then determines the frequency of the plotting, if -1 then only at the final time
Output Parameter:
. ctx - the context
Options Database Key:
. -ts_monitor_sp_eig - plot egienvalues of linearized right hand side
Notes:
Use TSMonitorSPEigCtxDestroy() to destroy.
Currently only works if the Jacobian is provided explicitly.
Currently only works for ODEs u_t - F(t,u) = 0; that is with no mass matrix.
Level: intermediate
.keywords: TS, monitor, line graph, residual, seealso
.seealso: TSMonitorSPEigTimeStep(), TSMonitorSet(), TSMonitorLGSolution(), TSMonitorLGError()
@*/
PetscErrorCode TSMonitorSPEigCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TSMonitorSPEigCtx *ctx)
{
PetscDraw win;
PetscErrorCode ierr;
PC pc;
PetscFunctionBegin;
ierr = PetscNew(ctx);CHKERRQ(ierr);
ierr = PetscRandomCreate(comm,&(*ctx)->rand);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions((*ctx)->rand);CHKERRQ(ierr);
ierr = PetscDrawCreate(comm,host,label,x,y,m,n,&win);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(win);CHKERRQ(ierr);
ierr = PetscDrawSPCreate(win,1,&(*ctx)->drawsp);CHKERRQ(ierr);
ierr = KSPCreate(comm,&(*ctx)->ksp);CHKERRQ(ierr);
ierr = KSPSetOptionsPrefix((*ctx)->ksp,"ts_monitor_sp_eig_");CHKERRQ(ierr); /* this is wrong, used use also prefix from the TS */
ierr = KSPSetType((*ctx)->ksp,KSPGMRES);CHKERRQ(ierr);
ierr = KSPGMRESSetRestart((*ctx)->ksp,200);CHKERRQ(ierr);
ierr = KSPSetTolerances((*ctx)->ksp,1.e-10,PETSC_DEFAULT,PETSC_DEFAULT,200);CHKERRQ(ierr);
ierr = KSPSetComputeSingularValues((*ctx)->ksp,PETSC_TRUE);CHKERRQ(ierr);
ierr = KSPSetFromOptions((*ctx)->ksp);CHKERRQ(ierr);
ierr = KSPGetPC((*ctx)->ksp,&pc);CHKERRQ(ierr);
ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
(*ctx)->howoften = howoften;
(*ctx)->computeexplicitly = PETSC_FALSE;
ierr = PetscOptionsGetBool(NULL,"-ts_monitor_sp_eig_explicitly",&(*ctx)->computeexplicitly,NULL);CHKERRQ(ierr);
(*ctx)->comm = comm;
(*ctx)->xmin = -2.1;
(*ctx)->xmax = 1.1;
(*ctx)->ymin = -1.1;
(*ctx)->ymax = 1.1;
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscDrawLG lg;
PetscDrawAxis axis;
PetscInt n = 15,i,x = 0,y = 0,width = 400,height = 300,nports = 1;
PetscBool useports,flg;
const char *xlabel,*ylabel,*toplabel,*legend;
PetscReal xd,yd;
PetscDrawViewPorts *ports = NULL;
PetscErrorCode ierr;
toplabel = "Top Label"; xlabel = "X-axis Label"; ylabel = "Y-axis Label"; legend = "Legend";
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-x",&x,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-y",&y,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-width",&width,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-height",&height,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-nports",&nports,&useports);CHKERRQ(ierr);
ierr = PetscOptionsHasName(NULL,NULL,"-nolegend",&flg);CHKERRQ(ierr);
if (flg) legend = NULL;
ierr = PetscOptionsHasName(NULL,NULL,"-notoplabel",&flg);CHKERRQ(ierr);
if (flg) toplabel = NULL;
ierr = PetscOptionsHasName(NULL,NULL,"-noxlabel",&flg);CHKERRQ(ierr);
if (flg) xlabel = NULL;
ierr = PetscOptionsHasName(NULL,NULL,"-noylabel",&flg);CHKERRQ(ierr);
if (flg) ylabel = NULL;
ierr = PetscOptionsHasName(NULL,NULL,"-nolabels",&flg);CHKERRQ(ierr);
if (flg) {toplabel = NULL; xlabel = NULL; ylabel = NULL;}
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
if (useports) {
ierr = PetscDrawViewPortsCreate(draw,nports,&ports);CHKERRQ(ierr);
ierr = PetscDrawViewPortsSet(ports,0);CHKERRQ(ierr);
}
ierr = PetscDrawLGCreate(draw,1,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGSetUseMarkers(lg,PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel);CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,&legend);CHKERRQ(ierr);
ierr = PetscDrawLGSetFromOptions(lg);CHKERRQ(ierr);
for (i=0; i<=n; i++) {
xd = (PetscReal)(i - 5); yd = xd*xd;
ierr = PetscDrawLGAddPoint(lg,&xd,&yd);CHKERRQ(ierr);
}
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
ierr = PetscDrawViewPortsDestroy(ports);CHKERRQ(ierr);
ierr = PetscDrawLGDestroy(&lg);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
/*@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);
}
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);
}
/*@C
PetscViewerDrawGetDraw - Returns PetscDraw object from PetscViewer object.
This PetscDraw object may then be used to perform graphics using
PetscDrawXXX() commands.
Not collective (but PetscDraw returned will be parallel object if PetscViewer is)
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 PETSC_DLLEXPORT PetscViewerDrawGetDraw(PetscViewer viewer,PetscInt windownumber,PetscDraw *draw)
{
PetscViewer_Draw *vdraw = (PetscViewer_Draw*)viewer->data;
PetscErrorCode ierr;
PetscTruth isdraw;
char *title;
PetscFunctionBegin;
PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,1);
if (draw) PetscValidPointer(draw,3);
ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
if (!isdraw) {
SETERRQ(PETSC_ERR_ARG_WRONG,"Must be draw type PetscViewer");
}
if (windownumber < 0) {
SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Window number cannot be negative");
}
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 = PetscMalloc3(vdraw->draw_max,PetscDraw,&vdraw->draw,vdraw->draw_max,PetscDrawLG,&vdraw->drawlg,vdraw->draw_max,PetscDrawAxis,&vdraw->drawaxis);CHKERRQ(ierr);
ierr = PetscMemzero(vdraw->draw,vdraw->draw_max*sizeof(PetscDraw));CHKERRQ(ierr);
ierr = PetscMemzero(vdraw->drawlg,vdraw->draw_max*sizeof(PetscDrawLG));CHKERRQ(ierr);
ierr = PetscMemzero(vdraw->drawaxis,vdraw->draw_max*sizeof(PetscDrawAxis));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]) {
if (!windownumber) {
title = vdraw->title;
} else {
char tmp_str[128];
ierr = PetscSNPrintf(tmp_str, 128, "%s:%d", vdraw->title,windownumber);CHKERRQ(ierr);
title = tmp_str;
}
ierr = PetscDrawCreate(((PetscObject)viewer)->comm,vdraw->display,title,PETSC_DECIDE,PETSC_DECIDE,vdraw->w,vdraw->h,&vdraw->draw[windownumber]);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(vdraw->draw[windownumber]);CHKERRQ(ierr);
}
if (draw) *draw = vdraw->draw[windownumber];
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
int x = 0,y = 0,width = 256,height = 256;
PetscErrorCode ierr;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_SELF,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawZoom(draw,zoomfunction,NULL);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return 0;
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscDrawLG lg;
PetscDrawAxis axis;
PetscInt n = 20,i,x = 0,y = 0,width = 300,height = 300,nports = 1;
PetscTruth flg;
const char *xlabel,*ylabel,*toplabel;
PetscReal xd,yd;
PetscDrawViewPorts *ports;
PetscErrorCode ierr;
xlabel = "X-axis Label";toplabel = "Top Label";ylabel = "Y-axis Label";
ierr = PetscInitialize(&argc,&argv,(char*)0,help);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(PETSC_NULL,"-width",&width,PETSC_NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(PETSC_NULL,"-height",&height,PETSC_NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(PETSC_NULL,"-n",&n,PETSC_NULL);CHKERRQ(ierr);
ierr = PetscOptionsHasName(PETSC_NULL,"-nolabels",&flg);CHKERRQ(ierr);
if (flg) {
xlabel = (char *)0; toplabel = (char *)0;
}
ierr = PetscDrawCreate(PETSC_COMM_SELF,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(PETSC_NULL,"-nports",&nports,PETSC_NULL);CHKERRQ(ierr);
ierr = PetscDrawViewPortsCreate(draw,nports,&ports);CHKERRQ(ierr);
ierr = PetscDrawViewPortsSet(ports,0);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(draw,1,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel);CHKERRQ(ierr);
for (i=0; i<n ; i++) {
xd = (PetscReal)(i - 5); yd = xd*xd;
ierr = PetscDrawLGAddPoint(lg,&xd,&yd);CHKERRQ(ierr);
}
ierr = PetscDrawLGIndicateDataPoints(lg);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = PetscDrawString(draw,-3.,150.0,PETSC_DRAW_BLUE,"A legend");CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscSleep(2);CHKERRQ(ierr);
ierr = PetscDrawViewPortsDestroy(ports);CHKERRQ(ierr);
ierr = PetscDrawLGDestroy(lg);CHKERRQ(ierr);
ierr = PetscDrawDestroy(draw);CHKERRQ(ierr);
ierr = PetscFinalize();CHKERRQ(ierr);
return 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;
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscDrawHG hist;
PetscDrawAxis axis;
PetscErrorCode ierr;
int n = 20,i,x = 0,y = 0,width = 400,height = 300,bins = 8;
PetscInt w = 400,h = 300,nn = 20,b = 8,c = PETSC_DRAW_GREEN;
int color = PETSC_DRAW_GREEN;
const char *xlabel,*ylabel,*toplabel;
PetscReal xd;
PetscBool flg;
xlabel = "X-axis Label"; toplabel = "Top Label"; ylabel = "Y-axis Label";
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-width",&w,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-height",&h,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-n",&nn,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-bins",&b,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-color",&c,NULL);CHKERRQ(ierr);
ierr = PetscOptionsHasName(NULL,NULL,"-nolabels",&flg);CHKERRQ(ierr);
width = (int) w; height = (int)h; n = (int)nn; bins = (int) b; color = (int) c;
if (flg) { xlabel = NULL; ylabel = NULL; toplabel = NULL; }
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawHGCreate(draw,bins,&hist);CHKERRQ(ierr);
ierr = PetscDrawHGSetColor(hist,color);CHKERRQ(ierr);
ierr = PetscDrawHGGetAxis(hist,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel);CHKERRQ(ierr);
/*ierr = PetscDrawHGSetFromOptions(hist);CHKERRQ(ierr);*/
for (i=0; i<n; i++) {
xd = (PetscReal)(i - 5);
ierr = PetscDrawHGAddValue(hist,xd*xd);CHKERRQ(ierr);
}
ierr = PetscDrawHGDraw(hist);CHKERRQ(ierr);
ierr = PetscDrawHGSave(hist);CHKERRQ(ierr);
ierr = PetscDrawHGDestroy(&hist);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
/*@C
KSPMonitorLGResidualNormCreate - Creates a line graph context for use with
KSP to monitor convergence of preconditioned residual norms.
Collective on KSP
Input Parameters:
+ comm - communicator context
. host - the X display to open, or null for the local machine
. label - the title to put in the title bar
. x, y - the screen coordinates of the upper left coordinate of
the window
- m, n - the screen width and height in pixels
Output Parameter:
. lgctx - the drawing context
Options Database Key:
. -ksp_monitor_lg_residualnorm - Sets line graph monitor
Notes:
Use PetscDrawLGDestroy() to destroy this line graph.
Level: intermediate
.keywords: KSP, monitor, line graph, residual, create
.seealso: KSPMonitorSet(), KSPMonitorLGTrueResidualCreate()
@*/
PetscErrorCode KSPMonitorLGResidualNormCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *lgctx)
{
PetscDraw draw;
PetscErrorCode ierr;
PetscDrawAxis axis;
PetscDrawLG lg;
PetscFunctionBegin;
ierr = PetscDrawCreate(comm,host,label,x,y,m,n,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(draw,1,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGSetFromOptions(lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Convergence","Iteration","Residual Norm");CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
*lgctx = lg;
PetscFunctionReturn(0);
}
/*@
PetscDrawGetPopup - Creates a popup window associated with a PetscDraw window.
Collective on PetscDraw
Input Parameter:
. draw - the original window
Output Parameter:
. popup - the new popup window
Level: advanced
@*/
PetscErrorCode PetscDrawGetPopup(PetscDraw draw,PetscDraw *popup)
{
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
PetscValidPointer(popup,2);
if (draw->popup) *popup = draw->popup;
else if (draw->ops->getpopup) {
ierr = (*draw->ops->getpopup)(draw,popup);CHKERRQ(ierr);
if (*popup) {
ierr = PetscObjectSetOptionsPrefix((PetscObject)*popup,"popup_");CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(*popup);CHKERRQ(ierr);
}
} else *popup = NULL;
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscErrorCode ierr;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_SELF,0,"Title",0,0,256,256,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawBoxedString(draw,.5,.5,PETSC_DRAW_BLUE,PETSC_DRAW_RED,"Greetings",NULL,NULL);CHKERRQ(ierr);
ierr = PetscDrawBoxedString(draw,.25,.75,PETSC_DRAW_BLUE,PETSC_DRAW_RED,"How\nare\nyou?",NULL,NULL);CHKERRQ(ierr);
ierr = PetscDrawBoxedString(draw,.25,.25,PETSC_DRAW_GREEN,PETSC_DRAW_RED,"Long line followed by a very\nshort line",NULL,NULL);CHKERRQ(ierr);
ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return 0;
}
/*@C
KSPMonitorLGResidualNormCreate - Creates a line graph context for use with
KSP to monitor convergence of preconditioned residual norms.
Collective on KSP
Input Parameters:
+ host - the X display to open, or null for the local machine
. label - the title to put in the title bar
. x, y - the screen coordinates of the upper left coordinate of
the window
- m, n - the screen width and height in pixels
Output Parameter:
. draw - the drawing context
Options Database Key:
. -ksp_monitor_lg_residualnorm - Sets line graph monitor
Notes:
Use KSPMonitorLGResidualNormDestroy() to destroy this line graph; do not use PetscDrawLGDestroy().
Level: intermediate
.keywords: KSP, monitor, line graph, residual, create
.seealso: KSPMonitorLGResidualNormDestroy(), KSPMonitorSet(), KSPMonitorLGTrueResidualCreate()
@*/
PetscErrorCode KSPMonitorLGResidualNormCreate(const char host[],const char label[],int x,int y,int m,int n,PetscObject **objs)
{
PetscDraw win;
PetscErrorCode ierr;
PetscDrawAxis axis;
PetscDrawLG draw;
PetscFunctionBegin;
ierr = PetscDrawCreate(PETSC_COMM_SELF,host,label,x,y,m,n,&win);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(win);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(win,1,&draw);CHKERRQ(ierr);
ierr = PetscDrawLGSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(draw,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Convergence","Iteration","Residual Norm");CHKERRQ(ierr);
ierr = PetscLogObjectParent((PetscObject)draw,(PetscObject)win);CHKERRQ(ierr);
ierr = PetscMalloc1(2,objs);CHKERRQ(ierr);
(*objs)[0] = (PetscObject)draw;
(*objs)[1] = (PetscObject)win;
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscDrawBar bar;
PetscDrawAxis axis;
PetscErrorCode ierr;
int color = PETSC_DRAW_ROTATE;
const char *xlabel,*ylabel,*toplabel;
const PetscReal values[] = {.3, .5, .05, .11};
const char *const labels[] = {"A","B","C","D",NULL};
PetscReal limits[2] = {0,0.55}; PetscInt nlimits = 2;
PetscBool nolabels,setlimits;
xlabel = "X-axis Label"; toplabel = "Top Label"; ylabel = "Y-axis Label";
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscOptionsHasName(NULL,NULL,"-nolabels",&nolabels);CHKERRQ(ierr);
if (nolabels) { xlabel = NULL; ylabel = NULL; toplabel = NULL; }
ierr = PetscOptionsGetRealArray(NULL,NULL,"-limits",limits,&nlimits,&setlimits);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,NULL,"Title",PETSC_DECIDE,PETSC_DECIDE,400,300,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawBarCreate(draw,&bar);CHKERRQ(ierr);
ierr = PetscDrawBarGetAxis(bar,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel);CHKERRQ(ierr);
ierr = PetscDrawBarSetColor(bar,color);CHKERRQ(ierr);
ierr = PetscDrawBarSetFromOptions(bar);CHKERRQ(ierr);
if (setlimits) {ierr = PetscDrawBarSetLimits(bar,limits[0],limits[1]);CHKERRQ(ierr);}
ierr = PetscDrawBarSetData(bar,4,values,labels);CHKERRQ(ierr);
ierr = PetscDrawBarDraw(bar);CHKERRQ(ierr);
ierr = PetscDrawBarSave(bar);CHKERRQ(ierr);
ierr = PetscDrawBarDestroy(&bar);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
/*@C
KSPMonitorSNESLGResidualNormCreate - Creates a line graph context for use with
KSP to monitor convergence of preconditioned residual norms.
Collective on KSP
Input Parameters:
+ comm - communicator context
. host - the X display to open, or null for the local machine
. label - the title to put in the title bar
. x, y - the screen coordinates of the upper left coordinate of
the window
- m, n - the screen width and height in pixels
Output Parameter:
. draw - the drawing context
Options Database Key:
. -ksp_monitor_lg_residualnorm - Sets line graph monitor
Notes:
Use KSPMonitorSNESLGResidualNormDestroy() to destroy this line graph; do not use PetscDrawLGDestroy().
Level: intermediate
.keywords: KSP, monitor, line graph, residual, create
.seealso: KSPMonitorSNESLGResidualNormDestroy(), KSPMonitorSet(), KSPMonitorSNESLGTrueResidualCreate()
@*/
PetscErrorCode KSPMonitorSNESLGResidualNormCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscObject **objs)
{
PetscDraw draw;
PetscErrorCode ierr;
PetscDrawAxis axis;
PetscDrawLG lg;
const char *names[] = {"Linear residual","Nonlinear residual"};
PetscFunctionBegin;
ierr = PetscDrawCreate(comm,host,label,x,y,m,n,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(draw,2,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(lg,names);CHKERRQ(ierr);
ierr = PetscDrawLGSetFromOptions(lg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Convergence of Residual Norm","Iteration","Residual Norm");CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscMalloc1(2,objs);CHKERRQ(ierr);
(*objs)[1] = (PetscObject)lg;
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
PetscDraw draw;
PetscMPIInt size,rank;
PetscErrorCode ierr;
int x = 0,y = 0,width = 256,height = 256,i;
ierr = PetscInitialize(&argc,&argv,NULL,help);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
for (i=rank; i<height; i+=size) {
PetscReal y = ((PetscReal)i)/(height-1);
ierr = PetscDrawLine(draw,0.0,y,1.0,y,i%256);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;
}
/*@C
KSPMonitorSNESLGResidualNormCreate - Creates a line graph context for use with
KSP to monitor convergence of preconditioned residual norms.
Collective on KSP
Input Parameters:
+ host - the X display to open, or null for the local machine
. label - the title to put in the title bar
. x, y - the screen coordinates of the upper left coordinate of
the window
- m, n - the screen width and height in pixels
Output Parameter:
. draw - the drawing context
Options Database Key:
. -ksp_monitor_lg_residualnorm - Sets line graph monitor
Notes:
Use KSPMonitorSNESLGResidualNormDestroy() to destroy this line graph; do not use PetscDrawLGDestroy().
Level: intermediate
.keywords: KSP, monitor, line graph, residual, create
.seealso: KSPMonitorSNESLGResidualNormDestroy(), KSPMonitorSet(), KSPMonitorSNESLGTrueResidualCreate()
@*/
PetscErrorCode KSPMonitorSNESLGResidualNormCreate(const char host[],const char label[],int x,int y,int m,int n,PetscObject **objs)
{
PetscDraw draw;
PetscErrorCode ierr;
PetscDrawAxis axis;
PetscDrawLG drawlg;
const char *names[] = {"Linear residual","Nonlinear residual"};
PetscFunctionBegin;
ierr = PetscDrawCreate(PETSC_COMM_SELF,host,label,x,y,m,n,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(draw,2,&drawlg);CHKERRQ(ierr);
ierr = PetscDrawLGSetFromOptions(drawlg);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(drawlg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,"Convergence of Residual Norm","Iteration","Residual Norm");CHKERRQ(ierr);
ierr = PetscDrawLGSetLegend(drawlg,names);CHKERRQ(ierr);
ierr = PetscLogObjectParent((PetscObject)drawlg,(PetscObject)draw);CHKERRQ(ierr);
ierr = PetscMalloc1(3,objs);CHKERRQ(ierr);
(*objs)[1] = (PetscObject)drawlg;
(*objs)[2] = (PetscObject)draw;
PetscFunctionReturn(0);
}
/*
Run with -build_twosided allreduce -pc_type bjacobi -sub_pc_type lu -q 16 -ksp_rtol 1.e-34 (or 1.e-14 for double precision)
-q <q> number of spectral elements to use
-N <N> maximum number of GLL points per element
*/
int main(int argc,char **args)
{
PetscErrorCode ierr;
PetscGLL gll;
PetscInt N = 80,n,q = 8,xs,xn,j,l;
PetscReal **A;
Mat K;
KSP ksp;
PC pc;
Vec x,b;
PetscInt *rows;
PetscReal norm,xc,yc,h;
PetscScalar *f;
PetscDraw draw;
PetscDrawLG lg;
PetscDrawAxis axis;
DM da;
PetscMPIInt rank,size;
ierr = PetscInitialize(&argc,&args,NULL,NULL);if (ierr) return ierr;
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-q",&q,NULL);CHKERRQ(ierr);
ierr = PetscDrawCreate(PETSC_COMM_WORLD,NULL,"Log(Error norm) vs Number of GLL points",0,0,500,500,&draw);CHKERRQ(ierr);
ierr = PetscDrawSetFromOptions(draw);CHKERRQ(ierr);
ierr = PetscDrawLGCreate(draw,1,&lg);CHKERRQ(ierr);
ierr = PetscDrawLGSetUseMarkers(lg,PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscDrawLGGetAxis(lg,&axis);CHKERRQ(ierr);
ierr = PetscDrawAxisSetLabels(axis,NULL,"Number of GLL points","Log(Error Norm)");CHKERRQ(ierr);
for (n=4; n<N; n+=2) {
/*
da contains the information about the parallel layout of the elements
*/
ierr = DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,q*(n-1)+1,1,1,NULL,&da);CHKERRQ(ierr);
ierr = DMSetFromOptions(da);CHKERRQ(ierr);
ierr = DMSetUp(da);CHKERRQ(ierr);
ierr = DMDAGetInfo(da,NULL,&q,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
q = (q-1)/(n-1); /* number of spectral elements */
/*
gll simply contains the GLL node and weight values
*/
ierr = PetscGLLCreate(n,PETSCGLL_VIA_LINEARALGEBRA,&gll);CHKERRQ(ierr);
ierr = DMDASetGLLCoordinates(da,&gll);CHKERRQ(ierr);
/*
Creates the element stiffness matrix for the given gll
*/
ierr = PetscGLLElementLaplacianCreate(&gll,&A);CHKERRQ(ierr);
/*
Scale the element stiffness and weights by the size of the element
*/
h = 2.0/q;
for (j=0; j<n; j++) {
gll.weights[j] *= .5*h;
for (l=0; l<n; l++) {
A[j][l] = 2.*A[j][l]/h;
}
}
/*
Create the global stiffness matrix and add the element stiffness for each local element
*/
ierr = DMCreateMatrix(da,&K);CHKERRQ(ierr);
ierr = MatSetOption(K,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
ierr = DMDAGetCorners(da,&xs,NULL,NULL,&xn,NULL,NULL);CHKERRQ(ierr);
xs = xs/(n-1);
xn = xn/(n-1);
ierr = PetscMalloc1(n,&rows);CHKERRQ(ierr);
/*
loop over local elements
*/
for (j=xs; j<xs+xn; j++) {
for (l=0; l<n; l++) rows[l] = j*(n-1)+l;
ierr = MatSetValues(K,n,rows,n,rows,&A[0][0],ADD_VALUES);CHKERRQ(ierr);
}
ierr = MatAssemblyBegin(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatCreateVecs(K,&x,&b);CHKERRQ(ierr);
ierr = ComputeRhs(da,&gll,b);CHKERRQ(ierr);
/*
Replace the first and last rows/columns of the matrix with the identity to obtain the zero Dirichlet boundary conditions
*/
rows[0] = 0;
rows[1] = q*(n-1);
ierr = MatZeroRowsColumns(K,2,rows,1.0,x,b);CHKERRQ(ierr);
ierr = PetscFree(rows);CHKERRQ(ierr);
ierr = KSPCreate(PETSC_COMM_WORLD,&ksp);CHKERRQ(ierr);
ierr = KSPSetOperators(ksp,K,K);CHKERRQ(ierr);
ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
ierr = PCSetType(pc,PCLU);CHKERRQ(ierr);
ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr);
ierr = KSPSolve(ksp,b,x);CHKERRQ(ierr);
/* compute the error to the continium problem */
ierr = ComputeSolution(da,&gll,b);CHKERRQ(ierr);
ierr = VecAXPY(x,-1.0,b);CHKERRQ(ierr);
/* compute the L^2 norm of the error */
ierr = VecGetArray(x,&f);CHKERRQ(ierr);
ierr = PetscGLLIntegrate(&gll,f,&norm);CHKERRQ(ierr);
ierr = VecRestoreArray(x,&f);CHKERRQ(ierr);
norm = PetscSqrtReal(norm);
ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"L^2 norm of the error %D %g\n",n,(double)norm);CHKERRQ(ierr);
if (n > 10 && norm > 1.e-8) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_PLIB,"Slower convergence than expected");
xc = (PetscReal)n;
yc = PetscLog10Real(norm);
ierr = PetscDrawLGAddPoint(lg,&xc,&yc);CHKERRQ(ierr);
ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
ierr = VecDestroy(&b);CHKERRQ(ierr);
ierr = VecDestroy(&x);CHKERRQ(ierr);
ierr = KSPDestroy(&ksp);CHKERRQ(ierr);
ierr = MatDestroy(&K);CHKERRQ(ierr);
ierr = PetscGLLElementLaplacianDestroy(&gll,&A);CHKERRQ(ierr);
ierr = PetscGLLDestroy(&gll);CHKERRQ(ierr);
ierr = DMDestroy(&da);CHKERRQ(ierr);
}
ierr = PetscDrawLGDestroy(&lg);CHKERRQ(ierr);
ierr = PetscDrawDestroy(&draw);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
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;
}
