EDIT

// $Id$

/**

* Copyright (C) 2017 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.workingSet;

import java.math.BigInteger;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.annotation.PostConstruct;

import javax.annotation.PreDestroy;

import org.bgbm.prometheus.xml.NEWQUALITATIVEPROPERTY;

import org.bgbm.prometheus.xml.ONTOLOGYDETAILS;

import org.bgbm.prometheus.xml.STRUCTURETERM;

import org.bgbm.prometheus.xml.Structure;

import org.bgbm.prometheus.xml.TYPETERM;

import org.eclipse.e4.ui.di.Focus;

import org.eclipse.jface.viewers.TreeViewer;

import org.eclipse.swt.SWT;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Tree;

import org.eclipse.ui.forms.widgets.TableWrapData;

import org.eclipse.ui.forms.widgets.TableWrapLayout;

/**

 * @author pplitzner

 * @date 24.05.2017

 *

 */

public class FeatureTreeBuilderViewPart {

    private Tree treeStructures;

    private Tree treeProperties;

    private Tree treeCharacters;

    private TreeViewer treeViewerCharacters;

    private TreeViewer treeViewerProperties;

    private TreeViewer treeViewerStructures;

    public FeatureTreeBuilderViewPart() {

        ontology = PrometheusOntology.createOntology();

        createIndex(ontology);

    }

    /**

     * Create contents of the view part.

     */

    @PostConstruct

    public void createControls(Composite parent) {

        {

            TableWrapLayout twl_parent = new TableWrapLayout();

            twl_parent.numColumns = 3;

            parent.setLayout(twl_parent);

        }

        treeViewerStructures = new TreeViewer(parent, SWT.BORDER);

        treeViewerStructures.setColumnProperties(new String[] {});

        treeStructures = treeViewerStructures.getTree();

        treeStructures.setLayoutData(new TableWrapData(TableWrapData.FILL_GRAB, TableWrapData.FILL_GRAB, 1, 1));

        treeViewerProperties = new TreeViewer(parent, SWT.BORDER);

        treeProperties = treeViewerProperties.getTree();

        treeProperties.setLayoutData(new TableWrapData(TableWrapData.FILL_GRAB, TableWrapData.FILL_GRAB, 1, 1));

        treeViewerCharacters = new TreeViewer(parent, SWT.BORDER);

        treeCharacters = treeViewerCharacters.getTree();

        treeCharacters.setLayoutData(new TableWrapData(TableWrapData.FILL_GRAB, TableWrapData.FILL_GRAB, 1, 1));

        init();

    }

    private void init(){

        treeViewerStructures.setContentProvider(new StructureContenProvider(this));

        treeViewerStructures.setLabelProvider(new StructureLabelProvider(this));

        treeViewerStructures.setInput("");

    }

    @PreDestroy

    public void dispose() {

    }

    @Focus

    public void setFocus() {

        // TODO	Set the focus to control

    }

    public TreeViewer getTreeViewerCharacters() {

        return treeViewerCharacters;

    }

    public TreeViewer getTreeViewerProperties() {

        return treeViewerProperties;

    }

    public TreeViewer getTreeViewerStructures() {

        return treeViewerStructures;

    }

    private final Map<String, List<String>> taxonToCharacterListMap = new HashMap<>();

    private final Map<BigInteger, STRUCTURETERM> structureTermMap = new HashMap<>();

    private final Map<BigInteger, TYPETERM> typeTermMap = new HashMap<>();

    private final Map<BigInteger, NEWQUALITATIVEPROPERTY> qualPropertyTermMap = new HashMap<>();

    private final Map<BigInteger, List<BigInteger>> structureTermToQualPropertyMap = new HashMap<>();

    private ONTOLOGYDETAILS ontology;

    private void createIndex(ONTOLOGYDETAILS ontology){

        //structures

        List<STRUCTURETERM> terms = ontology.getONTOLOGYDETAILSBody().getTERMS().getSTRUCTURETERMS().getSTRUCTURETERM();

        terms.forEach((STRUCTURETERM structureTerm) -> structureTermMap.put(structureTerm.getGLOBALID(), structureTerm));

        //type terms

        List<TYPETERM> typeTerms = ontology.getONTOLOGYDETAILSBody().getTERMS().getTYPEOFSTRUCTURETERMS().getTYPETERM();

        typeTerms.forEach((TYPETERM typeterm) -> typeTermMap.put(typeterm.getGLOBALID(), typeterm));

        //properties

        List<NEWQUALITATIVEPROPERTY> qualProperties = ontology.getONTOLOGYDETAILSBody().getNewQualitativeProperties().getNEWQUALITATIVEPROPERTY();

        qualProperties.forEach((NEWQUALITATIVEPROPERTY newqualitativeproperty) ->

        {

            BigInteger id = newqualitativeproperty.getIDSEQ();

            qualPropertyTermMap.put(id, newqualitativeproperty);

            List<Structure> structures = newqualitativeproperty.getStructuresLinkedToProperty().getStructure();

            structures.forEach((Structure structure) ->

            {

                List<BigInteger> properties = structureTermToQualPropertyMap.get(structure.getRefID());

                if(properties==null){

                    properties = new ArrayList<>();

                }

                properties.add(id);

                structureTermToQualPropertyMap.put(structure.getRefID(), properties);

            });

        });

    }

    /**

     * @return the ontology

     */

    public ONTOLOGYDETAILS getOntology() {

        return ontology;

    }

    public Map<BigInteger, STRUCTURETERM> getStructureTermMap() {

        return structureTermMap;

    }

}