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 #undef __FUNCT__ 335c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGCreate" 345c6c1daeSBarry Smith /*@C 355c6c1daeSBarry Smith PetscDrawHGCreate - Creates a histogram data structure. 365c6c1daeSBarry Smith 375c6c1daeSBarry Smith Collective over PetscDraw 385c6c1daeSBarry Smith 395c6c1daeSBarry Smith Input Parameters: 405c6c1daeSBarry Smith + draw - The window where the graph will be made 415c6c1daeSBarry Smith - bins - The number of bins to use 425c6c1daeSBarry Smith 435c6c1daeSBarry Smith Output Parameters: 445c6c1daeSBarry Smith . hist - The histogram context 455c6c1daeSBarry Smith 465c6c1daeSBarry Smith Level: intermediate 475c6c1daeSBarry Smith 485c6c1daeSBarry Smith Concepts: histogram^creating 495c6c1daeSBarry Smith 505c6c1daeSBarry Smith .seealso: PetscDrawHGDestroy() 515c6c1daeSBarry Smith 525c6c1daeSBarry Smith @*/ 530adebc6cSBarry Smith PetscErrorCode PetscDrawHGCreate(PetscDraw draw, int bins, PetscDrawHG *hist) 540adebc6cSBarry Smith { 555c6c1daeSBarry Smith PetscDrawHG h; 565c6c1daeSBarry Smith PetscBool isnull; 575c6c1daeSBarry Smith PetscErrorCode ierr; 585c6c1daeSBarry Smith 595c6c1daeSBarry Smith PetscFunctionBegin; 605c6c1daeSBarry Smith PetscValidHeaderSpecific(draw, PETSC_DRAW_CLASSID,1); 615c6c1daeSBarry Smith PetscValidPointer(hist,3); 62*73107ff1SLisandro Dalcin ierr = PetscHeaderCreate(h, PETSC_DRAWHG_CLASSID, "PetscDrawHG", "Histogram", "Draw", PetscObjectComm((PetscObject)draw), PetscDrawHGDestroy, NULL);CHKERRQ(ierr); 63a297a907SKarl Rupp 640298fd71SBarry Smith h->view = NULL; 650298fd71SBarry Smith h->destroy = NULL; 665c6c1daeSBarry Smith h->win = draw; 67a297a907SKarl Rupp 685c6c1daeSBarry Smith ierr = PetscObjectReference((PetscObject) draw);CHKERRQ(ierr); 69a297a907SKarl Rupp 705c6c1daeSBarry Smith h->color = PETSC_DRAW_GREEN; 715c6c1daeSBarry Smith h->xmin = PETSC_MAX_REAL; 725c6c1daeSBarry Smith h->xmax = PETSC_MIN_REAL; 735c6c1daeSBarry Smith h->ymin = 0.; 745c6c1daeSBarry Smith h->ymax = 1.; 755c6c1daeSBarry Smith h->numBins = bins; 765c6c1daeSBarry Smith h->maxBins = bins; 77a297a907SKarl Rupp 78785e854fSJed Brown ierr = PetscMalloc1(h->maxBins, &h->bins);CHKERRQ(ierr); 79a297a907SKarl Rupp 805c6c1daeSBarry Smith h->numValues = 0; 815c6c1daeSBarry Smith h->maxValues = CHUNKSIZE; 825c6c1daeSBarry Smith h->calcStats = PETSC_FALSE; 835c6c1daeSBarry Smith h->integerBins = PETSC_FALSE; 84a297a907SKarl Rupp 85785e854fSJed Brown ierr = PetscMalloc1(h->maxValues, &h->values);CHKERRQ(ierr); 863bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)h, (h->maxBins + h->maxValues)*sizeof(PetscReal));CHKERRQ(ierr); 875c6c1daeSBarry Smith ierr = PetscObjectTypeCompare((PetscObject) draw, PETSC_DRAW_NULL, &isnull);CHKERRQ(ierr); 88a297a907SKarl Rupp 895c6c1daeSBarry Smith if (!isnull) { 905c6c1daeSBarry Smith ierr = PetscDrawAxisCreate(draw, &h->axis);CHKERRQ(ierr); 913bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)h, (PetscObject)h->axis);CHKERRQ(ierr); 920298fd71SBarry Smith } else h->axis = NULL; 935c6c1daeSBarry Smith *hist = h; 945c6c1daeSBarry Smith PetscFunctionReturn(0); 955c6c1daeSBarry Smith } 965c6c1daeSBarry Smith 975c6c1daeSBarry Smith #undef __FUNCT__ 985c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGSetNumberBins" 995c6c1daeSBarry Smith /*@ 1005c6c1daeSBarry Smith PetscDrawHGSetNumberBins - Change the number of bins that are to be drawn. 1015c6c1daeSBarry Smith 1025c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 1035c6c1daeSBarry Smith 1045c6c1daeSBarry Smith Input Parameter: 1055c6c1daeSBarry Smith + hist - The histogram context. 1065c6c1daeSBarry Smith - dim - The number of curves. 1075c6c1daeSBarry Smith 1085c6c1daeSBarry Smith Level: intermediate 1095c6c1daeSBarry Smith 1105c6c1daeSBarry Smith Concepts: histogram^setting number of bins 1115c6c1daeSBarry Smith 1125c6c1daeSBarry Smith @*/ 1135c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetNumberBins(PetscDrawHG hist, int bins) 1145c6c1daeSBarry Smith { 1155c6c1daeSBarry Smith PetscErrorCode ierr; 1165c6c1daeSBarry Smith 1175c6c1daeSBarry Smith PetscFunctionBegin; 1185c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 1195c6c1daeSBarry Smith if (hist->maxBins < bins) { 1205c6c1daeSBarry Smith ierr = PetscFree(hist->bins);CHKERRQ(ierr); 121785e854fSJed Brown ierr = PetscMalloc1(bins, &hist->bins);CHKERRQ(ierr); 1223bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)hist, (bins - hist->maxBins) * sizeof(PetscReal));CHKERRQ(ierr); 123a297a907SKarl Rupp 1245c6c1daeSBarry Smith hist->maxBins = bins; 1255c6c1daeSBarry Smith } 1265c6c1daeSBarry Smith hist->numBins = bins; 1275c6c1daeSBarry Smith PetscFunctionReturn(0); 1285c6c1daeSBarry Smith } 1295c6c1daeSBarry Smith 1305c6c1daeSBarry Smith #undef __FUNCT__ 1315c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGReset" 1325c6c1daeSBarry Smith /*@ 1335c6c1daeSBarry Smith PetscDrawHGReset - Clears histogram to allow for reuse with new data. 1345c6c1daeSBarry Smith 1355c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 1365c6c1daeSBarry Smith 1375c6c1daeSBarry Smith Input Parameter: 1385c6c1daeSBarry Smith . hist - The histogram context. 1395c6c1daeSBarry Smith 1405c6c1daeSBarry Smith Level: intermediate 1415c6c1daeSBarry Smith 1425c6c1daeSBarry Smith Concepts: histogram^resetting 1435c6c1daeSBarry Smith @*/ 1445c6c1daeSBarry Smith PetscErrorCode PetscDrawHGReset(PetscDrawHG hist) 1455c6c1daeSBarry Smith { 1465c6c1daeSBarry Smith PetscFunctionBegin; 1475c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 1485c6c1daeSBarry Smith hist->xmin = PETSC_MAX_REAL; 1495c6c1daeSBarry Smith hist->xmax = PETSC_MIN_REAL; 1505c6c1daeSBarry Smith hist->ymin = 0.0; 1515c6c1daeSBarry Smith hist->ymax = 0.0; 1525c6c1daeSBarry Smith hist->numValues = 0; 1535c6c1daeSBarry Smith PetscFunctionReturn(0); 1545c6c1daeSBarry Smith } 1555c6c1daeSBarry Smith 1565c6c1daeSBarry Smith #undef __FUNCT__ 1575c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGDestroy" 1585c6c1daeSBarry Smith /*@C 1595c6c1daeSBarry Smith PetscDrawHGDestroy - Frees all space taken up by histogram data structure. 1605c6c1daeSBarry Smith 1615c6c1daeSBarry Smith Collective over PetscDrawHG 1625c6c1daeSBarry Smith 1635c6c1daeSBarry Smith Input Parameter: 1645c6c1daeSBarry Smith . hist - The histogram context 1655c6c1daeSBarry Smith 1665c6c1daeSBarry Smith Level: intermediate 1675c6c1daeSBarry Smith 1685c6c1daeSBarry Smith .seealso: PetscDrawHGCreate() 1695c6c1daeSBarry Smith @*/ 1705c6c1daeSBarry Smith PetscErrorCode PetscDrawHGDestroy(PetscDrawHG *hist) 1715c6c1daeSBarry Smith { 1725c6c1daeSBarry Smith PetscErrorCode ierr; 1735c6c1daeSBarry Smith 1745c6c1daeSBarry Smith PetscFunctionBegin; 1755c6c1daeSBarry Smith if (!*hist) PetscFunctionReturn(0); 1765c6c1daeSBarry Smith PetscValidHeader(*hist,1); 1775c6c1daeSBarry Smith 1785c6c1daeSBarry Smith if (--((PetscObject)(*hist))->refct > 0) PetscFunctionReturn(0); 1795c6c1daeSBarry Smith ierr = PetscDrawAxisDestroy(&(*hist)->axis);CHKERRQ(ierr); 1805c6c1daeSBarry Smith ierr = PetscDrawDestroy(&(*hist)->win);CHKERRQ(ierr); 1815c6c1daeSBarry Smith ierr = PetscFree((*hist)->bins);CHKERRQ(ierr); 1825c6c1daeSBarry Smith ierr = PetscFree((*hist)->values);CHKERRQ(ierr); 1835c6c1daeSBarry Smith ierr = PetscHeaderDestroy(hist);CHKERRQ(ierr); 1845c6c1daeSBarry Smith PetscFunctionReturn(0); 1855c6c1daeSBarry Smith } 1865c6c1daeSBarry Smith 1875c6c1daeSBarry Smith #undef __FUNCT__ 1885c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGAddValue" 1895c6c1daeSBarry Smith /*@ 1905c6c1daeSBarry Smith PetscDrawHGAddValue - Adds another value to the histogram. 1915c6c1daeSBarry Smith 1925c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 1935c6c1daeSBarry Smith 1945c6c1daeSBarry Smith Input Parameters: 1955c6c1daeSBarry Smith + hist - The histogram 1965c6c1daeSBarry Smith - value - The value 1975c6c1daeSBarry Smith 1985c6c1daeSBarry Smith Level: intermediate 1995c6c1daeSBarry Smith 2005c6c1daeSBarry Smith Concepts: histogram^adding values 2015c6c1daeSBarry Smith 2025c6c1daeSBarry Smith .seealso: PetscDrawHGAddValues() 2035c6c1daeSBarry Smith @*/ 2045c6c1daeSBarry Smith PetscErrorCode PetscDrawHGAddValue(PetscDrawHG hist, PetscReal value) 2055c6c1daeSBarry Smith { 2065c6c1daeSBarry Smith PetscFunctionBegin; 2075c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 2085c6c1daeSBarry Smith /* Allocate more memory if necessary */ 2095c6c1daeSBarry Smith if (hist->numValues >= hist->maxValues) { 2105c6c1daeSBarry Smith PetscReal *tmp; 2115c6c1daeSBarry Smith PetscErrorCode ierr; 2125c6c1daeSBarry Smith 213854ce69bSBarry Smith ierr = PetscMalloc1(hist->maxValues+CHUNKSIZE, &tmp);CHKERRQ(ierr); 2143bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)hist, CHUNKSIZE * sizeof(PetscReal));CHKERRQ(ierr); 2155c6c1daeSBarry Smith ierr = PetscMemcpy(tmp, hist->values, hist->maxValues * sizeof(PetscReal));CHKERRQ(ierr); 2165c6c1daeSBarry Smith ierr = PetscFree(hist->values);CHKERRQ(ierr); 217a297a907SKarl Rupp 2185c6c1daeSBarry Smith hist->values = tmp; 2195c6c1daeSBarry Smith hist->maxValues += CHUNKSIZE; 2205c6c1daeSBarry Smith } 2215c6c1daeSBarry Smith /* I disagree with the original Petsc implementation here. There should be no overshoot, but rather the 2225c6c1daeSBarry Smith stated convention of using half-open intervals (always the way to go) */ 2235c6c1daeSBarry Smith if (!hist->numValues) { 2245c6c1daeSBarry Smith hist->xmin = value; 2255c6c1daeSBarry Smith hist->xmax = value; 2265c6c1daeSBarry Smith #if 1 2275c6c1daeSBarry Smith } else { 2285c6c1daeSBarry Smith /* Update limits */ 229a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value; 230a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2315c6c1daeSBarry Smith #else 2325c6c1daeSBarry Smith } else if (hist->numValues == 1) { 2335c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 234a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value + 0.001*(value - hist->xmin)/hist->numBins; 2355c6c1daeSBarry Smith if (value < hist->xmin) { 2365c6c1daeSBarry Smith hist->xmin = value; 2375c6c1daeSBarry Smith hist->xmax = hist->xmax + 0.001*(hist->xmax - hist->xmin)/hist->numBins; 2385c6c1daeSBarry Smith } 2395c6c1daeSBarry Smith } else { 2405c6c1daeSBarry Smith /* Update limits -- We need to overshoot the largest value somewhat */ 241a297a907SKarl Rupp if (value > hist->xmax) hist->xmax = value + 0.001*(hist->xmax - hist->xmin)/hist->numBins; 242a297a907SKarl Rupp if (value < hist->xmin) hist->xmin = value; 2435c6c1daeSBarry Smith #endif 2445c6c1daeSBarry Smith } 2455c6c1daeSBarry Smith 2465c6c1daeSBarry Smith hist->values[hist->numValues++] = value; 2475c6c1daeSBarry Smith PetscFunctionReturn(0); 2485c6c1daeSBarry Smith } 2495c6c1daeSBarry Smith 2505c6c1daeSBarry Smith #undef __FUNCT__ 2515c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGDraw" 2525c6c1daeSBarry Smith /*@ 2535c6c1daeSBarry Smith PetscDrawHGDraw - Redraws a histogram. 2545c6c1daeSBarry Smith 2555c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 2565c6c1daeSBarry Smith 2575c6c1daeSBarry Smith Input Parameter: 2585c6c1daeSBarry Smith . hist - The histogram context 2595c6c1daeSBarry Smith 2605c6c1daeSBarry Smith Level: intermediate 2615c6c1daeSBarry Smith 2625c6c1daeSBarry Smith @*/ 2635c6c1daeSBarry Smith PetscErrorCode PetscDrawHGDraw(PetscDrawHG hist) 2645c6c1daeSBarry Smith { 2655c6c1daeSBarry Smith PetscDraw draw = hist->win; 2665c6c1daeSBarry Smith PetscBool isnull; 2675c6c1daeSBarry Smith PetscReal xmin,xmax,ymin,ymax,*bins,*values,binSize,binLeft,binRight,maxHeight,mean,var; 2685c6c1daeSBarry Smith char title[256]; 2695c6c1daeSBarry Smith char xlabel[256]; 2705c6c1daeSBarry Smith PetscInt numBins,numBinsOld,numValues,initSize,i,p,bcolor,color; 2715c6c1daeSBarry Smith PetscErrorCode ierr; 2725c6c1daeSBarry Smith 2735c6c1daeSBarry Smith PetscFunctionBegin; 2745c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 2755c6c1daeSBarry Smith ierr = PetscObjectTypeCompare((PetscObject) draw, PETSC_DRAW_NULL, &isnull);CHKERRQ(ierr); 2765c6c1daeSBarry Smith if (isnull) PetscFunctionReturn(0); 2775c6c1daeSBarry Smith if ((hist->xmin >= hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(0); 2785c6c1daeSBarry Smith if (hist->numValues < 1) PetscFunctionReturn(0); 2795c6c1daeSBarry Smith 2805c6c1daeSBarry Smith #if 0 281ce94432eSBarry Smith ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)hist),&rank);CHKERRQ(ierr); 2825c6c1daeSBarry Smith if (rank) PetscFunctionReturn(0); 2835c6c1daeSBarry Smith #endif 2845c6c1daeSBarry Smith 2855c6c1daeSBarry Smith color = hist->color; 286a297a907SKarl Rupp if (color == PETSC_DRAW_ROTATE) bcolor = 2; 287a297a907SKarl Rupp else bcolor = color; 288a297a907SKarl Rupp 2895c6c1daeSBarry Smith xmin = hist->xmin; 2905c6c1daeSBarry Smith xmax = hist->xmax; 2915c6c1daeSBarry Smith ymin = hist->ymin; 2925c6c1daeSBarry Smith ymax = hist->ymax; 2935c6c1daeSBarry Smith numValues = hist->numValues; 2945c6c1daeSBarry Smith values = hist->values; 2955c6c1daeSBarry Smith mean = 0.0; 2965c6c1daeSBarry Smith var = 0.0; 2975c6c1daeSBarry Smith 2985c6c1daeSBarry Smith ierr = PetscDrawClear(draw);CHKERRQ(ierr); 2995c6c1daeSBarry Smith if (xmin == xmax) { 3005c6c1daeSBarry Smith /* Calculate number of points in each bin */ 3015c6c1daeSBarry Smith bins = hist->bins; 3025c6c1daeSBarry Smith bins[0] = 0.; 3035c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 3045c6c1daeSBarry Smith if (values[p] == xmin) bins[0]++; 3055c6c1daeSBarry Smith mean += values[p]; 3065c6c1daeSBarry Smith var += values[p]*values[p]; 3075c6c1daeSBarry Smith } 3085c6c1daeSBarry Smith maxHeight = bins[0]; 3095c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 3105c6c1daeSBarry Smith xmax = xmin + 1; 3115c6c1daeSBarry Smith ierr = PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax);CHKERRQ(ierr); 3125c6c1daeSBarry Smith if (hist->calcStats) { 3135c6c1daeSBarry Smith mean /= numValues; 314a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 315a297a907SKarl Rupp else var = 0.0; 3165c6c1daeSBarry Smith ierr = PetscSNPrintf(title, 256, "Mean: %g Var: %g", (double)mean, (double)var);CHKERRQ(ierr); 3175c6c1daeSBarry Smith ierr = PetscSNPrintf(xlabel,256, "Total: %D", numValues);CHKERRQ(ierr); 3180298fd71SBarry Smith ierr = PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL);CHKERRQ(ierr); 3195c6c1daeSBarry Smith } 3205c6c1daeSBarry Smith ierr = PetscDrawAxisDraw(hist->axis);CHKERRQ(ierr); 3215c6c1daeSBarry Smith /* Draw bins */ 3225c6c1daeSBarry Smith binLeft = xmin; 3235c6c1daeSBarry Smith binRight = xmax; 3245c6c1daeSBarry Smith ierr = PetscDrawRectangle(draw,binLeft,ymin,binRight,bins[0],bcolor,bcolor,bcolor,bcolor);CHKERRQ(ierr); 325a297a907SKarl Rupp 326a297a907SKarl Rupp if (color == PETSC_DRAW_ROTATE && bins[0] != 0.0) bcolor++; 327a297a907SKarl Rupp if (bcolor > 31) bcolor = 2; 328a297a907SKarl Rupp 3295c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binLeft,ymin,binLeft,bins[0],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3305c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binRight,ymin,binRight,bins[0],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3315c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binLeft,bins[0],binRight,bins[0],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3325c6c1daeSBarry Smith } else { 3335c6c1daeSBarry Smith numBins = hist->numBins; 3345c6c1daeSBarry Smith numBinsOld = hist->numBins; 3355c6c1daeSBarry Smith if (hist->integerBins && (((int) xmax - xmin) + 1.0e-05 > xmax - xmin)) { 3365c6c1daeSBarry Smith initSize = (int) ((int) xmax - xmin)/numBins; 3375c6c1daeSBarry Smith while (initSize*numBins != (int) xmax - xmin) { 3385c6c1daeSBarry Smith initSize = PetscMax(initSize - 1, 1); 3395c6c1daeSBarry Smith numBins = (int) ((int) xmax - xmin)/initSize; 3405c6c1daeSBarry Smith ierr = PetscDrawHGSetNumberBins(hist, numBins);CHKERRQ(ierr); 3415c6c1daeSBarry Smith } 3425c6c1daeSBarry Smith } 3435c6c1daeSBarry Smith binSize = (xmax - xmin)/numBins; 3445c6c1daeSBarry Smith bins = hist->bins; 3455c6c1daeSBarry Smith 3465c6c1daeSBarry Smith ierr = PetscMemzero(bins, numBins * sizeof(PetscReal));CHKERRQ(ierr); 347a297a907SKarl Rupp 3485c6c1daeSBarry Smith maxHeight = 0.0; 3495c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3505c6c1daeSBarry Smith binLeft = xmin + binSize*i; 3515c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 3525c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 3535c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 3545c6c1daeSBarry Smith /* Handle last bin separately */ 3555c6c1daeSBarry Smith if ((i == numBins-1) && (values[p] == binRight)) bins[i]++; 3565c6c1daeSBarry Smith if (!i) { 3575c6c1daeSBarry Smith mean += values[p]; 3585c6c1daeSBarry Smith var += values[p]*values[p]; 3595c6c1daeSBarry Smith } 3605c6c1daeSBarry Smith } 3615c6c1daeSBarry Smith maxHeight = PetscMax(maxHeight, bins[i]); 3625c6c1daeSBarry Smith } 3635c6c1daeSBarry Smith if (maxHeight > ymax) ymax = hist->ymax = maxHeight; 3645c6c1daeSBarry Smith 3655c6c1daeSBarry Smith ierr = PetscDrawAxisSetLimits(hist->axis, xmin, xmax, ymin, ymax);CHKERRQ(ierr); 3665c6c1daeSBarry Smith if (hist->calcStats) { 3675c6c1daeSBarry Smith mean /= numValues; 368a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 369a297a907SKarl Rupp else var = 0.0; 370a297a907SKarl Rupp 3715c6c1daeSBarry Smith ierr = PetscSNPrintf(title, 256,"Mean: %g Var: %g", (double)mean, (double)var);CHKERRQ(ierr); 3725c6c1daeSBarry Smith ierr = PetscSNPrintf(xlabel,256, "Total: %D", numValues);CHKERRQ(ierr); 3730298fd71SBarry Smith ierr = PetscDrawAxisSetLabels(hist->axis, title, xlabel, NULL);CHKERRQ(ierr); 3745c6c1daeSBarry Smith } 3755c6c1daeSBarry Smith ierr = PetscDrawAxisDraw(hist->axis);CHKERRQ(ierr); 3765c6c1daeSBarry Smith /* Draw bins */ 3775c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 3785c6c1daeSBarry Smith binLeft = xmin + binSize*i; 3795c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 3805c6c1daeSBarry Smith ierr = PetscDrawRectangle(draw,binLeft,ymin,binRight,bins[i],bcolor,bcolor,bcolor,bcolor);CHKERRQ(ierr); 381a297a907SKarl Rupp if (color == PETSC_DRAW_ROTATE && bins[i]) bcolor++; 382a297a907SKarl Rupp if (bcolor > 31) bcolor = 2; 3835c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binLeft,ymin,binLeft,bins[i],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3845c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binRight,ymin,binRight,bins[i],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3855c6c1daeSBarry Smith ierr = PetscDrawLine(draw,binLeft,bins[i],binRight,bins[i],PETSC_DRAW_BLACK);CHKERRQ(ierr); 3865c6c1daeSBarry Smith } 3875c6c1daeSBarry Smith ierr = PetscDrawHGSetNumberBins(hist, numBinsOld);CHKERRQ(ierr); 3885c6c1daeSBarry Smith } 3895c6c1daeSBarry Smith ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr); 3905c6c1daeSBarry Smith ierr = PetscDrawPause(draw);CHKERRQ(ierr); 3915c6c1daeSBarry Smith PetscFunctionReturn(0); 3925c6c1daeSBarry Smith } 3935c6c1daeSBarry Smith 3945c6c1daeSBarry Smith #undef __FUNCT__ 39534a5a0e3SBarry Smith #define __FUNCT__ "PetscDrawHGView" 3965c6c1daeSBarry Smith /*@ 39734a5a0e3SBarry Smith PetscDrawHGView - Prints the histogram information. 3985c6c1daeSBarry Smith 3995c6c1daeSBarry Smith Not collective 4005c6c1daeSBarry Smith 4015c6c1daeSBarry Smith Input Parameter: 4025c6c1daeSBarry Smith . hist - The histogram context 4035c6c1daeSBarry Smith 4045c6c1daeSBarry Smith Level: beginner 4055c6c1daeSBarry Smith 4065c6c1daeSBarry Smith .keywords: draw, histogram 4075c6c1daeSBarry Smith @*/ 40834a5a0e3SBarry Smith PetscErrorCode PetscDrawHGView(PetscDrawHG hist,PetscViewer viewer) 4095c6c1daeSBarry Smith { 4105c6c1daeSBarry Smith PetscReal xmax,xmin,*bins,*values,binSize,binLeft,binRight,mean,var; 4115c6c1daeSBarry Smith PetscErrorCode ierr; 4125c6c1daeSBarry Smith PetscInt numBins,numBinsOld,numValues,initSize,i,p; 4135c6c1daeSBarry Smith 4145c6c1daeSBarry Smith PetscFunctionBegin; 4155c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 4165c6c1daeSBarry Smith if ((hist->xmin > hist->xmax) || (hist->ymin >= hist->ymax)) PetscFunctionReturn(0); 4175c6c1daeSBarry Smith if (hist->numValues < 1) PetscFunctionReturn(0); 4185c6c1daeSBarry Smith 41934a5a0e3SBarry Smith if (!viewer){ 42034a5a0e3SBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)hist),&viewer);CHKERRQ(ierr); 42134a5a0e3SBarry Smith } 42298c3331eSBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)hist,viewer);CHKERRQ(ierr); 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 */ 43934a5a0e3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer, "Bin %2d (%6.2g - %6.2g): %.0g\n", 0, (double)xmin, (double)xmax, (double)bins[0]);CHKERRQ(ierr); 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; 4485c6c1daeSBarry Smith ierr = PetscDrawHGSetNumberBins(hist, numBins);CHKERRQ(ierr); 4495c6c1daeSBarry Smith } 4505c6c1daeSBarry Smith } 4515c6c1daeSBarry Smith binSize = (xmax - xmin)/numBins; 4525c6c1daeSBarry Smith bins = hist->bins; 4535c6c1daeSBarry Smith 4545c6c1daeSBarry Smith /* Calculate number of points in each bin */ 4555c6c1daeSBarry Smith ierr = PetscMemzero(bins, numBins * sizeof(PetscReal));CHKERRQ(ierr); 4565c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4575c6c1daeSBarry Smith binLeft = xmin + binSize*i; 4585c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 4595c6c1daeSBarry Smith for (p = 0; p < numValues; p++) { 4605c6c1daeSBarry Smith if ((values[p] >= binLeft) && (values[p] < binRight)) bins[i]++; 4615c6c1daeSBarry Smith /* Handle last bin separately */ 4625c6c1daeSBarry Smith if ((i == numBins-1) && (values[p] == binRight)) bins[i]++; 4635c6c1daeSBarry Smith if (!i) { 4645c6c1daeSBarry Smith mean += values[p]; 4655c6c1daeSBarry Smith var += values[p]*values[p]; 4665c6c1daeSBarry Smith } 4675c6c1daeSBarry Smith } 4685c6c1daeSBarry Smith } 4695c6c1daeSBarry Smith /* Draw bins */ 4705c6c1daeSBarry Smith for (i = 0; i < numBins; i++) { 4715c6c1daeSBarry Smith binLeft = xmin + binSize*i; 4725c6c1daeSBarry Smith binRight = xmin + binSize*(i+1); 47334a5a0e3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer, "Bin %2d (%6.2g - %6.2g): %.0g\n", (int)i, (double)binLeft, (double)binRight, (double)bins[i]);CHKERRQ(ierr); 4745c6c1daeSBarry Smith } 4755c6c1daeSBarry Smith ierr = PetscDrawHGSetNumberBins(hist, numBinsOld);CHKERRQ(ierr); 4765c6c1daeSBarry Smith } 4775c6c1daeSBarry Smith 4785c6c1daeSBarry Smith if (hist->calcStats) { 4795c6c1daeSBarry Smith mean /= numValues; 480a297a907SKarl Rupp if (numValues > 1) var = (var - numValues*mean*mean) / (numValues-1); 481a297a907SKarl Rupp else var = 0.0; 48257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer, "Mean: %g Var: %g\n", (double)mean, (double)var);CHKERRQ(ierr); 48334a5a0e3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer, "Total: %D\n", numValues);CHKERRQ(ierr); 4845c6c1daeSBarry Smith } 4855c6c1daeSBarry Smith PetscFunctionReturn(0); 4865c6c1daeSBarry Smith } 4875c6c1daeSBarry Smith 4885c6c1daeSBarry Smith #undef __FUNCT__ 4895c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGSetColor" 4905c6c1daeSBarry Smith /*@ 4915c6c1daeSBarry Smith PetscDrawHGSetColor - Sets the color the bars will be drawn with. 4925c6c1daeSBarry Smith 4935c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 4945c6c1daeSBarry Smith 4955c6c1daeSBarry Smith Input Parameters: 4965c6c1daeSBarry Smith + hist - The histogram context 4975c6c1daeSBarry Smith - color - one of the colors defined in petscdraw.h or PETSC_DRAW_ROTATE to make each bar a 4985c6c1daeSBarry Smith different color 4995c6c1daeSBarry Smith 5005c6c1daeSBarry Smith Level: intermediate 5015c6c1daeSBarry Smith 5025c6c1daeSBarry Smith @*/ 5035c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetColor(PetscDrawHG hist, int color) 5045c6c1daeSBarry Smith { 5055c6c1daeSBarry Smith PetscFunctionBegin; 5065c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 5075c6c1daeSBarry Smith hist->color = color; 5085c6c1daeSBarry Smith PetscFunctionReturn(0); 5095c6c1daeSBarry Smith } 5105c6c1daeSBarry Smith 5115c6c1daeSBarry Smith #undef __FUNCT__ 5125c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGSetLimits" 5135c6c1daeSBarry Smith /*@ 5145c6c1daeSBarry Smith PetscDrawHGSetLimits - Sets the axis limits for a histogram. If more 5155c6c1daeSBarry Smith points are added after this call, the limits will be adjusted to 5165c6c1daeSBarry Smith include those additional points. 5175c6c1daeSBarry Smith 5185c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 5195c6c1daeSBarry Smith 5205c6c1daeSBarry Smith Input Parameters: 5215c6c1daeSBarry Smith + hist - The histogram context 5225c6c1daeSBarry Smith - x_min,x_max,y_min,y_max - The limits 5235c6c1daeSBarry Smith 5245c6c1daeSBarry Smith Level: intermediate 5255c6c1daeSBarry Smith 5265c6c1daeSBarry Smith Concepts: histogram^setting axis 5275c6c1daeSBarry Smith @*/ 5285c6c1daeSBarry Smith PetscErrorCode PetscDrawHGSetLimits(PetscDrawHG hist, PetscReal x_min, PetscReal x_max, int y_min, int y_max) 5295c6c1daeSBarry Smith { 5305c6c1daeSBarry Smith PetscFunctionBegin; 5315c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 5325c6c1daeSBarry Smith hist->xmin = x_min; 5335c6c1daeSBarry Smith hist->xmax = x_max; 5345c6c1daeSBarry Smith hist->ymin = y_min; 5355c6c1daeSBarry Smith hist->ymax = y_max; 5365c6c1daeSBarry Smith PetscFunctionReturn(0); 5375c6c1daeSBarry Smith } 5385c6c1daeSBarry Smith 5395c6c1daeSBarry Smith #undef __FUNCT__ 5405c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGCalcStats" 5415c6c1daeSBarry Smith /*@ 5425c6c1daeSBarry Smith PetscDrawHGCalcStats - Turns on calculation of descriptive statistics 5435c6c1daeSBarry Smith 5445c6c1daeSBarry Smith Not collective 5455c6c1daeSBarry Smith 5465c6c1daeSBarry Smith Input Parameters: 5475c6c1daeSBarry Smith + hist - The histogram context 5485c6c1daeSBarry Smith - calc - Flag for calculation 5495c6c1daeSBarry Smith 5505c6c1daeSBarry Smith Level: intermediate 5515c6c1daeSBarry Smith 5525c6c1daeSBarry Smith .keywords: draw, histogram, statistics 5535c6c1daeSBarry Smith 5545c6c1daeSBarry Smith @*/ 5555c6c1daeSBarry Smith PetscErrorCode PetscDrawHGCalcStats(PetscDrawHG hist, PetscBool calc) 5565c6c1daeSBarry Smith { 5575c6c1daeSBarry Smith PetscFunctionBegin; 5585c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 5595c6c1daeSBarry Smith hist->calcStats = calc; 5605c6c1daeSBarry Smith PetscFunctionReturn(0); 5615c6c1daeSBarry Smith } 5625c6c1daeSBarry Smith 5635c6c1daeSBarry Smith #undef __FUNCT__ 5645c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGIntegerBins" 5655c6c1daeSBarry Smith /*@ 5665c6c1daeSBarry Smith PetscDrawHGIntegerBins - Turns on integer width bins 5675c6c1daeSBarry Smith 5685c6c1daeSBarry Smith Not collective 5695c6c1daeSBarry Smith 5705c6c1daeSBarry Smith Input Parameters: 5715c6c1daeSBarry Smith + hist - The histogram context 5725c6c1daeSBarry Smith - ints - Flag for integer width bins 5735c6c1daeSBarry Smith 5745c6c1daeSBarry Smith Level: intermediate 5755c6c1daeSBarry Smith 5765c6c1daeSBarry Smith .keywords: draw, histogram, statistics 5775c6c1daeSBarry Smith @*/ 5785c6c1daeSBarry Smith PetscErrorCode PetscDrawHGIntegerBins(PetscDrawHG hist, PetscBool ints) 5795c6c1daeSBarry Smith { 5805c6c1daeSBarry Smith PetscFunctionBegin; 5815c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 5825c6c1daeSBarry Smith hist->integerBins = ints; 5835c6c1daeSBarry Smith PetscFunctionReturn(0); 5845c6c1daeSBarry Smith } 5855c6c1daeSBarry Smith 5865c6c1daeSBarry Smith #undef __FUNCT__ 5875c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGGetAxis" 5885c6c1daeSBarry Smith /*@C 5895c6c1daeSBarry Smith PetscDrawHGGetAxis - Gets the axis context associated with a histogram. 5905c6c1daeSBarry Smith This is useful if one wants to change some axis property, such as 5915c6c1daeSBarry Smith labels, color, etc. The axis context should not be destroyed by the 5925c6c1daeSBarry Smith application code. 5935c6c1daeSBarry Smith 5945c6c1daeSBarry Smith Not Collective (ignored except on processor 0 of PetscDrawHG) 5955c6c1daeSBarry Smith 5965c6c1daeSBarry Smith Input Parameter: 5975c6c1daeSBarry Smith . hist - The histogram context 5985c6c1daeSBarry Smith 5995c6c1daeSBarry Smith Output Parameter: 6005c6c1daeSBarry Smith . axis - The axis context 6015c6c1daeSBarry Smith 6025c6c1daeSBarry Smith Level: intermediate 6035c6c1daeSBarry Smith 6045c6c1daeSBarry Smith @*/ 6055c6c1daeSBarry Smith PetscErrorCode PetscDrawHGGetAxis(PetscDrawHG hist, PetscDrawAxis *axis) 6065c6c1daeSBarry Smith { 6075c6c1daeSBarry Smith PetscFunctionBegin; 6085c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 6095c6c1daeSBarry Smith PetscValidPointer(axis,2); 6105c6c1daeSBarry Smith *axis = hist->axis; 6115c6c1daeSBarry Smith PetscFunctionReturn(0); 6125c6c1daeSBarry Smith } 6135c6c1daeSBarry Smith 6145c6c1daeSBarry Smith #undef __FUNCT__ 6155c6c1daeSBarry Smith #define __FUNCT__ "PetscDrawHGGetDraw" 6165c6c1daeSBarry Smith /*@C 6175c6c1daeSBarry Smith PetscDrawHGGetDraw - Gets the draw context associated with a histogram. 6185c6c1daeSBarry Smith 6195c6c1daeSBarry Smith Not Collective, PetscDraw is parallel if PetscDrawHG is parallel 6205c6c1daeSBarry Smith 6215c6c1daeSBarry Smith Input Parameter: 6225c6c1daeSBarry Smith . hist - The histogram context 6235c6c1daeSBarry Smith 6245c6c1daeSBarry Smith Output Parameter: 6255c6c1daeSBarry Smith . win - The draw context 6265c6c1daeSBarry Smith 6275c6c1daeSBarry Smith Level: intermediate 6285c6c1daeSBarry Smith 6295c6c1daeSBarry Smith @*/ 6305c6c1daeSBarry Smith PetscErrorCode PetscDrawHGGetDraw(PetscDrawHG hist, PetscDraw *win) 6315c6c1daeSBarry Smith { 6325c6c1daeSBarry Smith PetscFunctionBegin; 6335c6c1daeSBarry Smith PetscValidHeaderSpecific(hist, PETSC_DRAWHG_CLASSID,1); 6345c6c1daeSBarry Smith PetscValidPointer(win,2); 6355c6c1daeSBarry Smith *win = hist->win; 6365c6c1daeSBarry Smith PetscFunctionReturn(0); 6375c6c1daeSBarry Smith } 6385c6c1daeSBarry Smith 639