commodity_producer_manager.h

Go to the documentation of this file.

#ifndef CYCLUS_SRC_TOOLKIT_COMMODITY_PRODUCER_MANAGER_H_
#define CYCLUS_SRC_TOOLKIT_COMMODITY_PRODUCER_MANAGER_H_
 
#include <set>
 
#include "agent_managed.h"
#include "commodity.h"
#include "commodity_producer.h"
 
namespace cyclus {
namespace toolkit {
 
/// A mixin to provide information about commodity producers
class CommodityProducerManager : public AgentManaged {
 public:
  CommodityProducerManager(Agent* agent = NULL) : AgentManaged(agent) {}
  virtual ~CommodityProducerManager() {}
 
  /// @return the total production capacity for a commodity amongst producers
  /// @param commodity the commodity in question
  double TotalCapacity(Commodity& commodity);
 
  /// Register a commodity producer with the manager
  /// @param producer the producer
  inline void Register(CommodityProducer* producer) {
    producers_.insert(producer);
  }
 
  /// Unregister a commodity producer with the manager
  /// @param producer the producer
  inline void Unregister(CommodityProducer* producer) {
    producers_.erase(producer);
  }
 
  inline const std::set<CommodityProducer*>& producers() const {
    return producers_;
  }
 
 private:
  /// The set of managed producers
  std::set<CommodityProducer*> producers_;
};
 
}  // namespace toolkit
}  // namespace cyclus
 
#endif  // CYCLUS_SRC_TOOLKIT_COMMODITY_PRODUCER_MANAGER_H_