xref: /petsc/src/sys/classes/draw/tests/ex3.c (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Plots a simple line graph.\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #if defined(PETSC_APPLE_FRAMEWORK)
5c4762a1bSJed Brown #import <PETSc/petscsys.h>
6c4762a1bSJed Brown #import <PETSc/petscdraw.h>
7c4762a1bSJed Brown #else
8c4762a1bSJed Brown 
9c4762a1bSJed Brown #include <petscsys.h>
10c4762a1bSJed Brown #include <petscdraw.h>
11c4762a1bSJed Brown #endif
12c4762a1bSJed Brown 
13c4762a1bSJed Brown int main(int argc,char **argv)
14c4762a1bSJed Brown {
15c4762a1bSJed Brown   PetscDraw          draw;
16c4762a1bSJed Brown   PetscDrawLG        lg;
17c4762a1bSJed Brown   PetscDrawAxis      axis;
18c4762a1bSJed Brown   PetscInt           n = 15,i,x = 0,y = 0,width = 400,height = 300,nports = 1;
19c4762a1bSJed Brown   PetscBool          useports,flg;
20c4762a1bSJed Brown   const char         *xlabel,*ylabel,*toplabel,*legend;
21c4762a1bSJed Brown   PetscReal          xd,yd;
22c4762a1bSJed Brown   PetscDrawViewPorts *ports = NULL;
23c4762a1bSJed Brown   PetscErrorCode     ierr;
24c4762a1bSJed Brown 
25c4762a1bSJed Brown   toplabel = "Top Label"; xlabel = "X-axis Label"; ylabel = "Y-axis Label"; legend = "Legend";
26c4762a1bSJed Brown 
27c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr;
28*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-x",&x,NULL));
29*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-y",&y,NULL));
30*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-width",&width,NULL));
31*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-height",&height,NULL));
32*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
33*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-nports",&nports,&useports));
34*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-nolegend",&flg));
35c4762a1bSJed Brown   if (flg) legend = NULL;
36*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-notoplabel",&flg));
37c4762a1bSJed Brown   if (flg) toplabel = NULL;
38*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-noxlabel",&flg));
39c4762a1bSJed Brown   if (flg) xlabel = NULL;
40*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-noylabel",&flg));
41c4762a1bSJed Brown   if (flg) ylabel = NULL;
42*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-nolabels",&flg));
43c4762a1bSJed Brown   if (flg) {toplabel = NULL; xlabel = NULL; ylabel = NULL;}
44c4762a1bSJed Brown 
45*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw));
46*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawSetFromOptions(draw));
47c4762a1bSJed Brown   if (useports) {
48*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawViewPortsCreate(draw,nports,&ports));
49*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawViewPortsSet(ports,0));
50c4762a1bSJed Brown   }
51*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGCreate(draw,1,&lg));
52*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetUseMarkers(lg,PETSC_TRUE));
53*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGGetAxis(lg,&axis));
54*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE));
55*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel));
56*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetLegend(lg,&legend));
57*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetFromOptions(lg));
58c4762a1bSJed Brown 
59c4762a1bSJed Brown   for (i=0; i<=n; i++) {
60c4762a1bSJed Brown     xd   = (PetscReal)(i - 5); yd = xd*xd;
61*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawLGAddPoint(lg,&xd,&yd));
62c4762a1bSJed Brown   }
63*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGDraw(lg));
64*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSave(lg));
65c4762a1bSJed Brown 
66*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawViewPortsDestroy(ports));
67*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGDestroy(&lg));
68*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawDestroy(&draw));
69c4762a1bSJed Brown   ierr = PetscFinalize();
70c4762a1bSJed Brown   return ierr;
71c4762a1bSJed Brown }
72c4762a1bSJed Brown 
73c4762a1bSJed Brown /*TEST
74c4762a1bSJed Brown 
75c4762a1bSJed Brown    build:
76c4762a1bSJed Brown      requires: x
77c4762a1bSJed Brown 
78c4762a1bSJed Brown    test:
79c4762a1bSJed Brown      output_file: output/ex1_1.out
80c4762a1bSJed Brown 
81c4762a1bSJed Brown TEST*/
82