#include "../Platform/stdafx.h"

#include "BehaviorRegistry.h"

BehaviorRegistry::BehaviorRegistry(DispenseItemBehavior* defaultValue) {
    defaultBehavior = defaultValue;
}

BehaviorRegistry::~BehaviorRegistry() {
    for (AUTO_VAR(it, storage.begin()); it != storage.end(); ++it) {
        delete it->second;
    }

    delete defaultBehavior;
}

DispenseItemBehavior* BehaviorRegistry::get(Item* key) {
    AUTO_VAR(it, storage.find(key));

    return (it == storage.end()) ? defaultBehavior : it->second;
}

void BehaviorRegistry::add(Item* key, DispenseItemBehavior* value) {
    storage.insert(std::make_pair(key, value));
}