initial commit

[kaka/cakelight.git] / src / kaka / cakelight / Frame.java

package kaka.cakelight;

import org.opencv.core.CvType;
import org.opencv.core.Mat;
import org.opencv.core.Size;
import org.opencv.imgproc.Imgproc;

import static kaka.cakelight.Main.saveFile;
import static kaka.cakelight.Main.timeIt;

public class Frame {
    private byte[] data;
    private Configuration config;

    private Frame(byte[] data) {
        this.data = data;
    }

    public static Frame of(byte[] data, Configuration config) {
        Frame frame = new Frame(data);
        frame.config = config;
        frame.convert();
        return frame;
    }

    private void convert() {
        Mat src = new Mat(config.video.height, config.video.width, CvType.CV_8UC2); // 8-bit, unsigned, 2 channels
        src.put(0, 0, data);

        Mat converted = new Mat();
        Mat resized = new Mat();

        timeIt("total", () -> {
            timeIt("yuyv2rgb", () -> Imgproc.cvtColor(src, converted, Imgproc.COLOR_YUV2RGB_YUYV)); // 3.5 - 4.0 ms
            timeIt("resizing", () -> Imgproc.resize(converted, resized, new Size(config.leds.cols, config.leds.rows), 0, 0, Imgproc.INTER_AREA)); // INTER_AREA is the best for shrinking, but also the slowest (~1.5 ms)
        });
//        save(converted, "/home/kaka/test-converted.data");
//        save(resized, "/home/kaka/test-resized.data");
    }

    private void save(Mat mat, String filepath) {
        byte[] data = new byte[mat.cols() * mat.rows() * mat.channels()];
        mat.get(0, 0, data);
        saveFile(data, filepath);
    }

    public byte[] getData() {
        return data;
    }
}