TagSet.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2021 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.tags;
 
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;
import org.turro.marker.MarkerHelper;
 
public class TagSet extends TreeSet<TagItem> {
 
  public TagSet(Collection<? extends TagItem> c) {
    super(c);
  }
 
  public void select(String tagName) {
    for(TagItem t : this) {
      if(t.getTagName().equals(tagName)) {
        t.setSelected(true);
      }
    }
  }
  
  public void unselect(String tagName) {
    for(TagItem t : this) {
      if(t.getTagName().equals(tagName)) {
        t.setSelected(false);
      }
    }
  }
  
  public void toggle(String tagName) {
    for(TagItem t : this) {
      if(t.getTagName().equals(tagName)) {
        if(t.isSelected()) {
          t.setSelected(false);
        } else if(t.isSibling()) {
          t.setSelected(true);
        } else {
          unselectAll();
          t.setSelected(true);
        }       
      }
    }
  }
  
  public void selectAll() {
    for(TagItem t : this) {
      t.setSelected(true);
    }
  }
  
  public void unselectAll() {
    for(TagItem t : this) {
      t.setSelected(false);
    }
  }
  
  public void toggle() {
    for(TagItem t : this) {
      t.setSelected(!t.isSelected());
    }
  }
  
  public String tagUrl(TagItem tag) {
    return MarkerHelper.setObfuscatedRightNowPars("tag=" + tag.getTagName());
  }
  
  public TreeSet<String> getTagNames() {
    return new TreeSet(stream().map(t -> t.getTagName()).collect(Collectors.toSet()));
  }
  
  public TreeSet<TagItem> getSelected() {
    return new TreeSet(stream().filter(t -> t.isSelected()).collect(Collectors.toSet()));
  }
  
  public TreeSet<TagItem> getUnselected() {
    return new TreeSet(stream().filter(t -> !t.isSelected()).collect(Collectors.toSet()));
  }
  
  public TreeSet<String> getSelectedTagNames() {
    return new TreeSet(stream().filter(t -> t.isSelected()).map(t -> t.getTagName()).collect(Collectors.toSet()));
  }
  
  public TreeSet<String> getUnselectedTagNames() {
    return new TreeSet(stream().filter(t -> !t.isSelected()).map(t -> t.getTagName()).collect(Collectors.toSet()));
  }
  
  public long getMaxUsage() {
    return stream().mapToLong(t -> t.getUsage()).max().orElse(0);
  }
 
  public List<String> getIdentifiers(String root) {
    Set<TagItem> set = getSelected();
    if(!(set == null && set.isEmpty())) {
      return Tags.getIdentifiers(root, set);
    }
    return Collections.EMPTY_LIST;
  }
 
  public void markSiblingsFromSelection(String root) {
    TagSet siblings = Tags.getSiblings(root, getSelectedTagNames());
    for(TagItem tag : getUnselected()) {
      tag.setSibling(siblings.contains(tag));
    }
  }
  
  public static TagSet empty() {
    return new TagSet(Collections.EMPTY_SET);
  }
  
}