File: AAIUnitStatistics.cpp

package info (click to toggle)
spring 105.0.1%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 108,860 kB
  • sloc: cpp: 467,785; ansic: 302,607; python: 12,925; java: 12,201; awk: 5,889; sh: 2,371; xml: 655; perl: 405; php: 276; objc: 194; makefile: 75; sed: 2
file content (116 lines) | stat: -rw-r--r-- 4,906 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// -------------------------------------------------------------------------
// AAI
//
// A skirmish AI for the Spring engine.
// Copyright Alexander Seizinger
//
// Released under GPL license: see LICENSE.html for more information.
// -------------------------------------------------------------------------

#include "AAIUnitStatistics.h"

AAIUnitStatistics::AAIUnitStatistics() 
{
	m_unitCostStatistics.resize( AAIUnitCategory::numberOfUnitCategories );
	m_unitBuildtimeStatistics.resize( AAIUnitCategory::numberOfUnitCategories );
	m_unitPrimaryAbilityStatistics.resize( AAIUnitCategory::numberOfUnitCategories );
	m_unitSecondaryAbilityStatistics.resize( AAIUnitCategory::numberOfUnitCategories );

	m_combatCostStatistics.resize( AAITargetType::numberOfMobileTargetTypes );
	m_combatBuildtimeStatistics.resize( AAITargetType::numberOfMobileTargetTypes );
	m_combatRangeStatistics.resize( AAITargetType::numberOfMobileTargetTypes );
	m_combatSpeedStatistics.resize( AAITargetType::numberOfMobileTargetTypes );
};

AAIUnitStatistics::~AAIUnitStatistics()
{
	m_unitCostStatistics.clear();
	m_unitBuildtimeStatistics.clear();
	m_unitPrimaryAbilityStatistics.clear();
	m_unitSecondaryAbilityStatistics.clear();

	m_combatCostStatistics.clear();
	m_combatBuildtimeStatistics.clear();
	m_combatRangeStatistics.clear();
	m_combatSpeedStatistics.clear();
};

void AAIUnitStatistics::Init(const std::vector<const springLegacyAI::UnitDef*>& unitDefs, const std::vector<UnitTypeProperties>& unitProperties, const std::vector< std::list<UnitDefId> >& unitsInCategory, const std::vector< std::list<UnitDefId> >& unitsInCombatCategory)
{
	//-----------------------------------------------------------------------------------------------------------------
	// calculate unit category statistics
	//-----------------------------------------------------------------------------------------------------------------
	for(uint32_t cat = 0; cat < AAIUnitCategory::numberOfUnitCategories; ++cat) 
	{
		for(auto unitDefId : unitsInCategory[cat])
		{
			m_unitBuildtimeStatistics[cat].AddValue( unitProperties[unitDefId.id].m_buildtime );
			m_unitCostStatistics[cat].AddValue( unitProperties[unitDefId.id].m_totalCost );
			m_unitPrimaryAbilityStatistics[cat].AddValue( unitProperties[unitDefId.id].m_primaryAbility );
			m_unitSecondaryAbilityStatistics[cat].AddValue( unitProperties[unitDefId.id].m_secondaryAbility );
		}

		// calculate average values after last value has been added
		m_unitBuildtimeStatistics[cat].Finalize();
		m_unitCostStatistics[cat].Finalize();
		m_unitPrimaryAbilityStatistics[cat].Finalize();
		m_unitSecondaryAbilityStatistics[cat].Finalize();
	}

	//-----------------------------------------------------------------------------------------------------------------
	// calculate combat category statistics
	//-----------------------------------------------------------------------------------------------------------------
	for(const auto combatUnitCategory : AAICombatUnitCategory::m_combatUnitCategories) 
	{
		const int i = AAICombatUnitCategory(combatUnitCategory).GetArrayIndex();

		for(auto unitDefId : unitsInCombatCategory[i])
		{
			m_combatCostStatistics[i].AddValue( unitProperties[unitDefId.id].m_totalCost );
			m_combatBuildtimeStatistics[i].AddValue( unitProperties[unitDefId.id].m_buildtime );
			m_combatRangeStatistics[i].AddValue( unitProperties[unitDefId.id].m_primaryAbility );
			m_combatSpeedStatistics[i].AddValue( unitProperties[unitDefId.id].m_secondaryAbility );
		}

		m_combatCostStatistics[i].Finalize();
		m_combatBuildtimeStatistics[i].Finalize();
		m_combatRangeStatistics[i].Finalize();
		m_combatSpeedStatistics[i].Finalize();
	}

	m_sensorStatistics.Init(unitDefs, unitProperties, unitsInCategory);
}

void SensorStatistics::Init(const std::vector<const springLegacyAI::UnitDef*>& unitDefs, const std::vector<UnitTypeProperties>& unitProperties, const std::vector< std::list<UnitDefId> >& unitsInCategory)
{
	const int index = AAIUnitCategory(EUnitCategory::STATIC_SENSOR).GetArrayIndex();

	for(auto defId = unitsInCategory[index].begin(); defId != unitsInCategory[index].end(); ++defId)
	{
		if(unitProperties[defId->id].m_unitType.IsRadar())
		{
			m_radarRanges.AddValue( unitDefs[defId->id]->radarRadius );
			m_radarCosts.AddValue( unitProperties[defId->id].m_totalCost );
		}
		
		if(unitProperties[defId->id].m_unitType.IsSonar())
		{
			m_sonarRanges.AddValue( unitDefs[defId->id]->sonarRadius );
			m_sonarCosts.AddValue( unitProperties[defId->id].m_totalCost );
		}

		if(unitProperties[defId->id].m_unitType.IsSeismicDetector())
		{
			m_seismicRanges.AddValue( unitDefs[defId->id]->seismicRadius );
			m_seismicCosts.AddValue( unitProperties[defId->id].m_totalCost );
		}
	}

	m_radarRanges.Finalize();
	m_sonarRanges.Finalize();
	m_seismicRanges.Finalize();

	m_radarCosts.Finalize();
	m_sonarCosts.Finalize();
	m_seismicCosts.Finalize();
}