372 lines
11 KiB
Java
372 lines
11 KiB
Java
/*
|
|
* Pixel Dungeon
|
|
* Copyright (C) 2012-2014 Oleg Dolya
|
|
*
|
|
* 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 3 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 <http://www.gnu.org/licenses/>
|
|
*/
|
|
package com.shatteredpixel.shatteredpixeldungeon.items;
|
|
|
|
import com.shatteredpixel.shatteredpixeldungeon.Dungeon;
|
|
import com.shatteredpixel.shatteredpixeldungeon.actors.hero.Hero;
|
|
import com.shatteredpixel.shatteredpixeldungeon.actors.mobs.npcs.Wandmaker.Rotberry;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.armor.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.artifacts.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.bags.Bag;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.food.Food;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.food.MysteryMeat;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.food.Pasty;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.potions.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.rings.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.scrolls.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.wands.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.weapon.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.weapon.melee.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.items.weapon.missiles.*;
|
|
import com.shatteredpixel.shatteredpixeldungeon.plants.*;
|
|
import com.watabou.utils.Bundle;
|
|
import com.watabou.utils.Random;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.HashMap;
|
|
|
|
public class Generator {
|
|
|
|
public static enum Category {
|
|
WEAPON ( 150, Weapon.class ),
|
|
ARMOR ( 100, Armor.class ),
|
|
POTION ( 500, Potion.class ),
|
|
SCROLL ( 400, Scroll.class ),
|
|
WAND ( 40, Wand.class ),
|
|
RING ( 15, Ring.class ),
|
|
ARTIFACT( 20, Artifact.class),
|
|
SEED ( 50, Plant.Seed.class ),
|
|
FOOD ( 0, Food.class ),
|
|
GOLD ( 500, Gold.class );
|
|
|
|
public Class<?>[] classes;
|
|
public float[] probs;
|
|
|
|
public float prob;
|
|
public Class<? extends Item> superClass;
|
|
|
|
private Category( float prob, Class<? extends Item> superClass ) {
|
|
this.prob = prob;
|
|
this.superClass = superClass;
|
|
}
|
|
|
|
public static int order( Item item ) {
|
|
for (int i=0; i < values().length; i++) {
|
|
if (values()[i].superClass.isInstance( item )) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return item instanceof Bag ? Integer.MAX_VALUE : Integer.MAX_VALUE - 1;
|
|
}
|
|
};
|
|
|
|
private static HashMap<Category,Float> categoryProbs = new HashMap<Generator.Category, Float>();
|
|
|
|
static {
|
|
|
|
Category.GOLD.classes = new Class<?>[]{
|
|
Gold.class };
|
|
Category.GOLD.probs = new float[]{ 1 };
|
|
|
|
Category.SCROLL.classes = new Class<?>[]{
|
|
ScrollOfIdentify.class,
|
|
ScrollOfTeleportation.class,
|
|
ScrollOfRemoveCurse.class,
|
|
ScrollOfUpgrade.class,
|
|
ScrollOfRecharging.class,
|
|
ScrollOfMagicMapping.class,
|
|
ScrollOfRage.class,
|
|
ScrollOfTerror.class,
|
|
ScrollOfLullaby.class,
|
|
ScrollOfWeaponUpgrade.class,
|
|
ScrollOfPsionicBlast.class,
|
|
ScrollOfMirrorImage.class };
|
|
Category.SCROLL.probs = new float[]{ 30, 10, 15, 0, 10, 15, 12, 8, 8, 0, 3, 6 };
|
|
|
|
Category.POTION.classes = new Class<?>[]{
|
|
PotionOfHealing.class,
|
|
PotionOfExperience.class,
|
|
PotionOfToxicGas.class,
|
|
PotionOfParalyticGas.class,
|
|
PotionOfLiquidFlame.class,
|
|
PotionOfLevitation.class,
|
|
PotionOfStrength.class,
|
|
PotionOfMindVision.class,
|
|
PotionOfPurity.class,
|
|
PotionOfInvisibility.class,
|
|
PotionOfMight.class,
|
|
PotionOfFrost.class };
|
|
Category.POTION.probs = new float[]{ 45, 4, 15, 10, 15, 10, 0, 20, 12, 10, 0, 10 };
|
|
|
|
Category.WAND.classes = new Class<?>[]{
|
|
WandOfTeleportation.class,
|
|
WandOfSlowness.class,
|
|
WandOfFirebolt.class,
|
|
WandOfRegrowth.class,
|
|
WandOfPoison.class,
|
|
WandOfBlink.class,
|
|
WandOfLightning.class,
|
|
WandOfAmok.class,
|
|
WandOfTelekinesis.class,
|
|
WandOfFlock.class,
|
|
WandOfMagicMissile.class,
|
|
WandOfDisintegration.class,
|
|
WandOfAvalanche.class };
|
|
Category.WAND.probs = new float[]{ 10, 10, 15, 6, 10, 11, 15, 10, 6, 10, 0, 5, 5 };
|
|
|
|
Category.WEAPON.classes = new Class<?>[]{
|
|
Dagger.class,
|
|
Knuckles.class,
|
|
Quarterstaff.class,
|
|
Spear.class,
|
|
Mace.class,
|
|
Sword.class,
|
|
Longsword.class,
|
|
BattleAxe.class,
|
|
WarHammer.class,
|
|
Glaive.class,
|
|
ShortSword.class,
|
|
Dart.class,
|
|
Javelin.class,
|
|
IncendiaryDart.class,
|
|
CurareDart.class,
|
|
Shuriken.class,
|
|
Boomerang.class,
|
|
Tamahawk.class };
|
|
Category.WEAPON.probs = new float[]{ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1 };
|
|
|
|
Category.ARMOR.classes = new Class<?>[]{
|
|
ClothArmor.class,
|
|
LeatherArmor.class,
|
|
MailArmor.class,
|
|
ScaleArmor.class,
|
|
PlateArmor.class };
|
|
Category.ARMOR.probs = new float[]{ 1, 1, 1, 1, 1 };
|
|
|
|
Category.FOOD.classes = new Class<?>[]{
|
|
Food.class,
|
|
Pasty.class,
|
|
MysteryMeat.class };
|
|
Category.FOOD.probs = new float[]{ 4, 1, 0 };
|
|
|
|
Category.RING.classes = new Class<?>[]{
|
|
RingOfAccuracy.class,
|
|
RingOfEvasion.class,
|
|
RingOfElements.class,
|
|
RingOfForce.class,
|
|
RingOfFuror.class,
|
|
RingOfHaste.class,
|
|
RingOfMagic.class,
|
|
RingOfMight.class,
|
|
RingOfSharpshooting.class,
|
|
RingOfTenacity.class,
|
|
RingOfWealth.class};
|
|
Category.RING.probs = new float[]{ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 };
|
|
|
|
Category.ARTIFACT.classes = new Class<?>[]{
|
|
CapeOfThorns.class,
|
|
ChaliceOfBlood.class,
|
|
CloakOfShadows.class,
|
|
HornOfPlenty.class,
|
|
MasterThievesArmband.class,
|
|
SandalsOfNature.class,
|
|
TalismanOfForesight.class};
|
|
Category.ARTIFACT.probs = new float[]{ 0, 1, 1, 1, 0, 1, 1 };
|
|
|
|
Category.SEED.classes = new Class<?>[]{
|
|
Firebloom.Seed.class,
|
|
Icecap.Seed.class,
|
|
Sorrowmoss.Seed.class,
|
|
Blindweed.Seed.class,
|
|
Sungrass.Seed.class,
|
|
Earthroot.Seed.class,
|
|
Fadeleaf.Seed.class,
|
|
Rotberry.Seed.class,
|
|
BlandfruitBush.Seed.class,
|
|
Dreamfoil.Seed.class,
|
|
Stormvine.Seed.class};
|
|
Category.SEED.probs = new float[]{ 3, 3, 3, 3, 3, 3, 3, 0, 1, 3, 3 };
|
|
}
|
|
|
|
public static void reset() {
|
|
for (Category cat : Category.values()) {
|
|
categoryProbs.put( cat, cat.prob );
|
|
}
|
|
}
|
|
|
|
public static Item random() {
|
|
return random( Random.chances( categoryProbs ) );
|
|
}
|
|
|
|
public static Item random( Category cat ) {
|
|
try {
|
|
|
|
categoryProbs.put( cat, categoryProbs.get( cat ) / 2 );
|
|
|
|
switch (cat) {
|
|
case ARMOR:
|
|
return randomArmor();
|
|
case WEAPON:
|
|
return randomWeapon();
|
|
case ARTIFACT:
|
|
Item item = randomArtifact();
|
|
//if we're out of artifacts, return a ring instead.
|
|
return item != null ? item : random(Category.RING);
|
|
default:
|
|
return ((Item)cat.classes[Random.chances( cat.probs )].newInstance()).random();
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
|
|
return null;
|
|
|
|
}
|
|
}
|
|
|
|
public static Item random( Class<? extends Item> cl ) {
|
|
try {
|
|
|
|
return ((Item)cl.newInstance()).random();
|
|
|
|
} catch (Exception e) {
|
|
|
|
return null;
|
|
|
|
}
|
|
}
|
|
|
|
public static Armor randomArmor(){
|
|
int curStr = Hero.STARTING_STR + Dungeon.potionOfStrength;
|
|
|
|
return randomArmor(curStr);
|
|
}
|
|
|
|
public static Armor randomArmor(int targetStr) {
|
|
|
|
Category cat = Category.ARMOR;
|
|
|
|
try {
|
|
Armor a1 = (Armor) cat.classes[Random.chances(cat.probs)].newInstance();
|
|
Armor a2 = (Armor) cat.classes[Random.chances(cat.probs)].newInstance();
|
|
|
|
a1.random();
|
|
a2.random();
|
|
|
|
return Math.abs(targetStr - a1.STR) < Math.abs(targetStr - a2.STR) ? a1 : a2;
|
|
} catch (Exception e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static Weapon randomWeapon(){
|
|
int curStr = Hero.STARTING_STR + Dungeon.potionOfStrength;
|
|
|
|
return randomWeapon(curStr);
|
|
}
|
|
|
|
public static Weapon randomWeapon(int targetStr) {
|
|
|
|
try {
|
|
Category cat = Category.WEAPON;
|
|
|
|
Weapon w1 = (Weapon)cat.classes[Random.chances( cat.probs )].newInstance();
|
|
Weapon w2 = (Weapon)cat.classes[Random.chances( cat.probs )].newInstance();
|
|
|
|
w1.random();
|
|
w2.random();
|
|
|
|
return Math.abs( targetStr - w1.STR ) < Math.abs( targetStr - w2.STR ) ? w1 : w2;
|
|
} catch (Exception e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
//enforces uniqueness of artifacts throughout a run.
|
|
public static Artifact randomArtifact() {
|
|
|
|
try {
|
|
Category cat = Category.ARTIFACT;
|
|
int i = Random.chances( cat.probs );
|
|
|
|
//if no artifacts are left, return null
|
|
if (i == -1){
|
|
return null;
|
|
}
|
|
|
|
Artifact artifact = (Artifact)cat.classes[i].newInstance();
|
|
|
|
//remove the chance of spawning this artifact.
|
|
cat.probs[i] = 0;
|
|
spawnedArtifacts.add(cat.classes[i].getSimpleName());
|
|
|
|
artifact.random();
|
|
|
|
return artifact;
|
|
|
|
} catch (Exception e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static boolean removeArtifact(Artifact artifact) {
|
|
if (spawnedArtifacts.contains(artifact.getClass().getSimpleName()))
|
|
return false;
|
|
|
|
Category cat = Category.ARTIFACT;
|
|
for (int i = 0; i < cat.classes.length; i++)
|
|
if (cat.classes[i].equals(artifact.getClass())) {
|
|
cat.probs[i] = 0;
|
|
spawnedArtifacts.add(artifact.getClass().getSimpleName());
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
//resets artifact probabilities, for new dungeons
|
|
public static void initArtifacts() {
|
|
Category.ARTIFACT.probs = new float[]{ 0, 1, 1, 1, 0, 1, 1 };
|
|
spawnedArtifacts = new ArrayList<String>();
|
|
}
|
|
|
|
private static ArrayList<String> spawnedArtifacts = new ArrayList<String>();
|
|
|
|
private static final String ARTIFACTS = "artifacts";
|
|
|
|
//used to store information on which artifacts have been spawned.
|
|
public static void storeInBundle(Bundle bundle) {
|
|
bundle.put( ARTIFACTS, spawnedArtifacts.toArray(new String[spawnedArtifacts.size()]));
|
|
}
|
|
|
|
public static void restoreFromBundle(Bundle bundle) {
|
|
initArtifacts();
|
|
|
|
if (bundle.contains(ARTIFACTS)) {
|
|
Collections.addAll(spawnedArtifacts, bundle.getStringArray(ARTIFACTS));
|
|
Category cat = Category.ARTIFACT;
|
|
|
|
for (String artifact : spawnedArtifacts)
|
|
for (int i = 0; i < cat.classes.length; i++)
|
|
if (cat.classes[i].getSimpleName().equals(artifact))
|
|
cat.probs[i] = 0;
|
|
}
|
|
}
|
|
}
|