pokeemerald-expansion/src/ow_abilities.c

#include "global.h"
#include "ow_abilities.h"
#include "pokemon.h"
#include "random.h"
#include "constants/pokemon.h"

const static enum Ability sForceNatureAbilities[] = {ABILITY_SYNCHRONIZE, ABILITY_NONE};
const static enum Ability sForceOppositeGenderAbilities[] = {ABILITY_CUTE_CHARM, ABILITY_NONE};

static UNUSED bool32 HasHalfChance(enum Species species);
static UNUSED bool32 HasTwoThirdsChance(enum Species species);
static UNUSED bool32 IsFalse(enum Species species);
static UNUSED bool32 IsTrue(enum Species species);
static UNUSED bool32 IsTrueIfUndiscoveredEggGroup(enum Species species);

static const bool32 (*const sSynchronizeModes[])(enum Species) = 
{
#if OW_SYNCHRONIZE_NATURE == GEN_3
    [WILDMON_ORIGIN] = HasHalfChance,
    [STATIC_WILDMON_ORIGIN] = IsFalse,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsFalse,
#elif OW_SYNCHRONIZE_NATURE <= GEN_5
    [WILDMON_ORIGIN] = HasHalfChance,
    [STATIC_WILDMON_ORIGIN] = HasHalfChance,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsFalse,
#elif OW_SYNCHRONIZE_NATURE == GEN_6
    [WILDMON_ORIGIN] = HasHalfChance,
    [STATIC_WILDMON_ORIGIN] = HasHalfChance,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsTrueIfUndiscoveredEggGroup,
#elif OW_SYNCHRONIZE_NATURE == GEN_7
    [WILDMON_ORIGIN] = HasHalfChance,
    [STATIC_WILDMON_ORIGIN] = HasHalfChance,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsTrue,
#elif OW_SYNCHRONIZE_NATURE == GEN_8
    [WILDMON_ORIGIN] = IsTrue,
    [STATIC_WILDMON_ORIGIN] = IsFalse,
    [ROAMER_ORIGIN] = IsTrue,
    [GIFTMON_ORIGIN] = IsFalse,
#else
    [WILDMON_ORIGIN] = IsFalse,
    [STATIC_WILDMON_ORIGIN] = IsFalse,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsFalse,
#endif
};

static const bool32 (*const sCuteCharmModes[])(enum Species) = 
{
    [WILDMON_ORIGIN] = HasTwoThirdsChance,
    [STATIC_WILDMON_ORIGIN] = HasTwoThirdsChance,
    [ROAMER_ORIGIN] = IsFalse,
    [GIFTMON_ORIGIN] = IsFalse,
};

static UNUSED bool32 HasHalfChance(enum Species species)
{
    return Random() % 2;
}

static UNUSED bool32 HasTwoThirdsChance(enum Species species)
{
    return Random() % 3;
}

static UNUSED bool32 IsFalse(enum Species species)
{
    return FALSE;
}

static UNUSED bool32 IsTrue(enum Species species)
{
    return TRUE;
}

static UNUSED bool32 IsTrueIfUndiscoveredEggGroup(enum Species species)
{
    return (gSpeciesInfo[species].eggGroups[0] == EGG_GROUP_NO_EGGS_DISCOVERED);
}

bool32 DoesLeadingMonHaveAbilityEffect(const enum Ability *abilityArray)
{
    if (GetMonData(&gParties[B_TRAINER_0][0], MON_DATA_SANITY_IS_EGG))
        return FALSE;
    enum Ability leadingMonAbility = GetMonAbility(&gParties[B_TRAINER_0][0]);
    for (u32 i = 0; abilityArray[i] != ABILITY_NONE; i++)
    {
        if (leadingMonAbility == abilityArray[i])
            return TRUE;
    }
    return FALSE;
}

UNUSED bool32 DoesPartyMemberHaveAbilityEffect(const enum Ability *abilityArray)
{
    for (u32 j = 0; j < gPartiesCount[B_TRAINER_0]; j++)
    {
        if (GetMonData(&gParties[B_TRAINER_0][j], MON_DATA_SANITY_IS_EGG))
            continue;
        enum Ability monAbility = GetMonAbility(&gParties[B_TRAINER_0][j]);
        for (u32 i = 0; abilityArray[i] != ABILITY_NONE; i++)
        {
            if (monAbility == abilityArray[i])
                return TRUE;
        }
    }
    return FALSE;
}

u32 GetSynchronizedNature(enum GeneratedMonOrigin origin, enum Species species)
{
    if (!DoesLeadingMonHaveAbilityEffect(sForceNatureAbilities))
        return NATURE_RANDOM;
    if (!(sSynchronizeModes[origin](species)))
        return NATURE_RANDOM;
    return GetMonData(&gParties[B_TRAINER_0][0], MON_DATA_PERSONALITY) % NUM_NATURES;
}

u32 GetSynchronizedGender(enum GeneratedMonOrigin origin, enum Species species)
{
    if (!DoesLeadingMonHaveAbilityEffect(sForceOppositeGenderAbilities))
        return MON_GENDER_RANDOM;
    if (!(sCuteCharmModes[origin](species)))
        return MON_GENDER_RANDOM;
    u8 leadingMonGender = GetMonGender(&gParties[B_TRAINER_0][0]);
    // misses mon is genderless check, although no genderless mon can have cute charm as ability
    if (leadingMonGender == MON_FEMALE)
        return MON_MALE;
    else
        return MON_FEMALE;
}