#include "common.h"
#include "element.h"
#include "equipment.h"

/* 45480 8016F580 */
u8 equipment_get_field_0x1b(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x1b;
}

/* 45498 8016F598 */
u8 equipment_get_field_0x1c(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x1c;
}

/* 454B0 8016F5B0 */
char* equipment_get_name(EquipmentId index) {
    return EQUIPMENT_DATA[index].name;
}

/* 454C8 8016F5C8 */
EquipmentType equipment_get_type(EquipmentId index) {
    return EQUIPMENT_DATA[index].type;
}

/* 454E0 8016F5E0 */
Element equipment_get_element(EquipmentId index) {
    Element element;

    element = EQUIPMENT_DATA[index].element;
    if (element == ELEMENT_UNKNOWN_0x10) {
        element = CHARACTER_SLOT[0].element;
    }
    return element;
}

/* 45514 8016F614 */
u8 equipment_get_field_0x6(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x6;
}

/* 4552C 8016F62C */
u16 equipment_get_buy_price(EquipmentId index) {
    return EQUIPMENT_DATA[index].buy_price;
}

/* 45544 8016F644 */
s8 equipment_get_field_0x10(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x10;
}

/* 4555C 8016F65C */
s8 equipment_get_strength(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].strength / 20;
    } else {
        ret = EQUIPMENT_DATA[index].strength;
    }
    return ret;
}

/* 455AC 8016F6AC */
s8 equipment_get_vitality(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].vitality / 20;
    } else {
        ret = EQUIPMENT_DATA[index].vitality;
    }
    return ret;
}

/* 455FC 8016F6FC */
s8 equipment_get_intelligence(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].intelligence / 20;
    } else {
        ret = EQUIPMENT_DATA[index].intelligence;
    }
    return ret;
}

/* 4564C 8016F74C */
s8 equipment_get_mentality(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].mentality / 20;
    } else {
        ret = EQUIPMENT_DATA[index].mentality;
    }
    return ret;
}

/* 4569C 8016F79C */
s8 equipment_get_agility(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].agility / 20;
    } else {
        ret = EQUIPMENT_DATA[index].agility;
    }
    return ret;
}

/* 456EC 8016F7EC */
s8 equipment_get_dexterity(EquipmentId index) {
    s8 ret;
    if (index == EQUIPMENT_ID_GALLANT_DOLL) {
        ret = CHARACTER_SLOT[0].dexterity / 20;
    } else {
        ret = EQUIPMENT_DATA[index].dexterity;
    }
    return ret;
}

/* 4573C 8016F83C */
s8 equipment_get_resistance_strike(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_strike;
}

/* 45754 8016F854 */
s8 equipment_get_resistance_wind(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_wind;
}

/* 4576C 8016F86C */
s8 equipment_get_resistance_fire(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_fire;
}

/* 45784 8016F884 */
s8 equipment_get_resistance_earth(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_earth;
}

/* 4579C 8016F89C */
s8 equipment_get_resistance_water(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_water;
}

/* 457B4 8016F8B4 */
s8 equipment_get_resistance_holy(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_holy;
}

/* 457CC 8016F8CC */
s8 equipment_get_resistance_dark(EquipmentId index) {
    return EQUIPMENT_DATA[index].res_dark;
}

/* 457E4 8016F8E4 */
u8 equipment_get_field_0x18_b68(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x18 >> 6;
}

/* 45800 8016F900 */
u8 equipment_get_field_0x18_b46(EquipmentId index) {
    return (EQUIPMENT_DATA[index].field_0x18 >> 4) & 3;
}

/* 45820 8016F920 */
u8 equipment_get_field_0x18_b24(EquipmentId index) {
    return (EQUIPMENT_DATA[index].field_0x18 >> 2) & 3;
}

/* 45840 8016F940 */
u8 equipment_get_field_0x18_b12(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x18 & 3;
}

/* 4585C 8016F95C */
u8 equipment_get_field_0x19_b68(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x19 >> 6;
}

/* 45878 8016F978 */
u8 equipment_get_field_0x19_b46(EquipmentId index) {
    return (EQUIPMENT_DATA[index].field_0x19 >> 4) & 3;
}

/* 45898 8016F998 */
u8 equipment_get_field_0x19_b24(EquipmentId index) {
    return (EQUIPMENT_DATA[index].field_0x19 >> 2) & 3;
}

/* 458B8 8016F9B8 */
u8 equipment_get_field_0x19_b12(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x19 & 3;
}

/* 458D4 8016F9D4 */
u8 equipment_get_field_0x1a_b7(EquipmentId index) {
    return EQUIPMENT_DATA[index].field_0x1a >> 7;
}

/* 458F0 8016F9F0 */
bool equipment_is_weapon(EquipmentId index) {
    s32 type = EQUIPMENT_DATA[index].type;
    
    if (type != 0) {
        if (type < EQUIPMENT_TYPE_SMALL_SHIELD) {
            return true;
        }
        if (type == EQUIPMENT_TYPE_FAN) {
            return true;
        }
    }

    return false;
}

INCLUDE_ASM(const s32, "equipment", equipment_get_first_weapon);

/* 45A50 8016FB50 */
EquipmentId equipment_get_first_spellbook(EquipmentId arg0, EquipmentId arg1, EquipmentId arg2, EquipmentId arg3) {    
    if (EQUIPMENT_DATA[arg0].type == EQUIPMENT_TYPE_SPELLBOOK)
        return arg0;
    
    if (EQUIPMENT_DATA[arg1].type == EQUIPMENT_TYPE_SPELLBOOK)
        return arg1;
        
    if (EQUIPMENT_DATA[arg2].type == EQUIPMENT_TYPE_SPELLBOOK)
        return arg2;
    
    if (EQUIPMENT_DATA[arg3].type == EQUIPMENT_TYPE_SPELLBOOK)
        return arg3;
    
    return 0;
}