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 java.util.UUID;

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.layout.FillLayout;

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.api.service.ITaxonNodeService;

import eu.etaxonomy.cdm.api.service.config.FindOccurrencesConfigurator;

import eu.etaxonomy.cdm.filter.TaxonNodeFilter;

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

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

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

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author pplitzner

 * @since Jan 3, 2018

 *

 */

public class SpecimenSelectionDialog extends Dialog {

    private TableViewer list;

    private WorkingSet workingSet;

    Collection<SpecimenOrObservationBase> selectedSpecimens = new ArrayList<>();

    protected SpecimenSelectionDialog(Shell parentShell, WorkingSet workingSet) {

        super(parentShell);

        this.workingSet = workingSet;

    }

    @Override

    protected Control createDialogArea(Composite parent) {

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

        composite.setLayout(new FillLayout());

        list = new TableViewer(composite);

        list.setContentProvider(new DescriptionContentProvider());

        list.setLabelProvider(new LabelProvider());

        //gather descriptions

        Set<SpecimenOrObservationBase> specimens = new HashSet<>();

        //set filter parameters

        TaxonNodeFilter filter = TaxonNodeFilter.NewRankInstance(workingSet.getMinRank(), workingSet.getMaxRank());

        workingSet.getGeoFilter().forEach(area -> filter.orArea(area.getUuid()));

        workingSet.getTaxonSubtreeFilter().forEach(node -> filter.orSubtree(node));

        ITaxonNodeService taxonNodeService = CdmStore.getService(ITaxonNodeService.class);

        List<UUID> filteredNodes = taxonNodeService.uuidList(filter);

        for (UUID uuid : filteredNodes) {

            //TODO implement occurrence service for taxon nodes

            // let it return UuidAndTitleCache

            Taxon taxon = taxonNodeService.load(uuid).getTaxon();

            if(taxon!=null){

                FindOccurrencesConfigurator config = new FindOccurrencesConfigurator();

                config.setAssociatedTaxonUuid(taxon.getUuid());

                specimens.addAll(CdmStore.getService(IOccurrenceService.class).findByTitle(config).getRecords());

            }

        }

        list.setInput(specimens);

        return composite;

    }

    @Override

    protected void configureShell(Shell newShell) {

        super.configureShell(newShell);

        newShell.setText("Select specimens");

        newShell.setMinimumSize(500, 400);

    }

    /**

     * {@inheritDoc}

     */

    @Override

    protected void okPressed() {

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

        super.okPressed();

    }

    public Collection<SpecimenOrObservationBase> getSpecimen(){

        return selectedSpecimens;

    }

    private class DescriptionContentProvider extends ArrayContentProvider{

        /**

         * {@inheritDoc}

         */

        @Override

        public Object[] getElements(Object inputElement) {

            return ((Collection<SpecimenOrObservationBase>) inputElement).stream().filter(specimen ->

                specimen.getDescriptions().stream().noneMatch(description ->

                workingSet.getDescriptions().contains(description))).toArray();

        }

    }

}