Progress on AI_CARGO_TROOPS

Moose Development/Moose/AI/AI_Cargo_Troops.lua

@@ -24,19 +24,16 @@ AI_CARGO_TROOPS = {
 
 --- Creates a new AI_CARGO_TROOPS object.
 -- @param #AI_CARGO_TROOPS self
+-- @param Wrapper.Unit#UNIT CargoCarrier
+-- @param Cargo.CargoGroup#CARGO_GROUP CargoGroup
+-- @param #number CombatRadius
 -- @return #AI_CARGO_TROOPS
 function AI_CARGO_TROOPS:New( CargoCarrier, CargoGroup, CombatRadius )
 
   local self = BASE:Inherit( self, FSM_CONTROLLABLE:New( ) ) -- #AI_CARGO_TROOPS
 
-  self.CargoCarrier = CargoCarrier -- Wrapper.Unit#UNIT
-  self.CargoGroup = CargoGroup -- Core.Cargo#CARGO_GROUP
+  self.CargoGroup = CargoGroup -- Cargo.CargoGroup#CARGO_GROUP
   self.CombatRadius = CombatRadius
-  
-  self.Zone = ZONE_UNIT:New( self.CargoCarrier:GetName() .. "-Zone", self.CargoCarrier, CombatRadius )
-  self.Coalition = self.CargoCarrier:GetCoalition()
-  
-  self:SetControllable( CargoCarrier )
 
   self:SetStartState( "UnLoaded" ) 
   
@@ -50,14 +47,91 @@ function AI_CARGO_TROOPS:New( CargoCarrier, CargoGroup, CombatRadius )
   self:AddTransition( "*", "Monitor", "*" )
   self:AddTransition( "*", "Follow", "Following" )
   self:AddTransition( "*", "Guard", "Guarding" )
+  
+  self:AddTransition( "*", "Destroyed", "Destroyed" )
 
   self:__Monitor( 1 )
-  self:__Load( 1 )
+
+  self:SetCarrier( CargoCarrier )
   
   return self
 end
 
 
+--- Set the Carrier.
+-- @param #AI_CARGO_TROOPS self
+-- @param Wrapper.Unit#UNIT CargoCarrier
+-- @return #AI_CARGO_TROOPS
+function AI_CARGO_TROOPS:SetCarrier( CargoCarrier )
+
+  self.CargoCarrier = CargoCarrier -- Wrapper.Unit#UNIT
+  self.CargoCarrier:SetState( self.CargoCarrier, "AI_CARGO_TROOPS", self )
+
+  CargoCarrier:HandleEvent( EVENTS.Dead )
+  CargoCarrier:HandleEvent( EVENTS.Hit )
+  
+  function CargoCarrier:OnEventDead( EventData )
+    self:F({"dead"})
+    local AICargoTroops = self:GetState( self, "AI_CARGO_TROOPS" )
+    self:F({AICargoTroops=AICargoTroops})
+    if AICargoTroops then
+      self:F({})
+      if not AICargoTroops:Is( "Loaded" ) then
+        -- There are enemies within combat range. Unload the CargoCarrier.
+        AICargoTroops:Destroyed()
+      end
+    end
+  end
+  
+  function CargoCarrier:OnEventHit( EventData )
+    self:F({"hit"})
+    local AICargoTroops = self:GetState( self, "AI_CARGO_TROOPS" )
+    if AICargoTroops then
+      self:F( { OnHitLoaded = AICargoTroops:Is( "Loaded" ) } )
+      if AICargoTroops:Is( "Loaded" ) or AICargoTroops:Is( "Boarding" ) then
+        -- There are enemies within combat range. Unload the CargoCarrier.
+        AICargoTroops:Unload()
+      end
+    end
+  end
+  
+  self.Zone = ZONE_UNIT:New( self.CargoCarrier:GetName() .. "-Zone", self.CargoCarrier, self.CombatRadius )
+  self.Coalition = self.CargoCarrier:GetCoalition()
+  
+  self:SetControllable( CargoCarrier )
+
+  self:Guard()
+
+  return self
+end
+
+
+--- Find a free Carrier within a range.
+-- @param #AI_CARGO_TROOPS self
+-- @param Core.Point#COORDINATE Coordinate
+-- @param #number Radius
+-- @return Wrapper.Unit#UNIT NewCarrier
+function AI_CARGO_TROOPS:FindCarrier( Coordinate, Radius )
+
+  local CoordinateZone = ZONE_RADIUS:New( "Zone" , Coordinate:GetVec2(), Radius )
+  CoordinateZone:Scan( { Object.Category.UNIT } )
+  for _, DCSUnit in pairs( CoordinateZone:GetScannedUnits() ) do
+    local NearUnit = UNIT:Find( DCSUnit )
+    self:F({NearUnit=NearUnit})
+    if not NearUnit:GetState( NearUnit, "AI_CARGO_TROOPS" ) then
+      local Attributes = NearUnit:GetDesc()
+      self:F({Desc=Attributes})
+      if NearUnit:HasAttribute( "Trucks" ) then
+        self:SetCarrier( NearUnit )
+        break
+      end
+    end
+  end
+
+end
+
+
+
 --- Follow Infantry to the Carrier.
 -- @param #AI_CARGO_TROOPS self
 -- @param #AI_CARGO_TROOPS Me
@@ -68,40 +142,44 @@ function AI_CARGO_TROOPS:FollowToCarrier( Me, CargoCarrier, InfantryGroup )
 
   self:F( { self = self:GetClassNameAndID(), InfantryGroup = InfantryGroup:GetName() } )
   
-  -- We check if the Cargo is near to the CargoCarrier.
-  if InfantryGroup:IsPartlyInZone( ZONE_UNIT:New( "Radius", CargoCarrier, 5 ) ) then
+  --if self:Is( "Following" ) then
 
-    -- The Cargo does not need to follow the Carrier.
-    Me:Guard()
+  if CargoCarrier:IsAlive() then
+    -- We check if the Cargo is near to the CargoCarrier.
+    if InfantryGroup:IsPartlyInZone( ZONE_UNIT:New( "Radius", CargoCarrier, 5 ) ) then
   
-  else
+      -- The Cargo does not need to follow the Carrier.
+      Me:Guard()
     
-    self:F( { InfantryGroup = InfantryGroup:GetName() } )
-  
-    if InfantryGroup:IsAlive() then
-          
+    else
+      
       self:F( { InfantryGroup = InfantryGroup:GetName() } )
-
-      local Waypoints = {}
-      
-      -- Calculate the new Route.
-      local FromCoord = InfantryGroup:GetCoordinate()
-      local FromGround = FromCoord:WaypointGround( 10, "Diamond" )
-      self:F({FromGround=FromGround})
-      table.insert( Waypoints, FromGround )
-
-      local ToCoord = CargoCarrier:GetCoordinate()
-      local ToGround = ToCoord:WaypointGround( 10, "Diamond" )
-      self:F({ToGround=ToGround})
-      table.insert( Waypoints, ToGround )
-      
-      local TaskRoute = InfantryGroup:TaskFunction( "AI_CARGO_TROOPS.FollowToCarrier", Me, CargoCarrier, InfantryGroup )
-      
-      self:F({Waypoints = Waypoints})
-      local Waypoint = Waypoints[#Waypoints]
-      InfantryGroup:SetTaskWaypoint( Waypoint, TaskRoute ) -- Set for the given Route at Waypoint 2 the TaskRouteToZone.
     
-      InfantryGroup:Route( Waypoints, 1 ) -- Move after a random seconds to the Route. See the Route method for details.
+      if InfantryGroup:IsAlive() then
+            
+        self:F( { InfantryGroup = InfantryGroup:GetName() } )
+  
+        local Waypoints = {}
+        
+        -- Calculate the new Route.
+        local FromCoord = InfantryGroup:GetCoordinate()
+        local FromGround = FromCoord:WaypointGround( 10, "Diamond" )
+        self:F({FromGround=FromGround})
+        table.insert( Waypoints, FromGround )
+  
+        local ToCoord = CargoCarrier:GetCoordinate():GetRandomCoordinateInRadius( 10, 5 )
+        local ToGround = ToCoord:WaypointGround( 10, "Diamond" )
+        self:F({ToGround=ToGround})
+        table.insert( Waypoints, ToGround )
+        
+        local TaskRoute = InfantryGroup:TaskFunction( "AI_CARGO_TROOPS.FollowToCarrier", Me, CargoCarrier, InfantryGroup )
+        
+        self:F({Waypoints = Waypoints})
+        local Waypoint = Waypoints[#Waypoints]
+        InfantryGroup:SetTaskWaypoint( Waypoint, TaskRoute ) -- Set for the given Route at Waypoint 2 the TaskRouteToZone.
+      
+        InfantryGroup:Route( Waypoints, 1 ) -- Move after a random seconds to the Route. See the Route method for details.
+      end
     end
   end
 end
@@ -132,6 +210,22 @@ function AI_CARGO_TROOPS:onafterMonitor( CargoCarrier, From, Event, To )
           self:Follow()
         end
       end
+      if self:Is( "Following" ) then
+        local Distance = Coordinate:Get2DDistance( self.CargoGroup:GetCoordinate() )
+        self:F( { Distance = Distance } )
+        if Distance > 40 then
+          CargoCarrier:RouteStop()
+          self.CarrierStopped = true
+        else
+          if self.CarrierStopped then
+            if self.CargoGroup:IsNear( CargoCarrier, 10 ) then
+              CargoCarrier:RouteResume()
+              self.CarrierStopped = nil
+            end
+          end
+        end
+      end
+      
     end
     self.CarrierCoordinate = CargoCarrier:GetCoordinate()
   end
@@ -149,7 +243,7 @@ function AI_CARGO_TROOPS:onafterLoad( CargoCarrier, From, Event, To )
   if CargoCarrier and CargoCarrier:IsAlive() then
     CargoCarrier:RouteStop()
     self:__Board( 10 ) 
-    self.CargoGroup:Board( CargoCarrier, 100 )
+    self.CargoGroup:Board( CargoCarrier, 10 )
   end
   
 end