TMap<FFlareResourceDescription*, WorldHelper::FlareResourceStats> SectorHelper::ComputeSectorResourceStats(UFlareSimulatedSector* Sector, bool IncludeStorage)
{
TMap<FFlareResourceDescription*, WorldHelper::FlareResourceStats> WorldStats;
// Init
for(int32 ResourceIndex = 0; ResourceIndex < Sector->GetGame()->GetResourceCatalog()->Resources.Num(); ResourceIndex++)
{
FFlareResourceDescription* Resource = &Sector->GetGame()->GetResourceCatalog()->Resources[ResourceIndex]->Data;
WorldHelper::FlareResourceStats ResourceStats;
ResourceStats.Production = 0;
ResourceStats.Consumption = 0;
ResourceStats.Balance = 0;
ResourceStats.Stock = 0;
ResourceStats.Capacity = 0;
WorldStats.Add(Resource, ResourceStats);
}
for (int SpacecraftIndex = 0; SpacecraftIndex < Sector->GetSectorSpacecrafts().Num(); SpacecraftIndex++)
{
UFlareSimulatedSpacecraft* Spacecraft = Sector->GetSectorSpacecrafts()[SpacecraftIndex];
if(!IncludeStorage && Spacecraft->HasCapability(EFlareSpacecraftCapability::Storage))
{
continue;
}
// Stock
TArray<FFlareCargo>& CargoBaySlots = Spacecraft->GetActiveCargoBay()->GetSlots();
for (int CargoIndex = 0; CargoIndex < CargoBaySlots.Num(); CargoIndex++)
{
FFlareCargo& Cargo = CargoBaySlots[CargoIndex];
if (!Cargo.Resource)
{
continue;
}
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Cargo.Resource];
FFlareResourceUsage Usage = Spacecraft->GetResourceUseType(Cargo.Resource);
if(Usage.HasUsage(EFlareResourcePriceContext::FactoryInput) || Usage.HasUsage(EFlareResourcePriceContext::ConsumerConsumption) || Usage.HasUsage(EFlareResourcePriceContext::MaintenanceConsumption) || Usage.HasUsage(EFlareResourcePriceContext::HubInput))
{
ResourceStats->Capacity += Spacecraft->GetActiveCargoBay()->GetSlotCapacity() - Cargo.Quantity;
}
if(Usage.HasUsage(EFlareResourcePriceContext::FactoryOutput) || Usage.HasUsage(EFlareResourcePriceContext::MaintenanceConsumption) || Usage.HasUsage(EFlareResourcePriceContext::HubOutput))
{
ResourceStats->Stock += Cargo.Quantity;
}
}
for (int32 FactoryIndex = 0; FactoryIndex < Spacecraft->GetFactories().Num(); FactoryIndex++)
{
UFlareFactory* Factory = Spacecraft->GetFactories()[FactoryIndex];
if(Factory->IsShipyard() && !Factory->IsActive())
{
const FFlareProductionData* ProductionData = Spacecraft->GetNextOrderShipProductionData(Factory->IsLargeShipyard()? EFlarePartSize::L : EFlarePartSize::S);
if (ProductionData)
{
for(const FFlareFactoryResource& FactoryResource : ProductionData->InputResources)
{
const FFlareResourceDescription* Resource = &FactoryResource.Resource->Data;
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Resource];
int64 ProductionDuration = ProductionData->ProductionTime;
float Flow = 0;
if (ProductionDuration == 0)
{
Flow = 1;
}
else
{
Flow = (float) FactoryResource.Quantity / float(ProductionDuration);
}
ResourceStats->Consumption += Flow;
}
}
continue;
}
if ((!Factory->IsActive() || !Factory->IsNeedProduction()))
{
// No resources needed
continue;
}
// Input flow
for (int32 ResourceIndex = 0; ResourceIndex < Factory->GetInputResourcesCount(); ResourceIndex++)
{
FFlareResourceDescription* Resource = Factory->GetInputResource(ResourceIndex);
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Resource];
int64 ProductionDuration = Factory->GetProductionDuration();
float Flow = 0;
if (ProductionDuration == 0)
{
Flow = 1;
}
else
{
Flow = (float) Factory->GetInputResourceQuantity(ResourceIndex) / float(ProductionDuration);
}
ResourceStats->Consumption += Flow;
}
// Ouput flow
for (int32 ResourceIndex = 0; ResourceIndex < Factory->GetOutputResourcesCount(); ResourceIndex++)
{
FFlareResourceDescription* Resource = Factory->GetOutputResource(ResourceIndex);
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Resource];
int64 ProductionDuration = Factory->GetProductionDuration();
if (ProductionDuration == 0)
{
ProductionDuration = 10;
}
float Flow = (float) Factory->GetOutputResourceQuantity(ResourceIndex) / float(ProductionDuration);
ResourceStats->Production+= Flow;
}
}
}
// FS
FFlareResourceDescription* FleetSupply = Sector->GetGame()->GetScenarioTools()->FleetSupply;
WorldHelper::FlareResourceStats *FSResourceStats = &WorldStats[FleetSupply];
FFlareFloatBuffer* Stats = &Sector->GetData()->FleetSupplyConsumptionStats;
float MeanConsumption = Stats->GetMean(0, Stats->MaxSize-1);
FSResourceStats->Consumption += MeanConsumption;
FSResourceStats->Capacity += Stats->GetValue(0);
// Customer flow
for (int32 ResourceIndex = 0; ResourceIndex < Sector->GetGame()->GetResourceCatalog()->ConsumerResources.Num(); ResourceIndex++)
{
FFlareResourceDescription* Resource = &Sector->GetGame()->GetResourceCatalog()->ConsumerResources[ResourceIndex]->Data;
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Resource];
ResourceStats->Consumption += Sector->GetPeople()->GetRessourceConsumption(Resource, false);
}
// Balance
for(int32 ResourceIndex = 0; ResourceIndex < Sector->GetGame()->GetResourceCatalog()->Resources.Num(); ResourceIndex++)
{
FFlareResourceDescription* Resource = &Sector->GetGame()->GetResourceCatalog()->Resources[ResourceIndex]->Data;
WorldHelper::FlareResourceStats *ResourceStats = &WorldStats[Resource];
ResourceStats->Balance = ResourceStats->Production - ResourceStats->Consumption;
/*FLOGV("World stats for %s: Production=%f Consumption=%f Balance=%f Stock=%d",
*Resource->Name.ToString(),
ResourceStats->Production,
ResourceStats->Consumption,
ResourceStats->Balance,
ResourceStats->Stock);*/
}
return WorldStats;
}
