fact.features.TimeSpread Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fact-tools Show documentation

The FACT telescope is a Cherenkov telescope on the island of La Palma. This set of tools is dedicated to read, parse and process raw data produced by the FACT telescope.

The newest version!

package fact.features;

import fact.Utils;
import stream.Data;
import stream.Processor;
import stream.annotations.Parameter;

public class TimeSpread implements Processor {
	
	@Parameter(required = true)
	private String arrivalTimeKey = null;
	@Parameter(required = true)
	private String weightsKey = null;
	@Parameter(required = true)
	private String showerKey = null;
	@Parameter(required = true)
	private String outputKey = null;
	
	private double[] arrivalTime = null;
	private double[] weights = null;
	private int[] shower = null;

	@Override
	public Data process(Data input) {
		
		Utils.mapContainsKeys( input, arrivalTimeKey, weightsKey, showerKey);
		
		arrivalTime = (double[]) input.get(arrivalTimeKey);
		weights = (double[]) input.get(weightsKey);
		shower = (int[]) input.get(showerKey);
		
		// NumberShowerPixel
		int n = shower.length;
		
		// Times of shower pixel
		double[] t = new double[n];
		// Weights of shower pixel
		double[] w = new double[n];
		for (int i = 0 ; i < n ; i++)
		{
			int chid = shower[i];
			t[i] = arrivalTime[chid];
			w[i] = weights[chid];
		}
		
		// Sum over the time array:
		double sumt = Utils.arraySum(t);
		// Sum over the weight array:
		double sumw = Utils.arraySum(w);
		// Sum over the weighted time array:
		double sumwt = Utils.arraySum(Utils.arrayMultiplication(w, t));
		// Sum over the element wise multiplication of t and t:
		double sumtt = Utils.arraySum(Utils.arrayMultiplication(t, t));
		// Sum over the element wise multiplication of t and t, weighted:
		double sumwtt = Utils.arraySum(Utils.arrayMultiplication(w, Utils.arrayMultiplication(t, t)));
		
		
		double timespread = Math.sqrt( sumtt / n - Math.pow(sumt/n,2));
		double weightedTimespread = Math.sqrt( sumwtt / sumw - Math.pow(sumwt/sumw,2));
		
		input.put(outputKey, timespread);
		input.put(outputKey+"_weighted", weightedTimespread);
		
		return input;
	}

	public String getArrivalTimeKey() {
		return arrivalTimeKey;
	}

	public void setArrivalTimeKey(String arrivalTimeKey) {
		this.arrivalTimeKey = arrivalTimeKey;
	}

	public String getWeightsKey() {
		return weightsKey;
	}

	public void setWeightsKey(String weightsKey) {
		this.weightsKey = weightsKey;
	}

	public String getShowerKey() {
		return showerKey;
	}

	public void setShowerKey(String showerKey) {
		this.showerKey = showerKey;
	}

	public String getOutputKey() {
		return outputKey;
	}

	public void setOutputKey(String outputKey) {
		this.outputKey = outputKey;
	}

}