BTHAI/SCProjects/BTHAIModule/Source/UpgradesPlanner.h

Go to the documentation of this file.

#ifndef __UPGRADESPLANNER_H__
#define __UPGRADESPLANNER_H__

#include <BWAPI.h>
#include "BaseAgent.h"

using namespace BWAPI;
using namespace std;

class UpgradesPlanner {

private:
        static UpgradesPlanner* instance;

        UpgradesPlanner();
        
        vector<UpgradeType> upgradesP1;
        vector<UpgradeType> upgradesP2;
        vector<UpgradeType> upgradesP3;
        vector<TechType> techsP1;
        vector<TechType> techsP2;
        vector<TechType> techsP3;

        bool canUpgrade(UpgradeType type, Unit* unit);
        bool canResearch(TechType type, Unit* unit);

public:
        ~UpgradesPlanner();

        static UpgradesPlanner* getInstance();

        bool checkUpgrade(BaseAgent* agent);
};

#endif