CEconomy.h

#ifndef CECONOMY_H
#define CECONOMY_H

#include <map>
#include <vector>
#include <stack>

#include "ARegistrar.h"

#include "headers/Defines.h"

class ATask;
class CGroup;
class CUnit;
class AIClasses;
class UnitType;
class float3;

const float alpha = 0.2f;
const float beta = 0.05f;

class CEconomy: public ARegistrar {
	public:
		CEconomy(AIClasses *ai);
		~CEconomy();

		/* overal mNow averaged over 5 logical frames */
		float mNow, mNowSummed;
		/* overal eNow averaged over 5 logical frames */
		float eNow, eNowSummed;
		/* overal mIncome averaged over 5 logical frames */
		float mIncome, mIncomeSummed;
		/* overal eIncome averaged over 5 logical frames */
		float eIncome, eIncomeSummed;
		/* total units mIncome averaged over 5 logical frames */
		float uMIncome, uMIncomeSummed;
		/* total units eIncome averaged over 5 logical frames */
		float uEIncome, uEIncomeSummed;
		/* metal usage averaged over 5 logical frames */
		float mUsage, mUsageSummed, mStart;
		/* energy usage averaged over 5 logical frames */
		float eUsage, eUsageSummed, eStart;
		/* metal storage */
		float mStorage;
		/* energy storage */
		float eStorage;
		
		/* State, a sort of measurement how far advanced we are */
		int state;

		/* stalling/exceeding vars, updated in updateIncomes() */
		bool mstall, estall, mexceeding, eexceeding, areMMakersEnabled;
		int latest_factory;

		/* Returns a fresh CGroup instance */
		CGroup* requestGroup();

		/* Overload */
		void remove(ARegistrar &group);

		/* Add a new unit on finished */
		void addUnitOnCreated(CUnit &unit);

		/* Add a new unit on created */
		void addUnitOnFinished(CUnit &unit);

		/* Initialize economy module */
		void init(CUnit &unit);

		/* Update the eco system */
		void update(int frame);

		/* Update averaged incomes */
		void updateIncomes(int frame = 100);

		/* See if this group has finished a building */
		bool hasFinishedBuilding(CGroup &group);

		/* See if this group begun building */
		bool hasBegunBuilding(CGroup &group);

		/* Can we afford to build this ? */
		bool canAffordToBuild(UnitType *builder, UnitType *utToBuild, int unitId);

		bool isInitialized() { return initialized; };

	private:
		bool initialized;
		AIClasses *ai;

		std::map<int, float3> takenMexes, takenGeo, upgradedMexes;

		/* Active groups ingame */
		std::map<int, CGroup*> activeGroups;

		/* Altered by canAfford() */
		bool eRequest, mRequest;

		/* Is this a windmap ? */
		bool windmap;

		/* updateIncomes counter */
		unsigned int incomes;

		/* Can we afford to assist a factory ? */
		ATask* canAssistFactory(CGroup &group);

		/* Do something with idle builder */
		void commandBuilderGroup(CGroup *group);

		/* See if we can help with a certain task */
		ATask* canAssist(buildType t, CGroup &group);

		float3 getBestSpot(CGroup &group, std::list<float3> &resources, std::map<int, float3> &tracker, bool metal);

		/* Fills takenMexes also */
		float3 getClosestOpenMetalSpot(CGroup &group);

		/* Fills both takenMexes and upgradedMexes */
		float3 getClosestUpgradeableMetalSpot(CGroup &group);

		float3 getClosestOpenGeoSpot(CGroup &group);

		/* Prevent stalling */
		void preventStalling();

		/* See what is best for our economy */
		void controlMetalMakers();

		/* build or assist on a certain task */
		bool buildOrAssist(CGroup &group, buildType bt, unsigned include, unsigned exclude = 0);

		/* See if a buildtask is in progress */
		bool taskInProgress(buildType bt);

		/* Get next allowed factory to build */
		unsigned int getNextFactoryToBuild(CUnit *unit, int maxteachlevel);
};

#endif