xref: /petsc/src/sys/classes/draw/tests/ex3.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
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*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&argv,NULL,help));
275f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-x",&x,NULL));
285f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-y",&y,NULL));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-width",&width,NULL));
305f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-height",&height,NULL));
315f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
325f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-nports",&nports,&useports));
335f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-nolegend",&flg));
34c4762a1bSJed Brown   if (flg) legend = NULL;
355f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-notoplabel",&flg));
36c4762a1bSJed Brown   if (flg) toplabel = NULL;
375f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-noxlabel",&flg));
38c4762a1bSJed Brown   if (flg) xlabel = NULL;
395f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-noylabel",&flg));
40c4762a1bSJed Brown   if (flg) ylabel = NULL;
415f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsHasName(NULL,NULL,"-nolabels",&flg));
42c4762a1bSJed Brown   if (flg) {toplabel = NULL; xlabel = NULL; ylabel = NULL;}
43c4762a1bSJed Brown 
445f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawCreate(PETSC_COMM_WORLD,0,"Title",x,y,width,height,&draw));
455f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawSetFromOptions(draw));
46c4762a1bSJed Brown   if (useports) {
475f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawViewPortsCreate(draw,nports,&ports));
485f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawViewPortsSet(ports,0));
49c4762a1bSJed Brown   }
505f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGCreate(draw,1,&lg));
515f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetUseMarkers(lg,PETSC_TRUE));
525f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGGetAxis(lg,&axis));
535f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawAxisSetColors(axis,PETSC_DRAW_BLACK,PETSC_DRAW_RED,PETSC_DRAW_BLUE));
545f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawAxisSetLabels(axis,toplabel,xlabel,ylabel));
555f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetLegend(lg,&legend));
565f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSetFromOptions(lg));
57c4762a1bSJed Brown 
58c4762a1bSJed Brown   for (i=0; i<=n; i++) {
59c4762a1bSJed Brown     xd   = (PetscReal)(i - 5); yd = xd*xd;
605f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDrawLGAddPoint(lg,&xd,&yd));
61c4762a1bSJed Brown   }
625f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGDraw(lg));
635f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGSave(lg));
64c4762a1bSJed Brown 
655f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawViewPortsDestroy(ports));
665f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawLGDestroy(&lg));
675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDrawDestroy(&draw));
68*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
69*b122ec5aSJacob 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