Commit | Line | Data |
---|---|---|
100b82fe TW |
1 | package kaka.cakelight; |
2 | ||
3 | import javafx.application.Application; | |
4 | import javafx.scene.Scene; | |
5 | import javafx.scene.canvas.Canvas; | |
6 | import javafx.scene.canvas.GraphicsContext; | |
7 | import javafx.scene.input.KeyCode; | |
8 | import javafx.scene.layout.Pane; | |
9 | import javafx.scene.paint.Color; | |
10 | import javafx.scene.paint.Paint; | |
11 | import javafx.stage.Stage; | |
12 | import org.opencv.core.Core; | |
13 | import org.opencv.core.Mat; | |
14 | ||
15 | import static kaka.cakelight.Main.log; | |
16 | ||
17 | public class GuiTest extends Application { | |
18 | private static final int BLOCK = 45; | |
19 | private static final int GUTTER = 3 * BLOCK; | |
20 | private Canvas canvas; | |
21 | private Configuration config; | |
22 | private CakeLight cakelight; | |
23 | private boolean paused; | |
24 | ||
25 | public static void main(String[] args) { | |
26 | System.loadLibrary(Core.NATIVE_LIBRARY_NAME); | |
27 | launch(args); | |
28 | } | |
29 | ||
30 | @Override | |
31 | public void start(Stage stage) throws Exception { | |
32 | config = Configuration.from("config.properties"); | |
33 | canvas = new Canvas(16 * BLOCK + 2 * GUTTER, 9 * BLOCK + 2 * GUTTER); | |
34 | ||
35 | Pane root = new Pane(); | |
36 | root.getChildren().add(canvas); | |
37 | ||
38 | Scene scene = new Scene(root); | |
39 | scene.setOnKeyPressed(keyEvent -> { | |
40 | if (keyEvent.getCode() == KeyCode.ESCAPE) { | |
41 | stage.close(); | |
42 | cakelight.cleanup(); | |
43 | } | |
44 | if (keyEvent.getCode() == KeyCode.SPACE) { | |
45 | paused = !paused; | |
46 | } | |
47 | }); | |
48 | ||
49 | stage.setTitle("Cakelight"); | |
50 | stage.setScene(scene); | |
51 | stage.setOnCloseRequest(windowEvent -> cakelight.cleanup()); | |
52 | stage.show(); | |
53 | ||
54 | setupCakeLight(); | |
55 | } | |
56 | ||
57 | private void setupCakeLight() { | |
58 | log("Running with config:\n" + config); | |
01ee91e7 | 59 | cakelight = new CakeLight(config); |
100b82fe TW |
60 | VideoMode mode = new VideoMode(); |
61 | cakelight.setMode(mode); | |
62 | cakelight.startLoop(); | |
03b67a73 | 63 | mode.onVideoFrame(frame -> drawFrame(canvas.getGraphicsContext2D(), frame)); |
100b82fe TW |
64 | } |
65 | ||
66 | private void drawFrame(GraphicsContext gc, Frame frame) { | |
67 | if (paused) return; | |
68 | System.out.println("Drawing a frame"); | |
69 | drawCols(gc, frame); | |
70 | drawRows(gc, frame); | |
71 | // drawVideo(gc, frame); | |
72 | drawBorderAndGrid(gc); | |
100b82fe TW |
73 | } |
74 | ||
03b67a73 | 75 | private void drawLEDs(GraphicsContext gc, LedFrame frame) { |
100b82fe TW |
76 | int ledLength = GUTTER; |
77 | float colSize = 16f * BLOCK / config.leds.cols; | |
78 | float rowSize = 9f * BLOCK / config.leds.rows; | |
79 | // DropShadow shadow = new DropShadow(BlurType.ONE_PASS_BOX, Color.RED, colSize * 2, colSize, 0, 0); | |
80 | for (int x = 0; x < config.leds.cols; x++) { | |
03b67a73 | 81 | gc.setFill(frame.getLedColor(x + config.leds.rows)); |
100b82fe | 82 | gc.fillRect(GUTTER + x * colSize, GUTTER - ledLength, colSize, ledLength); |
03b67a73 | 83 | gc.setFill(frame.getLedColor(config.leds.rows * 2 + config.leds.cols * 2 - 1 - x)); |
100b82fe TW |
84 | gc.fillRect(GUTTER + x * colSize, GUTTER + 9 * BLOCK, colSize, ledLength); |
85 | } | |
86 | for (int y = 0; y < config.leds.rows; y++) { | |
03b67a73 | 87 | gc.setFill(frame.getLedColor(config.leds.rows - 1 - y)); |
100b82fe | 88 | gc.fillRect(GUTTER - ledLength, GUTTER + y * rowSize, ledLength, rowSize); |
03b67a73 | 89 | gc.setFill(frame.getLedColor(y + config.leds.rows + config.leds.cols)); |
100b82fe TW |
90 | gc.fillRect(GUTTER + 16 * BLOCK, GUTTER + y * rowSize, ledLength, rowSize); |
91 | } | |
92 | } | |
93 | ||
94 | private void drawVideo(GraphicsContext gc, Frame frame) { | |
95 | byte[] rgb = new byte[3]; | |
96 | Mat img = frame.getConvertedImage(); | |
97 | float colSize = 16 * BLOCK / (float)img.cols(); | |
98 | float rowSize = 9 * BLOCK / (float)img.rows(); | |
99 | for (int x = 0, cols = img.cols(); x < cols; x++) { | |
100 | for (int y = 0, rows = img.rows(); y < rows; y++) { | |
101 | img.get(y, x, rgb); | |
102 | gc.setFill(Color.rgb(rgb[0] & 0xff, rgb[1] & 0xff, rgb[2] & 0xff, 0.5)); | |
103 | gc.fillRect(GUTTER + x * colSize, GUTTER + y * rowSize, colSize, rowSize); | |
104 | } | |
105 | } | |
106 | } | |
107 | ||
108 | private void drawCols(GraphicsContext gc, Frame frame) { | |
109 | byte[] rgb = new byte[3]; | |
110 | for (int x = 0; x < config.leds.cols; x++) { | |
111 | for (int y = 0; y < 9; y++) { | |
112 | frame.getColImage().get(y, x, rgb); | |
113 | drawColPixel(gc, x, y, rgb); | |
114 | } | |
115 | } | |
116 | } | |
117 | ||
118 | private void drawRows(GraphicsContext gc, Frame frame) { | |
119 | byte[] rgb = new byte[3]; | |
120 | for (int y = 0; y < config.leds.rows; y++) { | |
121 | for (int x = 0; x < 16; x++) { | |
122 | frame.getRowImage().get(y, x, rgb); | |
123 | drawRowPixel(gc, x, y, rgb); | |
124 | } | |
125 | } | |
126 | } | |
127 | ||
128 | private void drawColPixel(GraphicsContext gc, int x, int y, byte[] rgb) { | |
129 | float ledSize = 16f * BLOCK / config.leds.cols; | |
130 | gc.setFill(Color.rgb(rgb[0] & 0xff, rgb[1] & 0xff, rgb[2] & 0xff, 0.5)); | |
131 | gc.fillRect(GUTTER + x * ledSize, GUTTER + y * BLOCK, ledSize, BLOCK); | |
132 | } | |
133 | ||
134 | private void drawRowPixel(GraphicsContext gc, int x, int y, byte[] rgb) { | |
135 | float ledSize = 9f * BLOCK / config.leds.rows; | |
136 | gc.setFill(Color.rgb(rgb[0] & 0xff, rgb[1] & 0xff, rgb[2] & 0xff, 0.5)); | |
137 | gc.fillRect(GUTTER + x * BLOCK, GUTTER + y * ledSize, BLOCK, ledSize); | |
138 | } | |
139 | ||
140 | private void drawBorderAndGrid(GraphicsContext gc) { | |
141 | gc.setStroke(Color.BLACK); | |
142 | gc.setLineWidth(BLOCK * 0.1); | |
143 | gc.strokeRect(GUTTER, GUTTER, 16 * BLOCK, 9 * BLOCK); | |
144 | gc.setLineWidth(1); | |
145 | for (int x = 1; x < 16; x++) { | |
146 | gc.strokeLine(GUTTER + x * BLOCK, GUTTER, GUTTER + x * BLOCK, GUTTER + 9 * BLOCK); | |
147 | } | |
148 | for (int y = 1; y < 9; y++) { | |
149 | gc.strokeLine(GUTTER, GUTTER + y * BLOCK, GUTTER + 16 * BLOCK, GUTTER + y * BLOCK); | |
150 | } | |
151 | } | |
152 | ||
153 | private void drawPixel(GraphicsContext gc, int x, int y, Paint paint) { | |
154 | gc.setFill(paint); | |
155 | gc.fillRect(GUTTER + x * BLOCK, GUTTER + y * BLOCK, BLOCK, BLOCK); | |
156 | } | |
157 | } |