ReflectivityCalculatorTester.java

/*******************************************************************************
 * Copyright (c) 2013, 2014 UT-Battelle, LLC.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Initial API and implementation and/or initial documentation - 
 *   Jay Jay Billings, John Ankner
 *******************************************************************************/
package org.eclipse.ice.reflectivity.test;

import static org.junit.Assert.*;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspaceRoot;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.ice.datastructures.ICEObject.ListComponent;
import org.eclipse.ice.datastructures.form.Form;
import org.eclipse.ice.io.csv.CSVReader;
import org.eclipse.ice.reflectivity.ReflectivityCalculator;
import org.eclipse.ice.reflectivity.Tile;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * This class tests {@link org.eclipse.ice.reflectivity.ReflectivityCalculator}.
 * 
 * @author Jay Jay Billings, John Ankner
 *
 */
public class ReflectivityCalculatorTester {

	/**
	 * The CSV file that will be tested
	 */
	private static IFile testFile;
	/**
	 * The reader that will be tested.
	 */
	private static CSVReader reader;

	/**
	 * The project used by the test. It currently points to the
	 * CSVLoaderTesterWorkspace because the reflectivity CSV files are also used
	 * in the CSV tests.
	 */
	private static IProject project;

	/**
	 * This class loads the files for the test.
	 * 
	 * @throws java.lang.Exception
	 */
	@BeforeClass
	public static void setUpBeforeClass() throws Exception {

		// Get the file separator used on this system, which is different across
		// OSes.
		String separator = System.getProperty("file.separator");
		// Create the path for the reflectivity file in the ICE tests directory
		String userHome = System.getProperty("user.home");
		IWorkspaceRoot workspaceRoot = ResourcesPlugin.getWorkspace().getRoot();
		String projectName = "CSVLoaderTesterWorkspace";
		String filename = "getSpecRefSqrdMod_q841.csv";
		IPath projectPath = new Path(userHome + separator + "ICETests"
				+ separator + projectName + separator + ".project");

		// Setup the project
		try {
			// Create the project description
			IProjectDescription desc = ResourcesPlugin.getWorkspace()
					.loadProjectDescription(projectPath);
			// Get the project handle and create it
			project = workspaceRoot.getProject(desc.getName());
			project.create(desc, new NullProgressMonitor());
			// Open the project if it is not already open
			if (project.exists() && !project.isOpen()) {
				project.open(new NullProgressMonitor());
			}
			// Refresh the workspace
			project.refreshLocal(IResource.DEPTH_INFINITE,
					new NullProgressMonitor());
			// Create the IFile handle for the csv file
			testFile = project.getFile(filename);
		} catch (CoreException e) {
			// Catch exception for creating the project
			e.printStackTrace();
			fail();
		}

		// Create the reader
		reader = new CSVReader();

		return;

	}

	/**
	 * This class tests
	 * {@link ReflectivityCalculator#getExtensionLengths(double[], double, double, double, int, int, int)}
	 * .
	 */
	@Test
	public void testGetExtensionLengths() {
		// Load the convolution file
		Form form = reader.read(project.getFile("getExtensionLengths.csv"));
		ListComponent<String[]> lines = (ListComponent<String[]>) form
				.getComponent(1);
		assertEquals(403, lines.size());

		// Get the parameters and reference output values
		String line[] = lines.get(0);
		double delQ0 = Double.valueOf(line[0]);
		double delQ1oQ = Double.valueOf(line[1]);
		double wavelength = Double.valueOf(line[2]);
		int numPoints = Integer.valueOf(line[3]);
		int refNumLowPoints = Integer.valueOf(line[4]);
		int refNumHighPoints = Integer.valueOf(line[5]);

		// Load the q array. They are padded by numLowPoints for the
		// convolution.
		double[] waveVector = new double[lines.size() - 1];
		for (int i = 0; i < lines.size() - 1; i++) {
			line = lines.get(i + 1);
			waveVector[i] = Double.valueOf(line[0]);
		}
		assertEquals(numPoints, waveVector.length);
		assertEquals(4.401373320E-01, waveVector[numPoints - 1], 0.0);

		// Call the function
		int numLowPoints = 0, numHighPoints = 0;
		ReflectivityCalculator calculator = new ReflectivityCalculator();
		numLowPoints = calculator.getLowExtensionLength(waveVector, delQ0, delQ1oQ,
				numPoints);
		numHighPoints = calculator.getHighExtensionLength(waveVector, delQ0, delQ1oQ,
				numPoints);

		// Check the high and low extension lengths
		assertEquals(refNumLowPoints, numLowPoints);
		assertEquals(refNumHighPoints, numHighPoints);

		return;
	}

