BTHAI/SCProjects/BTHAIModule/Source/ResourceManager.h

Go to the documentation of this file.

#ifndef __RESOURCEMANAGER_H__
#define __RESOURCEMANAGER_H__

#include <BWAPI.h>

using namespace BWAPI;
using namespace std;

struct ResourceLock {
        UnitType unit;
        int mineralCost;
        int gasCost;

        ResourceLock(UnitType mUnit) {
                unit = mUnit;
                mineralCost = mUnit.mineralPrice();
                gasCost = mUnit.gasPrice();
        }
};

class ResourceManager {

private:
        static ResourceManager* instance;

        ResourceManager();

        bool hasProductionBuilding();
        
        vector<ResourceLock> locks;

        int calcLockedMinerals();
        int calcLockedGas();

public:
        ~ResourceManager();

        static ResourceManager* getInstance();

        bool needWorker();

        bool hasResources(UnitType type);

        bool hasResources(UpgradeType type);

        bool hasResources(TechType type);

        bool hasResources(int neededMinerals, int neededGas);

        void lockResources(UnitType type);

        void unlockResources(UnitType type);

        void printInfo();
};

#endif