package de.c3ma.ollo.mockup;

import java.awt.Color;
import java.io.File;
import java.util.ArrayList;

import javax.swing.SwingUtilities;

import org.luaj.vm2.LuaString;
import org.luaj.vm2.LuaTable;
import org.luaj.vm2.LuaValue;
import org.luaj.vm2.Varargs;
import org.luaj.vm2.lib.OneArgFunction;
import org.luaj.vm2.lib.TwoArgFunction;
import org.luaj.vm2.lib.VarArgFunction;
import org.luaj.vm2.lib.ZeroArgFunction;

import de.c3ma.ollo.LuaSimulation;
import de.c3ma.ollo.mockup.ui.WS2812Layout;

/**
 * created at 28.12.2017 - 23:34:04<br />
 * creator: ollo<br />
 * project: WS2812Emulation<br />
 * $Id: $<br />
 * 
 * @author ollo<br />
 */
public class ESP8266Ws2812 extends TwoArgFunction {

	private static WS2812Layout layout = null;

	@Override
	public LuaValue call(LuaValue modname, LuaValue env) {
		env.checkglobals();
		final LuaTable ws2812 = new LuaTable();
		ws2812.set("init", new init());
		ws2812.set("write", new write());
		ws2812.set("newBuffer", new newBuffer());
		env.set("ws2812", ws2812);
		env.get("package").get("loaded").set("ws2812", ws2812);
		return ws2812;
	}

	private class init extends ZeroArgFunction {

		@Override
		public LuaValue call() {
			System.out.println("[WS2812] init");
			return LuaValue.valueOf(true);
		}

	}

	private class write extends OneArgFunction {

		@Override
		public LuaValue call(LuaValue arg) {
			if (arg.isstring()) {
				LuaString jstring = arg.checkstring();
				final int length = jstring.rawlen();

				if (ESP8266Ws2812.layout == null) {
					System.err.println("[WS2812] Not initialized (" + length + "bytes to be updated)");
					return LuaValue.valueOf(false);
				}
				
				if ((length % 3) == 0) {
					final byte[] array = jstring.m_bytes;
					SwingUtilities.invokeLater(new Runnable() {
						@Override
						public void run() {
							for (int i = 0; i < length; i += 3) {
								if (ESP8266Ws2812.layout != null) {
									int r = array[i + 0]+(Byte.MIN_VALUE*-1);
									int b = array[i + 1]+(Byte.MIN_VALUE*-1);
									int g = array[i + 2]+(Byte.MIN_VALUE*-1);
									ESP8266Ws2812.layout.updateLED(i / 3, r, g, b);
								}
							}
						}
					});
				}
				return LuaValue.valueOf(true);
			} else {
				System.out.println("[WS2812] write no string given");
				return LuaValue.NIL;
			}
		}
	}

	public void setLayout(File file, LuaSimulation nodemcuSimu) {
		if (ESP8266Ws2812.layout == null) {
			ESP8266Ws2812.layout = WS2812Layout.parse(file, nodemcuSimu);
		}
	}
	
	private class newBuffer extends VarArgFunction {
    	
        public Varargs invoke(Varargs varargs) {
            if (varargs.narg() == 2) {
                final int leds = varargs.arg(1).toint();
                final int bytesPerLeds = varargs.arg(2).toint();
                final LuaTable rgbBuffer = new LuaTable();
                ArrayList<Color> ledList = new ArrayList<Color>();
                for(int i=0; i < leds; i++) {
                	ledList.add(new Color(0,0,0));
                }
                rgbBuffer.set("fill", new bufferFill(ledList));
                rgbBuffer.set("set", new bufferWrite(ledList));
                rgbBuffer.set("get", new bufferRead(ledList));
                System.out.println("[WS2812] " + leds + "leds (" + bytesPerLeds + "bytes per led)");                
                return rgbBuffer;
            } else {
            	return LuaValue.NIL;
            }
        }
    }
	
	private class bufferFill extends VarArgFunction {
		
		private ArrayList<Color> ledList = null;
		
		public bufferFill(ArrayList<Color> ledList) {
			this.ledList = ledList;
		}
    	
