majority-judgment-library-java/src/main/java/fr/mieuxvoter/mj/ProposalTally.java

package fr.mieuxvoter.mj;

import java.math.BigInteger;
import java.util.Arrays;

public class ProposalTally implements ProposalTallyInterface {

	/**
	 * Amounts of judgments received per grade, from "worst" grade to "best" grade.
	 * Those are BigIntegers because of our LCM-based normalization shenanigans.
	 */
	protected BigInteger[] tally;

	public ProposalTally() {}

	public ProposalTally(String[] tally) {
		setTally(tally);
	}

	public ProposalTally(Integer[] tally) {
		setTally(tally);
	}

	public ProposalTally(Long[] tally) {
		setTally(tally);
	}

	public ProposalTally(BigInteger[] tally) {
		setTally(tally);
	}

	public ProposalTally(ProposalTallyInterface proposalTally) {
		setTally(Arrays.copyOf(proposalTally.getTally(), proposalTally.getTally().length));
	}

	public void setTally(String[] tally) {
		int tallyLength = tally.length;
		BigInteger[] bigTally = new BigInteger[tallyLength];
		for (int i = 0 ; i < tallyLength ; i++) {
			bigTally[i] = new BigInteger(tally[i]);
		}
		setTally(bigTally);
	}

	public void setTally(Integer[] tally) {
		int tallyLength = tally.length;
		BigInteger[] bigTally = new BigInteger[tallyLength];
		for (int i = 0 ; i < tallyLength ; i++) {
			bigTally[i] = BigInteger.valueOf(tally[i]);
		}
		setTally(bigTally);
	}

	public void setTally(Long[] tally) {
		int tallyLength = tally.length;
		BigInteger[] bigTally = new BigInteger[tallyLength];
		for (int i = 0 ; i < tallyLength ; i++) {
			bigTally[i] = BigInteger.valueOf(tally[i]);
		}
		setTally(bigTally);
	}

	public void setTally(BigInteger[] tally) {
		this.tally = tally;
	}

	@Override
	public BigInteger[] getTally() {
		return this.tally;
	}

	@Override
	public ProposalTallyInterface duplicate() {
		return new ProposalTally(Arrays.copyOf(this.tally, this.tally.length));
	}

	@Override
	public void moveJudgments(Integer fromGrade, Integer intoGrade) {
		this.tally[intoGrade] = this.tally[intoGrade].add(this.tally[fromGrade]); 
		this.tally[fromGrade] = BigInteger.ZERO; 
	}

	@Override
	public BigInteger getAmountOfJudgments() {
		BigInteger sum = BigInteger.ZERO;
		int tallyLength = this.tally.length;
		for (int i = 0 ; i < tallyLength ; i++) {
			sum = sum.add(this.tally[i]);
		}
		return sum;
	}

}