-- | Simple counter.
module Perf.Count
  ( count,
    countN,
  )
where

import Perf.Types
import Prelude

-- | Register 1 as a performance measure
count :: (Applicative m) => StepMeasure m Int
count :: forall (m :: * -> *). Applicative m => StepMeasure m Int
count = forall (m :: * -> *) t i. m i -> (i -> m t) -> StepMeasure m t
StepMeasure m ()
start forall {f :: * -> *} {a} {p}. (Applicative f, Num a) => p -> f a
stop
  where
    start :: m ()
start = forall (f :: * -> *) a. Applicative f => a -> f a
pure ()
    stop :: p -> f a
stop p
_ = forall (f :: * -> *) a. Applicative f => a -> f a
pure a
1
{-# INLINEABLE count #-}

-- | Count the number of times measured.
countN :: Int -> Measure IO Int
countN :: Int -> Measure IO Int
countN Int
n = forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall (m :: * -> *) t.
Monad m =>
Int -> StepMeasure m t -> Measure m [t]
toMeasureN Int
n forall (m :: * -> *). Applicative m => StepMeasure m Int
count
{-# INLINEABLE countN #-}