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.workingSet.matrix;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import org.eclipse.jface.dialogs.Dialog;

import org.eclipse.jface.viewers.ArrayContentProvider;

import org.eclipse.jface.viewers.IStructuredSelection;

import org.eclipse.jface.viewers.LabelProvider;

import org.eclipse.jface.viewers.TableViewer;

import org.eclipse.swt.graphics.Image;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Shell;

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

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

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

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

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

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

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

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

import eu.etaxonomy.taxeditor.model.ImageResources;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author pplitzner

 * @since Jan 3, 2018

 *

 */

public class DescriptionSelectionDialog extends Dialog {

    private TableViewer list;

    private WorkingSet workingSet;

    Collection<DescriptionBase> selectedDescriptions = new ArrayList<>();

    protected DescriptionSelectionDialog(Shell parentShell, WorkingSet workingSet) {

        super(parentShell);

        this.workingSet = workingSet;

    }

    @Override

    protected Control createDialogArea(Composite parent) {

        Composite composite = (Composite) super.createDialogArea(parent);

        list = new TableViewer(composite);

        list.setContentProvider(new ArrayContentProvider());

        list.setLabelProvider(new DescriptionLabelProvider());

        //gather descriptions

        Set<TaxonNode> taxonNodes = workingSet.getTaxonSubtreeFilter();

        Set<SpecimenDescription> specimenDescriptions = new HashSet<>();

        for (TaxonNode node : taxonNodes) {

            specimenDescriptions.addAll(addDescriptionsRecursively(node));

        }

        list.setInput(specimenDescriptions);

        return composite;

    }

    private Set<SpecimenDescription> addDescriptionsRecursively(TaxonNode node){

        Set<SpecimenDescription> descriptions = new HashSet<>();

        if(node.getTaxon()!=null){

            Taxon taxon = HibernateProxyHelper.deproxy(node.getTaxon(), Taxon.class);

            Collection<SpecimenOrObservationBase> specimens = CdmStore.getService(IOccurrenceService.class).listByAssociatedTaxon(SpecimenOrObservationBase.class, null, taxon, null, null, null, null, null);

            for (SpecimenOrObservationBase specimen : specimens) {

                Set<SpecimenDescription> specimenDescriptions = specimen.getSpecimenDescriptions();

                for (SpecimenDescription specimenDescription : specimenDescriptions) {

                    descriptions.add(specimenDescription);

                }

            }

        }

        List<TaxonNode> childNodes = node.getChildNodes();

        for (TaxonNode childNode : childNodes) {

            descriptions.addAll(addDescriptionsRecursively(childNode));

        }

        return descriptions;

    }

    /**

     * {@inheritDoc}

     */

    @Override

    protected void okPressed() {

        selectedDescriptions = ((IStructuredSelection)list.getSelection()).toList();

        super.okPressed();

    }

    public Collection<DescriptionBase> getDescriptions(){

        return selectedDescriptions;

    }

    private class DescriptionLabelProvider extends LabelProvider{

        /**

         * {@inheritDoc}

         */

        @Override

        public Image getImage(Object element) {

            if(workingSet.getDescriptions().contains(element)){

                return ImageResources.getImage(ImageResources.SYNCED);

            }

            return null;

        }

    }

}