com.softwaremagico.tm.character.equipment.weapons.RandomWeapon 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.weapons;

/*-
 * #%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%
 */

import java.util.Collection;
import java.util.Set;

import com.softwaremagico.tm.InvalidXmlElementException;
import com.softwaremagico.tm.character.CharacterPlayer;
import com.softwaremagico.tm.character.equipment.EquipmentSelector;
import com.softwaremagico.tm.log.RandomGenerationLog;
import com.softwaremagico.tm.random.exceptions.InvalidRandomElementSelectedException;
import com.softwaremagico.tm.random.selectors.DifficultLevelPreferences;
import com.softwaremagico.tm.random.selectors.IRandomPreference;

public abstract class RandomWeapon extends EquipmentSelector {

	protected RandomWeapon(CharacterPlayer characterPlayer, Set preferences,
			Set mandatoryWeapons) throws InvalidXmlElementException {
		super(characterPlayer, preferences, mandatoryWeapons);
	}

	@Override
	public void assign() throws InvalidRandomElementSelectedException {
		final Weapon selectedWeapon = selectElementByWeight();
		if (!getCharacterPlayer().getAllWeapons().contains(selectedWeapon)) {
			getCharacterPlayer().addWeapon(selectedWeapon);
			RandomGenerationLog.info(this.getClass().getName(), "Selected weapon: " + selectedWeapon);
		}
	}

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

	protected abstract Set weaponTypesFilter();

	/**
	 * Similar tech level weapons preferred.
	 * 
	 * @param weapon
	 * @return
	 */
	protected abstract int getWeightTechModificator(Weapon weapon);

	@Override
	protected int getWeight(Weapon weapon) throws InvalidRandomElementSelectedException {
		super.getWeight(weapon);

		// Preferences

		// Only ranged weapons.
		if (!weaponTypesFilter().contains(weapon.getType())) {
			throw new InvalidRandomElementSelectedException("Only weapons of type '" + weaponTypesFilter()
					+ "' are accepted. Weapon '" + weapon + "' has not this type.");
		}

		int weight = 0;
		// Similar tech level preferred.
		final int weightTech = getWeightTechModificator(weapon);
		RandomGenerationLog.debug(this.getClass().getName(),
				"Weight value by tech level for '" + weapon + "' is '" + weightTech + "'.");
		weight += weightTech;

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

		// Skill modifications.
		final int skillMultiplier = getCharacterPlayer().getSkillTotalRanks(weapon.getSkill());
		if (skillMultiplier > 0) {
			RandomGenerationLog.debug(this.getClass().getName(),
					"Skill multiplication for weight for '" + weapon + "' is '" + skillMultiplier + "'.");
			weight = (weight + skillMultiplier) * skillMultiplier;
		}

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

	@Override
	public void validateElement(Weapon weapon) throws InvalidRandomElementSelectedException {
		super.validateElement(weapon);

		// Difficulty modification
		final DifficultLevelPreferences preference = DifficultLevelPreferences.getSelected(getPreferences());
		switch (preference) {
		case VERY_EASY:
			if (weapon.getMainDamage() >= 3) {
				throw new InvalidRandomElementSelectedException("Weapons with a damage '" + weapon.getDamage()
						+ "' are not allowed by selected preference '" + preference + "'.");
			}
			break;
		case EASY:
			if (weapon.getMainDamage() >= 5) {
				throw new InvalidRandomElementSelectedException("Weapons with a damage '" + weapon.getDamage()
						+ "' are not allowed by selected preference '" + preference + "'.");
			}
			break;
		case MEDIUM:
			if (weapon.getMainDamage() >= 8) {
				throw new InvalidRandomElementSelectedException("Weapons with a damage '" + weapon.getDamage()
						+ "' are not allowed by selected preference '" + preference + "'.");
			}
			break;
		case HARD:
			if (weapon.getMainDamage() <= 3) {
				throw new InvalidRandomElementSelectedException("Weapons with a damage '" + weapon.getDamage()
						+ "' are not allowed by selected preference '" + preference + "'.");
			}
			break;
		case VERY_HARD:
			if (weapon.getMainDamage() <= 4 || weapon.getDamageTypes().isEmpty()) {
				throw new InvalidRandomElementSelectedException("Weapons with a damage '" + weapon.getDamage()
						+ "' or without damage perforation are not allowed by selected preference '" + preference
						+ "'.");
			}
			break;
		}
	}

	@Override
	protected void assignIfMandatory(Weapon weapon) throws InvalidXmlElementException {
		return;
	}

	@Override
	protected void assignMandatoryValues(Set mandatoryValues) throws InvalidXmlElementException {
		for (final Weapon weapon : mandatoryValues) {
			if (weaponTypesFilter().contains(weapon.getType())) {
				getCharacterPlayer().addWeapon(weapon);
			}
		}
	}
}