teichmann@5831: package org.dive4elements.river.artifacts.model.map;
rrenkert@5307: 
rrenkert@5307: import java.util.ArrayList;
rrenkert@5307: import java.util.List;
rrenkert@5307: 
rrenkert@5307: import org.apache.log4j.Logger;
rrenkert@5307: 
rrenkert@5307: public class HWSContainer
rrenkert@5307: {
rrenkert@5307:     private static Logger logger = Logger.getLogger(HWSContainer.class);
rrenkert@5307:     private String river;
rrenkert@5307:     private HWS.TYPE type;
rrenkert@5307:     private List<HWS> hws;
rrenkert@5307: 
rrenkert@5307:     public HWSContainer() {
rrenkert@5307:         river = null;
rrenkert@5307:         hws = new ArrayList<HWS>();
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public HWSContainer(String river, HWS.TYPE type, List<HWS> hws) {
rrenkert@5307:         this.river = river;
rrenkert@5307:         this.hws = hws;
rrenkert@5307:         this.type = type;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public void setRiver(String river) {
rrenkert@5307:         this.river = river;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public String getRiver() {
rrenkert@5307:         return this.river;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public HWS.TYPE getType() {
rrenkert@5307:         return type;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public void setType(HWS.TYPE type) {
rrenkert@5307:         this.type = type;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public List<HWS> getHws() {
rrenkert@5307:         return hws;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public void addHws(HWS hws) {
rrenkert@5307:         logger.debug("add hws: " + hws.getName());
rrenkert@5307:         this.hws.add(hws);
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public void addHws(List<HWS> hws) {
rrenkert@5307:         this.hws.addAll(hws);
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public List<HWS> getOfficialHWS() {
rrenkert@5307:         if (hws == null || hws.size() == 0) {
rrenkert@5307:             return new ArrayList<HWS>();
rrenkert@5307:         }
rrenkert@5307:         List<HWS> results = new ArrayList<HWS>();
rrenkert@5307:         for (HWS h: hws) {
rrenkert@5307:             if (h.isOfficial()) {
rrenkert@5307:                 results.add(h);
rrenkert@5307:             }
rrenkert@5307:         }
rrenkert@5307:         return results;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public List<HWS> getHws(String name) {
rrenkert@5307:         logger.debug("find: " + name + " in " + hws.size() + " elements");
rrenkert@5307:         if (hws == null || hws.size() == 0) {
rrenkert@5307:             return new ArrayList<HWS>();
rrenkert@5307:         }
rrenkert@5307:         List<HWS> results = new ArrayList<HWS>();
rrenkert@5307:         for (HWS h: hws) {
rrenkert@5307:             if (h.getName().equals(name)) {
rrenkert@5307:                 results.add(h);
rrenkert@5307:             }
rrenkert@5307:         }
rrenkert@5307:         logger.debug("found: " + results.size());
rrenkert@5307:         return results;
rrenkert@5307:     }
rrenkert@5307: 
rrenkert@5307:     public List<HWS> getHws(List<String> list) {
rrenkert@5307:         if (hws == null || hws.size() == 0) {
rrenkert@5307:             return new ArrayList<HWS>();
rrenkert@5307:         }
rrenkert@5307:         List<HWS> results = new ArrayList<HWS>();
rrenkert@5307:         for (String name : list) {
rrenkert@5307:             results.addAll(getHws(name));
rrenkert@5307:         }
rrenkert@5307:         return results;
rrenkert@5307:     }
rrenkert@5307: }