EDIT

/**

 * Copyright (C) 2014 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.preference.wizard;

import java.util.Arrays;

import java.util.Collection;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.eclipse.jface.preference.IPreferenceStore;

import org.eclipse.jface.viewers.CheckStateChangedEvent;

import org.eclipse.jface.viewers.CheckboxTreeViewer;

import org.eclipse.jface.viewers.ICheckStateListener;

import org.eclipse.jface.viewers.StyledString.Styler;

import org.eclipse.jface.viewers.TreePath;

import org.eclipse.jface.viewers.ViewerComparator;

import org.eclipse.swt.SWT;

import org.eclipse.swt.graphics.TextStyle;

import org.eclipse.swt.layout.GridData;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Display;

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

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

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

import eu.etaxonomy.taxeditor.preference.PreferencesUtil;

import eu.etaxonomy.taxeditor.session.ICdmEntitySession;

import eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author a.oppermann

 * @date 21.07.2014

 *

 */

public class AvailableDistributionPage extends AbstractAreaSelectionWizard implements ICdmEntitySessionEnabled {

    /**

     * @param pageName

     */

    protected AvailableDistributionPage(String pageName) {

        super(pageName);

        this.localPref = true;

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.eclipse.jface.dialogs.IDialogPage#createControl(org.eclipse.swt.widgets

     * .Composite)

     */

    /** {@inheritDoc} */

    @Override

