@@ -80,6 +80,30 @@ elabType info syn doc fc opts n ty' = {- let ty' = piBind (params info) ty_in
8080 when corec $ do setAccessibility n Frozen
8181 addIBC (IBCAccess n Frozen)
8282
83+elabPostulate :: ElabInfo -> SyntaxInfo -> String ->
84+ FC -> FnOpts -> Name -> PTerm -> Idris ()
85+elabPostulate info syn doc fc opts n ty
86+ = do elabType info syn doc fc opts n ty
87+ -- make sure it's collapsible, so it is never needed at run time
88+ -- start by getting the elaborated type
89+ ctxt <- getContext
90+ fty <- case lookupTy Nothing n ctxt of
91+ [] -> tclift $ tfail $ (At fc (NoTypeDecl n)) -- can't happen!
92+ [ty] -> return ty
93+ ist <- get
94+ let (ap, _) = unApply (getRetTy fty)
95+ logLvl 5 $ "Checking collapsibility of " ++ show (ap, fty)
96+ let postOK = case ap of
97+ P _ tn _ -> case lookupCtxt Nothing tn
98+ (idris_optimisation ist) of
99+ [oi] -> collapsible oi
100+ _ -> False
101+ _ -> False
102+ when (not postOK)
103+ $ tclift $ tfail (At fc (NonCollapsiblePostulate n))
104+ -- remove it from the deferred definitions list
105+ solveDeferred n
106+
83107elabData :: ElabInfo -> SyntaxInfo -> String -> FC -> Bool -> PData -> Idris ()
84108elabData info syn doc fc codata (PLaterdecl n t_in)
85109 = do iLOG (show (fc, doc))
@@ -274,120 +298,123 @@ elabClauses :: ElabInfo -> FC -> FnOpts -> Name -> [PClause] -> Idris ()
274298elabClauses info fc opts n_in cs = let n = liftname info n_in in
275299 do ctxt <- getContext
276300 -- Check n actually exists
277− fty <- case lookupTy Nothing n ctxt of
278− [] -> -- TODO: turn into a CAF if there's no arguments
279− -- question: CAFs in where blocks?
280− tclift $ tfail $ (At fc (NoTypeDecl n))
281− [ty] -> return ty
282− pats_in <- mapM (elabClause info (TCGen `elem` opts))
283− (zip [0..] cs)
284− -- if the return type of 'ty' is collapsible, the optimised version should
285− -- just do nothing
286− ist <- get
287− let (ap, _) = unApply (getRetTy fty)
288− logLvl 5 $ "Checking collapsibility of " ++ show (ap, fty)
289− -- FIXME: Really ought to only do this for total functions!
290− let doNothing = case ap of
291− P _ tn _ -> case lookupCtxt Nothing tn
292− (idris_optimisation ist) of
293− [oi] -> collapsible oi
294− _ -> False
295− _ -> False
296− solveDeferred n
297− ist <- get
298− when doNothing $
299− case lookupCtxt Nothing n (idris_optimisation ist) of
300− [oi] -> do let opts = addDef n (oi { collapsible = True })
301− (idris_optimisation ist)
302− put (ist { idris_optimisation = opts })
303− _ -> do let opts = addDef n (Optimise True [] [])
304− (idris_optimisation ist)
305− put (ist { idris_optimisation = opts })
306− addIBC (IBCOpt n)
307− ist <- get
308− let pats = pats_in
309−-- logLvl 3 (showSep "\n" (map (\ (l,r) ->
310−-- show l ++ " = " ++
311−-- show r) pats))
312− let tcase = opt_typecase (idris_options ist)
313− let pdef = map debind $ map (simpl False (tt_ctxt ist)) pats
314−
315− numArgs <- tclift $ sameLength pdef
316−
317− optpats <- if doNothing
318− then return $ [Right (mkApp (P Bound n Erased)
319− (take numArgs (repeat Erased)), Erased)]
320− else stripCollapsed pats
321−
322− logLvl 5 $ "Patterns:\n" ++ show pats
323−
324− let optpdef = map debind $ map (simpl True (tt_ctxt ist)) optpats
325− tree@(CaseDef scargs sc _) <- tclift $
326− simpleCase tcase False CompileTime fc pdef
327− cov <- coverage
328− pmissing <-
329− if cov
330− then do missing <- genClauses fc n (map getLHS pdef) cs
331− -- missing <- genMissing n scargs sc
332− missing' <- filterM (checkPossible info fc True n) missing
333− logLvl 2 $ "Must be unreachable:\n" ++
334− showSep "\n" (map (showImp True) missing') ++
335− "\nAgainst: " ++
336− showSep "\n" (map (\t -> showImp True (delab ist t)) (map getLHS pdef))
337− return missing'
338− else return []
339− let pcover = null pmissing
340− pdef' <- applyOpts optpdef
341− logLvl 5 $ "Optimised patterns:\n" ++ show pdef'
342− ist <- get
343−-- let wf = wellFounded ist n sc
344− let tot = if pcover || AssertTotal `elem` opts
345− then Unchecked -- finish checking later
346− else Partial NotCovering -- already know it's not total
347−-- case lookupCtxt (namespace info) n (idris_flags ist) of
348−-- [fs] -> if TotalFn `elem` fs
349−-- then case tot of
350−-- Total _ -> return ()
351−-- t -> tclift $ tfail (At fc (Msg (show n ++ " is " ++ show t)))
352−-- else return ()
353−-- _ -> return ()
354− case tree of
355− CaseDef _ _ [] -> return ()
356− CaseDef _ _ xs -> mapM_ (\x ->
357− iputStrLn $ show fc ++
358− ":warning - Unreachable case: " ++
359− show (delab ist x)) xs
360− let knowncovering = pcover && cov
361−
362− tree' <- tclift $ simpleCase tcase knowncovering RunTime fc pdef'
363− logLvl 3 (show tree)
364− logLvl 3 $ "Optimised: " ++ show tree'
365− ctxt <- getContext
366− ist <- get
367− put (ist { idris_patdefs = addDef n pdef' (idris_patdefs ist) })
368− case lookupTy (namespace info) n ctxt of
369− [ty] -> do updateContext (addCasedef n (inlinable opts)
370− tcase knowncovering pats
371− pdef pdef' ty)
372− addIBC (IBCDef n)
373− setTotality n tot
374− totcheck (fc, n)
375− when (tot /= Unchecked) $ addIBC (IBCTotal n tot)
376− i <- get
377− case lookupDef Nothing n (tt_ctxt i) of
378− (CaseOp _ _ _ _ _ scargs sc scargs' sc' : _) ->
379− do let calls = findCalls sc' scargs'
380− let used = findUsedArgs sc' scargs'
381− -- let scg = buildSCG i sc scargs
382− -- add SCG later, when checking totality
383− let cg = CGInfo scargs' calls [] used []
384− logLvl 2 $ "Called names: " ++ show cg
385− addToCG n cg
386− addToCalledG n (nub (map fst calls)) -- plus names in type!
387− addIBC (IBCCG n)
388− _ -> return ()
389−-- addIBC (IBCTotal n tot)
390− [] -> return ()
301+ let tys = lookupTy Nothing n ctxt
302+ unless (length tys > 1) $ do
303+ fty <- case tys of
304+ [] -> -- TODO: turn into a CAF if there's no arguments
305+ -- question: CAFs in where blocks?
306+ tclift $ tfail $ At fc (NoTypeDecl n)
307+ [ty] -> return ty
308+ pats_in <- mapM (elabClause info (TCGen `elem` opts))
309+ (zip [0..] cs)
310+ -- if the return type of 'ty' is collapsible, the optimised version should
311+ -- just do nothing
312+ ist <- get
313+ let (ap, _) = unApply (getRetTy fty)
314+ logLvl 5 $ "Checking collapsibility of " ++ show (ap, fty)
315+ -- FIXME: Really ought to only do this for total functions!
316+ let doNothing = case ap of
317+ P _ tn _ -> case lookupCtxt Nothing tn
318+ (idris_optimisation ist) of
319+ [oi] -> collapsible oi
320+ _ -> False
321+ _ -> False
322+ solveDeferred n
323+ ist <- get
324+ when doNothing $
325+ case lookupCtxt Nothing n (idris_optimisation ist) of
326+ [oi] -> do let opts = addDef n (oi { collapsible = True })
327+ (idris_optimisation ist)
328+ put (ist { idris_optimisation = opts })
329+ _ -> do let opts = addDef n (Optimise True [] [])
330+ (idris_optimisation ist)
331+ put (ist { idris_optimisation = opts })
332+ addIBC (IBCOpt n)
333+ ist <- get
334+ let pats = pats_in
335+ -- logLvl 3 (showSep "\n" (map (\ (l,r) ->
336+ -- show l ++ " = " ++
337+ -- show r) pats))
338+ let tcase = opt_typecase (idris_options ist)
339+ let pdef = map debind $ map (simpl False (tt_ctxt ist)) pats
340+
341+ numArgs <- tclift $ sameLength pdef
342+
343+ optpats <- if doNothing
344+ then return $ [Right (mkApp (P Bound n Erased)
345+ (take numArgs (repeat Erased)), Erased)]
346+ else stripCollapsed pats
347+
348+ logLvl 5 $ "Patterns:\n" ++ show pats
349+
350+ let optpdef = map debind $ map (simpl True (tt_ctxt ist)) optpats
351+ tree@(CaseDef scargs sc _) <- tclift $
352+ simpleCase tcase False CompileTime fc pdef
353+ cov <- coverage
354+ pmissing <-
355+ if cov
356+ then do missing <- genClauses fc n (map getLHS pdef) cs
357+ -- missing <- genMissing n scargs sc
358+ missing' <- filterM (checkPossible info fc True n) missing
359+ logLvl 2 $ "Must be unreachable:\n" ++
360+ showSep "\n" (map (showImp True) missing') ++
361+ "\nAgainst: " ++
362+ showSep "\n" (map (\t -> showImp True (delab ist t)) (map getLHS pdef))
363+ return missing'
364+ else return []
365+ let pcover = null pmissing
366+ pdef' <- applyOpts optpdef
367+ logLvl 5 $ "Optimised patterns:\n" ++ show pdef'
368+ ist <- get
369+ -- let wf = wellFounded ist n sc
370+ let tot = if pcover || AssertTotal `elem` opts
371+ then Unchecked -- finish checking later
372+ else Partial NotCovering -- already know it's not total
373+ -- case lookupCtxt (namespace info) n (idris_flags ist) of
374+ -- [fs] -> if TotalFn `elem` fs
375+ -- then case tot of
376+ -- Total _ -> return ()
377+ -- t -> tclift $ tfail (At fc (Msg (show n ++ " is " ++ show t)))
378+ -- else return ()
379+ -- _ -> return ()
380+ case tree of
381+ CaseDef _ _ [] -> return ()
382+ CaseDef _ _ xs -> mapM_ (\x ->
383+ iputStrLn $ show fc ++
384+ ":warning - Unreachable case: " ++
385+ show (delab ist x)) xs
386+ let knowncovering = pcover && cov
387+
388+ tree' <- tclift $ simpleCase tcase knowncovering RunTime fc pdef'
389+ logLvl 3 (show tree)
390+ logLvl 3 $ "Optimised: " ++ show tree'
391+ ctxt <- getContext
392+ ist <- get
393+ put (ist { idris_patdefs = addDef n pdef' (idris_patdefs ist) })
394+ case lookupTy (namespace info) n ctxt of
395+ [ty] -> do updateContext (addCasedef n (inlinable opts)
396+ tcase knowncovering pats
397+ pdef pdef' ty)
398+ addIBC (IBCDef n)
399+ setTotality n tot
400+ totcheck (fc, n)
401+ when (tot /= Unchecked) $ addIBC (IBCTotal n tot)
402+ i <- get
403+ case lookupDef Nothing n (tt_ctxt i) of
404+ (CaseOp _ _ _ _ _ scargs sc scargs' sc' : _) ->
405+ do let calls = findCalls sc' scargs'
406+ let used = findUsedArgs sc' scargs'
407+ -- let scg = buildSCG i sc scargs
408+ -- add SCG later, when checking totality
409+ let cg = CGInfo scargs' calls [] used []
410+ logLvl 2 $ "Called names: " ++ show cg
411+ addToCG n cg
412+ addToCalledG n (nub (map fst calls)) -- plus names in type!
413+ addIBC (IBCCG n)
414+ _ -> return ()
415+ -- addIBC (IBCTotal n tot)
416+ [] -> return ()
417+ return ()
391418 where
392419 debind (Right (x, y)) = let (vs, x') = depat [] x
393420 (_, y') = depat [] y in
@@ -417,8 +444,16 @@ elabVal info aspat tm_in
417444 i <- get
418445 let tm = addImpl i tm_in
419446 logLvl 10 (showImp True tm)
420− ((tm', defer, is), _) <- tclift $ elaborate ctxt (MN 0 "val") infP []
421− (build i info aspat (MN 0 "val") (infTerm tm))
447+ -- try:
448+ -- * ordinary elaboration
449+ -- * elaboration as a Set
450+ -- * elaboration as a function a -> b
451+
452+ ((tm', defer, is), _) <-
453+ tctry (elaborate ctxt (MN 0 "val") (Set (UVal 0)) []
454+ (build i info aspat (MN 0 "val") tm))
455+ (elaborate ctxt (MN 0 "val") infP []
456+ (build i info aspat (MN 0 "val") (infTerm tm)))
422457 logLvl 3 ("Value: " ++ show tm')
423458 let vtm = getInferTerm tm'
424459 logLvl 2 (show vtm)
@@ -533,8 +568,9 @@ elabClause info tcgen (_, PWith fc fname lhs_in withs wval_in withblock)
533568 i <- get
534569 let lhs = addImpl i lhs_in
535570 logLvl 5 ("LHS: " ++ showImp True lhs)
536− ((lhs', dlhs, []), _) <- tclift $ elaborate ctxt (MN 0 "patLHS") infP []
537− (erun fc (buildTC i info True tcgen fname (infTerm lhs)))
571+ ((lhs', dlhs, []), _) <-
572+ tclift $ elaborate ctxt (MN 0 "patLHS") infP []
573+ (erun fc (buildTC i info True tcgen fname (infTerm lhs)))
538574 let lhs_tm = orderPats (getInferTerm lhs')
539575 let lhs_ty = getInferType lhs'
540576 let ret_ty = getRetTy lhs_ty
@@ -990,6 +1026,10 @@ elabDecl' what info (PTy doc s f o n ty)
9901026 | what /= EDefns
9911027 = do iLOG $ "Elaborating type decl " ++ show n ++ show o
9921028 elabType info s doc f o n ty
1029+elabDecl' what info (PPostulate doc s f o n ty)
1030+ | what /= EDefns
1031+ = do iLOG $ "Elaborating postulate " ++ show n ++ show o
1032+ elabPostulate info s doc f o n ty
9931033elabDecl' what info (PData doc s f co d)
9941034 | what /= ETypes
9951035 = do iLOG $ "Elaborating " ++ show (d_name d)