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.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.jface.viewers.StyledString.Styler;

import org.eclipse.swt.SWT;

import org.eclipse.swt.graphics.TextStyle;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Display;

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.taxeditor.editor.definedterm.TermContentProvider;

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

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<NamedArea> selectedAreas = new HashSet<>();

    private Collection<TermVocabulary<NamedArea>> namedAreas;

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

        super(parentShell);

        this.selectedAreas = selectedAreas;

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

    }

    @Override

    protected Control createDialogArea(Composite parent) {

        parent.setLayout(new GridLayout());

        Styler styler = new Styler() {

            @Override

            public void applyStyles(TextStyle textStyle) {

                textStyle.foreground = Display.getCurrent().getSystemColor(SWT.COLOR_BLACK);

            }

        };

        treeComposite = new CheckBoxTreeComposite(parent, new TermContentProvider(), new AreaLabelProvider(styler), SWT.NONE);

        treeComposite.getViewer().setInput(namedAreas);

        treeComposite.setCheckedElements(selectedAreas.toArray());

        treeComposite.getViewer().reveal(selectedAreas);

        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 NamedArea).collect(Collectors.toList());

        selectedAreas = new HashSet(checkedElements);

        super.okPressed();

    }

    @Override

    protected boolean isResizable() {

        return true;

    }

    public Set<NamedArea> getSelectedAreas(){

        return selectedAreas;

    }

}