#define PROG_NAME "testplot"
#define PROG_DESC "test of {pswr.h} (minus automatic layout)"
#define PROG_VERS "1.0"

/* Last edited on 2009-10-31 23:08:02 by stolfi */

#define testplot_COPYRIGHT \
  "Copyright © 2003  by the State University of Campinas (UNICAMP)"

/* Created by J. Stolfi, UNICAMP sometime before 2003-09-30. */


#include <pswr.h>
#include <pswr_iso.h>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>

#define OUT_PREFIX PROG_NAME "-out-"
  /* Prefix for output file names. */

int main (int argc, char **argv);
void DoEPSTests(void);
void DoPSTests(void);
void DrawThings(PSStream *ps);
void DrawTexts(PSStream *ps, double xc, double yc);
void DrawLines(PSStream *ps, double xc, double yc, bool_t arrowheads);
void DrawFigures(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo);
void DrawPolygon(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo);
void DrawBezierPolygon(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo);

int main (int argc, char **argv)
  { DoEPSTests();
    DoPSTests();
    return 0;
  }
  
void DoEPSTests(void)
  {
    PSStream *ps = pswr_new_stream(OUT_PREFIX, NULL, TRUE, "fig", NULL, 328.0, 246.0);
    pswr_new_canvas(ps, NULL);
    pswr_set_window
      ( ps, 
        -22.00, +22.00,  -16.50, +16.50, 
          4.00, 324.00,    3.00, 243.00
      );
    pswr_set_grid(ps, 44, 33);
    DrawThings(ps);
    pswr_close_stream(ps);
  }
      
void DoPSTests(void)
  { 
    double hSize, vSize; /* Paper dimensions in pt. */
    pswr_get_paper_dimensions("letter", &hSize, &vSize);
    
    PSStream *ps = pswr_new_stream(OUT_PREFIX, NULL, FALSE, "doc", "letter", 0, 0);
    pswr_new_canvas(ps, "one");
      pswr_set_window
        ( ps, 
          -22.00, +22.00,  -16.50, +16.50, 
          144.00, 464.00,  144.00, 384.00
        );
      pswr_set_grid(ps, 44, 33);
      DrawThings(ps);
      pswr_set_pen(ps, 0.000, 0.000, 1.000,  0.15,  0.0,0.0);
      pswr_add_caption(ps, "First Line Left Aligned\nSecond Line Left Aligned", 0.0);
      pswr_add_caption(ps, "Third Line Right Aligned", 1.0);
      pswr_add_caption(ps, "Fourth Line Centered", 0.5);

      pswr_set_window
        ( ps, 
          -22.00, +22.00,  -16.50, +16.50,
          144.00, 304.00,  432.00, 552.00
        );
      pswr_set_grid(ps, 44, 33);
      DrawThings(ps);
      pswr_add_caption(ps, "This Figure No Caption", 0.5);
    pswr_new_canvas(ps, "two");
      pswr_set_window
        ( ps, 
          -22.00, +22.00,  -16.50, +16.50, 
          144.00, 304.00,  432.00, 552.00
        );
      pswr_set_grid(ps, 44, 33);
      DrawThings(ps);
    pswr_close_stream(ps);
  }

void DrawThings(PSStream *ps)
  { pswr_comment(ps, "Thick solid red frame:");
    pswr_set_pen(ps, 1.000, 0.000, 0.000,  0.40,  0.0, 0.0);
    pswr_frame(ps);

    pswr_comment(ps, "Thin dashed light yellow gridlines:");
    pswr_set_pen(ps, 1.000, 1.000, 0.500,  0.10,  2.0, 1.0);
    pswr_grid_lines(ps);

    pswr_comment(ps, "Medium solid black coordinate lines:");
    pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.20,  0.0, 0.0);
    pswr_coord_line(ps, HOR, 0.17);
    pswr_coord_line(ps, VER, 3.14);

    pswr_comment(ps, "Text in various positions:");
    DrawTexts(ps, -20.0, -15.5);

    pswr_comment(ps, "Medium solid black segments:");
    pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.20,  0.0, 0.0);
    DrawLines(ps,  -6.0, -15.5, FALSE);

    pswr_comment(ps, "Thicker blue segments with arrowheads:");
    pswr_set_pen(ps, 0.000, 0.000, 1.000,  0.40,  0.0, 0.0);
    DrawLines(ps,  +8.0, -15.5, TRUE);

    pswr_comment(ps, "Thin solid black figures, yellow filled:");
    pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.10,  0.0, 0.0);
    pswr_set_fill_color(ps, 1.000, 1.000, 0.000);
    pswr_grid_cell(ps,  3, 2,  TRUE, TRUE);
    DrawFigures(ps, -20.0,  +0.5,  TRUE, TRUE, FALSE);

    pswr_comment(ps, "Medium solid red figures, unfilled:");
    pswr_set_pen(ps, 1.000, 0.000, 0.000,  0.20,  0.0, 0.0);
    pswr_set_fill_color(ps, -1.00, -1.00, -1.00);
    pswr_grid_cell(ps,  5, 2,  TRUE, TRUE);
    DrawFigures(ps,  -6.0,  +0.5,  TRUE, TRUE, FALSE);

    pswr_comment(ps, "Unstroked figures, pink e-o-filled:");
    pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.50,  0.0, 0.0);
    pswr_set_fill_color(ps, 1.000, 0.800, 0.700);
    pswr_grid_cell(ps,  7, 2,  TRUE, FALSE);
    DrawFigures(ps,  +8.0,  +0.5,  TRUE, FALSE, TRUE);
  }

