in the fade

main
Matthew Butterick 6 years ago
parent 21b380587d
commit 0bffe1b26c

@ -14,14 +14,14 @@ https://github.com/mbutterick/restructure/blob/master/src/Array.coffee
(define x:array% (define x:array%
(class xenobase% (class xenobase%
(super-new) (super-new)
(init-field [(@type type)] [(@len len)] [(@length-type length-type)]) (init-field [(@type type)] [(@len len)] [(@count-bytes? count-bytes?)])
(unless (xenomorphic-type? @type) (unless (xenomorphic-type? @type)
(raise-argument-error '+xarray "xenomorphic type" @type)) (raise-argument-error 'x:array "xenomorphic type" @type))
(unless (length-resolvable? @len) (unless (length-resolvable? @len)
(raise-argument-error '+xarray "length-resolvable?" @len)) (raise-argument-error 'x:array "length-resolvable?" @len))
(unless (memq @length-type '(bytes count)) (unless (boolean? @count-bytes?)
(raise-argument-error '+xarray "'bytes or 'count" @length-type)) (raise-argument-error 'x:array "boolean" @count-bytes?))
(define/augride (x:decode port parent) (define/augride (x:decode port parent)
(define new-parent (if (x:int? @len) (define new-parent (if (x:int? @len)
@ -32,19 +32,19 @@ https://github.com/mbutterick/restructure/blob/master/src/Array.coffee
parent)) parent))
(define len (resolve-length @len port parent)) (define len (resolve-length @len port parent))
(cond (cond
[(or (not len) (eq? @length-type 'bytes)) [(or (not len) @count-bytes?)
(define end-pos (cond (define end-pos (cond
;; resolved-len is byte length ;; len is byte length
[len (+ (pos port) len)] [len (+ (pos port) len)]
;; no resolved-len, but parent has length ;; no len, but parent has length
[(and parent (not (zero? (hash-ref parent x:length-key)))) [(and parent (not (zero? (hash-ref parent x:length-key))))
(+ (hash-ref parent x:start-offset-key) (hash-ref parent x:length-key))] (+ (hash-ref parent x:start-offset-key) (hash-ref parent x:length-key))]
;; no resolved-len or parent, so consume whole stream ;; no len or parent, so consume whole stream
[else +inf.0])) [else +inf.0]))
(for/list ([i (in-naturals)] (for/list ([i (in-naturals)]
#:break (or (eof-object? (peek-byte port)) (= (pos port) end-pos))) #:break (or (eof-object? (peek-byte port)) (= (pos port) end-pos)))
(send @type x:decode port new-parent))] (send @type x:decode port new-parent))]
;; we have resolved-len, which is treated as count of items ;; we have len, which is treated as count of items
[else (for/list ([i (in-range len)]) [else (for/list ([i (in-range len)])
(send @type x:decode port new-parent))])) (send @type x:decode port new-parent))]))
@ -94,7 +94,7 @@ https://github.com/mbutterick/restructure/blob/master/src/Array.coffee
#:post-decode [post-proc #f]) #:post-decode [post-proc #f])
(new (generate-subclass x:array% pre-proc post-proc) [type (or type-arg type-kwarg)] (new (generate-subclass x:array% pre-proc post-proc) [type (or type-arg type-kwarg)]
[len (or len-arg len-kwarg)] [len (or len-arg len-kwarg)]
[length-type (if count-bytes? 'bytes length-type-arg)])) [count-bytes? count-bytes?]))
(define (x:array? x) (is-a? x x:array%)) (define (x:array? x) (is-a? x x:array%))

@ -47,7 +47,7 @@ https://github.com/mbutterick/restructure/blob/master/src/LazyArray.coffee
(define len (or len-arg len-kwarg)) (define len (or len-arg len-kwarg))
(new (generate-subclass x:lazy-array% pre-proc post-proc) [type type] (new (generate-subclass x:lazy-array% pre-proc post-proc) [type type]
[len len] [len len]
[length-type 'count])) [count-bytes? #false]))
(module+ test (module+ test
(require rackunit "number.rkt" "generic.rkt") (require rackunit "number.rkt" "generic.rkt")

Loading…
Cancel
Save