print('--- Empty functions') func f() end print(f()) print('--- While loop') a = 1 while a < 10 do print(a) a += 1 end print("--- Range") func mul9(b) for i in range(#b) do b[i] *= 9 end end l = [1 2 3 4 5] print(range(#l)) print("--- Iter list") for i in l do print(i) end print("--- Index list") for i in range(#l) do print(i, l[i]) end print('--- mul9') mul9(l) for i in l do print(i) end print('--- Iter mul9') for i in range(#l) do print(i, l[i]) end print("--- Mapgen") PI = 3.14159265358979 d = { integer = 1 string = "hello" submap = { stamina = 100 health = 42.0 } sublist = [1 1 2 3 5 8], ["this time with spaces"] = PI*2, [1 + 5] = 1 + 5, [1 + 9] = 1 + 9 } print(d) print('--- Map iter') for i in d do print(i, d[i]) end print('--- try') func bad(x) print(x) return x.c() end test1 = {c = func() return 15 end} test2 = {} print(try(bad, test1)) print(try(bad, test2)) print(bad(test1)) --print(bad(test2)) print('--- Induced errors') func raise(x) error(x) end print(try(raise, "lp0 on fire")) print('--- Indexing') print(d["integer"]) print(d.integer) d.integer += 5 print(d.integer) print('--- Function binding') func outer(a) func inner(b) return a+b end inner.closure.a = a return inner end i = outer(5) print(i(3), i(4), i(5)) j = outer(8) print(j(3), j(4), j(5)) print('--- Iterators') func myiter() if i>10 then return StopIteration end i+=1 return i-1 end myiter.closure.i = 1 for i in myiter do print(i) end print('--- Method calls') d = {a = func(a, b) print(a, b) end} d.a(1, 2) d:a(3) print('--- Special methods') d = {__index = func(obj, key) print('Index', obj, key) return key end, __setindex = func(obj, key, val) print('SetIndex', obj, key, val) end, __call = func(obj, arg1, arg2) print('Call', obj, arg1, arg2) return arg1 end} print(d[3], d[5]) d.a = 7 print(d("q", "r")) e = {a=1, b=2} d = {__index = e, __setindex = e} print(d, d.a, d.b) d.c = 5 d.b = 7 print(d, e) print('--- Data sharing') d = {} e = [1 2 3 4 5] d.a = e d.b = e print(d) e[2]=7 e[3]="c" print(d) d.a:insert(#(d.a), "q") d.b:remove(1) d.a[3]="f" print(d) print('--- Arithmetic structure operations') print('ab'+'cd') print('l'+'ol'*32) print([1 2 3]+[4 5]) print([1 2 3]*5) print({a=1 b=2}+{c=3}) print('--- Map/filter') l = [1 2 3 4 5] print(l) l:map(func (i) return i*3 end) print(l) l:filter(func (i) return i & 1 end) print(l) print('--- Map/filter chain') print([1 2 3 4 5]:map(func (i) return i * 3 end):filter(func (i) return i & 1 end)) print('--- Exec/eval') exec('print("Hello from exec!")') print(eval('5 + 3')) execfile('programs/subtest.sol') print('--- Modulus') print(5%3) print(13%5) print(15%15) print('--- Special function manipulation') func foo(x) return x end print(foo) foo.name = "bar" print(foo) func something() return i end something.closure = {i=[1, 2, 3]} print(something()) cl = something.closure cl.i:insert(0, "b") print(something()) print('--- Function body swapping') func a() return 0 end func b() return 2 end print(a, a()) print(b, b()) print(a.stmt) print(b.stmt) temp = a.stmt a.stmt = b.stmt b.stmt = temp print(a, a()) print(b, b()) newbody = parse('return 4') print(newbody) a.stmt = newbody b.stmt = newbody print(a, a()) print(b, b()) print('--- More complicated ASTs') print(outer, outer.stmt, outer.stmt.stmtlist) print('--- Exec- and eval-by-parse') parse('print("Hello from parse()!")')() print(parse('5 + 3').stmtlist[0].expr()) print('--- Mutating ASTs') func f() return 5 + 7 end print(f, f.stmt, f()) f.stmt.stmtlist[0].ret.right.ival = 11 print(f, f.stmt, f()) func g() a=1 b=2 print("a=", a, ", b=", b) end print(g, g.stmt, g()) g.stmt.stmtlist[1].expr.value = parse('a').stmtlist[0].expr print(g, g.stmt, g()) print('--- AST Environments') code = parse('print("a is", a, "and b is", b"); a = 4; b = 5') print(code) code() d = {a=1, b=2} print(d) code(d) print(d) e = {a="hello", b=["world"]} print(e) code(e) print(e) e = {a=1, b=2} d = {__index = e} print(d) print(e) code(d) print(d) print(e) print('--- Basic buffers') print('(buffer.fromstring = ', buffer.fromstring, ')') b = buffer.fromstring("Hello, world!") print(b) print('(b.get = ', b.get, ')') print(b:get(buffer.type.cstr)) b:set(buffer.type.char, "Q") b:set(buffer.type.char, "L", 2) print(b:get(buffer.type.cstr)) print(b:get(buffer.type.cstr, 5)) print(b:get(buffer.type.uint32)) --b:set(buffer.type.double, 1.243416560929) b:set(buffer.type.uint32, 1886545252) print(b:get(buffer.type.uint32)) print(b:get(buffer.type.cstr)) q = buffer.fromaddress(b:address(), b:size()) print(q:get(buffer.type.cstr)) q:set(buffer.type.cstr, "Goodbye!") print(q:get(buffer.type.cstr), b:get(buffer.type.cstr)) s = "A string!" b = buffer.fromobject(s) prepr(s) print('...is a', buffer.objtype[b:get(buffer.type.int, 0)]) print('(buffer.sizeof.ptr = ', buffer.sizeof.ptr, ')') print('(buffer.sizeof.int = ', buffer.sizeof.int, ')') print('(buffer.sizeof.int*2 = ', buffer.sizeof.int*2, ')') print('(buffer.sizeof.int*2 + buffer.sizeof.ptr = ', buffer.sizeof.int*2 + (buffer.sizeof.ptr), ')') bs = b:get(buffer.type.ptr, buffer.sizeof.int*2 + (buffer.sizeof.ptr)) print('...string buffer:', bs) print('...with value:', bs:get(buffer.type.cstr)) print('--- IO redirection') oldstdout = io.stdout io.stdout = io.open('stdout', io.MODE_WRITE|io.MODE_TRUNCATE) print('A line!') print('An object:', {a=1, b=2, c="turkey"}) print('Something mysterious :o') io.stdout:write('Writing directly to a file :D') io.stdout:flush() io.stdout = oldstdout print('...restored stdout.') f = io.open('stdout', io.MODE_READ) s = f:read(io.ALL) print('Buffered output was:') prepr(s) f = None print('...second time.') io.stdout = io.open('stdout', io.MODE_WRITE|io.MODE_TRUNCATE) print('Hey there!') print('l'+'ol'*32) io.stdout:flush() io.stdout = oldstdout print('...restored.') print('Output was:') prepr(io.open('stdout', io.MODE_READ):read(io.ALL)) print('--- Substrings') s = 'This is a test!' prepr(s) prepr(s:sub(1, -1)) prepr(s:sub(3, -3)) prepr(s:sub(3, 5)) prepr(s:sub(3, 11)) prepr(s:sub(-1000, -1000)) print('--- Splitting') s = 'This is a test!' prepr(s) prepr(s:split(' ')) prepr(s:split('i')) prepr(s:split('0')) prepr(s:split('aeiou')) l = s:split(' ') for i in l do prepr(i, type(i)) end print('--- Continue/break') l = range(10) for i in l do print(i) if i >= 5 then break end end print('---') for i in l do if i%2 == 0 then continue end print(i) end print('--- Control expressions') print('1 is', if 1 then 'true' else 'false' end) print('0 is', if 0 then 'true' else 'false' end) print(for i in l do continue 1000 + 2 * i end) print(for i in l do print('(', i, ')') if i >= 5 then break i end end) print('--- All done!')