ywatds/tracks/conns.lisp

98 lines
3.1 KiB
Common Lisp

(in-package :tracks)
(defstruct trackside
(:pos (error "no coordinates specified") :type aux:v3d)
(:side (error "no side specified") :type (integer 0 15)))
(defmacro trackside (pos side)
(alexandria:once-only
(pos side)
`(if (and ,pos ,side) (make-trackside :pos ,pos :side ,side))))
(defmethod print-object ((obj trackside) s)
(with-accessors ((p trackside-pos) (side trackside-side)) obj
(print-unreadable-object (obj s)
(format s "~a/~a" p (aref *dirnames* side)))))
(defmethod json:encode-json ((obj trackside) &optional json:*json-output*)
(with-accessors ((p trackside-pos) (side trackside-side)) obj
(json:encode-json (list (cons "p" p) (cons "s" side)))))
(defstruct dir
(:c 0 :type (integer 0 15))
(:r 0 :type (integer 0 1)))
(defmethod print-object ((obj dir) s)
(print-unreadable-object (obj s)
(format s "~a~@d" (aref *dirnames* (dir-c obj)) (dir-r obj))))
(defstruct conns
(:from (error "from-field not specified for conns entry") :type dir)
(:to (error "to-field not specified for conns entry") :type dir))
(defmethod print-object ((obj conns) s)
(print-unreadable-object (obj s)
(format s "FROM ~a TO ~a" (conns-from obj) (conns-to obj))))
(defmacro dir (c &optional r)
`(make-dir :c (or ,c 0) :r (or ,r 0)))
(defmacro conns (c1 c2 &optional r1 r2)
`(make-conns :from (dir ,c1 ,r1) :to (dir ,c2 ,r2)))
(defmacro conns-t ((c1 &body cb) (&optional r1 &body rb))
``(,,@(loop for c in cb and ri = rb then (cdr ri) for r = (car ri)
collect `(conns ,c1 ,c ,r1 ,r))))
(defmacro conns-x ((&body cb) (&body rb))
``(,,@(loop for (c1 c2) on cb by #'cddr and r = rb then (cddr rb)
for r1 = (car r) and r2 = (cadr r)
collect `(conns ,c1 ,c2 ,r1 ,r2))))
(defmacro adjust-dir (dir diff)
(alexandria:once-only
(dir)
`(dir (mod (+ 16 (or ,diff 0) (dir-c ,dir)) 16) (dir-r ,dir))))
(defmacro adjust-conns (conns diff)
(alexandria:once-only
(conns diff)
`(make-conns :from (adjust-dir (conns-from ,conns) ,diff)
:to (adjust-dir (conns-to ,conns) ,diff))))
(defmacro adjust-conns-list (clist diff)
(alexandria:with-gensyms (i)
(alexandria:once-only
(diff)
`(loop for ,i in ,clist collect (adjust-conns ,i ,diff)))))
(defmacro adjust-dir-with-param2 (dir param2)
`(adjust-dir ,dir (* 4 ,param2)))
(defmacro adjust-conns-with-param2 (conns param2)
`(adjust-conns ,conns (* 4 ,param2)))
(defmacro adjust-conns-list-with-param2 (clist param2)
`(adjust-conns-list ,clist (* 4 ,param2)))
(defmacro apply-dir-to-v3d (pos dir)
(alexandria:with-gensyms (h)
(alexandria:once-only
(pos dir)
`(let ((,h (aref *hdiff* (dir-c ,dir))))
(aux:make-v3d :x (+ (aux:v3d-x ,pos) (car ,h))
:y (+ (aux:v3d-y ,pos) (dir-r ,dir))
:z (+ (aux:v3d-z ,pos) (cdr ,h)))))))
(defmacro conns-list-dirs (clist)
(alexandria:with-gensyms (i)
`(remove-duplicates (loop for ,i in ,clist collect (conns-from ,i) collect (conns-to ,i))
:test #'equalp)))
(defmacro dir-length (dir)
(alexandria:with-gensyms (h)
(alexandria:once-only
(dir)
`(let ((,h (aref *hdiff* (dir-c ,dir))))
(sqrt (+ (expt (car ,h) 2) (expt (cdr ,h) 2) (expt (dir-r ,dir) 2) 0.0))))))