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.ITreeViewerListener;

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

import org.eclipse.jface.viewers.TreeExpansionEvent;

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.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 DistributionContentProvider());

        Styler styler = new Styler() {

            @Override

            public void applyStyles(TextStyle textStyle) {

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

            }

        };

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

        getViewer().setComparator(new AreaViewerComparator());

        ((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();

            }

        });

        ((CheckboxTreeViewer)getViewer()).addTreeListener(new ITreeViewerListener() {

            @Override

            public void treeCollapsed(TreeExpansionEvent event) {

            }

            @Override

            public void treeExpanded(TreeExpansionEvent event) {

                final Object element = event.getElement();

                final Object[] children = ((DistributionContentProvider)getViewer().getContentProvider()).getChildren(element);

                for (Object child : children) {

                    if (child instanceof NamedAreaWrapper){

                        if (getListCheckedTerms().contains(((NamedAreaWrapper) child).getNamedArea()) ){

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

                        }

                        if (getListGrayedTerms().contains(((NamedAreaWrapper) child).getNamedArea())&& !((NamedAreaWrapper) child).isBaseElement){

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

                        }

                    }

                }

            }

        });

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

        rememberCheckedValues(checkedValues, grayedValues);

        ((CheckboxTreeViewer)getViewer()).expandToLevel(2);

        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) {

       if (element == null) {

           return;

       }

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

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

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

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

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

               ((CheckboxTreeViewer)getViewer()).setGrayChecked(element, !checked);

           }

//           if (element instanceof NamedAreaWrapper){

//               NamedAreaWrapper baseElement = new NamedAreaWrapper(((NamedAreaWrapper)element).getNamedArea(), true, ((NamedAreaWrapper)element).parent);

//               ((CheckboxTreeViewer)getViewer()).setChecked(baseElement, !checked);

//               ((CheckboxTreeViewer)getViewer()).setGrayed(element, checked);

//           }

       }else{

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

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

       }

   }

   /**

   *

   * @param path

   * @param checked

   */

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

      DistributionContentProvider tcp = new DistributionContentProvider();

      if (element == null) {

          return;

      }

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

      Object parent = tcp.getParent(element);

      if (parent != null) {

          boolean allUnchecked = true;

          boolean allChecked = 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)) {

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

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

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

              }

              if (allUnchecked) {

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

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

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

//                  }else{

//                      checked = true;

//                  }

              } else if (allChecked){

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

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

              }else {

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

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

                    ((CheckboxTreeViewer)getViewer()).setChecked(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;

    }

}