/*-
 * Copyright 2016 Diamond Light Source Ltd.
 *
 * 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
 */

package uk.ac.diamond.scisoft.analysis.processing.operations.mask;

import org.eclipse.dawnsci.analysis.api.metadata.IDiffractionMetadata;
import org.eclipse.dawnsci.analysis.api.processing.Atomic;
import org.eclipse.dawnsci.analysis.api.processing.OperationData;
import org.eclipse.dawnsci.analysis.api.processing.OperationException;
import org.eclipse.dawnsci.analysis.api.processing.OperationRank;
import org.eclipse.dawnsci.analysis.dataset.operations.AbstractOperation;
import org.eclipse.january.IMonitor;
import org.eclipse.january.MetadataException;
import org.eclipse.january.dataset.Dataset;
import org.eclipse.january.dataset.DatasetUtils;
import org.eclipse.january.dataset.IDataset;
import org.eclipse.january.metadata.MaskMetadata;
import org.eclipse.january.metadata.MetadataFactory;

import uk.ac.diamond.scisoft.analysis.diffraction.powder.PixelIntegration;
import uk.ac.diamond.scisoft.analysis.diffraction.powder.PixelIntegrationBean;
import uk.ac.diamond.scisoft.analysis.diffraction.powder.PixelIntegrationCache;
import uk.ac.diamond.scisoft.analysis.io.DiffractionMetadata;
import uk.ac.diamond.scisoft.analysis.roi.XAxis;

@Atomic
public class MaskOutliersInQOperation extends AbstractOperation<MaskOutliersInQModel, OperationData> {

	private volatile PixelIntegrationCache cache;
	private IDiffractionMetadata metadata;
	
	@Override
	public String getId() {
		return "uk.ac.diamond.scisoft.analysis.processing.operations.mask.MaskOutliersInQOperation";
	}

	@Override
	public OperationRank getInputRank() {
		return OperationRank.TWO;
	}

	@Override
	public OperationRank getOutputRank() {
		return OperationRank.TWO;
	}

	@Override
	protected OperationData process(IDataset input, IMonitor monitor) throws OperationException {
		
		DiffractionMetadata meta = input.getFirstMetadata(DiffractionMetadata.class);
		
		if (meta == null) throw new OperationException(this, "Does not contain calibration information!");
		
		if (metadata == null) {
			metadata = meta;
			cache = null;
		} else {
			boolean dee = metadata.getDiffractionCrystalEnvironment().equals(meta.getDiffractionCrystalEnvironment());
			boolean dpe = metadata.getDetector2DProperties().equals(meta.getDetector2DProperties());
			
			if (!dpe || !dee) {
				metadata = meta;
				cache = null;
			}
		}
		
		PixelIntegrationCache lcache = getCache(input, meta);
		
		IDataset mask = null;
		
		if (input.getFirstMetadata(MaskMetadata.class) != null) {
			MaskMetadata mm = input.getFirstMetadata(MaskMetadata.class);
			if (mm.getMask() != null) {
				mask = mm.getMask();
			}
		}
		
		Dataset calculateOutlierMask = PixelIntegration.calculateOutlierMask(DatasetUtils.convertToDataset(input), mask, lcache, model.getScale(),model.isLow(),model.isHigh());
		
		MaskMetadata mm;
		try {
			mm = MetadataFactory.createMetadata(MaskMetadata.class, calculateOutlierMask);
		} catch (MetadataException e) {
			throw new OperationException(this, e);
		}
		input.setMetadata(mm);

		return new OperationData(input);
	}
	
	
	private PixelIntegrationCache getCache(IDataset input, IDiffractionMetadata meta) {

		PixelIntegrationCache lcache = cache;
		if (lcache == null) {
			synchronized(this) {
				lcache = cache;
				if (lcache == null) {
					PixelIntegrationBean bean = new PixelIntegrationBean();
					bean.setUsePixelSplitting(false);
					bean.setxAxis(XAxis.Q);
					bean.setAzimuthalIntegration(true);
					bean.setTo1D(true);
					bean.setShape(input.getShape());
					cache = lcache = new PixelIntegrationCache(meta, bean);
				}
			}
		}
		return lcache;
	}

}