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;

import java.util.Arrays;

import java.util.Collection;

import java.util.Collections;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import java.util.stream.Collectors;

import org.eclipse.jface.dialogs.Dialog;

import org.eclipse.jface.layout.GridLayoutFactory;

import org.eclipse.swt.SWT;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Shell;

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

import eu.etaxonomy.cdm.model.common.TermType;

import eu.etaxonomy.cdm.model.common.TermVocabulary;

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

import eu.etaxonomy.cdm.persistence.dto.TermDto;

import eu.etaxonomy.taxeditor.editor.definedterm.TermDtoContentProvider;

import eu.etaxonomy.taxeditor.editor.definedterm.TermDtoLabelProvider;

import eu.etaxonomy.taxeditor.preference.wizard.CheckBoxTreeComposite;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author pplitzner

 * @since Oct 29, 2018

 *

 */

public class AreasSelectionDialog extends Dialog{

    private CheckBoxTreeComposite treeComposite;

    private Set<TermDto> selectedAreas = new HashSet<>();

    private Collection<TermVocabulary<NamedArea>> areaVocabularies;

    protected AreasSelectionDialog(Shell parentShell, Set<TermDto> selectedAreas) {

        super(parentShell);

        this.selectedAreas = selectedAreas;

        this.areaVocabularies = CdmStore.getService(IVocabularyService.class).findByTermType(TermType.NamedArea, null);

    }

    @Override

    protected Control createDialogArea(Composite parent) {

        parent.setLayout(new GridLayout());

        treeComposite = new CheckBoxTreeComposite(parent, new TermDtoContentProvider(), new TermDtoLabelProvider(), SWT.NONE);

        treeComposite.getViewer().setInput(areaVocabularies);

        treeComposite.setCheckedElements(selectedAreas.toArray());

        GridLayoutFactory.fillDefaults().applyTo(treeComposite);

        return treeComposite;

    }

    @Override

    protected void configureShell(Shell newShell) {

        super.configureShell(newShell);

        newShell.setText("Choose areas");

        newShell.setSize(400, 600);

    }

    @Override

    protected void okPressed() {

        selectedAreas.clear();

        List<Object> checkedElements = Arrays.asList(treeComposite.getViewer().getCheckedElements());

        checkedElements = checkedElements.stream().filter(element->element instanceof TermDto).collect(Collectors.toList());

        Collections.sort(checkedElements, (o1, o2)->((TermDto)o1).getOrderIndex()-((TermDto)o2).getOrderIndex());

        selectedAreas = new HashSet(checkedElements);

        super.okPressed();

    }

    @Override

    protected boolean isResizable() {

        return true;

    }

    public Set<TermDto> getSelectedAreas(){

        return selectedAreas;

    }

}