Working version of the DETECTION classes

Moose Development/Moose/Process_BAI.lua

@@ -0,0 +1,143 @@
+--- @module Process_BAI
+
+--- PROCESS_BAI class
+-- @type PROCESS_BAI
+-- @field Unit#UNIT ProcessUnit
+-- @field Set#SET_UNIT TargetSetUnit
+-- @extends Process#PROCESS
+PROCESS_BAI = { 
+  ClassName = "PROCESS_BAI",
+  Fsm = {},
+  TargetSetUnit = nil,
+}
+
+
+--- Creates a new BAI task.
+-- @param #PROCESS_BAI self
+-- @param Task#TASK Task
+-- @param Unit#UNIT ProcessUnit
+-- @param Set#SET_UNIT TargetSetUnit
+-- @return #PROCESS_BAI self
+function PROCESS_BAI:New( Task, ProcessUnit, TargetSetUnit )
+
+  -- Inherits from BASE
+  local self = BASE:Inherit( self, PROCESS:New( "BAI", Task, ProcessUnit ) ) -- #PROCESS_BAI
+  
+  self.TargetSetUnit = TargetSetUnit
+
+  self.Fsm = STATEMACHINE_PROCESS:New( self, {
+    initial = 'Assigned',
+    events = {
+      { name = 'Start', from = 'Assigned', to = 'Waiting'    },
+      { name = 'HitTarget',  from = 'Waiting',    to = 'Destroy' },
+      { name = 'MoreTargets', from = 'Destroy', to = 'Waiting'  },
+      { name = 'Destroyed', from = 'Destroy', to = 'Success' },      
+      { name = 'Fail', from = 'Assigned', to = 'Failed' },
+      { name = 'Fail', from = 'Waiting', to = 'Failed' },
+      { name = 'Fail', from = 'Destroy', to = 'Failed' },
+    },
+    callbacks = {
+      onStart =  self.OnStart,
+      onHitTarget =  self.OnHitTarget,
+      onMoreTargets = self.OnMoreTargets,
+      onDestroyed = self.OnDestroyed,
+      onKilled = self.OnKilled,
+    },
+    endstates = { 'Success', 'Failed' }
+  } )
+
+
+  _EVENTDISPATCHER:OnDead( self.EventDead, self )
+  
+  return self
+end
+
+--- Process Events
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+function PROCESS_BAI:OnStart( Fsm, Event, From, To )
+  self:E( { Event, From, To, self.ProcessUnit.UnitName} )
+
+  self:NextEvent( Fsm.Start )
+end
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+-- @param Event#EVENTDATA Event
+function PROCESS_BAI:OnHitTarget( Fsm, Event, From, To, Event )
+
+  if self.TargetSetUnit:Count() > 0 then
+    self:NextEvent( Fsm.MoreTargets )
+  else
+    self:NextEvent( Fsm.Destroyed )
+  end
+end
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+function PROCESS_BAI:OnMoreTargets( Fsm, Event, From, To )
+
+
+end
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+-- @param Event#EVENTDATA DCSEvent
+function PROCESS_BAI:OnKilled( Fsm, Event, From, To )
+
+  self:NextEvent( Fsm.Restart )
+
+end
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+function PROCESS_BAI:OnRestart( Fsm, Event, From, To )
+
+  self:NextEvent( Fsm.Menu )
+
+end
+
+--- StateMachine callback function for a PROCESS
+-- @param #PROCESS_BAI self
+-- @param StateMachine#STATEMACHINE_PROCESS Fsm
+-- @param #string Event
+-- @param #string From
+-- @param #string To
+function PROCESS_BAI:OnDestroyed( Fsm, Event, From, To )
+
+end
+
+--- DCS Events
+
+--- @param #PROCESS_BAI self
+-- @param Event#EVENTDATA Event
+function PROCESS_BAI:EventDead( Event )
+
+  if Event.IniDCSUnit then
+    self.TargetSetUnit:Remove( Event.IniDCSUnitName )
+    self:NextEvent( self.Fsm.HitTarget, Event )
+  end
+end
+
+