npc_solo3v3.h

/*
 *
 * Copyright (C) 2014 Teiby
 * Written by Teiby <http://www.teiby.de/>
 *
 */

#ifndef SOLO_3V3_H
#define SOLO_3V3_H

// SOLO_3V3_TALENTS found in: TalentTab.dbc -> TalentTabID


const uint32 SOLO_3V3_TALENTS_MELEE[] =
{
	182, // assasination
	398, // blood
	399, // frost DK
	746, // arms
	181, // combat
	263, // enhancement
	750, // feral
	815, // fury
	839, // protection (warrior)
	183, // subtlety
	400, // unholy
	845, // protection (paladin)
	855, // retribution
	261, // elemental
	811, // beast mastery
	807, // marskmanship
	809, // survival
	752, // balance
	799, // arcane
	871, // affliction
	851, // fire
	867, // demonology
	795, // shadow
	823, // frost mage
	865, // destruction
	0 // End
};

const uint32 SOLO_3V3_TALENTS_CASTER[] =
{
	831, // holy (paladin)
	813, // holy (priest)
	760, // discipline
	262, // restoration (shaman)
	748, // restoration (druid)
	0 // End
};

const uint32 SOLO_3V3_TALENTS_HEALER[] =
{
	831, // holy (paladin)
	813, // holy (priest)
	760, // discipline
	262, // restoration (shaman)
	748, // restoration (druid)
	0 // End
};

enum Solo3v3TalentCat
{
    MELEE = 0,
    CASTER,
    HEALER,
    MAX_TALENT_CAT
};

// TalentTab.dbc -> TalentTabID
const uint32 FORBIDDEN_TALENTS_IN_1V1_ARENA[] =
{
    // Healer
    201, // PriestDiscipline
    202, // PriestHoly
    382, // PaladinHoly
    262, // ShamanRestoration
    282, // DruidRestoration

    // Tanks
    //383, // PaladinProtection
    //163, // WarriorProtection

    0 // End
};

// Returns MELEE, CASTER or HEALER (depends on talent builds)
static Solo3v3TalentCat GetTalentCatForSolo3v3(Player* player)
{
    if (!player || sWorld->getBoolConfig(CONFIG_SOLO_3V3_FILTER_TALENTS) == false)
        return MELEE;

    uint32 count[MAX_TALENT_CAT];
    for (int i = 0; i < MAX_TALENT_CAT; i++)
        count[i] = 0;

    for (uint32 talentId = 0; talentId < sTalentStore.GetNumRows(); ++talentId)
    {
        TalentEntry const* talentInfo = sTalentStore.LookupEntry(talentId);

        if (!talentInfo)
            continue;

        for (int8 rank = MAX_TALENT_RANK-1; rank >= 0; --rank)
        {
            if (talentInfo->RankID[rank] == 0)
                continue;

            if (player->HasTalent(talentInfo->RankID[rank], player->GetActiveSpec()))
            {
                for (int8 i = 0; SOLO_3V3_TALENTS_MELEE[i] != 0; i++)
                    if (SOLO_3V3_TALENTS_MELEE[i] == talentInfo->TalentTab)
                        count[MELEE] += rank + 1;

                for (int8 i = 0; SOLO_3V3_TALENTS_CASTER[i] != 0; i++)
                    if (SOLO_3V3_TALENTS_CASTER[i] == talentInfo->TalentTab)
                        count[CASTER] += rank + 1;

                for (int8 i = 0; SOLO_3V3_TALENTS_HEAL[i] != 0; i++)
                    if (SOLO_3V3_TALENTS_HEAL[i] == talentInfo->TalentTab)
                        count[HEALER] += rank + 1;
            }
        }
    }


    uint32 prevCount = 0;
    Solo3v3TalentCat talCat = MELEE; // Default MELEE (if no talent points set)
    for (int i = 0; i < MAX_TALENT_CAT; i++)
    {
        if (count[i] > prevCount)
        {
            talCat = (Solo3v3TalentCat)i;
            prevCount = count[i];
        }
    }

    return talCat;
}


// Return false, if player have invested more than 35 talentpoints in a forbidden talenttree.
static bool Arena1v1CheckTalents(Player* player)
{
	if(!player)
		return false;

//	if(sWorld->getBoolConfig(CONFIG_ARENA_1V1_BLOCK_FORBIDDEN_TALENTS) == false)
		return true;

	uint32 count = 0;
	for (uint32 talentId = 0; talentId < sTalentStore.GetNumRows(); ++talentId)
	{
		TalentEntry const* talentInfo = sTalentStore.LookupEntry(talentId);

		if (!talentInfo)
			continue;

		for (int8 rank = MAX_TALENT_RANK-1; rank >= 0; --rank)
		{
			if (talentInfo->RankID[rank] == 0)
				continue;

			if (player->HasTalent(talentInfo->RankID[rank], player->GetActiveSpec()))
			{
				for(int8 i = 0; FORBIDDEN_TALENTS_IN_1V1_ARENA[i] != 0; i++)
					if(FORBIDDEN_TALENTS_IN_1V1_ARENA[i] == talentInfo->TalentTab)
						count += rank + 1;
			}
		}
	}

	if(count >= 36)
	{
		// Dont show error message for healers already in
		// arena because of the bonus rewards for healers
		if (player->InArena())
			return false;
		else
		{
		    player->GetSession()->SendAreaTriggerMessage("You can't join, because you have invested too many points in a forbidden talent. Please edit your talents.");
		    return false;
		}
	}
	else
		return true;
}

#endif