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 grabberThread;
  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         grabberThread.interrupt();
  28         deviceListener.stopListening();
  29     }
  30
  31     private void startGrabberThread(File videoDevice) {
  32         assert frameConsumer != null;
  33         grabberThread = 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                     }
  42                 } catch (IOException e) {
  43                     e.printStackTrace();
  44                 }
  45             }
  46         };
  47         grabberThread.start();
  48     }
  49
  50     public void onVideoFrame(Consumer<Frame> consumer) {
  51         frameConsumer = consumer;
  52     }
  53
  54     private void onFrame(Frame frame) {
  55         updateWithFrame(frame.getLedFrame());
  56     }
  57
  58     public void onVideoDeviceChange(Optional<File> videoDevice) {
  59         // Should only happen when this mode is active!
  60         if (grabberThread != null) {
  61             grabberThread.interrupt();
  62         }
  63         videoDevice.ifPresent(this::startGrabberThread);
  64     }
  65 }