cz/vutbr/fit/dudka/SGVis/Data/RelationStorage.java

Go to the documentation of this file.

package cz.vutbr.fit.dudka.SGVis.Data;

import java.net.URL;
import java.util.Iterator;

public class RelationStorage implements
  Cloneable,
  IContainer<Relation>
{
  private RelationSet all;
  private UrlSet urlSet;
  private UrlMap<String> urlToHostMap;
  private RelationMap<URL> fromMap;
  private RelationMap<URL> toMap;
  private RelationMap<String> edgeTypeMap;
  
  public static RelationStorage fromRelationList(RelationList relationList) {
    RelationStorage relationSet = new RelationStorage();
    relationSet.add(relationList);
    return relationSet;
  }
  
  public static String urlToHost(URL url) {
    return url.getHost();
  }
  
  public RelationStorage() {
    all =   new RelationSet();
    urlSet = new UrlSet();
    urlToHostMap = new UrlMap<String>();
    fromMap = new RelationMap<URL>();
    toMap = new RelationMap<URL>();
    edgeTypeMap = new RelationMap<String>();
  }
  
  public RelationStorage(RelationStorage ref) {
    all =   new RelationSet(ref.all);
    urlSet = new UrlSet(ref.urlSet);
    urlToHostMap = new UrlMap<String>(ref.urlToHostMap);
    fromMap = new RelationMap<URL>(ref.fromMap);
    toMap = new RelationMap<URL>(ref.toMap);
    edgeTypeMap = new RelationMap<String>(ref.edgeTypeMap);
  }
  
  public RelationStorage clone() throws CloneNotSupportedException {
    return new RelationStorage(this);
  }
  
  public void add(Relation relation) {
    all.add(relation);
    addUrl(relation.from);
    addUrl(relation.to);
    fromMap.addToSet(relation.from, relation);
    toMap.addToSet(relation.to, relation);
    edgeTypeMap.addToSet(relation.edgeType, relation);
  }
  
  public void add(Iterable<Relation> relationList) {
    for(Relation relation: relationList)
      this.add(relation);
  }
  
  public IContainer<Relation> getAllFrom(URL from) {
    return fromMap.get(from);
  }
  
  public IContainer<Relation> getAllTo(URL to) {
    return toMap.get(to);
  }
  
  public IContainer<Relation> getAllIncidents(URL url) {
    RelationList list = new RelationList();
    list.add(getAllFrom(url));
    list.add(getAllTo(url));
    return list;
  }
  
  public IContainer<Relation> getAllOfEdgeType(String edgeType) {
    return edgeTypeMap.get(edgeType);
  }
  
  public Iterator<Relation> iterator() {
    return all.iterator();
  }
  
  public Iterable<URL> getUrls() {
    return urlSet;
  }
  
  public Iterable<URL> getUrls(String host) {
    return urlToHostMap.get(host);
  }
  
  public Iterable<String> getHosts() {
    return urlToHostMap.getKeys();
  }
  
  public Statistics getStatistics() {
    int nHosts = 0;
    for (@SuppressWarnings("unused")
    String s:urlToHostMap.getKeys())
      nHosts++;
    return new Statistics(
        nHosts,
        urlSet.size(),
        all.size());
  }
  private void addUrl(URL url) {
    urlSet.add(url);
    urlToHostMap.addToSet(urlToHost(url), url);
  }
}

---