diff options
Diffstat (limited to 'src/CHAD/AST/Count.hs')
| -rw-r--r-- | src/CHAD/AST/Count.hs | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/CHAD/AST/Count.hs b/src/CHAD/AST/Count.hs index 46173d2..8923e13 100644 --- a/src/CHAD/AST/Count.hs +++ b/src/CHAD/AST/Count.hs @@ -880,6 +880,16 @@ occCountX initialS topexpr k = case topexpr of EError _ t msg -> k OccEnd $ \_ -> EError ext (applySubstruc s t) msg + + EUser _ t e -> + occCountX SsFull e $ \env1 mke -> + k env1 $ \env' -> + projectSmallerSubstruc SsFull s $ EUser ext t (mke env') + + EUnUser _ e -> + occCountX SsFull e $ \env1 mke -> + k env1 $ \env' -> + projectSmallerSubstruc SsFull s $ EUnUser ext (mke env') where s = simplifySubstruc (typeOf topexpr) initialS |