#include "gd.h"
#include <vector>
#include <set>
#include <fstream>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <string>

vector<float> x,y,z;
vector<int> colors(256);
vector<vector<int> > nbd;
set<int> boundary;
string infile,outfile;


  /* Declare the image */
gdImagePtr im;
int size=200;
int black;
float minx,maxx,miny,maxy,minz,maxz;

void readfile(void) {
int i,k;
vector<int> nbs;
float xx,yy,zz;
int a,b,c,d,e,f;
ifstream BE(infile.c_str());
while (BE.good()) {
  BE >> i;
  nbs.clear();
  int c=0;;
  for (int j=0;j<6;j++) {
     BE >> k;
     if (k) nbs.push_back(k-1);
  }
  if (nbs.size()<6) boundary.insert(i-1);
  nbd.push_back(nbs);
  BE >>xx >>yy >>zz;
  x.push_back(xx);
  y.push_back(yy);
  z.push_back(zz);
//  cout << ".";
}
for (int i=0;i<x.size();i++) {
  vector<int> nnbs;
  for (int j=0;j<nbd[i].size();j++) {
    if (boundary.count(nbd[i][j])) nnbs.push_back(nbd[i][j]);
  }
//  if (nnbs.size()!=nbd[i].size()) cout << ".";
  nbd[i]=nnbs;
}
cout << x.size() << " points\n";
cout << boundary.size() <<" boundary points\n";
}

void makecolors(void){
black=gdImageColorAllocate (im, 0,0,0);
for (int i=1;i<256;i++) {
 colors[i]= gdImageColorAllocate (im, i,255-i,256/2);
}
}

int getcolor(float xx, float yy) {
//return 150;
float closez;
float closed=10000;
for (int i=0;i<x.size();i++) {
  float d=pow(xx-x[i],2)+pow(yy-y[i],2);
  if (d<closed) {
     closed=d;
     closez=z[i];
  }
}
// cout << ",\n";
float col=(closez-minz)*254.0/(maxz-minz)+1;
return colors[int(col)];
}

int
main (int argc, char **argv)
{
  size=atoi(argv[1]);
  infile=argv[2];
  outfile=argv[3];
  cout << infile << " -> " << outfile << "\n";
  readfile();
  /* Declare output files */
  FILE *pngout, *jpegout;
  /* Declare color indexes */

  /* Allocate the image: ? pixels across by ? pixels tall */
  im = gdImageCreate (size, size);

  makecolors();

  minx=*min_element(x.begin(),x.end()); 
  cout << "minx " << minx << "\n";
  miny=*min_element(y.begin(),y.end()); 
  cout << "miny " << miny << "\n";
  minz=*min_element(z.begin(),z.end()); 
  cout << "minz " << minz << "\n";

  maxx=*max_element(x.begin(),x.end()); 
  cout << "maxx " << maxx << "\n";
  maxy=*max_element(y.begin(),y.end()); 
  cout << "maxy " << maxy << "\n";
  maxz=*max_element(z.begin(),z.end()); 
  cout << "maxz " << maxz << "\n";

  float rsize=max(maxx-minx,maxy-miny);
  float step=rsize/size;
  int i=0;
  for (float xx=minx;i<size;xx+=step,i++){
     int j=0;
     for (float yy=miny;j<size;yy+=step,j++){
       int k=getcolor(xx,yy);
       gdImageSetPixel (im, i,j, k);
     }  
  }


for (int i=0;i<x.size();i++) {
   gdImageSetPixel (im, int((x[i]-minx)/step),int((y[i]-miny)/step), black);
   if (nbd[i].size()==2 && boundary.count(i)) {
      for (int j=0; j<nbd[i].size();j++) {
         int other=nbd[i][j];
       //  if (nbd[other].size()<6 ) {
             gdImageLine(im,int((x[i    ]-minx)/step),int((y[i    ]-miny)/step),
                            int((x[other]-minx)/step),int((y[other]-miny)/step),black);
       //  }
      }
   }  
}  
  pngout = fopen (outfile.c_str(), "wb");
//  jpegout = fopen ("test.jpg", "wb");
  gdImagePng (im, pngout);
//  gdImageJpeg (im, jpegout, -1);
  fclose (pngout);
//  fclose (jpegout);
  gdImageDestroy (im);
}