Commit | Line | Data |
---|---|---|
67b0a758 TW |
1 | package kaka.cakelight.mode; |
2 | ||
3 | import kaka.cakelight.Configuration; | |
4 | import kaka.cakelight.FrameGrabber; | |
5 | import kaka.cakelight.VideoDeviceListener; | |
6 | import kaka.cakelight.VideoFrame; | |
4a2d6056 | 7 | |
03670958 | 8 | import java.io.File; |
4a2d6056 TW |
9 | import java.io.IOException; |
10 | import java.util.Optional; | |
100b82fe | 11 | import java.util.function.Consumer; |
4a2d6056 | 12 | |
d182b8cc | 13 | public class VideoMode extends Mode { |
4a2d6056 | 14 | private Configuration config; |
8418fbda | 15 | private Thread grabberThread; |
adc29b9a | 16 | private Consumer<VideoFrame> frameConsumer; |
03670958 | 17 | private VideoDeviceListener deviceListener; |
fa9808cd | 18 | private boolean isPaused = false; |
03670958 TW |
19 | |
20 | public VideoMode() { | |
21 | deviceListener = new VideoDeviceListener(); | |
d182b8cc | 22 | deviceListener.onVideoDeviceChange(this::onVideoDeviceChange); |
03670958 | 23 | } |
4a2d6056 TW |
24 | |
25 | @Override | |
26 | public void enter(Configuration config) { | |
27 | this.config = config; | |
03670958 | 28 | deviceListener.startListening(); |
4a2d6056 TW |
29 | } |
30 | ||
31 | @Override | |
d0afa6fb | 32 | public void pause() { |
fa9808cd | 33 | isPaused = true; |
d0afa6fb TW |
34 | } |
35 | ||
36 | @Override | |
37 | public void resume() { | |
fa9808cd | 38 | isPaused = false; |
d0afa6fb TW |
39 | grabberThread.notify(); |
40 | } | |
41 | ||
42 | @Override | |
4a2d6056 | 43 | public void exit() { |
8418fbda | 44 | grabberThread.interrupt(); |
03670958 | 45 | deviceListener.stopListening(); |
4a2d6056 TW |
46 | } |
47 | ||
03670958 | 48 | private void startGrabberThread(File videoDevice) { |
100b82fe | 49 | assert frameConsumer != null; |
8418fbda | 50 | grabberThread = new Thread() { |
4a2d6056 | 51 | public void run() { |
03670958 | 52 | try (FrameGrabber grabber = FrameGrabber.from(videoDevice, config)) { |
4a2d6056 | 53 | while (!isInterrupted()) { |
adc29b9a | 54 | Optional<VideoFrame> frame = grabber.grabFrame(); |
fa9808cd TW |
55 | if (isPaused) { |
56 | wait(); | |
57 | } | |
03b67a73 | 58 | if (frameConsumer != null) frame.ifPresent(frameConsumer); |
adc29b9a | 59 | frame.ifPresent(VideoMode.this::onVideoFrame); |
100b82fe | 60 | // timeIt("frame", grabber::grabFrame); |
4a2d6056 | 61 | } |
fa9808cd | 62 | } catch (IOException | InterruptedException e) { |
4a2d6056 TW |
63 | e.printStackTrace(); |
64 | } | |
65 | } | |
66 | }; | |
8418fbda | 67 | grabberThread.start(); |
4a2d6056 | 68 | } |
100b82fe | 69 | |
adc29b9a | 70 | public void onVideoFrame(Consumer<VideoFrame> consumer) { |
100b82fe TW |
71 | frameConsumer = consumer; |
72 | } | |
03670958 | 73 | |
adc29b9a | 74 | private void onVideoFrame(VideoFrame frame) { |
6b569670 | 75 | updateWithFrame(frame.getLedFrame()); |
03b67a73 TW |
76 | } |
77 | ||
d182b8cc | 78 | public void onVideoDeviceChange(Optional<File> videoDevice) { |
03670958 | 79 | // Should only happen when this mode is active! |
8418fbda TW |
80 | if (grabberThread != null) { |
81 | grabberThread.interrupt(); | |
03670958 TW |
82 | } |
83 | videoDevice.ifPresent(this::startGrabberThread); | |
84 | } | |
4a2d6056 | 85 | } |