Improvements

Moose Development/Moose/Tasking/Task.lua

@@ -283,7 +283,6 @@ function TASK:AbortUnit( PlayerUnit )
         if #PlayerGroup:GetUnits() == 1 then
           self:UnAssignFromGroup( PlayerGroup )
           PlayerGroup:SetState( PlayerGroup, "Assigned", nil )
-          self:RemoveMenuForGroup( PlayerGroup )
         end
         self:Abort()
         self:PlayerAborted( PlayerUnit )
@@ -322,7 +321,6 @@ function TASK:CrashUnit( PlayerUnit )
         self:E( { TaskGroup = PlayerGroup:GetName(), GetUnits = PlayerGroup:GetUnits() } )
         if #PlayerGroup:GetUnits() == 1 then
           PlayerGroup:SetState( PlayerGroup, "Assigned", nil )
-          self:RemoveMenuForGroup( PlayerGroup )
         end
         self:PlayerCrashed( PlayerUnit )
       end
@@ -357,17 +355,18 @@ end
 -- @param Wrapper.Group#GROUP TaskGroup
 -- @return #TASK
 function TASK:AssignToGroup( TaskGroup )
-  self:F2( TaskGroup:GetName() )
+  self:F( TaskGroup:GetName() )
   
   local TaskGroupName = TaskGroup:GetName()
   
   TaskGroup:SetState( TaskGroup, "Assigned", self )
   
+  self:E("Task is assigned to " .. TaskGroup:GetName() )
+  
   local Mission = self:GetMission()
   local MissionMenu = Mission:GetMenu( TaskGroup )
   MissionMenu:RemoveSubMenus()
   
-  --self:RemoveMenuForGroup( TaskGroup )
   self:SetAssignedMenuForGroup( TaskGroup )
   
   local TaskUnits = TaskGroup:GetUnits()
@@ -482,6 +481,8 @@ function TASK:UnAssignFromGroup( TaskGroup )
   
   TaskGroup:SetState( TaskGroup, "Assigned", nil )
 
+  self:E("Task is unassigned from " .. TaskGroup:GetName() )
+
   self:RemoveAssignedMenuForGroup( TaskGroup )
 
   local TaskUnits = TaskGroup:GetUnits()

Moose Development/Moose/Tasking/Task_A2G.lua

@@ -93,8 +93,6 @@ do -- TASK_A2G
     self.TargetSetUnit = TargetSetUnit
     self.TaskType = TaskType
 
-    Mission:AddTask( self )
-    
     local Fsm = self:GetUnitProcess()
     
 
@@ -374,6 +372,8 @@ do -- TASK_SEAD
     local self = BASE:Inherit( self, TASK_A2G:New( Mission, SetGroup, TaskName, TargetSetUnit, "SEAD" ) ) -- #TASK_SEAD
     self:F()
     
+    Mission:AddTask( self )
+    
     return self
   end 
 
@@ -403,6 +403,8 @@ do -- TASK_BAI
     local self = BASE:Inherit( self, TASK_A2G:New( Mission, SetGroup, TaskName, TargetSetUnit, "BAI" ) ) -- #TASK_BAI
     self:F()
     
+    Mission:AddTask( self )
+    
     return self
   end 
 
@@ -432,6 +434,8 @@ do -- TASK_CAS
     local self = BASE:Inherit( self, TASK_A2G:New( Mission, SetGroup, TaskName, TargetSetUnit, "CAS" ) ) -- #TASK_CAS
     self:F()
     
+    Mission:AddTask( self )
+    
     return self
   end