| 1 | package kaka.cakelight; |
| 2 | |
| 3 | import kaka.cakelight.mode.Mode; |
| 4 | import org.jnativehook.GlobalScreen; |
| 5 | import org.jnativehook.NativeHookException; |
| 6 | import org.jnativehook.keyboard.NativeKeyAdapter; |
| 7 | import org.jnativehook.keyboard.NativeKeyEvent; |
| 8 | import org.jnativehook.mouse.NativeMouseEvent; |
| 9 | import org.jnativehook.mouse.NativeMouseMotionAdapter; |
| 10 | |
| 11 | import java.util.Objects; |
| 12 | import java.util.Stack; |
| 13 | |
| 14 | public class CakeLight { |
| 15 | private Configuration config; |
| 16 | private Stack<Mode> modes = new Stack<>(); |
| 17 | private LedController ledController; |
| 18 | |
| 19 | public CakeLight(Configuration config, LedController ledController) { |
| 20 | this.config = config; |
| 21 | this.ledController = ledController; |
| 22 | Color.calculateGammaCorrection(config.gamma); |
| 23 | } |
| 24 | |
| 25 | public void setMode(Mode mode) { |
| 26 | cleanup(); |
| 27 | pushMode(mode); |
| 28 | } |
| 29 | |
| 30 | public void cleanup() { |
| 31 | while (popMode() != null); |
| 32 | } |
| 33 | |
| 34 | public void pushMode(Mode mode) { |
| 35 | Objects.requireNonNull(mode); |
| 36 | if (!modes.isEmpty()) { |
| 37 | pauseMode(modes.peek()); |
| 38 | } |
| 39 | modes.push(mode); |
| 40 | startMode(mode); |
| 41 | // TODO: create a composite fading mode of top of stack and new mode |
| 42 | } |
| 43 | |
| 44 | public Mode popMode() { |
| 45 | if (modes.size() > 1) { |
| 46 | Mode mode = modes.pop(); |
| 47 | stopMode(mode); |
| 48 | resumeMode(modes.peek()); |
| 49 | return mode; |
| 50 | } |
| 51 | return null; |
| 52 | // TODO: create a composite fading mode of popped mode and top of stack, unless doing cleanup |
| 53 | } |
| 54 | |
| 55 | private void startMode(Mode mode) { |
| 56 | mode.setFrameListener(ledController::onFrame); |
| 57 | mode.enter(config); |
| 58 | } |
| 59 | |
| 60 | private void pauseMode(Mode mode) { |
| 61 | mode.pause(); |
| 62 | } |
| 63 | |
| 64 | private void resumeMode(Mode mode) { |
| 65 | mode.resume(); |
| 66 | } |
| 67 | |
| 68 | private void stopMode(Mode mode) { |
| 69 | mode.setFrameListener(ledFrame -> {}); // To avoid any frame being sent to the controller while the thread is exiting |
| 70 | mode.exit(); |
| 71 | } |
| 72 | |
| 73 | public void startLoop() { |
| 74 | Console console = Console.start(this, config); |
| 75 | PipeController.start(console); |
| 76 | // initNativeHook(); |
| 77 | // TODO |
| 78 | // FrameGrabber grabber = FrameGrabber.from(config); |
| 79 | // grabber.prepare(); |
| 80 | // VideoFrame frame = grabber.grabFrame(); |
| 81 | // double time = 0; |
| 82 | // for (int i = 0; i < 100; i++) { |
| 83 | // time += timeIt("frame", () -> grabber.grabFrame()); |
| 84 | // } |
| 85 | // System.out.println("time = " + time); |
| 86 | // grabber.close(); |
| 87 | // byte[] data = frame.getData(); |
| 88 | // saveFile(data, "/home/kaka/test.img"); |
| 89 | } |
| 90 | |
| 91 | private void initNativeHook() { |
| 92 | try { |
| 93 | GlobalScreen.registerNativeHook(); |
| 94 | GlobalScreen.addNativeKeyListener(new NativeKeyAdapter() { |
| 95 | @Override |
| 96 | public void nativeKeyPressed(NativeKeyEvent e) { |
| 97 | System.out.println("key code = " + e.getKeyCode() + ", key text = '" + NativeKeyEvent.getKeyText(e.getKeyCode()) + "'"); |
| 98 | } |
| 99 | }); |
| 100 | GlobalScreen.addNativeMouseMotionListener(new NativeMouseMotionAdapter() { |
| 101 | @Override |
| 102 | public void nativeMouseMoved(NativeMouseEvent e) { |
| 103 | System.out.println("mouse point = " + e.getPoint()); |
| 104 | } |
| 105 | }); |
| 106 | } catch (NativeHookException e) { |
| 107 | e.printStackTrace(); |
| 108 | } |
| 109 | } |
| 110 | |
| 111 | public void turnOff() { |
| 112 | cleanup(); |
| 113 | ledController.onFrame(LedFrame.from(config).fillColor(0, 0, 0)); |
| 114 | } |
| 115 | } |