/***************************************************************************
 *   Copyright (C) 2009 by Douglas Castro   *
 *   douglas@ime.unicamp.br   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include "imprime.h"

void desenha_folhas(PSStream *ps, No *r, int ind, int prof, double xmin, double xmax, double ymin, double ymax)
{
  assert((r->fil[0] == NULL) == (r->fil[1] == NULL));
  if(r->fil[0]==NULL)
  {
    DrawCells(ps, xmin, xmax, ymin, ymax,  TRUE, TRUE, FALSE);
    return;
  }
  if(prof%2==0)
  {
    desenha_folhas(ps, r->fil[0], 2*ind + 0, prof + 1, xmin, (xmax+xmin)/2.0, ymin, ymax);
    desenha_folhas(ps, r->fil[1], 2*ind + 1, prof + 1, (xmax+xmin)/2.0, xmax, ymin, ymax);
  }
  else
  {
    desenha_folhas(ps, r->fil[0], 2*ind + 0, prof + 1, xmin, xmax, ymin, (ymax+ymin)/2.0);
    desenha_folhas(ps, r->fil[1], 2*ind + 1, prof + 1, xmin, xmax, (ymin+ymax)/2.0, ymax);
  }
}

/**
 * A chamada dessa funcao retorna a grade formada por todas as celulas folha da arvore
 */
void DoMesh(char *prefixo, No *u)
{
  PSStream *ps = pswr_new_stream(TRUE, prefixo, NULL, NULL, 328.0, 246.0);
  pswr_new_page(ps, "mesh");
  pswr_set_window
      ( ps, 
        -22.00, +22.00,  -16.50, +16.50, 
        4.00, 324.00,    3.00, 243.00,
        44, 33
      );

  pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.20,  0.0, 0.0);
  pswr_set_fill_color(ps, 1.000, 1.000, 1.000);
  //    chamar uma funcao que sai a procura de folhas
  desenha_folhas(ps, u, 1, 0, 0.0, 1.0, 0.0, 1.0);
  
  pswr_close_stream(ps);
}

void DrawCells(PSStream *ps, double xmin, double xmax, double ymin, double ymax, bool_t fill, bool_t draw, bool_t eo)
{
  /* Usable area [0 _ 12]�[0 _ 15] */
  //  Aqui e feita uma mudanca de escala para que o retangulo se ajuste ao
  //  dominio onde e feito o desenho [-22,22]X[-16.5,16,5]
  double xe = 44.0*xmin-22.0;
  double xd = 44.0*xmax-22.0;
  double ye = 33.0*ymin-16.5;
  double yd = 33.0*ymax-16.5;
  pswr_rectangle(ps, xe, xd, ye, yd, fill, draw);

}

void MakeTree(char *prefixo, No *u, int niv)
{
  PSStream *ps = pswr_new_stream(TRUE, prefixo, NULL, NULL, 328.0, 246.0);
  pswr_new_page(ps, "tree");
  pswr_set_window
      ( ps, 
        -22.00, +22.00,  -16.50, +16.50, 
        4.00, 324.00,    3.00, 243.00,
        44, 33
      );
  double x=0.0,y=12.0;
  int prof = 0;
  
  DrawNodes(ps,u,prof,niv,x,y);
  
  pswr_close_stream(ps);
}

void DrawNodes(PSStream *ps, No *u, int prof, int niv, double x, double y)
{
  assert((u->fil[0]==NULL)==(u->fil[1]==NULL));
  if(u->fil[0]==NULL){ return;}
//   if(u->/*fil[0]->*/leaf){ return;}
  if(prof==0)
  {
    fprintf(stderr,"\natencao para criterio de parada em DrawNodes -imprime.c- \n");
    fprintf(stderr,"  se eh u.fil == null tem-se arvore com folhas virtuais\n");
    fprintf(stderr,"  se eh u.fil == leaf tem-se arvore com folhas normais\n");
//     fprintf(stderr,"  em uso u.fil == leaf\n\n");
    fprintf(stderr,"  em uso u == null\n\n");
  }
  
  int qual;
  for(qual = 0;qual<=1;qual++)
  {
    double s = qual == 0 ? -1.0 : 1.0;
    DrawNodes(ps, u->fil[qual], prof+1,niv, x + s*20.0/(pow(2.0,prof+1)), y-1.0);
    if(u->fil[qual]->virtual_leaf/* && prof!=niv-2*/)
    {
      pswr_comment(ps, "Thicker blue segments with arrowheads:");
      pswr_set_pen(ps, 0.000, 0.000, 1.000,  0.20,  0.5, 0.5);
    }
    else
    {
      pswr_comment(ps, "Medium solid black segments:");
      pswr_set_pen(ps, 0.000, 0.000, 0.000,  0.20,  0.0, 0.0);
    }
    MakeLines(ps, qual, prof, x, y, FALSE);
  }
  
}