void DrawTexts(PSStream *ps, double xc, double yc)
  {
    /* Usable area [0 _ 12]×[0 _ 15] */
    pswr_rectangle(ps, 0.25+xc, 11.75+xc, 0.25+yc, 14.75+yc, FALSE, TRUE);
    
    auto void do_lab(char *text, double xd, double yd, double xalign, double yalign);
    void do_lab(char *text, double xd, double yd, double xalign, double yalign)
      { pswr_set_fill_color(ps, 0.000, 0.700, 1.000);
        pswr_dot(ps, xd+xc, yd+yc, 0.3,  TRUE, FALSE);
        pswr_label(ps, text,  xd+xc, yd+yc,  xalign, yalign);
      }
    
    pswr_set_pen(ps, 1.000, 0.000, 0.000, 0.50,  0.0, 0.0);

    pswr_set_label_font(ps, "Courier", 10.0);
    do_lab("red C10",   +1.00,  +3.00,  0.0, 0.0);

    pswr_set_label_font(ps, "Times-Roman", 12.0);
    do_lab("red TR12",  +6.00,  +6.00,  0.5, 0.0);

    pswr_set_pen(ps, 0.000, 0.000, 1.000, 0.50,  0.0, 0.0);

    pswr_set_label_font(ps, "Helvetica", 8.0);
    do_lab("blu H8",   +11.00,  +9.00,  1.0, 0.5);

    pswr_set_label_font(ps, "Courier", 12.0);
    do_lab("blu C12",   +1.00, +12.00, 0.0, 0.5);
  }

void DrawLines(PSStream *ps, double xc, double yc, bool_t arrowheads)
  {
    /* Usable area [0 _ 12]×[0 _ 15] */
    pswr_rectangle(ps, 0.25+xc, 11.75+xc, 0.25+yc, 14.75+yc, FALSE, TRUE);
    
    auto void do_seg(double xa, double ya, double b, double yb);
    void do_seg(double xa, double ya, double xb, double yb)
      { pswr_segment(ps, xa+xc, ya+yc, xb+xc, yb+yc);
        pswr_dot(ps, xa+xc, ya+yc, 1.0, FALSE, TRUE);
        pswr_dot(ps, xb+xc, yb+yc, 1.0, FALSE, TRUE);
        if (arrowheads)
          { pswr_arrowhead(ps, xa+xc, ya+yc, xb+xc, yb+yc, 2.0, 3.0, 0.85, TRUE, TRUE); }
      }
    
    do_seg(1.0, 1.0, 11.0, 3.0);
    do_seg(1.0, 3.0, 11.0, 1.0);
    
    pswr_segment(ps,  1.0+xc,  5.0+yc,  1.0+xc, 13.0+yc);
    pswr_segment(ps, 11.0+xc,  5.0+yc, 11.0+xc, 13.0+yc);
    pswr_segment(ps,  2.0+xc, 14.0+yc, 10.0+xc, 14.0+yc);
    int i;
    for (i = -7; i <= +7; i++)
      { double align = ((i % 2) == 0)*0.5 - ((i % 4) == 0)*0.25;
        double ticsz = 0.5 + ((i % 2) == 0)*0.5 + ((i % 4) == 0)*1.0; 
        pswr_tic(ps, VER,  1.0+xc, 9.0+0.5*i+yc, ticsz, align); 
        pswr_tic(ps, VER, 11.0+xc, 9.0+0.5*i+yc, ticsz, 1-align); 
        pswr_tic(ps, HOR, 6.0 + 0.5*i+xc, 14.0+yc, ticsz, 1-align); 
      }
    
    pswr_curve(ps,
       4.0+xc,  4.0+yc,  
      11.0+xc, 14.0+yc, 
       1.0+xc, 14.0+yc, 
       8.0+xc,  4.0+yc
    );
    pswr_square(ps,  4.0+xc,  4.0+yc, 1.0, FALSE, TRUE);  
    pswr_dot   (ps, 11.0+xc, 14.0+yc, 0.5, FALSE, TRUE);
    pswr_dot   (ps,  1.0+xc, 14.0+yc, 0.5, FALSE, TRUE); 
    pswr_square(ps,  8.0+xc,  4.0+yc, 1.0, FALSE, TRUE);
  }
  
