veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00
Code Issues Projects Releases Wiki Activity

Fix up breeding

Browse source
This commit is contained in:
Petr Viktorin 2011-04-27 00:51:40 +03:00
parent 8dee48d966
commit 4f8108d61b
1 changed files with 62 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

68
pokedex/util/movesets.py
View file

@ -292,7 +292,7 @@ class MovesetSearch(object):
self.evolution_moves[evolution_chain] = move required for evolution
self.babies[egg_group_id] = set of baby pokemon
self.hatch_counters[pokemon] = hatch counter
self.gender_rates[pokemon] = gender rate
self.gender_rates[evolution_chain] = gender rate
"""
eg1 = tables.PokemonEggGroup
eg2 = aliased(tables.PokemonEggGroup)
@ -324,7 +324,7 @@ class MovesetSearch(object):
item_baby_chains = set() # evolution chains with baby-trigger items
for pokemon, evolution_chain, parent, g1, g2, baby_item, hatch_counter, gender_rate in query:
self.hatch_counters[pokemon] = hatch_counter
self.gender_rates[pokemon] = gender_rate
self.gender_rates[evolution_chain] = gender_rate
if g1 in bad_groups:
unbreedable[pokemon] = evolution_chain
else:
@ -483,9 +483,9 @@ class MovesetSearch(object):
return success
for group in goal_egg_groups:
handle(group, self.goal_moves, ())
for moves in powerset(self.goal_moves.difference(self.egg_moves)):
for moves in powerset(self.goal_moves):
if moves:
breeds_required[group][frozenset(moves) | self.egg_moves] = 1
breeds_required[group][frozenset(moves)] = 1
self.breeds_required = breeds_required
if self.debug:
@ -567,7 +567,7 @@ class MovesetSearch(object):
###
default_costs = {
# Costs for learning a move in verious ways
# Costs for learning a move in various ways
'level-up': 20, # The normal way
'machine': 40, # Machines are slightly inconvenient.
'machine-once': 2000, # before gen. 5, TMs only work once. Avoid.
@ -592,7 +592,7 @@ default_costs = {
'form-change': 100,
# Other actions.
# Breeding should cost more than 3 times than a lv-up/machine/tutor move.
# Breeding should cost more than 3 times a lv-up/machine/tutor move.
'evolution': 100, # We have to do this anyway, usually.
'evolution-delayed': 50, # *in addition* to evolution. Who wants to mash B on every level.
'breed': 400, # Breeding's a pain.
@ -620,7 +620,7 @@ default_costs = {
class Facade(object):
@property
def pokemon(self):
return self.search.get_by_id(tables.Pokemon, self.pokemon_)
return self.search.session.query(tables.Pokemon).filter_by(id=self.pokemon_).one()
@property
def version_group(self):
@ -768,7 +768,7 @@ class PokemonNode(Node, Facade, namedtuple('PokemonNode',
level_difference = level - self.level
if level_difference > 0 or (
level_difference == 0 and self.new_level):
cost += level_difference * search.costs['per-level']
cost += level - self.level * search.costs['per-level']
yield self._learn(move, method, cost,
level=level, new_level=True)
else:
@ -870,14 +870,29 @@ class PokemonNode(Node, Facade, namedtuple('PokemonNode',
cost = search.costs['breed']
cost += search.costs['egg'] * len(moves)
cost += search.costs['breed-penalty'] * len(search.egg_moves - moves)
gender_rate = search.gender_rates[evo_chain]
if 0 <= gender_rate:
# Only pokemon that have males can pas down moves to other species
# (and the other species must have females: checked in BreedNode)
for group in egg_groups:
if moves in breeds_required[group]:
yield cost, None, BreedNode(search=self.search, dummy='breed',
yield cost, None, BreedNode(search=self.search, dummy='b',
group_=group, version_group_=self.version_group_,
moves_=self.moves_)
# Since the target family is not included in our breed graph, we
# breed with it explicitly. But again, there must be a female to
# breed with.
if search.gender_rates[search.goal_evolution_chain] > 0:
yield cost, None, GoalBreedNode(search=self.search, dummy='g',
version_group_=self.version_group_, moves_=self.moves_)
elif evo_chain == search.goal_evolution_chain:
# Single-gender & genderless pokemon can pass on moves via
# breeding with Ditto, to produce the same species again. Obviously
# this is only useful when breeding the goal species.
yield cost, None, GoalBreedNode(search=self.search, dummy='g',
version_group_=self.version_group_, moves_=self.moves_)
class BreedNode(Node, namedtuple('BreedNode',
'search dummy group_ version_group_ moves_')):
class BaseBreedNode(Node):
"""Breed node
This serves to prevent duplicate breeds, by storing only the needed info
in the namedtuple.
@ -889,13 +904,12 @@ class BreedNode(Node, namedtuple('BreedNode',
vg = self.version_group_
gen = search.generation_id_by_version_group[vg]
hatch_level = 5 if (gen < 4) else 1
for baby in search.babies[self.group_]:
for baby in self.babies():
bred_moves = self.moves_
gender_rate = search.gender_rates[baby]
baby_chain = search.evolution_chains[baby]
if bred_moves.issubset(search.movepools[baby_chain]):
moves = search.pokemon_moves[baby][vg]
if not bred_moves.issubset(moves):
continue
if len(bred_moves) < 4:
moves = search.pokemon_moves[baby][self.version_group_]
for move, methods in moves.items():
if 'light-ball-pichu' in methods:
bred_moves.add(move)
@ -903,8 +917,6 @@ class BreedNode(Node, namedtuple('BreedNode',
yield 0, BreedAction(self.search, baby, bred_moves), PokemonNode(
search=self.search, pokemon_=baby, level=hatch_level,
version_group_=vg, moves_=bred_moves, new_level=True)
return
yield
@property
def pokemon(self):
@ -913,6 +925,26 @@ class BreedNode(Node, namedtuple('BreedNode',
def estimate(self, g):
return 0
class BreedNode(BaseBreedNode, namedtuple('BreedNode',
'search dummy group_ version_group_ moves_')):
def babies(self):
search = self.search
for baby in search.babies[self.group_]:
baby_chain = search.evolution_chains[baby]
if self.moves_.issubset(search.movepools[baby_chain]) and (
search.gender_rates[baby_chain] > 0):
yield baby
class GoalBreedNode(BaseBreedNode, namedtuple('InbreedNode',
'search dummy version_group_ moves_')):
def babies(self):
search = self.search
goal_family = search.goal_evolution_chain
group = search.egg_groups[goal_family][0]
for baby in search.pokemon_by_evolution_chain[goal_family]:
if baby in search.babies[group]:
yield baby
class GoalNode(PokemonNode):
def expand(self):
return ()