EDIT

/**

 * Copyright (C) 2017 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.workingSet.matrix;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashMap;

import java.util.Map;

import java.util.Set;

import eu.etaxonomy.cdm.api.service.IOccurrenceService;

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

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

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

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

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

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

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

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

import eu.etaxonomy.cdm.model.location.NamedArea;

import eu.etaxonomy.cdm.model.occurrence.DerivedUnit;

import eu.etaxonomy.cdm.model.occurrence.FieldUnit;

import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;

import eu.etaxonomy.cdm.model.taxon.Taxon;

import eu.etaxonomy.cdm.model.taxon.TaxonBase;

import eu.etaxonomy.cdm.model.taxon.TaxonNode;

import eu.etaxonomy.taxeditor.model.MessagingUtils;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * Wrapper object representing one row in the character matrix

 * containing a specimen description and supplemental information

 * about the taxon and the specimen

 * @author pplitzner

 * @since Dec 14, 2017

 *

 */

public class RowWrapper {

    private SpecimenDescription description;

    private TaxonNode taxonNode;

    private FieldUnit fieldUnit;

    private String identifier;

    private NamedArea country;

    private Map<Feature, DescriptionElementBase> featureToElementMap = new HashMap<>();

    public RowWrapper(SpecimenDescription description, WorkingSet workingSet) {

        this.description = description;

        IOccurrenceService occurrenceService = CdmStore.getService(IOccurrenceService.class);

        SpecimenOrObservationBase<?> specimen = description.getDescribedSpecimenOrObservation();

        //supplemental information

        if(specimen!=null){

            Collection<TaxonBase<?>> associatedTaxa = occurrenceService.listAssociatedTaxa(specimen, null, null, null, null);

            if(associatedTaxa!=null){

                //FIXME: what about multiple associated taxa

                taxonNode = ((Taxon) associatedTaxa.iterator().next()).getTaxonNode(MatrixUtility.getClassificationForWorkingSet(workingSet));

            }

            Collection<FieldUnit> fieldUnits = occurrenceService.getFieldUnits(specimen.getUuid());

            if(fieldUnits.size()!=1){

                MessagingUtils.error(RowWrapper.class, "More than one or no field unit found for specimen", null); //$NON-NLS-1$

            }

            else{

                fieldUnit = fieldUnits.iterator().next();

            }

            if(specimen instanceof DerivedUnit){

                identifier = occurrenceService.getMostSignificantIdentifier(HibernateProxyHelper.deproxy(specimen, DerivedUnit.class));

            }

            if(fieldUnit!=null && fieldUnit.getGatheringEvent()!=null){

                country = fieldUnit.getGatheringEvent().getCountry();

            }

        }

        //description elements

        Set<DescriptionElementBase> elements = description.getElements();

        for (DescriptionElementBase descriptionElementBase : elements) {

            Feature feature = descriptionElementBase.getFeature();

            featureToElementMap.put(feature, descriptionElementBase);

        }

    }

    public QuantitativeData addQuantitativeData(Feature feature){

        QuantitativeData data = QuantitativeData.NewInstance(feature);

        description.addElement(data);

        featureToElementMap.put(feature, data);

        return data;

    }

    public CategoricalData addCategoricalData(Feature feature){

        CategoricalData data = CategoricalData.NewInstance(feature);

        description.addElement(data);

        featureToElementMap.put(feature, data);

        return data;

    }

    public SpecimenDescription getSpecimenDescription() {

        return description;

    }

    public TaxonNode getTaxonNode() {

        return taxonNode;

    }

    public FieldUnit getFieldUnit() {

        return fieldUnit;

    }

    public String getIdentifier() {

        return identifier;

    }

    public NamedArea getCountry() {

        return country;

    }

    public Object getDataValueForFeature(Feature feature){

        DescriptionElementBase descriptionElementBase = featureToElementMap.get(feature);

        return descriptionElementBase;

    }

    public void setDataValueForFeature(Feature feature, Object newValue){

        /* Only CategoricalData is handled here because for QuantitativeData the value

         * is set in the ModifyListener of the swt.Text in the CellEditor

         * for each StatisticalMeasure. So no need to set it again here.

         */

        DescriptionElementBase descriptionElementBase = featureToElementMap.get(feature);

        if(descriptionElementBase!=null && descriptionElementBase.isInstanceOf(CategoricalData.class) && newValue instanceof Collection){

            CategoricalData categoricalData = HibernateProxyHelper.deproxy(descriptionElementBase, CategoricalData.class);

            categoricalData.setStateDataOnly(new ArrayList<>((Collection<State>) newValue));

        }

    }

}