EDIT

/**

* Copyright (C) 2022 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.concept.e4;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.Map.Entry;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.IStructuredSelection;

import org.eclipse.jface.viewers.ListViewer;

import org.eclipse.jface.viewers.StructuredSelection;

import org.eclipse.swt.widgets.Composite;

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

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

class ConceptRelationViewer extends ListViewer {

    public ConceptRelationViewer(Composite parent) {

		super(parent);

	}

	@Override

	public ISelection getSelection() {

		ISelection selection = super.getSelection();

		List<TaxonRelationship> relationships = new ArrayList<>();

		if(selection instanceof IStructuredSelection && !selection.isEmpty()){

		    for(Iterator<?> iter = ((IStructuredSelection) selection).iterator();iter.hasNext();){

		        Object object = iter.next();

		        relationships.add(((Entry<TaxonRelationship, Taxon>)object).getKey());

		    }

		    return new StructuredSelection(relationships);

		}

		return selection;

	}

	@Override

	public void setSelection(ISelection selection) {

		super.setSelection(selection);

	}

}