package com.volmit.iris.gui;

import java.awt.Color;
import java.awt.Dimension;
import java.awt.EventQueue;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Point;
import java.awt.Rectangle;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseMotionListener;
import java.awt.event.MouseWheelEvent;
import java.awt.event.MouseWheelListener;
import java.util.concurrent.locks.ReentrantLock;

import javax.swing.JComboBox;
import javax.swing.JComponent;
import javax.swing.JFrame;
import javax.swing.JLayeredPane;
import javax.swing.JPanel;
import javax.swing.JViewport;

import com.volmit.iris.Iris;
import com.volmit.iris.gen.IrisChunkGenerator;
import com.volmit.iris.noise.CNG;
import com.volmit.iris.object.NoiseStyle;
import com.volmit.iris.util.Function2;
import com.volmit.iris.util.GroupedExecutor;
import com.volmit.iris.util.J;
import com.volmit.iris.util.KList;
import com.volmit.iris.util.M;
import com.volmit.iris.util.PrecisionStopwatch;
import com.volmit.iris.util.RNG;
import com.volmit.iris.util.RollingSequence;

public class NoiseView extends JPanel implements MouseWheelListener {

	private static final long serialVersionUID = 2094606939770332040L;

	static JComboBox<String> combo;
	RollingSequence r = new RollingSequence(90);
	boolean colorMode = true;
	double scale = 1;
	static boolean hd = false;
	static double ascale = 10;
	CNG cng = NoiseStyle.STATIC.create(new RNG(RNG.r.nextLong()));
	GroupedExecutor gx = new GroupedExecutor(Runtime.getRuntime().availableProcessors(), Thread.MAX_PRIORITY,
			"Iris Renderer");
	ReentrantLock l = new ReentrantLock();
	int[][] co;
	int w = 0;
	int h = 0;
	static Function2<Double, Double, Color> renderer;
	static double oxp = 0;
	static double ozp = 0;
	double ox = 0;
	double oz = 0;
	double mx = 0;
	double mz = 0;
	static double mxx = 0;
	static double mzz = 0;
	static boolean down = false;

	double lx = Double.MAX_VALUE;
	double lz = Double.MAX_VALUE;
	double tz = 1D;
	double t = 1D;

	public NoiseView() {

		addMouseWheelListener((MouseWheelListener) this);
		addMouseMotionListener(new MouseMotionListener() {

			@Override
			public void mouseMoved(MouseEvent e) {
				Point cp = e.getPoint();

				lx = (cp.getX());
				lz = (cp.getY());
				mx = lx;
				mz = lz;
			}

			@Override
			public void mouseDragged(MouseEvent e) {
				Point cp = e.getPoint();
				ox += (lx - cp.getX()) * scale;
				oz += (lz - cp.getY()) * scale;
				lx = cp.getX();
				lz = cp.getY();

			}
		});
	}

	public void mouseWheelMoved(MouseWheelEvent e) {

		int notches = e.getWheelRotation();
		if (e.isControlDown()) {
			t = t + ((0.001 * t) * notches);
			return;
		}

		scale = scale + ((0.044 * scale) * notches);
		scale = scale < 0.00001 ? 0.00001 : scale;
	}

