src/kaka/cakelight/VideoMode.java

   1 package kaka.cakelight;
   2
   3 import java.io.File;
   4 import java.io.IOException;
   5 import java.util.Optional;
   6 import java.util.function.Consumer;
   7
   8 public class VideoMode extends Mode {
   9     private Configuration config;
  10     private Thread thread;
  11     private Consumer<Frame> frameConsumer;
  12     private VideoDeviceListener deviceListener;
  13
  14     public VideoMode() {
  15         deviceListener = new VideoDeviceListener();
  16         deviceListener.onVideoDeviceChange(this::onVideoDeviceChange);
  17     }
  18
  19     @Override
  20     public void enter(Configuration config) {
  21         this.config = config;
  22         deviceListener.startListening();
  23     }
  24
  25     @Override
  26     public void exit() {
  27         thread.interrupt();
  28         deviceListener.stopListening();
  29     }
  30
  31     private void startGrabberThread(File videoDevice) {
  32         assert frameConsumer != null;
  33         thread = new Thread() {
  34             public void run() {
  35                 try (FrameGrabber grabber = FrameGrabber.from(videoDevice, config)) {
  36                     while (!isInterrupted()) {
  37                         Optional<Frame> frame = grabber.grabFrame();
  38                         if (frameConsumer != null) frame.ifPresent(frameConsumer);
  39                         frame.ifPresent(VideoMode.this::onFrame);
  40 //                        timeIt("frame", grabber::grabFrame);
  41                         // TODO: process frame
  42                         // TODO: save where the LedController can access it
  43                     }
  44                 } catch (IOException e) {
  45                     e.printStackTrace();
  46                 }
  47             }
  48         };
  49         thread.start();
  50     }
  51
  52     public void onVideoFrame(Consumer<Frame> consumer) {
  53         frameConsumer = consumer;
  54     }
  55
  56     private void onFrame(Frame frame) {
  57         assert frameListener != null;
  58         frameListener.accept(frame.getLedFrame());
  59     }
  60
  61     public void onVideoDeviceChange(Optional<File> videoDevice) {
  62         // Should only happen when this mode is active!
  63         if (thread != null) {
  64             thread.interrupt();
  65         }
  66         videoDevice.ifPresent(this::startGrabberThread);
  67     }
  68 }