EDIT

// $Id$

/**

* Copyright (C) 2018 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.matrix;

import java.util.Comparator;

import java.util.List;

import java.util.Set;

import ca.odell.glazedlists.TreeList;

import eu.etaxonomy.cdm.model.name.Rank;

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

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

/**

 * @author pplitzner

 * @date 23.01.2018

 *

 */

 public class DescriptionTreeFormat implements TreeList.Format<Object> {

     private Rank maxRank;

     public DescriptionTreeFormat(Rank maxRank) {

         this.maxRank = maxRank;

     }

     @Override

     public void getPath(List path, Object element) {

         if(element instanceof RowWrapper){

             //TODO: check for multiple taxon nodes in multiple classifications

             Taxon taxon = ((RowWrapper) element).getAssociatedTaxon();

             Set<TaxonNode> taxonNodes = taxon.getTaxonNodes();

             if(taxonNodes!=null){

                 TaxonNode node = taxonNodes.iterator().next();

                 addPathRecursive(path, node);

             }

         }

         path.add(element);

     }

     private void addPathRecursive(List path, TaxonNode node){

         if(node.getParent()!=null

                 && node.getTaxon()!=null

                 && node.getTaxon().getName()!=null

                 && node.getTaxon().getName().getRank()!=null){

             Rank rank = node.getTaxon().getName().getRank();

             if(maxRank.equals(rank)){

                 path.add(node);

                 return;

             }

             addPathRecursive(path, node.getParent());

             path.add(node);

         }

     }

     @Override

     public boolean allowsChildren(Object element) {

         if(element instanceof TaxonNode){

             return true;

         }

         return false;

     }

     @Override

     public Comparator<Object> getComparator(int depth) {

         return new MatrixRowComparator();

     }

 }