OptiTrackShotDetector.java

/*
 * ShootOFF - Software for Laser Dry Fire Training
 * Copyright (C) 2016 phrack
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.shootoff.camera.shotdetection;

import java.awt.Point;
import java.util.Timer;
import java.util.TimerTask;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.shootoff.camera.CameraManager;
import com.shootoff.camera.CameraView;
import com.shootoff.camera.cameratypes.Camera.CameraState;
import com.shootoff.camera.shot.ShotColor;
import com.shootoff.camera.cameratypes.OptiTrackCamera;

public class OptiTrackShotDetector extends ShotYieldingShotDetector implements CameraStateListener {
	private static final Logger logger = LoggerFactory.getLogger(OptiTrackShotDetector.class);

	private final CameraManager cameraManager;

	private final long startTime = System.currentTimeMillis();

	public OptiTrackShotDetector(final CameraManager cameraManager, final CameraView cameraView) {
		super(cameraManager, cameraView);

		this.cameraManager = cameraManager;

		cameraManager.registerCameraStateListener(this);
	}

	public static boolean isSystemSupported() {
		return OptiTrackCamera.initialized();
	}

	@Override
	public void cameraStateChange(CameraState state) {
		if (logger.isDebugEnabled()) logger.debug("got state change {}", state);
		switch (state) {
		case DETECTING_CALIBRATED:
			Timer timer = new Timer();
			timer.schedule(new TimerTask() {
				@Override
				public void run() {
					if (cameraManager.isDetecting()) enableDetection();
				}
			}, 100);
			break;
		default:
			disableDetection();
			break;

		}
	}

	private native void startDetectionModeNative();

	private native void enableDetection();

	private native void disableDetection();

	@Override
	public void setFrameSize(int width, int height) {}

	/**
	 * Called by the native code to notify this class when a shot is detected.
	 *
	 * @param x
	 *            the x coordinate of the new shot
	 * @param y
	 *            the y coordinate of the new shot
	 * @param timestamp
	 *            the timestamp of the shot not adjusted for the shot timer
	 * @param rgb
	 *            the rgb color of the new shot
	 */
	public void foundShot(int x, int y, long timestamp, int rgb) {
		if (!cameraManager.isDetecting()) return;

		final Point undist = cameraManager.undistortCoords(x, y);

		if (logger.isTraceEnabled()) logger.trace("Translation: {} {} to {}", x, y, undist);

		super.addShot(ShotColor.INFRARED, undist.x, undist.y, startTime + timestamp, true);
	}

	@Override
	public void initDetecting() {
		startDetectionModeNative();
	}

	@Override
	protected boolean handlesBounds() {
		return false;
	}
}