void MakeLines(PSStream *ps, int qual, int prof, double xc, double yc, bool_t arrowheads)
{
  auto void do_seg(double xa, double ya, double xb, 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, 0.2, FALSE, TRUE);
  pswr_dot(ps, xb+xc, yb+yc, 0.2, FALSE, TRUE);
  if (arrowheads)
  { pswr_arrowhead(ps, xa+xc, ya+yc, xb+xc, yb+yc, 2.0, 3.0, 0.85, TRUE, TRUE); }
  }

  if(qual==0)
  { do_seg(-20.0/(pow(2.0,prof+1)), -1.0, 0.0, 0.0); /*esq*/ }
  else
  { do_seg(0.0, 0.0, 20.0/(pow(2.0,prof+1)), -1.0);  /*dir*/ }
  
}

void escrevearquivo1d(FILE *arq, No *u, int prof, double *xlo)
{
  if(u/*->fil[0]*/==NULL){return;}
  assert((u->fil[0] == NULL) == (u->fil[1] == NULL));
  if(u->leaf /*|| u->fil[0]==NULL*/)
  {
    int tam = pow(2,prof);
    double Dx = 2.0/tam;
    double xmin = *xlo; 
    double xmax = *xlo + Dx;
    fprintf(arq,"%10.6f %10.6f\n", xmin , u->fval);
    fprintf(arq,"%10.6f %10.6f\n", xmax , u->fval);
    *xlo = xmax;
    return;
  }
  int qual;
  for(qual=0;qual<2;qual++)
    escrevearquivo1d( arq, u->fil[qual], prof+1, xlo);
}

void DoEPSTests(char *prefixo)
{
  PSStream *ps = pswr_new_stream(TRUE, prefixo, NULL, NULL, 328.0, 246.0);
  pswr_new_page(ps, NULL);
  pswr_set_window
      ( ps, 
        -22.00, +22.00,  -16.50, +16.50, 
        4.00, 324.00,    3.00, 243.00,
        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 ticksz = 0.5 + ((i % 2) == 0)*0.5 + ((i % 4) == 0)*1.0; 
  pswr_tick(ps, VER,  1.0+xc, 9.0+0.5*i+yc, ticksz, align); 
  pswr_tick(ps, VER, 11.0+xc, 9.0+0.5*i+yc, ticksz, 1-align); 
  pswr_tick(ps, HOR, 6.0 + 0.5*i+xc, 14.0+yc, ticksz, 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_dot(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_dot(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);
  int n = 17;
  double x[n], y[n];
  int i;
  for (i = 0; i < n; i++)
  { 
    double t = 2.0*3.1415926*((double)i)/((double)n);
    double r = (0.75 +0.25*cos(t));
    x[i] = xc + 3.0 + 2.0*r*cos(2*t); 
    y[i] = yc + 2.0 + 1.5*r*sin(2*t);
  }
  pswr_rectangle(ps, 7.0+xc, 9.0+xc, 1.5+yc, 3.5+yc, fill, draw);
  pswr_circle(ps, 3.0+xc, 7.0+yc, 2.0, fill, draw);
  pswr_dot(ps, 3.0+xc, 7.0+yc, 2.0, 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_polygon(ps, x, y, n, fill, draw, eo);
  pswr_triangle(ps, 7.0+xc, 10.0+yc, 11.0+xc, 12.0+yc, 9.0+xc, 14.0+yc, fill, draw);
}

void teste(No *r, int prof)
{
  if(prof == 0){fprintf(stderr,"imprime.c - teste -\n");}
  if(r->fil[0] == NULL) {return;}
  int qual;
  for(qual = 0;qual<2;qual++)
    teste(r->fil[qual],prof+1);
  if(r->deletable)
    fprintf(stderr, "%*st\n", 2*prof, "-");
}