MOOSE_MISSIONS/Core/Set/SET-GRP/SET-GRP-000 - Add and Remove/SET-101 - Group Sets.lua

SetVehicles = SET_GROUP:New()

SetVehicles:AddGroupsByName( { "Vehicle A", "Vehicle B", "Vehicle C" } )

SetVehicles:ForEachGroup( 
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeGreen()
    end
  end 
)



SetBluePlanesGroup = SET_GROUP:New()
  :FilterCoalitions( "blue" )
  :FilterCategories( "plane" )
  :FilterStart()
  
SetNorthKoreaGroup = SET_GROUP:New()
  :FilterCountries( "RUSSIA" )
  :FilterStart()

SetSAMGroup = SET_GROUP:New()
  :FilterPrefixes( "SAM" )
  :FilterStart()
  :SetIteratorIntervals( 5, 10 )
  
SetGroundGroup = SET_GROUP:New()
  :FilterCategories( "ground" )
  :FilterStart()
  
SetGroundGroup:Flush()

SpawnUS_Plane = SPAWN:New( 'Spawn Test USA Plane')
GroupUS_Plane = SpawnUS_Plane:Spawn()

SpawnUS_Vehicle = SPAWN:New( 'Spawn Test USA Vehicle')
GroupUS_Vehicle = SpawnUS_Vehicle:Spawn()

SpawnUS_Ship = SPAWN:New( 'Spawn Test USA Ship')
GroupUS_Ship = SpawnUS_Ship:Spawn()

SpawnRU_Vehicle = SPAWN:New( 'Spawn Test RUSSIA Vehicle')
GroupRU_Vehicle = SpawnRU_Vehicle:Spawn()

SpawnRU_Ship = SPAWN:New( 'Spawn Test RUSSIA Ship')
GroupRU_Ship = SpawnRU_Ship:Spawn()

SpawnM2A2_AttackVehicle = SPAWN:New( 'Spawn Test M2A2 Attack Vehicle' ):InitRandomizeUnits( true, 10, 4 )
SpawnSAM_AttackVehicle = SPAWN:New( 'Spawn Test SAM Attack Vehicle' ):InitRandomizeUnits( true, 10, 4 )

for i = 1, 30 do
  GroupM2A2_AttackVehicle = SpawnM2A2_AttackVehicle:SpawnInZone( ZONE:New("Spawn Zone") )
  GroupSAM_AttackVehicle = SpawnSAM_AttackVehicle:SpawnInZone( ZONE:New("Spawn Zone") )
end

SetVehicleCompletely = SET_GROUP:New()
  :FilterPrefixes( "Spawn Vehicle Zone Completely" )
  :FilterStart() 

SetVehiclePartly = SET_GROUP:New()
  :FilterPrefixes( "Spawn Vehicle Zone Partly" )
  :FilterStart() 

SetVehicleNot = SET_GROUP:New()
  :FilterPrefixes( "Spawn Vehicle Zone Not" )
  :FilterStart() 

Spawn_Vehicle_Zone_Completely = SPAWN:New( 'Spawn Vehicle Zone Completely' ):InitRandomizeUnits( true, 10, 4)
Spawn_Vehicle_Zone_Partly     = SPAWN:New( 'Spawn Vehicle Zone Partly' ):InitRandomizeUnits( true, 10, 4 )
Spawn_Vehicle_Zone_Not        = SPAWN:New( 'Spawn Vehicle Zone Not' ):InitRandomizeUnits( true, 10, 4 )
for i = 1, 30 do
  Spawn_Vehicle_Zone_Completely:SpawnInZone( ZONE:New("Spawn Zone Completely") )
  Spawn_Vehicle_Zone_Partly:SpawnInZone( ZONE:New("Spawn Zone Partly") )
  Spawn_Vehicle_Zone_Not:SpawnInZone( ZONE:New("Spawn Zone Not") )
end

--DBBlue:TraceDatabase()
--SCHEDULER:New( DBBluePlanes, DBBluePlanes.Flush, {  }, 1 )
--SCHEDULER:New( DBRedVehicles, DBRedVehicles.Flush, {  }, 1 )
--SCHEDULER:New( DBShips, DBShips.Flush, {  }, 1 )
--SCHEDULER:New( DBBelgium, DBBelgium.Flush, {  }, 1 )
--SCHEDULER:New( DBNorthKorea, DBNorthKorea.Flush, {  }, 1 )
--SCHEDULER:New( DBKA50Vinson, DBKA50Vinson.Flush, {  }, 1 )
--
--SCHEDULER:New( DBBluePlanesGroup, DBBluePlanesGroup.Flush, { }, 1 )
--SCHEDULER:New( DBNorthKoreaGroup, DBNorthKoreaGroup.Flush, { }, 1 )

SetBluePlanesGroup:ForEachGroup( 
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeBlue()
    end
  end 
)

SetNorthKoreaGroup:ForEachGroup( 
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeRed()
    end
  end 
)

SetSAMGroup:ForEachGroup( 
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeOrange()
    end
  end 
)

GroupZoneCompletely = GROUP:FindByName( "Zone Completely" )
GroupZonePartly = GROUP:FindByName( "Zone Partly" )
GroupZoneNot = GROUP:FindByName( "Zone Not" )

ZoneCompletely = ZONE_POLYGON:New( "Zone Completely", GroupZoneCompletely ):SmokeZone( SMOKECOLOR.White )
ZonePartly = ZONE_POLYGON:New( "Zone Partly", GroupZonePartly ):SmokeZone( SMOKECOLOR.White )
ZoneNot = ZONE_POLYGON:New( "Zone Not", GroupZoneNot ):SmokeZone( SMOKECOLOR.White )

SetVehicleCompletely:ForEachGroupCompletelyInZone( ZoneCompletely,
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeBlue()
    end
  end 
)
  
SetVehiclePartly:ForEachGroupPartlyInZone( ZonePartly,
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeBlue()
    end
  end 
)
    
SetVehicleNot:ForEachGroupNotInZone( ZoneNot,
  --- @param Wrapper.Group#GROUP MooseGroup
  function( MooseGroup ) 
    for UnitId, UnitData in pairs( MooseGroup:GetUnits() ) do
      local UnitAction = UnitData -- Wrapper.Unit#UNIT
      UnitAction:SmokeBlue()
    end
  end 
)