#include <stdio.h>
#include <iostream.h>
#include <math.h>
#include "module.h"

// prototype for tiff write routine
int tiff_raster2file(char filename[], uint32 *raster, uint32 nrow, uint32 ncol);

// main routine demonstrating a hierarchical modeling system
main(int argc, char *argv[]) {
  ColorVector_t spec = ColorVector(0.1, 0.1, 0.1);
  ColorVector_t Grey = ColorVector(0.5, 0.5, 0.5);
  ColorVector_t Purple = ColorVector(0.8, 0.1, 0.8);

  // define other variables;
  Image_t im;
  long nPixels;

  // set up the 3D viewing geometry
  ViewStructure3D vs;
  vs.vrp = Vector(0, 0, -40);
  vs.vpn = Vector(0, 0, 1);
  vs.vup = Vector(0, 1, 0);
  vs.cop = Vector(0, 0, -6);
  vs.umax = vs.vmax = 2;
  vs.umin = vs.vmin = -2;
  vs.F = 1;
  vs.B = 50;
  vs.screenX = vs.screenY = 500;

  // get an angle for the faces from the command line
  double angle1 = M_PI / 5.0;
  if(argc > 1) {
    sscanf(argv[1], "%lf", &angle1);
    angle1 = angle1 / 180.0 * M_PI;
  }

  double angle2 = M_PI / 5.0;
  if(argc > 2) {
    sscanf(argv[2], "%lf", &angle2);
    angle2 = angle2 / 180.0 * M_PI;
  }

  // allocate the image
  im.image = new uint32[(long)vs.screenX * (long)vs.screenY];
  im.depth = new float[(long)vs.screenX * (long)vs.screenY];
  im.rows = (long)vs.screenY;
  im.cols = (long)vs.screenX;

  long i;
  nPixels = im.rows * im.cols;
  for(i=0;i<nPixels;i++) {
    im.depth[i] = 100000.0;
    im.image[i] = 0x00000000;
  }

  Module box;

  box.MakeBox(1, 1, 1, Purple, spec);

  Module background;
  Vector_t square[4];
  Vector_t normal[4];
  Vector_t color[4];
  Vector_t surface[4];
  

  square[0] = Vector(-100, -100, 10);
  square[1] = Vector(100, -100, 10);
  square[2] = Vector(100, 100, 10);
  square[3] = Vector(-100, 100, 10);

  for(i=0;i<4;i++) {
    normal[i] = Vector(0, 0, -1);
    color[i] = Vector(01.0, 1.0, 1.0);
    surface[i] = Vector(0.1, 0.1, 0.1);
  }

  background.Begin();

    background.Polygon(square, normal, color, surface, 4);

  background.End();

  Module block;

  block.Begin();

    block.RotateY(cos(angle1), sin(angle1));

    block.RotateX(cos(angle2), sin(angle2));

    block.Insert(&box);

    block.Translate(2, 2, 0);
    block.Insert(&box);

    block.Translate(-4, 0, 0);
    block.Insert(&box);

    block.Translate(0, -4, 0);
    block.Insert(&box);

    block.Translate(4, 0, 0);
    block.Insert(&box);

  block.End();

  Module blocks;
  blocks.Begin();

    blocks.Insert(&block);

    blocks.Translate(8, 8, 0);
    blocks.Insert(&block);

    blocks.Translate(-16, 0, 0);
    blocks.Insert(&block);

    blocks.Translate(0, -16, 0);
    blocks.Insert(&block);

    blocks.Translate(16, 0, 0);
    blocks.Insert(&block);

  blocks.End();

  Module scene;
  Light_t amb, light;

  amb.Set(AMBIENT, Vector(0, 0, 0), Vector(0, 0, 0), ColorVector(0.3, 0.3, 0.3), 0.0);
  light.Set(POINT, Vector(5, 5, -10), Vector(0, 0, 0), ColorVector(1.0, 1.0, 1.0), 0.0);

  scene.Begin();
    scene.Set3DPerspView(&vs);

    scene.AddLight(&amb);
    scene.AddLight(&light);

    scene.Insert(&background);

    //    scene.Insert(&box);

    //    scene.Translate(-2, 2, 0); fig 2
    //    scene.Scale(2, 2, 2); fig 3
    //    scene.RotateZ(cos(M_PI/3), sin(M_PI/3)); fig 4

    /* figure 5
    scene.RotateZ(cos(M_PI/3), sin(M_PI/3));
    scene.Scale(2, 2, 2);
    scene.Translate(-1, 1, 0);
    */
    
    scene.RotateZ(cos(angle1*2), sin(angle1*2));
    scene.Insert(&blocks);

  scene.End();

  // draw the scene on the image
  scene.ZBufferOn();
  scene.Draw(&im);

  // define an output file name
  char buffer[64];
  if(argc > 1)
    sprintf(buffer, "math.%03d.tif", (int)(angle1*180 / M_PI + 0.5));
  else
    strcpy(buffer, "tmp.tif");

  // write out the image
  tiff_raster2file(buffer, im.image, im.rows, im.cols);
}