def expand_list_rfn(ret, item):
if is_unquote(item):
ret = rt.conj(ret, rt.vector(rt.first(rt.next(item))))
elif is_unquote_splicing(item):
ret = rt.conj(ret, rt.first(rt.next(item)))
else:
ret = rt.conj(ret, rt.vector(SyntaxQuoteReader.syntax_quote(item)))
return ret
def expand_list_rfn(ret, item):
if is_unquote(item):
ret = rt.conj(ret, rt.vector(rt.first(rt.next(item))))
elif is_unquote_splicing(item):
ret = rt.conj(ret, rt.first(rt.next(item)))
else:
ret = rt.conj(ret, rt.vector(SyntaxQuoteReader.syntax_quote(item)))
return ret
def _reduce(self, f, init):
assert isinstance(self, FileList)
for dirpath, dirnames, filenames in os.walk(str(self._top)):
for dirname in dirnames:
init = f.invoke([init, rt.vector(rt.wrap(dirpath), KW_DIR, rt.wrap(dirname))])
if rt.reduced_QMARK_(init):
return rt.deref(init)
for filename in filenames:
init = f.invoke([init, rt.vector(rt.wrap(dirpath), KW_FILE, rt.wrap(filename))])
if rt.reduced_QMARK_(init):
return rt.deref(init)
return init
def compile_form(form, ctx):
if form is nil:
ctx.push_const(nil)
return
if rt.instance_QMARK_(rt.ISeq.deref(), form) and form is not nil:
return compile_cons(form, ctx)
if isinstance(form, numbers.Integer):
ctx.push_const(form)
return
if isinstance(form, symbol.Symbol):
name = form._str
loc = resolve_local(ctx, name)
if loc is None:
var = resolve_var(ctx, form)
if var is None:
var = NS_VAR.deref().intern_or_make(name)
ctx.push_const(var)
ctx.bytecode.append(code.DEREF_VAR)
return
loc.emit(ctx)
return
if isinstance(form, Bool) or form is nil:
ctx.push_const(form)
return
if isinstance(form, Keyword):
ctx.push_const(form)
return
if isinstance(form, PersistentVector):
vector_var = rt.vector()
size = rt.count(form).int_val()
#assert rt.count(form).int_val() == 0
ctx.push_const(code.intern_var(u"pixie.stdlib", u"vector"))
for x in range(size):
compile_form(rt.nth(form, numbers.Integer(x)), ctx)
ctx.bytecode.append(code.INVOKE)
ctx.bytecode.append(r_uint(size + 1))
ctx.sub_sp(size)
return
if rt.instance_QMARK_(rt.IMap.deref(), form):
compile_map_literal(form, ctx)
return
if isinstance(form, String):
ctx.push_const(form)
return
raise Exception("Can't compile ")
def invoke(self, args):
affirm(not self._used, u"Can only call a given stacklet handle once.")
affirm(len(args) == 1, u"Only one arg should be handed to a stacklet handle")
self._used = True
global_state._val = args[0]
new_h = StackletHandle(global_state._th.switch(self._stacklet_handle))
val = global_state._val
global_state._val = None
return rt.vector(new_h, val)
def invoke(self, args):
affirm(not self._used, u"Can only call a given stacklet handle once.")
affirm(len(args) == 1, u"Only one arg should be handed to a stacklet handle")
self._used = True
global_state._val = args[0]
new_h = StackletHandle(global_state._th.switch(self._stacklet_handle))
val = global_state._val
global_state._val = None
return rt.vector(new_h, val)
def split(a, b):
affirm(rt.count(b) > 0, u"separator can't be empty")
v = rt.vector()
for s in rstring.split(rt.name(a), rt.name(b)):
v = rt.conj(v, rt.wrap(s))
return v
def compile_form(form, ctx):
if form is nil:
ctx.push_const(nil)
return
if rt._satisfies_QMARK_(rt.ISeq.deref(), form) and form is not nil:
form = macroexpand(form)
return compile_cons(form, ctx)
if isinstance(form, numbers.Integer):
ctx.push_const(form)
return
if isinstance(form, numbers.BigInteger):
ctx.push_const(form)
return
if isinstance(form, numbers.Float):
ctx.push_const(form)
return
if isinstance(form, numbers.Ratio):
ctx.push_const(form)
return
if isinstance(form, symbol.Symbol):
name = rt.name(form)
ns = rt.namespace(form)
loc = resolve_local(ctx, name)
var = resolve_var(ctx, form)
if var is None and loc:
loc.emit(ctx)
return
if var and loc and ns is None:
loc.emit(ctx)
return
if var is None:
name = rt.name(form)
var = NS_VAR.deref().intern_or_make(name)
ctx.push_const(var)
meta = rt.meta(form)
if meta is not nil:
ctx.debug_points[len(ctx.bytecode)] = rt.interpreter_code_info(meta)
ctx.bytecode.append(code.DEREF_VAR)
return
if isinstance(form, Bool) or form is nil:
ctx.push_const(form)
return
if isinstance(form, Keyword):
ctx.push_const(form)
return
if isinstance(form, PersistentVector):
vector_var = rt.vector()
size = rt.count(form)
#assert rt.count(form).int_val() == 0
ctx.push_const(code.intern_var(u"pixie.stdlib", u"vector"))
for x in range(size):
compile_form(rt.nth(form, rt.wrap(x)), ctx)
ctx.bytecode.append(code.INVOKE)
ctx.bytecode.append(r_uint(size + 1))
ctx.sub_sp(size)
compile_meta(rt.meta(form), ctx)
return
if isinstance(form, PersistentHashSet):
compile_set_literal(form, ctx)
return
if rt._satisfies_QMARK_(rt.IMap.deref(), form):
compile_map_literal(form, ctx)
return
if isinstance(form, String):
ctx.push_const(form)
return
if isinstance(form, Character):
ctx.push_const(form)
return
raise Exception("Can't compile ")
def compile_form(form, ctx):
if form is nil:
ctx.push_const(nil)
return
if rt.satisfies_QMARK_(rt.ISeq.deref(), form) and form is not nil:
form = macroexpand(form)
return compile_cons(form, ctx)
if isinstance(form, numbers.Integer):
ctx.push_const(form)
return
if isinstance(form, numbers.BigInteger):
ctx.push_const(form)
return
if isinstance(form, numbers.Float):
ctx.push_const(form)
return
if isinstance(form, numbers.Ratio):
ctx.push_const(form)
return
if isinstance(form, symbol.Symbol):
name = rt.name(form)
ns = rt.namespace(form)
loc = resolve_local(ctx, name)
var = resolve_var(ctx, form)
if var is None and loc:
loc.emit(ctx)
return
if var and loc and ns is None:
loc.emit(ctx)
return
if var is None:
name = rt.name(form)
var = NS_VAR.deref().intern_or_make(name)
ctx.push_const(var)
meta = rt.meta(form)
if meta is not nil:
ctx.debug_points[len(
ctx.bytecode)] = rt.interpreter_code_info(meta)
ctx.bytecode.append(code.DEREF_VAR)
return
if isinstance(form, Bool) or form is nil:
ctx.push_const(form)
return
if isinstance(form, Keyword):
ctx.push_const(form)
return
if isinstance(form, PersistentVector):
vector_var = rt.vector()
size = rt.count(form)
#assert rt.count(form).int_val() == 0
ctx.push_const(code.intern_var(u"pixie.stdlib", u"vector"))
for x in range(size):
compile_form(rt.nth(form, rt.wrap(x)), ctx)
ctx.bytecode.append(code.INVOKE)
ctx.bytecode.append(r_uint(size + 1))
ctx.sub_sp(size)
compile_meta(rt.meta(form), ctx)
return
if isinstance(form, PersistentHashSet):
compile_set_literal(form, ctx)
return
if rt.satisfies_QMARK_(rt.IMap.deref(), form):
compile_map_literal(form, ctx)
return
if isinstance(form, String):
ctx.push_const(form)
return
if isinstance(form, Character):
ctx.push_const(form)
return
raise Exception("Can't compile ")
def ratio_write(obj):
assert isinstance(obj, Ratio)
return rt.vector(rt.wrap(obj.numerator()), rt.wrap(obj.denominator()))
def list_dir(self):
assert isinstance(self, Path)
result = rt.vector()
for item in os.listdir(str(self._path)):
result = rt.conj(result, rt.wrap(str(self._path) + "/" + str(item)))
return result
def list_dir(self):
assert isinstance(self, Path)
result = rt.vector()
for item in os.listdir(str(self._path)):
result = rt.conj(result, rt.wrap(str(self._path) + "/" + str(item)))
return result
def split(a, b):
affirm(rt.count(b) > 0, u"separator can't be empty")
v = rt.vector()
for s in rstring.split(rt.name(a), rt.name(b)):
v = rt.conj(v, rt.wrap(s))
return v
def ratio_write(obj):
assert isinstance(obj, Ratio)
return rt.vector(rt.wrap(obj.numerator()), rt.wrap(obj.denominator()))
