2d/2d-lib/private/readtable.rkt
Robby Findler a350659906 export some readtable functions
and finish up the docs
2016-07-31 02:51:48 -05:00

174 lines
6.4 KiB
Racket

#lang racket/base
#|
ideas:
- 2dcond
- 2dmatch
- literal tables in scribble layout?
- something for graphics?
example uses:
- unifier
- subtyping relation
|#
(require "read-util.rkt"
racket/set
;syntax/rect
"../dir-chars.rkt"
racket/port)
(provide make-2d-readtable
2d-readtable-dispatch-proc)
(define (make-2d-readtable)
(define previous-readtable (current-readtable))
(make-readtable
previous-readtable
#\2
'dispatch-macro
(case-lambda
[(char port)
(define-values (line col pos) (port-next-location port))
;; the "-2"s here are because the initial line and column
;; are supposed be at the beginning of the thing read, not
;; after the "#2" has been consumed.
(2d-readtable-dispatch-proc char port #f line
(and col (- col 2))
(and pos (- pos 2))
read/recursive previous-readtable)]
[(char port source _line _col _pos)
(2d-readtable-dispatch-proc char port source _line _col _pos
(λ (a b c) (read-syntax/recursive source a b c))
previous-readtable)])))
(define (2d-readtable-dispatch-proc char port source _line _col _pos /recursive previous-readtable)
(define next-char (peek-char port))
(cond
[(equal? next-char #\d)
(define chars-read 2) ;; account for the # and the 2
(define (rc)
(set! chars-read (+ chars-read 1))
(read-char port))
(rc) ;; get the #\d
(define kwd-chars
(let loop ()
(define c (rc))
(cond
[(eof-object? c)
(raise (make-exn:fail:read:eof
"expected a newline to follow #2d"
(current-continuation-marks)
(list (srcloc source
_line _col _pos
(+ _pos chars-read)))))]
[(equal? c #\newline) '()]
[(equal? c #\return)
(when (equal? #\newline (peek-char port))
(rc))
'()]
[else (cons c (loop))])))
(define-values (post-2d-line post-2d-col post-2d-span) (port-next-location port))
(define-values (cell-connections
lines
table-column-breaks
initial-space-count
position-of-first-cell)
(parse-2dcond port source _line _col _pos chars-read))
(define lhses (close-cell-graph cell-connections
(length table-column-breaks)
(vector-length lines)))
(define scratch-string (make-string (for/sum ([ss (in-vector lines)])
(for/sum ([s (in-list ss)])
(string-length s)))
#\space))
(define heights
(for/list ([line (in-vector lines)])
(length line)))
(define kwd-str (string-append "2d" (apply string kwd-chars)))
(define kwd-port (open-input-string kwd-str))
(port-count-lines! kwd-port)
(set-port-next-location! kwd-port _line (and _col (+ _col 1)) (and _pos (+ _pos 1)))
(define kwd-stx (read-syntax source kwd-port))
(define line-width (+ initial-space-count
(apply + table-column-breaks)
(max 0 (- (length table-column-breaks) 1))))
(define (add-srclocs indicies)
(for/list ([index (in-list indicies)])
(define srcloc (hash-ref position-of-first-cell index))
(datum->syntax #f
index
(vector (srcloc-source srcloc)
#f ;; line
#f ;; col
(srcloc-position srcloc)
1))))
`(,kwd-stx
,table-column-breaks
,heights
,@(for/list ([set-of-indicies (in-list (sort (set->list lhses) compare/xy
#:key smallest-representative))])
(fill-scratch-string set-of-indicies
lines
scratch-string
table-column-breaks
initial-space-count)
(define scratch-port (open-input-string scratch-string))
(when post-2d-line (port-count-lines! scratch-port))
(set-port-next-location! scratch-port post-2d-line post-2d-col post-2d-span)
`[,(add-srclocs (sort (set->list set-of-indicies) compare/xy))
,@(read-subparts source scratch-port
initial-space-count table-column-breaks heights set-of-indicies
/recursive)]))]
[else
(/recursive
(input-port-append #f (open-input-string "#2") port)
#f
previous-readtable)]))
(define (read-subparts source scratch-port
initial-space-count table-column-breaks heights lhs
/recursive)
(with-handlers (#;
[exn:fail:read?
(λ (exn)
(define constructor
(cond
[(exn:fail:read:eof? exn) exn:fail:read:eof/rects]
[(exn:fail:read:non-char? exn) exn:fail:read:non-char/rects]
[else exn:fail:read/rects]))
(raise
(constructor (exn-message exn)
(exn-continuation-marks exn)
(exn:fail:read-srclocs exn)
(build-rectangles
source
initial-space-count table-column-breaks heights lhs))))])
(let loop ()
(define o (/recursive scratch-port #f (current-readtable)))
(cond
[(eof-object? o) '()]
[else (cons o (loop))]))))
#;
(define (build-rectangles source table-column-breaks heights set-of-indicies)
(for/list ([pr (in-set set-of-indicies)])
(define x (list-ref pr 0))
(define y (list-ref pr 1))
(srcloc-rect source
?-start-position
(list-ref table-column-breaks x)
(list-ref heights y))))