xref: /petsc/src/sys/classes/draw/tests/ex9.c (revision 40badf4fbc550ac1f60bd080eaff6de6d55b946d)
1 
2 static char help[] = "Makes a simple histogram.\n";
3 
4 #include <petscsys.h>
5 #include <petscdraw.h>
6 
7 int main(int argc,char **argv)
8 {
9   PetscDraw      draw;
10   PetscDrawHG    hist;
11   PetscDrawAxis  axis;
12   PetscErrorCode ierr;
13   int            n     = 20,i,x = 0,y = 0,width = 400,height = 300,bins = 8;
14   PetscInt       w     = 400,h = 300,nn = 20,b = 8,c = PETSC_DRAW_GREEN;
15   int            color = PETSC_DRAW_GREEN;
16   const char     *xlabel,*ylabel,*toplabel;
17   PetscReal      xd;
18   PetscBool      flg;
19 
20   xlabel = "X-axis Label"; toplabel = "Top Label"; ylabel = "Y-axis Label";
21 
22   ierr  = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr;
23   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-width",&w,NULL));
24   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-height",&h,NULL));
25   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&nn,NULL));
26   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-bins",&b,NULL));
27   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-color",&c,NULL));
28   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-nolabels",&flg));
29   width = (int) w; height = (int)h; n = (int)nn; bins = (int) b; color = (int) c;
30   if (flg) { xlabel = NULL; ylabel = NULL; toplabel = NULL; }
31 
32   CHKERRQ(PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw));
33   CHKERRQ(PetscDrawSetFromOptions(draw));
34   CHKERRQ(PetscDrawHGCreate(draw,bins,&hist));
35   CHKERRQ(PetscDrawHGSetColor(hist,color));
36   CHKERRQ(PetscDrawHGGetAxis(hist,&axis));
37   CHKERRQ(PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE));
38   CHKERRQ(PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel));
39   /*CHKERRQ(PetscDrawHGSetFromOptions(hist));*/
40 
41   for (i=0; i<n; i++) {
42     xd   = (PetscReal)(i - 5);
43     CHKERRQ(PetscDrawHGAddValue(hist,xd*xd));
44   }
45   CHKERRQ(PetscDrawHGDraw(hist));
46   CHKERRQ(PetscDrawHGSave(hist));
47 
48   CHKERRQ(PetscDrawHGDestroy(&hist));
49   CHKERRQ(PetscDrawDestroy(&draw));
50   ierr = PetscFinalize();
51   return ierr;
52 }
53 
54 /*TEST
55 
56    build:
57      requires: x
58 
59    test:
60      output_file: output/ex1_1.out
61 
62 TEST*/
63