aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/lux/tool/compiler/phase/generation/python/loop.lux
blob: a58f02141db198a9f19ad8acfa9fb54a35c53141 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
(.module:
  [lux (#- Scope)
   [abstract
    ["." monad (#+ do)]]
   [data
    ["." product]
    [text
     format]
    [collection
     ["." list ("#@." functor)]]]
   [target
    ["_" python (#+ Expression)]]]
  ["." // #_
   [runtime (#+ Operation Phase)]
   ["#." case]
   ["#/" //
    ["#/" //
     [//
      [synthesis (#+ Scope Synthesis)]]]]])

(def: #export (scope generate [start initsS+ bodyS])
  (-> Phase (Scope Synthesis) (Operation (Expression Any)))
  (do ////.monad
    [@loop (:: @ map (|>> %n (format "loop") _.var) ///.next)
     initsO+ (monad.map @ generate initsS+)
     bodyO (///.with-anchor @loop
             (generate bodyS))
     _ (///.save! true ["" (_.code @loop)]
                  (_.def @loop (|> initsS+
                                   list.enumerate
                                   (list@map (|>> product.left (n/+ start) //case.register)))
                         (_.return bodyO)))]
    (wrap (_.apply/* @loop initsO+))))

(def: #export (recur generate argsS+)
  (-> Phase (List Synthesis) (Operation (Expression Any)))
  (do ////.monad
    [@scope ///.anchor
     argsO+ (monad.map @ generate argsS+)]
    (wrap (_.apply/* @scope argsO+))))