nat-optics-1.0.0.3: src/NatOptics/Positive/Math.hs

{-# language Trustworthy #-}

module NatOptics.Positive.Math
  (
    minus,
    plus,
  )
  where

import Data.Maybe          ( Maybe (..) )
import Data.Ord            ( Ord, (<) )
import Optics.Core         ( preview, review )
import Prelude             ( Num, abs, (+), (-) )

import NatOptics.Positive.Unsafe ( Positive (..) )
import qualified NatOptics.Positive as Positive

import NatOptics.Signed ( Signed )
import qualified NatOptics.Signed as Signed

plus :: Num n => Positive n -> Positive n -> Positive n
plus (PositiveUnsafe a) (PositiveUnsafe b) = PositiveUnsafe (a + b)

minus :: (Num n, Ord n) => Positive n -> Positive n -> Signed n
minus a b =
    let
        a' = review Positive.refine a
        b' = review Positive.refine b
        diff = a' - b'
    in
        case preview Positive.refine (abs diff) of
            Nothing -> Signed.Zero
            Just diff' -> if diff < 0 then Signed.Minus diff' else Signed.Plus diff'