
package de.ivu.elect.business.ergebniskonflikt.entity;

import java.math.BigInteger;
import java.util.*;

import com.google.common.math.BigIntegerMath;

import de.ivu.elect.business.ergebniskonflikt.control.util.SublistList;
import de.ivu.elect.business.gebietsbaum.boundary.AbstractWahldurchfuehrungsgebietDAO;
import de.ivu.elect.business.gebietsbaum.entity.GebietPK;
import de.ivu.elect.business.gebietsbaum.entity.Wahldurchfuehrungsgebiet;
import de.ivu.elect.business.gruppe.boundary.AbstractPolitischeGruppeDAO;
import de.ivu.elect.business.gruppe.entity.GruppePK;
import de.ivu.elect.business.gruppe.entity.PolitischeGruppe;
import de.ivu.elect.business.kandidatur.boundary.AbstractKandidaturDAO;
import de.ivu.elect.business.kandidatur.entity.Kandidatur;
import de.ivu.elect.business.person.entity.PersonendatenPK;
import de.ivu.elect.business.wahlvorschlag.boundary.AbstractListenverbindungDAO;
import de.ivu.elect.business.wahlvorschlag.boundary.AbstractWahlvorschlagDAO;
import de.ivu.elect.business.wahlvorschlag.entity.*;
import de.ivu.elect.jee.entity.EntityLongPK;
import de.ivu.elect.jee.faces.Messages;

