com.softwaremagico.tm.character.equipment.shields.RandomShield Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of think-machine-random Show documentation
Think Machine - A Fading Suns character generator (random character creation)
There is a newer version: 0.10.5
package com.softwaremagico.tm.character.equipment.shields;

import com.softwaremagico.tm.InvalidXmlElementException;
import com.softwaremagico.tm.character.CharacterPlayer;
import com.softwaremagico.tm.character.equipment.EquipmentSelector;
import com.softwaremagico.tm.character.exceptions.RestrictedElementException;
import com.softwaremagico.tm.character.exceptions.UnofficialElementNotAllowedException;
import com.softwaremagico.tm.log.RandomGenerationLog;
import com.softwaremagico.tm.random.exceptions.InvalidCostElementSelectedException;
import com.softwaremagico.tm.random.exceptions.InvalidRandomElementSelectedException;
import com.softwaremagico.tm.random.selectors.CombatPreferences;
import com.softwaremagico.tm.random.selectors.DifficultLevelPreferences;
import com.softwaremagico.tm.random.selectors.IRandomPreference;
import com.softwaremagico.tm.random.selectors.ShieldPreferences;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

/*-
 * #%L
 * Think Machine (Core)
 * %%
 * Copyright (C) 2017 - 2018 Softwaremagico
 * %%
 * This software is designed by Jorge Hortelano Otero. Jorge Hortelano Otero
 *  Valencia (Spain).
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option) any later
 * version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; If not, see .
 * #L%
 */

public class RandomShield extends EquipmentSelector {

    public RandomShield(CharacterPlayer characterPlayer, Set> preferences,
                        Set mandatoryShields) throws InvalidXmlElementException, RestrictedElementException,
            UnofficialElementNotAllowedException {
        super(characterPlayer, preferences, mandatoryShields);
    }

    @Override
    public void assign() throws InvalidRandomElementSelectedException, InvalidShieldException, UnofficialElementNotAllowedException {
        final ShieldPreferences shieldPreferences = ShieldPreferences.getSelected(getPreferences());
        if (shieldPreferences != null && random.nextFloat() < shieldPreferences.getShieldProbability()) {
            final Shield selectedShield = selectElementByWeight();
            if (getCharacterPlayer().getShield() == null) {
                if (selectedShield.getCost() > getCharacterPlayer().getMoney()) {
                    removeElementWeight(selectedShield);
                    throw new InvalidCostElementSelectedException(selectedShield, "Shield '" + selectedShield + "' has a cost of '" +
                            selectedShield.getCost() + "'. Current money is '" + getCharacterPlayer().getMoney() + "'.");
                }
                getCharacterPlayer().setShield(selectedShield);
                RandomGenerationLog.info(this.getClass().getName(), "Selected shield '{}'.", selectedShield);
            }
        }
    }

    @Override
    protected Collection getAllElements() throws InvalidXmlElementException {
        return ShieldFactory.getInstance().getElements(getCharacterPlayer().getLanguage(), getCharacterPlayer().getModuleName());
    }

    /**
     * Not so expensive shields.
     *
     * @param shield the shield
     * @return the weight
     */
    @Override
    protected int getWeightCostModificator(Shield shield) throws InvalidRandomElementSelectedException {
        if (shield.getCost() > getCurrentMoney()) {
            throw new InvalidRandomElementSelectedException("Not enough money!");
        } else if (shield.getCost() > getCurrentMoney() / (double) 2) {
            return 5;
        } else {
            return 1;
        }
    }

    @Override
    protected int getWeight(Shield shield) throws InvalidRandomElementSelectedException {
        int weight = super.getWeight(shield);

        // Difficulty modification
        final DifficultLevelPreferences preference = DifficultLevelPreferences.getSelected(getPreferences());
        switch (preference) {
            case VERY_EASY:
            case EASY:
            case MEDIUM:
                break;
            case HARD:
            case VERY_HARD:
                weight *= shield.getHits();
                break;
        }

        // More protection is better.
        if (getPreferences().contains(CombatPreferences.BELLIGERENT)) {
            weight *= shield.getHits();
            RandomGenerationLog.debug(this.getClass().getName(),
                    "Protection multiplier for '" + shield + "' is '" + shield.getHits() + "'.");
        }

        // shields depending on the purchasing power of the character.
        final int costModificator = getWeightCostModificator(shield);
        RandomGenerationLog.debug(this.getClass().getName(),
                "Cost multiplication for weight for '" + shield + "' is '" + costModificator + "'.");
        weight /= (double) costModificator;

        RandomGenerationLog.debug(this.getClass().getName(), "Total weight for '{}' is '{}'.", shield, weight);
        return weight;
    }

    @Override
    public void validateElement(Shield shield) throws InvalidRandomElementSelectedException {
        super.validateElement(shield);

        // Difficulty modification
        final DifficultLevelPreferences preference = DifficultLevelPreferences.getSelected(getPreferences());
        switch (preference) {
            case VERY_EASY:
            case EASY:
                throw new InvalidRandomElementSelectedException(
                        "Shield '" + shield + "' are not allowed by selected preference '" + preference + "'.");
            default:
                break;
        }
    }

    @Override
    protected void assignIfMandatory(Shield element) throws InvalidXmlElementException {
        //Nothing
    }

    @Override
    protected void assignMandatoryValues(Set mandatoryValues) throws InvalidXmlElementException, UnofficialElementNotAllowedException {
        // We only assign the most expensive one.
        if (!mandatoryValues.isEmpty()) {
            final List sortedShields = new ArrayList<>(mandatoryValues);
            sortedShields.sort((shield0, shield1) -> Float.compare(shield0.getCost(), shield1.getCost()));
            getCharacterPlayer().setShield(sortedShields.get(0));
        }
    }
}