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