public void AddDesiredPerBase(BuildingAtBase key, int val)
{
if (!DesiredPerBase.ContainsKey(key))
{
DesiredPerBase.Add(key, 0);
}
DesiredPerBase[key] += val;
}
public override bool Equals(object obj)
{
if (obj.GetType() != typeof(BuildingAtBase))
{
return(false);
}
BuildingAtBase other = (BuildingAtBase)obj;
return(Type == other.Type && B == other.B);
}
public bool Construct()
{
BuiltThisFrame = false;
Dictionary <uint, int> desired = new Dictionary <uint, int>();
Dictionary <BuildingAtBase, int> desiredPerBase = new Dictionary <BuildingAtBase, int>();
for (int pos = 0; pos < Steps.Count; pos++)
{
BuildStep step = null;
try
{
step = Steps[pos];
if (step.GetType() == typeof(BuildingStep))
{
BuildingStep building = (BuildingStep)step;
if (!building.Condition.Invoke())
{
continue;
}
if (UnitTypes.LookUp[building.UnitType].TechRequirement != 0 &&
Tyr.Bot.UnitManager.Completed(UnitTypes.LookUp[building.UnitType].TechRequirement) == 0 &&
UnitTypes.LookUp[building.UnitType].TechRequirement != UnitTypes.HATCHERY)
{
Tyr.Bot.DrawText("Skipping list. Build tech for " + UnitTypes.LookUp[building.UnitType].Name + " not available: " + UnitTypes.LookUp[building.UnitType].TechRequirement);
return(true);
}
Add(desired, building.UnitType, building.Number);
if (building.Exact)
{
bool built = false;
foreach (BuildRequest request in ConstructionTask.Task.BuildRequests)
{
if (request.Type == building.UnitType && SC2Util.DistanceSq(request.Pos, building.DesiredPos) <= 2)
{
built = true;
break;
}
}
foreach (BuildRequest request in ConstructionTask.Task.UnassignedRequests)
{
if (request.Type == building.UnitType && SC2Util.DistanceSq(request.Pos, building.DesiredPos) <= 2)
{
built = true;
break;
}
}
if (!built)
{
foreach (Agent agent in Tyr.Bot.UnitManager.Agents.Values)
{
if (agent.Unit.UnitType == building.UnitType && SC2Util.DistanceSq(agent.Unit.Pos, building.DesiredPos) <= 2)
{
built = true;
break;
}
}
}
if (!built)
{
if (!Construct(building, 1))
{
return(false);
}
}
}
if (building.DesiredBase == null && desired[building.UnitType] > Tyr.Bot.UnitManager.Count(building.UnitType) &&
!building.Exact)
{
if (!Construct(building, desired[building.UnitType] - Tyr.Bot.UnitManager.Count(building.UnitType)))
{
return(false);
}
}
if (building.DesiredBase != null)
{
BuildingAtBase key = new BuildingAtBase(building.UnitType, building.DesiredBase);
Add(desiredPerBase, key, building.Number);
if (desiredPerBase[key] > Build.Count(building.DesiredBase, building.UnitType) &&
!building.Exact)
{
Tyr.Bot.DrawText("Building " + UnitTypes.LookUp[building.UnitType].Name + " at base.");
if (!Construct(building, desiredPerBase[key] - Build.Count(building.DesiredBase, building.UnitType)))
{
return(false);
}
}
}
}
else if (step.GetType() == typeof(ConditionalStep))
{
if (!((ConditionalStep)step).Check())
{
Tyr.Bot.DrawText("Skipping list. Condition not met.");
return(true);
}
}
else if (step.GetType() == typeof(GotoStep))
{
if (BuiltThisFrame)
{
return(true);
}
if (((GotoStep)step).Pos < 0)
{
pos += ((GotoStep)step).Pos;
}
else
{
pos = ((GotoStep)step).Pos - 1;
}
}
if (step.GetType() == typeof(MorphStep))
{
MorphStep morph = (MorphStep)step;
if (!morph.Condition.Invoke())
{
continue;
}
if (UnitTypes.LookUp[morph.UnitType].TechRequirement != 0 &&
Tyr.Bot.UnitManager.Completed(UnitTypes.LookUp[morph.UnitType].TechRequirement) == 0 &&
UnitTypes.LookUp[morph.UnitType].TechRequirement != UnitTypes.HATCHERY)
{
Tyr.Bot.DrawText("Skipping list. Morph tech for " + UnitTypes.LookUp[morph.UnitType].Name + " not available: " + UnitTypes.LookUp[morph.UnitType].TechRequirement);
return(true);
}
Add(desired, morph.UnitType, morph.Number);
if (desired[morph.UnitType] > Tyr.Bot.UnitManager.Count(morph.UnitType))
{
if (!Morph(morph, desired[morph.UnitType] - Tyr.Bot.UnitManager.Count(morph.UnitType)))
{
return(false);
}
}
}
}
catch (Exception e)
{
throw new Exception("Exception in buildlist at step " + pos + ": " + step, e);
}
}
return(true);
}
public StepResult Perform(BuildListState state)
{
if (!Condition.Invoke())
{
return(new NextItem());
}
if (UnitTypes.LookUp[UnitType].TechRequirement != 0 &&
Tyr.Bot.UnitManager.Completed(UnitTypes.LookUp[UnitType].TechRequirement) == 0 &&
UnitTypes.LookUp[UnitType].TechRequirement != UnitTypes.HATCHERY)
{
Tyr.Bot.DrawText("Skipping list. Build tech for " + UnitTypes.LookUp[UnitType].Name + " not available: " + UnitTypes.LookUp[UnitType].TechRequirement);
return(new NextList());
}
state.AddDesired(UnitType, Number);
if (DesiredPos != null)
{
bool built = false;
foreach (BuildRequest request in ConstructionTask.Task.BuildRequests)
{
if (request.Type == UnitType && request.Exact == Exact && request.AroundLocation == DesiredPos)
{
built = true;
break;
}
}
foreach (BuildRequest request in ConstructionTask.Task.UnassignedRequests)
{
if (request.Type == UnitType && request.Exact == Exact && request.AroundLocation == DesiredPos)
{
built = true;
break;
}
}
if (!built)
{
foreach (Agent agent in Tyr.Bot.UnitManager.Agents.Values)
{
if (CheckTypeMatches(UnitType, agent.Unit.UnitType) && agent.Exact == Exact && agent.AroundLocation == DesiredPos)
{
built = true;
break;
}
}
}
if (!built)
{
if (!Construct(state, 1))
{
return(new WaitForResources());
}
}
}
if (DesiredBase == null && state.Desired[UnitType] > Tyr.Bot.UnitManager.Count(UnitType) &&
!Exact)
{
if (!Construct(state, state.Desired[UnitType] - Tyr.Bot.UnitManager.Count(UnitType)))
{
return(new WaitForResources());
}
}
if (DesiredBase != null)
{
BuildingAtBase key = new BuildingAtBase(UnitType, DesiredBase);
state.AddDesiredPerBase(key, Number);
if (state.DesiredPerBase[key] > Build.Count(DesiredBase, UnitType) &&
!Exact)
{
if (!Construct(state, state.DesiredPerBase[key] - Build.Count(DesiredBase, UnitType)))
{
return(new WaitForResources());
}
}
}
return(new NextItem());
}
