fact.filter.DrsTimeCalibration 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.filter;

import fact.Utils;
import fact.io.FitsStream;

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

import stream.Data;
import stream.Processor;
import stream.annotations.Parameter;
import stream.io.SourceURL;

import java.net.URL;


public class DrsTimeCalibration implements Processor {
	static Logger log = LoggerFactory.getLogger(DrsCalibration.class);
	
	@Parameter(required=false,description="Key of the StartCellData in the data fits file",defaultValue="StartCellData")
	private String startCellKey = "StartCellData";
	@Parameter(required=true,description="Key of the time calibration constants (relative to the start cell of each pixel)")
	private String outputKey = null;
	
	private int numberOfSlices = 1024;
	private int numberOfTimemarker = 160;
	
	private int npix;

	private String drsTimeKey = "SamplingTimeDeviation";
	
	Data drsTimeData = null;
	private double[] absoluteTimeOffsets = new double[numberOfSlices*numberOfTimemarker];
	
	@Override
	public Data process(Data input) {		
		Utils.isKeyValid(input, "NPIX", Integer.class);
		Utils.mapContainsKeys(input, startCellKey, "NROI");
		
		npix = (Integer) input.get("NPIX");
		int roi = (Integer) input.get("NROI");
		short[] startCell = (short[]) input.get(startCellKey);
		if (startCell==null)
		{
			log.info("Couldn't find StartCellData");
		}
		double[] relativeTimeOffsets = new double[roi*npix];
		for (int px = 0 ; px < npix ; px++){
			int patch = px / 9;
			double offsetAtStartCell = absoluteTimeOffsets[patch*numberOfSlices + startCell[px]];
			for (int slice = 0 ; slice < roi ; slice++){
				int cell = patch*numberOfSlices + (slice + startCell[px])%numberOfSlices;
				relativeTimeOffsets[px*roi+slice] = absoluteTimeOffsets[cell]-offsetAtStartCell;
			}
		}
		
		input.put(outputKey, relativeTimeOffsets);
		return input;
	}
		
	protected void loadDrsTimeCalibConstants(SourceURL  in) {
		try {

			FitsStream stream = new FitsStream(in);
			stream.init();
			drsTimeData = stream.readNext();
			log.debug("Read DRS Time data: {}", drsTimeData);
			
			if (!drsTimeData.containsKey(drsTimeKey))
			{
				throw new RuntimeException("Drs time data is missing key + " + drsTimeKey + "!");
			}
			this.absoluteTimeOffsets = (double[]) drsTimeData.get(drsTimeKey);

		} catch (Exception e) {

			log.error("Failed to load DRS data: {}", e.getMessage());
			if (log.isDebugEnabled())
				e.printStackTrace();
			this.absoluteTimeOffsets = null;

			throw new RuntimeException(e.getMessage());
		}
	}

	public String getStartCellKey() {
		return startCellKey;
	}

	public void setStartCellKey(String startCellKey) {
		this.startCellKey = startCellKey;
	}

	public String getOutputKey() {
		return outputKey;
	}

	public void setOutputKey(String outputKey) {
		this.outputKey = outputKey;
	}
	
	public int getNumberOfSlices() {
		return numberOfSlices;
	}

	public void setNumberOfSlices(int numberOfSlices) {
		this.numberOfSlices = numberOfSlices;
	}

	public int getNumberOfTimemarker() {
		return numberOfTimemarker;
	}

	public void setNumberOfTimemarker(int numberOfTimemarker) {
		this.numberOfTimemarker = numberOfTimemarker;
	}

	public void setUrl(URL url) {
		try {
			loadDrsTimeCalibConstants(new SourceURL(url));
		} catch (Exception e) {
			throw new RuntimeException(e.getMessage());
		}
	}

	public String getDrsTimeKey() {
		return drsTimeKey;
	}

	public void setDrsTimeKey(String drsTimeKey) {
		this.drsTimeKey = drsTimeKey;
	}

}