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_