51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
#ifndef TARGETGENERATOR_HPP
|
|
#define TARGETGENERATOR_HPP
|
|
|
|
# include <iostream>
|
|
# include <string>
|
|
# include "ATarget.hpp"
|
|
# include <map>
|
|
|
|
class TargetGenerator {
|
|
private:
|
|
TargetGenerator(TargetGenerator const & other);
|
|
TargetGenerator & operator=(TargetGenerator const & other);
|
|
|
|
std::map<std::string, ATarget *> arr;
|
|
|
|
public:
|
|
TargetGenerator() {
|
|
};
|
|
~TargetGenerator() {
|
|
std::map<std::string, ATarget *>::iterator it_begin = this->arr.begin();
|
|
std::map<std::string, ATarget *>::iterator it_end = this->arr.end();
|
|
while (it_begin != it_end) {
|
|
delete it_begin->second;
|
|
++it_begin;
|
|
}
|
|
this->arr.clear();
|
|
};
|
|
|
|
void learnTargetType(ATarget * atarget) {
|
|
if (atarget)
|
|
arr.insert(std::pair<std::string, ATarget *>(
|
|
atarget->getType(),
|
|
atarget->clone()
|
|
));
|
|
};
|
|
void forgetTargetType(std::string const & name) {
|
|
std::map<std::string, ATarget *>::iterator it = arr.find(name);
|
|
if (it != arr.end())
|
|
delete it->second;
|
|
arr.erase(name);
|
|
};
|
|
ATarget * createTarget(std::string const & name) {
|
|
std::map<std::string, ATarget *>::iterator it = arr.find(name);
|
|
if (it != arr.end())
|
|
return arr[name];
|
|
return NULL;
|
|
};
|
|
};
|
|
|
|
#endif
|