Tariff Region

CHANGELOG.rst

@@ -6,6 +6,7 @@ Since last release
 ======================
 
 **Added:**
+* Added TariffRegion Region (#661)
 * Added Conversion Facility (#657)
 * Replaced manual matl_buy/sell_policy code in storage with code injection (#639)
 * Added package parameter to storage (#603, #612, #616)

src/CMakeLists.txt

@@ -26,6 +26,8 @@ USE_CYCLUS("cycamore" "manager_inst")
 
 USE_CYCLUS("cycamore" "growth_region")
 
+USE_CYCLUS("cycamore" "tariff_region")
+
 USE_CYCLUS("cycamore" "storage")
 
 INSTALL_CYCLUS_MODULE("cycamore" "" "NONE")

src/tariff_region.cc

@@ -0,0 +1,88 @@
+#include "tariff_region.h"
+
+namespace cycamore {
+
+TariffRegion::TariffRegion(cyclus::Context* ctx)
+: cyclus::Region(ctx) {}
+
+TariffRegion::~TariffRegion() {}
+
+void TariffRegion::EnterNotify() {
+  Region::EnterNotify();
+
+}
+
+void TariffRegion::AdjustMatlPrefs(cyclus::PrefMap<cyclus::Material>::type& prefs) {
+  for (auto& req_pair : prefs) {
+
+    // Iterate over the bids in the request portfolio
+    for (auto& bid_pair : req_pair.second) {
+
+      // Get the bid
+      cyclus::Bid<cyclus::Material>* bid = bid_pair.first;
+
+      // Get the supplier - use manager() to get the Agent* from the Trader*
+      cyclus::Agent* supplier = bid->bidder()->manager();
+
+      // Traverse up the hierarchy to get the supplier's region
+      cyclus::Region* supplier_region = nullptr;
+      cyclus::Agent* current = supplier;
+      while (current != nullptr) {
+        supplier_region = dynamic_cast<cyclus::Region*>(current);
+        if (supplier_region != nullptr) {
+          break;  // Found a region
+        }
+        current = current->parent();
+      }
+
+      // If the supplier is in the region list, apply the appropriate adjustment
+      auto it = std::find(region_names.begin(), region_names.end(), 
+          supplier_region->prototype());
+      if (it != region_names.end()) {
+        double cost_multiplier = 1.0 + adjustments[it - region_names.begin()];
+        bid_pair.second *= cost_multiplier > 0.0 ? 1.0 / cost_multiplier : std::numeric_limits<double>::infinity();
+      }
+    }
+  }
+}
+
+void TariffRegion::AdjustProductPrefs(cyclus::PrefMap<cyclus::Product>::type& prefs) {
+
+  // Iterate over the preferences
+  for (auto& req_pair : prefs) {
+    // Iterate over the bids in the request portfolio
+    for (auto& bid_pair : req_pair.second) {
+
+      // Get the bid
+      cyclus::Bid<cyclus::Product>* bid = bid_pair.first;
+
+      // Get the supplier - use manager() to get the Agent* from the Trader*
+      cyclus::Agent* supplier = bid->bidder()->manager();
+
+      // Traverse up the hierarchy to get the supplier's region
+      cyclus::Region* supplier_region = nullptr;
+      cyclus::Agent* current = supplier;
+      while (current != nullptr) {
+        supplier_region = dynamic_cast<cyclus::Region*>(current);
+        if (supplier_region != nullptr) {
+          break;  // Found a region
+        }
+        current = current->parent();
+      }
+
+      // If the supplier is in the region list, apply the appropriate adjustment
+      auto it = std::find(region_names.begin(), region_names.end(), 
+                      supplier_region->prototype());
+      if (it != region_names.end()) {
+        double cost_multiplier = 1.0 + adjustments[it - region_names.begin()];
+        bid_pair.second *= cost_multiplier > 0.0 ? 1.0 / cost_multiplier : std::numeric_limits<double>::infinity();
+      }
+    }
+  }
+}
+
+extern "C" cyclus::Agent* ConstructTariffRegion(cyclus::Context* ctx) {
+  return new TariffRegion(ctx);
+}
+
+}  // namespace cycamore

src/tariff_region.h

@@ -0,0 +1,43 @@
+#ifndef CYCAMORE_SRC_TARIFF_REGION_H_
+#define CYCAMORE_SRC_TARIFF_REGION_H_
+
+#include "cyclus.h"
+#include <string>
+
+namespace cycamore {
+
+class TariffRegion : public cyclus::Region {
+ public:
+  TariffRegion(cyclus::Context* ctx);
+  virtual ~TariffRegion();
+
+  virtual void EnterNotify();
+  virtual void AdjustMatlPrefs(cyclus::PrefMap<cyclus::Material>::type& prefs);
+  virtual void AdjustProductPrefs(cyclus::PrefMap<cyclus::Product>::type& prefs);
+
+  #pragma cyclus
+
+ private:
+  cyclus::Region* FindRegionByName(const std::string& name);
+  cyclus::Region* FindRegionInHierarchy(cyclus::Agent* agent, const std::string& name);
+
+  #pragma cyclus var { \
+    "default": [], \
+    "doc": "List of regions that will have a trade adjustment " \
+    "(subsidy)" \
+  }
+  std::vector<std::string> region_names;
+
+  #pragma cyclus var { \
+    "default": [], \
+    "doc": "Adjustments (1+val) to apply to trades from affected " \
+    "regions (percent as decimal, must be in same order as " \
+    "region_names). Positive values are tariffs, negative values are subsidies." \
+  }
+  std::vector<double> adjustments;
+
+};
+
+} // namespace cycamore
+
+#endif  // CYCAMORE_SRC_TARIFF_REGION_H_