Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds fuseSetters #118

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
127 changes: 103 additions & 24 deletions src/Data/Lens/Setter.purs
Original file line number Diff line number Diff line change
@@ -1,60 +1,101 @@
-- | This module defines functions for working with setters.
module Data.Lens.Setter
( (%~), over, iover
, (.~), set
, (+~), addOver
, (-~), subOver
, (*~), mulOver
, (//~), divOver
, (||~), disjOver
, (&&~), conjOver
, (<>~), appendOver
( (%~)
, over
, iover
, (.~)
, set
, (+~)
, addOver
, (-~)
, subOver
, (*~)
, mulOver
, (//~)
, divOver
, (||~)
, disjOver
, (&&~)
, conjOver
, (<>~)
, appendOver
, (++~)
, (?~), setJust
, (.=), assign
, (%=), modifying
, (+=), addModifying
, (*=), mulModifying
, (-=), subModifying
, (//=), divModifying
, (||=), disjModifying
, (&&=), conjModifying
, (<>=), appendModifying
, (?~)
, setJust
, (.=)
, assign
, (%=)
, modifying
, (+=)
, addModifying
, (*=)
, mulModifying
, (-=)
, subModifying
, (//=)
, divModifying
, (||=)
, disjModifying
, (&&=)
, conjModifying
, (<>=)
, appendModifying
, (++=)
, (?=), assignJust
, (?=)
, assignJust
, fuseSetters
, cTuple
, module Data.Lens.Types
) where

import Prelude

import Control.Monad.State.Class (class MonadState, modify)

import Data.Lens.Types (IndexedSetter, Indexed(..), Setter, Setter')
import Data.Maybe (Maybe(..))
import Data.Tuple (uncurry)
import Data.Tuple (uncurry, Tuple(..), fst, snd)

infixr 4 over as %~

infixr 4 set as .~

infixr 4 addOver as +~

infixr 4 subOver as -~

infixr 4 mulOver as *~

infixr 4 divOver as //~

infixr 4 disjOver as ||~

infixr 4 conjOver as &&~

infixr 4 appendOver as <>~

infixr 4 appendOver as ++~

infixr 4 setJust as ?~

infix 4 assign as .=

infix 4 modifying as %=

infix 4 addModifying as +=

infix 4 mulModifying as *=

infix 4 subModifying as -=

infix 4 divModifying as //=

infix 4 disjModifying as ||=

infix 4 conjModifying as &&=

infix 4 appendModifying as <>=

infix 4 appendModifying as ++=

infix 4 assignJust as ?=

-- | Apply a function to the foci of a `Setter`.
Expand Down Expand Up @@ -94,7 +135,6 @@ setJust :: forall s t a b. Setter s t a (Maybe b) -> b -> s -> t
setJust p = set p <<< Just

-- Stateful

-- | Set the foci of a `Setter` in a monadic state to a constant value.
assign :: forall s a b m. MonadState s m => Setter s s a b -> b -> m Unit
assign p b = void (modify (set p b))
Expand Down Expand Up @@ -126,3 +166,42 @@ appendModifying p = modifying p <<< flip append

assignJust :: forall s a b m. MonadState s m => Setter s s a (Maybe b) -> b -> m Unit
assignJust p = assign p <<< Just

-- | Add two setters as "branches" to a trunk setter.
-- |
-- | Useful when there are multiple setters acting on the same data structure.
-- | and you need to optimize performance. For large operations (ie a setter
-- | over an audio file or a photo), this can have a 1.5-2x performance increase
-- | for each fused setter. The performance increase compounds with each nested
-- | fused setter.
-- |
-- | ```purescript
-- |over
-- | (_2 <<< (fuseSetters _1 (_2 <<< (fuseSetters _1 _2))))
-- | ( cTuple
-- | ((+) 55)
-- | (cTuple (flip (-) 101) ((*) 57))
-- | )
-- | (Tuple 0 (Tuple 0 (Tuple 1 2)))
-- | ```
-- |
-- | This yields:
-- |
-- | ```bash
-- | (Tuple 0 (Tuple 55 (Tuple -100 114)))
-- | ```
-- |
-- | Use `cTuple` along with `fuseSetters` to create the tree of functions used
-- | by `over`.
fuseSetters :: forall a b c. Setter' a b -> Setter' a c -> Setter' a (Tuple (b -> b) (c -> c))
fuseSetters ba ca l = (over ba fa <<< over ca fb)
where
t = l (Tuple identity identity)

fa = fst t

fb = snd t

-- | For use with `fuseSetters`
cTuple :: forall a b c. a -> b -> c -> Tuple a b
cTuple a b _ = Tuple a b