View Javadoc
1   package fr.ifremer.tutti.ui.swing.content.protocol.zones.tree;
2   
3   /*
4    * #%L
5    * Tutti :: UI
6    * $Id:$
7    * $HeadURL:$
8    * %%
9    * Copyright (C) 2012 - 2016 Ifremer
10   * %%
11   * This program is free software: you can redistribute it and/or modify
12   * it under the terms of the GNU General Public License as
13   * published by the Free Software Foundation, either version 3 of the
14   * License, or (at your option) any later version.
15   * 
16   * This program is distributed in the hope that it will be useful,
17   * but WITHOUT ANY WARRANTY; without even the implied warranty of
18   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19   * GNU General Public License for more details.
20   * 
21   * You should have received a copy of the GNU General Public
22   * License along with this program.  If not, see
23   * <http://www.gnu.org/licenses/gpl-3.0.html>.
24   * #L%
25   */
26  
27  import fr.ifremer.tutti.persistence.entities.protocol.Zone;
28  import fr.ifremer.tutti.ui.swing.content.protocol.zones.tree.node.StrataNode;
29  import fr.ifremer.tutti.ui.swing.content.protocol.zones.tree.node.ZoneNode;
30  import org.apache.commons.logging.Log;
31  import org.apache.commons.logging.LogFactory;
32  
33  import java.util.Enumeration;
34  import java.util.LinkedHashSet;
35  import java.util.Set;
36  
37  import static org.nuiton.i18n.I18n.t;
38  
39  /**
40   * @author Kevin Morin (Code Lutin)
41   * @author Tony Chemit - chemit@codelutin.com
42   * @since 4.5
43   */
44  public class ZonesTreeModel extends ZoneEditorTreeModelSupport {
45  
46      /** Logger. */
47      private static final Log log = LogFactory.getLog(ZonesTreeModel.class);
48  
49      public ZonesTreeModel() {
50          super(t("tutti.zoneEditor.zones.root.label"));
51      }
52  
53      public ZoneNode addZone(Zone zone) {
54  
55          if (log.isInfoEnabled()) {
56              log.info("Add zone: " + zone);
57          }
58          ZoneNode zoneNode = new ZoneNode(zone.getId(), zone.getLabel());
59          addNode(getRoot(), zoneNode);
60  
61          zone.getStrata().forEach(strata -> {
62  
63              StrataNode strataNode = addStrata(zoneNode, strata.getId());
64              strata.getSubstrata().forEach(subStrata -> addSubsStrata(strataNode, subStrata.getId()));
65  
66          });
67  
68          return zoneNode;
69  
70      }
71  
72      public Set<Zone> getZones() {
73  
74          Set<Zone> zones = new LinkedHashSet<>();
75          Enumeration zoneNodes = getRoot().children();
76          while (zoneNodes.hasMoreElements()) {
77              Object zoneNode = zoneNodes.nextElement();
78              zones.add(((ZoneNode) zoneNode).toBean());
79          }
80          return zones;
81  
82      }
83  
84  }