import java.awt.*;
import graphics.*;

public class hw7c extends ThreeDApplet
{
  CurvedTriangle unitCurvedTriangle;
  Tube unitTube;
  Torus unitTorus;

  CurvedTriangle[] curvedTriangles;
  Tube stem;
  Torus handle;

  Animation unfold;

  int[] color = {255, 255, 255};

  public void initialize() {
    super.initialize();

    //set background
    for (int w=0; w<W; w++)
    for (int h=0; h<H; h++)
    {
      if (h % 50 == 0)
      {
        bgbuffer[h][w][0] = 255;
        bgbuffer[h][w][1] = 255;
        bgbuffer[h][w][2] = 255;
      }
      else
      {
        bgbuffer[h][w][0] = 75;
        bgbuffer[h][w][1] = 75;
        bgbuffer[h][w][2] = 200;
      }
    }

    lights = new Light[1];

    lights[0] = new Light(0, -2, 1);
    lights[0].color[0] = lights[0].color[1] = lights[0].color[2] = 255;

    ambient[0] = ambient[1] = ambient[2] = 40;

    unfold = new Animation(2);
    unfold.looping = false;
    unfold.addKeyFrame(3, Math.PI/5);
    unfold.addKeyFrame(7, Math.PI/8);

    BezierCurve curve = new BezierCurve(1, 0.75, 0.75, 1);
    unitCurvedTriangle = new CurvedTriangle( curve.getSpline() );
    unitCurvedTriangle.calculate();
    unitCurvedTriangle.flipNormals();

    curvedTriangles = new CurvedTriangle[8];

//unitCurvedTriangle.dumpVertices();

    unitTube = new Tube();
    unitTube.calculate();

    //inner radius, outer radius
    unitTorus = new Torus(1, 2);
    unitTorus.vStart = 0.5; //draw only half the torus
    unitTorus.calculate();
    unitTorus.flipNormals();
  }

  void draw()
  {
    for (int i=0; i<curvedTriangles.length; i++)
    {
      //drawNormals(curvedTriangles[i], color, 0.25);
      drawShape(curvedTriangles[i]);
    }

    drawShape(stem);
    //drawNormals(handle, color, 0.25);
    drawShape(handle);
  }

  void animate(long time)
  {
    double theta;

    stem = (Tube)unitTube.clone();
    stem.setDiffuse(80, 0, 0);
    stem.setSpecular(1, 255, 255, 255);

    handle = (Torus)unitTorus.clone();
    handle.setDiffuse(80, 0, 0);
    handle.setSpecular(1, 255, 255, 255);

    push(); //scene
      //move everything out
      translate(0,0,-15);
      rotateZ(Math.PI); //flip camera upside-down

      push(); //umbrella
          //theta = 0.00025*time;
          //rotateY(theta);

        push(); //stem
          translate(0, 2, 0);
          scale(0.2, 4, 0.2);
          rotateX(Math.PI/2); //set upright

          transform(stem);
        pop();

        push(); //handle
          translate(0.4, -2, 0);
          //rotateY(theta);
          scale(0.2, 0.2, 0.2);
          transform(handle);
        pop();

        push(); //top part of the umbrella
          translate(0, 2.5, 0);
          theta = 0.00025*time;
          //rotateY(theta);

          //rotateZ(theta);
          //rotateY(Math.PI);

          rotateX(Math.PI/1.5);

          for (int i=0; i<curvedTriangles.length; i++)
          {
            curvedTriangles[i] = (CurvedTriangle)unitCurvedTriangle.clone();
            int red = 150 * (1 - i%2); //alternate blue and red
            int blue = 150 * (i%2);

            curvedTriangles[i].setDiffuse(red, 0, blue);
            curvedTriangles[i].setSpecular(100, 100, 100, 100);
            //curvedTriangles[i].setSpecular(5, red, 0, blue);

            push();
              rotateZ(i * Math.PI/4);

              //theta = 0.00025*time;
              //rotateY(theta);
              rotateX( unfold.get(time/1000.0) );

              scale(1, 2.8, 2.8);

              transform(curvedTriangles[i]);
//curvedTriangle.dumpVertices();
            pop();
          }

        pop(); //top part of the umbrella
      pop(); //umbrella
    pop(); //scene
  }
}