xref: /petsc/src/sys/classes/draw/tests/ex3.c (revision 327415f76d85372a4417cf1aaa14db707d4d6c04)
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*327415f7SBarry Smith   PetscFunctionBeginUser;
279566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc,&argv,NULL,help));
289566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-x",&x,NULL));
299566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-y",&y,NULL));
309566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-width",&width,NULL));
319566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-height",&height,NULL));
329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
339566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,NULL,"-nports",&nports,&useports));
349566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-nolegend",&flg));
35c4762a1bSJed Brown   if (flg) legend = NULL;
369566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-notoplabel",&flg));
37c4762a1bSJed Brown   if (flg) toplabel = NULL;
389566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-noxlabel",&flg));
39c4762a1bSJed Brown   if (flg) xlabel = NULL;
409566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-noylabel",&flg));
41c4762a1bSJed Brown   if (flg) ylabel = NULL;
429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(NULL,NULL,"-nolabels",&flg));
43c4762a1bSJed Brown   if (flg) {toplabel = NULL; xlabel = NULL; ylabel = NULL;}
44c4762a1bSJed Brown 
459566063dSJacob Faibussowitsch   PetscCall(PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw));
469566063dSJacob Faibussowitsch   PetscCall(PetscDrawSetFromOptions(draw));
47c4762a1bSJed Brown   if (useports) {
489566063dSJacob Faibussowitsch     PetscCall(PetscDrawViewPortsCreate(draw,nports,&ports));
499566063dSJacob Faibussowitsch     PetscCall(PetscDrawViewPortsSet(ports,0));
50c4762a1bSJed Brown   }
519566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGCreate(draw,1,&lg));
529566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetUseMarkers(lg,PETSC_TRUE));
539566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGGetAxis(lg,&axis));
549566063dSJacob Faibussowitsch   PetscCall(PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE));
559566063dSJacob Faibussowitsch   PetscCall(PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel));
569566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetLegend(lg,&legend));
579566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSetFromOptions(lg));
58c4762a1bSJed Brown 
59c4762a1bSJed Brown   for (i=0; i<=n; i++) {
60c4762a1bSJed Brown     xd   = (PetscReal)(i - 5); yd = xd*xd;
619566063dSJacob Faibussowitsch     PetscCall(PetscDrawLGAddPoint(lg,&xd,&yd));
62c4762a1bSJed Brown   }
639566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGDraw(lg));
649566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGSave(lg));
65c4762a1bSJed Brown 
669566063dSJacob Faibussowitsch   PetscCall(PetscDrawViewPortsDestroy(ports));
679566063dSJacob Faibussowitsch   PetscCall(PetscDrawLGDestroy(&lg));
689566063dSJacob Faibussowitsch   PetscCall(PetscDrawDestroy(&draw));
699566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
70b122ec5aSJacob Faibussowitsch   return 0;
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