builder.h

Go to the documentation of this file.

#ifndef CYCLUS_SRC_TOOLKIT_BUILDER_H_
#define CYCLUS_SRC_TOOLKIT_BUILDER_H_
 
#include <set>
 
#include "agent_managed.h"
#include "commodity_producer.h"
 
namespace cyclus {
namespace toolkit {
 
/// A mixin to provide information about commodity producers that can
/// be built.
class Builder : public AgentManaged {
 public:
  Builder(Agent* agent = NULL) : AgentManaged(agent) {}
  virtual ~Builder() {}
 
  /// Register a commodity producer with the agent
  /// @param producer the producer
  inline void Register(CommodityProducer* producer) {
    producers_.insert(producer);
  }
 
  /// Unregister a commodity producer with the agent
  /// @param producer the producer
  inline void Unregister(CommodityProducer* producer) {
    producers_.erase(producer);
  }
 
  inline const std::set<CommodityProducer*>& producers() const {
    return producers_;
  }
 
 private:
  std::set<CommodityProducer*> producers_;
};
 
}  // namespace toolkit
}  // namespace cyclus
 
#endif  // CYCLUS_SRC_TOOLKIT_BUILDER_H_