/*  plot_laphi.c  plot Lagrange Phi functions */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <GL/glut.h>
#include "laphi.h"

static GLfloat points[130][3];
static GLfloat colors[11][3] = {1.0, 0.0, 0.0,  0.0, 1.0, 0.0,  0.0, 0.0, 1.0,
                                1.0, 1.0, 0.0,  1.0, 0.0, 1.0,  0.0, 1.0, 1.0,
                                1.0, 0.5, 0.5,  0.5, 0.5, 1.0,  0.5, 1.0, 0.5,
                                0.7, 0.7, 0.7,  0.5, 0.5, 0.5};
static int n_points = 0;
static int n = 0;
static int width=800;
static int height=600;
static double xg[11] = {-0.5, -0.4, -0.3, -0.2, -0.1, 0.0, 0.1, 0.2, 0.3, 0.4, 0.5};
static int np = 11;
static double xmin = -0.5;
static double xmax = 0.5;
static double ymin = 99.0;
static double ymax = -99.0;
static float offset = 0.0;
static int ifphi = 1;

void compute_phi(int i)
{
  double x, y, h;
  double z = 0.0;
  int j;

  n = 60;
  n_points = n;
  h = (xmax-xmin)/(double)(n-1);
  x = xmin;
  for(j=0; j<n; j++)
  {
    y = phi(x, i, np-1, xg);
    if(i==0 && j==0) {ymin=y; ymax=y;}
    if(y>ymax) ymax=y;
    if(y<ymin) ymin=y;
    points[j][0] = (float)3.0*x-2.0*xmin+offset;
    points[j][1] = (float)y/1.5+4.0;
    points[j][2] = (float)z;
    x = x+h;
  }
  printf("phi i=%d, ymin=%g, ymax=%g \n", i, ymin, ymax);
} /* end compute_phi11 */

void compute_phip(int i)
{
  double x, y, h;
  double z = 0.0;
  int j;

  n = 50;
  n_points = n;
  h = (xmax-xmin)/(double)(n-1);
  x = xmin;
  for(j=0; j<n; j++)
  {
    y = phip(x, i, np-1, xg);
    if(i==0 && j==0) {ymin=y; ymax=y;}
    if(y>ymax) ymax=y;
    if(y<ymin) ymin=y;
    points[j][0] = (float)3.0*x-2.0*xmin+offset;
    points[j][1] = (float)(y/16.0+4.0);
    points[j][2] = (float)z;
    x = x+h;
  }
  printf("phip i=%d, ymin=%g, ymax=%g \n", i, ymin, ymax);
} /* end computer_phip11 */


void draw_points(int i)
{
  int j;

  glColor3fv(colors[i]);
  for(j=0; j<n_points-1; j++)
  {
    glBegin(GL_LINES);
      glVertex3fv(points[j]);
      glVertex3fv(points[j+1]);
    glEnd();
  }
} /* end draw_points */

static void drawText(float x, float y, char * msg)
{
  int   i, len;

  glRasterPos2f(x, y);
  len = strlen(msg);
  for (i=0; i<len; i++)
    glutBitmapCharacter(GLUT_BITMAP_HELVETICA_18, msg[i]);
}

void display(void)
{
  char text1[]="laphi(x,i,np-1,xg[ ]) np=11, xmin=-0.5, xmax=0.5";
  char text1p[]="laphip(x,i,np-1,xg[ ]) np=11, xmin=-0.5, xmax=0.5";
  char text1y[]="ymin=dd.dd  ymax=dd.dd ";
  char text2[]="i =   ";
  char text3[]="0";
  char text4[]="1";
  char text4m[]="-1";
  char text44[]="4";
  char text44m[]="-4";
  char num[5];
  int i, j;
  
  /* clear window */
  glClear(GL_COLOR_BUFFER_BIT); 
  glLoadIdentity ();
  glColor3f(0.0, 0.0, 0.0);
  
  /* draw title text */
  glColor3f(0.0, 0.0, 0.0);

  for(i=0; i<10; i++)
  {
    glColor3fv(colors[i]);
    text2[4] = (char)(48+i);
    text2[5] = ' ';
    drawText(-0.9, -0.20+i*0.6, text2);
  }
  glColor3fv(colors[10]);
  text2[4] = '1';
  text2[5] = '0';
  drawText(-0.9, -0.20+10*0.6, text2);
  glColor3f(0.0, 0.0, 0.0);
  
  /* Draw the points */
  if(ifphi==1)
  {
    offset = 0.25;
    for(i=0; i<np; i++)
    {
      compute_phi(i);
      draw_points(i);
    }
    glColor3f(0.0, 0.0, 0.0);
    drawText(-0.95, -0.90, text1);
    drawText(-0.33 ,  3.9 , text3);
    drawText(-0.33 ,  4.6 , text4);
    drawText(-0.33 ,  3.2 , text4m);
  }
  else
  {
    offset = 0.25;
    for(i=0; i<np; i++)
    {
      compute_phip(i);
      draw_points(i);
    }
    glColor3f(0.0, 0.0, 0.0);
    drawText(-0.95, -0.90, text1p);
    drawText(-0.33 ,  3.9 , text3);
    drawText(-0.33 ,  7.2 , text44);
    drawText(-0.33 ,  0.6 , text44m);
  }
  glColor3f(0.0, 0.0, 0.0);
  sprintf(num, "%5.2f", ymin);
  for(i=0; i<5; i++) text1y[i+5]=num[i];
  sprintf(num, "%5.2f", ymax);
  for(i=0; i<5; i++) text1y[i+17]=num[i];
  drawText(2.0, -0.90, text1y);

  glFlush(); 
}

/* This routine handels mouse events */
static void mouse(int button, int state, int x, int y)
{
  float wx, wy;
  if (button == GLUT_LEFT_BUTTON && state == GLUT_DOWN)
  {
    ifphi = 1-ifphi;
    ymin = 99.0;
    ymax = -99.0;
  }
  if (button == GLUT_RIGHT_BUTTON && state == GLUT_DOWN)
  {
    ifphi = 1-ifphi;
    ymin = 99.0;
    ymax = -99.0;
  }
  glutPostRedisplay();
}

/* This routine handles window resizes */
void reshape(int w, int h)
{
  width = w;
  height = h;
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
  glOrtho(-1.0, 3.0, -1.0, 10.0, -1.0, 1.0);
  glMatrixMode(GL_MODELVIEW);
  glLoadIdentity();
  glViewport(0, 0, w, h);
}

void init()
{
  /* set clear color to white */
  glClearColor (1.0, 1.0, 1.0, 0.0);
  /* set fill  color to black */
  glColor3f(0.0, 0.0, 0.0);
  /* Set the transformations */
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
  glOrtho(-1.0, 3.0, -1.0, 10.0, -1.0, 1.0);
  glMatrixMode(GL_MODELVIEW);
  glLoadIdentity();
  glViewport(0, 0, width, height);

}

int main(int argc, char* argv[])
{
  glutInit(&argc,argv);
  glutInitDisplayMode (GLUT_SINGLE | GLUT_RGB);  
  glutInitWindowSize(width, height);
  glutInitWindowPosition(0,0); 
  
  glutCreateWindow(argv[0]); 
  glutDisplayFunc(display);
  glutMouseFunc(mouse);
  init();
  glutMainLoop();
  return 0;
}