15c6c1daeSBarry Smith /* 25c6c1daeSBarry Smith Contains the data structure for plotting a histogram in a window with an axis. 35c6c1daeSBarry Smith */ 49804daf3SBarry Smith #include <petscdraw.h> /*I "petscdraw.h" I*/ 5af0996ceSBarry Smith #include <petsc/private/petscimpl.h> /*I "petscsys.h" I*/ 634a5a0e3SBarry Smith #include <petscviewer.h> /*I "petscviewer.h" I*/ 75c6c1daeSBarry Smith 85c6c1daeSBarry Smith PetscClassId PETSC_DRAWHG_CLASSID = 0; 95c6c1daeSBarry Smith 105c6c1daeSBarry Smith struct _p_PetscDrawHG { 115c6c1daeSBarry Smith PETSCHEADER(int); 125c6c1daeSBarry Smith PetscErrorCode (*destroy)(PetscDrawSP); 135c6c1daeSBarry Smith PetscErrorCode (*view)(PetscDrawSP, PetscViewer); 145c6c1daeSBarry Smith PetscDraw win; 155c6c1daeSBarry Smith PetscDrawAxis axis; 165c6c1daeSBarry Smith PetscReal xmin, xmax; 175c6c1daeSBarry Smith PetscReal ymin, ymax; 185c6c1daeSBarry Smith int numBins; 195c6c1daeSBarry Smith int maxBins; 205c6c1daeSBarry Smith PetscReal *bins; 215c6c1daeSBarry Smith int numValues; 225c6c1daeSBarry Smith int maxValues; 235c6c1daeSBarry Smith PetscReal *values; 245c6c1daeSBarry Smith int color; 255c6c1daeSBarry Smith PetscBool calcStats; 265c6c1daeSBarry Smith PetscBool integerBins; 275c6c1daeSBarry Smith }; 285c6c1daeSBarry Smith 295c6c1daeSBarry Smith #define CHUNKSIZE 100 305c6c1daeSBarry Smith 31cc4c1da9SBarry Smith /*@ 325c6c1daeSBarry Smith PetscDrawHGCreate - Creates a histogram data structure. 335c6c1daeSBarry Smith 34c3339decSBarry Smith Collective 355c6c1daeSBarry Smith 365c6c1daeSBarry Smith Input Parameters: 375c6c1daeSBarry Smith + draw - The window where the graph will be made 385c6c1daeSBarry Smith - bins - The number of bins to use 395c6c1daeSBarry Smith 4020f4b53cSBarry Smith Output Parameter: 415c6c1daeSBarry Smith . hist - The histogram context 425c6c1daeSBarry Smith 4320f4b53cSBarry Smith Level: intermediate 4420f4b53cSBarry Smith 4595452b02SPatrick Sanan Notes: 461d27aa22SBarry Smith The difference between a bar chart, `PetscDrawBar`, and a histogram, `PetscDrawHG`, is explained here <https://stattrek.com/statistics/charts/histogram.aspx?Tutorial=AP> 470afdd333SBarry Smith 48811af0c4SBarry Smith The histogram is only displayed when `PetscDrawHGDraw()` is called. 490afdd333SBarry Smith 50811af0c4SBarry 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 511d27aa22SBarry Smith zeroth MPI process in the communicator. All MPI processes in the communicator must call `PetscDrawHGDraw()` to display the updated graph. 527e25d57eSBarry Smith 53db781477SPatrick Sanan .seealso: `PetscDrawHGDestroy()`, `PetscDrawHG`, `PetscDrawBarCreate()`, `PetscDrawBar`, `PetscDrawLGCreate()`, `PetscDrawLG`, `PetscDrawSPCreate()`, `PetscDrawSP`, 54db781477SPatrick Sanan `PetscDrawHGSetNumberBins()`, `PetscDrawHGReset()`, `PetscDrawHGAddValue()`, `PetscDrawHGDraw()`, `PetscDrawHGSave()`, `PetscDrawHGView()`, `PetscDrawHGSetColor()`, 55db781477SPatrick Sanan `PetscDrawHGSetLimits()`, `PetscDrawHGCalcStats()`, `PetscDrawHGIntegerBins()`, `PetscDrawHGGetAxis()`, `PetscDrawAxis`, `PetscDrawHGGetDraw()` 565c6c1daeSBarry Smith @*/ 57d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGCreate(PetscDraw draw, int bins, PetscDrawHG *hist) 58d71ae5a4SJacob Faibussowitsch { 59e118a51fSLisandro Dalcin PetscDrawHG h; 605c6c1daeSBarry Smith 615c6c1daeSBarry Smith PetscFunctionBegin; 625c6c1daeSBarry Smith PetscValidHeaderSpecific(draw, PETSC_DRAW_CLASSID, 1); 63e118a51fSLisandro Dalcin PetscValidLogicalCollectiveInt(draw, bins, 2); 644f572ea9SToby Isaac PetscAssertPointer(hist, 3); 65e118a51fSLisandro Dalcin 669566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(h, PETSC_DRAWHG_CLASSID, "DrawHG", "Histogram", "Draw", PetscObjectComm((PetscObject)draw), PetscDrawHGDestroy, NULL)); 679566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)draw)); 68e118a51fSLisandro Dalcin h->win = draw; 690298fd71SBarry Smith h->view = NULL; 700298fd71SBarry Smith h->destroy = NULL; 715c6c1daeSBarry Smith h->color = PETSC_DRAW_GREEN; 725c6c1daeSBarry Smith h->xmin = PETSC_MAX_REAL; 735c6c1daeSBarry Smith h->xmax = PETSC_MIN_REAL; 745c6c1daeSBarry Smith h->ymin = 0.; 755c6c1daeSBarry Smith h->ymax = 1.; 765c6c1daeSBarry Smith h->numBins = bins; 775c6c1daeSBarry Smith h->maxBins = bins; 789566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(h->maxBins, &h->bins)); 795c6c1daeSBarry Smith h->numValues = 0; 805c6c1daeSBarry Smith h->maxValues = CHUNKSIZE; 815c6c1daeSBarry Smith h->calcStats = PETSC_FALSE; 825c6c1daeSBarry Smith h->integerBins = PETSC_FALSE; 839566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(h->maxValues, &h->values)); 849566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisCreate(draw, &h->axis)); 855c6c1daeSBarry Smith *hist = h; 863ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 875c6c1daeSBarry Smith } 885c6c1daeSBarry Smith 895c6c1daeSBarry Smith /*@ 90811af0c4SBarry Smith PetscDrawHGSetNumberBins - Change the number of bins that are to be drawn in the histogram 915c6c1daeSBarry Smith 92c3339decSBarry Smith Logically Collective 935c6c1daeSBarry Smith 94d8d19677SJose E. Roman Input Parameters: 955c6c1daeSBarry Smith + hist - The histogram context. 96e118a51fSLisandro Dalcin - bins - The number of bins. 975c6c1daeSBarry Smith 985c6c1daeSBarry Smith Level: intermediate 995c6c1daeSBarry Smith 100db781477SPatrick Sanan .seealso: `PetscDrawHGCreate()`, `PetscDrawHG`, `PetscDrawHGDraw()`, `PetscDrawHGIntegerBins()` 1015c6c1daeSBarry Smith @*/ 102d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGSetNumberBins(PetscDrawHG hist, int bins) 103d71ae5a4SJacob Faibussowitsch { 1045c6c1daeSBarry Smith PetscFunctionBegin; 1055c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 106e118a51fSLisandro Dalcin PetscValidLogicalCollectiveInt(hist, bins, 2); 107e118a51fSLisandro Dalcin 1085c6c1daeSBarry Smith if (hist->maxBins < bins) { 1099566063dSJacob Faibussowitsch PetscCall(PetscFree(hist->bins)); 1109566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(bins, &hist->bins)); 1115c6c1daeSBarry Smith hist->maxBins = bins; 1125c6c1daeSBarry Smith } 1135c6c1daeSBarry Smith hist->numBins = bins; 1143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1155c6c1daeSBarry Smith } 1165c6c1daeSBarry Smith 1175c6c1daeSBarry Smith /*@ 1185c6c1daeSBarry Smith PetscDrawHGReset - Clears histogram to allow for reuse with new data. 1195c6c1daeSBarry Smith 120c3339decSBarry Smith Logically Collective 1215c6c1daeSBarry Smith 1225c6c1daeSBarry Smith Input Parameter: 1235c6c1daeSBarry Smith . hist - The histogram context. 1245c6c1daeSBarry Smith 1255c6c1daeSBarry Smith Level: intermediate 1265c6c1daeSBarry Smith 127db781477SPatrick Sanan .seealso: `PetscDrawHGCreate()`, `PetscDrawHG`, `PetscDrawHGDraw()`, `PetscDrawHGAddValue()` 1285c6c1daeSBarry Smith @*/ 129d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGReset(PetscDrawHG hist) 130d71ae5a4SJacob Faibussowitsch { 1315c6c1daeSBarry Smith PetscFunctionBegin; 1325c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 133e118a51fSLisandro Dalcin 1345c6c1daeSBarry Smith hist->xmin = PETSC_MAX_REAL; 1355c6c1daeSBarry Smith hist->xmax = PETSC_MIN_REAL; 1365c6c1daeSBarry Smith hist->ymin = 0.0; 13760e16b1bSMatthew G. Knepley hist->ymax = 1.0; 1385c6c1daeSBarry Smith hist->numValues = 0; 1393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1405c6c1daeSBarry Smith } 1415c6c1daeSBarry Smith 1425d83a8b1SBarry Smith /*@ 1435c6c1daeSBarry Smith PetscDrawHGDestroy - Frees all space taken up by histogram data structure. 1445c6c1daeSBarry Smith 145c3339decSBarry Smith Collective 1465c6c1daeSBarry Smith 1475c6c1daeSBarry Smith Input Parameter: 1485c6c1daeSBarry Smith . hist - The histogram context 1495c6c1daeSBarry Smith 1505c6c1daeSBarry Smith Level: intermediate 1515c6c1daeSBarry Smith 152db781477SPatrick Sanan .seealso: `PetscDrawHGCreate()`, `PetscDrawHG` 1535c6c1daeSBarry Smith @*/ 154d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGDestroy(PetscDrawHG *hist) 155d71ae5a4SJacob Faibussowitsch { 1565c6c1daeSBarry Smith PetscFunctionBegin; 1573ba16761SJacob Faibussowitsch if (!*hist) PetscFunctionReturn(PETSC_SUCCESS); 158e118a51fSLisandro Dalcin PetscValidHeaderSpecific(*hist, PETSC_DRAWHG_CLASSID, 1); 159f4f49eeaSPierre Jolivet if (--((PetscObject)*hist)->refct > 0) { 1609371c9d4SSatish Balay *hist = NULL; 1613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1629371c9d4SSatish Balay } 1635c6c1daeSBarry Smith 1649566063dSJacob Faibussowitsch PetscCall(PetscFree((*hist)->bins)); 1659566063dSJacob Faibussowitsch PetscCall(PetscFree((*hist)->values)); 1669566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDestroy(&(*hist)->axis)); 1679566063dSJacob Faibussowitsch PetscCall(PetscDrawDestroy(&(*hist)->win)); 1689566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(hist)); 1693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1705c6c1daeSBarry Smith } 1715c6c1daeSBarry Smith 1725c6c1daeSBarry Smith /*@ 1735c6c1daeSBarry Smith PetscDrawHGAddValue - Adds another value to the histogram. 1745c6c1daeSBarry Smith 175c3339decSBarry Smith Logically Collective 1765c6c1daeSBarry Smith 1775c6c1daeSBarry Smith Input Parameters: 1785c6c1daeSBarry Smith + hist - The histogram 1795c6c1daeSBarry Smith - value - The value 1805c6c1daeSBarry Smith 1815c6c1daeSBarry Smith Level: intermediate 1825c6c1daeSBarry Smith 18342747ad1SJacob Faibussowitsch .seealso: `PetscDrawHGCreate()`, `PetscDrawHG`, `PetscDrawHGDraw()`, `PetscDrawHGReset()` 1845c6c1daeSBarry Smith @*/ 185d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGAddValue(PetscDrawHG hist, PetscReal value) 186d71ae5a4SJacob Faibussowitsch { 1875c6c1daeSBarry Smith PetscFunctionBegin; 1885c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 189e118a51fSLisandro Dalcin 1905c6c1daeSBarry Smith /* Allocate more memory if necessary */ 1915c6c1daeSBarry Smith if (hist->numValues >= hist->maxValues) { 1925c6c1daeSBarry Smith PetscReal *tmp; 1935c6c1daeSBarry Smith 1949566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(hist->maxValues + CHUNKSIZE, &tmp)); 1959566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(tmp, hist->values, hist->maxValues)); 1969566063dSJacob Faibussowitsch PetscCall(PetscFree(hist->values)); 197a297a907SKarl Rupp 1985c6c1daeSBarry Smith hist->values = tmp; 1995c6c1daeSBarry Smith hist->maxValues += CHUNKSIZE; 2005c6c1daeSBarry Smith } 2015c6c1daeSBarry Smith /* I disagree with the original Petsc implementation here. There should be no overshoot, but rather the 2025c6c1daeSBarry Smith stated convention of using half-open intervals (always the way to go) */ 203d0c080abSJoseph Pusztay if (!hist->numValues && (hist->xmin == PETSC_MAX_REAL) && (hist->xmax == PETSC_MIN_REAL)) { 2045c6c1daeSBarry Smith hist->xmin = value; 2055c6c1daeSBarry Smith hist->xmax = value; 2065c6c1daeSBarry Smith #if 1 2075c6c1daeSBarry Smith } else { 2085c6c1daeSBarry Smith /* Update limits */ 209a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value; 210a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2115c6c1daeSBarry Smith #else 2125c6c1daeSBarry Smith } else if (hist->numValues == 1) { 2135c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 2146497c311SBarry Smith if (value > hist->xmax) hist->xmax = value + 0.001 * (value - hist->xmin) / (PetscReal)hist->numBins; 2155c6c1daeSBarry Smith if (value < hist->xmin) { 2165c6c1daeSBarry Smith hist->xmin = value; 2176497c311SBarry Smith hist->xmax = hist->xmax + 0.001 * (hist->xmax - hist->xmin) / (PetscReal)hist->numBins; 2185c6c1daeSBarry Smith } 2195c6c1daeSBarry Smith } else { 2205c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 2216497c311SBarry Smith if (value > hist->xmax) hist->xmax = value + 0.001 * (hist->xmax - hist->xmin) / (PetscReal)hist->numBins; 222a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2235c6c1daeSBarry Smith #endif 2245c6c1daeSBarry Smith } 2255c6c1daeSBarry Smith 2265c6c1daeSBarry Smith hist->values[hist->numValues++] = value; 2273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2285c6c1daeSBarry Smith } 2295c6c1daeSBarry Smith 2305c6c1daeSBarry Smith /*@ 2315c6c1daeSBarry Smith PetscDrawHGDraw - Redraws a histogram. 2325c6c1daeSBarry Smith 233c3339decSBarry Smith Collective 2345c6c1daeSBarry Smith 2355c6c1daeSBarry Smith Input Parameter: 2365c6c1daeSBarry Smith . hist - The histogram context 2375c6c1daeSBarry Smith 2385c6c1daeSBarry Smith Level: intermediate 2395c6c1daeSBarry Smith 24042747ad1SJacob Faibussowitsch .seealso: `PetscDrawHGCreate()`, `PetscDrawHG`, `PetscDrawHGAddValue()`, `PetscDrawHGReset()` 2415c6c1daeSBarry Smith @*/ 242d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGDraw(PetscDrawHG hist) 243d71ae5a4SJacob Faibussowitsch { 244e118a51fSLisandro Dalcin PetscDraw draw; 2455c6c1daeSBarry Smith PetscBool isnull; 2465c6c1daeSBarry Smith PetscReal xmin, xmax, ymin, ymax, *bins, *values, binSize, binLeft, binRight, maxHeight, mean, var; 2475c6c1daeSBarry Smith char title[256]; 2485c6c1daeSBarry Smith char xlabel[256]; 2496497c311SBarry Smith PetscInt numValues, initSize, i, p; 2506497c311SBarry Smith int bcolor, color, numBins, numBinsOld; 251e118a51fSLisandro Dalcin PetscMPIInt rank; 2525c6c1daeSBarry Smith 2535c6c1daeSBarry Smith PetscFunctionBegin; 2545c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 2559566063dSJacob Faibussowitsch PetscCall(PetscDrawIsNull(hist->win, &isnull)); 2563ba16761SJacob Faibussowitsch if (isnull) PetscFunctionReturn(PETSC_SUCCESS); 2579566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)hist), &rank)); 258e118a51fSLisandro Dalcin 2593ba16761SJacob Faibussowitsch if ((hist->xmin >= hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(PETSC_SUCCESS); 2603ba16761SJacob Faibussowitsch if (hist->numValues < 1) PetscFunctionReturn(PETSC_SUCCESS); 2615c6c1daeSBarry Smith 2625c6c1daeSBarry Smith color = hist->color; 26371917b75SLisandro Dalcin if (color == PETSC_DRAW_ROTATE) bcolor = PETSC_DRAW_BLACK + 1; 264a297a907SKarl Rupp else bcolor = color; 265a297a907SKarl Rupp 2665c6c1daeSBarry Smith xmin = hist->xmin; 2675c6c1daeSBarry Smith xmax = hist->xmax; 2685c6c1daeSBarry Smith ymin = hist->ymin; 2695c6c1daeSBarry Smith ymax = hist->ymax; 2705c6c1daeSBarry Smith numValues = hist->numValues; 2715c6c1daeSBarry Smith values = hist->values; 2725c6c1daeSBarry Smith mean = 0.0; 2735c6c1daeSBarry Smith var = 0.0; 2745c6c1daeSBarry Smith 2755b399a63SLisandro Dalcin draw = hist->win; 2769566063dSJacob Faibussowitsch PetscCall(PetscDrawCheckResizedWindow(draw)); 2779566063dSJacob Faibussowitsch PetscCall(PetscDrawClear(draw)); 278e118a51fSLisandro Dalcin 2795c6c1daeSBarry Smith if (xmin == xmax) { 2805c6c1daeSBarry Smith /* Calculate number of points in each bin */ 2815c6c1daeSBarry Smith bins = hist->bins; 2825c6c1daeSBarry Smith bins[0] = 0.; 2835c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 2845c6c1daeSBarry Smith if (values[p] == xmin) bins[0]++; 2855c6c1daeSBarry Smith mean += values[p]; 2865c6c1daeSBarry Smith var += values[p] * values[p]; 2875c6c1daeSBarry Smith } 2885c6c1daeSBarry Smith maxHeight = bins[0]; 2895c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 2905c6c1daeSBarry Smith xmax = xmin + 1; 2919566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax)); 2925c6c1daeSBarry Smith if (hist->calcStats) { 2936497c311SBarry Smith mean /= (PetscReal)numValues; 2946497c311SBarry Smith if (numValues > 1) var = (var - ((PetscReal)numValues) * mean * mean) / ((PetscReal)numValues - 1); 295a297a907SKarl Rupp else var = 0.0; 2969566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(title, 256, "Mean: %g Var: %g", (double)mean, (double)var)); 2979566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(xlabel, 256, "Total: %" PetscInt_FMT, numValues)); 2989566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL)); 2995c6c1daeSBarry Smith } 3009566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDraw(hist->axis)); 301d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 302dd400576SPatrick Sanan if (rank == 0) { /* Draw bins */ 3035c6c1daeSBarry Smith binLeft = xmin; 3045c6c1daeSBarry Smith binRight = xmax; 3059566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw, binLeft, ymin, binRight, bins[0], bcolor, bcolor, bcolor, bcolor)); 3069566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binLeft, ymin, binLeft, bins[0], PETSC_DRAW_BLACK)); 3079566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binRight, ymin, binRight, bins[0], PETSC_DRAW_BLACK)); 3089566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binLeft, bins[0], binRight, bins[0], PETSC_DRAW_BLACK)); 309e118a51fSLisandro Dalcin } 310d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 3115c6c1daeSBarry Smith } else { 3125c6c1daeSBarry Smith numBins = hist->numBins; 3135c6c1daeSBarry Smith numBinsOld = hist->numBins; 3145c6c1daeSBarry Smith if (hist->integerBins && (((int)xmax - xmin) + 1.0e-05 > xmax - xmin)) { 315*835f2295SStefano Zampini initSize = ((int)(xmax - xmin)) / numBins; 3165c6c1daeSBarry Smith while (initSize * numBins != (int)xmax - xmin) { 3175c6c1daeSBarry Smith initSize = PetscMax(initSize - 1, 1); 318*835f2295SStefano Zampini numBins = ((int)(xmax - xmin)) / initSize; 3199566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist, numBins)); 3205c6c1daeSBarry Smith } 3215c6c1daeSBarry Smith } 3226497c311SBarry Smith binSize = (xmax - xmin) / (PetscReal)numBins; 3235c6c1daeSBarry Smith bins = hist->bins; 3245c6c1daeSBarry Smith 3259566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(bins, numBins)); 326a297a907SKarl Rupp 3275c6c1daeSBarry Smith maxHeight = 0.0; 3285c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3295c6c1daeSBarry Smith binLeft = xmin + binSize * i; 3305c6c1daeSBarry Smith binRight = xmin + binSize * (i + 1); 3315c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 3325c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 3335c6c1daeSBarry Smith /* Handle last bin separately */ 3345c6c1daeSBarry Smith if ((i == numBins - 1) && (values[p] == binRight)) bins[i]++; 3355c6c1daeSBarry Smith if (!i) { 3365c6c1daeSBarry Smith mean += values[p]; 3375c6c1daeSBarry Smith var += values[p] * values[p]; 3385c6c1daeSBarry Smith } 3395c6c1daeSBarry Smith } 3405c6c1daeSBarry Smith maxHeight = PetscMax(maxHeight, bins[i]); 3415c6c1daeSBarry Smith } 3425c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 3435c6c1daeSBarry Smith 3449566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax)); 3455c6c1daeSBarry Smith if (hist->calcStats) { 3465c6c1daeSBarry Smith mean /= numValues; 347a297a907SKarl Rupp if (numValues > 1) var = (var - numValues * mean * mean) / (numValues - 1); 348a297a907SKarl Rupp else var = 0.0; 3499566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(title, 256, "Mean: %g Var: %g", (double)mean, (double)var)); 3509566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(xlabel, 256, "Total: %" PetscInt_FMT, numValues)); 3519566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL)); 3525c6c1daeSBarry Smith } 3539566063dSJacob Faibussowitsch PetscCall(PetscDrawAxisDraw(hist->axis)); 354d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 355dd400576SPatrick Sanan if (rank == 0) { /* Draw bins */ 3565c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3575c6c1daeSBarry Smith binLeft = xmin + binSize * i; 3585c6c1daeSBarry Smith binRight = xmin + binSize * (i + 1); 3599566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw, binLeft, ymin, binRight, bins[i], bcolor, bcolor, bcolor, bcolor)); 3609566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binLeft, ymin, binLeft, bins[i], PETSC_DRAW_BLACK)); 3619566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binRight, ymin, binRight, bins[i], PETSC_DRAW_BLACK)); 3629566063dSJacob Faibussowitsch PetscCall(PetscDrawLine(draw, binLeft, bins[i], binRight, bins[i], PETSC_DRAW_BLACK)); 363e118a51fSLisandro Dalcin if (color == PETSC_DRAW_ROTATE && bins[i]) bcolor++; 36471917b75SLisandro Dalcin if (bcolor > PETSC_DRAW_BASIC_COLORS - 1) bcolor = PETSC_DRAW_BLACK + 1; 365e118a51fSLisandro Dalcin } 3665c6c1daeSBarry Smith } 367d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 3689566063dSJacob Faibussowitsch PetscCall(PetscDrawHGSetNumberBins(hist, numBinsOld)); 3695c6c1daeSBarry Smith } 370e118a51fSLisandro Dalcin 3719566063dSJacob Faibussowitsch PetscCall(PetscDrawFlush(draw)); 3729566063dSJacob Faibussowitsch PetscCall(PetscDrawPause(draw)); 3733ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3745c6c1daeSBarry Smith } 3755c6c1daeSBarry Smith 37657fd6651SLisandro Dalcin /*@ 37757fd6651SLisandro Dalcin PetscDrawHGSave - Saves a drawn image 37857fd6651SLisandro Dalcin 379c3339decSBarry Smith Collective 38057fd6651SLisandro Dalcin 38157fd6651SLisandro Dalcin Input Parameter: 382aec76313SJacob Faibussowitsch . hg - The histogram context 38357fd6651SLisandro Dalcin 38457fd6651SLisandro Dalcin Level: intermediate 38557fd6651SLisandro Dalcin 386aec76313SJacob Faibussowitsch .seealso: `PetscDrawSave()`, `PetscDrawHGCreate()`, `PetscDrawHGGetDraw()`, `PetscDrawSetSave()`, `PetscDrawHGDraw()` 38757fd6651SLisandro Dalcin @*/ 388d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGSave(PetscDrawHG hg) 389d71ae5a4SJacob Faibussowitsch { 39057fd6651SLisandro Dalcin PetscFunctionBegin; 39157fd6651SLisandro Dalcin PetscValidHeaderSpecific(hg, PETSC_DRAWHG_CLASSID, 1); 3929566063dSJacob Faibussowitsch PetscCall(PetscDrawSave(hg->win)); 3933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 39457fd6651SLisandro Dalcin } 39557fd6651SLisandro Dalcin 3965c6c1daeSBarry Smith /*@ 397811af0c4SBarry Smith PetscDrawHGView - Prints the histogram information to a viewer 3985c6c1daeSBarry Smith 39920f4b53cSBarry Smith Not Collective 4005c6c1daeSBarry Smith 40110450e9eSJacob Faibussowitsch Input Parameters: 40210450e9eSJacob Faibussowitsch + hist - The histogram context 40310450e9eSJacob Faibussowitsch - viewer - The viewer to view it with 4045c6c1daeSBarry Smith 4055c6c1daeSBarry Smith Level: beginner 4065c6c1daeSBarry Smith 407811af0c4SBarry Smith .seealso: `PetscDrawHG`, `PetscViewer`, `PetscDrawHGCreate()`, `PetscDrawHGGetDraw()`, `PetscDrawSetSave()`, `PetscDrawSave()`, `PetscDrawHGDraw()` 4085c6c1daeSBarry Smith @*/ 409d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGView(PetscDrawHG hist, PetscViewer viewer) 410d71ae5a4SJacob Faibussowitsch { 4115c6c1daeSBarry Smith PetscReal xmax, xmin, *bins, *values, binSize, binLeft, binRight, mean, var; 4125c6c1daeSBarry Smith PetscInt numBins, numBinsOld, numValues, initSize, i, p; 413*835f2295SStefano Zampini int inumBins; 4145c6c1daeSBarry Smith 4155c6c1daeSBarry Smith PetscFunctionBegin; 4165c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 417e118a51fSLisandro Dalcin 4183ba16761SJacob Faibussowitsch if ((hist->xmin > hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(PETSC_SUCCESS); 4193ba16761SJacob Faibussowitsch if (hist->numValues < 1) PetscFunctionReturn(PETSC_SUCCESS); 4205c6c1daeSBarry Smith 42148a46eb9SPierre Jolivet if (!viewer) PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)hist), &viewer)); 4229566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)hist, viewer)); 4235c6c1daeSBarry Smith xmax = hist->xmax; 4245c6c1daeSBarry Smith xmin = hist->xmin; 4255c6c1daeSBarry Smith numValues = hist->numValues; 4265c6c1daeSBarry Smith values = hist->values; 4275c6c1daeSBarry Smith mean = 0.0; 4285c6c1daeSBarry Smith var = 0.0; 4295c6c1daeSBarry Smith if (xmax == xmin) { 4305c6c1daeSBarry Smith /* Calculate number of points in the bin */ 4315c6c1daeSBarry Smith bins = hist->bins; 4325c6c1daeSBarry Smith bins[0] = 0.; 4335c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 4345c6c1daeSBarry Smith if (values[p] == xmin) bins[0]++; 4355c6c1daeSBarry Smith mean += values[p]; 4365c6c1daeSBarry Smith var += values[p] * values[p]; 4375c6c1daeSBarry Smith } 4385c6c1daeSBarry Smith /* Draw bins */ 4399566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Bin %2d (%6.2g - %6.2g): %.0g\n", 0, (double)xmin, (double)xmax, (double)bins[0])); 4405c6c1daeSBarry Smith } else { 4415c6c1daeSBarry Smith numBins = hist->numBins; 4425c6c1daeSBarry Smith numBinsOld = hist->numBins; 4435c6c1daeSBarry Smith if (hist->integerBins && (((int)xmax - xmin) + 1.0e-05 > xmax - xmin)) { 4445c6c1daeSBarry Smith initSize = (int)((int)xmax - xmin) / numBins; 4455c6c1daeSBarry Smith while (initSize * numBins != (int)xmax - xmin) { 4465c6c1daeSBarry Smith initSize = PetscMax(initSize - 1, 1); 4475c6c1daeSBarry Smith numBins = (int)((int)xmax - xmin) / initSize; 448*835f2295SStefano Zampini PetscCall(PetscCIntCast(numBins, &inumBins)); 449*835f2295SStefano Zampini PetscCall(PetscDrawHGSetNumberBins(hist, inumBins)); 4505c6c1daeSBarry Smith } 4515c6c1daeSBarry Smith } 4525c6c1daeSBarry Smith binSize = (xmax - xmin) / numBins; 4535c6c1daeSBarry Smith bins = hist->bins; 4545c6c1daeSBarry Smith 4555c6c1daeSBarry Smith /* Calculate number of points in each bin */ 4569566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(bins, numBins)); 4575c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4585c6c1daeSBarry Smith binLeft = xmin + binSize * i; 4595c6c1daeSBarry Smith binRight = xmin + binSize * (i + 1); 4605c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 4615c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 4625c6c1daeSBarry Smith /* Handle last bin separately */ 4635c6c1daeSBarry Smith if ((i == numBins - 1) && (values[p] == binRight)) bins[i]++; 4645c6c1daeSBarry Smith if (!i) { 4655c6c1daeSBarry Smith mean += values[p]; 4665c6c1daeSBarry Smith var += values[p] * values[p]; 4675c6c1daeSBarry Smith } 4685c6c1daeSBarry Smith } 4695c6c1daeSBarry Smith } 4705c6c1daeSBarry Smith /* Draw bins */ 4715c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4725c6c1daeSBarry Smith binLeft = xmin + binSize * i; 4735c6c1daeSBarry Smith binRight = xmin + binSize * (i + 1); 474*835f2295SStefano Zampini PetscCall(PetscViewerASCIIPrintf(viewer, "Bin %2" PetscInt_FMT " (%6.2g - %6.2g): %.0g\n", i, (double)binLeft, (double)binRight, (double)bins[i])); 4755c6c1daeSBarry Smith } 476*835f2295SStefano Zampini PetscCall(PetscCIntCast(numBinsOld, &inumBins)); 477*835f2295SStefano Zampini PetscCall(PetscDrawHGSetNumberBins(hist, inumBins)); 4785c6c1daeSBarry Smith } 4795c6c1daeSBarry Smith 4805c6c1daeSBarry Smith if (hist->calcStats) { 4815c6c1daeSBarry Smith mean /= numValues; 482a297a907SKarl Rupp if (numValues > 1) var = (var - numValues * mean * mean) / (numValues - 1); 483a297a907SKarl Rupp else var = 0.0; 4849566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Mean: %g Var: %g\n", (double)mean, (double)var)); 4859566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Total: %" PetscInt_FMT "\n", numValues)); 4865c6c1daeSBarry Smith } 4873ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4885c6c1daeSBarry Smith } 4895c6c1daeSBarry Smith 4905c6c1daeSBarry Smith /*@ 4915c6c1daeSBarry Smith PetscDrawHGSetColor - Sets the color the bars will be drawn with. 4925c6c1daeSBarry Smith 493c3339decSBarry Smith Logically Collective 4945c6c1daeSBarry Smith 4955c6c1daeSBarry Smith Input Parameters: 4965c6c1daeSBarry Smith + hist - The histogram context 4972fe279fdSBarry Smith - color - one of the colors defined in petscdraw.h or `PETSC_DRAW_ROTATE` to make each bar a different color 4985c6c1daeSBarry Smith 4995c6c1daeSBarry Smith Level: intermediate 5005c6c1daeSBarry Smith 501811af0c4SBarry Smith .seealso: `PetscDrawHG`, `PetscDrawHGCreate()`, `PetscDrawHGGetDraw()`, `PetscDrawSetSave()`, `PetscDrawSave()`, `PetscDrawHGDraw()`, `PetscDrawHGGetAxis()` 5025c6c1daeSBarry Smith @*/ 503d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGSetColor(PetscDrawHG hist, int color) 504d71ae5a4SJacob Faibussowitsch { 5055c6c1daeSBarry Smith PetscFunctionBegin; 5065c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 507e118a51fSLisandro Dalcin 5085c6c1daeSBarry Smith hist->color = color; 5093ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5105c6c1daeSBarry Smith } 5115c6c1daeSBarry Smith 5125c6c1daeSBarry Smith /*@ 5135c6c1daeSBarry Smith PetscDrawHGSetLimits - Sets the axis limits for a histogram. If more 5145c6c1daeSBarry Smith points are added after this call, the limits will be adjusted to 5155c6c1daeSBarry Smith include those additional points. 5165c6c1daeSBarry Smith 517c3339decSBarry Smith Logically Collective 5185c6c1daeSBarry Smith 5195c6c1daeSBarry Smith Input Parameters: 5205c6c1daeSBarry Smith + hist - The histogram context 5212fe279fdSBarry Smith . x_min - the horizontal lower limit 522aaa8cc7dSPierre Jolivet . x_max - the horizontal upper limit 5232fe279fdSBarry Smith . y_min - the vertical lower limit 5242fe279fdSBarry Smith - y_max - the vertical upper limit 5255c6c1daeSBarry Smith 5265c6c1daeSBarry Smith Level: intermediate 5275c6c1daeSBarry Smith 528811af0c4SBarry Smith .seealso: `PetscDrawHG`, `PetscDrawHGCreate()`, `PetscDrawHGGetDraw()`, `PetscDrawSetSave()`, `PetscDrawSave()`, `PetscDrawHGDraw()`, `PetscDrawHGGetAxis()` 5295c6c1daeSBarry Smith @*/ 530d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGSetLimits(PetscDrawHG hist, PetscReal x_min, PetscReal x_max, int y_min, int y_max) 531d71ae5a4SJacob Faibussowitsch { 5325c6c1daeSBarry Smith PetscFunctionBegin; 5335c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 534e118a51fSLisandro Dalcin 5355c6c1daeSBarry Smith hist->xmin = x_min; 5365c6c1daeSBarry Smith hist->xmax = x_max; 5375c6c1daeSBarry Smith hist->ymin = y_min; 5385c6c1daeSBarry Smith hist->ymax = y_max; 5393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5405c6c1daeSBarry Smith } 5415c6c1daeSBarry Smith 5425c6c1daeSBarry Smith /*@ 543811af0c4SBarry Smith PetscDrawHGCalcStats - Turns on calculation of descriptive statistics associated with the histogram 5445c6c1daeSBarry Smith 54520f4b53cSBarry Smith Not Collective 5465c6c1daeSBarry Smith 5475c6c1daeSBarry Smith Input Parameters: 5485c6c1daeSBarry Smith + hist - The histogram context 5495c6c1daeSBarry Smith - calc - Flag for calculation 5505c6c1daeSBarry Smith 5515c6c1daeSBarry Smith Level: intermediate 5525c6c1daeSBarry Smith 553811af0c4SBarry Smith .seealso: `PetscDrawHG`, `PetscDrawHGCreate()`, `PetscDrawHGAddValue()`, `PetscDrawHGView()`, `PetscDrawHGDraw()` 5545c6c1daeSBarry Smith @*/ 555d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGCalcStats(PetscDrawHG hist, PetscBool calc) 556d71ae5a4SJacob Faibussowitsch { 5575c6c1daeSBarry Smith PetscFunctionBegin; 5585c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 559e118a51fSLisandro Dalcin 5605c6c1daeSBarry Smith hist->calcStats = calc; 5613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5625c6c1daeSBarry Smith } 5635c6c1daeSBarry Smith 5645c6c1daeSBarry Smith /*@ 5655c6c1daeSBarry Smith PetscDrawHGIntegerBins - Turns on integer width bins 5665c6c1daeSBarry Smith 56720f4b53cSBarry Smith Not Collective 5685c6c1daeSBarry Smith 5695c6c1daeSBarry Smith Input Parameters: 5705c6c1daeSBarry Smith + hist - The histogram context 5715c6c1daeSBarry Smith - ints - Flag for integer width bins 5725c6c1daeSBarry Smith 5735c6c1daeSBarry Smith Level: intermediate 5745c6c1daeSBarry Smith 575811af0c4SBarry Smith .seealso: `PetscDrawHG`, `PetscDrawHGCreate()`, `PetscDrawHGAddValue()`, `PetscDrawHGView()`, `PetscDrawHGDraw()`, `PetscDrawHGSetColor()` 5765c6c1daeSBarry Smith @*/ 577d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGIntegerBins(PetscDrawHG hist, PetscBool ints) 578d71ae5a4SJacob Faibussowitsch { 5795c6c1daeSBarry Smith PetscFunctionBegin; 5805c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 581e118a51fSLisandro Dalcin 5825c6c1daeSBarry Smith hist->integerBins = ints; 5833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5845c6c1daeSBarry Smith } 5855c6c1daeSBarry Smith 5865d83a8b1SBarry Smith /*@ 5875c6c1daeSBarry Smith PetscDrawHGGetAxis - Gets the axis context associated with a histogram. 5885c6c1daeSBarry Smith This is useful if one wants to change some axis property, such as 5895c6c1daeSBarry Smith labels, color, etc. The axis context should not be destroyed by the 5905c6c1daeSBarry Smith application code. 5915c6c1daeSBarry Smith 592811af0c4SBarry Smith Not Collective, axis is parallel if hist is parallel 5935c6c1daeSBarry Smith 5945c6c1daeSBarry Smith Input Parameter: 5955c6c1daeSBarry Smith . hist - The histogram context 5965c6c1daeSBarry Smith 5975c6c1daeSBarry Smith Output Parameter: 5985c6c1daeSBarry Smith . axis - The axis context 5995c6c1daeSBarry Smith 6005c6c1daeSBarry Smith Level: intermediate 6015c6c1daeSBarry Smith 602aec76313SJacob Faibussowitsch .seealso: `PetscDrawHG`, `PetscDrawAxis`, `PetscDrawHGCreate()`, `PetscDrawHGAddValue()`, `PetscDrawHGView()`, `PetscDrawHGDraw()`, `PetscDrawHGSetColor()`, `PetscDrawHGSetLimits()` 6035c6c1daeSBarry Smith @*/ 604d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGGetAxis(PetscDrawHG hist, PetscDrawAxis *axis) 605d71ae5a4SJacob Faibussowitsch { 6065c6c1daeSBarry Smith PetscFunctionBegin; 607e118a51fSLisandro Dalcin PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 6084f572ea9SToby Isaac PetscAssertPointer(axis, 2); 6095c6c1daeSBarry Smith *axis = hist->axis; 6103ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6115c6c1daeSBarry Smith } 6125c6c1daeSBarry Smith 6135d83a8b1SBarry Smith /*@ 6145c6c1daeSBarry Smith PetscDrawHGGetDraw - Gets the draw context associated with a histogram. 6155c6c1daeSBarry Smith 616811af0c4SBarry Smith Not Collective, draw is parallel if hist is parallel 6175c6c1daeSBarry Smith 6185c6c1daeSBarry Smith Input Parameter: 6195c6c1daeSBarry Smith . hist - The histogram context 6205c6c1daeSBarry Smith 6215c6c1daeSBarry Smith Output Parameter: 622e118a51fSLisandro Dalcin . draw - The draw context 6235c6c1daeSBarry Smith 6245c6c1daeSBarry Smith Level: intermediate 6255c6c1daeSBarry Smith 626811af0c4SBarry Smith .seealso: `PetscDraw`, `PetscDrawHG`, `PetscDrawHGCreate()`, `PetscDrawHGAddValue()`, `PetscDrawHGView()`, `PetscDrawHGDraw()`, `PetscDrawHGSetColor()`, `PetscDrawAxis`, `PetscDrawHGSetLimits()` 6275c6c1daeSBarry Smith @*/ 628d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscDrawHGGetDraw(PetscDrawHG hist, PetscDraw *draw) 629d71ae5a4SJacob Faibussowitsch { 6305c6c1daeSBarry Smith PetscFunctionBegin; 6315c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID, 1); 6324f572ea9SToby Isaac PetscAssertPointer(draw, 2); 633e118a51fSLisandro Dalcin *draw = hist->win; 6343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6355c6c1daeSBarry Smith } 636