Added a custom Color class
[kaka/cakelight.git] / src / kaka / cakelight / FrameGrabber.java
1 package kaka.cakelight;
2
3 import java.io.*;
4 import java.util.Optional;
5
6 import static kaka.cakelight.Main.log;
7
8 public class FrameGrabber implements Closeable {
9     private Configuration config;
10     private File file;
11     private int bytesPerFrame;
12     private InputStream fileStream;
13
14     private FrameGrabber() {
15     }
16
17     public static FrameGrabber from(File videoDevice, Configuration config) {
18         FrameGrabber fg = new FrameGrabber();
19         fg.config = config;
20         fg.file = videoDevice;
21         fg.bytesPerFrame = config.video.width * config.video.height * config.video.bpp;
22         fg.prepare();
23         return fg;
24     }
25
26     private boolean prepare() {
27         try {
28             fileStream = new FileInputStream(file);
29             return true;
30         } catch (FileNotFoundException e) {
31             e.printStackTrace();
32             return false;
33         }
34     }
35
36     /**
37      * Must be run in the same thread as {@link #prepare}.
38      */
39     public Optional<Frame> grabFrame() {
40         try {
41             byte[] data = new byte[bytesPerFrame];
42             int count = fileStream.read(data);
43             log("# of bytes read = " + count);
44             return Optional.of(Frame.of(data, config));
45         } catch (IOException e) {
46             e.printStackTrace();
47         }
48
49         return Optional.empty();
50     }
51
52     @Override
53     public void close() throws IOException {
54         fileStream.close();
55     }
56 }