void DrawFigures(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo)
  {
    /* Usable area [0 _ 12]×[0 _ 15] */
    pswr_rectangle(ps, 0.25+xc, 11.75+xc, 0.25+yc, 14.75+yc, FALSE, TRUE);
    
    pswr_rectangle(ps, 1.5+xc, 4.5+xc, 10.0+yc, 11.5+yc, fill, draw);
    pswr_circle(ps, 3.0+xc, 8.0+yc, 1.75, fill, draw);
    pswr_dot(ps, 3.0+xc, 8.0+yc, 1.5, fill, draw);
    pswr_lune(ps, 9.0+xc, 7.0+yc, 1.5, 45.0, fill, draw);
    pswr_slice(ps, 3.0+xc, 12.0+yc, 2.0, 30.0, 135.0, fill, draw);
    pswr_quadrilateral(ps, 5.0+xc, 8.0+yc, 9.0+xc, 10.0+yc, 7.0+xc, 12.0+yc, 6.0+xc, 9.0+yc, fill, draw);
    pswr_triangle(ps, 7.0+xc, 10.0+yc, 11.0+xc, 12.0+yc, 9.0+xc, 14.0+yc, fill, draw);
    
    /* Ordinary polygon {eo} filled: */
    DrawPolygon(ps, 3.0+xc, 3.0+yc, fill, draw, eo);
    
    /* Bézier polygon {eo} filled: */
    DrawBezierPolygon(ps, 7.5+xc, 3.0+yc, fill, draw, eo);
  }

void DrawPolygon(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo)
  {
    int n = 7;
    double x[n], y[n];
    int i;
    for (i = 0; i < n; i++)
      { double t = 2.0*M_PI*((double)i)/((double)n);
        double r = 1.000;
        x[i] = xc + 2.0*r*cos(2*t); 
        y[i] = yc + 2.5*r*sin(2*t);
      }
    pswr_polygon(ps, x, y, n, fill, draw, eo);
  }

void DrawBezierPolygon(PSStream *ps, double xc, double yc, bool_t fill, bool_t draw, bool_t eo)
  {
    /* Prototypical S-shaped arc: */
    double xs[4], ys[4];
    xs[0] = +1.200; ys[0] = 00.000;
    xs[1] = +0.400; ys[1] = -0.800;
    xs[2] = -0.400; ys[2] = +0.800;
    xs[3] = -1.200; ys[3] = 00.000;
    
    /* Prototypical convex arc: */
    double xv[4], yv[4];
    xv[0] = +1.200; yv[0] = 00.000;
    xv[1] = +0.400; yv[1] = +0.800;
    xv[2] = -0.400; yv[2] = +0.800;
    xv[3] = -1.200; yv[3] = 00.000;
    
    double mag = 1.500;   /* Magnification factor. */
    double shift = 0.600; /* Side shift from center. */
    
    int n = 5;
    int np = 4*n;
    double x[np], y[np];
    int i, j;
    for (i = 0; i < n; i++)
      { /* Rotation angle: */
        double t = 2.0*M_PI*((double)i)/((double)n);
        double ct = cos(2*t), st = sin(2*t);
        for (j = 0; j < 4; j++)
          { /* Pick Bezier point from chosen arc: */
            double xb, yb;
            if (i != 99)
              { xb = xs[j]; yb = ys[j]; }
            else
              { xb = xv[j]; yb = yv[j]; }
            /* Displace in {y}: */
            yb += shift;
            /* Rotate and add center: */
            x[4*i + j] = xc + mag*(+ ct*xb - st*yb);
            y[4*i + j] = yc + mag*(+ st*xb + ct*yb);
          }
      }
    
    /* Draw the control polygon: */
    /* pswr_polygon(ps, x, y, np, FALSE, draw, FALSE); */
    
    /* Draw the Bézier polygon: */
    pswr_bezier_polygon(ps, x, y, n, fill, draw, eo);
  }