
package de.ivu.elect.business.entschaedigung.control;

import java.util.HashMap;
import java.util.Map;

import de.ivu.elect.business.entschaedigung.entity.Entschaedigungsart;
import de.ivu.elect.business.wahl.entity.Wahl;
import de.ivu.elect.business.wahl.entity.Wahltermin;
import de.ivu.elect.jee.entity.tk.EntityTK;

public class AuszahlungImportContext extends EntschaedigungExchangeContext {

  private static final long serialVersionUID = 1L;

  private final Map<EntityTK, Entschaedigungsart> zusaetzlicheEntschaedigungsartMap;

  public AuszahlungImportContext(Wahl wahl, Wahltermin wahltermin) {
    super(wahl, wahltermin);

    this.zusaetzlicheEntschaedigungsartMap = new HashMap<>();
  }

  public Map<EntityTK, Entschaedigungsart> getZusaetzlicheEntschaedigungsartMap() {
    return zusaetzlicheEntschaedigungsartMap;
  }

  public void addZusaetzlicheEntschaedigungsart(Entschaedigungsart entity) {
    zusaetzlicheEntschaedigungsartMap.put(EntschaedigungsartImportTKFactory.create(entity), entity);
  }
}
