fact.cleaning.ProbabilityClean 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.cleaning;

import fact.Utils;
import fact.hexmap.FactPixelMapping;
import fact.hexmap.ui.overlays.PixelSetOverlay;
import stream.Data;
import stream.Processor;

import java.util.ArrayList;

/**
 * Wolfgangs idea after previous cleaning make a new cleaning depending on the distance to showeraxis.
 * TODO: find thresholds and check code. unit test?
 * @author Fabian Temme
 */
public class ProbabilityClean extends BasicCleaning implements Processor {

	private String photonChargeKey = null;
	
	private String outputKey = null;
	
	private String deltaKey = null;
	private String cogxKey = null;
	private String cogyKey = null;
	
	private double probabilityThreshold;
	
	private double distanceCoeff = 1.0;
	private double distanceExp = -1.0;
	
	private double[] photoncharge = null;
	private double delta;
	private double cogx;
	private double cogy;
	
	private int npix;
	
	FactPixelMapping pixelMap = FactPixelMapping.getInstance();
	
	@Override
	public Data process(Data input) {
		Utils.isKeyValid(input, "NPIX", Integer.class);
		npix = (Integer) input.get("NPIX");
		Utils.mapContainsKeys(input, photonChargeKey, deltaKey);
		
		photoncharge = (double[]) input.get(photonChargeKey);
		
		delta = (Double) input.get(deltaKey);
		
		cogx = (Double) input.get(cogxKey);
		cogy = (Double) input.get(cogyKey);
		
		ArrayList showerlist = new ArrayList();
		
		for (int px = 0 ; px < npix ; px++)
		{
			double xpos = pixelMap.getPixelFromId(px).getXPositionInMM();
			double ypos = pixelMap.getPixelFromId(px).getYPositionInMM();
			double dist = Utils.calculateDistancePointToShowerAxis(cogx, cogy, delta, xpos, ypos);
			double weight = photoncharge[px] * Math.pow(distanceCoeff*dist, distanceExp); 
			
			if (weight > probabilityThreshold)
			{
				showerlist.add(px);
			}
			
		}
		
		showerlist = removeSmallCluster(showerlist,2);
		
		PixelSetOverlay cleanedPixelSet = new PixelSetOverlay();
		if (showerlist.size() > 0)
		{
			Integer[] showerArray = new Integer[showerlist.size()];
			for (int i = 0; i < showerlist.size(); i++) {
                cleanedPixelSet.addById(showerlist.get(i));
                showerArray[i] =  showerlist.get(i);
            }
			
			input.put(outputKey, showerArray);
			input.put(outputKey+"Set", cleanedPixelSet);
		}
		
		return input;
	}

	public String getPhotonChargeKey() {
		return photonChargeKey;
	}

	public void setPhotonChargeKey(String photonChargeKey) {
		this.photonChargeKey = photonChargeKey;
	}

	public String getOutputKey() {
		return outputKey;
	}

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

	public String getDeltaKey() {
		return deltaKey;
	}

	public void setDeltaKey(String deltaKey) {
		this.deltaKey = deltaKey;
	}

	public String getCogxKey() {
		return cogxKey;
	}

	public void setCogxKey(String cogxKey) {
		this.cogxKey = cogxKey;
	}

	public String getCogyKey() {
		return cogyKey;
	}

	public void setCogyKey(String cogyKey) {
		this.cogyKey = cogyKey;
	}

	public double getProbabilityThreshold() {
		return probabilityThreshold;
	}

	public void setProbabilityThreshold(double probabilityThreshold) {
		this.probabilityThreshold = probabilityThreshold;
	}

	public double getDistanceCoeff() {
		return distanceCoeff;
	}

	public void setDistanceCoeff(double distanceCoeff) {
		this.distanceCoeff = distanceCoeff;
	}

	public double getDistanceExp() {
		return distanceExp;
	}

	public void setDistanceExp(double distanceExp) {
		this.distanceExp = distanceExp;
	}

}