public abstract class AbstractListenUndListenverbindungenAuswaehlenErgebnisKonfliktCustomData
    extends
      AbstractErgebnisKonfliktLosoptionenCustomData {

  private static final long serialVersionUID = 1L;

  private final Set<WahlvorschlagPK> listePKsZurAuswahl = new LinkedHashSet<>();

  private final Set<WahlvorschlagPK> listePKsGelost = new LinkedHashSet<>();

  private final Set<ListenverbindungPK> listenverbindungPKsZurAuswahl = new LinkedHashSet<>();

  private final Set<ListenverbindungPK> listenverbindungPKsGelost = new LinkedHashSet<>();

  @Override
  protected Collection<PersonendatenPK> getBeteiligteKandidaturenPKs() {
    return Collections.emptySet();
  }

  @Override
  protected Collection<PersonendatenPK> getGelosteKandidaturenPKs() {
    return Collections.emptySet();
  }

  @Override
  protected Collection<WahlvorschlagPK> getBeteiligteListenPKs() {
    return listePKsZurAuswahl;
  }

  @Override
  protected Collection<WahlvorschlagPK> getGelosteListenPKs() {
    return listePKsGelost;
  }

  @Override
  protected Collection<ListenverbindungPK> getBeteiligteListenverbindungenPKs() {
    return listenverbindungPKsZurAuswahl;
  }

  @Override
  protected Collection<ListenverbindungPK> getGelosteListenverbindungenPKs() {
    return listenverbindungPKsGelost;
  }

  @Override
  protected Collection<GruppePK> getBeteiligtePolitischeGruppenPKs() {
    return Collections.emptySet();
  }

  @Override
  protected Collection<GruppePK> getGelostePolitischeGruppenPKs() {
    return Collections.emptySet();
  }

  @Override
  protected Collection<GebietPK> getBeteiligteGebietPKs() {
    return null;
  }

  @Override
  protected Collection<GebietPK> getGelosteGebietPKs() {
    return null;
  }

  @Override
  public boolean isFachlichGleich(ErgebnisKonfliktCustomData other) {
    return super.isFachlichGleich(other) && equals(other);
  }

  public Set<WahlvorschlagPK> getListePKsZurAuswahl() {
    return listePKsZurAuswahl;
  }

  public void setListePKsZurAuswahl(Set<WahlvorschlagPK> listePKsZurAuswahl) {
    this.listePKsZurAuswahl.clear();
    this.listePKsZurAuswahl.addAll(listePKsZurAuswahl);

    notifyEntity();
  }

  public void setListenZurAuswahl(Collection<Liste> listenZurAuswahl) {
    this.listePKsZurAuswahl.clear();
    for (Liste liste : listenZurAuswahl) {
      listePKsZurAuswahl.add(liste.getPk());
    }

    notifyEntity();
  }

  public Set<WahlvorschlagPK> getListePKsGelost() {
    return listePKsGelost;
  }

  public void setListePKsGelost(Set<WahlvorschlagPK> listePKsGelost) {
    this.listePKsGelost.clear();
    this.listePKsGelost.addAll(listePKsGelost);

    notifyEntity();
  }

  public void setListenGelost(Collection<Liste> listenGelost) {
    this.listePKsGelost.clear();
    for (Liste liste : listenGelost) {
      listePKsGelost.add(liste.getPk());
    }

    notifyEntity();
  }

  public Set<ListenverbindungPK> getListenverbindungPKsZurAuswahl() {
    return listenverbindungPKsZurAuswahl;
  }

  public void setListenverbindungPKsZurAuswahl(Set<ListenverbindungPK> listenverbindungPKsZurAuswahl) {
    this.listenverbindungPKsZurAuswahl.clear();
    this.listenverbindungPKsZurAuswahl.addAll(listenverbindungPKsZurAuswahl);

    notifyEntity();
  }

  public Set<ListenverbindungPK> getListenverbindungPKsGelost() {
    return listenverbindungPKsGelost;
  }

  public void setListenverbindungPKsGelost(Set<ListenverbindungPK> listenverbindungPKsGelost) {
    this.listenverbindungPKsGelost.clear();
    this.listenverbindungPKsGelost.addAll(listenverbindungPKsGelost);

    notifyEntity();
  }

  @Override
  public List<ErgebnisKonfliktLosoption> getLosoptionenVerfuegbar(AbstractWahlvorschlagDAO wahlvorschlagDao,
                                                                  AbstractKandidaturDAO kandidaturDao, AbstractListenverbindungDAO listenverbindungDao,
                                                                  AbstractPolitischeGruppeDAO politischeGruppeDAO, AbstractWahldurchfuehrungsgebietDAO wahldurchfuehrungsgebietDAO, Messages messages) {
    return createLosoptionen(wahlvorschlagDao, getListePKsZurAuswahl(), listenverbindungDao, getListenverbindungPKsZurAuswahl(), messages);
  }

  @Override
  public List<ErgebnisKonfliktLosoption> getLosoptionenGelost(AbstractWahlvorschlagDAO wahlvorschlagDao,
                                                              AbstractKandidaturDAO kandidaturDao, AbstractListenverbindungDAO listenverbindungDao,
                                                              AbstractPolitischeGruppeDAO politischeGruppeDAO,
                                                              AbstractWahldurchfuehrungsgebietDAO wahldurchfuehrungsgebietDAO, Messages messages) {
    return createLosoptionen(wahlvorschlagDao, getListePKsGelost(), listenverbindungDao, getListenverbindungPKsGelost(), messages);
  }

  @Override
  public void setLosoptionenGelost(List<ErgebnisKonfliktLosoption> losoptionenGelost) {
    Set<WahlvorschlagPK> listePks = new LinkedHashSet<>();
    Set<ListenverbindungPK> listenverbindungPks = new LinkedHashSet<>();

    for (ErgebnisKonfliktLosoption losoption : losoptionenGelost) {
      if (losoption.isInstanceOf(Liste.class)) {
        listePks.add((WahlvorschlagPK) losoption.getPk());
      } else if (losoption.isInstanceOf(Listenverbindung.class)) {
        listenverbindungPks.add((ListenverbindungPK) losoption.getPk());
      }
    }

    setListePKsGelost(listePks);
    setListenverbindungPKsGelost(listenverbindungPks);
  }

  /**
   * equals() bezieht sich nur auf die Attribute, die die zu entscheidene Situation beschreiben, aber nicht auf die
   * Attribute, die die Lösung / Auswahl / Entscheidung betreffen.
   * 
   * @see java.lang.Object#equals(java.lang.Object)
   */
  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }

    if (!(o instanceof AbstractListenUndListenverbindungenAuswaehlenErgebnisKonfliktCustomData)) {
      return false;
    }

    AbstractListenUndListenverbindungenAuswaehlenErgebnisKonfliktCustomData that = (AbstractListenUndListenverbindungenAuswaehlenErgebnisKonfliktCustomData) o;

    if (getAnzahlZuLosen() != that.getAnzahlZuLosen()) {
      return false;
    }

    if (!listePKsZurAuswahl.equals(that.listePKsZurAuswahl)) {
      return false;
    }

    if (!listenverbindungPKsZurAuswahl.equals(that.listenverbindungPKsZurAuswahl)) {
      return false;
    }

    return true;
  }

  @Override
  public int hashCode() {
    int result = (int) (getAnzahlZuLosen() ^ getAnzahlZuLosen() >>> 32);

    result = 31 * result + listePKsZurAuswahl.hashCode();
    result = 31 * result + listenverbindungPKsZurAuswahl.hashCode();

    return result;
  }

  @Override
  public void loeseKonfliktAutomatisch() {
    long anzahl = getAnzahlZuLosen();
    Set<ListenverbindungPK> listenverbindungPksGelost = waehleDieErstenAus(anzahl, getBeteiligteListenverbindungenPKs());
    setListenverbindungPKsGelost(listenverbindungPksGelost);

    anzahl -= listenverbindungPksGelost.size();
    Set<WahlvorschlagPK> listePksGelost = waehleDieErstenAus(anzahl, getBeteiligteListenPKs());
    setListePKsGelost(listePksGelost);
  }

  @Override
  public void loeseKonfliktAutomatisch(int indexDerLoesung) {
    List<EntityLongPK> allePKs = getGelosteListenverbindungPKsUndListenPKs();
    List<EntityLongPK> gelost = new SublistList<>(allePKs, (int) getAnzahlZuLosen()).get(indexDerLoesung);
    setGelosteListenverbindungPKsUndListenPKs(gelost);
  }

  @Override
  public void loeseKonfliktAutomatischZyklisch(int indexDerLoesung) {
    List<EntityLongPK> allePKs = getGelosteListenverbindungPKsUndListenPKs();
    List<EntityLongPK> gelost = permutiereZyklisch(indexDerLoesung, allePKs);
    setGelosteListenverbindungPKsUndListenPKs(gelost);
  }

  private List<EntityLongPK> getGelosteListenverbindungPKsUndListenPKs() {
    List<EntityLongPK> allePKs = new ArrayList<>();
    allePKs.addAll(asSortedList(getBeteiligteListenverbindungenPKs()));
    allePKs.addAll(asSortedList(getBeteiligteListenPKs()));
    return allePKs;
  }

  private void setGelosteListenverbindungPKsUndListenPKs(List<EntityLongPK> allePKs) {
    Set<ListenverbindungPK> listenverbindungPksGelost = new LinkedHashSet<>();
    Set<WahlvorschlagPK> listePksGelost = new LinkedHashSet<>();
    for (EntityLongPK pk : allePKs) {
      if (pk instanceof ListenverbindungPK) {
        listenverbindungPksGelost.add((ListenverbindungPK) pk);
      } else if (pk instanceof WahlvorschlagPK) {
        listePksGelost.add((WahlvorschlagPK) pk);
      }
    }

    setListenverbindungPKsGelost(listenverbindungPksGelost);
    setListePKsGelost(listePksGelost);
  }

  @Override
  public BigInteger getAnzahlLoesungen() {
    return BigIntegerMath.binomial(getAnzahlZyklischerLoesungen(), (int) getAnzahlZuLosen());
  }

  @Override
  public int getAnzahlZyklischerLoesungen() {
    return getBeteiligteListenverbindungenPKs().size() + getBeteiligteListenPKs().size();
  }

  @Override
  public String createLoesungBeschreibung(Collection<Wahlvorschlag> wahlvorschlaege, Collection<Kandidatur> ignore1,
                                          Collection<Listenverbindung> listenverbindungen, Collection<PolitischeGruppe> ignore2,
                                          Collection<Wahldurchfuehrungsgebiet> ignore3, Messages messages) {
    return createLoesungBeschreibungFuerListenUndListenverbindungen(wahlvorschlaege, getGelosteListenPKs(),
        listenverbindungen, getGelosteListenverbindungenPKs(), messages);
  }

}