    public void createControl(Composite parent) {

        IPreferenceStore preferenceStore = PreferencesUtil.getPreferenceStore();

        String checkedValues = preferenceStore.getString(PreferencesUtil.DISTRIBUTION_AREA_OCCURENCE_STATUS);

        String grayedValues = preferenceStore.getString(PreferencesUtil.DISTRIBUTION_AREA_OCCURENCE_STATUS_GRAYED);

        setTitle("Select areas for Distribution Editor");

        setDescription("In order to be able to modify and see the distribution status of taxa,\n"

                + "you have to select the areas which you like to see.");

        Composite composite = new Composite(parent, SWT.NULL);

        composite.setLayout(new GridLayout());

        setViewer(new CheckboxTreeViewer(composite, SWT.NULL));

        ((CheckboxTreeViewer)getViewer()).getTree().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));

        getViewer().setContentProvider(new TermContentProvider());

        Styler styler = new Styler() {

            @Override

            public void applyStyles(TextStyle textStyle) {

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

            }

        };

        getViewer().setLabelProvider(new TermLabelProvider(styler));

        getViewer().setComparator(new ViewerComparator());

        ((CheckboxTreeViewer)getViewer()).addCheckStateListener(new ICheckStateListener() {

            private boolean ignoreCheckEvent = false;

            @Override

            public void checkStateChanged(CheckStateChangedEvent event) {

                if (ignoreCheckEvent ) {

                    return;

                }

                ignoreCheckEvent = true;

                try {

                    checkChildren(event.getElement(), event.getChecked());

                    grayParents(event.getElement(), event.getChecked());

                    ((CheckboxTreeViewer)getViewer()).refresh();

                }

                finally {

                    ignoreCheckEvent = false;

                }

                checkNoneChecked();

            }

        });

        // TODO: write listener or toggle button to select all named areas

        rememberCheckedValues(checkedValues, grayedValues);

        setControl(composite);

    }

    private boolean checkNoneChecked() {

        if (((CheckboxTreeViewer)getViewer()).getCheckedElements().length == 0) {

            setMessage("Please check at least one item", WARNING);

            return true;

        } else {

            setMessage(null);

            return false;

        }

    }

    /**

     *

     * @param path

     */

    private void checkParents(final TreePath path) {

        if (path == null) {

            return;

        }

        TermContentProvider tcp = new TermContentProvider();

        Object treeElement = path.getLastSegment();

        if (treeElement != null) {

            boolean allChecked = true;

            boolean allUnchecked = true;

            for (final Object fieldElement : tcp.getChildren(treeElement)) {

                allChecked = allChecked && ((CheckboxTreeViewer)getViewer()).getChecked(fieldElement);

                allUnchecked = allUnchecked && !((CheckboxTreeViewer)getViewer()).getChecked(fieldElement);

            }

            if (allUnchecked) {

                ((CheckboxTreeViewer)getViewer()).setChecked(treeElement, false);

                ((CheckboxTreeViewer)getViewer()).setGrayed(treeElement, false);

            } else if (allChecked) {

                ((CheckboxTreeViewer)getViewer()).setChecked(treeElement, true);

                ((CheckboxTreeViewer)getViewer()).setGrayed(treeElement, false);

            } else if (!allUnchecked && !allChecked) {

                ((CheckboxTreeViewer)getViewer()).setChecked(treeElement, true);

                ((CheckboxTreeViewer)getViewer()).setGrayed(treeElement, true);

            }

        }

        checkParents(path.getParentPath());

    }

    /**

    *

    * @param path

    * @param checked

    */

   private void checkChildren(final Object element, final boolean checked) {

       TermContentProvider tcp = new TermContentProvider();

       if (element == null) {

           return;

       }

       ((CheckboxTreeViewer)getViewer()).setChecked(element, checked);

       ((CheckboxTreeViewer)getViewer()).setGrayed(element, false);

       if (((CheckboxTreeViewer)getViewer()).isExpandable(element)){

           ((CheckboxTreeViewer)getViewer()).expandToLevel(element, CheckboxTreeViewer.ALL_LEVELS);

           ((CheckboxTreeViewer)getViewer()).setSubtreeChecked(element, checked);

       }

   }

   /**

   *

   * @param path

   * @param checked

   */

  private void grayParents(final Object element, boolean checked) {

      TermContentProvider tcp = new TermContentProvider();

      if (element == null) {

          return;

      }

   //   final Object element = path.getLastSegment();

      Object parent = tcp.getParent(element);

      if (parent != null) {

          boolean allUnchecked = true;

          //if element is checked then the parent should be grayed if it is unchecked it have to be computed whether the parent should be checked

          if (checked){

              if (!((CheckboxTreeViewer)getViewer()).getChecked(parent)){

                  ((CheckboxTreeViewer)getViewer()).setGrayChecked(parent, true);

              }

          }else{

              for (final Object fieldElement : tcp.getChildren(parent)) {

                  allUnchecked = allUnchecked && !((CheckboxTreeViewer)getViewer()).getChecked(fieldElement);

                  allUnchecked = allUnchecked && !((CheckboxTreeViewer)getViewer()).getGrayed(fieldElement);

              }

              if (allUnchecked) {

                  ((CheckboxTreeViewer)getViewer()).setChecked(parent, false);

                  ((CheckboxTreeViewer)getViewer()).setGrayed(parent, false);

              } else {

                if (!((CheckboxTreeViewer)getViewer()).getChecked(parent)){

                    ((CheckboxTreeViewer)getViewer()).setGrayChecked(parent, true);

                }

                checked = true;

              }

          }

        grayParents(parent, checked);

      }

  }

    @Override

    public void dispose() {

        CdmStore.getCurrentSessionManager().dispose(this);

        super.dispose();

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getCdmEntitySession()

     */

    @Override

    public ICdmEntitySession getCdmEntitySession() {

        return CdmStore.getCurrentSessionManager().getNullSession();

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getRootEntities()

     */

    @Override

    public <T extends CdmBase> Collection<T> getRootEntities() {

        return null;

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getPropertyPathsMap()

     */

    @Override

    public Map<Object, List<String>> getPropertyPathsMap() {

        Map<Object, List<String>> propertyPathsMap = new HashMap<Object, List<String>>();

        List<String> termsPropertyPaths = Arrays.asList(new String[] {

                "includes"

        });

        propertyPathsMap.put("includes", termsPropertyPaths);

        propertyPathsMap.put("terms", termsPropertyPaths);

        return propertyPathsMap;

    }

}