BTHAI/SCProjects/BTHAIModule/Source/UpgradesFileReader.h

Go to the documentation of this file.

#ifndef __UPGRADESFILEREADER_H__
#define __UPGRADESFILEREADER_H__

#include <BWAPI.h>

using namespace BWAPI;
using namespace std;

struct Tokens {
        string key;
        string value;
};

class UpgradesFileReader {

private:
        bool fileExists(string filename);
        string getFilename();
        void addUpgrade(string line);
        void replace(string &line);
        int toInt(string &str);
        Tokens split(string line, string delimiter);

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

public:
        UpgradesFileReader();

        void readUpgrades();

        vector<UpgradeType> getUpgradesP1();
        vector<UpgradeType> getUpgradesP2();
        vector<UpgradeType> getUpgradesP3();
        vector<TechType> getTechsP1();
        vector<TechType> getTechsP2();
        vector<TechType> getTechsP3();
};

#endif