FastaJobResults.java

package uk.co.flax.biosolr.pdbe.fasta;

import java.util.LinkedHashMap;
import java.util.Map;

public class FastaJobResults {

  private Map<PDb.Id, Map<String, PDb.Alignment>> alignments = new LinkedHashMap<>();
  private int numChains = 0;

  /*package*/ void addAlignment(PDb.Alignment alignment) {
    Map<String, PDb.Alignment> map = alignments.get(alignment.getPdbId());
    if (map == null) {
      map = new LinkedHashMap<>();
      alignments.put(alignment.getPdbId(), map);
    }
    if (map.put(alignment.getChain(), alignment) == null) {
      ++numChains;
    }
  }
  
  public PDb.Alignment getAlignment(String pdbId, String chain) {
    Map<String, PDb.Alignment> map = alignments.get(new PDb.Id(pdbId));
    if (map == null) return null;
    return map.get(chain);
  }
  
  public Map<PDb.Id, Map<String, PDb.Alignment>> getAlignments() {
    return alignments;
  }

  public int getNumChains() {
    return numChains;
  }

  public int getNumEntries() {
    return alignments.size();
  }

}