
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 AbstractListenUndPolitischeGruppenAuswaehlenErgebnisKonfliktCustomData
    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<GruppePK> politischeGruppePKsZurAuswahl = new LinkedHashSet<>();

  private final Set<GruppePK> politischeGruppePKsGelost = 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 Collections.emptySet();
  }

  @Override
  protected Collection<ListenverbindungPK> getGelosteListenverbindungenPKs() {
    return Collections.emptySet();
  }

  @Override
  protected Collection<GruppePK> getBeteiligtePolitischeGruppenPKs() {
    return politischeGruppePKsZurAuswahl;
  }

  @Override
  protected Collection<GruppePK> getGelostePolitischeGruppenPKs() {
    return politischeGruppePKsGelost;
  }

  @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<GruppePK> getPolitischeGruppePKsZurAuswahl() {
    return politischeGruppePKsZurAuswahl;
  }

  public void setPolitischeGruppePKsZurAuswahl(Set<GruppePK> politischeGruppePKsZurAuswahl) {
    this.politischeGruppePKsZurAuswahl.clear();
    this.politischeGruppePKsZurAuswahl.addAll(politischeGruppePKsZurAuswahl);

    notifyEntity();
  }

  public void setPolitischeGruppenZurAuswahl(Collection<PolitischeGruppe> politischeGruppenZurAuswahl) {
    this.politischeGruppePKsZurAuswahl.clear();
    for (PolitischeGruppe politischeGruppe : politischeGruppenZurAuswahl) {
      politischeGruppePKsZurAuswahl.add(politischeGruppe.getPk());
    }

    notifyEntity();
  }

  public Set<GruppePK> getPolitischeGruppePKsGelost() {
    return politischeGruppePKsGelost;
  }

  public void setPolitischeGruppePKsGelost(Set<GruppePK> politischeGruppePKsGelost) {
    this.politischeGruppePKsGelost.clear();
    this.politischeGruppePKsGelost.addAll(politischeGruppePKsGelost);

    notifyEntity();
  }

  public void setPolitischeGruppenGelost(Collection<PolitischeGruppe> politischeGruppenGelost) {
    this.politischeGruppePKsGelost.clear();
    for (PolitischeGruppe politischeGruppe : politischeGruppenGelost) {
      politischeGruppePKsGelost.add(politischeGruppe.getPk());
    }

    notifyEntity();
  }

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

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

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

    for (ErgebnisKonfliktLosoption losoption : losoptionenGelost) {
      if (losoption.isInstanceOf(Liste.class)) {
        listePks.add((WahlvorschlagPK) losoption.getPk());
      } else if (losoption.isInstanceOf(PolitischeGruppe.class)) {
        politischeGruppePks.add((GruppePK) losoption.getPk());
      }
    }

    setListePKsGelost(listePks);
    setPolitischeGruppePKsGelost(politischeGruppePks);
  }

  /**
   * 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 Object#equals(Object)
   */
  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }

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

    AbstractListenUndPolitischeGruppenAuswaehlenErgebnisKonfliktCustomData that = (AbstractListenUndPolitischeGruppenAuswaehlenErgebnisKonfliktCustomData) o;

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

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

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

    return true;
  }

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

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

    return result;
  }

  @Override
  public void loeseKonfliktAutomatisch() {
    long anzahl = getAnzahlZuLosen();
    Set<GruppePK> politischeGruppePksGelost = waehleDieErstenAus(anzahl, getBeteiligtePolitischeGruppenPKs());
    setPolitischeGruppePKsGelost(politischeGruppePksGelost);

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

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

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

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

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

    setPolitischeGruppePKsGelost(politischeGruppePksGelost);
    setListePKsGelost(listePksGelost);
  }

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

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

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

}
