mirror of
https://github.com/nim-lang/Nim.git
synced 2025-12-29 09:24:36 +00:00
116 lines
3.3 KiB
Nim
116 lines
3.3 KiB
Nim
#
|
|
#
|
|
# The Nim Compiler
|
|
# (c) Copyright 2023 Andreas Rumpf
|
|
#
|
|
# See the file "copying.txt", included in this
|
|
# distribution, for details about the copyright.
|
|
#
|
|
|
|
## Nim Intermediate Representation, designed to capture all of Nim's semantics without losing too much
|
|
## precious information. Can easily be translated into C. And to JavaScript, hopefully.
|
|
|
|
from os import addFileExt, `/`, createDir
|
|
|
|
import ".." / [ast, modulegraphs, renderer, transf, options, msgs, lineinfos]
|
|
import nirtypes, nirinsts, ast2ir, nirlineinfos
|
|
|
|
import ".." / ic / [rodfiles, bitabs]
|
|
|
|
type
|
|
PCtx* = ref object of TPassContext
|
|
m: ModuleCon
|
|
c: ProcCon
|
|
oldErrorCount: int
|
|
|
|
proc newCtx*(module: PSym; g: ModuleGraph; idgen: IdGenerator): PCtx =
|
|
let m = initModuleCon(g, g.config, idgen, module)
|
|
PCtx(m: m, c: initProcCon(m, nil, g.config), idgen: idgen)
|
|
|
|
proc refresh*(c: PCtx; module: PSym; idgen: IdGenerator) =
|
|
c.m = initModuleCon(c.m.graph, c.m.graph.config, idgen, module)
|
|
c.c = initProcCon(c.m, nil, c.m.graph.config)
|
|
c.idgen = idgen
|
|
|
|
proc setupGlobalCtx*(module: PSym; graph: ModuleGraph; idgen: IdGenerator) =
|
|
if graph.repl.isNil:
|
|
graph.repl = newCtx(module, graph, idgen)
|
|
#registerAdditionalOps(PCtx graph.repl)
|
|
else:
|
|
refresh(PCtx graph.repl, module, idgen)
|
|
|
|
proc setupNirReplGen*(graph: ModuleGraph; module: PSym; idgen: IdGenerator): PPassContext =
|
|
setupGlobalCtx(module, graph, idgen)
|
|
result = PCtx graph.repl
|
|
|
|
proc evalStmt(c: PCtx; n: PNode) =
|
|
let n = transformExpr(c.m.graph, c.idgen, c.m.module, n)
|
|
let pc = genStmt(c.c, n)
|
|
|
|
var res = ""
|
|
if pc < c.c.code.len:
|
|
toString c.c.code, NodePos(pc), c.m.lit.strings, c.m.lit.numbers, res
|
|
#res.add "\n--------------------------\n"
|
|
#toString res, c.m.types.g
|
|
echo res
|
|
|
|
|
|
proc runCode*(c: PPassContext; n: PNode): PNode =
|
|
let c = PCtx(c)
|
|
# don't eval errornous code:
|
|
if c.oldErrorCount == c.m.graph.config.errorCounter:
|
|
evalStmt(c, n)
|
|
result = newNodeI(nkEmpty, n.info)
|
|
else:
|
|
result = n
|
|
c.oldErrorCount = c.m.graph.config.errorCounter
|
|
|
|
type
|
|
NirPassContext* = ref object of TPassContext
|
|
m: ModuleCon
|
|
c: ProcCon
|
|
|
|
proc openNirBackend*(g: ModuleGraph; module: PSym; idgen: IdGenerator): PPassContext =
|
|
let m = initModuleCon(g, g.config, idgen, module)
|
|
NirPassContext(m: m, c: initProcCon(m, nil, g.config), idgen: idgen)
|
|
|
|
proc gen(c: NirPassContext; n: PNode) =
|
|
let n = transformExpr(c.m.graph, c.idgen, c.m.module, n)
|
|
let pc = genStmt(c.c, n)
|
|
|
|
proc nirBackend*(c: PPassContext; n: PNode): PNode =
|
|
gen(NirPassContext(c), n)
|
|
result = n
|
|
|
|
proc closeNirBackend*(c: PPassContext; finalNode: PNode) =
|
|
discard nirBackend(c, finalNode)
|
|
|
|
let c = NirPassContext(c)
|
|
let nimcache = getNimcacheDir(c.c.config).string
|
|
createDir nimcache
|
|
let outp = nimcache / c.m.module.name.s.addFileExt("nir")
|
|
var r = rodfiles.create(outp)
|
|
try:
|
|
r.storeHeader(nirCookie)
|
|
r.storeSection stringsSection
|
|
r.store c.m.lit.strings
|
|
|
|
r.storeSection numbersSection
|
|
r.store c.m.lit.numbers
|
|
|
|
r.storeSection bodiesSection
|
|
r.store c.c.code
|
|
|
|
r.storeSection typesSection
|
|
r.store c.m.types.g
|
|
|
|
r.storeSection sideChannelSection
|
|
r.store c.m.man
|
|
|
|
finally:
|
|
r.close()
|
|
if r.err != ok:
|
|
rawMessage(c.c.config, errFatal, "serialization failed: " & outp)
|
|
else:
|
|
echo "created: ", outp
|