TestTargetIO.java

/*
 * Copyright (C) 2016 phrack. All rights reserved.
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

package com.shootoff.targets.io;

import static org.junit.Assert.*;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;

import javafx.scene.Node;
import javafx.scene.paint.Color;

import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

import com.shootoff.camera.CamerasSupervisor;
import com.shootoff.config.Configuration;
import com.shootoff.config.ConfigurationException;
import com.shootoff.gui.JavaFXThreadingRule;
import com.shootoff.gui.MockCanvasManager;
import com.shootoff.gui.controller.MockProjectorArenaController;
import com.shootoff.gui.controller.ShootOFFController;
import com.shootoff.plugins.ProjectorTrainingExerciseBase;
import com.shootoff.targets.EllipseRegion;
import com.shootoff.targets.ImageRegion;
import com.shootoff.targets.PolygonRegion;
import com.shootoff.targets.RectangleRegion;
import com.shootoff.targets.Target;
import com.shootoff.targets.TargetRegion;
import com.shootoff.targets.io.TargetIO.TargetComponents;

public class TestTargetIO {
	@Rule public JavaFXThreadingRule javafxRule = new JavaFXThreadingRule();

	private Map<String, String> targetTags = new HashMap<>();
	private List<Node> regions = new ArrayList<>();
	private ImageRegion img;
	private RectangleRegion rec;
	private EllipseRegion ell;
	private PolygonRegion pol;
	private File tempXMLTarget;

	@Before
	public void setUp() throws FileNotFoundException {
		System.setProperty("shootoff.home", System.getProperty("user.dir"));

		Map<String, String> imgTags = new HashMap<>();
		imgTags.put("1", "2");
		img = new ImageRegion(6, 6, new File("targets" + File.separator + "circle-plate.gif"));
		img.setTags(imgTags);

		Map<String, String> recTags = new HashMap<>();
		recTags.put("a", "b");
		recTags.put("c", "d");
		rec = new RectangleRegion(10, 40, 20, 90);
		rec.setFill(Color.ORANGE);
		rec.setTags(recTags);

		Map<String, String> ellTags = new HashMap<>();
		ellTags.put("name", "value");
		ell = new EllipseRegion(0, 20, 5, 5);
		ell.setFill(Color.RED);
		ell.setTags(ellTags);

		Map<String, String> polTags = new HashMap<>();
		polTags.put("points", "3");
		pol = new PolygonRegion(300, 0, 400, 30, 300, 100);
		pol.setTags(polTags);

		regions.add(img);
		regions.add(rec);
		regions.add(ell);
		regions.add(pol);

		targetTags.put("A", "B");
		targetTags.put("C", "D");
		
		tempXMLTarget = new File("temp.target");
		TargetIO.saveTarget(targetTags, regions, tempXMLTarget);
	}

	@After
	public void tearDown() {
		if (!tempXMLTarget.delete()) System.err.println("Failed to delete " + tempXMLTarget.getPath());
	}

	private void checkTarget(TargetComponents targetComponents) {
		for (Node node : targetComponents.getTargetGroup().getChildren()) {
			TargetRegion region = (TargetRegion) node;

			switch (region.getType()) {
			case IMAGE:
				ImageRegion img = (ImageRegion) region;
				assertEquals(this.img.getX(), img.getX(), 0.5);
				assertEquals(this.img.getY(), img.getY(), 0.5);
				assertEquals(this.img.getImageFile().getAbsolutePath(), img.getImageFile().getAbsolutePath());
				assertEquals("2", this.img.getTag("1"));
				break;
			case RECTANGLE:
				RectangleRegion rec = (RectangleRegion) region;
				assertEquals(this.rec.getX(), rec.getX(), 0.5);
				assertEquals(this.rec.getY(), rec.getY(), 0.5);
				assertEquals(this.rec.getWidth(), rec.getWidth(), 0.5);
				assertEquals(this.rec.getHeight(), rec.getHeight(), 0.5);
				assertEquals(this.rec.getFill(), rec.getFill());
				assertEquals("b", this.rec.getTag("a"));
				assertEquals("d", this.rec.getTag("c"));
				break;
			case ELLIPSE:
				EllipseRegion ell = (EllipseRegion) region;
				assertEquals(this.ell.getCenterX(), ell.getCenterX(), 0.5);
				assertEquals(this.ell.getCenterY(), ell.getCenterY(), 0.5);
				assertEquals(this.ell.getRadiusX(), ell.getRadiusX(), 0.5);
				assertEquals(this.ell.getRadiusY(), ell.getRadiusY(), 0.5);
				assertEquals(this.ell.getFill(), ell.getFill());
				break;
			case POLYGON:
				PolygonRegion pol = (PolygonRegion) region;
				assertEquals(this.pol.getPoints(), pol.getPoints());
				assertEquals(this.pol.getFill(), pol.getFill());
				break;
			}
		}
		
		for (Entry<String, String> entry : targetComponents.getTargetTags().entrySet()) {
			assertTrue(targetTags.containsKey(entry.getKey()));
			assertEquals(targetTags.get(entry.getKey()), entry.getValue());
		}
	}

	@Test
	public void testXMLSerializationFile() {
		Optional<TargetComponents> targetComponents = TargetIO.loadTarget(tempXMLTarget);

		assertTrue(targetComponents.isPresent());

		TargetComponents tc = targetComponents.get();

		assertEquals(4, tc.getTargetGroup().getChildren().size());

		checkTarget(tc);
	}

	@Test
	public void testXMLSerializationStream() throws FileNotFoundException {
		Optional<TargetComponents> targetComponents = TargetIO.loadTarget(new FileInputStream(tempXMLTarget), null);

		assertTrue(targetComponents.isPresent());

		TargetComponents tc = targetComponents.get();

		assertEquals(4, tc.getTargetGroup().getChildren().size());

		checkTarget(tc);
	}

	@Test(expected = AssertionError.class)
	public void testXMLSerializationExerciseStream() throws FileNotFoundException, ConfigurationException {
		Configuration config = new Configuration(new String[0]);

		CamerasSupervisor cs = new CamerasSupervisor(config);

		MockProjectorArenaController pac = new MockProjectorArenaController(config, new MockCanvasManager(config));

		ProjectorTrainingExerciseBase pteb = new ProjectorTrainingExerciseBase(new ArrayList<Target>());
		pteb.init(cs, new ShootOFFController(), pac);

		pteb.addTarget(new File("@" + tempXMLTarget.getName()), 0, 0);
	}
}