module Optics.AffineFold
(
AffineFold
, afolding
, preview
, previews
, afoldVL
, filtered
, atraverseOf_
, isn't
, afailing
, An_AffineFold
) where
import Data.Maybe
import Data.Profunctor.Indexed
import Optics.Internal.Bi
import Optics.Internal.Optic
type AffineFold s a = Optic' An_AffineFold NoIx s a
afoldVL
:: (forall f. Functor f => (forall r. r -> f r) -> (a -> f u) -> s -> f v)
-> AffineFold s a
afoldVL :: (forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f u) -> s -> f v)
-> AffineFold s a
afoldVL forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f u) -> s -> f v
f = (forall (p :: * -> * -> * -> *) i.
Profunctor p =>
Optic_ An_AffineFold p i (Curry NoIx i) s s a a)
-> AffineFold s a
forall k (is :: IxList) s t a b.
(forall (p :: * -> * -> * -> *) i.
Profunctor p =>
Optic_ k p i (Curry is i) s t a b)
-> Optic k is s t a b
Optic (p i s v -> p i s s
forall (p :: * -> * -> * -> *) i c a b.
(Profunctor p, Bicontravariant p) =>
p i c a -> p i c b
rphantom (p i s v -> p i s s) -> (p i a a -> p i s v) -> p i a a -> p i s s
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f u) -> s -> f v)
-> p i a u -> p i s v
forall (p :: * -> * -> * -> *) i s t a b.
Visiting p =>
(forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f b) -> s -> f t)
-> p i a b -> p i s t
visit forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f u) -> s -> f v
f (p i a u -> p i s v) -> (p i a a -> p i a u) -> p i a a -> p i s v
forall b c a. (b -> c) -> (a -> b) -> a -> c
. p i a a -> p i a u
forall (p :: * -> * -> * -> *) i c a b.
(Profunctor p, Bicontravariant p) =>
p i c a -> p i c b
rphantom)
{-# INLINE afoldVL #-}
preview :: Is k An_AffineFold => Optic' k is s a -> s -> Maybe a
preview :: Optic' k is s a -> s -> Maybe a
preview Optic' k is s a
o = Optic' k is s a -> (a -> a) -> s -> Maybe a
forall k (is :: IxList) s a r.
Is k An_AffineFold =>
Optic' k is s a -> (a -> r) -> s -> Maybe r
previews Optic' k is s a
o a -> a
forall a. a -> a
id
{-# INLINE preview #-}
previews :: Is k An_AffineFold => Optic' k is s a -> (a -> r) -> s -> Maybe r
previews :: Optic' k is s a -> (a -> r) -> s -> Maybe r
previews Optic' k is s a
o = \a -> r
f -> ForgetM r (Curry is Any) s s -> s -> Maybe r
forall r i a b. ForgetM r i a b -> a -> Maybe r
runForgetM (ForgetM r (Curry is Any) s s -> s -> Maybe r)
-> ForgetM r (Curry is Any) s s -> s -> Maybe r
forall a b. (a -> b) -> a -> b
$
Optic An_AffineFold is s s a a
-> Optic_ An_AffineFold (ForgetM r) Any (Curry is Any) s s a a
forall (p :: * -> * -> * -> *) k (is :: IxList) s t a b i.
Profunctor p =>
Optic k is s t a b -> Optic_ k p i (Curry is i) s t a b
getOptic (Optic' k is s a -> Optic An_AffineFold is s s a a
forall destKind srcKind (is :: IxList) s t a b.
Is srcKind destKind =>
Optic srcKind is s t a b -> Optic destKind is s t a b
castOptic @An_AffineFold Optic' k is s a
o) Optic__ (ForgetM r) Any (Curry is Any) s s a a
-> Optic__ (ForgetM r) Any (Curry is Any) s s a a
forall a b. (a -> b) -> a -> b
$ (a -> Maybe r) -> ForgetM r Any a a
forall r i a b. (a -> Maybe r) -> ForgetM r i a b
ForgetM (r -> Maybe r
forall a. a -> Maybe a
Just (r -> Maybe r) -> (a -> r) -> a -> Maybe r
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> r
f)
{-# INLINE previews #-}
atraverseOf_
:: (Is k An_AffineFold, Functor f)
=> Optic' k is s a
-> (forall r. r -> f r) -> (a -> f u) -> s -> f ()
atraverseOf_ :: Optic' k is s a -> (forall r. r -> f r) -> (a -> f u) -> s -> f ()
atraverseOf_ Optic' k is s a
o forall r. r -> f r
point a -> f u
f s
s = case Optic' k is s a -> s -> Maybe a
forall k (is :: IxList) s a.
Is k An_AffineFold =>
Optic' k is s a -> s -> Maybe a
preview Optic' k is s a
o s
s of
Just a
a -> () () -> f u -> f ()
forall (f :: * -> *) a b. Functor f => a -> f b -> f a
<$ a -> f u
f a
a
Maybe a
Nothing -> () -> f ()
forall r. r -> f r
point ()
{-# INLINE atraverseOf_ #-}
afolding :: (s -> Maybe a) -> AffineFold s a
afolding :: (s -> Maybe a) -> AffineFold s a
afolding s -> Maybe a
f = (forall (p :: * -> * -> * -> *) i.
Profunctor p =>
Optic_ An_AffineFold p i (Curry NoIx i) s s a a)
-> AffineFold s a
forall k (is :: IxList) s t a b.
(forall (p :: * -> * -> * -> *) i.
Profunctor p =>
Optic_ k p i (Curry is i) s t a b)
-> Optic k is s t a b
Optic ((s -> Either s a)
-> (s -> Either s a) -> p i (Either s a) (Either s a) -> p i s s
forall (p :: * -> * -> * -> *) b a d c i.
Bicontravariant p =>
(b -> a) -> (d -> c) -> p i a c -> p i b d
contrabimap (\s
s -> Either s a -> (a -> Either s a) -> Maybe a -> Either s a
forall b a. b -> (a -> b) -> Maybe a -> b
maybe (s -> Either s a
forall a b. a -> Either a b
Left s
s) a -> Either s a
forall a b. b -> Either a b
Right (s -> Maybe a
f s
s)) s -> Either s a
forall a b. a -> Either a b
Left (p i (Either s a) (Either s a) -> p i s s)
-> (p i a a -> p i (Either s a) (Either s a)) -> p i a a -> p i s s
forall b c a. (b -> c) -> (a -> b) -> a -> c
. p i a a -> p i (Either s a) (Either s a)
forall (p :: * -> * -> * -> *) i a b c.
Choice p =>
p i a b -> p i (Either c a) (Either c b)
right')
{-# INLINE afolding #-}
filtered :: (a -> Bool) -> AffineFold a a
filtered :: (a -> Bool) -> AffineFold a a
filtered a -> Bool
p = (forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f a) -> a -> f a)
-> AffineFold a a
forall a u s v.
(forall (f :: * -> *).
Functor f =>
(forall r. r -> f r) -> (a -> f u) -> s -> f v)
-> AffineFold s a
afoldVL (\forall r. r -> f r
point a -> f a
f a
a -> if a -> Bool
p a
a then a -> f a
f a
a else a -> f a
forall r. r -> f r
point a
a)
{-# INLINE filtered #-}
afailing
:: (Is k An_AffineFold, Is l An_AffineFold)
=> Optic' k is s a
-> Optic' l js s a
-> AffineFold s a
afailing :: Optic' k is s a -> Optic' l js s a -> AffineFold s a
afailing Optic' k is s a
a Optic' l js s a
b = (s -> Maybe a) -> AffineFold s a
forall s a. (s -> Maybe a) -> AffineFold s a
afolding ((s -> Maybe a) -> AffineFold s a)
-> (s -> Maybe a) -> AffineFold s a
forall a b. (a -> b) -> a -> b
$ \s
s -> Maybe a -> (a -> Maybe a) -> Maybe a -> Maybe a
forall b a. b -> (a -> b) -> Maybe a -> b
maybe (Optic' l js s a -> s -> Maybe a
forall k (is :: IxList) s a.
Is k An_AffineFold =>
Optic' k is s a -> s -> Maybe a
preview Optic' l js s a
b s
s) a -> Maybe a
forall a. a -> Maybe a
Just (Optic' k is s a -> s -> Maybe a
forall k (is :: IxList) s a.
Is k An_AffineFold =>
Optic' k is s a -> s -> Maybe a
preview Optic' k is s a
a s
s)
infixl 3 `afailing`
{-# INLINE afailing #-}
isn't :: Is k An_AffineFold => Optic' k is s a -> s -> Bool
isn't :: Optic' k is s a -> s -> Bool
isn't Optic' k is s a
k s
s = Bool -> Bool
not (Maybe a -> Bool
forall a. Maybe a -> Bool
isJust (Optic' k is s a -> s -> Maybe a
forall k (is :: IxList) s a.
Is k An_AffineFold =>
Optic' k is s a -> s -> Maybe a
preview Optic' k is s a
k s
s))
{-# INLINE isn't #-}