xref: /petsc/src/sys/classes/draw/tests/ex3.c (revision 9566063d113dddea24716c546802770db7481bc0)
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 
24c4762a1bSJed Brown   toplabel = "Top Label"; xlabel = "X-axis Label"; ylabel = "Y-axis Label"; legend = "Legend";
25c4762a1bSJed Brown 
26*9566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc,&argv,NULL,help));
27*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-x",&x,NULL));
28*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-y",&y,NULL));
29*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-width",&width,NULL));
30*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-height",&height,NULL));
31*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
32*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-nports",&nports,&useports));
33*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-nolegend",&flg));
34c4762a1bSJed Brown   if (flg) legend = NULL;
35*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-notoplabel",&flg));
36c4762a1bSJed Brown   if (flg) toplabel = NULL;
37*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-noxlabel",&flg));
38c4762a1bSJed Brown   if (flg) xlabel = NULL;
39*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-noylabel",&flg));
40c4762a1bSJed Brown   if (flg) ylabel = NULL;
41*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-nolabels",&flg));
42c4762a1bSJed Brown   if (flg) {toplabel = NULL; xlabel = NULL; ylabel = NULL;}
43c4762a1bSJed Brown 
44*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw));
45*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawSetFromOptions(draw));
46c4762a1bSJed Brown   if (useports) {
47*9566063dSJacob Faibussowitsch     PetscCall(PetscDrawViewPortsCreate(draw,nports,&ports));
48*9566063dSJacob Faibussowitsch     PetscCall(PetscDrawViewPortsSet(ports,0));
49c4762a1bSJed Brown   }
50*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGCreate(draw,1,&lg));
51*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetUseMarkers(lg,PETSC_TRUE));
52*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGGetAxis(lg,&axis));
53*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE));
54*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel));
55*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetLegend(lg,&legend));
56*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetFromOptions(lg));
57c4762a1bSJed Brown 
58c4762a1bSJed Brown   for (i=0; i<=n; i++) {
59c4762a1bSJed Brown     xd   = (PetscReal)(i - 5); yd = xd*xd;
60*9566063dSJacob Faibussowitsch     PetscCall(PetscDrawLGAddPoint(lg,&xd,&yd));
61c4762a1bSJed Brown   }
62*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGDraw(lg));
63*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSave(lg));
64c4762a1bSJed Brown 
65*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawViewPortsDestroy(ports));
66*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGDestroy(&lg));
67*9566063dSJacob Faibussowitsch   PetscCall(PetscDrawDestroy(&draw));
68*9566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
69b122ec5aSJacob Faibussowitsch   return 0;
70c4762a1bSJed Brown }
71c4762a1bSJed Brown 
72c4762a1bSJed Brown /*TEST
73c4762a1bSJed Brown 
74c4762a1bSJed Brown    build:
75c4762a1bSJed Brown      requires: x
76c4762a1bSJed Brown 
77c4762a1bSJed Brown    test:
78c4762a1bSJed Brown      output_file: output/ex1_1.out
79c4762a1bSJed Brown 
80c4762a1bSJed Brown TEST*/
81