        public Varargs invoke(Varargs varargs) {
            if (varargs.narg() >= 3) {
                final int red = varargs.arg(1).toint();
                final int green = varargs.arg(2).toint();
                final int blue = varargs.arg(3).toint();
                /* update local buffer */ 
                for(int i=0; i < ledList.size(); i++) {
                	ledList.set(i, new Color(red, green, blue));
                }
                /* Update GUI */
                if (ESP8266Ws2812.layout != null) {
                	SwingUtilities.invokeLater(new Runnable() {
						@Override
						public void run() {
							ESP8266Ws2812.layout.fillLEDs(red, green, blue);
						}
                	});
				}
                System.out.println("[WS2812] buffer fill with " + red + "," + green + "," +  blue);
                return LuaValue.valueOf(true);
            } else if (varargs.isstring(2)) {
            	final LuaString color = varargs.arg(2).checkstring();
            	
 				final int length = color.rawlen();
 				if ((length == 3) && (ESP8266Ws2812.layout != null)) {

 					final byte[] array = color.m_bytes;
 					final int r = array[0]+(Byte.MIN_VALUE*-1);
 					final int b = array[1]+(Byte.MIN_VALUE*-1);
 					final int g = array[2]+(Byte.MIN_VALUE*-1);
 					/* update local buffer */ 
 	                for(int i=0; i < ledList.size(); i++) {
 	                	ledList.set(i, new Color(r, g, b));
 	                }
 	                /* Update GUI */
 					SwingUtilities.invokeLater(new Runnable() {
						@Override
						public void run() {
		 					ESP8266Ws2812.layout.fillLEDs(r, g, b);
						} 
					});
 	 				//System.out.println("[WS2812] buffer fill with " + r + "," + g + "," +  b);
 	 				return LuaValue.valueOf(true);
 				} else {
 					System.err.println("[WS2812] buffer not initialized ("+varargs.narg() +"args) , length "+ length + ", raw:" + color.toString());
 	 				return LuaValue.NIL;
 				}
            } else {
            	System.err.println("[WS2812] fill with " + varargs.narg() + " arguments undefined.");
            	return LuaValue.NIL;
            }
        }
    }
	
	private class bufferWrite extends VarArgFunction {
		private ArrayList<Color> ledList = null;
		
		public bufferWrite(ArrayList<Color> ledList) {
			this.ledList = ledList;
		}
		
        public Varargs invoke(Varargs varargs) {
            if (varargs.narg() == 3) {
                final int index = varargs.arg(2).toint();
                final LuaString color = varargs.arg(3).checkstring();
 				final int length = color.rawlen();
				if (length == 3) {
					final byte[] array = color.m_bytes;
					final int r = array[0]+(Byte.MIN_VALUE*-1);
					final int b = array[1]+(Byte.MIN_VALUE*-1);
					final int g = array[2]+(Byte.MIN_VALUE*-1);
					// update buffer
	                ledList.set(index - 1, new Color(r, g, b));
	                
	                // update GUI
					SwingUtilities.invokeLater(new Runnable() {
						@Override
						public void run() {
							ESP8266Ws2812.layout.updateLED(index - 1, r, g, b);
						}
					});
	                return LuaValue.valueOf(true);
				} else {					
					System.err.println("[WS2812] set with " + varargs.narg() + " arguments at index="+ index + " and "+ length + " charactes not matching");
	            	return LuaValue.NIL;	
				}
            } else if (varargs.narg() == 5) {
                final int index = varargs.arg(2).toint();
                final int green = varargs.arg(3).toint();
                final int red = varargs.arg(4).toint();
                final int blue = varargs.arg(5).toint();
                // update buffer
                ledList.set(index - 1, new Color(red, green, blue));
                
                // update GUI
				SwingUtilities.invokeLater(new Runnable() {
					@Override
					public void run() {
						ESP8266Ws2812.layout.updateLED(index - 1, red, green, blue);
					}
				});
				return LuaValue.valueOf(true);
            } else {
				for(int i=0; i <= varargs.narg(); i++) {
					System.err.println("[WS2812] write ["+(i) + "] (" + varargs.arg(i).typename() + ") " + varargs.arg(i).toString() );
				}
            	System.err.println("[WS2812] set with " + varargs.narg() + " arguments undefined.");
            	return LuaValue.NIL;
            }
        }
    }
	
	private class bufferRead extends VarArgFunction {
		private ArrayList<Color> ledList = null;
		
		public bufferRead(ArrayList<Color> ledList) {
			this.ledList = ledList;
		}
		
		@Override
		public Varargs invoke(Varargs varargs) {	
			final int offset = varargs.arg(2).toint();
			
			if (ledList != null) {
				// receiver from buffer
				Color color = ledList.get(offset - 1);
				final char[] array = new char[3];
				array[0] = (char) (color.getRed() );
				array[1] = (char) (color.getGreen() );
				array[2] = (char) (color.getBlue() );
				
//				System.err.println("[WS2812] reading " + offset + ":" + ((int)array[0]) +"," + ((int) array[1]) + "," + ((int) array[2])  + " from " + color);
				return LuaString.valueOf(array);
			}

			System.err.println("[WS2812] reading " + offset + " impossible");
			return LuaValue.NIL;
		}
	}
}