EncounterViewModel.java

package com.lateensoft.pathfinder.toolkit.views.encounter;

import com.google.common.collect.Lists;
import com.lateensoft.pathfinder.toolkit.model.IdNamePair;
import com.lateensoft.pathfinder.toolkit.model.party.Encounter;
import com.lateensoft.pathfinder.toolkit.model.party.EncounterParticipant;

import java.util.Collection;
import java.util.Collections;
import java.util.List;

public class EncounterViewModel extends Encounter<EncounterParticipantRowModel> {

    public EncounterViewModel() {
        super("New Encounter");
    }

    public EncounterViewModel(Encounter<EncounterParticipant> encounter) {
        super(encounter.getId(), encounter.getName(), rowModelsForParticipants(encounter));
        setInCombat(encounter.isInCombat());

        if (encounter.getCurrentTurn() != null) {
            setCurrentTurn(getRowWithId(encounter.getCurrentTurn().getId()));
        }

        Collections.sort(this);
    }

    private EncounterParticipantRowModel getRowWithId(long id) {
        for (EncounterParticipantRowModel row : this) {
            if (row.getParticipant().getId() == id) {
                return row;
            }
        }
        return null;
    }

    public void updateTurnOrders() {
        for (int i = 0; i < size(); i++) {
            get(i).getParticipant().setTurnOrder(i);
        }
    }

    private static List<EncounterParticipantRowModel> rowModelsForParticipants(Collection<EncounterParticipant> participants) {
        List<EncounterParticipantRowModel> rows = Lists.newArrayListWithCapacity(participants.size());
        for (EncounterParticipant participant : participants) {
            rows.add(new EncounterParticipantRowModel(participant));
        }
        return rows;
    }

    public Encounter<EncounterParticipant> buildEncounterForModel() {
        List<EncounterParticipant> participants = Lists.newArrayListWithCapacity(this.size());
        for (EncounterParticipantRowModel row : this) {
            participants.add(row.getParticipant());
        }

        Encounter<EncounterParticipant> encounter = new Encounter<EncounterParticipant>(this.getId(), this.getName(), participants);
        encounter.setInCombat(this.isInCombat());
        encounter.setCurrentTurn(this.getCurrentTurn() != null ? this.getCurrentTurn().getParticipant() : null);
        return encounter;
    }

    public List<IdNamePair> getParticipantIds() {
        List<IdNamePair> participantIds = Lists.newArrayList();
        for (EncounterParticipantRowModel row : this) {
            EncounterParticipant participant = row.getParticipant();
            participantIds.add(new IdNamePair(participant.getId(), participant.getName()));
        }
        return participantIds;
    }
}