15c6c1daeSBarry Smith 25c6c1daeSBarry Smith /* 35c6c1daeSBarry Smith Contains the data structure for plotting a histogram in a window with an axis. 45c6c1daeSBarry Smith */ 59804daf3SBarry Smith #include <petscdraw.h> /*I "petscdraw.h" I*/ 6af0996ceSBarry Smith #include <petsc/private/petscimpl.h> /*I "petscsys.h" I*/ 734a5a0e3SBarry Smith #include <petscviewer.h> /*I "petscviewer.h" I*/ 85c6c1daeSBarry Smith 95c6c1daeSBarry Smith PetscClassId PETSC_DRAWHG_CLASSID = 0; 105c6c1daeSBarry Smith 115c6c1daeSBarry Smith struct _p_PetscDrawHG { 125c6c1daeSBarry Smith PETSCHEADER(int); 135c6c1daeSBarry Smith PetscErrorCode (*destroy)(PetscDrawSP); 145c6c1daeSBarry Smith PetscErrorCode (*view)(PetscDrawSP,PetscViewer); 155c6c1daeSBarry Smith PetscDraw win; 165c6c1daeSBarry Smith PetscDrawAxis axis; 175c6c1daeSBarry Smith PetscReal xmin,xmax; 185c6c1daeSBarry Smith PetscReal ymin,ymax; 195c6c1daeSBarry Smith int numBins; 205c6c1daeSBarry Smith int maxBins; 215c6c1daeSBarry Smith PetscReal *bins; 225c6c1daeSBarry Smith int numValues; 235c6c1daeSBarry Smith int maxValues; 245c6c1daeSBarry Smith PetscReal *values; 255c6c1daeSBarry Smith int color; 265c6c1daeSBarry Smith PetscBool calcStats; 275c6c1daeSBarry Smith PetscBool integerBins; 285c6c1daeSBarry Smith }; 295c6c1daeSBarry Smith 305c6c1daeSBarry Smith #define CHUNKSIZE 100 315c6c1daeSBarry Smith 325c6c1daeSBarry Smith /*@C 335c6c1daeSBarry Smith PetscDrawHGCreate - Creates a histogram data structure. 345c6c1daeSBarry Smith 355b399a63SLisandro Dalcin Collective on PetscDraw 365c6c1daeSBarry Smith 375c6c1daeSBarry Smith Input Parameters: 385c6c1daeSBarry Smith + draw - The window where the graph will be made 395c6c1daeSBarry Smith - bins - The number of bins to use 405c6c1daeSBarry Smith 415c6c1daeSBarry Smith Output Parameters: 425c6c1daeSBarry Smith . hist - The histogram context 435c6c1daeSBarry Smith 4495452b02SPatrick Sanan Notes: 45a8d69d7bSBarry Smith The difference between a bar chart, PetscDrawBar, and a histogram, PetscDrawHG, is explained here https://stattrek.com/statistics/charts/histogram.aspx?Tutorial=AP 460afdd333SBarry Smith 470afdd333SBarry Smith The histogram is only displayed when PetscDrawHGDraw() is called. 480afdd333SBarry Smith 497e25d57eSBarry Smith The MPI communicator that owns the PetscDraw owns this PetscDrawHG, but the calls to set options and add data are ignored on all processes except the 507e25d57eSBarry Smith zeroth MPI process in the communicator. All MPI processes in the communicator must call PetscDrawHGDraw() to display the updated graph. 517e25d57eSBarry Smith 525c6c1daeSBarry Smith Level: intermediate 535c6c1daeSBarry Smith 540afdd333SBarry Smith .seealso: PetscDrawHGDestroy(), PetscDrawHG, PetscDrawBarCreate(), PetscDrawBar, PetscDrawLGCreate(), PetscDrawLG, PetscDrawSPCreate(), PetscDrawSP, 550afdd333SBarry Smith PetscDrawHGSetNumberBins(), PetscDrawHGReset(), PetscDrawHGAddValue(), PetscDrawHGDraw(), PetscDrawHGSave(), PetscDrawHGView(), PetscDrawHGSetColor(), 560afdd333SBarry Smith PetscDrawHGSetLimits(), PetscDrawHGCalcStats(), PetscDrawHGIntegerBins(), PetscDrawHGGetAxis(), PetscDrawAxis, PetscDrawHGGetDraw() 575c6c1daeSBarry Smith 585c6c1daeSBarry Smith @*/ 590adebc6cSBarry Smith PetscErrorCode PetscDrawHGCreate(PetscDraw draw,int bins,PetscDrawHG *hist) 600adebc6cSBarry Smith { 61e118a51fSLisandro Dalcin PetscDrawHG h; 625c6c1daeSBarry Smith 635c6c1daeSBarry Smith PetscFunctionBegin; 645c6c1daeSBarry Smith PetscValidHeaderSpecific(draw, PETSC_DRAW_CLASSID,1); 65e118a51fSLisandro Dalcin PetscValidLogicalCollectiveInt(draw,bins,2); 665c6c1daeSBarry Smith PetscValidPointer(hist,3); 67e118a51fSLisandro Dalcin 68*9566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(h,PETSC_DRAWHG_CLASSID,"DrawHG","Histogram","Draw",PetscObjectComm((PetscObject)draw),PetscDrawHGDestroy,NULL)); 69*9566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)draw,(PetscObject)h)); 70e118a51fSLisandro Dalcin 71*9566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)draw)); 72e118a51fSLisandro Dalcin h->win = draw; 73a297a907SKarl Rupp 740298fd71SBarry Smith h->view = NULL; 750298fd71SBarry Smith h->destroy = NULL; 765c6c1daeSBarry Smith h->color = PETSC_DRAW_GREEN; 775c6c1daeSBarry Smith h->xmin = PETSC_MAX_REAL; 785c6c1daeSBarry Smith h->xmax = PETSC_MIN_REAL; 795c6c1daeSBarry Smith h->ymin = 0.; 805c6c1daeSBarry Smith h->ymax = 1.; 815c6c1daeSBarry Smith h->numBins = bins; 825c6c1daeSBarry Smith h->maxBins = bins; 83a297a907SKarl Rupp 84*9566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(h->maxBins,&h->bins)); 85a297a907SKarl Rupp 865c6c1daeSBarry Smith h->numValues = 0; 875c6c1daeSBarry Smith h->maxValues = CHUNKSIZE; 885c6c1daeSBarry Smith h->calcStats = PETSC_FALSE; 895c6c1daeSBarry Smith h->integerBins = PETSC_FALSE; 90a297a907SKarl Rupp 91*9566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(h->maxValues,&h->values)); 92*9566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)h,(h->maxBins + h->maxValues)*sizeof(PetscReal))); 93a297a907SKarl Rupp 94*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisCreate(draw,&h->axis)); 95*9566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)h,(PetscObject)h->axis)); 96e118a51fSLisandro Dalcin 975c6c1daeSBarry Smith *hist = h; 985c6c1daeSBarry Smith PetscFunctionReturn(0); 995c6c1daeSBarry Smith } 1005c6c1daeSBarry Smith 1015c6c1daeSBarry Smith /*@ 1025c6c1daeSBarry Smith PetscDrawHGSetNumberBins - Change the number of bins that are to be drawn. 1035c6c1daeSBarry Smith 1045b399a63SLisandro Dalcin Logically Collective on PetscDrawHG 1055c6c1daeSBarry Smith 106d8d19677SJose E. Roman Input Parameters: 1075c6c1daeSBarry Smith + hist - The histogram context. 108e118a51fSLisandro Dalcin - bins - The number of bins. 1095c6c1daeSBarry Smith 1105c6c1daeSBarry Smith Level: intermediate 1115c6c1daeSBarry Smith 1120afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHG, PetscDrawHGDraw(), PetscDrawHGIntegerBins() 1130afdd333SBarry Smith 1145c6c1daeSBarry Smith @*/ 1155c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetNumberBins(PetscDrawHG hist, int bins) 1165c6c1daeSBarry Smith { 1175c6c1daeSBarry Smith PetscFunctionBegin; 1185c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 119e118a51fSLisandro Dalcin PetscValidLogicalCollectiveInt(hist,bins,2); 120e118a51fSLisandro Dalcin 1215c6c1daeSBarry Smith if (hist->maxBins < bins) { 122*9566063dSJacob Faibussowitsch PetscCall(PetscFree(hist->bins)); 123*9566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(bins, &hist->bins)); 124*9566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)hist, (bins - hist->maxBins) * sizeof(PetscReal))); 1255c6c1daeSBarry Smith hist->maxBins = bins; 1265c6c1daeSBarry Smith } 1275c6c1daeSBarry Smith hist->numBins = bins; 1285c6c1daeSBarry Smith PetscFunctionReturn(0); 1295c6c1daeSBarry Smith } 1305c6c1daeSBarry Smith 1315c6c1daeSBarry Smith /*@ 1325c6c1daeSBarry Smith PetscDrawHGReset - Clears histogram to allow for reuse with new data. 1335c6c1daeSBarry Smith 1345b399a63SLisandro Dalcin Logically Collective on PetscDrawHG 1355c6c1daeSBarry Smith 1365c6c1daeSBarry Smith Input Parameter: 1375c6c1daeSBarry Smith . hist - The histogram context. 1385c6c1daeSBarry Smith 1395c6c1daeSBarry Smith Level: intermediate 1405c6c1daeSBarry Smith 1410afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHG, PetscDrawHGDraw(), PetscDrawHGAddValue() 1420afdd333SBarry Smith 1435c6c1daeSBarry Smith @*/ 1445c6c1daeSBarry Smith PetscErrorCode PetscDrawHGReset(PetscDrawHG hist) 1455c6c1daeSBarry Smith { 1465c6c1daeSBarry Smith PetscFunctionBegin; 1475c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 148e118a51fSLisandro Dalcin 1495c6c1daeSBarry Smith hist->xmin = PETSC_MAX_REAL; 1505c6c1daeSBarry Smith hist->xmax = PETSC_MIN_REAL; 1515c6c1daeSBarry Smith hist->ymin = 0.0; 1525c6c1daeSBarry Smith hist->ymax = 0.0; 1535c6c1daeSBarry Smith hist->numValues = 0; 1545c6c1daeSBarry Smith PetscFunctionReturn(0); 1555c6c1daeSBarry Smith } 1565c6c1daeSBarry Smith 1575c6c1daeSBarry Smith /*@C 1585c6c1daeSBarry Smith PetscDrawHGDestroy - Frees all space taken up by histogram data structure. 1595c6c1daeSBarry Smith 1605b399a63SLisandro Dalcin Collective on PetscDrawHG 1615c6c1daeSBarry Smith 1625c6c1daeSBarry Smith Input Parameter: 1635c6c1daeSBarry Smith . hist - The histogram context 1645c6c1daeSBarry Smith 1655c6c1daeSBarry Smith Level: intermediate 1665c6c1daeSBarry Smith 1670afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHG 1685c6c1daeSBarry Smith @*/ 1695c6c1daeSBarry Smith PetscErrorCode PetscDrawHGDestroy(PetscDrawHG *hist) 1705c6c1daeSBarry Smith { 1715c6c1daeSBarry Smith PetscFunctionBegin; 1725c6c1daeSBarry Smith if (!*hist) PetscFunctionReturn(0); 173e118a51fSLisandro Dalcin PetscValidHeaderSpecific(*hist,PETSC_DRAWHG_CLASSID,1); 174e118a51fSLisandro Dalcin if (--((PetscObject)(*hist))->refct > 0) {*hist = NULL; PetscFunctionReturn(0);} 1755c6c1daeSBarry Smith 176*9566063dSJacob Faibussowitsch PetscCall(PetscFree((*hist)->bins)); 177*9566063dSJacob Faibussowitsch PetscCall(PetscFree((*hist)->values)); 178*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDestroy(&(*hist)->axis)); 179*9566063dSJacob Faibussowitsch PetscCall(PetscDrawDestroy(&(*hist)->win)); 180*9566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(hist)); 1815c6c1daeSBarry Smith PetscFunctionReturn(0); 1825c6c1daeSBarry Smith } 1835c6c1daeSBarry Smith 1845c6c1daeSBarry Smith /*@ 1855c6c1daeSBarry Smith PetscDrawHGAddValue - Adds another value to the histogram. 1865c6c1daeSBarry Smith 1875b399a63SLisandro Dalcin Logically Collective on PetscDrawHG 1885c6c1daeSBarry Smith 1895c6c1daeSBarry Smith Input Parameters: 1905c6c1daeSBarry Smith + hist - The histogram 1915c6c1daeSBarry Smith - value - The value 1925c6c1daeSBarry Smith 1935c6c1daeSBarry Smith Level: intermediate 1945c6c1daeSBarry Smith 1950afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHG, PetscDrawHGDraw(), PetscDrawHGAddValue(), PetscDrawHGReset() 1965c6c1daeSBarry Smith @*/ 1975c6c1daeSBarry Smith PetscErrorCode PetscDrawHGAddValue(PetscDrawHG hist, PetscReal value) 1985c6c1daeSBarry Smith { 1995c6c1daeSBarry Smith PetscFunctionBegin; 2005c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 201e118a51fSLisandro Dalcin 2025c6c1daeSBarry Smith /* Allocate more memory if necessary */ 2035c6c1daeSBarry Smith if (hist->numValues >= hist->maxValues) { 2045c6c1daeSBarry Smith PetscReal *tmp; 2055c6c1daeSBarry Smith 206*9566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(hist->maxValues+CHUNKSIZE, &tmp)); 207*9566063dSJacob Faibussowitsch PetscCall(PetscLogObjectMemory((PetscObject)hist, CHUNKSIZE * sizeof(PetscReal))); 208*9566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, hist->values, hist->maxValues)); 209*9566063dSJacob Faibussowitsch PetscCall(PetscFree(hist->values)); 210a297a907SKarl Rupp 2115c6c1daeSBarry Smith hist->values = tmp; 2125c6c1daeSBarry Smith hist->maxValues += CHUNKSIZE; 2135c6c1daeSBarry Smith } 2145c6c1daeSBarry Smith /* I disagree with the original Petsc implementation here. There should be no overshoot, but rather the 2155c6c1daeSBarry Smith stated convention of using half-open intervals (always the way to go) */ 216d0c080abSJoseph Pusztay if (!hist->numValues && (hist->xmin == PETSC_MAX_REAL) && (hist->xmax == PETSC_MIN_REAL)) { 2175c6c1daeSBarry Smith hist->xmin = value; 2185c6c1daeSBarry Smith hist->xmax = value; 2195c6c1daeSBarry Smith #if 1 2205c6c1daeSBarry Smith } else { 2215c6c1daeSBarry Smith /* Update limits */ 222a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value; 223a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2245c6c1daeSBarry Smith #else 2255c6c1daeSBarry Smith } else if (hist->numValues == 1) { 2265c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 227a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value + 0.001*(value - hist->xmin)/hist->numBins; 2285c6c1daeSBarry Smith if (value < hist->xmin) { 2295c6c1daeSBarry Smith hist->xmin = value; 2305c6c1daeSBarry Smith hist->xmax = hist->xmax + 0.001*(hist->xmax - hist->xmin)/hist->numBins; 2315c6c1daeSBarry Smith } 2325c6c1daeSBarry Smith } else { 2335c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 234a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value + 0.001*(hist->xmax - hist->xmin)/hist->numBins; 235a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2365c6c1daeSBarry Smith #endif 2375c6c1daeSBarry Smith } 2385c6c1daeSBarry Smith 2395c6c1daeSBarry Smith hist->values[hist->numValues++] = value; 2405c6c1daeSBarry Smith PetscFunctionReturn(0); 2415c6c1daeSBarry Smith } 2425c6c1daeSBarry Smith 2435c6c1daeSBarry Smith /*@ 2445c6c1daeSBarry Smith PetscDrawHGDraw - Redraws a histogram. 2455c6c1daeSBarry Smith 2465b399a63SLisandro Dalcin Collective on PetscDrawHG 2475c6c1daeSBarry Smith 2485c6c1daeSBarry Smith Input Parameter: 2495c6c1daeSBarry Smith . hist - The histogram context 2505c6c1daeSBarry Smith 2515c6c1daeSBarry Smith Level: intermediate 2525c6c1daeSBarry Smith 2530afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHG, PetscDrawHGDraw(), PetscDrawHGAddValue(), PetscDrawHGReset() 2540afdd333SBarry Smith 2555c6c1daeSBarry Smith @*/ 2565c6c1daeSBarry Smith PetscErrorCode PetscDrawHGDraw(PetscDrawHG hist) 2575c6c1daeSBarry Smith { 258e118a51fSLisandro Dalcin PetscDraw draw; 2595c6c1daeSBarry Smith PetscBool isnull; 2605c6c1daeSBarry Smith PetscReal xmin,xmax,ymin,ymax,*bins,*values,binSize,binLeft,binRight,maxHeight,mean,var; 2615c6c1daeSBarry Smith char title[256]; 2625c6c1daeSBarry Smith char xlabel[256]; 2635c6c1daeSBarry Smith PetscInt numBins,numBinsOld,numValues,initSize,i,p,bcolor,color; 264e118a51fSLisandro Dalcin PetscMPIInt rank; 2655c6c1daeSBarry Smith PetscErrorCode ierr; 2665c6c1daeSBarry Smith 2675c6c1daeSBarry Smith PetscFunctionBegin; 2685c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 269*9566063dSJacob Faibussowitsch PetscCall(PetscDrawIsNull(hist->win,&isnull)); 2708f69470aSLisandro Dalcin if (isnull) PetscFunctionReturn(0); 271*9566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)hist),&rank)); 272e118a51fSLisandro Dalcin 2735c6c1daeSBarry Smith if ((hist->xmin >= hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(0); 2745c6c1daeSBarry Smith if (hist->numValues < 1) PetscFunctionReturn(0); 2755c6c1daeSBarry Smith 2765c6c1daeSBarry Smith color = hist->color; 27771917b75SLisandro Dalcin if (color == PETSC_DRAW_ROTATE) bcolor = PETSC_DRAW_BLACK+1; 278a297a907SKarl Rupp else bcolor = color; 279a297a907SKarl Rupp 2805c6c1daeSBarry Smith xmin = hist->xmin; 2815c6c1daeSBarry Smith xmax = hist->xmax; 2825c6c1daeSBarry Smith ymin = hist->ymin; 2835c6c1daeSBarry Smith ymax = hist->ymax; 2845c6c1daeSBarry Smith numValues = hist->numValues; 2855c6c1daeSBarry Smith values = hist->values; 2865c6c1daeSBarry Smith mean = 0.0; 2875c6c1daeSBarry Smith var = 0.0; 2885c6c1daeSBarry Smith 2895b399a63SLisandro Dalcin draw = hist->win; 290*9566063dSJacob Faibussowitsch PetscCall(PetscDrawCheckResizedWindow(draw)); 291*9566063dSJacob Faibussowitsch PetscCall(PetscDrawClear(draw)); 292e118a51fSLisandro Dalcin 2935c6c1daeSBarry Smith if (xmin == xmax) { 2945c6c1daeSBarry Smith /* Calculate number of points in each bin */ 2955c6c1daeSBarry Smith bins = hist->bins; 2965c6c1daeSBarry Smith bins[0] = 0.; 2975c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 2985c6c1daeSBarry Smith if (values[p] == xmin) bins[0]++; 2995c6c1daeSBarry Smith mean += values[p]; 3005c6c1daeSBarry Smith var += values[p]*values[p]; 3015c6c1daeSBarry Smith } 3025c6c1daeSBarry Smith maxHeight = bins[0]; 3035c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 3045c6c1daeSBarry Smith xmax = xmin + 1; 305*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax)); 3065c6c1daeSBarry Smith if (hist->calcStats) { 3075c6c1daeSBarry Smith mean /= numValues; 308a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 309a297a907SKarl Rupp else var = 0.0; 310*9566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(title, 256, "Mean: %g Var: %g", (double)mean, (double)var)); 311*9566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(xlabel,256, "Total: %" PetscInt_FMT, numValues)); 312*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL)); 3135c6c1daeSBarry Smith } 314*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDraw(hist->axis)); 315*9566063dSJacob Faibussowitsch ierr = PetscDrawCollectiveBegin(draw);PetscCall(ierr); 316dd400576SPatrick Sanan if (rank == 0) { /* Draw bins */ 3175c6c1daeSBarry Smith binLeft = xmin; 3185c6c1daeSBarry Smith binRight = xmax; 319*9566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,binLeft,ymin,binRight,bins[0],bcolor,bcolor,bcolor,bcolor)); 320*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binLeft,ymin,binLeft,bins[0],PETSC_DRAW_BLACK)); 321*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binRight,ymin,binRight,bins[0],PETSC_DRAW_BLACK)); 322*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binLeft,bins[0],binRight,bins[0],PETSC_DRAW_BLACK)); 323e118a51fSLisandro Dalcin } 324*9566063dSJacob Faibussowitsch ierr = PetscDrawCollectiveEnd(draw);PetscCall(ierr); 3255c6c1daeSBarry Smith } else { 3265c6c1daeSBarry Smith numBins = hist->numBins; 3275c6c1daeSBarry Smith numBinsOld = hist->numBins; 3285c6c1daeSBarry Smith if (hist->integerBins && (((int) xmax - xmin) + 1.0e-05 > xmax - xmin)) { 3295c6c1daeSBarry Smith initSize = (int) ((int) xmax - xmin)/numBins; 3305c6c1daeSBarry Smith while (initSize*numBins != (int) xmax - xmin) { 3315c6c1daeSBarry Smith initSize = PetscMax(initSize - 1, 1); 3325c6c1daeSBarry Smith numBins = (int) ((int) xmax - xmin)/initSize; 333*9566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist, numBins)); 3345c6c1daeSBarry Smith } 3355c6c1daeSBarry Smith } 3365c6c1daeSBarry Smith binSize = (xmax - xmin)/numBins; 3375c6c1daeSBarry Smith bins = hist->bins; 3385c6c1daeSBarry Smith 339*9566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(bins, numBins)); 340a297a907SKarl Rupp 3415c6c1daeSBarry Smith maxHeight = 0.0; 3425c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3435c6c1daeSBarry Smith binLeft = xmin + binSize*i; 3445c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 3455c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 3465c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 3475c6c1daeSBarry Smith /* Handle last bin separately */ 3485c6c1daeSBarry Smith if ((i == numBins-1) && (values[p] == binRight)) bins[i]++; 3495c6c1daeSBarry Smith if (!i) { 3505c6c1daeSBarry Smith mean += values[p]; 3515c6c1daeSBarry Smith var += values[p]*values[p]; 3525c6c1daeSBarry Smith } 3535c6c1daeSBarry Smith } 3545c6c1daeSBarry Smith maxHeight = PetscMax(maxHeight, bins[i]); 3555c6c1daeSBarry Smith } 3565c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 3575c6c1daeSBarry Smith 358*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax)); 3595c6c1daeSBarry Smith if (hist->calcStats) { 3605c6c1daeSBarry Smith mean /= numValues; 361a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 362a297a907SKarl Rupp else var = 0.0; 363*9566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(title, 256,"Mean: %g Var: %g", (double)mean, (double)var)); 364*9566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(xlabel,256, "Total: %" PetscInt_FMT, numValues)); 365*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL)); 3665c6c1daeSBarry Smith } 367*9566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDraw(hist->axis)); 368*9566063dSJacob Faibussowitsch ierr = PetscDrawCollectiveBegin(draw);PetscCall(ierr); 369dd400576SPatrick Sanan if (rank == 0) { /* Draw bins */ 3705c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3715c6c1daeSBarry Smith binLeft = xmin + binSize*i; 3725c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 373*9566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw,binLeft,ymin,binRight,bins[i],bcolor,bcolor,bcolor,bcolor)); 374*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binLeft,ymin,binLeft,bins[i],PETSC_DRAW_BLACK)); 375*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binRight,ymin,binRight,bins[i],PETSC_DRAW_BLACK)); 376*9566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw,binLeft,bins[i],binRight,bins[i],PETSC_DRAW_BLACK)); 377e118a51fSLisandro Dalcin if (color == PETSC_DRAW_ROTATE && bins[i]) bcolor++; 37871917b75SLisandro Dalcin if (bcolor > PETSC_DRAW_BASIC_COLORS-1) bcolor = PETSC_DRAW_BLACK+1; 379e118a51fSLisandro Dalcin } 3805c6c1daeSBarry Smith } 381*9566063dSJacob Faibussowitsch ierr = PetscDrawCollectiveEnd(draw);PetscCall(ierr); 382*9566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist,numBinsOld)); 3835c6c1daeSBarry Smith } 384e118a51fSLisandro Dalcin 385*9566063dSJacob Faibussowitsch PetscCall(PetscDrawFlush(draw)); 386*9566063dSJacob Faibussowitsch PetscCall(PetscDrawPause(draw)); 3875c6c1daeSBarry Smith PetscFunctionReturn(0); 3885c6c1daeSBarry Smith } 3895c6c1daeSBarry Smith 39057fd6651SLisandro Dalcin /*@ 39157fd6651SLisandro Dalcin PetscDrawHGSave - Saves a drawn image 39257fd6651SLisandro Dalcin 39357fd6651SLisandro Dalcin Collective on PetscDrawHG 39457fd6651SLisandro Dalcin 39557fd6651SLisandro Dalcin Input Parameter: 39657fd6651SLisandro Dalcin . hist - The histogram context 39757fd6651SLisandro Dalcin 39857fd6651SLisandro Dalcin Level: intermediate 39957fd6651SLisandro Dalcin 4000afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGGetDraw(), PetscDrawSetSave(), PetscDrawSave(), PetscDrawHGDraw() 40157fd6651SLisandro Dalcin @*/ 40257fd6651SLisandro Dalcin PetscErrorCode PetscDrawHGSave(PetscDrawHG hg) 40357fd6651SLisandro Dalcin { 40457fd6651SLisandro Dalcin PetscFunctionBegin; 40557fd6651SLisandro Dalcin PetscValidHeaderSpecific(hg,PETSC_DRAWHG_CLASSID,1); 406*9566063dSJacob Faibussowitsch PetscCall(PetscDrawSave(hg->win)); 40757fd6651SLisandro Dalcin PetscFunctionReturn(0); 40857fd6651SLisandro Dalcin } 40957fd6651SLisandro Dalcin 4105c6c1daeSBarry Smith /*@ 41134a5a0e3SBarry Smith PetscDrawHGView - Prints the histogram information. 4125c6c1daeSBarry Smith 4135c6c1daeSBarry Smith Not collective 4145c6c1daeSBarry Smith 4155c6c1daeSBarry Smith Input Parameter: 4165c6c1daeSBarry Smith . hist - The histogram context 4175c6c1daeSBarry Smith 4185c6c1daeSBarry Smith Level: beginner 4195c6c1daeSBarry Smith 4200afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGGetDraw(), PetscDrawSetSave(), PetscDrawSave(), PetscDrawHGDraw() 4210afdd333SBarry Smith 4225c6c1daeSBarry Smith @*/ 42334a5a0e3SBarry Smith PetscErrorCode PetscDrawHGView(PetscDrawHG hist,PetscViewer viewer) 4245c6c1daeSBarry Smith { 4255c6c1daeSBarry Smith PetscReal xmax,xmin,*bins,*values,binSize,binLeft,binRight,mean,var; 4265c6c1daeSBarry Smith PetscInt numBins,numBinsOld,numValues,initSize,i,p; 4275c6c1daeSBarry Smith 4285c6c1daeSBarry Smith PetscFunctionBegin; 4295c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 430e118a51fSLisandro Dalcin 4315c6c1daeSBarry Smith if ((hist->xmin > hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(0); 4325c6c1daeSBarry Smith if (hist->numValues < 1) PetscFunctionReturn(0); 4335c6c1daeSBarry Smith 43434a5a0e3SBarry Smith if (!viewer) { 435*9566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)hist),&viewer)); 43634a5a0e3SBarry Smith } 437*9566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)hist,viewer)); 4385c6c1daeSBarry Smith xmax = hist->xmax; 4395c6c1daeSBarry Smith xmin = hist->xmin; 4405c6c1daeSBarry Smith numValues = hist->numValues; 4415c6c1daeSBarry Smith values = hist->values; 4425c6c1daeSBarry Smith mean = 0.0; 4435c6c1daeSBarry Smith var = 0.0; 4445c6c1daeSBarry Smith if (xmax == xmin) { 4455c6c1daeSBarry Smith /* Calculate number of points in the bin */ 4465c6c1daeSBarry Smith bins = hist->bins; 4475c6c1daeSBarry Smith bins[0] = 0.; 4485c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 4495c6c1daeSBarry Smith if (values[p] == xmin) bins[0]++; 4505c6c1daeSBarry Smith mean += values[p]; 4515c6c1daeSBarry Smith var += values[p]*values[p]; 4525c6c1daeSBarry Smith } 4535c6c1daeSBarry Smith /* Draw bins */ 454*9566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Bin %2d (%6.2g - %6.2g): %.0g\n", 0, (double)xmin, (double)xmax, (double)bins[0])); 4555c6c1daeSBarry Smith } else { 4565c6c1daeSBarry Smith numBins = hist->numBins; 4575c6c1daeSBarry Smith numBinsOld = hist->numBins; 4585c6c1daeSBarry Smith if (hist->integerBins && (((int) xmax - xmin) + 1.0e-05 > xmax - xmin)) { 4595c6c1daeSBarry Smith initSize = (int) ((int) xmax - xmin)/numBins; 4605c6c1daeSBarry Smith while (initSize*numBins != (int) xmax - xmin) { 4615c6c1daeSBarry Smith initSize = PetscMax(initSize - 1, 1); 4625c6c1daeSBarry Smith numBins = (int) ((int) xmax - xmin)/initSize; 463*9566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist, numBins)); 4645c6c1daeSBarry Smith } 4655c6c1daeSBarry Smith } 4665c6c1daeSBarry Smith binSize = (xmax - xmin)/numBins; 4675c6c1daeSBarry Smith bins = hist->bins; 4685c6c1daeSBarry Smith 4695c6c1daeSBarry Smith /* Calculate number of points in each bin */ 470*9566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(bins, numBins)); 4715c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4725c6c1daeSBarry Smith binLeft = xmin + binSize*i; 4735c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 4745c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 4755c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 4765c6c1daeSBarry Smith /* Handle last bin separately */ 4775c6c1daeSBarry Smith if ((i == numBins-1) && (values[p] == binRight)) bins[i]++; 4785c6c1daeSBarry Smith if (!i) { 4795c6c1daeSBarry Smith mean += values[p]; 4805c6c1daeSBarry Smith var += values[p]*values[p]; 4815c6c1daeSBarry Smith } 4825c6c1daeSBarry Smith } 4835c6c1daeSBarry Smith } 4845c6c1daeSBarry Smith /* Draw bins */ 4855c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4865c6c1daeSBarry Smith binLeft = xmin + binSize*i; 4875c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 488*9566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Bin %2d (%6.2g - %6.2g): %.0g\n", (int)i, (double)binLeft, (double)binRight, (double)bins[i])); 4895c6c1daeSBarry Smith } 490*9566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist, numBinsOld)); 4915c6c1daeSBarry Smith } 4925c6c1daeSBarry Smith 4935c6c1daeSBarry Smith if (hist->calcStats) { 4945c6c1daeSBarry Smith mean /= numValues; 495a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 496a297a907SKarl Rupp else var = 0.0; 497*9566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Mean: %g Var: %g\n", (double)mean, (double)var)); 498*9566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Total: %" PetscInt_FMT "\n", numValues)); 4995c6c1daeSBarry Smith } 5005c6c1daeSBarry Smith PetscFunctionReturn(0); 5015c6c1daeSBarry Smith } 5025c6c1daeSBarry Smith 5035c6c1daeSBarry Smith /*@ 5045c6c1daeSBarry Smith PetscDrawHGSetColor - Sets the color the bars will be drawn with. 5055c6c1daeSBarry Smith 5065b399a63SLisandro Dalcin Logically Collective on PetscDrawHG 5075c6c1daeSBarry Smith 5085c6c1daeSBarry Smith Input Parameters: 5095c6c1daeSBarry Smith + hist - The histogram context 5105c6c1daeSBarry Smith - color - one of the colors defined in petscdraw.h or PETSC_DRAW_ROTATE to make each bar a 5115c6c1daeSBarry Smith different color 5125c6c1daeSBarry Smith 5135c6c1daeSBarry Smith Level: intermediate 5145c6c1daeSBarry Smith 5150afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGGetDraw(), PetscDrawSetSave(), PetscDrawSave(), PetscDrawHGDraw(), PetscDrawHGGetAxis() 5160afdd333SBarry Smith 5175c6c1daeSBarry Smith @*/ 5185c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetColor(PetscDrawHG hist,int color) 5195c6c1daeSBarry Smith { 5205c6c1daeSBarry Smith PetscFunctionBegin; 5215c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 522e118a51fSLisandro Dalcin 5235c6c1daeSBarry Smith hist->color = color; 5245c6c1daeSBarry Smith PetscFunctionReturn(0); 5255c6c1daeSBarry Smith } 5265c6c1daeSBarry Smith 5275c6c1daeSBarry Smith /*@ 5285c6c1daeSBarry Smith PetscDrawHGSetLimits - Sets the axis limits for a histogram. If more 5295c6c1daeSBarry Smith points are added after this call, the limits will be adjusted to 5305c6c1daeSBarry Smith include those additional points. 5315c6c1daeSBarry Smith 5325b399a63SLisandro Dalcin Logically Collective on PetscDrawHG 5335c6c1daeSBarry Smith 5345c6c1daeSBarry Smith Input Parameters: 5355c6c1daeSBarry Smith + hist - The histogram context 5365c6c1daeSBarry Smith - x_min,x_max,y_min,y_max - The limits 5375c6c1daeSBarry Smith 5385c6c1daeSBarry Smith Level: intermediate 5395c6c1daeSBarry Smith 5400afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGGetDraw(), PetscDrawSetSave(), PetscDrawSave(), PetscDrawHGDraw(), PetscDrawHGGetAxis() 5410afdd333SBarry Smith 5425c6c1daeSBarry Smith @*/ 5435c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetLimits(PetscDrawHG hist, PetscReal x_min, PetscReal x_max, int y_min, int y_max) 5445c6c1daeSBarry Smith { 5455c6c1daeSBarry Smith PetscFunctionBegin; 5465c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 547e118a51fSLisandro Dalcin 5485c6c1daeSBarry Smith hist->xmin = x_min; 5495c6c1daeSBarry Smith hist->xmax = x_max; 5505c6c1daeSBarry Smith hist->ymin = y_min; 5515c6c1daeSBarry Smith hist->ymax = y_max; 5525c6c1daeSBarry Smith PetscFunctionReturn(0); 5535c6c1daeSBarry Smith } 5545c6c1daeSBarry Smith 5555c6c1daeSBarry Smith /*@ 5565c6c1daeSBarry Smith PetscDrawHGCalcStats - Turns on calculation of descriptive statistics 5575c6c1daeSBarry Smith 5585c6c1daeSBarry Smith Not collective 5595c6c1daeSBarry Smith 5605c6c1daeSBarry Smith Input Parameters: 5615c6c1daeSBarry Smith + hist - The histogram context 5625c6c1daeSBarry Smith - calc - Flag for calculation 5635c6c1daeSBarry Smith 5645c6c1daeSBarry Smith Level: intermediate 5655c6c1daeSBarry Smith 5660afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGAddValue(), PetscDrawHGView(), PetscDrawHGDraw() 5670afdd333SBarry Smith 5685c6c1daeSBarry Smith @*/ 5695c6c1daeSBarry Smith PetscErrorCode PetscDrawHGCalcStats(PetscDrawHG hist, PetscBool calc) 5705c6c1daeSBarry Smith { 5715c6c1daeSBarry Smith PetscFunctionBegin; 5725c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 573e118a51fSLisandro Dalcin 5745c6c1daeSBarry Smith hist->calcStats = calc; 5755c6c1daeSBarry Smith PetscFunctionReturn(0); 5765c6c1daeSBarry Smith } 5775c6c1daeSBarry Smith 5785c6c1daeSBarry Smith /*@ 5795c6c1daeSBarry Smith PetscDrawHGIntegerBins - Turns on integer width bins 5805c6c1daeSBarry Smith 5815c6c1daeSBarry Smith Not collective 5825c6c1daeSBarry Smith 5835c6c1daeSBarry Smith Input Parameters: 5845c6c1daeSBarry Smith + hist - The histogram context 5855c6c1daeSBarry Smith - ints - Flag for integer width bins 5865c6c1daeSBarry Smith 5875c6c1daeSBarry Smith Level: intermediate 5885c6c1daeSBarry Smith 5890afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGAddValue(), PetscDrawHGView(), PetscDrawHGDraw(), PetscDrawHGSetColor() 5900afdd333SBarry Smith 5915c6c1daeSBarry Smith @*/ 5925c6c1daeSBarry Smith PetscErrorCode PetscDrawHGIntegerBins(PetscDrawHG hist, PetscBool ints) 5935c6c1daeSBarry Smith { 5945c6c1daeSBarry Smith PetscFunctionBegin; 5955c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 596e118a51fSLisandro Dalcin 5975c6c1daeSBarry Smith hist->integerBins = ints; 5985c6c1daeSBarry Smith PetscFunctionReturn(0); 5995c6c1daeSBarry Smith } 6005c6c1daeSBarry Smith 6015c6c1daeSBarry Smith /*@C 6025c6c1daeSBarry Smith PetscDrawHGGetAxis - Gets the axis context associated with a histogram. 6035c6c1daeSBarry Smith This is useful if one wants to change some axis property, such as 6045c6c1daeSBarry Smith labels, color, etc. The axis context should not be destroyed by the 6055c6c1daeSBarry Smith application code. 6065c6c1daeSBarry Smith 6075b399a63SLisandro Dalcin Not Collective, PetscDrawAxis is parallel if PetscDrawHG is parallel 6085c6c1daeSBarry Smith 6095c6c1daeSBarry Smith Input Parameter: 6105c6c1daeSBarry Smith . hist - The histogram context 6115c6c1daeSBarry Smith 6125c6c1daeSBarry Smith Output Parameter: 6135c6c1daeSBarry Smith . axis - The axis context 6145c6c1daeSBarry Smith 6155c6c1daeSBarry Smith Level: intermediate 6165c6c1daeSBarry Smith 6170afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGAddValue(), PetscDrawHGView(), PetscDrawHGDraw(), PetscDrawHGSetColor(), PetscDrawAxis, PetscDrawHGSetLimits() 6180afdd333SBarry Smith 6195c6c1daeSBarry Smith @*/ 6205c6c1daeSBarry Smith PetscErrorCode PetscDrawHGGetAxis(PetscDrawHG hist,PetscDrawAxis *axis) 6215c6c1daeSBarry Smith { 6225c6c1daeSBarry Smith PetscFunctionBegin; 623e118a51fSLisandro Dalcin PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 62445f3bb6eSLisandro Dalcin PetscValidPointer(axis,2); 6255c6c1daeSBarry Smith *axis = hist->axis; 6265c6c1daeSBarry Smith PetscFunctionReturn(0); 6275c6c1daeSBarry Smith } 6285c6c1daeSBarry Smith 6295c6c1daeSBarry Smith /*@C 6305c6c1daeSBarry Smith PetscDrawHGGetDraw - Gets the draw context associated with a histogram. 6315c6c1daeSBarry Smith 6325c6c1daeSBarry Smith Not Collective, PetscDraw is parallel if PetscDrawHG is parallel 6335c6c1daeSBarry Smith 6345c6c1daeSBarry Smith Input Parameter: 6355c6c1daeSBarry Smith . hist - The histogram context 6365c6c1daeSBarry Smith 6375c6c1daeSBarry Smith Output Parameter: 638e118a51fSLisandro Dalcin . draw - The draw context 6395c6c1daeSBarry Smith 6405c6c1daeSBarry Smith Level: intermediate 6415c6c1daeSBarry Smith 6420afdd333SBarry Smith .seealso: PetscDrawHGCreate(), PetscDrawHGAddValue(), PetscDrawHGView(), PetscDrawHGDraw(), PetscDrawHGSetColor(), PetscDrawAxis, PetscDrawHGSetLimits() 6430afdd333SBarry Smith 6445c6c1daeSBarry Smith @*/ 645e118a51fSLisandro Dalcin PetscErrorCode PetscDrawHGGetDraw(PetscDrawHG hist,PetscDraw *draw) 6465c6c1daeSBarry Smith { 6475c6c1daeSBarry Smith PetscFunctionBegin; 6485c6c1daeSBarry Smith PetscValidHeaderSpecific(hist,PETSC_DRAWHG_CLASSID,1); 64945f3bb6eSLisandro Dalcin PetscValidPointer(draw,2); 650e118a51fSLisandro Dalcin *draw = hist->win; 6515c6c1daeSBarry Smith PetscFunctionReturn(0); 6525c6c1daeSBarry Smith } 653