public void SetVariable(SC2GameState gameState) { this.gameState = gameState; this.newObservation = gameState.NewObservation; this.gameLoop = (int)newObservation.Observation.GameLoop; this.allUnits = newObservation.Observation.RawData.Units.ToList(); this.upgradeIDs = newObservation.Observation.RawData.Player.UpgradeIds; this.enemyUnit = GetEnemyUnits(); this.myUnit = GetMyUnits(); this.baseLocations = allUnits.FindBaseLocation(); }
private EnemyUnits GetEnemyUnits() { EnemyUnits ret = new EnemyUnits(); foreach (Unit u in allUnits) { if (u.Alliance == Alliance.Enemy) { ret.all.Add(u); if (u.IsBaseBuilding()) { ret.baseBuilding.Add(u); } else { if (u.IsArmyUnit()) { ret.armyUnit.Add(u); } } } } return(ret); }