import processing.opengl.*;
import org.gdal.ogr.*;
import org.gdal.gdal.*;

import java.util.*;


float xmin,xmax,ymin,ymax;
int redval = 0;

DataSource src;
Layer lyr;
ArrayList<Feature> ftrlist; //contains all our features/suburbs

PGraphics buffer;   //backbuffer which allows for picking!
//suburb[] suburbs;   //suburbs vector to hold the mapped vertices

LinkedList<PFeature>[] mygeoms;

void setup()
{
  size(600,600,OPENGL);//P3D);
  background(127);
  frameRate(30);

  //  // buffer is created using applet dimensions
  //  buffer = createGraphics(width, height, P3D);

  // Let the user select a Shapefile
  String loadPath = selectInput();
  if(loadPath == null)
  {
    println("No file was selected!");
    exit();
    return;
  }
  println(loadPath);

  String[] arr = {
    "--debug", "on"
  };
  ogr.GeneralCmdLineProcessor(arr);
  ogr.RegisterAll();
  print("No. of available drivers : ");
  println(ogr.GetDriverCount());

  src = ogr.Open(loadPath);
  if(src == null)
  {
    println("unable to open!!==");
    println(gdal.GetLastErrorMsg());
    println("==");
    return;
  }

  System.out.printf(":::: %s information ::::\n", (new File(loadPath)).getName());
  System.out.printf("No. of Layers : %d\n", src.GetLayerCount());
  double[] extents;
  for(int l=0; l<src.GetLayerCount(); l++)
  {
    System.out.printf("Layer %d\n", l);
    lyr = src.GetLayer(l);
    System.out.printf("  Name = %s\n", lyr.GetName());
    System.out.printf("  No. of features : %d\n", lyr.GetFeatureCount());
    extents = lyr.GetExtent();
    System.out.printf("  %f  %f  %f  %f\n", extents[0], extents[1], extents[2], extents[3]);
  }
  println("=================");

  // use first layer of data  
  lyr = src.GetLayer(0);
  extents = lyr.GetExtent();
  xmin = (float)extents[0];
  xmax = (float)extents[1];
  ymin = (float)extents[2];
  ymax = (float)extents[3];

  lyr.ResetReading();
  Feature ftr;
  ftrlist = new ArrayList<Feature>(lyr.GetFeatureCount());
  for(int f=0; f<lyr.GetFeatureCount(); f++)
  {
    ftr = lyr.GetNextFeature();
    System.out.printf("%d with %d fields\n", ftr.GetFID(), ftr.GetFieldCount());
    ftrlist.add(ftr);
  }
  System.out.printf("ftrlist.size()==%d\n", ftrlist.size());

  //allocate to store the geometry data
  mygeoms = new LinkedList[lyr.GetFeatureCount()];

  Geometry ftrgeom;
  Geometry geom;
  ftr = lyr.GetFeature(0);
  ftrgeom = ftr.GetGeometryRef();
  System.out.printf("Feature : %d\n", 0);
  System.out.printf("  Geometry : %s", ftrgeom.GetGeometryName());
  System.out.printf(" consisting of %d geometries\n", ftrgeom.GetGeometryCount());
  for(int i=0; i<ftrgeom.GetGeometryCount(); i++)
  {
    geom = ftrgeom.GetGeometryRef(i);
    System.out.printf("    %d - %s (%d)\n", i, geom.GetGeometryName(),geom.GetGeometryCount());
    if(geom.GetGeometryCount()==1)
    {
      System.out.printf("      %d points\n", geom.GetGeometryRef(0).GetPointCount());
    }
  }
  //  ftrgeom = ftrgeom.GetGeometryRef(0);
  //  System.out.printf("   %d points\n", ftrgeom.GetPointCount());
  //  System.out.printf("   %s <- name\n", ftrgeom.GetGeometryName());
  //  ftrgeom.GetEnvelope(extents);
  //  System.out.printf("  %f  %f  %f  %f\n", extents[0], extents[1], extents[2], extents[3]);


  //Process the features
  System.out.printf("\n\n\n");
//  myftrs = new PFeature[8];
  for(int f=0; f<lyr.GetFeatureCount(); f++)//later update limits to accomodate all features
  {
    mygeoms[f] = new LinkedList();
    ftr = lyr.GetFeature(f);
    ftrgeom = ftr.GetGeometryRef();
    System.out.printf("Feature %d::\n", f);
    processFeature(ftrgeom, f);
  }

}



