package collembola;
import java.awt.color.ColorSpace;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.HashMap;
import java.util.Random;

import javax.swing.JFrame;

import mcmas.core.MCMChrono;

import org.geotools.data.FileDataStore;
import org.geotools.data.FileDataStoreFinder;
import org.geotools.data.shapefile.ng.ShapefileDataStore;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.data.simple.SimpleFeatureIterator;
import org.geotools.data.simple.SimpleFeatureSource;
import org.geotools.geometry.jts.ReferencedEnvelope;
import org.opengis.feature.simple.SimpleFeature;

import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.MultiPolygon;
import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Point;

public class MapMain {
	
	/*
	public void displayLayers(File shpFile) throws IOException {
		System.out.println(shpFile);
		
		FileDataStore store = FileDataStoreFinder.getDataStore(shpFile);
		
		System.out.println(shpFile.length());
		SimpleFeatureSource shapefileSource = store.getFeatureSource();
		
		Style style = SLD.createPolygonStyle(Color.GREEN, null, 0.0f);
		
		final MapContent map = new MapContent();
		map.setTitle("Test display GIS");
		
		Layer shpLayer = new FeatureLayer(shapefileSource, style);
		map.addLayer(shpLayer);
		
		JMapFrame frame = new JMapFrame(map);
		frame.setSize(800, 600);
		frame.enableStatusBar(true);
		frame.enableToolBar(true);
		frame.setVisible(true);
	}*/
	
	public void displayLayers2(URL shpFile) throws IOException {
		ShapefileDataStore store = new ShapefileDataStore(shpFile);
		String name = store.getTypeNames()[0];
		SimpleFeatureSource source = store.getFeatureSource(name);
		SimpleFeatureCollection collection = source.getFeatures();
		SimpleFeatureIterator it = collection.features();
		GeometryFactory gf = new GeometryFactory();
		
		ReferencedEnvelope env = source.getBounds();
		
		int pict_width = (int) Math.ceil(env.getMaxX() - env.getMinX());
		int pict_height = (int) Math.ceil(env.getMaxY() - env.getMinY());
		
		System.out.println(pict_width);
		System.out.println(pict_height);
		
		//Point [][] points = new Point[256][256];
		String [][] grid = new String[256][256];
		
		HashMap<String, Integer> colorMap = new HashMap<String, Integer>();
		Random rng = new Random();
		
		//int index = 0;
		
		MCMChrono ch = new MCMChrono("interections").start();
		
		/*
		for (int i = 0; i < points.length; i++) {
			for (int j = 0; j < points[i].length; j++) {
				final double xpos = transform(i + 0.5, 0, grid.length, env.getMinX(), env.getMaxX());
				final double ypos = transform(j + 0.5, 0, grid[i].length, env.getMinY(), env.getMaxY());
				points[i][j] = gf.createPoint(new Coordinate(xpos, ypos));
			}
		}*/
		
		while (it.hasNext()) {
			//index++;
			SimpleFeature f = it.next();
			//Polygon p = new Polygon();
			
			//System.out.println(f.getProperty("level1"));
			//System.out.println(f.getProperties());
			MultiPolygon poly = (MultiPolygon) f.getDefaultGeometry();
			//Coordinate[] c = poly.getCoordinates();
			colorMap.put(f.getID(), rng.nextInt());
			
			System.out.println(f.getID());
			System.out.println(f.getAttribute("Level1"));
			
			//System.out.println(poly.getCentroid());
			//System.out.println(c.length);
			
			for (int i = 0; i < grid.length; i++) {
				for (int j = 0; j < grid[i].length; j++) {
					if (grid[i][j] != null) continue;
					
					final double xpos = transform(i + 0.5, 0, grid.length, env.getMinX(), env.getMaxX());
					final double ypos = transform(j + 0.5, 0, grid[i].length, env.getMinY(), env.getMaxY());
					
					//if (grid[i][j] != null) continue;
					
					//System.out.println("xpos: " + xpos);
					//System.out.println("ypos: " + ypos);
					Point point = gf.createPoint(new Coordinate(xpos, ypos));
					//final Point point = points[i][j];
					
					if (poly.contains(point)) {
						/*System.out.println("xpos: " + xpos);
						System.out.println("ypos: " + ypos);
						System.out.println("true");*/
						grid[i][j] = f.getID();
					}
				}
			}
		}
		
		System.out.println(ch.stop());
		
		ch = new MCMChrono("image generation").start();
		
		BufferedImage img = new BufferedImage(grid.length, grid.length, ColorSpace.TYPE_RGB);
		
		for (int i = 0; i < grid.length; i++) {
			for (int j = 0; j < grid[i].length; j++) {
				/*if (! colorMap.containsKey(grid[i][j])) {
					colorMap.put(grid[i][j], rng.nextInt());
				}*/
				
				//System.out.print("" + j + " ");
				//System.out.println(grid.length - j);
				Integer c = colorMap.get(grid[i][j]);
				
				if (c != null)
					img.setRGB(i, grid.length - 1 - j, c);
				//System.out.print(grid[i][j] + " ");
			}
			//System.out.println();
		}
		
		System.out.println(ch.stop());
		
		//ImageIO.write(img, "png", new File("out.png"));
		
		System.out.println(source.getBounds());
		//BufferedImage img = new BufferedImage(20000, height, imageType)
		
		System.out.println(transform(128, 0, 256, 0, 1000));
		
		System.out.println(img);
		ImagePanel panel = new ImagePanel(img);
		JFrame frame = new JFrame("MapGIS");
		frame.setContentPane(panel);
		frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
		frame.pack();
		frame.setVisible(true);
		
	}
	
	public double transform(double n, double min1, double max1, double min2, double max2) {
		final double range1 = max1 - min1;
		final double range2 = max2 - min2;
		
		//System.out.println(range1);
		//System.out.println(range2);
		return ((n - min1) / range1) * range2 + min2;
	}
	
	public static void test(String[] args) throws IOException {
		File file = new File("110m_cultural\\110m_admin_0_countries.shp");
        FileDataStore store = FileDataStoreFinder.getDataStore(file);
        SimpleFeatureSource featureSource = store.getFeatureSource();
        SimpleFeatureCollection c = featureSource.getFeatures();
        SimpleFeatureIterator featuresIterator = c.features();
       
        while (featuresIterator.hasNext()) {
            SimpleFeature o = featuresIterator.next();
            
            String name = (String) o.getAttribute("NAME");
            Object geometry = o.getDefaultGeometry();
            System.out.println(name + ", " + geometry.getClass());
        }
	}
	
	public static void main(String[] args) throws IOException {
		//ClassLoader cl = ClassLoader.getSystemClassLoader();
		 
        //URL[] urls = ((URLClassLoader)cl).getURLs();
 
        /*for(URL url: urls){
        	System.out.println(url.getFile());
        }*/
        
		//new MapMain().displayLayers(new File("site4.shp"));
        new MapMain().displayLayers2(new URL("file://data/site4.shp"));
	}
}