// This function does all the hard work of working out which engines are burning, which tanks are being drained
// and setting the drain rates
private void UpdateResourceDrains()
{
// Update the active engines
UpdateActiveEngines();
// Empty the draining resources set
drainingResources.Clear();
// Reset the resource drains of all draining parts
foreach (PartSim partSim in drainingParts)
{
partSim.resourceDrains.Reset();
}
// Empty the draining parts set
drainingParts.Clear();
// Loop through all the active engine modules
for (int i = 0; i < activeEngines.Count; ++i)
{
EngineSim engine = activeEngines[i];
// Set the resource drains for this engine
if (engine.SetResourceDrains(log, allParts, allFuelLines, drainingParts))
{
// If it is active then add the consumed resource types to the set
for (int j = 0; j < engine.ResourceConsumptionsForMass.Types.Count; ++j)
{
drainingResources.Add(engine.ResourceConsumptionsForMass.Types[j]);
}
}
}
//foreach (RCSSim r in allRCS) {
// r.SetResourceDrains(log, allParts);
//}
// Update the active engines again to remove any engines that have no fuel supply
UpdateActiveEngines();
if (log != null)
{
log.AppendLine("Active engines = ", activeEngines.Count);
int i = 0;
for (int j = 0; j < activeEngines.Count; j++)
{
EngineSim engine = activeEngines[j];
log.Append("Engine " + (i++) + ":");
engine.DumpEngineToLog(log);
}
log.Flush();
}
}
public void DumpToLog(LogMsg log)
{
if (attachedPartSim == null)
{
log.Append("<staged>:<n>");
}
else
{
log.Append(attachedPartSim.name, ":", attachedPartSim.partId);
}
log.Append("#", nodeType, ":", id);
}
public void DumpToLog(LogMsg log)
{
if (attachedPartSim == null)
{
log.Append("<staged>:<n>");
}
else
{
log.Append(attachedPartSim.name, ":", attachedPartSim.partId);
}
log.Append("#", nodeType, ":", id);
}
public void SetupParent(Dictionary <Part, PartSim> partSimLookup, LogMsg log)
{
if (part.parent != null)
{
parent = null;
if (partSimLookup.TryGetValue(part.parent, out parent))
{
if (log != null)
{
log.AppendLine("Parent part is ", parent.name, ":", parent.partId);
}
if (part.attachMode == AttachModes.SRF_ATTACH && part.attachRules.srfAttach && part.fuelCrossFeed && part.parent.fuelCrossFeed)
{
if (log != null)
{
log.Append("Added (", name, ":", partId)
.AppendLine(", ", parent.name, ":", parent.partId, ") to surface mounted fuel targets.");
}
parent.surfaceMountFuelTargets.Add(this);
surfaceMountFuelTargets.Add(parent);
}
}
else
{
if (log != null)
{
log.AppendLine("No PartSim for parent part (", part.parent.partInfo.name, ")");
}
}
}
}
public bool SetPossibleResourceDrains(LogMsg log, List <PartSim> allParts, HashSet <PartSim> drainingParts)
{
//DumpSourcePartSets(log, "before clear");
foreach (HashSet <PartSim> sourcePartSet in fullSourcePartSets.Values)
{
sourcePartSet.Clear();
}
//DumpSourcePartSets(log, "after clear");
for (int index = 0; index < this.resourceConsumptions.Types.Count; index++)
{
int type = this.resourceConsumptions.Types[index];
if (!fullSourcePartSets.TryGetValue(type, out HashSet <PartSim> sourcePartSet))
{
sourcePartSet = new HashSet <PartSim>();
fullSourcePartSets.Add(type, sourcePartSet);
}
switch ((ResourceFlowMode)this.resourceFlowModes[type])
{
case ResourceFlowMode.NO_FLOW:
if (partSim.maxResources[type] > SimManager.RESOURCE_MIN)
{
sourcePartSet.Add(partSim);
}
break;
case ResourceFlowMode.ALL_VESSEL:
case ResourceFlowMode.ALL_VESSEL_BALANCE:
for (int i = 0; i < allParts.Count; i++)
{
PartSim aPartSim = allParts[i];
if (aPartSim.maxResources[type] > SimManager.RESOURCE_MIN)
{
sourcePartSet.Add(aPartSim);
}
}
break;
case ResourceFlowMode.STAGE_PRIORITY_FLOW:
case ResourceFlowMode.STAGE_PRIORITY_FLOW_BALANCE:
//All vessel ordered by stage
if (log != null)
{
log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":", partSim.partId);
}
foreach (HashSet <PartSim> stagePartSet in stagePartSets.Values)
{
stagePartSet.Clear();
}
var maxStage = -1;
for (int i = 0; i < allParts.Count; i++)
{
var aPartSim = allParts[i];
//if (log != null) log.Append(aPartSim.name, ":" + aPartSim.partId, " contains ", aPartSim.resources[type])
// .AppendLine((aPartSim.resourceFlowStates[type] == 0) ? " (disabled)" : "");
if (aPartSim.maxResources[type] > SimManager.RESOURCE_MIN)
{
int stage = aPartSim.inverseStage;
if (stage > maxStage)
{
maxStage = stage;
}
if (!stagePartSets.TryGetValue(stage, out HashSet <PartSim> tempPartSet))
{
tempPartSet = new HashSet <PartSim>();
stagePartSets.Add(stage, tempPartSet);
}
tempPartSet.Add(aPartSim);
}
}
//Add resource containers in order by stage
for (int j = maxStage; j >= -1; j--)
{
//if (log != null) log.AppendLine("Testing stage ", j);
if (stagePartSets.TryGetValue(j, out HashSet <PartSim> stagePartSet) && stagePartSet.Count > 0)
{
//if (log != null) log.AppendLine("Not empty");
// We have to copy the contents of the set here rather than copying the set reference or
// bad things (tm) happen
foreach (PartSim aPartSim in stagePartSet)
{
sourcePartSet.Add(aPartSim);
}
break;
}
}
break;
case ResourceFlowMode.STACK_PRIORITY_SEARCH:
case ResourceFlowMode.STAGE_STACK_FLOW:
case ResourceFlowMode.STAGE_STACK_FLOW_BALANCE:
visited.Clear();
//Resource containers limited to current stage
if (log != null)
{
log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":", partSim.partId);
}
partSim.GetSourceSet(type, true, allParts, visited, sourcePartSet, true, log, "");
break;
default:
if (log != null)
{
log.Append("SetResourceDrains(", partSim.name, ":", partSim.partId)
.AppendLine(") Unexpected flow type for ", ResourceContainer.GetResourceName(type), ")");
}
break;
}
}
// If we don't have sources for all the needed resources then return false without setting up any drains
for (int i = 0; i < this.resourceConsumptions.Types.Count; i++)
{
int type = this.resourceConsumptions.Types[i];
if (!fullSourcePartSets.TryGetValue(type, out HashSet <PartSim> sourcePartSet) || sourcePartSet.Count == 0)
{
if (log != null)
{
log.AppendLine("No source of ", ResourceContainer.GetResourceName(type));
}
return(false);
}
}
// Now we set the drains on the members of the sets and update the draining parts set
for (int i = 0; i < this.resourceConsumptions.Types.Count; i++)
{
HashSet <PartSim> sourcePartSet = fullSourcePartSets[resourceConsumptions.Types[i]];
// Loop through the members of the set
foreach (PartSim partSim in sourcePartSet)
{
drainingParts.Add(partSim);
}
}
return(true);
}
public void SetupParent(Dictionary<Part, PartSim> partSimLookup, LogMsg log)
{
if (part.parent != null)
{
parent = null;
if (partSimLookup.TryGetValue(part.parent, out parent))
{
if (log != null) log.AppendLine("Parent part is ", parent.name, ":", parent.partId);
if (part.attachMode == AttachModes.SRF_ATTACH && part.attachRules.srfAttach && part.fuelCrossFeed && part.parent.fuelCrossFeed)
{
if (log != null) log.Append("Added (", name, ":", partId)
.AppendLine(", ", parent.name, ":", parent.partId, ") to surface mounted fuel targets.");
parent.surfaceMountFuelTargets.Add(this);
surfaceMountFuelTargets.Add(parent);
}
}
else
{
if (log != null) log.AppendLine("No PartSim for parent part (", part.parent.partInfo.name, ")");
}
}
}
// This is the old recursive function for STACK_PRIORITY_SEARCH
public void GetSourceSet_Old(int type, bool includeSurfaceMountedParts, List<PartSim> allParts, HashSet<PartSim> visited, HashSet<PartSim> allSources, LogMsg log, String indent)
{
if (log != null)
{
log.Append(indent, "GetSourceSet_Old(", ResourceContainer.GetResourceName(type), ") for ")
.AppendLine(name, ":", partId);
indent += " ";
}
// Rule 1: Each part can be only visited once, If it is visited for second time in particular search it returns as is.
if (visited.Contains(this))
{
if (log != null) log.Append(indent, "Returning empty set, already visited (", name, ":")
.AppendLine(partId + ")");
return;
}
if (log != null) log.AppendLine(indent, "Adding this to visited");
visited.Add(this);
// Rule 2: Part performs scan on start of every fuel pipe ending in it. This scan is done in order in which pipes were installed.
// Then it makes an union of fuel tank sets each pipe scan returned. If the resulting list is not empty, it is returned as result.
//MonoBehaviour.print("for each fuel line");
int lastCount = allSources.Count;
for (int i = 0; i < this.fuelTargets.Count; i++)
{
PartSim partSim = this.fuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null) log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
partSim.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
}
}
}
// check surface mounted fuel targets
if (includeSurfaceMountedParts)
{
for (int i = 0; i < surfaceMountFuelTargets.Count; i++)
{
PartSim partSim = this.surfaceMountFuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null) log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
partSim.GetSourceSet_Old(type, true, allParts, visited, allSources, log, indent);
}
}
}
}
if (allSources.Count > lastCount)
{
if (log != null) log.Append(indent, "Returning ", (allSources.Count - lastCount), " fuel target sources (")
.AppendLine(this.name, ":", this.partId, ")");
return;
}
// Rule 3: This rule has been removed and merged with rules 4 and 7 to fix issue with fuel tanks with disabled crossfeed
// Rule 4: Part performs scan on each of its axially mounted neighbors.
// Couplers (bicoupler, tricoupler, ...) are an exception, they only scan one attach point on the single attachment side,
// skip the points on the side where multiple points are. [Experiment]
// Again, the part creates union of scan lists from each of its neighbor and if it is not empty, returns this list.
// The order in which mount points of a part are scanned appears to be fixed and defined by the part specification file. [Experiment]
if (fuelCrossFeed)
{
lastCount = allSources.Count;
//MonoBehaviour.print("for each attach node");
for (int i = 0; i < this.attachNodes.Count; i++)
{
AttachNodeSim attachSim = this.attachNodes[i];
if (attachSim.attachedPartSim != null)
{
if (attachSim.nodeType == AttachNode.NodeType.Stack)
{
if ((string.IsNullOrEmpty(noCrossFeedNodeKey) == false && attachSim.id.Contains(noCrossFeedNodeKey)) == false)
{
if (visited.Contains(attachSim.attachedPartSim))
{
if (log != null) log.Append(indent, "Attached part already visited, skipping (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding attached part as source (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
attachSim.attachedPartSim.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
}
}
}
}
}
if (allSources.Count > lastCount)
{
if (log != null) log.Append(indent, "Returning " + (allSources.Count - lastCount) + " attached sources (")
.AppendLine(this.name, ":", this.partId, ")");
return;
}
}
// Rule 5: If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled [Experiment]) and it contains fuel, it returns itself.
// Rule 6: If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled) but it does not contain the requested fuel, it returns empty list. [Experiment]
if (resources.HasType(type) && resourceFlowStates[type] > 0.0)
{
if (resources[type] > SimManager.RESOURCE_MIN)
{
allSources.Add(this);
if (log != null) log.Append(indent, "Returning enabled tank as only source (", name, ":")
.AppendLine(partId, ")");
return;
}
}
else
{
if (log != null) log.Append(indent, "Not fuel tank or disabled. HasType = ", resources.HasType(type))
.AppendLine(" FlowState = " + resourceFlowStates[type]);
}
// Rule 7: If the part is radially attached to another part and it is child of that part in the ship's tree structure, it scans its
// parent and returns whatever the parent scan returned. [Experiment] [Experiment]
if (parent != null && parentAttach == AttachModes.SRF_ATTACH)
{
if (fuelCrossFeed)
{
if (visited.Contains(parent))
{
if (log != null) log.Append(indent, "Parent part already visited, skipping (", parent.name, ":")
.AppendLine(parent.partId, ")");
}
else
{
lastCount = allSources.Count;
this.parent.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
if (allSources.Count > lastCount)
{
if (log != null) log.Append(indent, "Returning ", (allSources.Count - lastCount), " parent sources (")
.AppendLine(this.name, ":", this.partId, ")");
return;
}
}
}
}
// Rule 8: If all preceding rules failed, part returns empty list.
if (log != null) log.Append(indent, "Returning empty set, no sources found (", name, ":")
.AppendLine(partId, ")");
return;
}
public void GetSourceSet_Internal(int type, bool includeSurfaceMountedParts, List<PartSim> allParts, HashSet<PartSim> visited, HashSet<PartSim> allSources, ref int priMax, LogMsg log, String indent)
{
if (log != null)
{
log.Append(indent, "GetSourceSet_Internal(", ResourceContainer.GetResourceName(type), ") for ")
.AppendLine(name, ":", partId);
indent += " ";
}
// Rule 1: Each part can be only visited once, If it is visited for second time in particular search it returns as is.
if (visited.Contains(this))
{
if (log != null) log.Append(indent, "Nothing added, already visited (", name, ":")
.AppendLine(partId + ")");
return;
}
if (log != null) log.AppendLine(indent, "Adding this to visited");
visited.Add(this);
// Rule 2: Part performs scan on start of every fuel pipe ending in it. This scan is done in order in which pipes were installed.
// Then it makes an union of fuel tank sets each pipe scan returned. If the resulting list is not empty, it is returned as result.
//MonoBehaviour.print("for each fuel line");
int lastCount = allSources.Count;
for (int i = 0; i < this.fuelTargets.Count; i++)
{
PartSim partSim = this.fuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null) log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
partSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
if (fuelCrossFeed)
{
if (includeSurfaceMountedParts)
{
// check surface mounted fuel targets
for (int i = 0; i < surfaceMountFuelTargets.Count; i++)
{
PartSim partSim = this.surfaceMountFuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null) log.Append(indent, "Surface part already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding surface part as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
partSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
}
lastCount = allSources.Count;
//MonoBehaviour.print("for each attach node");
for (int i = 0; i < this.attachNodes.Count; i++)
{
AttachNodeSim attachSim = this.attachNodes[i];
if (attachSim.attachedPartSim != null)
{
if (attachSim.nodeType == AttachNode.NodeType.Stack)
{
if ((string.IsNullOrEmpty(noCrossFeedNodeKey) == false && attachSim.id.Contains(noCrossFeedNodeKey)) == false)
{
if (visited.Contains(attachSim.attachedPartSim))
{
if (log != null) log.Append(indent, "Attached part already visited, skipping (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
else
{
if (log != null) log.Append(indent, "Adding attached part as source (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
attachSim.attachedPartSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
}
}
}
// If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled) and it contains fuel, it adds itself.
if (resources.HasType(type) && resourceFlowStates[type] > 0.0)
{
if (resources[type] > resRequestRemainingThreshold)
{
// Get the priority of this tank
int pri = GetResourcePriority();
if (pri > priMax)
{
// This tank is higher priority than the previously added ones so we clear the sources
// and set the priMax to this priority
allSources.Clear();
priMax = pri;
}
// If this is the correct priority then add this to the sources
if (pri == priMax)
{
if (log != null) log.Append(indent, "Adding enabled tank as source (", name, ":")
.AppendLine(partId, ")");
allSources.Add(this);
}
}
}
else
{
if (log != null) log.Append(indent, "Not fuel tank or disabled. HasType = ", resources.HasType(type))
.AppendLine(" FlowState = " + resourceFlowStates[type]);
}
}
public void DumpPartToLog(LogMsg log, String prefix, List<PartSim> allParts = null)
{
if (log == null)
return;
log.Append(prefix);
log.Append(name);
log.Append(":[id = ", partId, ", decouple = ", decoupledInStage);
log.Append(", invstage = ", inverseStage);
//log.Append(", vesselName = '", vesselName, "'");
//log.Append(", vesselType = ", SimManager.GetVesselTypeString(vesselType));
//log.Append(", initialVesselName = '", initialVesselName, "'");
log.Append(", isNoPhys = ", isNoPhysics);
log.buf.AppendFormat(", baseMass = {0}", baseMass);
log.buf.AppendFormat(", baseMassForCoM = {0}", baseMassForCoM);
log.Append(", fuelCF = {0}", fuelCrossFeed);
log.Append(", noCFNKey = '{0}'", noCrossFeedNodeKey);
log.Append(", isSep = {0}", isSepratron);
for (int i = 0; i < resources.Types.Count; i++)
{
int type = resources.Types[i];
log.buf.AppendFormat(", {0} = {1:g6}", ResourceContainer.GetResourceName(type), resources[type]);
}
if (attachNodes.Count > 0)
{
log.Append(", attached = <");
attachNodes[0].DumpToLog(log);
for (int i = 1; i < attachNodes.Count; i++)
{
log.Append(", ");
attachNodes[i].DumpToLog(log);
}
log.Append(">");
}
if (surfaceMountFuelTargets.Count > 0)
{
log.Append(", surface = <");
log.Append(surfaceMountFuelTargets[0].name, ":", surfaceMountFuelTargets[0].partId);
for (int i = 1; i < surfaceMountFuelTargets.Count; i++)
{
log.Append(", ", surfaceMountFuelTargets[i].name, ":", surfaceMountFuelTargets[i].partId);
}
log.Append(">");
}
// Add more info here
log.AppendLine("]");
if (allParts != null)
{
String newPrefix = prefix + " ";
for (int i = 0; i < allParts.Count; i++)
{
PartSim partSim = allParts[i];
if (partSim.parent == this)
partSim.DumpPartToLog(log, newPrefix, allParts);
}
}
}
public bool SetResourceDrains(LogMsg log, List <PartSim> allParts, List <PartSim> allFuelLines, HashSet <PartSim> drainingParts)
{
//DumpSourcePartSets(log, "before clear");
foreach (HashSet <PartSim> sourcePartSet in sourcePartSets.Values)
{
sourcePartSet.Clear();
}
//DumpSourcePartSets(log, "after clear");
for (int index = 0; index < this.resourceConsumptionsForMass.Types.Count; index++)
{
int type = this.resourceConsumptionsForMass.Types[index];
HashSet <PartSim> sourcePartSet;
if (!sourcePartSets.TryGetValue(type, out sourcePartSet))
{
sourcePartSet = new HashSet <PartSim>();
sourcePartSets.Add(type, sourcePartSet);
}
switch ((ResourceFlowMode)this.resourceFlowModes[type])
{
case ResourceFlowMode.NO_FLOW:
if (partSim.resources[type] > SimManager.RESOURCE_MIN && partSim.resourceFlowStates[type] != 0)
{
sourcePartSet.Add(partSim);
}
break;
case ResourceFlowMode.ALL_VESSEL:
case ResourceFlowMode.ALL_VESSEL_BALANCE:
for (int i = 0; i < allParts.Count; i++)
{
PartSim aPartSim = allParts[i];
if (aPartSim.resources[type] > SimManager.RESOURCE_MIN && aPartSim.resourceFlowStates[type] != 0)
{
sourcePartSet.Add(aPartSim);
}
}
break;
case ResourceFlowMode.STAGE_PRIORITY_FLOW:
case ResourceFlowMode.STAGE_PRIORITY_FLOW_BALANCE:
if (log != null)
{
log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":", partSim.partId);
}
foreach (HashSet <PartSim> stagePartSet in stagePartSets.Values)
{
stagePartSet.Clear();
}
var maxStage = -1;
for (int i = 0; i < allParts.Count; i++)
{
var aPartSim = allParts[i];
//if (log != null) log.Append(aPartSim.name, ":" + aPartSim.partId, " contains ", aPartSim.resources[type])
// .AppendLine((aPartSim.resourceFlowStates[type] == 0) ? " (disabled)" : "");
if (aPartSim.resources[type] <= SimManager.RESOURCE_MIN || aPartSim.resourceFlowStates[type] == 0)
{
continue;
}
int stage = aPartSim.inverseStage;
if (stage > maxStage)
{
maxStage = stage;
}
HashSet <PartSim> tempPartSet;
if (!stagePartSets.TryGetValue(stage, out tempPartSet))
{
tempPartSet = new HashSet <PartSim>();
stagePartSets.Add(stage, tempPartSet);
}
tempPartSet.Add(aPartSim);
}
for (int j = maxStage; j >= -1; j--)
{
//if (log != null) log.AppendLine("Testing stage ", j);
HashSet <PartSim> stagePartSet;
if (stagePartSets.TryGetValue(j, out stagePartSet) && stagePartSet.Count > 0)
{
//if (log != null) log.AppendLine("Not empty");
// We have to copy the contents of the set here rather than copying the set reference or
// bad things (tm) happen
foreach (PartSim aPartSim in stagePartSet)
{
sourcePartSet.Add(aPartSim);
}
break;
}
}
break;
case ResourceFlowMode.STACK_PRIORITY_SEARCH:
case ResourceFlowMode.STAGE_STACK_FLOW:
case ResourceFlowMode.STAGE_STACK_FLOW_BALANCE:
visited.Clear();
if (log != null)
{
log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":", partSim.partId);
}
partSim.GetSourceSet(type, allParts, visited, sourcePartSet, log, "");
break;
default:
if (log != null)
{
log.Append("SetResourceDrains(", partSim.name, ":", partSim.partId)
.AppendLine(") Unexpected flow type for ", ResourceContainer.GetResourceName(type), ")");
}
break;
}
if (log != null && sourcePartSet.Count > 0)
{
log.AppendLine("Source parts for ", ResourceContainer.GetResourceName(type), ":");
foreach (PartSim partSim in sourcePartSet)
{
log.AppendLine(partSim.name, ":", partSim.partId);
}
}
//DumpSourcePartSets(log, "after " + ResourceContainer.GetResourceName(type));
}
// If we don't have sources for all the needed resources then return false without setting up any drains
for (int i = 0; i < this.resourceConsumptionsForMass.Types.Count; i++)
{
int type = this.resourceConsumptionsForMass.Types[i];
HashSet <PartSim> sourcePartSet;
if (!sourcePartSets.TryGetValue(type, out sourcePartSet) || sourcePartSet.Count == 0)
{
if (log != null)
{
log.AppendLine("No source of ", ResourceContainer.GetResourceName(type));
}
isActive = false;
return(false);
}
}
// Now we set the drains on the members of the sets and update the draining parts set
for (int i = 0; i < this.resourceConsumptionsForMass.Types.Count; i++)
{
int type = this.resourceConsumptionsForMass.Types[i];
HashSet <PartSim> sourcePartSet = sourcePartSets[type];
ResourceFlowMode mode = (ResourceFlowMode)resourceFlowModes[type];
double consumption = resourceConsumptionsForMass[type];
double amount = 0d;
double total = 0d;
if (mode == ResourceFlowMode.ALL_VESSEL_BALANCE ||
mode == ResourceFlowMode.STAGE_PRIORITY_FLOW_BALANCE ||
mode == ResourceFlowMode.STAGE_STACK_FLOW_BALANCE ||
mode == ResourceFlowMode.STACK_PRIORITY_SEARCH)
{
foreach (PartSim partSim in sourcePartSet)
{
total += partSim.resources[type];
}
}
else
{
amount = consumption / sourcePartSet.Count;
}
// Loop through the members of the set
foreach (PartSim partSim in sourcePartSet)
{
if (total != 0d)
{
amount = consumption * partSim.resources[type] / total;
}
if (log != null)
{
log.Append("Adding drain of ", amount, " ", ResourceContainer.GetResourceName(type))
.AppendLine(" to ", partSim.name, ":", partSim.partId);
}
partSim.resourceDrains.Add(type, amount);
drainingParts.Add(partSim);
}
}
return(true);
}
// This function runs the simulation and returns a newly created array of Stage objects
public Stage[] RunSimulation(LogMsg _log)
{
log = _log;
if (log != null)
{
log.AppendLine("RunSimulation started");
}
_timer.Reset();
_timer.Start();
// Start with the last stage to simulate
// (this is in a member variable so it can be accessed by AllowedToStage and ActivateStage)
currentStage = lastStage;
// Work out which engines would be active if just doing the staging and if this is different to the
// currently active engines then generate an extra stage
// Loop through all the engines
bool anyActive = false;
for (int i = 0; i < allEngines.Count; ++i)
{
EngineSim engine = allEngines[i];
if (log != null)
{
log.AppendLine("Testing engine mod of ", engine.partSim.name, ":", engine.partSim.partId);
}
bool bActive = engine.isActive;
bool bStage = (engine.partSim.inverseStage >= currentStage);
if (log != null)
{
log.AppendLine("bActive = ", bActive, " bStage = ", bStage);
}
if (HighLogic.LoadedSceneIsFlight)
{
if (bActive)
{
anyActive = true;
}
if (bActive != bStage)
{
// If the active state is different to the state due to staging
if (log != null)
{
log.AppendLine("Need to do current active engines first");
}
doingCurrent = true;
}
}
else
{
if (bStage)
{
if (log != null)
{
log.AppendLine("Marking as active");
}
engine.isActive = true;
}
}
}
// If we need to do current because of difference in engine activation and there actually are active engines
// then we do the extra stage otherwise activate the next stage and don't treat it as current
if (doingCurrent && anyActive)
{
currentStage++;
}
else
{
ActivateStage();
doingCurrent = false;
}
// Create a list of lists of PartSims that prevent decoupling
BuildDontStageLists(log);
if (log != null)
{
log.Flush();
}
// Create the array of stages that will be returned
Stage[] stages = new Stage[currentStage + 1];
int startStage = currentStage;
// Loop through the stages
while (currentStage >= 0)
{
if (log != null)
{
log.AppendLine("Simulating stage ", currentStage);
log.Flush();
_timer.Reset();
_timer.Start();
}
// Update active engines and resource drains
UpdateResourceDrains();
// Update the masses of the parts to correctly handle "no physics" parts
stageStartMass = UpdatePartMasses();
if (log != null)
{
allParts[0].DumpPartToLog(log, "", allParts);
}
// Create the Stage object for this stage
Stage stage = new Stage();
stageTime = 0d;
vecStageDeltaV = Vector3.zero;
stageStartCom = ShipCom;
stepStartMass = stageStartMass;
stepEndMass = 0;
CalculateThrustAndISP();
// Store various things in the Stage object
stage.thrust = totalStageThrust;
stage.thrustToWeight = totalStageThrust / (stageStartMass * gravity);
stage.maxThrustToWeight = stage.thrustToWeight;
stage.actualThrust = totalStageActualThrust;
stage.actualThrustToWeight = totalStageActualThrust / (stageStartMass * gravity);
CalculateRCS(gravity, false);
stage.RCSIsp = RCSIsp;
stage.RCSThrust = RCSThrust;
stage.RCSdeltaVStart = RCSDeltaV;
stage.RCSTWRStart = RCSTWR;
stage.RCSBurnTime = RCSBurnTime;
if (log != null)
{
log.AppendLine("stage.thrust = ", stage.thrust);
log.AppendLine("StageMass = ", stageStartMass);
log.AppendLine("Initial maxTWR = ", stage.maxThrustToWeight);
}
// calculate torque and associates
stage.maxThrustTorque = totalStageThrustForce.TorqueAt(stageStartCom).magnitude;
// torque divided by thrust. imagine that all engines are at the end of a lever that tries to turn the ship.
// this numerical value, in meters, would represent the length of that lever.
double torqueLeverArmLength = (stage.thrust <= 0) ? 0 : stage.maxThrustTorque / stage.thrust;
// how far away are the engines from the CoM, actually?
double thrustDistance = (stageStartCom - totalStageThrustForce.GetAverageForceApplicationPoint()).magnitude;
// the combination of the above two values gives an approximation of the offset angle.
double sinThrustOffsetAngle = 0;
if (thrustDistance > 1e-7)
{
sinThrustOffsetAngle = torqueLeverArmLength / thrustDistance;
if (sinThrustOffsetAngle > 1)
{
sinThrustOffsetAngle = 1;
}
}
stage.thrustOffsetAngle = Math.Asin(sinThrustOffsetAngle) * 180 / Math.PI;
// Calculate the total cost of the vessel at this point
stage.totalCost = 0d;
for (int i = 0; i < allParts.Count; ++i)
{
if (currentStage > allParts[i].decoupledInStage)
{
stage.totalCost += allParts[i].GetCost(currentStage);
}
}
// The total mass is simply the mass at the start of the stage
stage.totalMass = stageStartMass;
// If we have done a previous stage
if (currentStage < startStage)
{
// Calculate what the previous stage's mass and cost were by subtraction
Stage prev = stages[currentStage + 1];
prev.cost = prev.totalCost - stage.totalCost;
prev.mass = prev.totalMass - stage.totalMass;
}
// The above code will never run for the last stage so set those directly
if (currentStage == 0)
{
stage.cost = stage.totalCost;
stage.mass = stage.totalMass;
}
dontStageParts = dontStagePartsLists[currentStage];
if (log != null)
{
log.AppendLine("Stage setup took ", _timer.ElapsedMilliseconds, "ms");
if (dontStageParts.Count > 0)
{
log.AppendLine("Parts preventing staging:");
for (int i = 0; i < dontStageParts.Count; i++)
{
PartSim partSim = dontStageParts[i];
partSim.DumpPartToLog(log, "");
}
}
else
{
log.AppendLine("No parts preventing staging");
}
log.Flush();
}
// Now we will loop until we are allowed to stage
int loopCounter = 0;
while (!AllowedToStage())
{
loopCounter++;
//if (log != null) log.AppendLine("loop = ", loopCounter);
// Calculate how long each draining tank will take to drain and run for the minimum time
double resourceDrainTime = double.MaxValue;
PartSim partMinDrain = null;
foreach (PartSim partSim in drainingParts)
{
double time = partSim.TimeToDrainResource(log);
if (time < resourceDrainTime)
{
resourceDrainTime = time;
partMinDrain = partSim;
}
}
if (log != null)
{
log.Append("Drain time = ", resourceDrainTime, " (", partMinDrain.name)
.AppendLine(":", partMinDrain.partId, ")");
}
foreach (PartSim partSim in drainingParts)
{
partSim.DrainResources(resourceDrainTime, log);
}
// Get the mass after draining
stepEndMass = ShipMass;
stageTime += resourceDrainTime;
double stepEndTWR = totalStageThrust / (stepEndMass * gravity);
/*if (log != null)
* {
* log.AppendLine("After drain mass = ", stepEndMass);
* log.AppendLine("currentThrust = ", totalStageThrust);
* log.AppendLine("currentTWR = ", stepEndTWR);
* }*/
if (stepEndTWR > stage.maxThrustToWeight)
{
stage.maxThrustToWeight = stepEndTWR;
}
//if (log != null) log.AppendLine("newMaxTWR = ", stage.maxThrustToWeight);
// If we have drained anything and the masses make sense then add this step's deltaV to the stage total
if (resourceDrainTime > 0d && stepStartMass > stepEndMass && stepStartMass > 0d && stepEndMass > 0d)
{
vecStageDeltaV += vecThrust * (float)((currentisp * Units.GRAVITY * Math.Log(stepStartMass / stepEndMass)) / simpleTotalThrust);
}
// Update the active engines and resource drains for the next step
UpdateResourceDrains();
// Recalculate the current thrust and isp for the next step
CalculateThrustAndISP();
// Check if we actually changed anything
if (stepStartMass == stepEndMass)
{
//MonoBehaviour.print("No change in mass");
break;
}
// Check to stop rampant looping
if (loopCounter == 1000)
{
if (log != null)
{
log.AppendLine("exceeded loop count");
log.AppendLine("stageStartMass = " + stageStartMass);
log.AppendLine("stepStartMass = " + stepStartMass);
log.AppendLine("StepEndMass = " + stepEndMass);
}
break;
}
// The next step starts at the mass this one ended at
stepStartMass = stepEndMass;
}
// Store more values in the Stage object and stick it in the array
// Store the magnitude of the deltaV vector
stage.deltaV = vecStageDeltaV.magnitude;
stage.resourceMass = stageStartMass - stepEndMass;
if (HighLogic.LoadedSceneIsEditor) //this is only needed in the VAB.
{
CalculateRCS(gravity, true);
}
stage.RCSdeltaVEnd = RCSDeltaV;
stage.RCSTWREnd = RCSTWR;
// Recalculate effective stage isp from the stage deltaV (flip the standard deltaV calculation around)
// Note: If the mass doesn't change then this is a divide by zero
if (stageStartMass != stepStartMass)
{
stage.isp = stage.deltaV / (Units.GRAVITY * Math.Log(stageStartMass / stepStartMass));
}
else
{
stage.isp = 0;
}
// Zero stage time if more than a day (this should be moved into the window code)
stage.time = (stageTime < SECONDS_PER_DAY) ? stageTime : 0d;
stage.number = doingCurrent ? -1 : currentStage; // Set the stage number to -1 if doing current engines
stage.totalPartCount = allParts.Count;
stage.maxMach = maxMach;
stages[currentStage] = stage;
// Now activate the next stage
currentStage--;
doingCurrent = false;
if (log != null)
{
// Log how long the stage took
_timer.Stop();
log.AppendLine("Simulating stage took ", _timer.ElapsedMilliseconds, "ms");
stage.Dump(log);
_timer.Reset();
_timer.Start();
}
// Activate the next stage
ActivateStage();
if (log != null)
{
// Log how long it took to activate
_timer.Stop();
log.AppendLine("ActivateStage took ", _timer.ElapsedMilliseconds, "ms");
}
}
// Now we add up the various total fields in the stages
for (int i = 0; i < stages.Length; i++)
{
// For each stage we total up the cost, mass, deltaV and time for this stage and all the stages above
for (int j = i; j >= 0; j--)
{
stages[i].totalDeltaV += stages[j].deltaV;
stages[i].totalResourceMass += stages[j].resourceMass;
stages[i].totalTime += stages[j].time;
stages[i].partCount = i > 0 ? stages[i].totalPartCount - stages[i - 1].totalPartCount : stages[i].totalPartCount;
}
// We also total up the deltaV for stage and all stages below
for (int j = i; j < stages.Length; j++)
{
stages[i].inverseTotalDeltaV += stages[j].deltaV;
}
// Zero the total time if the value will be huge (24 hours?) to avoid the display going weird
// (this should be moved into the window code)
if (stages[i].totalTime > SECONDS_PER_DAY)
{
stages[i].totalTime = 0d;
}
}
FreePooledObject();
_timer.Stop();
if (log != null)
{
log.AppendLine("RunSimulation: ", _timer.ElapsedMilliseconds, "ms");
log.Flush();
}
log = null;
return(stages);
}
// This is the old recursive function for STACK_PRIORITY_SEARCH
public void GetSourceSet_Old(int type, bool includeSurfaceMountedParts, List <PartSim> allParts, HashSet <PartSim> visited, HashSet <PartSim> allSources, LogMsg log, String indent)
{
if (log != null)
{
log.Append(indent, "GetSourceSet_Old(", ResourceContainer.GetResourceName(type), ") for ")
.AppendLine(name, ":", partId);
indent += " ";
}
// Rule 1: Each part can be only visited once, If it is visited for second time in particular search it returns as is.
if (visited.Contains(this))
{
if (log != null)
{
log.Append(indent, "Returning empty set, already visited (", name, ":")
.AppendLine(partId + ")");
}
return;
}
if (log != null)
{
log.AppendLine(indent, "Adding this to visited");
}
visited.Add(this);
// Rule 2: Part performs scan on start of every fuel pipe ending in it. This scan is done in order in which pipes were installed.
// Then it makes an union of fuel tank sets each pipe scan returned. If the resulting list is not empty, it is returned as result.
//MonoBehaviour.print("for each fuel line");
int lastCount = allSources.Count;
for (int i = 0; i < this.fuelTargets.Count; i++)
{
PartSim partSim = this.fuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null)
{
log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
partSim.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
}
}
}
// check surface mounted fuel targets
if (includeSurfaceMountedParts)
{
for (int i = 0; i < surfaceMountFuelTargets.Count; i++)
{
PartSim partSim = this.surfaceMountFuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null)
{
log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
partSim.GetSourceSet_Old(type, true, allParts, visited, allSources, log, indent);
}
}
}
}
if (allSources.Count > lastCount)
{
if (log != null)
{
log.Append(indent, "Returning ", (allSources.Count - lastCount), " fuel target sources (")
.AppendLine(this.name, ":", this.partId, ")");
}
return;
}
// Rule 3: This rule has been removed and merged with rules 4 and 7 to fix issue with fuel tanks with disabled crossfeed
// Rule 4: Part performs scan on each of its axially mounted neighbors.
// Couplers (bicoupler, tricoupler, ...) are an exception, they only scan one attach point on the single attachment side,
// skip the points on the side where multiple points are. [Experiment]
// Again, the part creates union of scan lists from each of its neighbor and if it is not empty, returns this list.
// The order in which mount points of a part are scanned appears to be fixed and defined by the part specification file. [Experiment]
if (fuelCrossFeed)
{
lastCount = allSources.Count;
//MonoBehaviour.print("for each attach node");
for (int i = 0; i < this.attachNodes.Count; i++)
{
AttachNodeSim attachSim = this.attachNodes[i];
if (attachSim.attachedPartSim != null)
{
if (attachSim.nodeType == AttachNode.NodeType.Stack)
{
if ((string.IsNullOrEmpty(noCrossFeedNodeKey) == false && attachSim.id.Contains(noCrossFeedNodeKey)) == false)
{
if (visited.Contains(attachSim.attachedPartSim))
{
if (log != null)
{
log.Append(indent, "Attached part already visited, skipping (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding attached part as source (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
attachSim.attachedPartSim.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
}
}
}
}
}
if (allSources.Count > lastCount)
{
if (log != null)
{
log.Append(indent, "Returning " + (allSources.Count - lastCount) + " attached sources (")
.AppendLine(this.name, ":", this.partId, ")");
}
return;
}
}
// Rule 5: If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled [Experiment]) and it contains fuel, it returns itself.
// Rule 6: If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled) but it does not contain the requested fuel, it returns empty list. [Experiment]
if (resources.HasType(type) && resourceFlowStates[type] > 0.0)
{
if (resources[type] > SimManager.RESOURCE_MIN)
{
allSources.Add(this);
if (log != null)
{
log.Append(indent, "Returning enabled tank as only source (", name, ":")
.AppendLine(partId, ")");
}
return;
}
}
else
{
if (log != null)
{
log.Append(indent, "Not fuel tank or disabled. HasType = ", resources.HasType(type))
.AppendLine(" FlowState = " + resourceFlowStates[type]);
}
}
// Rule 7: If the part is radially attached to another part and it is child of that part in the ship's tree structure, it scans its
// parent and returns whatever the parent scan returned. [Experiment] [Experiment]
if (parent != null && parentAttach == AttachModes.SRF_ATTACH)
{
if (fuelCrossFeed)
{
if (visited.Contains(parent))
{
if (log != null)
{
log.Append(indent, "Parent part already visited, skipping (", parent.name, ":")
.AppendLine(parent.partId, ")");
}
}
else
{
lastCount = allSources.Count;
this.parent.GetSourceSet_Old(type, includeSurfaceMountedParts, allParts, visited, allSources, log, indent);
if (allSources.Count > lastCount)
{
if (log != null)
{
log.Append(indent, "Returning ", (allSources.Count - lastCount), " parent sources (")
.AppendLine(this.name, ":", this.partId, ")");
}
return;
}
}
}
}
// Rule 8: If all preceding rules failed, part returns empty list.
if (log != null)
{
log.Append(indent, "Returning empty set, no sources found (", name, ":")
.AppendLine(partId, ")");
}
return;
}
public void GetSourceSet_Internal(int type, bool includeSurfaceMountedParts, List <PartSim> allParts, HashSet <PartSim> visited, HashSet <PartSim> allSources, ref int priMax, LogMsg log, String indent)
{
if (log != null)
{
log.Append(indent, "GetSourceSet_Internal(", ResourceContainer.GetResourceName(type), ") for ")
.AppendLine(name, ":", partId);
indent += " ";
}
// Rule 1: Each part can be only visited once, If it is visited for second time in particular search it returns as is.
if (visited.Contains(this))
{
if (log != null)
{
log.Append(indent, "Nothing added, already visited (", name, ":")
.AppendLine(partId + ")");
}
return;
}
if (log != null)
{
log.AppendLine(indent, "Adding this to visited");
}
visited.Add(this);
// Rule 2: Part performs scan on start of every fuel pipe ending in it. This scan is done in order in which pipes were installed.
// Then it makes an union of fuel tank sets each pipe scan returned. If the resulting list is not empty, it is returned as result.
//MonoBehaviour.print("for each fuel line");
int lastCount = allSources.Count;
for (int i = 0; i < this.fuelTargets.Count; i++)
{
PartSim partSim = this.fuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null)
{
log.Append(indent, "Fuel target already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding fuel target as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
partSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
if (fuelCrossFeed)
{
if (includeSurfaceMountedParts)
{
// check surface mounted fuel targets
for (int i = 0; i < surfaceMountFuelTargets.Count; i++)
{
PartSim partSim = this.surfaceMountFuelTargets[i];
if (partSim != null)
{
if (visited.Contains(partSim))
{
if (log != null)
{
log.Append(indent, "Surface part already visited, skipping (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding surface part as source (", partSim.name, ":")
.AppendLine(partSim.partId, ")");
}
partSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
}
lastCount = allSources.Count;
//MonoBehaviour.print("for each attach node");
for (int i = 0; i < this.attachNodes.Count; i++)
{
AttachNodeSim attachSim = this.attachNodes[i];
if (attachSim.attachedPartSim != null)
{
if (attachSim.nodeType == AttachNode.NodeType.Stack)
{
if ((string.IsNullOrEmpty(noCrossFeedNodeKey) == false && attachSim.id.Contains(noCrossFeedNodeKey)) == false)
{
if (visited.Contains(attachSim.attachedPartSim))
{
if (log != null)
{
log.Append(indent, "Attached part already visited, skipping (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
}
else
{
if (log != null)
{
log.Append(indent, "Adding attached part as source (", attachSim.attachedPartSim.name, ":")
.AppendLine(attachSim.attachedPartSim.partId, ")");
}
attachSim.attachedPartSim.GetSourceSet_Internal(type, includeSurfaceMountedParts, allParts, visited, allSources, ref priMax, log, indent);
}
}
}
}
}
}
// If the part is fuel container for searched type of fuel (i.e. it has capability to contain that type of fuel and the fuel
// type was not disabled) and it contains fuel, it adds itself.
if (resources.HasType(type) && resourceFlowStates[type] > 0.0)
{
if (resources[type] > resRequestRemainingThreshold)
{
// Get the priority of this tank
int pri = GetResourcePriority();
if (pri > priMax)
{
// This tank is higher priority than the previously added ones so we clear the sources
// and set the priMax to this priority
allSources.Clear();
priMax = pri;
}
// If this is the correct priority then add this to the sources
if (pri == priMax)
{
if (log != null)
{
log.Append(indent, "Adding enabled tank as source (", name, ":")
.AppendLine(partId, ")");
}
allSources.Add(this);
}
}
}
else
{
if (log != null)
{
log.Append(indent, "Not fuel tank or disabled. HasType = ", resources.HasType(type))
.AppendLine(" FlowState = " + resourceFlowStates[type]);
}
}
}
public void DumpPartToLog(LogMsg log, String prefix, List <PartSim> allParts = null)
{
if (log == null)
{
return;
}
log.Append(prefix);
log.Append(name);
log.Append(":[id = ", partId, ", decouple = ", decoupledInStage);
log.Append(", invstage = ", inverseStage);
//log.Append(", vesselName = '", vesselName, "'");
//log.Append(", vesselType = ", SimManager.GetVesselTypeString(vesselType));
//log.Append(", initialVesselName = '", initialVesselName, "'");
log.Append(", isNoPhys = ", isNoPhysics);
log.buf.AppendFormat(", baseMass = {0}", baseMass);
log.buf.AppendFormat(", baseMassForCoM = {0}", baseMassForCoM);
log.Append(", fuelCF = {0}", fuelCrossFeed);
log.Append(", noCFNKey = '{0}'", noCrossFeedNodeKey);
log.Append(", isSep = {0}", isSepratron);
for (int i = 0; i < resources.Types.Count; i++)
{
int type = resources.Types[i];
log.buf.AppendFormat(", {0} = {1:g6}", ResourceContainer.GetResourceName(type), resources[type]);
}
if (attachNodes.Count > 0)
{
log.Append(", attached = <");
attachNodes[0].DumpToLog(log);
for (int i = 1; i < attachNodes.Count; i++)
{
log.Append(", ");
attachNodes[i].DumpToLog(log);
}
log.Append(">");
}
if (surfaceMountFuelTargets.Count > 0)
{
log.Append(", surface = <");
log.Append(surfaceMountFuelTargets[0].name, ":", surfaceMountFuelTargets[0].partId);
for (int i = 1; i < surfaceMountFuelTargets.Count; i++)
{
log.Append(", ", surfaceMountFuelTargets[i].name, ":", surfaceMountFuelTargets[i].partId);
}
log.Append(">");
}
// Add more info here
log.AppendLine("]");
if (allParts != null)
{
String newPrefix = prefix + " ";
for (int i = 0; i < allParts.Count; i++)
{
PartSim partSim = allParts[i];
if (partSim.parent == this)
{
partSim.DumpPartToLog(log, newPrefix, allParts);
}
}
}
}
// This function runs the simulation and returns a newly created array of Stage objects
public Stage[] RunSimulation(LogMsg _log)
{
log = _log;
if (log != null) log.AppendLine("RunSimulation started");
_timer.Reset();
_timer.Start();
// Start with the last stage to simulate
// (this is in a member variable so it can be accessed by AllowedToStage and ActivateStage)
currentStage = lastStage;
// Work out which engines would be active if just doing the staging and if this is different to the
// currently active engines then generate an extra stage
// Loop through all the engines
bool anyActive = false;
for (int i = 0; i < allEngines.Count; ++i)
{
EngineSim engine = allEngines[i];
if (log != null) log.AppendLine("Testing engine mod of ", engine.partSim.name, ":", engine.partSim.partId);
bool bActive = engine.isActive;
bool bStage = (engine.partSim.inverseStage >= currentStage);
if (log != null) log.AppendLine("bActive = ", bActive, " bStage = ", bStage);
if (HighLogic.LoadedSceneIsFlight)
{
if (bActive)
{
anyActive = true;
}
if (bActive != bStage)
{
// If the active state is different to the state due to staging
if (log != null) log.AppendLine("Need to do current active engines first");
doingCurrent = true;
}
}
else
{
if (bStage)
{
if (log != null) log.AppendLine("Marking as active");
engine.isActive = true;
}
}
}
// If we need to do current because of difference in engine activation and there actually are active engines
// then we do the extra stage otherwise activate the next stage and don't treat it as current
if (doingCurrent && anyActive)
{
currentStage++;
}
else
{
ActivateStage();
doingCurrent = false;
}
// Create a list of lists of PartSims that prevent decoupling
BuildDontStageLists(log);
if (log != null) log.Flush();
// Create the array of stages that will be returned
Stage[] stages = new Stage[currentStage + 1];
int startStage = currentStage;
// Loop through the stages
while (currentStage >= 0)
{
if (log != null)
{
log.AppendLine("Simulating stage ", currentStage);
log.Flush();
_timer.Reset();
_timer.Start();
}
// Update active engines and resource drains
UpdateResourceDrains();
// Update the masses of the parts to correctly handle "no physics" parts
stageStartMass = UpdatePartMasses();
if (log != null)
allParts[0].DumpPartToLog(log, "", allParts);
// Create the Stage object for this stage
Stage stage = new Stage();
stageTime = 0d;
vecStageDeltaV = Vector3.zero;
stageStartCom = ShipCom;
stepStartMass = stageStartMass;
stepEndMass = 0;
CalculateThrustAndISP();
// Store various things in the Stage object
stage.thrust = totalStageThrust;
stage.thrustToWeight = totalStageThrust / (stageStartMass * gravity);
stage.maxThrustToWeight = stage.thrustToWeight;
stage.actualThrust = totalStageActualThrust;
stage.actualThrustToWeight = totalStageActualThrust / (stageStartMass * gravity);
if (log != null)
{
log.AppendLine("stage.thrust = ", stage.thrust);
log.AppendLine("StageMass = ", stageStartMass);
log.AppendLine("Initial maxTWR = ", stage.maxThrustToWeight);
}
// calculate torque and associates
stage.maxThrustTorque = totalStageThrustForce.TorqueAt(stageStartCom).magnitude;
// torque divided by thrust. imagine that all engines are at the end of a lever that tries to turn the ship.
// this numerical value, in meters, would represent the length of that lever.
double torqueLeverArmLength = (stage.thrust <= 0) ? 0 : stage.maxThrustTorque / stage.thrust;
// how far away are the engines from the CoM, actually?
double thrustDistance = (stageStartCom - totalStageThrustForce.GetAverageForceApplicationPoint()).magnitude;
// the combination of the above two values gives an approximation of the offset angle.
double sinThrustOffsetAngle = 0;
if (thrustDistance > 1e-7) {
sinThrustOffsetAngle = torqueLeverArmLength / thrustDistance;
if (sinThrustOffsetAngle > 1) {
sinThrustOffsetAngle = 1;
}
}
stage.thrustOffsetAngle = Math.Asin(sinThrustOffsetAngle) * 180 / Math.PI;
// Calculate the total cost of the vessel at this point
stage.totalCost = 0d;
for (int i = 0; i < allParts.Count; ++i)
{
if (currentStage > allParts[i].decoupledInStage)
stage.totalCost += allParts[i].GetCost(currentStage);
}
// The total mass is simply the mass at the start of the stage
stage.totalMass = stageStartMass;
// If we have done a previous stage
if (currentStage < startStage)
{
// Calculate what the previous stage's mass and cost were by subtraction
Stage prev = stages[currentStage + 1];
prev.cost = prev.totalCost - stage.totalCost;
prev.mass = prev.totalMass - stage.totalMass;
}
// The above code will never run for the last stage so set those directly
if (currentStage == 0)
{
stage.cost = stage.totalCost;
stage.mass = stage.totalMass;
}
dontStageParts = dontStagePartsLists[currentStage];
if (log != null)
{
log.AppendLine("Stage setup took ", _timer.ElapsedMilliseconds, "ms");
if (dontStageParts.Count > 0)
{
log.AppendLine("Parts preventing staging:");
for (int i = 0; i < dontStageParts.Count; i++)
{
PartSim partSim = dontStageParts[i];
partSim.DumpPartToLog(log, "");
}
}
else
{
log.AppendLine("No parts preventing staging");
}
log.Flush();
}
// Now we will loop until we are allowed to stage
int loopCounter = 0;
while (!AllowedToStage())
{
loopCounter++;
//if (log != null) log.AppendLine("loop = ", loopCounter);
// Calculate how long each draining tank will take to drain and run for the minimum time
double resourceDrainTime = double.MaxValue;
PartSim partMinDrain = null;
foreach (PartSim partSim in drainingParts)
{
double time = partSim.TimeToDrainResource(log);
if (time < resourceDrainTime)
{
resourceDrainTime = time;
partMinDrain = partSim;
}
}
if (log != null) log.Append("Drain time = ", resourceDrainTime, " (", partMinDrain.name)
.AppendLine(":", partMinDrain.partId, ")");
foreach (PartSim partSim in drainingParts)
{
partSim.DrainResources(resourceDrainTime, log);
}
// Get the mass after draining
stepEndMass = ShipMass;
stageTime += resourceDrainTime;
double stepEndTWR = totalStageThrust / (stepEndMass * gravity);
/*if (log != null)
{
log.AppendLine("After drain mass = ", stepEndMass);
log.AppendLine("currentThrust = ", totalStageThrust);
log.AppendLine("currentTWR = ", stepEndTWR);
}*/
if (stepEndTWR > stage.maxThrustToWeight)
{
stage.maxThrustToWeight = stepEndTWR;
}
//if (log != null) log.AppendLine("newMaxTWR = ", stage.maxThrustToWeight);
// If we have drained anything and the masses make sense then add this step's deltaV to the stage total
if (resourceDrainTime > 0d && stepStartMass > stepEndMass && stepStartMass > 0d && stepEndMass > 0d)
{
vecStageDeltaV += vecThrust * (float)((currentisp * Units.GRAVITY * Math.Log(stepStartMass / stepEndMass)) / simpleTotalThrust);
}
// Update the active engines and resource drains for the next step
UpdateResourceDrains();
// Recalculate the current thrust and isp for the next step
CalculateThrustAndISP();
// Check if we actually changed anything
if (stepStartMass == stepEndMass)
{
//MonoBehaviour.print("No change in mass");
break;
}
// Check to stop rampant looping
if (loopCounter == 1000)
{
if (log != null)
{
log.AppendLine("exceeded loop count");
log.AppendLine("stageStartMass = " + stageStartMass);
log.AppendLine("stepStartMass = " + stepStartMass);
log.AppendLine("StepEndMass = " + stepEndMass);
}
break;
}
// The next step starts at the mass this one ended at
stepStartMass = stepEndMass;
}
// Store more values in the Stage object and stick it in the array
// Store the magnitude of the deltaV vector
stage.deltaV = vecStageDeltaV.magnitude;
stage.resourceMass = stageStartMass - stepEndMass;
// Recalculate effective stage isp from the stage deltaV (flip the standard deltaV calculation around)
// Note: If the mass doesn't change then this is a divide by zero
if (stageStartMass != stepStartMass)
{
stage.isp = stage.deltaV / (Units.GRAVITY * Math.Log(stageStartMass / stepStartMass));
}
else
{
stage.isp = 0;
}
// Zero stage time if more than a day (this should be moved into the window code)
stage.time = (stageTime < SECONDS_PER_DAY) ? stageTime : 0d;
stage.number = doingCurrent ? -1 : currentStage; // Set the stage number to -1 if doing current engines
stage.totalPartCount = allParts.Count;
stage.maxMach = maxMach;
stages[currentStage] = stage;
// Now activate the next stage
currentStage--;
doingCurrent = false;
if (log != null)
{
// Log how long the stage took
_timer.Stop();
log.AppendLine("Simulating stage took ", _timer.ElapsedMilliseconds, "ms");
stage.Dump(log);
_timer.Reset();
_timer.Start();
}
// Activate the next stage
ActivateStage();
if (log != null)
{
// Log how long it took to activate
_timer.Stop();
log.AppendLine("ActivateStage took ", _timer.ElapsedMilliseconds, "ms");
}
}
// Now we add up the various total fields in the stages
for (int i = 0; i < stages.Length; i++)
{
// For each stage we total up the cost, mass, deltaV and time for this stage and all the stages above
for (int j = i; j >= 0; j--)
{
stages[i].totalDeltaV += stages[j].deltaV;
stages[i].totalTime += stages[j].time;
stages[i].partCount = i > 0 ? stages[i].totalPartCount - stages[i - 1].totalPartCount : stages[i].totalPartCount;
}
// We also total up the deltaV for stage and all stages below
for (int j = i; j < stages.Length; j++)
{
stages[i].inverseTotalDeltaV += stages[j].deltaV;
}
// Zero the total time if the value will be huge (24 hours?) to avoid the display going weird
// (this should be moved into the window code)
if (stages[i].totalTime > SECONDS_PER_DAY)
{
stages[i].totalTime = 0d;
}
}
FreePooledObject();
_timer.Stop();
if (log != null)
{
log.AppendLine("RunSimulation: ", _timer.ElapsedMilliseconds, "ms");
log.Flush();
}
log = null;
return stages;
}
public bool SetResourceDrains(LogMsg log, List<PartSim> allParts, List<PartSim> allFuelLines, HashSet<PartSim> drainingParts)
{
//DumpSourcePartSets(log, "before clear");
foreach (HashSet<PartSim> sourcePartSet in sourcePartSets.Values)
{
sourcePartSet.Clear();
}
//DumpSourcePartSets(log, "after clear");
for (int index = 0; index < this.resourceConsumptions.Types.Count; index++)
{
int type = this.resourceConsumptions.Types[index];
HashSet<PartSim> sourcePartSet;
if (!sourcePartSets.TryGetValue(type, out sourcePartSet))
{
sourcePartSet = new HashSet<PartSim>();
sourcePartSets.Add(type, sourcePartSet);
}
switch ((ResourceFlowMode)this.resourceFlowModes[type])
{
case ResourceFlowMode.NO_FLOW:
if (partSim.resources[type] > SimManager.RESOURCE_MIN && partSim.resourceFlowStates[type] != 0)
{
sourcePartSet.Add(partSim);
}
break;
case ResourceFlowMode.ALL_VESSEL:
case ResourceFlowMode.ALL_VESSEL_BALANCE:
for (int i = 0; i < allParts.Count; i++)
{
PartSim aPartSim = allParts[i];
if (aPartSim.resources[type] > SimManager.RESOURCE_MIN && aPartSim.resourceFlowStates[type] != 0)
{
sourcePartSet.Add(aPartSim);
}
}
break;
case ResourceFlowMode.STAGE_PRIORITY_FLOW:
case ResourceFlowMode.STAGE_PRIORITY_FLOW_BALANCE:
if (log != null) log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":" , partSim.partId);
foreach (HashSet<PartSim> stagePartSet in stagePartSets.Values)
{
stagePartSet.Clear();
}
var maxStage = -1;
for (int i = 0; i < allParts.Count; i++)
{
var aPartSim = allParts[i];
//if (log != null) log.Append(aPartSim.name, ":" + aPartSim.partId, " contains ", aPartSim.resources[type])
// .AppendLine((aPartSim.resourceFlowStates[type] == 0) ? " (disabled)" : "");
if (aPartSim.resources[type] <= SimManager.RESOURCE_MIN || aPartSim.resourceFlowStates[type] == 0)
{
continue;
}
int stage = aPartSim.inverseStage;
if (stage > maxStage)
{
maxStage = stage;
}
HashSet<PartSim> tempPartSet;
if (!stagePartSets.TryGetValue(stage, out tempPartSet))
{
tempPartSet = new HashSet<PartSim>();
stagePartSets.Add(stage, tempPartSet);
}
tempPartSet.Add(aPartSim);
}
for (int j = maxStage; j >= -1; j--)
{
//if (log != null) log.AppendLine("Testing stage ", j);
HashSet<PartSim> stagePartSet;
if (stagePartSets.TryGetValue(j, out stagePartSet) && stagePartSet.Count > 0)
{
//if (log != null) log.AppendLine("Not empty");
// We have to copy the contents of the set here rather than copying the set reference or
// bad things (tm) happen
foreach (PartSim aPartSim in stagePartSet)
{
sourcePartSet.Add(aPartSim);
}
break;
}
}
break;
case ResourceFlowMode.STACK_PRIORITY_SEARCH:
case ResourceFlowMode.STAGE_STACK_FLOW:
case ResourceFlowMode.STAGE_STACK_FLOW_BALANCE:
visited.Clear();
if (log != null) log.Append("Find ", ResourceContainer.GetResourceName(type), " sources for ", partSim.name)
.AppendLine(":", partSim.partId);
partSim.GetSourceSet(type, true, allParts, visited, sourcePartSet, log, "");
break;
default:
if (log != null) log.Append("SetResourceDrains(", partSim.name, ":", partSim.partId)
.AppendLine(") Unexpected flow type for ", ResourceContainer.GetResourceName(type), ")");
break;
}
if (log != null && sourcePartSet.Count > 0)
{
log.AppendLine("Source parts for ", ResourceContainer.GetResourceName(type), ":");
foreach (PartSim partSim in sourcePartSet)
{
log.AppendLine(partSim.name, ":", partSim.partId);
}
}
//DumpSourcePartSets(log, "after " + ResourceContainer.GetResourceName(type));
}
// If we don't have sources for all the needed resources then return false without setting up any drains
for (int i = 0; i < this.resourceConsumptions.Types.Count; i++)
{
int type = this.resourceConsumptions.Types[i];
HashSet<PartSim> sourcePartSet;
if (!sourcePartSets.TryGetValue(type, out sourcePartSet) || sourcePartSet.Count == 0)
{
if (log != null) log.AppendLine("No source of ", ResourceContainer.GetResourceName(type));
isActive = false;
return false;
}
}
// Now we set the drains on the members of the sets and update the draining parts set
for (int i = 0; i < this.resourceConsumptions.Types.Count; i++)
{
int type = this.resourceConsumptions.Types[i];
HashSet<PartSim> sourcePartSet = sourcePartSets[type];
ResourceFlowMode mode = (ResourceFlowMode)resourceFlowModes[type];
double consumption = resourceConsumptions[type];
double amount = 0d;
double total = 0d;
if (mode == ResourceFlowMode.ALL_VESSEL_BALANCE ||
mode == ResourceFlowMode.STAGE_PRIORITY_FLOW_BALANCE ||
mode == ResourceFlowMode.STAGE_STACK_FLOW_BALANCE ||
mode == ResourceFlowMode.STACK_PRIORITY_SEARCH)
{
foreach (PartSim partSim in sourcePartSet)
total += partSim.resources[type];
}
else
amount = consumption / sourcePartSet.Count;
// Loop through the members of the set
foreach (PartSim partSim in sourcePartSet)
{
if (total != 0d)
amount = consumption * partSim.resources[type] / total;
if (log != null) log.Append("Adding drain of ", amount, " ", ResourceContainer.GetResourceName(type))
.AppendLine(" to ", partSim.name, ":", partSim.partId);
partSim.resourceDrains.Add(type, amount);
drainingParts.Add(partSim);
}
}
return true;
}