void draw()
{
  //flip y-axis to upwards and setup origin on lower left corner
  scale(1,-1);
  translate(0,-height+1);

  //now features are processed; try plotting them!
  for(int f=0; f<lyr.GetFeatureCount(); f++)//later update limits to accomodate all features
  {
//    System.out.printf("Feature %d::\t", f);
//    System.out.printf("%d geometries\n", mygeoms[f].size());
    ListIterator<PFeature> li = mygeoms[f].listIterator();
//    mygeoms[f].get(0).draw(this.g);
    fill(246,46,78);
    int ll = 0;
    while(li.hasNext())
    {
      PFeature tmpf = li.next();
//      System.out.printf("  geometry %d -> ", ll);
//      System.out.printf("%d points\n", (tmpf.closed ? tmpf.npts+1 : tmpf.npts));
      tmpf.draw(this.g);
//      li.next().draw(this.g);
      ll++;
    }

    //draw the BBs for the features
    noFill();
    double[] extents = new double[4];
    Geometry ftrgeom = lyr.GetFeature(f).GetGeometryRef();
    ftrgeom.GetEnvelope(extents);
//    System.out.printf("  %f  %f  %f  %f\n", extents[0], extents[1], extents[2], extents[3]);
    float x1,y1,x2,y2;
    x1 = map((float)extents[0],xmin,xmax,0.0,width-1.0);
    x2 = map((float)extents[1],xmin,xmax,0.0,width-1.0);
    y1 = map((float)extents[2],ymin,ymax,0.0,height-1.0);
    y2 = map((float)extents[3],ymin,ymax,0.0,height-1.0);
    stroke(127,127,255);
    fill(127,127,255,50);
    rectMode(CORNERS);
    rect(x1,y1,x2,y2);
  }
}








void processFeature(Geometry ftrgeom, int f)
{
  PFeature pf = null;
  Geometry g;

  System.out.printf("  %s::", ftrgeom.GetGeometryName());
  switch(ftrgeom.GetGeometryType())
  {
  case ogr.wkbUnknown:
  case ogr.wkbNone:
    System.out.printf("  Unknown/None geometry type found!!\n");
    break;
  case ogr.wkbPoint:
  case ogr.wkbPoint25D:
    System.out.printf("  Point/Point25D geometry; ignoring!!\n");
    break;
  case ogr.wkbLineString:
  case ogr.wkbLineString25D:
    System.out.printf("  LineString found. Dumping now...\n");
    System.out.printf("    %d points\n", ftrgeom.GetPointCount());
    pf = new PFeature(ftrgeom.GetPointCount());
    for(int i=0; i<pf.npts; i++)
    {
      pf.tmpx[i] = map((float)ftrgeom.GetX(i),xmin,xmax,0.0,width-1.0);
      pf.tmpy[i] = map((float)ftrgeom.GetY(i),ymin,ymax,0.0,height-1.0);
    }
    pf.closed = false;
    mygeoms[f].add(pf);
    break;
  case ogr.wkbPolygon:
  case ogr.wkbPolygon25D:
    System.out.printf("  Polygon found. Dumping now...");
    System.out.printf("  %d rings\n",ftrgeom.GetGeometryCount());
    for(int r=0;r<ftrgeom.GetGeometryCount(); r++)
    {
      System.out.printf("    ring %d -> %d points\n", r, ftrgeom.GetGeometryRef(r).GetPointCount());
      g = ftrgeom.GetGeometryRef(r);
      //last pt is the same as first, better to let processing complete
      //the polygon. otherwise polygon might appear unclosed!
      pf = new PFeature(g.GetPointCount()-1);
      for(int i=0; i<pf.npts; i++)
      {
        pf.tmpx[i] = map((float)g.GetX(i),xmin,xmax,0.0,width-1.0);
        pf.tmpy[i] = map((float)g.GetY(i),ymin,ymax,0.0,height-1.0);
      }
      pf.closed = true;
      mygeoms[f].add(pf);
    }
    break;
  case ogr.wkbMultiPoint:
  case ogr.wkbMultiPoint25D:
  case ogr.wkbMultiLineString:
  case ogr.wkbMultiLineString25D:
  case ogr.wkbMultiPolygon:
  case ogr.wkbMultiPolygon25D:
  case ogr.wkbGeometryCollection:
  case ogr.wkbGeometryCollection25D:
//    System.out.printf("  Multi<something> found. Info...\n");
//    System.out.printf("    %d geometries\n",ftrgeom.GetGeometryCount());
    System.out.printf("  %d geometries:\n",ftrgeom.GetGeometryCount());
    for(int ig=0; ig<ftrgeom.GetGeometryCount(); ig++)//ig=innergeometry
    {
      System.out.printf("  (%d)", ig);
      processFeature(ftrgeom.GetGeometryRef(ig),f);
    }
    break;
  case ogr.wkbLinearRing:
    break;
  }
}