15c6c1daeSBarry Smith 25ccc9f3cSMatthew G Knepley #include <petscviewer.h> 3e118a51fSLisandro Dalcin #include <../src/sys/classes/draw/utils/lgimpl.h> /*I "petscdraw.h" I*/ 45c6c1daeSBarry Smith PetscClassId PETSC_DRAWLG_CLASSID = 0; 55c6c1daeSBarry Smith 65c6c1daeSBarry Smith /*@ 75c6c1daeSBarry Smith PetscDrawLGGetAxis - Gets the axis context associated with a line graph. 85c6c1daeSBarry Smith This is useful if one wants to change some axis property, such as 95c6c1daeSBarry Smith labels, color, etc. The axis context should not be destroyed by the 105c6c1daeSBarry Smith application code. 115c6c1daeSBarry Smith 125c6c1daeSBarry Smith Not Collective, if PetscDrawLG is parallel then PetscDrawAxis is parallel 135c6c1daeSBarry Smith 145c6c1daeSBarry Smith Input Parameter: 155c6c1daeSBarry Smith . lg - the line graph context 165c6c1daeSBarry Smith 175c6c1daeSBarry Smith Output Parameter: 185c6c1daeSBarry Smith . axis - the axis context 195c6c1daeSBarry Smith 205c6c1daeSBarry Smith Level: advanced 215c6c1daeSBarry Smith 22ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawAxis 23ba1e01c4SBarry Smith 245c6c1daeSBarry Smith @*/ 255c6c1daeSBarry Smith PetscErrorCode PetscDrawLGGetAxis(PetscDrawLG lg,PetscDrawAxis *axis) 265c6c1daeSBarry Smith { 275c6c1daeSBarry Smith PetscFunctionBegin; 28e118a51fSLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 2945f3bb6eSLisandro Dalcin PetscValidPointer(axis,2); 305c6c1daeSBarry Smith *axis = lg->axis; 315c6c1daeSBarry Smith PetscFunctionReturn(0); 325c6c1daeSBarry Smith } 335c6c1daeSBarry Smith 345c6c1daeSBarry Smith /*@ 355c6c1daeSBarry Smith PetscDrawLGGetDraw - Gets the draw context associated with a line graph. 365c6c1daeSBarry Smith 375c6c1daeSBarry Smith Not Collective, if PetscDrawLG is parallel then PetscDraw is parallel 385c6c1daeSBarry Smith 395c6c1daeSBarry Smith Input Parameter: 405c6c1daeSBarry Smith . lg - the line graph context 415c6c1daeSBarry Smith 425c6c1daeSBarry Smith Output Parameter: 435c6c1daeSBarry Smith . draw - the draw context 445c6c1daeSBarry Smith 455c6c1daeSBarry Smith Level: intermediate 465c6c1daeSBarry Smith 47ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDraw 485c6c1daeSBarry Smith @*/ 495c6c1daeSBarry Smith PetscErrorCode PetscDrawLGGetDraw(PetscDrawLG lg,PetscDraw *draw) 505c6c1daeSBarry Smith { 515c6c1daeSBarry Smith PetscFunctionBegin; 525c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 5345f3bb6eSLisandro Dalcin PetscValidPointer(draw,2); 545c6c1daeSBarry Smith *draw = lg->win; 555c6c1daeSBarry Smith PetscFunctionReturn(0); 565c6c1daeSBarry Smith } 575c6c1daeSBarry Smith 585c6c1daeSBarry Smith 595c6c1daeSBarry Smith /*@ 605c6c1daeSBarry Smith PetscDrawLGSPDraw - Redraws a line graph. 615c6c1daeSBarry Smith 625b399a63SLisandro Dalcin Collective on PetscDrawLG 635c6c1daeSBarry Smith 645c6c1daeSBarry Smith Input Parameter: 655c6c1daeSBarry Smith . lg - the line graph context 665c6c1daeSBarry Smith 675c6c1daeSBarry Smith Level: intermediate 685c6c1daeSBarry Smith 695c6c1daeSBarry Smith .seealso: PetscDrawLGDraw(), PetscDrawSPDraw() 705c6c1daeSBarry Smith 71*95452b02SPatrick Sanan Developer Notes: 72*95452b02SPatrick Sanan This code cheats and uses the fact that the LG and SP structs are the same 735c6c1daeSBarry Smith 745c6c1daeSBarry Smith @*/ 755c6c1daeSBarry Smith PetscErrorCode PetscDrawLGSPDraw(PetscDrawLG lg,PetscDrawSP spin) 765c6c1daeSBarry Smith { 775c6c1daeSBarry Smith PetscDrawLG sp = (PetscDrawLG)spin; 785c6c1daeSBarry Smith PetscReal xmin,xmax,ymin,ymax; 795c6c1daeSBarry Smith PetscErrorCode ierr; 80e118a51fSLisandro Dalcin PetscBool isnull; 81e118a51fSLisandro Dalcin PetscMPIInt rank; 82e118a51fSLisandro Dalcin PetscDraw draw; 835c6c1daeSBarry Smith 845c6c1daeSBarry Smith PetscFunctionBegin; 855c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 865c6c1daeSBarry Smith PetscValidHeaderSpecific(sp,PETSC_DRAWSP_CLASSID,2); 875b399a63SLisandro Dalcin ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr); 88e118a51fSLisandro Dalcin if (isnull) PetscFunctionReturn(0); 89e118a51fSLisandro Dalcin ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)lg),&rank);CHKERRQ(ierr); 905c6c1daeSBarry Smith 915b399a63SLisandro Dalcin draw = lg->win; 92e118a51fSLisandro Dalcin ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr); 935b399a63SLisandro Dalcin ierr = PetscDrawClear(draw);CHKERRQ(ierr); 94e118a51fSLisandro Dalcin 95e118a51fSLisandro Dalcin xmin = PetscMin(lg->xmin,sp->xmin); ymin = PetscMin(lg->ymin,sp->ymin); 96e118a51fSLisandro Dalcin xmax = PetscMax(lg->xmax,sp->xmax); ymax = PetscMax(lg->ymax,sp->ymax); 975c6c1daeSBarry Smith ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr); 985c6c1daeSBarry Smith ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr); 995c6c1daeSBarry Smith 1005b399a63SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 1015c6c1daeSBarry Smith if (!rank) { 102e118a51fSLisandro Dalcin int i,j,dim,nopts; 1035c6c1daeSBarry Smith dim = lg->dim; 1045c6c1daeSBarry Smith nopts = lg->nopts; 1055c6c1daeSBarry Smith for (i=0; i<dim; i++) { 1065c6c1daeSBarry Smith for (j=1; j<nopts; j++) { 1075c6c1daeSBarry Smith 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],PETSC_DRAW_BLACK+i);CHKERRQ(ierr); 108b6fe0379SLisandro Dalcin if (lg->use_markers) { 10978e9f83eSLisandro Dalcin ierr = PetscDrawMarker(draw,lg->x[j*dim+i],lg->y[j*dim+i],PETSC_DRAW_RED);CHKERRQ(ierr); 1105c6c1daeSBarry Smith } 1115c6c1daeSBarry Smith } 1125c6c1daeSBarry Smith } 1135c6c1daeSBarry Smith dim = sp->dim; 1145c6c1daeSBarry Smith nopts = sp->nopts; 1155c6c1daeSBarry Smith for (i=0; i<dim; i++) { 1165c6c1daeSBarry Smith for (j=0; j<nopts; j++) { 11778e9f83eSLisandro Dalcin ierr = PetscDrawMarker(draw,sp->x[j*dim+i],sp->y[j*dim+i],PETSC_DRAW_RED);CHKERRQ(ierr); 1185c6c1daeSBarry Smith } 1195c6c1daeSBarry Smith } 1205c6c1daeSBarry Smith } 121e118a51fSLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 1225b399a63SLisandro Dalcin 1235b399a63SLisandro Dalcin ierr = PetscDrawFlush(draw);CHKERRQ(ierr); 1245b399a63SLisandro Dalcin ierr = PetscDrawPause(draw);CHKERRQ(ierr); 1255c6c1daeSBarry Smith PetscFunctionReturn(0); 1265c6c1daeSBarry Smith } 1275c6c1daeSBarry Smith 1285c6c1daeSBarry Smith 1295c6c1daeSBarry Smith /*@ 1305c6c1daeSBarry Smith PetscDrawLGCreate - Creates a line graph data structure. 1315c6c1daeSBarry Smith 1325b399a63SLisandro Dalcin Collective on PetscDraw 1335c6c1daeSBarry Smith 1345c6c1daeSBarry Smith Input Parameters: 1355c6c1daeSBarry Smith + draw - the window where the graph will be made. 1365c6c1daeSBarry Smith - dim - the number of curves which will be drawn 1375c6c1daeSBarry Smith 1385c6c1daeSBarry Smith Output Parameters: 139e118a51fSLisandro Dalcin . outlg - the line graph context 1405c6c1daeSBarry Smith 1415c6c1daeSBarry Smith Level: intermediate 1425c6c1daeSBarry Smith 143*95452b02SPatrick Sanan Notes: 144*95452b02SPatrick Sanan The MPI communicator that owns the PetscDraw owns this PetscDrawLG, but the calls to set options and add points are ignored on all processes except the 1457e25d57eSBarry Smith zeroth MPI process in the communicator. All MPI processes in the communicator must call PetscDrawLGDraw() to display the updated graph. 1467e25d57eSBarry Smith 1475c6c1daeSBarry Smith Concepts: line graph^creating 1485c6c1daeSBarry Smith 149ba1e01c4SBarry Smith .seealso: PetscDrawLGDestroy(), PetscDrawLGAddPoint(), PetscDrawLGAddCommonPoint(), PetscDrawLGAddPoints(), PetscDrawLGDraw(), PetscDrawLGSave(), 150ba1e01c4SBarry Smith PetscDrawLGView(), PetscDrawLGReset(), PetscDrawLGSetDimension(), PetscDrawLGGetDimension(), PetscDrawLGSetLegend(), PetscDrawLGGetAxis(), 151ba1e01c4SBarry Smith PetscDrawLGGetDraw(), PetscDrawLGSetUseMarkers(), PetscDrawLGSetLimits(), PetscDrawLGSetColors(), PetscDrawLGSetOptionsPrefix(), PetscDrawLGSetFromOptions() 1525c6c1daeSBarry Smith @*/ 153e118a51fSLisandro Dalcin PetscErrorCode PetscDrawLGCreate(PetscDraw draw,PetscInt dim,PetscDrawLG *outlg) 1545c6c1daeSBarry Smith { 1555c6c1daeSBarry Smith PetscDrawLG lg; 156e118a51fSLisandro Dalcin PetscErrorCode ierr; 1575c6c1daeSBarry Smith 1585c6c1daeSBarry Smith PetscFunctionBegin; 1595c6c1daeSBarry Smith PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1); 160e118a51fSLisandro Dalcin PetscValidLogicalCollectiveInt(draw,dim,2); 161e118a51fSLisandro Dalcin PetscValidPointer(outlg,3); 162e118a51fSLisandro Dalcin 16371917b75SLisandro Dalcin ierr = PetscHeaderCreate(lg,PETSC_DRAWLG_CLASSID,"DrawLG","Line Graph","Draw",PetscObjectComm((PetscObject)draw),PetscDrawLGDestroy,NULL);CHKERRQ(ierr); 164e118a51fSLisandro Dalcin ierr = PetscLogObjectParent((PetscObject)draw,(PetscObject)lg);CHKERRQ(ierr); 165c2bac407SLisandro Dalcin ierr = PetscDrawLGSetOptionsPrefix(lg,((PetscObject)draw)->prefix);CHKERRQ(ierr); 166e118a51fSLisandro Dalcin 167e118a51fSLisandro Dalcin ierr = PetscObjectReference((PetscObject)draw);CHKERRQ(ierr); 168e118a51fSLisandro Dalcin lg->win = draw; 169a297a907SKarl Rupp 17069c83917SLisandro Dalcin lg->view = NULL; 17169c83917SLisandro Dalcin lg->destroy = NULL; 1725c6c1daeSBarry Smith lg->nopts = 0; 1735c6c1daeSBarry Smith lg->dim = dim; 1745c6c1daeSBarry Smith lg->xmin = 1.e20; 1755c6c1daeSBarry Smith lg->ymin = 1.e20; 1765c6c1daeSBarry Smith lg->xmax = -1.e20; 1775c6c1daeSBarry Smith lg->ymax = -1.e20; 178a297a907SKarl Rupp 179dcca6d9dSJed Brown ierr = PetscMalloc2(dim*CHUNCKSIZE,&lg->x,dim*CHUNCKSIZE,&lg->y);CHKERRQ(ierr); 1803bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)lg,2*dim*CHUNCKSIZE*sizeof(PetscReal));CHKERRQ(ierr); 181a297a907SKarl Rupp 1825c6c1daeSBarry Smith lg->len = dim*CHUNCKSIZE; 1835c6c1daeSBarry Smith lg->loc = 0; 184b6fe0379SLisandro Dalcin lg->use_markers = PETSC_FALSE; 185a297a907SKarl Rupp 1865c6c1daeSBarry Smith ierr = PetscDrawAxisCreate(draw,&lg->axis);CHKERRQ(ierr); 1873bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)lg,(PetscObject)lg->axis);CHKERRQ(ierr); 188a297a907SKarl Rupp 189e118a51fSLisandro Dalcin *outlg = lg; 1905c6c1daeSBarry Smith PetscFunctionReturn(0); 1915c6c1daeSBarry Smith } 1925c6c1daeSBarry Smith 1935c6c1daeSBarry Smith /*@ 1945c6c1daeSBarry Smith PetscDrawLGSetColors - Sets the color of each line graph drawn 1955c6c1daeSBarry Smith 1965b399a63SLisandro Dalcin Logically Collective on PetscDrawLG 1975c6c1daeSBarry Smith 1985c6c1daeSBarry Smith Input Parameter: 1995c6c1daeSBarry Smith + lg - the line graph context. 2005c6c1daeSBarry Smith - colors - the colors 2015c6c1daeSBarry Smith 2025c6c1daeSBarry Smith Level: intermediate 2035c6c1daeSBarry Smith 2045c6c1daeSBarry Smith Concepts: line graph^setting number of lines 2055c6c1daeSBarry Smith 206ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate() 207ba1e01c4SBarry Smith 2085c6c1daeSBarry Smith @*/ 20945f3bb6eSLisandro Dalcin PetscErrorCode PetscDrawLGSetColors(PetscDrawLG lg,const int colors[]) 2105c6c1daeSBarry Smith { 2115c6c1daeSBarry Smith PetscErrorCode ierr; 2125c6c1daeSBarry Smith 2135c6c1daeSBarry Smith PetscFunctionBegin; 2145c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 21545f3bb6eSLisandro Dalcin if (lg->dim) PetscValidIntPointer(colors,2); 216e118a51fSLisandro Dalcin 2175c6c1daeSBarry Smith ierr = PetscFree(lg->colors);CHKERRQ(ierr); 218785e854fSJed Brown ierr = PetscMalloc1(lg->dim,&lg->colors);CHKERRQ(ierr); 2195c6c1daeSBarry Smith ierr = PetscMemcpy(lg->colors,colors,lg->dim*sizeof(int));CHKERRQ(ierr); 2205c6c1daeSBarry Smith PetscFunctionReturn(0); 2215c6c1daeSBarry Smith } 2225c6c1daeSBarry Smith 2235c6c1daeSBarry Smith /*@C 2245c6c1daeSBarry Smith PetscDrawLGSetLegend - sets the names of each curve plotted 2255c6c1daeSBarry Smith 2265b399a63SLisandro Dalcin Logically Collective on PetscDrawLG 2275c6c1daeSBarry Smith 2285c6c1daeSBarry Smith Input Parameter: 2295c6c1daeSBarry Smith + lg - the line graph context. 2305c6c1daeSBarry Smith - names - the names for each curve 2315c6c1daeSBarry Smith 2325c6c1daeSBarry Smith Level: intermediate 2335c6c1daeSBarry Smith 234*95452b02SPatrick Sanan Notes: 235*95452b02SPatrick Sanan Call PetscDrawLGGetAxis() and then change properties of the PetscDrawAxis for detailed control of the plot 236ba1e01c4SBarry Smith 2375c6c1daeSBarry Smith Concepts: line graph^setting number of lines 2385c6c1daeSBarry Smith 239ba1e01c4SBarry Smith .seealso: PetscDrawLGGetAxis(), PetscDrawAxis, PetscDrawAxisSetColors(), PetscDrawAxisSetLabels(), PetscDrawAxisSetHoldLimits() 240ba1e01c4SBarry Smith 2415c6c1daeSBarry Smith @*/ 2425c6c1daeSBarry Smith PetscErrorCode PetscDrawLGSetLegend(PetscDrawLG lg,const char *const *names) 2435c6c1daeSBarry Smith { 2445c6c1daeSBarry Smith PetscErrorCode ierr; 2455c6c1daeSBarry Smith PetscInt i; 2465c6c1daeSBarry Smith 2475c6c1daeSBarry Smith PetscFunctionBegin; 2485c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 24945f3bb6eSLisandro Dalcin if (names) PetscValidPointer(names,2); 2505c6c1daeSBarry Smith 2515c6c1daeSBarry Smith if (lg->legend) { 2525c6c1daeSBarry Smith for (i=0; i<lg->dim; i++) { 2535c6c1daeSBarry Smith ierr = PetscFree(lg->legend[i]);CHKERRQ(ierr); 2545c6c1daeSBarry Smith } 2555c6c1daeSBarry Smith ierr = PetscFree(lg->legend);CHKERRQ(ierr); 2565c6c1daeSBarry Smith } 2575c6c1daeSBarry Smith if (names) { 258785e854fSJed Brown ierr = PetscMalloc1(lg->dim,&lg->legend);CHKERRQ(ierr); 2595c6c1daeSBarry Smith for (i=0; i<lg->dim; i++) { 2605c6c1daeSBarry Smith ierr = PetscStrallocpy(names[i],&lg->legend[i]);CHKERRQ(ierr); 2615c6c1daeSBarry Smith } 2625c6c1daeSBarry Smith } 2635c6c1daeSBarry Smith PetscFunctionReturn(0); 2645c6c1daeSBarry Smith } 2655c6c1daeSBarry Smith 2665c6c1daeSBarry Smith /*@ 2675c6c1daeSBarry Smith PetscDrawLGGetDimension - Change the number of lines that are to be drawn. 2685c6c1daeSBarry Smith 2695b399a63SLisandro Dalcin Not Collective 2705c6c1daeSBarry Smith 2715c6c1daeSBarry Smith Input Parameter: 2725c6c1daeSBarry Smith . lg - the line graph context. 2735c6c1daeSBarry Smith 2745c6c1daeSBarry Smith Output Parameter: 2755c6c1daeSBarry Smith . dim - the number of curves. 2765c6c1daeSBarry Smith 2775c6c1daeSBarry Smith Level: intermediate 2785c6c1daeSBarry Smith 2795c6c1daeSBarry Smith Concepts: line graph^setting number of lines 2805c6c1daeSBarry Smith 281ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawLGSetDimension() 282ba1e01c4SBarry Smith 2835c6c1daeSBarry Smith @*/ 2845c6c1daeSBarry Smith PetscErrorCode PetscDrawLGGetDimension(PetscDrawLG lg,PetscInt *dim) 2855c6c1daeSBarry Smith { 2865c6c1daeSBarry Smith PetscFunctionBegin; 2875c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 28845f3bb6eSLisandro Dalcin PetscValidIntPointer(dim,2); 2895c6c1daeSBarry Smith *dim = lg->dim; 2905c6c1daeSBarry Smith PetscFunctionReturn(0); 2915c6c1daeSBarry Smith } 2925c6c1daeSBarry Smith 2935c6c1daeSBarry Smith /*@ 2945c6c1daeSBarry Smith PetscDrawLGSetDimension - Change the number of lines that are to be drawn. 2955c6c1daeSBarry Smith 2965b399a63SLisandro Dalcin Logically Collective on PetscDrawLG 2975c6c1daeSBarry Smith 2985c6c1daeSBarry Smith Input Parameter: 2995c6c1daeSBarry Smith + lg - the line graph context. 3005c6c1daeSBarry Smith - dim - the number of curves. 3015c6c1daeSBarry Smith 3025c6c1daeSBarry Smith Level: intermediate 3035c6c1daeSBarry Smith 3045c6c1daeSBarry Smith Concepts: line graph^setting number of lines 3055c6c1daeSBarry Smith 306ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawLGGetDimension() 3075c6c1daeSBarry Smith @*/ 3085c6c1daeSBarry Smith PetscErrorCode PetscDrawLGSetDimension(PetscDrawLG lg,PetscInt dim) 3095c6c1daeSBarry Smith { 3105c6c1daeSBarry Smith PetscErrorCode ierr; 3115c6c1daeSBarry Smith PetscInt i; 3125c6c1daeSBarry Smith 3135c6c1daeSBarry Smith PetscFunctionBegin; 3145c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 3155c6c1daeSBarry Smith PetscValidLogicalCollectiveInt(lg,dim,2); 3165c6c1daeSBarry Smith if (lg->dim == dim) PetscFunctionReturn(0); 3175c6c1daeSBarry Smith 3185c6c1daeSBarry Smith ierr = PetscFree2(lg->x,lg->y);CHKERRQ(ierr); 3195c6c1daeSBarry Smith if (lg->legend) { 3205c6c1daeSBarry Smith for (i=0; i<lg->dim; i++) { 3215c6c1daeSBarry Smith ierr = PetscFree(lg->legend[i]);CHKERRQ(ierr); 3225c6c1daeSBarry Smith } 3235c6c1daeSBarry Smith ierr = PetscFree(lg->legend);CHKERRQ(ierr); 3245c6c1daeSBarry Smith } 3255c6c1daeSBarry Smith ierr = PetscFree(lg->colors);CHKERRQ(ierr); 3265c6c1daeSBarry Smith lg->dim = dim; 327dcca6d9dSJed Brown ierr = PetscMalloc2(dim*CHUNCKSIZE,&lg->x,dim*CHUNCKSIZE,&lg->y);CHKERRQ(ierr); 3283bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)lg,2*dim*CHUNCKSIZE*sizeof(PetscReal));CHKERRQ(ierr); 3295c6c1daeSBarry Smith lg->len = dim*CHUNCKSIZE; 3305c6c1daeSBarry Smith PetscFunctionReturn(0); 3315c6c1daeSBarry Smith } 3325c6c1daeSBarry Smith 33371917b75SLisandro Dalcin 33471917b75SLisandro Dalcin /*@ 33571917b75SLisandro Dalcin PetscDrawLGSetLimits - Sets the axis limits for a line graph. If more 33671917b75SLisandro Dalcin points are added after this call, the limits will be adjusted to 33771917b75SLisandro Dalcin include those additional points. 33871917b75SLisandro Dalcin 33971917b75SLisandro Dalcin Logically Collective on PetscDrawLG 34071917b75SLisandro Dalcin 34171917b75SLisandro Dalcin Input Parameters: 34271917b75SLisandro Dalcin + xlg - the line graph context 34371917b75SLisandro Dalcin - x_min,x_max,y_min,y_max - the limits 34471917b75SLisandro Dalcin 34571917b75SLisandro Dalcin Level: intermediate 34671917b75SLisandro Dalcin 34771917b75SLisandro Dalcin Concepts: line graph^setting axis 34871917b75SLisandro Dalcin 349ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate() 350ba1e01c4SBarry Smith 35171917b75SLisandro Dalcin @*/ 35271917b75SLisandro Dalcin PetscErrorCode PetscDrawLGSetLimits(PetscDrawLG lg,PetscReal x_min,PetscReal x_max,PetscReal y_min,PetscReal y_max) 35371917b75SLisandro Dalcin { 35471917b75SLisandro Dalcin PetscFunctionBegin; 35571917b75SLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 35671917b75SLisandro Dalcin 35771917b75SLisandro Dalcin (lg)->xmin = x_min; 35871917b75SLisandro Dalcin (lg)->xmax = x_max; 35971917b75SLisandro Dalcin (lg)->ymin = y_min; 36071917b75SLisandro Dalcin (lg)->ymax = y_max; 36171917b75SLisandro Dalcin PetscFunctionReturn(0); 36271917b75SLisandro Dalcin } 36371917b75SLisandro Dalcin 3645c6c1daeSBarry Smith /*@ 3655c6c1daeSBarry Smith PetscDrawLGReset - Clears line graph to allow for reuse with new data. 3665c6c1daeSBarry Smith 3675b399a63SLisandro Dalcin Logically Collective on PetscDrawLG 3685c6c1daeSBarry Smith 3695c6c1daeSBarry Smith Input Parameter: 3705c6c1daeSBarry Smith . lg - the line graph context. 3715c6c1daeSBarry Smith 3725c6c1daeSBarry Smith Level: intermediate 3735c6c1daeSBarry Smith 3745c6c1daeSBarry Smith Concepts: line graph^restarting 3755c6c1daeSBarry Smith 376ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate() 377ba1e01c4SBarry Smith 3785c6c1daeSBarry Smith @*/ 3795c6c1daeSBarry Smith PetscErrorCode PetscDrawLGReset(PetscDrawLG lg) 3805c6c1daeSBarry Smith { 3815c6c1daeSBarry Smith PetscFunctionBegin; 3825c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 3835c6c1daeSBarry Smith lg->xmin = 1.e20; 3845c6c1daeSBarry Smith lg->ymin = 1.e20; 3855c6c1daeSBarry Smith lg->xmax = -1.e20; 3865c6c1daeSBarry Smith lg->ymax = -1.e20; 3875c6c1daeSBarry Smith lg->loc = 0; 3885c6c1daeSBarry Smith lg->nopts = 0; 3895c6c1daeSBarry Smith PetscFunctionReturn(0); 3905c6c1daeSBarry Smith } 3915c6c1daeSBarry Smith 3925c6c1daeSBarry Smith /*@ 3935c6c1daeSBarry Smith PetscDrawLGDestroy - Frees all space taken up by line graph data structure. 3945c6c1daeSBarry Smith 3955b399a63SLisandro Dalcin Collective on PetscDrawLG 3965c6c1daeSBarry Smith 3975c6c1daeSBarry Smith Input Parameter: 3985c6c1daeSBarry Smith . lg - the line graph context 3995c6c1daeSBarry Smith 4005c6c1daeSBarry Smith Level: intermediate 4015c6c1daeSBarry Smith 4025c6c1daeSBarry Smith .seealso: PetscDrawLGCreate() 4035c6c1daeSBarry Smith @*/ 4045c6c1daeSBarry Smith PetscErrorCode PetscDrawLGDestroy(PetscDrawLG *lg) 4055c6c1daeSBarry Smith { 4065c6c1daeSBarry Smith PetscErrorCode ierr; 4075c6c1daeSBarry Smith PetscInt i; 4085c6c1daeSBarry Smith 4095c6c1daeSBarry Smith PetscFunctionBegin; 4105c6c1daeSBarry Smith if (!*lg) PetscFunctionReturn(0); 411e118a51fSLisandro Dalcin PetscValidHeaderSpecific(*lg,PETSC_DRAWLG_CLASSID,1); 412e118a51fSLisandro Dalcin if (--((PetscObject)(*lg))->refct > 0) {*lg = NULL; PetscFunctionReturn(0);} 4135c6c1daeSBarry Smith 4145c6c1daeSBarry Smith if ((*lg)->legend) { 4155c6c1daeSBarry Smith for (i=0; i<(*lg)->dim; i++) { 4165c6c1daeSBarry Smith ierr = PetscFree((*lg)->legend[i]);CHKERRQ(ierr); 4175c6c1daeSBarry Smith } 4185c6c1daeSBarry Smith ierr = PetscFree((*lg)->legend);CHKERRQ(ierr); 4195c6c1daeSBarry Smith } 4205c6c1daeSBarry Smith ierr = PetscFree((*lg)->colors);CHKERRQ(ierr); 4215c6c1daeSBarry Smith ierr = PetscFree2((*lg)->x,(*lg)->y);CHKERRQ(ierr); 422e118a51fSLisandro Dalcin ierr = PetscDrawAxisDestroy(&(*lg)->axis);CHKERRQ(ierr); 423e118a51fSLisandro Dalcin ierr = PetscDrawDestroy(&(*lg)->win);CHKERRQ(ierr); 4245c6c1daeSBarry Smith ierr = PetscHeaderDestroy(lg);CHKERRQ(ierr); 4255c6c1daeSBarry Smith PetscFunctionReturn(0); 4265c6c1daeSBarry Smith } 4275c6c1daeSBarry Smith /*@ 428b6fe0379SLisandro Dalcin PetscDrawLGSetUseMarkers - Causes LG to draw a marker for each data-point. 4295c6c1daeSBarry Smith 4305b399a63SLisandro Dalcin Logically Collective on PetscDrawLG 4315c6c1daeSBarry Smith 4325c6c1daeSBarry Smith Input Parameters: 433287de1a7SBarry Smith + lg - the linegraph context 434287de1a7SBarry Smith - flg - should mark each data point 435287de1a7SBarry Smith 436287de1a7SBarry Smith Options Database: 437b6fe0379SLisandro Dalcin . -lg_use_markers <true,false> 4385c6c1daeSBarry Smith 4395c6c1daeSBarry Smith Level: intermediate 4405c6c1daeSBarry Smith 4415c6c1daeSBarry Smith Concepts: line graph^showing points 4425c6c1daeSBarry Smith 443ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate() 444ba1e01c4SBarry Smith 4455c6c1daeSBarry Smith @*/ 446b6fe0379SLisandro Dalcin PetscErrorCode PetscDrawLGSetUseMarkers(PetscDrawLG lg,PetscBool flg) 4475c6c1daeSBarry Smith { 4485c6c1daeSBarry Smith PetscFunctionBegin; 449e118a51fSLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 45045f3bb6eSLisandro Dalcin PetscValidLogicalCollectiveBool(lg,flg,2); 451b6fe0379SLisandro Dalcin lg->use_markers = flg; 4525c6c1daeSBarry Smith PetscFunctionReturn(0); 4535c6c1daeSBarry Smith } 4545c6c1daeSBarry Smith 4555c6c1daeSBarry Smith /*@ 4565c6c1daeSBarry Smith PetscDrawLGDraw - Redraws a line graph. 4575c6c1daeSBarry Smith 4585b399a63SLisandro Dalcin Collective on PetscDrawLG 4595c6c1daeSBarry Smith 4605c6c1daeSBarry Smith Input Parameter: 4615c6c1daeSBarry Smith . lg - the line graph context 4625c6c1daeSBarry Smith 4635c6c1daeSBarry Smith Level: intermediate 4645c6c1daeSBarry Smith 465ba1e01c4SBarry Smith .seealso: PetscDrawSPDraw(), PetscDrawLGSPDraw(), PetscDrawLGReset() 4665c6c1daeSBarry Smith 4675c6c1daeSBarry Smith @*/ 4685c6c1daeSBarry Smith PetscErrorCode PetscDrawLGDraw(PetscDrawLG lg) 4695c6c1daeSBarry Smith { 470e118a51fSLisandro Dalcin PetscReal xmin,xmax,ymin,ymax; 4715c6c1daeSBarry Smith PetscErrorCode ierr; 472e118a51fSLisandro Dalcin PetscMPIInt rank; 473e118a51fSLisandro Dalcin PetscDraw draw; 4745c6c1daeSBarry Smith PetscBool isnull; 4755c6c1daeSBarry Smith 4765c6c1daeSBarry Smith PetscFunctionBegin; 4775c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 4788f69470aSLisandro Dalcin ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr); 4798f69470aSLisandro Dalcin if (isnull) PetscFunctionReturn(0); 480e118a51fSLisandro Dalcin ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)lg),&rank);CHKERRQ(ierr); 4815c6c1daeSBarry Smith 4825b399a63SLisandro Dalcin draw = lg->win; 4835c6c1daeSBarry Smith ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr); 4845b399a63SLisandro Dalcin ierr = PetscDrawClear(draw);CHKERRQ(ierr); 485e118a51fSLisandro Dalcin 486e118a51fSLisandro Dalcin xmin = lg->xmin; xmax = lg->xmax; ymin = lg->ymin; ymax = lg->ymax; 4875c6c1daeSBarry Smith ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr); 4885c6c1daeSBarry Smith ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr); 4895c6c1daeSBarry Smith 4905b399a63SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 4915c6c1daeSBarry Smith if (!rank) { 492e118a51fSLisandro Dalcin int i,j,dim=lg->dim,nopts=lg->nopts,cl; 4935c6c1daeSBarry Smith for (i=0; i<dim; i++) { 4945c6c1daeSBarry Smith for (j=1; j<nopts; j++) { 4950ed3bfb6SBarry Smith cl = lg->colors ? lg->colors[i] : ((PETSC_DRAW_BLACK + i) % PETSC_DRAW_MAXCOLOR); 4965c6c1daeSBarry Smith 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); 49771917b75SLisandro Dalcin if (lg->use_markers) {ierr = PetscDrawMarker(draw,lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);} 4985c6c1daeSBarry Smith } 4995c6c1daeSBarry Smith } 5005c6c1daeSBarry Smith } 50156a02f31SLisandro Dalcin if (!rank && lg->legend) { 502e118a51fSLisandro Dalcin int i,dim=lg->dim,cl; 5035c6c1daeSBarry Smith PetscReal xl,yl,xr,yr,tw,th; 50471917b75SLisandro Dalcin size_t slen,len=0; 50571917b75SLisandro Dalcin ierr = PetscDrawAxisGetLimits(lg->axis,&xl,&xr,&yl,&yr);CHKERRQ(ierr); 5065c6c1daeSBarry Smith ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr); 5075c6c1daeSBarry Smith for (i=0; i<dim; i++) { 50871917b75SLisandro Dalcin ierr = PetscStrlen(lg->legend[i],&slen);CHKERRQ(ierr); 50971917b75SLisandro Dalcin len = PetscMax(len,slen); 5105c6c1daeSBarry Smith } 51171917b75SLisandro Dalcin xr = xr - 1.5*tw; xl = xr - (len + 7)*tw; 51271917b75SLisandro Dalcin yr = yr - 1.0*th; yl = yr - (dim + 1)*th; 51371917b75SLisandro Dalcin ierr = PetscDrawLine(draw,xl,yl,xr,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr); 51471917b75SLisandro Dalcin ierr = PetscDrawLine(draw,xr,yl,xr,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr); 51571917b75SLisandro Dalcin ierr = PetscDrawLine(draw,xr,yr,xl,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr); 51671917b75SLisandro Dalcin ierr = PetscDrawLine(draw,xl,yr,xl,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr); 5175c6c1daeSBarry Smith for (i=0; i<dim; i++) { 51871917b75SLisandro Dalcin cl = lg->colors ? lg->colors[i] : (PETSC_DRAW_BLACK + i); 51971917b75SLisandro Dalcin ierr = PetscDrawLine(draw,xl + 1*tw,yr - (i + 1)*th,xl + 5*tw,yr - (i + 1)*th,cl);CHKERRQ(ierr); 52071917b75SLisandro Dalcin ierr = PetscDrawString(draw,xl + 6*tw,yr - (i + 1.5)*th,PETSC_DRAW_BLACK,lg->legend[i]);CHKERRQ(ierr); 5215c6c1daeSBarry Smith } 5225c6c1daeSBarry Smith } 523e118a51fSLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 5245b399a63SLisandro Dalcin 5255b399a63SLisandro Dalcin ierr = PetscDrawFlush(draw);CHKERRQ(ierr); 526e118a51fSLisandro Dalcin ierr = PetscDrawPause(draw);CHKERRQ(ierr); 5275c6c1daeSBarry Smith PetscFunctionReturn(0); 5285c6c1daeSBarry Smith } 5295c6c1daeSBarry Smith 53057fd6651SLisandro Dalcin /*@ 53157fd6651SLisandro Dalcin PetscDrawLGSave - Saves a drawn image 53257fd6651SLisandro Dalcin 53357fd6651SLisandro Dalcin Collective on PetscDrawLG 53457fd6651SLisandro Dalcin 53557fd6651SLisandro Dalcin Input Parameter: 53657fd6651SLisandro Dalcin . lg - The line graph context 53757fd6651SLisandro Dalcin 53857fd6651SLisandro Dalcin Level: intermediate 53957fd6651SLisandro Dalcin 54057fd6651SLisandro Dalcin Concepts: line graph^saving 54157fd6651SLisandro Dalcin 54257fd6651SLisandro Dalcin .seealso: PetscDrawLGCreate(), PetscDrawLGGetDraw(), PetscDrawSetSave(), PetscDrawSave() 54357fd6651SLisandro Dalcin @*/ 54457fd6651SLisandro Dalcin PetscErrorCode PetscDrawLGSave(PetscDrawLG lg) 54557fd6651SLisandro Dalcin { 54657fd6651SLisandro Dalcin PetscErrorCode ierr; 54757fd6651SLisandro Dalcin 54857fd6651SLisandro Dalcin PetscFunctionBegin; 54957fd6651SLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 55057fd6651SLisandro Dalcin ierr = PetscDrawSave(lg->win);CHKERRQ(ierr); 55157fd6651SLisandro Dalcin PetscFunctionReturn(0); 55257fd6651SLisandro Dalcin } 55357fd6651SLisandro Dalcin 5545c6c1daeSBarry Smith /*@ 55534a5a0e3SBarry Smith PetscDrawLGView - Prints a line graph. 5565c6c1daeSBarry Smith 5575b399a63SLisandro Dalcin Collective on PetscDrawLG 5585c6c1daeSBarry Smith 5595c6c1daeSBarry Smith Input Parameter: 5605c6c1daeSBarry Smith . lg - the line graph context 5615c6c1daeSBarry Smith 5625c6c1daeSBarry Smith Level: beginner 5635c6c1daeSBarry Smith 564ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate() 565ba1e01c4SBarry Smith 5665c6c1daeSBarry Smith .keywords: draw, line, graph 5675c6c1daeSBarry Smith @*/ 56834a5a0e3SBarry Smith PetscErrorCode PetscDrawLGView(PetscDrawLG lg,PetscViewer viewer) 5695c6c1daeSBarry Smith { 5705c6c1daeSBarry Smith PetscReal xmin=lg->xmin, xmax=lg->xmax, ymin=lg->ymin, ymax=lg->ymax; 57134a5a0e3SBarry Smith PetscInt i, j, dim = lg->dim, nopts = lg->nopts; 57234a5a0e3SBarry Smith PetscErrorCode ierr; 5735c6c1daeSBarry Smith 5745c6c1daeSBarry Smith PetscFunctionBegin; 5755c6c1daeSBarry Smith PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 576e118a51fSLisandro Dalcin 5775c6c1daeSBarry Smith if (nopts < 1) PetscFunctionReturn(0); 5785c6c1daeSBarry Smith if (xmin > xmax || ymin > ymax) PetscFunctionReturn(0); 5795c6c1daeSBarry Smith 58034a5a0e3SBarry Smith if (!viewer){ 58134a5a0e3SBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)lg),&viewer);CHKERRQ(ierr); 58234a5a0e3SBarry Smith } 58398c3331eSBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)lg,viewer);CHKERRQ(ierr); 5845c6c1daeSBarry Smith for (i = 0; i < dim; i++) { 58534a5a0e3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer, "Line %D>\n", i);CHKERRQ(ierr); 5865c6c1daeSBarry Smith for (j = 0; j < nopts; j++) { 58734a5a0e3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer, " X: %g Y: %g\n", (double)lg->x[j*dim+i], (double)lg->y[j*dim+i]);CHKERRQ(ierr); 5885c6c1daeSBarry Smith } 5895c6c1daeSBarry Smith } 5905c6c1daeSBarry Smith PetscFunctionReturn(0); 5915c6c1daeSBarry Smith } 592287de1a7SBarry Smith 593c2bac407SLisandro Dalcin /*@C 594c2bac407SLisandro Dalcin PetscDrawLGSetOptionsPrefix - Sets the prefix used for searching for all 595c2bac407SLisandro Dalcin PetscDrawLG options in the database. 596c2bac407SLisandro Dalcin 597c2bac407SLisandro Dalcin Logically Collective on PetscDrawLG 598c2bac407SLisandro Dalcin 599c2bac407SLisandro Dalcin Input Parameter: 600c2bac407SLisandro Dalcin + lg - the line graph context 601c2bac407SLisandro Dalcin - prefix - the prefix to prepend to all option names 602c2bac407SLisandro Dalcin 603c2bac407SLisandro Dalcin Level: advanced 604c2bac407SLisandro Dalcin 605c2bac407SLisandro Dalcin .keywords: PetscDrawLG, set, options, prefix, database 606c2bac407SLisandro Dalcin 607ba1e01c4SBarry Smith .seealso: PetscDrawLGSetFromOptions(), PetscDrawLGCreate() 608c2bac407SLisandro Dalcin @*/ 609c2bac407SLisandro Dalcin PetscErrorCode PetscDrawLGSetOptionsPrefix(PetscDrawLG lg,const char prefix[]) 610c2bac407SLisandro Dalcin { 611c2bac407SLisandro Dalcin PetscErrorCode ierr; 612c2bac407SLisandro Dalcin 613c2bac407SLisandro Dalcin PetscFunctionBegin; 614c2bac407SLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 615c2bac407SLisandro Dalcin ierr = PetscObjectSetOptionsPrefix((PetscObject)lg,prefix);CHKERRQ(ierr); 616c2bac407SLisandro Dalcin PetscFunctionReturn(0); 617c2bac407SLisandro Dalcin } 618c2bac407SLisandro Dalcin 619287de1a7SBarry Smith /*@ 620287de1a7SBarry Smith PetscDrawLGSetFromOptions - Sets options related to the PetscDrawLG 621287de1a7SBarry Smith 6225b399a63SLisandro Dalcin Collective on PetscDrawLG 623287de1a7SBarry Smith 624287de1a7SBarry Smith Options Database: 625287de1a7SBarry Smith 626287de1a7SBarry Smith Level: intermediate 627287de1a7SBarry Smith 628287de1a7SBarry Smith Concepts: line graph^creating 629287de1a7SBarry Smith 630287de1a7SBarry Smith .seealso: PetscDrawLGDestroy(), PetscDrawLGCreate() 631287de1a7SBarry Smith @*/ 632287de1a7SBarry Smith PetscErrorCode PetscDrawLGSetFromOptions(PetscDrawLG lg) 633287de1a7SBarry Smith { 634287de1a7SBarry Smith PetscErrorCode ierr; 63571917b75SLisandro Dalcin PetscBool usemarkers,set; 63671917b75SLisandro Dalcin PetscDrawMarkerType markertype; 637287de1a7SBarry Smith 638287de1a7SBarry Smith PetscFunctionBegin; 639e118a51fSLisandro Dalcin PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1); 640e118a51fSLisandro Dalcin 64171917b75SLisandro Dalcin ierr = PetscDrawGetMarkerType(lg->win,&markertype);CHKERRQ(ierr); 64271917b75SLisandro Dalcin ierr = PetscOptionsGetEnum(((PetscObject)lg)->options,((PetscObject)lg)->prefix,"-lg_marker_type",PetscDrawMarkerTypes,(PetscEnum*)&markertype,&set);CHKERRQ(ierr); 64371917b75SLisandro Dalcin if (set) { 64471917b75SLisandro Dalcin ierr = PetscDrawLGSetUseMarkers(lg,PETSC_TRUE);CHKERRQ(ierr); 64571917b75SLisandro Dalcin ierr = PetscDrawSetMarkerType(lg->win,markertype);CHKERRQ(ierr); 64671917b75SLisandro Dalcin } 64771917b75SLisandro Dalcin usemarkers = lg->use_markers; 64871917b75SLisandro Dalcin ierr = PetscOptionsGetBool(((PetscObject)lg)->options,((PetscObject)lg)->prefix,"-lg_use_markers",&usemarkers,&set);CHKERRQ(ierr); 64971917b75SLisandro Dalcin if (set) {ierr = PetscDrawLGSetUseMarkers(lg,usemarkers);CHKERRQ(ierr);} 650287de1a7SBarry Smith PetscFunctionReturn(0); 651287de1a7SBarry Smith } 652