@@ -19,6 +19,7 @@
19{-# LANGUAGE UndecidableInstances #-}
19{-# LANGUAGE UndecidableInstances #-}
20{-# LANGUAGE PolyKinds #-}
20{-# LANGUAGE PolyKinds #-}
21{-# LANGUAGE InstanceSigs #-}
21{-# LANGUAGE InstanceSigs #-}
22#ifdef USE_REFLEX_OPTIMIZER
23#ifdef USE_REFLEX_OPTIMIZER
23{-# OPTIONS_GHC -fplugin=Reflex.Optimizer #-}
24{-# OPTIONS_GHC -fplugin=Reflex.Optimizer #-}
@@ -374,7 +375,7 @@ eventRoot !k !r = Event $ wrap eventSubscribedRoot $ liftIO . getRootSubscribed
374eventNever :: Event x a
375eventNever :: Event x a
375eventNever = Event $ \_ -> return (EventSubscription (return ()) eventSubscribedNever, Nothing)
376eventNever = Event $ \_ -> return (EventSubscription (return ()) eventSubscribedNever, Nothing)
377−eventFan :: (GCompare k, HasSpiderTimeline x) => k a -> Fan x k -> Event x a
378+eventFan :: (GCompare k, HasSpiderTimeline x) => k a -> Fan x k v -> Event x (v a)
378eventFan !k !f = Event $ wrap eventSubscribedFan $ getFanSubscribed k f
379eventFan !k !f = Event $ wrap eventSubscribedFan $ getFanSubscribed k f
380eventSwitch :: HasSpiderTimeline x => Switch x a -> Event x a
381eventSwitch :: HasSpiderTimeline x => Switch x a -> Event x a
@@ -426,14 +427,14 @@ newSubscriberHold h = return $ Subscriber
426 , subscriberRecalculateHeight = \_ -> return ()
427 , subscriberRecalculateHeight = \_ -> return ()
429−newSubscriberFan :: forall x k. (HasSpiderTimeline x, GCompare k) => FanSubscribed x k -> IO (Subscriber x (DMap k Identity))
430+newSubscriberFan :: forall x k v. (HasSpiderTimeline x, GCompare k) => FanSubscribed x k v -> IO (Subscriber x (DMap k v))
430newSubscriberFan subscribed = return $ Subscriber
431newSubscriberFan subscribed = return $ Subscriber
431 { subscriberPropagate = \a -> {-# SCC "traverseFan" #-} do
432 { subscriberPropagate = \a -> {-# SCC "traverseFan" #-} do
432 subs <- liftIO $ readIORef $ fanSubscribedSubscribers subscribed
433 subs <- liftIO $ readIORef $ fanSubscribedSubscribers subscribed
433 tracePropagate (Proxy :: Proxy x) $ "SubscriberFan" <> showNodeId subscribed <> ": " ++ show (DMap.size subs) ++ " keys subscribed, " ++ show (DMap.size a) ++ " keys firing"
434 tracePropagate (Proxy :: Proxy x) $ "SubscriberFan" <> showNodeId subscribed <> ": " ++ show (DMap.size subs) ++ " keys subscribed, " ++ show (DMap.size a) ++ " keys firing"
434 liftIO $ writeIORef (fanSubscribedOccurrence subscribed) $ Just a
435 liftIO $ writeIORef (fanSubscribedOccurrence subscribed) $ Just a
435 scheduleClear $ fanSubscribedOccurrence subscribed
436 scheduleClear $ fanSubscribedOccurrence subscribed
436− let f _ (Pair (Identity v) subsubs) = do
437+ let f _ (Pair v subsubs) = do
437 propagate v $ _fanSubscribedChildren_list subsubs
438 propagate v $ _fanSubscribedChildren_list subsubs
438 return $ Constant ()
439 return $ Constant ()
439 _ <- DMap.traverseWithKey f $ DMap.intersectionWithKey (\_ -> Pair) a subs --TODO: Would be nice to have DMap.traverse_
440 _ <- DMap.traverseWithKey f $ DMap.intersectionWithKey (\_ -> Pair) a subs --TODO: Would be nice to have DMap.traverse_
@@ -581,7 +582,7 @@ eventSubscribedNever = EventSubscribed
584−eventSubscribedFan :: FanSubscribed x k -> EventSubscribed x
585+eventSubscribedFan :: FanSubscribed x k v -> EventSubscribed x
585eventSubscribedFan !subscribed = EventSubscribed
586eventSubscribedFan !subscribed = EventSubscribed
586 { eventSubscribedHeightRef = eventSubscribedHeightRef $ _eventSubscription_subscribed $ fanSubscribedParent subscribed
587 { eventSubscribedHeightRef = eventSubscribedHeightRef $ _eventSubscription_subscribed $ fanSubscribedParent subscribed
587 , eventSubscribedRetained = toAny subscribed
588 , eventSubscribedRetained = toAny subscribed
@@ -990,7 +991,7 @@ data RootSubscribed x a = forall k. GCompare k => RootSubscribed
993−data Root x (k :: * -> *)
994+data Root x k
994 = Root { rootOccurrence :: !(IORef (DMap k Identity)) -- The currently-firing occurrence of this event
995 = Root { rootOccurrence :: !(IORef (DMap k Identity)) -- The currently-firing occurrence of this event
995 , rootSubscribed :: !(IORef (DMap k (RootSubscribed x)))
996 , rootSubscribed :: !(IORef (DMap k (RootSubscribed x)))
996 , rootInit :: !(forall a. k a -> RootTrigger x a -> IO (IO ()))
997 , rootInit :: !(forall a. k a -> RootTrigger x a -> IO (IO ()))
@@ -1060,25 +1061,25 @@ heightBagVerify b@(HeightBag s c) = if
1060heightBagVerify = id
1061heightBagVerify = id
1063−data FanSubscribedChildren (x :: *) k a = FanSubscribedChildren
1064+data FanSubscribedChildren x k v a = FanSubscribedChildren
1064− { _fanSubscribedChildren_list :: !(WeakBag (Subscriber x a))
1065+ { _fanSubscribedChildren_list :: !(WeakBag (Subscriber x (v a)))
1065− , _fanSubscribedChildren_self :: {-# NOUNPACK #-} !(k a, FanSubscribed x k)
1066+ , _fanSubscribedChildren_self :: {-# NOUNPACK #-} !(k a, FanSubscribed x k v)
1066− , _fanSubscribedChildren_weakSelf :: !(IORef (Weak (k a, FanSubscribed x k)))
1067+ , _fanSubscribedChildren_weakSelf :: !(IORef (Weak (k a, FanSubscribed x k v)))
1069−data FanSubscribed (x :: *) k
1070+data FanSubscribed x k v
1070− = FanSubscribed { fanSubscribedCachedSubscribed :: !(IORef (Maybe (FanSubscribed x k)))
1071+ = FanSubscribed { fanSubscribedCachedSubscribed :: !(IORef (Maybe (FanSubscribed x k v)))
1071− , fanSubscribedOccurrence :: !(IORef (Maybe (DMap k Identity)))
1072+ , fanSubscribedOccurrence :: !(IORef (Maybe (DMap k v)))
1072− , fanSubscribedSubscribers :: !(IORef (DMap k (FanSubscribedChildren x k))) -- This DMap should never be empty
1073+ , fanSubscribedSubscribers :: !(IORef (DMap k (FanSubscribedChildren x k v))) -- This DMap should never be empty
1073 , fanSubscribedParent :: !(EventSubscription x)
1074 , fanSubscribedParent :: !(EventSubscription x)
1074#ifdef DEBUG_NODEIDS
1075#ifdef DEBUG_NODEIDS
1075 , fanSubscribedNodeId :: Int
1076 , fanSubscribedNodeId :: Int
1079−data Fan x k
1080+data Fan x k v
1080− = Fan { fanParent :: !(Event x (DMap k Identity))
1081+ = Fan { fanParent :: !(Event x (DMap k v))
1081− , fanSubscribed :: !(IORef (Maybe (FanSubscribed x k)))
1082+ , fanSubscribed :: !(IORef (Maybe (FanSubscribed x k v)))
1084data SwitchSubscribed x a
1085data SwitchSubscribed x a
@@ -1525,7 +1526,7 @@ fanInt p =
1525 return (EventSubscription (FastWeakBag.remove t) $! EventSubscribed heightRef $! toAny (_fanInt_subscriptionRef self, t), IntMap.lookup k currentOcc)
1526 return (EventSubscription (FastWeakBag.remove t) $! EventSubscribed heightRef $! toAny (_fanInt_subscriptionRef self, t), IntMap.lookup k currentOcc)
1527{-# INLINABLE getFanSubscribed #-}
1528{-# INLINABLE getFanSubscribed #-}
1528−getFanSubscribed :: (HasSpiderTimeline x, GCompare k) => k a -> Fan x k -> Subscriber x a -> EventM x (WeakBagTicket, FanSubscribed x k, Maybe a)
1529+getFanSubscribed :: (HasSpiderTimeline x, GCompare k) => k a -> Fan x k v -> Subscriber x (v a) -> EventM x (WeakBagTicket, FanSubscribed x k v, Maybe (v a))
1529getFanSubscribed k f sub = do
1530getFanSubscribed k f sub = do
1530 mSubscribed <- liftIO $ readIORef $ fanSubscribed f
1531 mSubscribed <- liftIO $ readIORef $ fanSubscribed f
1531 case mSubscribed of
1532 case mSubscribed of
@@ -1559,7 +1560,7 @@ getFanSubscribed k f sub = do
1559 liftIO $ writeIORef (fanSubscribed f) $ Just subscribed
1560 liftIO $ writeIORef (fanSubscribed f) $ Just subscribed
1560 return (slnForSub, subscribed, coerce $ DMap.lookup k =<< parentOcc)
1561 return (slnForSub, subscribed, coerce $ DMap.lookup k =<< parentOcc)
1562−cleanupFanSubscribed :: GCompare k => (k a, FanSubscribed x k) -> IO ()
1563+cleanupFanSubscribed :: GCompare k => (k a, FanSubscribed x k v) -> IO ()
1563cleanupFanSubscribed (k, subscribed) = do
1564cleanupFanSubscribed (k, subscribed) = do
1564 subscribers <- readIORef $ fanSubscribedSubscribers subscribed
1565 subscribers <- readIORef $ fanSubscribedSubscribers subscribed
1565 let reducedSubscribers = DMap.delete k subscribers
1566 let reducedSubscribers = DMap.delete k subscribers
@@ -1571,7 +1572,7 @@ cleanupFanSubscribed (k, subscribed) = do
1571 else writeIORef (fanSubscribedSubscribers subscribed) $! reducedSubscribers
1572 else writeIORef (fanSubscribedSubscribers subscribed) $! reducedSubscribers
1573{-# INLINE subscribeFanSubscribed #-}
1574{-# INLINE subscribeFanSubscribed #-}
1574−subscribeFanSubscribed :: GCompare k => k a -> FanSubscribed x k -> Subscriber x a -> IO WeakBagTicket
1575+subscribeFanSubscribed :: GCompare k => k a -> FanSubscribed x k v -> Subscriber x (v a) -> IO WeakBagTicket
1575subscribeFanSubscribed k subscribed sub = do
1576subscribeFanSubscribed k subscribed sub = do
1576 subscribers <- readIORef $ fanSubscribedSubscribers subscribed
1577 subscribers <- readIORef $ fanSubscribedSubscribers subscribed
1577 case DMap.lookup k subscribers of
1578 case DMap.lookup k subscribers of
@@ -2047,14 +2048,15 @@ mergeIntCheap d = Event $ \sub -> do
2049newtype EventSelector x k = EventSelector { select :: forall a. k a -> Event x a }
2050newtype EventSelector x k = EventSelector { select :: forall a. k a -> Event x a }
2051+newtype EventSelectorG x k v = EventSelectorG { selectG :: forall a. k a -> Event x (v a) }
2051−fan :: (HasSpiderTimeline x, GCompare k) => Event x (DMap k Identity) -> EventSelector x k
2053+fanG :: (HasSpiderTimeline x, GCompare k) => Event x (DMap k v) -> EventSelectorG x k v
2052−fan e =
2054+fanG e =
2053 let f = Fan
2055 let f = Fan
2054 { fanParent = e
2056 { fanParent = e
2055 , fanSubscribed = unsafeNewIORef e Nothing
2057 , fanSubscribed = unsafeNewIORef e Nothing
2057− in EventSelector $ \k -> eventFan k f
2059+ in EventSelectorG $ \k -> eventFan k f
2059runHoldInits :: HasSpiderTimeline x => IORef [SomeHoldInit x] -> IORef [SomeDynInit x] -> IORef [SomeMergeInit x] -> EventM x ()
2061runHoldInits :: HasSpiderTimeline x => IORef [SomeHoldInit x] -> IORef [SomeDynInit x] -> IORef [SomeMergeInit x] -> EventM x ()
2060runHoldInits holdInitRef dynInitRef mergeInitRef = do
2062runHoldInits holdInitRef dynInitRef mergeInitRef = do
@@ -2523,6 +2525,8 @@ instance HasSpiderTimeline x => R.Reflex (SpiderTimeline x) where
2523 pushCheap f = SpiderEvent . pushCheap (coerce f) . unSpiderEvent
2525 pushCheap f = SpiderEvent . pushCheap (coerce f) . unSpiderEvent
2524 {-# INLINABLE pull #-}
2526 {-# INLINABLE pull #-}
2525 pull = SpiderBehavior . pull . coerce
2527 pull = SpiderBehavior . pull . coerce
2528+ {-# INLINABLE fanG #-}
2529+ fanG e = R.EventSelectorG $ SpiderEvent . selectG (fanG (unSpiderEvent e))
2526 {-# INLINABLE mergeG #-}
2530 {-# INLINABLE mergeG #-}
2528 :: forall (k :: k2 -> *) q (v :: k2 -> *). GCompare k
2532 :: forall (k :: k2 -> *) q (v :: k2 -> *). GCompare k
@@ -2530,8 +2534,6 @@ instance HasSpiderTimeline x => R.Reflex (SpiderTimeline x) where
2530 -> DMap k q
2534 -> DMap k q
2531 -> R.Event (SpiderTimeline x) (DMap k v)
2535 -> R.Event (SpiderTimeline x) (DMap k v)
2532 mergeG nt = SpiderEvent . mergeG (unSpiderEvent #. nt) . dynamicConst
2536 mergeG nt = SpiderEvent . mergeG (unSpiderEvent #. nt) . dynamicConst
2533− {-# INLINABLE fan #-}
2534− fan e = R.EventSelector $ SpiderEvent . select (fan (unSpiderEvent e))
2535 {-# INLINABLE switch #-}
2537 {-# INLINABLE switch #-}
2536 switch = SpiderEvent . switch . (coerce :: Behavior x (R.Event (SpiderTimeline x) a) -> Behavior x (Event x a)) . unSpiderBehavior
2538 switch = SpiderEvent . switch . (coerce :: Behavior x (R.Event (SpiderTimeline x) a) -> Behavior x (Event x a)) . unSpiderBehavior
2537 {-# INLINABLE coincidence #-}
2539 {-# INLINABLE coincidence #-}