Run in video mode unless arguments are supplied

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

package kaka.cakelight;

import org.opencv.core.Core;

import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;

public class Main {
    public static void main(String[] args) {
	System.loadLibrary(Core.NATIVE_LIBRARY_NAME);

	Configuration config = Configuration.from("config.properties");
	log("Running with config:\n" + config);

	CakeLight cakelight = new CakeLight(config, new LedController(config));
	if (args.length > 0) {
	    cakelight.setMode(new AmbientMode(args));
	} else {
	    cakelight.setMode(new VideoMode());
	}
	cakelight.startLoop();
	Runtime.getRuntime().addShutdownHook(new Thread(Main::printTimeStats));
    }

    public static void saveFile(byte[] data, String filepath) {
	try {
	    FileOutputStream fos = new FileOutputStream(filepath);
	    fos.write(data);
	    fos.close();
	} catch (IOException e) {
	    e.printStackTrace();
	}
    }

    public static void log(String msg, Object... args) {
	System.out.println(String.format(msg, args));
    }

    private static HashMap<String, Double> timeDurations = new HashMap<>();
    private static HashMap<String, Integer> timeCounts = new HashMap<>();
    public static double timeIt(String tag, Runnable lambda) {
	long start = System.nanoTime();
	lambda.run();
	long end = System.nanoTime();
	double duration = (end - start) * 0.000001;
//	log("duration (ms): " + tag + " = " + duration);

	if (!timeDurations.containsKey(tag)) {
	    timeDurations.put(tag, 0.0);
	    timeCounts.put(tag, 0);
	}
	timeDurations.put(tag, timeDurations.get(tag) + duration);
	timeCounts.put(tag, timeCounts.get(tag) + 1);
	return duration;
    }

    private static void printTimeStats() {
        log("Average times in ms:");
        timeDurations.forEach((tag, duration) -> {
	    log("%s: %s", tag, duration / timeCounts.get(tag));
	});
    }
}

/*
FrameGrabber läser frames asynkront
skickar frame till FrameConverter
sparas i huvudklassen
läses av FrameProcessor/CakeLight
 */