(in-package #:clnl-transpiler)
(defparameter *prims* nil)
-(defparameter *prim-aliases* nil)
(defvar *local-variables* nil)
+(defvar *dynamic-prims* nil)
(defun prim-name (prim) (getf prim :name))
(defun prim-type (prim) (getf prim :type))
(defun prim-func (prim) (getf prim :func))
-(defun is-reporter (prim) (eql :reporter (getf prim :type)))
-(defun is-command (prim) (eql :command (getf prim :type)))
+(defun prim-reporter-p (prim) (eql :reporter (getf prim :type)))
+(defun prim-command-p (prim) (eql :command (getf prim :type)))
(defun find-prim (symb)
(when symb
- (or
- (find symb *prims* :key #'prim-name)
- (find-prim (getf (find symb *prim-aliases* :key #'prim-name) :real-symb)))))
+ (find-if
+ (lambda (prim-name) (or (eql symb prim-name) (and (listp prim-name) (find symb prim-name))))
+ (append *prims* *dynamic-prims*)
+ :key #'prim-name)))
-; Let this grow, slowly but surely, eventually taking on calling context, etc.
-; For now, it's just a
-(defun transpile-commands (parsed-ast)
- "TRANSPILE-COMMANDS PARSED-AST => AST
+(defun transpile (parsed-ast &optional dynamic-prims)
+ "TRANSPILE PARSED-AST &optional DYNAMIC-PRIMS => AST
+
+ DYNAMIC-PRIMS: DYNAMIC-PRIM*
+ DYNAMIC-PRIM: (:name NAME :type TYPE :macro MACRO :func FUNC)
+ TYPE: :reporter | :command
ARGUMENTS AND VALUES:
PARSED-AST: An ast as returned by the parser
AST: An common lisp AST that can be actually run in a common lisp instance
+ NAME: A symbol in the keyword package
+ MACRO: A macro that will be called with the arguments ast
+ FUNC: A function that will be called with the transpiled arguments
DESCRIPTION:
- TRANSPILE-COMMANDS takes a unambigious PARSED-AST and converts it to
- Common Lisp code.
+ TRANSPILE takes a unambigious PARSED-AST and converts it to
+ Common Lisp code. The PARSED-AST must be either a list of commands,
+ or a single reporter.
+
+ When a set of DYNAMIC-PRIMS is included, external language constructs
+ can be also transpiled. The provided functions will be inserted into
+ the returned AST with a call to FUNCALL. If :macro is included, instead
+ of having a call to FUNCALL provided, the macro will be run at netlogo
+ transpile time, with the arguments it should have specified to the
+ parser. The result of that function call will then be dropped into
+ the ast.
Calling eval on that code should work correctly as long as you have a
- running engine. This is the entry point for commands, so it does
- extra checking to ensure that commands are actually in the PARSED-AST."
+ running engine."
+ (let
+ ((*dynamic-prims*
+ (mapcar
+ (lambda (prim)
+ (if (getf prim :macro)
+ (append (list :func (getf prim :macro)) prim)
+ (append (list :func (lambda (&rest args) `(funcall ,(getf prim :func) ,@args))) prim)))
+ dynamic-prims)))
+ (let
+ ((deoptionalized-ast (deoptionalize parsed-ast)))
+ (cond
+ ((command-list-p deoptionalized-ast) (transpile-commands deoptionalized-ast))
+ ((and (listp deoptionalized-ast) (= 1 (length deoptionalized-ast)) (reporter-p (car deoptionalized-ast)))
+ (transpile-reporter (car deoptionalized-ast)))
+ (t (error "Is neither a list of commands nor a reporter: ~S" deoptionalized-ast))))))
+
+(defun command-list-p (parsed-ast)
+ "COMMAND-LIST-P PARSED-AST => RESULT
+
+ARGUMENTS AND VALUES:
+
+ PARSED-AST: An ast as returned by the parser
+ RESULT: A boolean
+
+DESCRIPTION:
+
+ COMMAND-LIST-P returns whether the parsed-ast is a valid list
+ of commands."
+ (every #'command-p parsed-ast))
+
+(defun command-p (parsed-ast)
+ (and
+ (listp parsed-ast)
+ (prim-command-p (find-prim (car parsed-ast)))))
+
+(defun reporter-p (parsed-ast)
+ "REPORTER-P PARSED-AST => RESULT
+
+ARGUMENTS AND VALUES:
+
+ PARSED-AST: An ast as returned by the parser
+ RESULT: A boolean
+
+DESCRIPTION:
+
+ REPORTER-P returns whether the parsed-ast is a valid reporter."
+ (and
+ (listp parsed-ast)
+ (symbolp (car parsed-ast))
+ (prim-reporter-p (find-prim (car parsed-ast)))))
+
+; Let this grow, slowly but surely, eventually taking on calling context, etc.
+; For now, it's just a
+(defun transpile-commands (parsed-ast)
`(progn
,@(transpile-commands-inner parsed-ast)))
+; This makes some assumptions about the precedence of optionality
+; Consider the following:
+; - command-1 <optional> <optional>
+; - reporter-1 <optional>
+;
+; And consider:
+; - command-1 reporter-1 "foo"
+;
+; Does the "foo" belong to command-1 or reporter-1?
+;
+; I'm not sure how netlogo answers this at this time, but for the purposes
+; of this deoptionalizer, "foo" belongs to reporter-1. In the case that the
+; language tests treat this as incorrect, I'll fix it. I'm not sure if this
+; case exists in the base netlogo language, and I don't feel like creating
+; a specific test case at this time to find out.
+(defun deoptionalize (parsed-ast)
+ (let
+ ((first-thing (car parsed-ast)))
+ (cond
+ ((not parsed-ast) nil)
+ ((not (listp first-thing)) (cons first-thing (deoptionalize (cdr parsed-ast))))
+ ((and
+ (find :optional first-thing)
+ (cadr parsed-ast)
+ (not (command-p (cadr parsed-ast))))
+ (let
+ ((deoptionalized-rest (deoptionalize (cdr parsed-ast)))
+ (optional-pos (position :optional first-thing)))
+ (deoptionalize
+ (cons
+ (append
+ (subseq first-thing 0 optional-pos)
+ (list (car deoptionalized-rest))
+ (subseq first-thing (1+ optional-pos)))
+ (cdr deoptionalized-rest)))))
+ ((find :optional first-thing)
+ (deoptionalize
+ (cons
+ (subseq first-thing 0 (position :optional first-thing))
+ (cdr parsed-ast))))
+ (t
+ (cons
+ (deoptionalize first-thing)
+ (deoptionalize (cdr parsed-ast)))))))
+
(defun transpile-commands-inner (parsed-ast)
(cond
((not parsed-ast) nil)
(cond
((not (listp command)) (error "Expected a statement of some sort"))
((not (find-prim (car command))) (error "Couldn't find the command for ~S" (car command)))
- ((not (is-command (find-prim (car command)))) (error "Expected command, got ~S" (car command)))
+ ((not (prim-command-p (find-prim (car command)))) (error "Expected command, got ~S" (car command)))
(t (apply (prim-func (find-prim (car command))) (mapcar #'transpile-reporter (cdr command))))))
(defun transpile-reporter (reporter)
- "TRANSPILE-REPORTER REPORTER => AST
-
-ARGUMENTS AND VALUES:
-
- REPORTER: An ast returned from the parser.
- AST: An common lisp AST that can be actually run in a common lisp instance
-
-DESCRIPTION:
-
- TRANSPILE-REPORTER takes a unambigious PARSED-AST and converts it to
- Common Lisp code.
-
- Calling eval on that code should work correctly as long as you have a
- running engine. This is the entry point for reporters, so it does
- extra checking to ensure that the reporter is actually in the REPORTER.
-
- The Common lisp code that is returned, when run, will return some value."
(cond
((numberp reporter) reporter) ; The parser converts to double for us
((stringp reporter) reporter)
((eql :command-block (car reporter)) (transpile-command-block reporter))
((eql :list-literal (car reporter)) (cons 'list (mapcar #'transpile-reporter (cdr reporter))))
((eql :reporter-block (car reporter)) (transpile-reporter-block reporter))
+ ((eql :token (car reporter)) (cadr reporter))
((and (symbolp (car reporter)) (find (car reporter) *local-variables*))
(intern (symbol-name (car reporter)) clnl:*model-package*))
((not (find-prim (car reporter))) (error "Couldn't find the reporter for ~S" (car reporter)))
- ((not (is-reporter (find-prim (car reporter)))) (error "Expected reporter, got ~S" (car reporter)))
+ ((not (prim-reporter-p (find-prim (car reporter)))) (error "Expected reporter, got ~S" (car reporter)))
(t (apply (prim-func (find-prim (car reporter))) (mapcar #'transpile-reporter (cdr reporter))))))
(defun transpile-command-block (block)
(defmacro defkeywordprim (name)
`(defprim ,name :reporter (lambda () ',name)))
-(defmacro defprim-alias (name real-symb)
- `(push (list :name ,name :real-symb ,real-symb) *prim-aliases*))
-
(defmacro defagentvalueprim (name)
`(defprim ,name :reporter (lambda () `(clnl-nvm:agent-value ,,name))))
(defprim :any? :reporter (lambda (agentset) `(> (clnl-nvm:count ,agentset) 0)))
(defsimpleprim :ask :command clnl-nvm:ask)
(defagentvalueprim :color)
+(defsimpleprim '(:clear-all :ca) :command clnl-nvm:clear-all)
(defsimpleprim :count :reporter clnl-nvm:count)
-(defsimpleprim :crt :command clnl-nvm:create-turtles)
+(defprim '(:crt :create-turtles) :command (lambda (num &optional fn) `(clnl-nvm:create-turtles ,num nil ,fn)))
(defsimpleprim :die :command clnl-nvm:die)
+(defsimpleprim :display :command clnl-nvm:display)
(defsimpleprim :fd :command clnl-nvm:forward)
(defsimpleprim :hatch :command clnl-nvm:hatch)
(defprim :if :command (lambda (pred a) `(when ,pred ,@(make-command-block-inline a))))
-(defprim :ifelse :command (lambda (pred a b)
- `(if ,pred
- ,@(make-command-block-inline a)
- ,@(make-command-block-inline b))))
+(defprim '(:ifelse :if-else)
+ :command (lambda (pred a b)
+ (let
+ ((then (make-command-block-inline a))
+ (else (make-command-block-inline b)))
+ `(if ,pred
+ ,@(if (= (length then) 1) then `((progn ,@then)))
+ ,@(if (= (length else) 1) else `((progn ,@else)))))))
-(defprim-alias :if-else :ifelse)
(defagentvalueprim :label)
(defagentvalueprim :label-color)
+(defsimpleprim :let :command nil)
(defsimpleprim :lt :command clnl-nvm:turn-left)
(defsimpleprim :not :reporter cl:not)
(defkeywordprim :nobody)
(defsimpleprim :random-float :reporter clnl-nvm:random-float)
(defsimpleprim :random-xcor :reporter clnl-nvm:random-xcor)
(defsimpleprim :random-ycor :reporter clnl-nvm:random-ycor)
+(defprim :round :reporter (lambda (n) `(ffloor (+ ,n 0.5d0))))
(defsimpleprim :rt :command clnl-nvm:turn-right)
(defsimpleprim :set :command cl:setf)
(defsimpleprim :set-default-shape :command clnl-nvm:set-default-shape)
(defsimpleprim :setxy :command clnl-nvm:setxy)
(defsimpleprim :show :command clnl-nvm:show)
+(defsimpleprim :stop :command clnl-nvm:stop)
(defagentvalueprim :size)
(defsimpleprim :tick :command clnl-nvm:tick)
(defsimpleprim :ticks :reporter clnl-nvm:ticks)
(defsimpleprim :turtles :reporter clnl-nvm:turtles)
+(defsimpleprim :turtles-here :reporter clnl-nvm:turtles-here)
(defagentvalueprim :who)
(defsimpleprim :with :reporter clnl-nvm:with)
(defcolorprim :brown)
(defcolorprim :green)
(defcolorprim :white)
+
+; Boleans
+(defprim :true :reporter (lambda () t))
+(defprim :false :reporter (lambda () nil))