initial commit

aghanim_singleplayer/scripts/vscripts/encounters/encounter_wrath.lua

@@ -0,0 +1,138 @@
+
+require( "map_encounter" )
+require( "aghanim_utility_functions" )
+require( "spawner" )
+
+--------------------------------------------------------------------------------
+
+if CMapEncounter_Wrath == nil then
+	CMapEncounter_Wrath = class( {}, {}, CMapEncounter )
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:GetPreviewUnit()
+	return "npc_dota_hero_zuus"
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:OnEncounterLoaded()
+	CMapEncounter.OnEncounterLoaded( self )
+
+	self.bGreenLight = true
+
+	self.HeroesOnGoal = {}
+end
+
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:GetMaxSpawnedUnitCount()
+	return 0
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:Start()
+	CMapEncounter.Start( self )
+
+	local hGreenLights = self:GetRoom():FindAllEntitiesInRoomByName( "green_light", true )
+	if #hGreenLights > 0 then
+		self.hGreenLight = hGreenLights[ 1 ]
+	end
+	if self.hGreenLight == nil then
+		print( "Unable to find \"green_light\"" )
+	end
+
+	local hRedLights = self:GetRoom():FindAllEntitiesInRoomByName( "red_light", true )
+	if #hRedLights > 0 then
+		self.hRedLight = hRedLights[ 1 ]
+	end
+	if self.hRedLight == nil then
+		print( "Unable to find \"red_light\"" )
+	end
+
+	self.fEncounterStartTime = GameRules:GetGameTime()
+	self.fNextLightChangeTime = self.fEncounterStartTime + self:GetNextLightChangeTime()
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:OnThink()
+	CMapEncounter.OnThink( self )
+
+	if GameRules:GetGameTime() >= self.fNextLightChangeTime then
+		printf( "self:ToggleLights()" )
+		self:ToggleLights()
+	end
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:GetNextLightChangeTime()
+	return RandomFloat( 1.0, 7.0 )
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:ToggleLights()
+	if self.bGreenLight then
+		self.hGreenLight:Disable()
+		self.hRedLight:Enable()
+		self.bGreenLight = false
+	else
+		self.hGreenLight:Enable()
+		self.hRedLight:Disable()
+		self.bGreenLight = true
+	end
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:CheckForCompletion()
+	local nHeroesAlive = 0
+	local hHeroes = HeroList:GetAllHeroes()
+	for _, hHero in pairs ( hHeroes ) do
+		if hHero ~= nil and hHero:IsRealHero() and hHero:GetTeamNumber() == DOTA_TEAM_GOODGUYS and ( hHero:IsAlive() or hHero:IsReincarnating() )  then
+			nHeroesAlive = nHeroesAlive + 1
+		end
+	end
+
+	return nHeroesAlive > 0 and #self.HeroesOnGoal == nHeroesAlive
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:OnSpawnerFinished( hSpawner, hSpawnedUnits )
+	CMapEncounter.OnSpawnerFinished( self, hSpawner, hSpawnedUnits )
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:OnTriggerStartTouch( event )
+
+	local hUnit = EntIndexToHScript( event.activator_entindex )
+	local hTriggerEntity = EntIndexToHScript( event.caller_entindex )
+	if hUnit ~= nil and hUnit:IsRealHero() and hUnit:IsControllableByAnyPlayer() and event.trigger_name == "goal_trigger" then
+		table.insert( self.HeroesOnGoal, hUnit )
+	end
+end
+
+--------------------------------------------------------------------------------
+
+function CMapEncounter_Wrath:OnTriggerEndTouch( event )
+	local hUnit = EntIndexToHScript( event.activator_entindex )
+	local hTriggerEntity = EntIndexToHScript( event.caller_entindex )
+	if hUnit ~= nil and hUnit:IsRealHero() and hUnit:IsControllableByAnyPlayer() and event.trigger_name == "goal_trigger" then
+		for k,hHero in pairs( self.HeroesOnGoal ) do
+			if hHero == hUnit then
+				table.remove( self.HeroesOnGoal, k )
+			end
+		end
+	end
+end
+
+--------------------------------------------------------------------------------
+
+return CMapEncounter_Wrath