Re-tabbed to 2 spaces to follow convention.
This commit is contained in:
Joey Payne
parent
52730a6564
commit
957d8dbd66
1 changed files with 288 additions and 288 deletions
576
oop_macro.nim
576
oop_macro.nim
|
|
@ -4,327 +4,327 @@ import strutils
|
|||
{.hint[XDeclaredButNotUsed]: off.}
|
||||
|
||||
macro new*(obj: untyped): untyped {.immediate.}=
|
||||
if obj.kind == nnkObjConstr or obj.kind == nnkCall:
|
||||
var args: seq[NimNode] = @[]
|
||||
var new_obj = copyNimTree(obj)
|
||||
if obj.kind == nnkObjConstr or obj.kind == nnkCall:
|
||||
var args: seq[NimNode] = @[]
|
||||
var new_obj = copyNimTree(obj)
|
||||
|
||||
# delete the parameters from the object
|
||||
# since we are going to use init() instead
|
||||
new_obj.del(1, len(new_obj)-1)
|
||||
# delete the parameters from the object
|
||||
# since we are going to use init() instead
|
||||
new_obj.del(1, len(new_obj)-1)
|
||||
|
||||
# create a symbol that we can use
|
||||
var sym = genSym(nskVar, "init_obj")
|
||||
# create a symbol that we can use
|
||||
var sym = genSym(nskVar, "init_obj")
|
||||
|
||||
# self = init_obj
|
||||
var self_eq = newNimNode(nnkExprEqExpr).add(ident"self", sym)
|
||||
# self = init_obj
|
||||
var self_eq = newNimNode(nnkExprEqExpr).add(ident"self", sym)
|
||||
|
||||
args.add(self_eq)
|
||||
args.add(self_eq)
|
||||
|
||||
for ch in obj.children:
|
||||
if ch.kind == nnkExprColonExpr:
|
||||
# some_arg = some_val
|
||||
args.add(newNimNode(nnkExprEqExpr).add(ch[0], ch[1]))
|
||||
for ch in obj.children:
|
||||
if ch.kind == nnkExprColonExpr:
|
||||
# some_arg = some_val
|
||||
args.add(newNimNode(nnkExprEqExpr).add(ch[0], ch[1]))
|
||||
|
||||
# init(self=init_obj, some_arg=someval, ...)
|
||||
var init_call = newCall(ident"init", args)
|
||||
# init(self=init_obj, some_arg=someval, ...)
|
||||
var init_call = newCall(ident"init", args)
|
||||
|
||||
template new_object(symbol, obj_node, init_func)=
|
||||
var symbol = obj_node
|
||||
when compiles(init_func):
|
||||
init_func
|
||||
symbol
|
||||
template new_object(symbol, obj_node, init_func)=
|
||||
var symbol = obj_node
|
||||
when compiles(init_func):
|
||||
init_func
|
||||
symbol
|
||||
|
||||
result = getAst(new_object(sym, new_obj, init_call))
|
||||
else:
|
||||
# otherwise, just call system.new on the object
|
||||
# since we don't care about it
|
||||
template new_object(obj)=
|
||||
var init_obj = obj
|
||||
system.new(init_obj)
|
||||
result = getAst(new_object(obj))
|
||||
result = getAst(new_object(sym, new_obj, init_call))
|
||||
else:
|
||||
# otherwise, just call system.new on the object
|
||||
# since we don't care about it
|
||||
template new_object(obj)=
|
||||
var init_obj = obj
|
||||
system.new(init_obj)
|
||||
result = getAst(new_object(obj))
|
||||
|
||||
|
||||
macro class*(head: untyped, body: untyped): untyped =
|
||||
|
||||
# object reference name inside methods.
|
||||
# ie: self, self
|
||||
let obj_reference = "self"
|
||||
var export_class: bool = false
|
||||
|
||||
# object reference name inside methods.
|
||||
# ie: self, self
|
||||
let obj_reference = "self"
|
||||
var export_class: bool = false
|
||||
|
||||
var typeName, baseName: NimNode
|
||||
var typeName, baseName: NimNode
|
||||
|
||||
if head.kind == nnkIdent:
|
||||
# `head` is expression `typeName`
|
||||
# echo head.treeRepr
|
||||
# --------------------
|
||||
# Ident !"Animal"
|
||||
typeName = head
|
||||
|
||||
elif head.kind == nnkInfix and $head[0] == "of":
|
||||
# `head` is expression `typeName of baseClass`
|
||||
# echo head.treeRepr
|
||||
# --------------------
|
||||
# Infix
|
||||
# Ident !"of"
|
||||
# Ident !"Animal"
|
||||
# Ident !"RootObj"
|
||||
typeName = head[1]
|
||||
baseName = head[2]
|
||||
|
||||
elif head.kind == nnkInfix and $head[0] == "*" and $head[1] == "of":
|
||||
# echo head.treeRepr
|
||||
# -----------
|
||||
# Infix
|
||||
# Ident !"*"
|
||||
# Ident !"Animal
|
||||
# Prefix
|
||||
# Ident !"of"
|
||||
# Ident !"RootObj"
|
||||
export_class = true
|
||||
typeName = head[1]
|
||||
baseName = head[2][1]
|
||||
elif head.kind == nnkInfix and $head[0] == "*":
|
||||
export_class = true
|
||||
typeName = head[1]
|
||||
else:
|
||||
quit "Invalid node: " & head.lispRepr
|
||||
|
||||
# echo treeRepr(body)
|
||||
if head.kind == nnkIdent:
|
||||
# `head` is expression `typeName`
|
||||
# echo head.treeRepr
|
||||
# --------------------
|
||||
# StmtList
|
||||
# VarSection
|
||||
# IdentDefs
|
||||
# Ident !"name"
|
||||
# Ident !"string"
|
||||
# Empty
|
||||
# IdentDefs
|
||||
# Ident !"age"
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
# MethodDef
|
||||
# Ident !"vocalize"
|
||||
# Empty
|
||||
# Empty
|
||||
# FormalParams
|
||||
# Ident !"string"
|
||||
# Empty
|
||||
# Empty
|
||||
# StmtList
|
||||
# StrLit ...
|
||||
# MethodDef
|
||||
# Ident !"age_human_yrs"
|
||||
# Empty
|
||||
# Empty
|
||||
# FormalParams
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
# Empty
|
||||
# StmtList
|
||||
# DotExpr
|
||||
# Ident !"self"
|
||||
# Ident !"age"
|
||||
# Ident !"Animal"
|
||||
typeName = head
|
||||
|
||||
# create a new stmtList for the result
|
||||
result = newStmtList()
|
||||
|
||||
# var declarations will be turned into object fields
|
||||
var recList = newNimNode(nnkRecList)
|
||||
|
||||
# add a super function to simulate OOP
|
||||
# inheritance tree (Doesn't do what is expected because of dynamic binding)
|
||||
#if not isNil(`baseName`):
|
||||
# var super = quote do:
|
||||
# proc super(self: `typeName`): `baseName`=
|
||||
# return `baseName`(self)
|
||||
# result.add(super)
|
||||
|
||||
template set_node_name(n2, proc_name, type_name)=
|
||||
if n2.name.kind == nnkIdent:
|
||||
proc_name = $(n2.name.toStrLit())
|
||||
n2.name = ident(proc_name & type_name)
|
||||
elif n2.name.kind == nnkPostFix:
|
||||
if n2.name[1].kind == nnkIdent:
|
||||
proc_name = $(n2.name[1].toStrLit())
|
||||
n2.name[1] = ident(proc_name & type_name)
|
||||
elif n2.name[1].kind == nnkAccQuoted:
|
||||
proc_name = $(n2.name[1][0].toStrLit())
|
||||
n2.name[1][0] = ident(proc_name & type_name)
|
||||
elif n2.name.kind == nnkAccQuoted:
|
||||
proc_name = $(n2.name[0].toStrLit())
|
||||
n2.name[0] = ident(proc_name & type_name)
|
||||
result.add(n2)
|
||||
|
||||
# Make forward declarations so that function order
|
||||
# does not matter, just like in real OOP!
|
||||
for node in body.children:
|
||||
case node.kind:
|
||||
of nnkMethodDef, nnkProcDef:
|
||||
# inject `self: T` into the arguments
|
||||
let n = copyNimTree(node)
|
||||
n.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
# clear the body so we only get a
|
||||
# declaration
|
||||
n.body = newEmptyNode()
|
||||
result.add(n)
|
||||
|
||||
# forward declare the inheritable method
|
||||
let n2 = copyNimTree(n)
|
||||
let type_name = $(typeName.toStrLit())
|
||||
var proc_name = ""
|
||||
|
||||
set_node_name(n2, proc_name, type_name)
|
||||
else:
|
||||
discard
|
||||
|
||||
# Iterate over the statements, adding `self: T`
|
||||
# to the parameters of functions
|
||||
for node in body.children:
|
||||
case node.kind:
|
||||
of nnkMethodDef, nnkProcDef:
|
||||
# inject `self: T` into the arguments
|
||||
let n = copyNimTree(node)
|
||||
n.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
|
||||
# Copy the proc or method for inheritance
|
||||
# ie: procName_ClassName()
|
||||
let n2 = copyNimTree(node)
|
||||
n2.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
|
||||
let type_name = $(typeName.toStrLit())
|
||||
var proc_name = $(n2.name.toStrLit())
|
||||
var is_assignment = proc_name.contains("=")
|
||||
|
||||
set_node_name(n2, proc_name, type_name)
|
||||
|
||||
# simply call the class method from here
|
||||
# proc procName=
|
||||
# procName_ClassName()
|
||||
var p: seq[NimNode] = @[]
|
||||
for i in 1..n.params.len-1:
|
||||
p.add(n.params[i][0])
|
||||
if is_assignment:
|
||||
let dot = newDotExpr(ident(obj_reference), ident(proc_name & type_name))
|
||||
n.body = newStmtList(newAssignment(dot, p[1]))
|
||||
else:
|
||||
n.body = newStmtList(newCall(proc_name & type_name, p))
|
||||
|
||||
result.add(n)
|
||||
|
||||
of nnkVarSection:
|
||||
# variables get turned into fields of the type.
|
||||
for n in node.children:
|
||||
recList.add(n)
|
||||
else:
|
||||
result.add(node)
|
||||
|
||||
# The following prints out the AST structure:
|
||||
#
|
||||
# import macros
|
||||
# dumptree:
|
||||
# type X = ref object of Y
|
||||
# z: int
|
||||
elif head.kind == nnkInfix and $head[0] == "of":
|
||||
# `head` is expression `typeName of baseClass`
|
||||
# echo head.treeRepr
|
||||
# --------------------
|
||||
# TypeSection
|
||||
# TypeDef
|
||||
# Ident !"X"
|
||||
# Empty
|
||||
# RefTy
|
||||
# ObjectTy
|
||||
# Empty
|
||||
# OfInherit
|
||||
# Ident !"Y"
|
||||
# RecList
|
||||
# IdentDefs
|
||||
# Ident !"z"
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
# Infix
|
||||
# Ident !"of"
|
||||
# Ident !"Animal"
|
||||
# Ident !"RootObj"
|
||||
typeName = head[1]
|
||||
baseName = head[2]
|
||||
|
||||
var type_decl: NimNode
|
||||
elif head.kind == nnkInfix and $head[0] == "*" and $head[1] == "of":
|
||||
# echo head.treeRepr
|
||||
# -----------
|
||||
# Infix
|
||||
# Ident !"*"
|
||||
# Ident !"Animal
|
||||
# Prefix
|
||||
# Ident !"of"
|
||||
# Ident !"RootObj"
|
||||
export_class = true
|
||||
typeName = head[1]
|
||||
baseName = head[2][1]
|
||||
elif head.kind == nnkInfix and $head[0] == "*":
|
||||
export_class = true
|
||||
typeName = head[1]
|
||||
else:
|
||||
quit "Invalid node: " & head.lispRepr
|
||||
|
||||
template declare_type_export(tname, bname)=
|
||||
type tname* = ref object of bname
|
||||
template declare_type(tname, bname)=
|
||||
type tname = ref object of bname
|
||||
# echo treeRepr(body)
|
||||
# --------------------
|
||||
# StmtList
|
||||
# VarSection
|
||||
# IdentDefs
|
||||
# Ident !"name"
|
||||
# Ident !"string"
|
||||
# Empty
|
||||
# IdentDefs
|
||||
# Ident !"age"
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
# MethodDef
|
||||
# Ident !"vocalize"
|
||||
# Empty
|
||||
# Empty
|
||||
# FormalParams
|
||||
# Ident !"string"
|
||||
# Empty
|
||||
# Empty
|
||||
# StmtList
|
||||
# StrLit ...
|
||||
# MethodDef
|
||||
# Ident !"age_human_yrs"
|
||||
# Empty
|
||||
# Empty
|
||||
# FormalParams
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
# Empty
|
||||
# StmtList
|
||||
# DotExpr
|
||||
# Ident !"self"
|
||||
# Ident !"age"
|
||||
|
||||
if baseName == nil:
|
||||
if export_class:
|
||||
type_decl = getAst(declare_type_export(typeName, RootObj))
|
||||
# create a new stmtList for the result
|
||||
result = newStmtList()
|
||||
|
||||
# var declarations will be turned into object fields
|
||||
var recList = newNimNode(nnkRecList)
|
||||
|
||||
# add a super function to simulate OOP
|
||||
# inheritance tree (Doesn't do what is expected because of dynamic binding)
|
||||
#if not isNil(`baseName`):
|
||||
# var super = quote do:
|
||||
# proc super(self: `typeName`): `baseName`=
|
||||
# return `baseName`(self)
|
||||
# result.add(super)
|
||||
|
||||
template set_node_name(n2, proc_name, type_name)=
|
||||
if n2.name.kind == nnkIdent:
|
||||
proc_name = $(n2.name.toStrLit())
|
||||
n2.name = ident(proc_name & type_name)
|
||||
elif n2.name.kind == nnkPostFix:
|
||||
if n2.name[1].kind == nnkIdent:
|
||||
proc_name = $(n2.name[1].toStrLit())
|
||||
n2.name[1] = ident(proc_name & type_name)
|
||||
elif n2.name[1].kind == nnkAccQuoted:
|
||||
proc_name = $(n2.name[1][0].toStrLit())
|
||||
n2.name[1][0] = ident(proc_name & type_name)
|
||||
elif n2.name.kind == nnkAccQuoted:
|
||||
proc_name = $(n2.name[0].toStrLit())
|
||||
n2.name[0] = ident(proc_name & type_name)
|
||||
result.add(n2)
|
||||
|
||||
# Make forward declarations so that function order
|
||||
# does not matter, just like in real OOP!
|
||||
for node in body.children:
|
||||
case node.kind:
|
||||
of nnkMethodDef, nnkProcDef:
|
||||
# inject `self: T` into the arguments
|
||||
let n = copyNimTree(node)
|
||||
n.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
# clear the body so we only get a
|
||||
# declaration
|
||||
n.body = newEmptyNode()
|
||||
result.add(n)
|
||||
|
||||
# forward declare the inheritable method
|
||||
let n2 = copyNimTree(n)
|
||||
let type_name = $(typeName.toStrLit())
|
||||
var proc_name = ""
|
||||
|
||||
set_node_name(n2, proc_name, type_name)
|
||||
else:
|
||||
discard
|
||||
|
||||
# Iterate over the statements, adding `self: T`
|
||||
# to the parameters of functions
|
||||
for node in body.children:
|
||||
case node.kind:
|
||||
of nnkMethodDef, nnkProcDef:
|
||||
# inject `self: T` into the arguments
|
||||
let n = copyNimTree(node)
|
||||
n.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
|
||||
# Copy the proc or method for inheritance
|
||||
# ie: procName_ClassName()
|
||||
let n2 = copyNimTree(node)
|
||||
n2.params.insert(1, newIdentDefs(ident(obj_reference), typeName))
|
||||
|
||||
let type_name = $(typeName.toStrLit())
|
||||
var proc_name = $(n2.name.toStrLit())
|
||||
var is_assignment = proc_name.contains("=")
|
||||
|
||||
set_node_name(n2, proc_name, type_name)
|
||||
|
||||
# simply call the class method from here
|
||||
# proc procName=
|
||||
# procName_ClassName()
|
||||
var p: seq[NimNode] = @[]
|
||||
for i in 1..n.params.len-1:
|
||||
p.add(n.params[i][0])
|
||||
if is_assignment:
|
||||
let dot = newDotExpr(ident(obj_reference), ident(proc_name & type_name))
|
||||
n.body = newStmtList(newAssignment(dot, p[1]))
|
||||
else:
|
||||
type_decl = getAst(declare_type(typeName, RootObj))
|
||||
else:
|
||||
if export_class:
|
||||
type_decl = getAst(declare_type_export(typeName, baseName))
|
||||
else:
|
||||
type_decl = getAst(declare_type(typeName, baseName))
|
||||
n.body = newStmtList(newCall(proc_name & type_name, p))
|
||||
|
||||
# Inspect the tree structure:
|
||||
#
|
||||
# echo type_decl.treeRepr
|
||||
# --------------------
|
||||
# StmtList
|
||||
# TypeSection
|
||||
# TypeDef
|
||||
# Ident !"Animal"
|
||||
# Empty
|
||||
# RefTy
|
||||
# ObjectTy
|
||||
# Empty
|
||||
# OfInherit
|
||||
# Ident !"RootObj"
|
||||
# Empty <= We want to replace self
|
||||
type_decl[0][0][2][0][2] = recList
|
||||
result.insert(0, type_decl)
|
||||
result.add(n)
|
||||
|
||||
of nnkVarSection:
|
||||
# variables get turned into fields of the type.
|
||||
for n in node.children:
|
||||
recList.add(n)
|
||||
else:
|
||||
result.add(node)
|
||||
|
||||
# The following prints out the AST structure:
|
||||
#
|
||||
# import macros
|
||||
# dumptree:
|
||||
# type X = ref object of Y
|
||||
# z: int
|
||||
# --------------------
|
||||
# TypeSection
|
||||
# TypeDef
|
||||
# Ident !"X"
|
||||
# Empty
|
||||
# RefTy
|
||||
# ObjectTy
|
||||
# Empty
|
||||
# OfInherit
|
||||
# Ident !"Y"
|
||||
# RecList
|
||||
# IdentDefs
|
||||
# Ident !"z"
|
||||
# Ident !"int"
|
||||
# Empty
|
||||
|
||||
var type_decl: NimNode
|
||||
|
||||
template declare_type_export(tname, bname)=
|
||||
type tname* = ref object of bname
|
||||
template declare_type(tname, bname)=
|
||||
type tname = ref object of bname
|
||||
|
||||
if baseName == nil:
|
||||
if export_class:
|
||||
type_decl = getAst(declare_type_export(typeName, RootObj))
|
||||
else:
|
||||
type_decl = getAst(declare_type(typeName, RootObj))
|
||||
else:
|
||||
if export_class:
|
||||
type_decl = getAst(declare_type_export(typeName, baseName))
|
||||
else:
|
||||
type_decl = getAst(declare_type(typeName, baseName))
|
||||
|
||||
# Inspect the tree structure:
|
||||
#
|
||||
# echo type_decl.treeRepr
|
||||
# --------------------
|
||||
# StmtList
|
||||
# TypeSection
|
||||
# TypeDef
|
||||
# Ident !"Animal"
|
||||
# Empty
|
||||
# RefTy
|
||||
# ObjectTy
|
||||
# Empty
|
||||
# OfInherit
|
||||
# Ident !"RootObj"
|
||||
# Empty <= We want to replace self
|
||||
type_decl[0][0][2][0][2] = recList
|
||||
result.insert(0, type_decl)
|
||||
|
||||
|
||||
class Animal of RootObj:
|
||||
var
|
||||
name: string
|
||||
age: int
|
||||
var
|
||||
name: string
|
||||
age: int
|
||||
|
||||
method init*(name: string, age: int)=
|
||||
self.name = name
|
||||
self.age = age
|
||||
echo "I am a new Animal, ", self.name
|
||||
method init*(name: string, age: int)=
|
||||
self.name = name
|
||||
self.age = age
|
||||
echo "I am a new Animal, ", self.name
|
||||
|
||||
method stuff(s:string): string = s
|
||||
method vocalize: string = "..."
|
||||
method age_human_yrs: int = self.age # `self` is injected
|
||||
method stuff(s:string): string = s
|
||||
method vocalize: string = "..."
|
||||
method age_human_yrs: int = self.age # `self` is injected
|
||||
|
||||
class Dog of Animal:
|
||||
method vocalize: string = "woof"
|
||||
method age_human_yrs: int = self.age * 7
|
||||
method vocalize: string = "woof"
|
||||
method age_human_yrs: int = self.age * 7
|
||||
|
||||
class Cat of Animal:
|
||||
method vocalize: string =
|
||||
# call the base class method
|
||||
self.vocalize_animal() & "meow"
|
||||
method vocalize: string =
|
||||
# call the base class method
|
||||
self.vocalize_animal() & "meow"
|
||||
|
||||
class Tiger of Cat:
|
||||
method init(name: string="Bob", age: int)=
|
||||
self.init_animal(name, age)
|
||||
echo "I am a new tiger"
|
||||
method vocalize: string =
|
||||
# no need for super.super!
|
||||
self.vocalize_animal() & "Rawr!"
|
||||
method init(name: string="Bob", age: int)=
|
||||
self.init_animal(name, age)
|
||||
echo "I am a new tiger"
|
||||
method vocalize: string =
|
||||
# no need for super.super!
|
||||
self.vocalize_animal() & "Rawr!"
|
||||
|
||||
if isMainModule:
|
||||
var animals: seq[Animal] = @[]
|
||||
animals.add(new Dog(name: "Sparky", age: 10))
|
||||
animals.add(new Cat(name: "Mitten", age: 10))
|
||||
animals.add(new Tiger(name: "Jean", age: 2))
|
||||
var animals: seq[Animal] = @[]
|
||||
animals.add(new Dog(name: "Sparky", age: 10))
|
||||
animals.add(new Cat(name: "Mitten", age: 10))
|
||||
animals.add(new Tiger(name: "Jean", age: 2))
|
||||
|
||||
for a in animals:
|
||||
echo a.name, " says ", a.vocalize()
|
||||
echo a.age_human_yrs()
|
||||
for a in animals:
|
||||
echo a.name, " says ", a.vocalize()
|
||||
echo a.age_human_yrs()
|
||||
|
||||
# prints:
|
||||
# I am a new Animal, Sparky
|
||||
# I am a new Animal, Mitten
|
||||
# I am a new Animal, Jean
|
||||
# I am a new tiger
|
||||
# Sparky says woof
|
||||
# 70
|
||||
# Mitten says ...meow
|
||||
# 10
|
||||
# Jean says ...Rawr!
|
||||
# 2
|
||||
# prints:
|
||||
# I am a new Animal, Sparky
|
||||
# I am a new Animal, Mitten
|
||||
# I am a new Animal, Jean
|
||||
# I am a new tiger
|
||||
# Sparky says woof
|
||||
# 70
|
||||
# Mitten says ...meow
|
||||
# 10
|
||||
# Jean says ...Rawr!
|
||||
# 2
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue