EDIT

/**

* Copyright (C) 2018 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.taxeditor.editor.descriptiveDataSet.matrix.quantitative;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import java.util.stream.Collectors;

import org.eclipse.jface.window.Window;

import org.eclipse.nebula.widgets.nattable.config.IConfigRegistry;

import org.eclipse.nebula.widgets.nattable.edit.editor.ICellEditor;

import org.eclipse.nebula.widgets.nattable.edit.gui.CellEditDialog;

import org.eclipse.nebula.widgets.nattable.layer.cell.ILayerCell;

import org.eclipse.swt.SWT;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Shell;

import org.eclipse.swt.widgets.Text;

import eu.etaxonomy.cdm.api.service.dto.RowWrapperDTO;

import eu.etaxonomy.cdm.model.description.QuantitativeData;

import eu.etaxonomy.cdm.model.description.StatisticalMeasure;

import eu.etaxonomy.taxeditor.editor.descriptiveDataSet.matrix.CharacterMatrix;

/**

 * @author pplitzner

 * @since Jul 19, 2018

 *

 */

public class QuantitativeDataDialog extends CellEditDialog {

    private Character initialInput;

    private CharacterMatrix matrix;

    public QuantitativeDataDialog(Character initialInput, CharacterMatrix matrix, Shell parentShell, Object originalCanonicalValue, ILayerCell cell,

            ICellEditor cellEditor, IConfigRegistry configRegistry) {

        super(parentShell, originalCanonicalValue, cell, cellEditor, configRegistry);

        this.initialInput = initialInput;

        this.matrix = matrix;

    }

    private QuantitativeData editorValue;

    private RowWrapperDTO<?> rowWrapperDTO;

    private QuantitativeDataDialogComposite composite;

    @Override

    protected Control createDialogArea(Composite parent) {

        composite = new QuantitativeDataDialogComposite(initialInput, editorValue, parent, SWT.NONE);

        return composite;

    }

    void setEditorValue(QuantitativeData editorValue) {

        this.editorValue = editorValue;

    }

    void setRowWrapper(RowWrapperDTO<?> rowWrapperDTO) {

        this.rowWrapperDTO = rowWrapperDTO;

    }

    @Override

    public boolean close() {

        if(getReturnCode()==Window.OK){

            //clear values

            editorValue.getStatisticalValues().clear();

            //add back all values from text fields

            Map<StatisticalMeasure, List<String>> measureToValueMap = new HashMap<>();

            Map<StatisticalMeasure, List<Text>> textFields = composite.getTextFields();

            Set<Entry<StatisticalMeasure,List<Text>>> entrySet = textFields.entrySet();

            for (Entry<StatisticalMeasure, List<Text>> entry : entrySet) {

                StatisticalMeasure statisticalMeasure = entry.getKey();

                List<String> values = entry.getValue().stream()

                        .filter(text->text.isEnabled())

                        .map(text->text.getText())

                        .collect(Collectors.toList());

                measureToValueMap.put(statisticalMeasure, values);

            }

            rowWrapperDTO.setDataValueForQuantitativeData(editorValue.getFeature(), measureToValueMap, composite.getEditorValue().getUnit());

//            if (composite.getEditorValue().getUnit() == null && (composite.getEditorValue().getFeature().getRecommendedMeasurementUnits() != null && composite.getEditorValue().getFeature().getRecommendedMeasurementUnits().size() == 1)){

//                editorValue.setUnit(editorValue.getFeature().getRecommendedMeasurementUnits().iterator().next());

//            }else{

//                editorValue.setUnit(composite.getEditorValue().getUnit());

//            }

//            ((QuantitativeData)rowWrapperDTO.getDataValueForFeature(editorValue.getFeature())).setUnit(editorValue.getUnit());

            matrix.addRowToSave(rowWrapperDTO);

            matrix.setDirty();

        }

        return super.close();

    }

}