Project

General

Profile

Download (3.13 KB) Statistics
| Branch: | Tag: | Revision:
1
// $Id$
2
/**
3
* Copyright (C) 2018 EDIT
4
* European Distributed Institute of Taxonomy
5
* http://www.e-taxonomy.eu
6
*
7
* The contents of this file are subject to the Mozilla Public License Version 1.1
8
* See LICENSE.TXT at the top of this package for the full license terms.
9
*/
10
package eu.etaxonomy.taxeditor.editor.descriptiveDataSet.matrix;
11

    
12
import java.util.Comparator;
13
import java.util.HashMap;
14
import java.util.List;
15
import java.util.Map;
16
import java.util.UUID;
17

    
18
import ca.odell.glazedlists.TreeList;
19
import eu.etaxonomy.cdm.api.service.ITaxonNodeService;
20
import eu.etaxonomy.cdm.api.service.dto.RowWrapperDTO;
21
import eu.etaxonomy.cdm.model.description.DescriptiveDataSet;
22
import eu.etaxonomy.cdm.model.name.Rank;
23
import eu.etaxonomy.cdm.model.taxon.TaxonNode;
24
import eu.etaxonomy.cdm.persistence.dto.TaxonNodeDto;
25
import eu.etaxonomy.taxeditor.store.CdmStore;
26

    
27
/**
28
 * The tree format defines how to build the hierarchy path for the
29
 * taxon column in the character matrix
30
 *
31
 * @author pplitzner
32
 * @date 23.01.2018
33
 *
34
 */
35
 public class DescriptionTreeFormat implements TreeList.Format<Object> {
36

    
37
     private Rank maxRank;
38

    
39
     private Map<UUID, TaxonNodeDto> uuidToTaxonNodeDtoMap = new HashMap<>();
40

    
41
     public DescriptionTreeFormat(DescriptiveDataSet descriptiveDataSet) {
42
         this.maxRank = descriptiveDataSet.getMaxRank();
43
     }
44

    
45
     @Override
46
     public void getPath(List path, Object element) {
47
         if(element instanceof RowWrapperDTO){
48
             //TODO: check for multiple taxon nodes in multiple classifications
49
             TaxonNodeDto taxonNode = ((RowWrapperDTO) element).getTaxonNode();
50
             //use dto from map cache because the elements have to be
51
             //exactly the same for the tree format to work
52
             if(uuidToTaxonNodeDtoMap.get(taxonNode.getUuid())!=null){
53
                 taxonNode = uuidToTaxonNodeDtoMap.get(taxonNode.getUuid());
54
             }
55
             else{
56
                 uuidToTaxonNodeDtoMap.put(taxonNode.getUuid(), taxonNode);
57
             }
58
             addPathRecursive(path, taxonNode);
59
         }
60
         path.add(element);
61
     }
62

    
63
     private void addPathRecursive(List path, TaxonNodeDto node){
64
         if(node!=null){
65
             if(maxRank!=null && node.getRankOrderIndex()==maxRank.getOrderIndex()){
66
                 path.add(node);
67
                 return;
68
             }
69
             TaxonNodeDto parentNodeDto = uuidToTaxonNodeDtoMap.get(node.getParentUUID());
70
             if(parentNodeDto==null){
71
                 TaxonNode parent = CdmStore.getService(ITaxonNodeService.class).load(node.getParentUUID());
72
                 if(parent!=null){
73
                     parentNodeDto = new TaxonNodeDto(parent);
74
                     uuidToTaxonNodeDtoMap.put(node.getParentUUID(), parentNodeDto);
75
                 }
76
             }
77
             if(parentNodeDto!=null){
78
                 addPathRecursive(path, parentNodeDto);
79
                 path.add(node);
80
             }
81
         }
82
     }
83

    
84
     @Override
85
     public boolean allowsChildren(Object element) {
86
         return true;
87
     }
88

    
89
     @Override
90
     public Comparator<Object> getComparator(int depth) {
91
         return new MatrixRowComparator();
92
     }
93

    
94
 }
(13-13/19)