	/**
	 * This class tests
	 * {@link ReflectivityCalculator#convolute(double[], double, double, double, int, int, double[])}
	 * .
	 */
	@Test
	public void testConvolute() {
		// Load the convolution file
		Form form = reader.read(project.getFile("convolute.csv"));
		ListComponent<String[]> lines = (ListComponent<String[]>) form
				.getComponent(1);
		assertEquals(417, lines.size());

		// The first line of the file contains all of the parameters
		String[] line = lines.get(0);
		double delQ0 = Double.valueOf(line[0]);
		double delQ1oQ = Double.valueOf(line[1]);
		double wavelength = Double.valueOf(line[2]);
		int numPoints = Integer.valueOf(line[3]);
		int numLowPoints = Integer.valueOf(line[4]);
		int numHighPoints = Integer.valueOf(line[5]);

		// Load the q and refFit arrays. They are padded by numLowPoints for the
		// convolution.
		double[] waveVector = new double[lines.size() - 1];
		double[] refRefFit = new double[lines.size() - 1]; // Reference!
		for (int i = 0; i < lines.size() - 1; i++) {
			line = lines.get(i + 1);
			waveVector[i] = Double.valueOf(line[0]);
			refRefFit[i] = Double.valueOf(line[1]);
		}
		assertEquals(0.461926308814, waveVector[415], 0.0);
		assertEquals(3.87921357905325E-09, refRefFit[415], 0.0);

		// Load the tiles
		form = reader.read(project.getFile("getSpecRefSqrdMod_q841.csv"));
		ListComponent<String[]> tileLines = (ListComponent<String[]>) form
				.getComponent(1);
		Tile[] tiles = loadTiles(tileLines);
		assertEquals(173, tiles.length);

		// Compute the initial value of refFit
		double[] refFit = new double[lines.size() - 1];
		double qEff = 0.0;
		ReflectivityCalculator calculator = new ReflectivityCalculator();
		for (int i = 0; i < numHighPoints + numLowPoints + numPoints; i++) {
			if (waveVector[i] < 1.0e-10) {
				qEff = 1.0e-10;
			} else {
				qEff = waveVector[i];
			}
			refFit[i] = calculator.getModSqrdSpecRef(qEff, wavelength, tiles);
		}

		// Do the convolution and check the result
		calculator.convolute(waveVector, delQ0, delQ1oQ, wavelength, numPoints,
				numLowPoints, numHighPoints, refFit);
		for (int i = 0; i < refFit.length; i++) {
			assertEquals(refRefFit[i], refFit[i],
					Math.abs(refRefFit[i]) / 1.0e-4);
		}

		return;
	}

	/**
	 * This class tests
	 * {@link ReflectivityCalculator#getModSqrdSpecRef(double, double, Tile[])}.
	 */
	@Test
	public void testGetSpecRefSqrdMod() {

		// Load the file
		Form form = reader.read(project.getFile("getSpecRefSqrdMod_q841.csv"));
		ListComponent<String[]> lines = (ListComponent<String[]>) form
				.getComponent(1);

		// Get the two single parameters and the final result out of the data
		// for the first test case
		String[] line = lines.get(0);
		double waveVectorQ, wavelength, expectedSpecRefSqrd;
		waveVectorQ = Double.valueOf(line[0]);
		wavelength = Double.valueOf(line[1]);
		expectedSpecRefSqrd = Double.valueOf(line[2]);

		// Load the tiles from the rest of the data
		Tile[] tiles = loadTiles(lines);
		assertEquals(173, tiles.length);

		// Get the squared modulus of the specular reflectivity with the values
		// from the file for the first case.
		ReflectivityCalculator calculator = new ReflectivityCalculator();
		double specRefSqrd = calculator.getModSqrdSpecRef(waveVectorQ,
				wavelength, tiles);
		System.out.println(specRefSqrd + " " + expectedSpecRefSqrd);
		System.out.println("RERR = " + (specRefSqrd - expectedSpecRefSqrd)
				/ expectedSpecRefSqrd);
		assertEquals(expectedSpecRefSqrd, specRefSqrd,
				Math.abs(expectedSpecRefSqrd) / 1.0e-4);

		// Get the two single parameters and the final result out of the data
		// for the second test case
		line = lines.get(1);
		waveVectorQ = Double.valueOf(line[0]);
		wavelength = Double.valueOf(line[1]);
		expectedSpecRefSqrd = Double.valueOf(line[2]);
		System.out.println(waveVectorQ + " " + wavelength + " "
				+ expectedSpecRefSqrd);

		// Get the squared modulus of the specular reflectivity with the values
		// from the file for the second case.
		specRefSqrd = calculator.getModSqrdSpecRef(waveVectorQ, wavelength,
				tiles);
		System.out.println(specRefSqrd + " " + expectedSpecRefSqrd);
		System.out.println("RERR = " + (specRefSqrd - expectedSpecRefSqrd)
				/ expectedSpecRefSqrd);
		assertEquals(expectedSpecRefSqrd, specRefSqrd,
				Math.abs(expectedSpecRefSqrd / 1.0e-4));

		return;
	}

	/**
	 * This operation loads the set of Tiles from the reference file, ignoring
	 * the first and second lines that store the reference values.
	 * 
	 * @param lines
	 *            The ListComponent with the lines from the file
	 * @return the list of Tiles from the file
	 */
	private Tile[] loadTiles(ListComponent<String[]> lines) {

		// Load all of the tiles
		Tile[] tiles = new Tile[lines.size() - 2];
		for (int i = 2; i < lines.size(); i++) {
			// Load the line
			String[] line = lines.get(i);
			// Create the tile and load the data from the line
			Tile tile = new Tile();
			tile.scatteringLength = Double.valueOf(line[0]);
			tile.trueAbsLength = Double.valueOf(line[1]);
			tile.incAbsLength = Double.valueOf(line[2]);
			tile.thickness = Double.valueOf(line[3]);
			// Load the tile into the array
			tiles[i - 2] = tile;
		}

		return tiles;
	}

}