From e5cb082fb5c6a71897446090b56490fb58d77efc Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Thu, 23 May 2019 12:15:54 -0700
Subject: [PATCH] lib: amountToCost, mixedAmountToCost

---
 hledger-lib/Hledger/Data/Amount.hs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/hledger-lib/Hledger/Data/Amount.hs b/hledger-lib/Hledger/Data/Amount.hs
index 8485d39a8..c83cbb69a 100644
--- a/hledger-lib/Hledger/Data/Amount.hs
+++ b/hledger-lib/Hledger/Data/Amount.hs
@@ -57,6 +57,7 @@ module Hledger.Data.Amount (
   amountWithCommodity,
   -- ** arithmetic
   costOfAmount,
+  amountToCost,
   divideAmount,
   multiplyAmount,
   divideAmountAndPrice,
@@ -93,6 +94,7 @@ module Hledger.Data.Amount (
   normaliseMixedAmount,
   -- ** arithmetic
   costOfMixedAmount,
+  mixedAmountToCost,
   divideMixedAmount,
   multiplyMixedAmount,
   divideMixedAmountAndPrice,
@@ -216,6 +218,10 @@ costOfAmount a@Amount{aquantity=q, aprice=price} =
       UnitPrice  p@Amount{aquantity=pq} -> p{aquantity=pq * q}
       TotalPrice p@Amount{aquantity=pq} -> p{aquantity=pq * signum q}
 
+-- | Convert this amount to cost, and apply the appropriate amount style.
+amountToCost :: M.Map CommoditySymbol AmountStyle -> Amount -> Amount
+amountToCost styles = styleAmount styles . costOfAmount
+
 -- | Replace an amount's TotalPrice, if it has one, with an equivalent UnitPrice.
 -- Has no effect on amounts without one.
 -- Also increases the unit price's display precision to show one extra decimal place,
@@ -581,6 +587,10 @@ mapMixedAmount f (Mixed as) = Mixed $ map f as
 costOfMixedAmount :: MixedAmount -> MixedAmount
 costOfMixedAmount (Mixed as) = Mixed $ map costOfAmount as
 
+-- | Convert all component amounts to cost, and apply the appropriate amount styles.
+mixedAmountToCost :: M.Map CommoditySymbol AmountStyle -> MixedAmount -> MixedAmount
+mixedAmountToCost styles (Mixed as) = Mixed $ map (amountToCost styles) as
+
 -- | Divide a mixed amount's quantities by a constant.
 divideMixedAmount :: Quantity -> MixedAmount -> MixedAmount
 divideMixedAmount n = mapMixedAmount (divideAmount n)