EDIT

/**

* Copyright (C) 2007 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.view.media;

import java.util.Collections;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import org.eclipse.core.runtime.Assert;

import org.eclipse.jface.viewers.ITreeContentProvider;

import org.eclipse.jface.viewers.Viewer;

import eu.etaxonomy.cdm.api.facade.DerivedUnitFacade;

import eu.etaxonomy.cdm.api.facade.DerivedUnitFacadeNotSupportedException;

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

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

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

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

import eu.etaxonomy.cdm.model.media.Media;

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

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

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

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

import eu.etaxonomy.taxeditor.editor.e4.TaxonEditorInputE4;

import eu.etaxonomy.taxeditor.model.MessagingUtils;

/**

 * <p>MediaContentProvider class.</p>

 *

 * @author n.hoffmann

 * @created Jun 15, 2010

 */

public class MediaContentProvider implements ITreeContentProvider{

	private static final Object[] NO_CHILDREN = new Object[0];

	/** {@inheritDoc} */

	@Override

    public Object[] getChildren(Object parentElement) {

		if (parentElement instanceof TaxonEditorInputE4 || parentElement instanceof Taxon) {

			Taxon taxon = null;

			if (parentElement instanceof TaxonEditorInputE4){

				taxon = ((TaxonEditorInputE4) parentElement).getTaxon();

			} else{

				taxon = HibernateProxyHelper.deproxy(parentElement, Taxon.class);

			}

			if(taxon == null){

				MessagingUtils.error(getClass(), "Taxon is null", null); //$NON-NLS-1$

				return NO_CHILDREN;

			}

			HashSet<DescriptionBase<?>> imageGalleries = new HashSet<>();

			for(DescriptionBase<?> description : taxon.getDescriptions()){

				if(description.isImageGallery()){

					imageGalleries.add(description);

				}

			}

			return imageGalleries.toArray();

		}

		else if (parentElement instanceof DescriptionBase) {

			if (((DescriptionBase<?>) parentElement).isImageGallery()) {

				List<Media> images =  getImages((DescriptionBase<?>) parentElement);

				if (images != null){

					return images.toArray();

				}

				return null;

			}

		}

		else if (parentElement instanceof DerivedUnit){

			try {

				DerivedUnitFacade facade = DerivedUnitFacade.NewInstance((DerivedUnit) parentElement);

				SpecimenDescription derivedUnitImageGallery = facade.getDerivedUnitImageGallery(false);

				if(derivedUnitImageGallery != null){

					return Collections.singleton(derivedUnitImageGallery).toArray();

				}

			} catch (DerivedUnitFacadeNotSupportedException e) {

				MessagingUtils.error(this.getClass(), "DerivedUnitFacadeNotSupportedException when trying to instantiate DerivedUnitFacade", e); //$NON-NLS-1$

			}

		}

		else if (parentElement instanceof FieldUnit){

		        DerivedUnitFacade facade = DerivedUnitFacade.NewInstance(SpecimenOrObservationType.FieldUnit, (FieldUnit) parentElement);

		        SpecimenDescription fieldObjectImageGallery = facade.getFieldObjectImageGallery(false);

		        if(fieldObjectImageGallery != null){

		            return Collections.singleton(fieldObjectImageGallery).toArray();

		        }

		}

		return NO_CHILDREN;

	}

	/** {@inheritDoc} */

	@Override

    public Object getParent(Object element) {

		// TODO Auto-generated method stub

		return null;

	}

	/** {@inheritDoc} */

	@Override

    public boolean hasChildren(Object element) {

		if (getChildren(element)== null) {

			return false;

		}

		return (getChildren(element).length > 0);

	}

	/** {@inheritDoc} */

	@Override

    public Object[] getElements(Object inputElement) {

		return getChildren(inputElement);

	}

	/**

	 * <p>dispose</p>

	 */

	@Override

    public void dispose() {}

	/** {@inheritDoc} */

	@Override

    public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {}

	private List<Media> getImages(DescriptionBase description){

		Assert.isTrue(description.isImageGallery(), "Description should have the imageGallery flag set."); //$NON-NLS-1$

		Set<DescriptionElementBase> elements = description.getElements();

		if (elements != null) {

			// by definition, image galleries have only one description element

			if(elements.size() > 1){

				MessagingUtils.error(this.getClass(), "There should be one and only one description element to hold the images. Found: " + elements.size() + " InDescription ID: " + description.getUuid(), null); //$NON-NLS-1$ //$NON-NLS-2$

			}

			if (!elements.isEmpty()){

				DescriptionElementBase element = elements.iterator().next();

				return element.getMedia();

			}

		}

		return null;

	}

	private SpecimenDescription createDerivedUnitFacadeImageGallery(DerivedUnitFacade facade){

		SpecimenDescription description = SpecimenDescription.NewInstance();

		description.setImageGallery(true);

		return description;

	}

}