initial commit

aghanim_singleplayer/scripts/vscripts/ai/troll_warlord_melee.lua

@@ -0,0 +1,182 @@
+--[[ Troll Warlord Melee AI ]]
+
+require( "ai/ai_core" )
+
+function Spawn( entityKeyValues )
+	thisEntity:AddNewModifier( thisEntity, nil, "modifier_troll_warlord_berserkers_rage", {} )
+	thisEntity:SetContextThink( "AIThink", AIThink, 0.25 )
+    behaviorSystem = AICore:CreateBehaviorSystem( thisEntity, { BehaviorNone, BehaviorWhirlingAxes, BehaviorBattleTrance, BehaviorBKB } )
+end
+
+function AIThink() -- For some reason AddThinkToEnt doesn't accept member functions
+	return behaviorSystem:Think( )
+end
+
+--------------------------------------------------------------------------------------------------------
+
+BehaviorNone = {}
+
+function BehaviorNone:Evaluate()
+	return 1 -- must return a value > 0, so we have a default
+end
+
+function BehaviorNone:Begin()
+
+	local orders = nil
+	local hTarget = AICore:ClosestEnemyHeroInRange( thisEntity, 1000 )
+	if hTarget ~= nil then
+		thisEntity.lastTargetPosition = hTarget:GetAbsOrigin()
+		hTarget:MakeVisibleDueToAttack( DOTA_TEAM_BADGUYS, 100 )
+		orders =
+		{
+			UnitIndex = thisEntity:entindex(),
+			OrderType = DOTA_UNIT_ORDER_ATTACK_TARGET,
+			TargetIndex = hTarget:entindex()
+		}
+	elseif thisEntity.lastTargetPosition ~= nil then
+		orders =
+		{
+			UnitIndex = thisEntity:entindex(),
+			OrderType = DOTA_UNIT_ORDER_ATTACK_MOVE,
+			Position = thisEntity.lastTargetPosition
+		}
+	else
+		orders =
+		{
+			UnitIndex = thisEntity:entindex(),
+			OrderType = DOTA_UNIT_ORDER_STOP
+		}
+	end
+
+	return orders
+end
+
+BehaviorNone.Continue = BehaviorNone.Begin
+
+--------------------------------------------------------------------------------------------------------
+
+BehaviorWhirlingAxes = {}
+
+function BehaviorWhirlingAxes:Evaluate()
+	--print( "BehaviorWhirlingAxes:Evaluate()" )
+	local desire = 0
+
+	-- let's not choose this twice in a row
+	if behaviorSystem.currentBehavior == self then 
+		return desire 
+	end
+
+	self.whirlingAxesAbility = thisEntity:FindAbilityByName( "troll_warlord_whirling_axes_melee" )
+	if self.whirlingAxesAbility and self.whirlingAxesAbility:IsFullyCastable() then
+		local enemies = FindUnitsInRadius( DOTA_TEAM_BADGUYS, thisEntity:GetOrigin(), nil, 450, DOTA_UNIT_TARGET_TEAM_ENEMY, DOTA_UNIT_TARGET_HERO, DOTA_UNIT_TARGET_FLAG_NO_INVIS, 0, false )
+		if ( #enemies >= 0 )  then
+			desire = #enemies + 1
+		end
+	end
+
+	return desire
+end
+
+function BehaviorWhirlingAxes:Begin()
+	--print( "BehaviorWhirlingAxes:Begin()" )
+	if self.whirlingAxesAbility and self.whirlingAxesAbility:IsFullyCastable() then
+		local order =
+			{
+				UnitIndex = thisEntity:entindex(),
+				OrderType = DOTA_UNIT_ORDER_CAST_NO_TARGET,
+				AbilityIndex = self.whirlingAxesAbility:entindex(),
+			}
+		return order
+	end
+
+	return nil
+end
+
+BehaviorWhirlingAxes.Continue = BehaviorWhirlingAxes.Begin
+
+--------------------------------------------------------------------------------------------------------
+
+BehaviorBattleTrance = {}
+
+function BehaviorBattleTrance:Evaluate()
+	--print( "BehaviorBattleTrance:Evaluate()" )
+	local desire = 0
+
+	self.battleTranceAbility = thisEntity:FindAbilityByName( "troll_warlord_battle_trance" )
+	if self.battleTranceAbility and self.battleTranceAbility:IsFullyCastable() then
+		local enemies = FindUnitsInRadius( DOTA_TEAM_BADGUYS, thisEntity:GetOrigin(), nil, 300, DOTA_UNIT_TARGET_TEAM_ENEMY, DOTA_UNIT_TARGET_HERO, DOTA_UNIT_TARGET_FLAG_NO_INVIS, 0, false )
+		if ( thisEntity:GetHealthPercent() < 30 ) then
+			if ( #enemies >= 0 )  then
+				desire = #enemies + 1
+			end
+		end
+	end
+
+	return desire
+end
+
+function BehaviorBattleTrance:Begin()
+	--print( "BehaviorBattleTrance:Begin()" )
+	if self.battleTranceAbility and self.battleTranceAbility:IsFullyCastable() then
+		--print( "Casting Battle Trance" )
+		local order =
+		{
+			UnitIndex = thisEntity:entindex(),
+			OrderType = DOTA_UNIT_ORDER_CAST_NO_TARGET,
+			AbilityIndex = self.battleTranceAbility:entindex(),
+		}
+		return order
+	end
+
+	return nil
+end
+
+BehaviorBattleTrance.Continue = BehaviorBattleTrance.Begin
+
+--------------------------------------------------------------------------------------------------------
+
+BehaviorBKB = {}
+
+function BehaviorBKB:Evaluate()
+	--print( "BehaviorBKB:Evaluate()" )
+	local desire = 0
+	for i = 0, DOTA_ITEM_MAX - 1 do
+		local item = thisEntity:GetItemInSlot( i )
+		if item and item:GetAbilityName() == "item_creature_black_king_bar" then
+			self.blackKingBarAbility = item
+		end
+	end
+	if self.blackKingBarAbility and self.blackKingBarAbility:IsFullyCastable() then
+		local enemies = FindUnitsInRadius( DOTA_TEAM_BADGUYS, thisEntity:GetOrigin(), nil, 300, DOTA_UNIT_TARGET_TEAM_ENEMY, DOTA_UNIT_TARGET_HERO, DOTA_UNIT_TARGET_FLAG_NO_INVIS, 0, false )
+		if ( thisEntity:GetHealthPercent() < 50 ) then
+			if ( #enemies >= 0 )  then
+				desire = #enemies + 1
+			end
+		end
+	end
+	
+
+	return desire
+end
+
+function BehaviorBKB:Begin()
+	--print( "BehaviorBKB:Begin()" )
+	if self.blackKingBarAbility and self.blackKingBarAbility:IsFullyCastable() then
+		--print( "Casting Primal Roar" )
+		local order =
+		{
+			UnitIndex = thisEntity:entindex(),
+			OrderType = DOTA_UNIT_ORDER_CAST_NO_TARGET,
+			AbilityIndex = self.blackKingBarAbility:entindex(),
+		}
+		return order
+	end
+
+	return nil
+end
+
+BehaviorBKB.Continue = BehaviorBKB.Begin
+
+--------------------------------------------------------------------------------------------------------
+
+AICore.possibleBehaviors = { BehaviorNone, BehaviorWhirlingAxes, BehaviorBattleTrance, BehaviorBKB }