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 ca.odell.glazedlists.TreeList;

import eu.etaxonomy.cdm.api.service.dto.RowWrapperDTO;

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

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

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

/**

 * The tree format defines how to build the hierarchy path for the

 * taxon column in the character matrix

 *

 * @author pplitzner

 * @date 23.01.2018

 *

 */

 public class DescriptionTreeFormat implements TreeList.Format<Object> {

     private Rank maxRank;

     public DescriptionTreeFormat(WorkingSet workingSet) {

         this.maxRank = workingSet.getMaxRank();

     }

     @Override

     public void getPath(List path, Object element) {

         if(element instanceof RowWrapperDTO){

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

             TaxonNode taxonNode = ((RowWrapperDTO) element).getTaxonNode();

             addPathRecursive(path, taxonNode);

         }

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

     }

 }