From 7a6a0697e392d40c56880ecc16e81c7058a2f765 Mon Sep 17 00:00:00 2001
From: Jeremy Saklad <saklad5@me.com>
Date: Fri, 25 Jun 2021 18:37:41 -0500
Subject: [PATCH] Add the Trifling Diplomat

This buyer actually changes based on a world quality. As such, it has
been implemented in a similar fashion to Occasional Buyer.

The scaling for this buyer's difficulty level has not been established
at the time of this writing, so it is being treated as 0.
---
 Bone Market Solver.py | 51 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 49 insertions(+), 2 deletions(-)

diff --git a/Bone Market Solver.py b/Bone Market Solver.py
index 7f6e420..f6b13e6 100644
--- a/Bone Market Solver.py	
+++ b/Bone Market Solver.py	
@@ -1159,6 +1159,11 @@ class Buyer(Enum):
             cost = Cost.ACTION.value
             )
 
+    THE_TRIFLING_DIPLOMAT_ANTIQUE = Action(
+            "Sell the Diplomat an antique skeleton",
+            cost = Cost.ACTION.value
+            )
+
     def __str__(self):
         return str(self.value)
 
@@ -1184,7 +1189,13 @@ class OccasionalBuyer(Enum):
     AN_INGENUOUS_MALACOLOGIST = [Buyer.AN_INGENUOUS_MALACOLOGIST]
 
 
-def Solve(shadowy_level, bone_market_fluctuations, zoological_mania, occasional_buyer = None, desired_buyers = [], maximum_cost = cp_model.INT32_MAX, maximum_exhaustion = cp_model.INT32_MAX, time_limit = float('inf'), workers = cpu_count(), stdscr = None):
+class DiplomatFascination(Enum):
+    """The current requirements of the Trifling Diplomat."""
+
+    ANTIQUITY = Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE
+
+
+def Solve(shadowy_level, bone_market_fluctuations, zoological_mania, occasional_buyer = None, diplomat_fascination = None, desired_buyers = [], maximum_cost = cp_model.INT32_MAX, maximum_exhaustion = cp_model.INT32_MAX, time_limit = float('inf'), workers = cpu_count(), stdscr = None):
     model = cp_model.CpModel()
 
     actions = {}
@@ -1231,6 +1242,10 @@ def Solve(shadowy_level, bone_market_fluctuations, zoological_mania, occasional_
         for unavailable_buyer in OccasionalBuyer if unavailable_buyer != occasional_buyer
         for buyer in unavailable_buyer.value if buyer not in desired_buyers
         ])
+    model.AddAssumptions([
+        actions[outmoded_fascination.value].Not()
+        for outmoded_fascination in DiplomatFascination if outmoded_fascination != diplomat_fascination and outmoded_fascination.value not in desired_buyers
+        ])
 
     # Restrict to desired buyers
     if desired_buyers:
@@ -2103,6 +2118,30 @@ def Solve(shadowy_level, bone_market_fluctuations, zoological_mania, occasional_
     model.Add(added_exhaustion == 0).OnlyEnforceIf(actions[Buyer.THE_CARPENTERS_GRANDDAUGHTER])
 
 
+    # The Trifling Diplomat - Antique
+    model.Add(skeleton_in_progress >= 100).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+    model.Add(antiquity >= 5).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+
+    antiquity_squared = model.NewIntVar(0, cp_model.INT32_MAX, '{}: {}'.format(Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE.name, 'antiquity squared'))
+    model.AddMultiplicationEquality(antiquity_squared, [antiquity, antiquity])
+
+    value_remainder = model.NewIntVar(0, 49, '{}: {}'.format(Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE.name, 'value remainder'))
+    model.AddModuloEquality(value_remainder, value, 50)
+
+    model.Add(primary_revenue == value - value_remainder + 50).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+    model.Add(secondary_revenue == 50*antiquity_squared).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+
+    # TODO: Add actual difficulty level
+    model.Add(difficulty_level == 0).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+
+    # The indirection is necessary for applying an enforcement literal
+    derived_exhaustion = model.NewIntVar(cp_model.INT32_MIN, cp_model.INT32_MAX, '{}: {}'.format(Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE.name, 'derived exhaustion'))
+    model.AddDivisionEquality(derived_exhaustion, antiquity_squared, 100)
+    model.Add(added_exhaustion == derived_exhaustion).OnlyEnforceIf(actions[Buyer.THE_TRIFLING_DIPLOMAT_ANTIQUE])
+
+    del antiquity_squared, value_remainder, derived_exhaustion
+
+
     # Maximize profit margin
     net_profit = model.NewIntVar(cp_model.INT32_MIN, cp_model.INT32_MAX, 'net profit')
     model.Add(net_profit == total_revenue - cost)
@@ -2272,13 +2311,21 @@ def main():
             )
 
     buyer = parser.add_mutually_exclusive_group()
-    buyer.add_argument(
+    transient_buyers = buyer.add_argument_group()
+    transient_buyers.add_argument(
             "-o", "--occasional-buyer",
             action=EnumAction,
             type=OccasionalBuyer,
             help="current value of Occasional Buyer, which allows access to a buyer that is not otherwise available",
             dest='occasional_buyer'
             )
+    transient_buyers.add_argument(
+            "-d", "--diplomat-fascination",
+            action=EnumAction,
+            type=DiplomatFascination,
+            help="current value of The Diplomat's Current Fascination, which determines what the Trifling Diplomat is interested in",
+            dest='diplomat_fascination'
+            )
     buyer.add_argument(
             "-b", "--buyer", "--desired-buyer",
             action=EnumAction,