@@ -1475,31 +1475,34 @@ implicitise syn ignore ist tm = -- trace ("INCOMING " ++ showImp True tm) $
14751475
14761476-- Add implicit arguments in function calls
14771477addImplPat :: IState -> PTerm -> PTerm
1478−addImplPat = addImpl' True [] []
1478+addImplPat = addImpl' True [] [] []
14791479
14801480addImplBound :: IState -> [Name] -> PTerm -> PTerm
1481−addImplBound ist ns = addImpl' False ns [] ist
1481+addImplBound ist ns = addImpl' False ns [] [] ist
14821482
14831483addImplBoundInf :: IState -> [Name] -> [Name] -> PTerm -> PTerm
1484−addImplBoundInf ist ns inf = addImpl' False ns inf ist
1484+addImplBoundInf ist ns inf = addImpl' False ns inf [] ist
14851485
14861486-- | Add the implicit arguments to applications in the term
1487−addImpl :: IState -> PTerm -> PTerm
1487+-- [Name] gives the names to always expend, even when under a binder of
1488+-- that name (this is to expand methods with implicit arguments in dependent
1489+-- type classes).
1490+addImpl :: [Name] -> IState -> PTerm -> PTerm
14881491addImpl = addImpl' False [] []
14891492
14901493-- TODO: in patterns, don't add implicits to function names guarded by constructors
14911494-- and *not* inside a PHidden
14921495
1493−addImpl' :: Bool -> [Name] -> [Name] -> IState -> PTerm -> PTerm
1494−addImpl' inpat env infns ist ptm
1496+addImpl' :: Bool -> [Name] -> [Name] -> [Name] -> IState -> PTerm -> PTerm
1497+addImpl' inpat env infns imp_meths ist ptm
14951498 = mkUniqueNames env (ai False (zip env (repeat Nothing)) [] ptm)
14961499 where
14971500 ai :: Bool -> [(Name, Maybe PTerm)] -> [[T.Text]] -> PTerm -> PTerm
14981501 ai qq env ds (PRef fc f)
14991502 | f `elem` infns = PInferRef fc f
1500− | not (f `elem` map fst env) = handleErr $ aiFn inpat inpat qq ist fc f ds []
1503+ | not (f `elem` map fst env) = handleErr $ aiFn inpat inpat qq imp_meths ist fc f ds []
15011504 ai qq env ds (PHidden (PRef fc f))
1502− | not (f `elem` map fst env) = PHidden (handleErr $ aiFn inpat False qq ist fc f ds [])
1505+ | not (f `elem` map fst env) = PHidden (handleErr $ aiFn inpat False qq imp_meths ist fc f ds [])
15031506 ai qq env ds (PEq fc lt rt l r)
15041507 = let lt' = ai qq env ds lt
15051508 rt' = ai qq env ds rt
@@ -1535,7 +1538,7 @@ addImpl' inpat env infns ist ptm
15351538 | f `elem` infns = ai qq env ds (PApp fc (PInferRef fc f) as)
15361539 | not (f `elem` map fst env)
15371540 = let as' = map (fmap (ai qq env ds)) as in
1538− handleErr $ aiFn inpat False qq ist fc f ds as'
1541+ handleErr $ aiFn inpat False qq imp_meths ist fc f ds as'
15391542 | Just (Just ty) <- lookup f env =
15401543 let as' = map (fmap (ai qq env ds)) as
15411544 arity = getPArity ty in
@@ -1570,7 +1573,9 @@ addImpl' inpat env infns ist ptm
15701573 _ -> ai qq env ds (PCase fc val [(PRef fc n, sc)])
15711574 ai qq env ds (PPi p n ty sc)
15721575 = let ty' = ai qq env ds ty
1573− sc' = ai qq ((n, Just ty):env) ds sc in
1576+ env' = if n `elem` imp_meths then env
1577+ else ((n, Just ty) : env)
1578+ sc' = ai qq env' ds sc in
15741579 PPi p n ty' sc'
15751580 ai qq env ds (PGoal fc r n sc)
15761581 = let r' = ai qq env ds r
@@ -1594,8 +1599,8 @@ addImpl' inpat env infns ist ptm
15941599-- if in a pattern, and there are no arguments, and there's no possible
15951600-- names with zero explicit arguments, don't add implicits.
15961601
1597−aiFn :: Bool -> Bool -> Bool -> IState -> FC -> Name -> [[T.Text]] -> [PArg] -> Either Err PTerm
1598−aiFn inpat True qq ist fc f ds []
1602+aiFn :: Bool -> Bool -> Bool -> [Name] -> IState -> FC -> Name -> [[T.Text]] -> [PArg] -> Either Err PTerm
1603+aiFn inpat True qq imp_meths ist fc f ds []
15991604 = case lookupDef f (tt_ctxt ist) of
16001605 [] -> Right $ PPatvar fc f
16011606 alts -> let ialts = lookupCtxtName f (idris_implicits ist) in
@@ -1603,7 +1608,7 @@ aiFn inpat True qq ist fc f ds []
16031608 if (not (vname f) || tcname f
16041609 || any (conCaf (tt_ctxt ist)) ialts)
16051610-- any constructor alts || any allImp ialts))
1606− then aiFn inpat False qq ist fc f ds [] -- use it as a constructor
1611+ then aiFn inpat False qq imp_meths ist fc f ds [] -- use it as a constructor
16071612 else Right $ PPatvar fc f
16081613 where imp (PExp _ _ _ _) = False
16091614 imp _ = True
@@ -1617,9 +1622,9 @@ aiFn inpat True qq ist fc f ds []
16171622 vname (UN n) = True -- non qualified
16181623 vname _ = False
16191624
1620−aiFn inpat expat qq ist fc f ds as
1625+aiFn inpat expat qq imp_meths ist fc f ds as
16211626 | f `elem` primNames = Right $ PApp fc (PRef fc f) as
1622−aiFn inpat expat qq ist fc f ds as
1627+aiFn inpat expat qq imp_meths ist fc f ds as
16231628 -- This is where namespaces get resolved by adding PAlternative
16241629 = do let ns = lookupCtxtName f (idris_implicits ist)
16251630 let nh = filter (\(n, _) -> notHidden n) ns
@@ -1627,15 +1632,20 @@ aiFn inpat expat qq ist fc f ds as
16271632 [] -> nh
16281633 x -> x
16291634 case ns' of
1630− [(f',ns)] -> Right $ mkPApp fc (length ns) (PRef fc f') (insertImpl ns as)
1635+ [(f',ns)] -> Right $ mkPApp fc (length ns) (PRef fc (isImpName f f')) (insertImpl ns as)
16311636 [] -> if f `elem` (map fst (idris_metavars ist))
16321637 then Right $ PApp fc (PRef fc f) as
16331638 else Right $ mkPApp fc (length as) (PRef fc f) as
16341639 alts -> Right $
16351640 PAlternative True $
1636− map (\(f', ns) -> mkPApp fc (length ns) (PRef fc f')
1641+ map (\(f', ns) -> mkPApp fc (length ns) (PRef fc (isImpName f f'))
16371642 (insertImpl ns as)) alts
16381643 where
1644+ -- if the name is in imp_meths, we should actually refer to the bound
1645+ -- name rather than the global one after expanding implicits
1646+ isImpName f f' | f `elem` imp_meths = f
1647+ | otherwise = f'
1648+
16391649 trimAlts [] alts = alts
16401650 trimAlts ns alts
16411651 = filter (\(x, _) -> any (\d -> d `isPrefixOf` nspace x) ns) alts
@@ -1679,7 +1689,7 @@ aiFn inpat expat qq ist fc f ds as
16791689 PImp p True l n Placeholder :
16801690 insImpAcc (M.insert n Placeholder pnas) ps given imps
16811691 insImpAcc pnas (PTacImplicit p l n sc' ty : ps) given imps =
1682− let sc = addImpl ist (substMatches (M.toList pnas) sc') in
1692+ let sc = addImpl imp_meths ist (substMatches (M.toList pnas) sc') in
16831693 case find n imps [] of
16841694 Just (tm, imps') ->
16851695 PTacImplicit p l n sc tm :