de.gwdg.metadataqa.marc.analysis.SerialScores Maven / Gradle / Ivy

Go to download
package de.gwdg.metadataqa.marc.analysis;

import de.gwdg.metadataqa.marc.Utils;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class SerialScores {
  private Map scores;

  public SerialScores() {
    scores = new LinkedHashMap<>();
    for (SerialFields field : SerialFields.values()) {
      if (!field.equals(SerialFields.ID)) {
        scores.put(field, 0);
      }
    }
  }

  public void count(SerialFields key) {
    Utils.count(key, scores);
  }

  public void set(SerialFields key, int value) {
    scores.put(key, value);
  }

  public int get(SerialFields key) {
    return scores.getOrDefault(key, null);
  }

  public void calculateTotal() {
    var total = 0;
    for (Map.Entry entry : scores.entrySet()) {
      var field = entry.getKey();
      if (!field.equals(SerialFields.TOTAL)) {
        if (field.isClassification())
          total += Math.min(entry.getValue(), 10);
        else
          total += entry.getValue();
      }
    }
    set(SerialFields.TOTAL, total);
  }

  public List asList() {
    List list = new ArrayList<>();
    for (Map.Entry entry : scores.entrySet()) {
      SerialFields field = entry.getKey();
      if (field.isClassification())
        list.add(Math.min(entry.getValue(), 10));
      else
        list.add(entry.getValue());
    }
    return list;
  }

  public Map getScores() {
    return scores;
  }
}