	@Override
	public void paint(Graphics g) {

		if (scale < ascale) {
			ascale -= Math.abs(scale - ascale) * 0.16;
		}

		if (scale > ascale) {
			ascale += Math.abs(ascale - scale) * 0.16;
		}

		if (t < tz) {
			tz -= Math.abs(t - tz) * 0.29;
		}

		if (t > tz) {
			tz += Math.abs(tz - t) * 0.29;
		}

		if (ox < oxp) {
			oxp -= Math.abs(ox - oxp) * 0.16;
		}

		if (ox > oxp) {
			oxp += Math.abs(oxp - ox) * 0.16;
		}

		if (oz < ozp) {
			ozp -= Math.abs(oz - ozp) * 0.16;
		}

		if (oz > ozp) {
			ozp += Math.abs(ozp - oz) * 0.16;
		}

		if (mx < mxx) {
			mxx -= Math.abs(mx - mxx) * 0.16;
		}

		if (mx > mxx) {
			mxx += Math.abs(mxx - mx) * 0.16;
		}

		if (mz < mzz) {
			mzz -= Math.abs(mz - mzz) * 0.16;
		}

		if (mz > mzz) {
			mzz += Math.abs(mzz - mz) * 0.16;
		}

		PrecisionStopwatch p = PrecisionStopwatch.start();
		int accuracy = hd ? 1 : M.clip((r.getAverage() / 6D), 1D, 128D).intValue();
		accuracy = down ? accuracy * 4 : accuracy;
		int v = 1000;

		Iris.proj.getCurrentProject().getCache().targetChunk(0, 0);

		if (g instanceof Graphics2D) {
			Graphics2D gg = (Graphics2D) g;

			if (getParent().getWidth() != w || getParent().getHeight() != h) {
				w = getParent().getWidth();
				h = getParent().getHeight();
				co = null;
			}

			if (co == null) {
				co = new int[getParent().getWidth()][getParent().getHeight()];
			}

			for (int x = 0; x < getParent().getWidth(); x += accuracy) {
				int xx = x;

				for (int z = 0; z < getParent().getHeight(); z += accuracy) {
					int zz = z;
					gx.queue("a", () -> {
						if (renderer != null) {
							co[xx][zz] = renderer.apply((xx * ascale) + oxp, (zz * ascale) + ozp).getRGB();
						}

						else {
							double n = cng.noise((xx * ascale) + oxp, tz, (zz * ascale) + ozp);

							if (n > 1 || n < 0) {
								System.out.println("EXCEEDED " + n);
								return;
							}

							Color color = colorMode
									? Color.getHSBColor((float) (n), 1f - (float) (n * n * n * n * n * n),
											1f - (float) n)
									: Color.getHSBColor(0f, 0f, (float) n);
							int rgb = color.getRGB();
							co[xx][zz] = rgb;
						}
					});

				}

				gx.waitFor("a");

				if (hd && p.getMilliseconds() > v) {
					break;
				}
			}

			for (int x = 0; x < getParent().getWidth(); x += accuracy) {
				for (int z = 0; z < getParent().getHeight(); z += accuracy) {
					gg.setColor(new Color(co[x][z]));
					gg.fillRect(x, z, accuracy, accuracy);
				}
			}

			if (renderer != null) {
				String text = Iris.proj.getCurrentProject().textFor((mxx * ascale) + oxp, (mzz * ascale) + ozp);
				gg.setColor(Color.black);
				gg.setFont(new Font("TimesRoman", Font.PLAIN, 24));
				gg.drawString(text, (float) mxx, (float) mzz);
			}
		}

		p.end();

		t += 1D;
		r.put(p.getMilliseconds());

		if (!isVisible()) {
			return;
		}

		if (!getParent().isVisible()) {
			return;
		}

		if (!getParent().getParent().isVisible()) {
			return;
		}

		EventQueue.invokeLater(() -> {
			J.sleep((long) Math.max(0, 32 - r.getAverage()));
			repaint();
		});
	}

	private static void createAndShowGUI(IrisChunkGenerator g) {
		JFrame frame = new JFrame("Vision");
		NoiseView nv = new NoiseView();
		frame.setDefaultCloseOperation(JFrame.HIDE_ON_CLOSE);
		KList<String> li = new KList<NoiseStyle>(NoiseStyle.values()).toStringList().qadd("PROJECT");
		combo = new JComboBox<String>(li.toArray(new String[li.size()]));
		combo.setSelectedItem(g != null ? "PROJECT" : "STATIC");

		if (g != null) {
			renderer = Iris.proj.getCurrentProject().createRenderer();
		}

		combo.setFocusable(false);
		combo.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent e) {
				@SuppressWarnings("unchecked")
				String b = (String) (((JComboBox<String>) e.getSource()).getSelectedItem());
				if (b.equals("PROJECT")) {
					renderer = Iris.proj.getCurrentProject().createRenderer();
					return;
				}
				renderer = null;
				NoiseStyle s = NoiseStyle.valueOf(b);
				nv.cng = s.create(RNG.r.nextParallelRNG(RNG.r.imax()));
			}
		});

		combo.setSize(500, 30);
		JLayeredPane pane = new JLayeredPane();
		nv.setSize(new Dimension(1440, 820));
		pane.add(nv, 1, 0);
		pane.add(combo, 2, 0);
		frame.add(pane);
		frame.setSize(1440, 820);
		frame.setVisible(true);
	}

	public static void launch(IrisChunkGenerator g) {
		EventQueue.invokeLater(new Runnable() {
			public void run() {
				createAndShowGUI(g);
			}
		});
	}

	static class HandScrollListener extends MouseAdapter {
		private static final Point pp = new Point();

		@Override
		public void mouseDragged(MouseEvent e) {
			JViewport vport = (JViewport) e.getSource();
			JComponent label = (JComponent) vport.getView();
			Point cp = e.getPoint();
			Point vp = vport.getViewPosition();
			vp.translate(pp.x - cp.x, pp.y - cp.y);
			label.scrollRectToVisible(new Rectangle(vp, vport.getSize()));

			pp.setLocation(cp);
		}

		@Override
		public void mousePressed(MouseEvent e) {
			pp.setLocation(e.getPoint());
		}
	}
}