|
|
#lang racket/base
|
|
|
(require "racket.rkt")
|
|
|
(require "number.rkt" "utils.rkt")
|
|
|
(provide (all-defined-out))
|
|
|
|
|
|
#|
|
|
|
approximates
|
|
|
https://github.com/mbutterick/restructure/blob/master/src/Array.coffee
|
|
|
|#
|
|
|
|
|
|
(define-subclass xenomorph-base% (ArrayT type [len #f] [length-type 'count])
|
|
|
|
|
|
(define/augride (decode port [parent #f])
|
|
|
(define ctx (if (NumberT? len)
|
|
|
(mhasheq 'parent parent
|
|
|
'_startOffset (pos port)
|
|
|
'_currentOffset 0
|
|
|
'_length len)
|
|
|
parent))
|
|
|
|
|
|
(define decoded-len (resolve-length len port parent))
|
|
|
(cond
|
|
|
[(or (not decoded-len) (eq? length-type 'bytes))
|
|
|
(define end-pos (cond
|
|
|
;; decoded-len is byte length
|
|
|
[decoded-len (+ (pos port) decoded-len)]
|
|
|
;; no decoded-len, but parent has length
|
|
|
[(and parent (not (zero? (· parent _length)))) (+ (· parent _startOffset) (· parent _length))]
|
|
|
;; no decoded-len or parent, so consume whole stream
|
|
|
[else +inf.0]))
|
|
|
(for/list ([i (in-naturals)]
|
|
|
#:break (or (eof-object? (peek-byte port)) (= (pos port) end-pos)))
|
|
|
(send type decode port ctx))]
|
|
|
;; we have decoded-len, which is treated as count of items
|
|
|
[else (for/list ([i (in-range decoded-len)])
|
|
|
(send type decode port ctx))]))
|
|
|
|
|
|
|
|
|
(define/augride (size [val #f] [ctx #f])
|
|
|
(when val (unless (countable? val)
|
|
|
(raise-argument-error 'Array:size "countable" val)))
|
|
|
(cond
|
|
|
[val (let-values ([(ctx len-size) (if (NumberT? len)
|
|
|
(values (mhasheq 'parent ctx) (send len size))
|
|
|
(values ctx 0))])
|
|
|
(+ len-size (for/sum ([item (in-list (countable->list val))])
|
|
|
(send type size item ctx))))]
|
|
|
[else (let ([item-count (resolve-length len #f ctx)]
|
|
|
[item-size (send type size #f ctx)])
|
|
|
(* item-size item-count))]))
|
|
|
|
|
|
|
|
|
(define/augride (encode port array [parent #f])
|
|
|
(when array (unless (countable? array)
|
|
|
(raise-argument-error 'Array:encode "list or countable" array)))
|
|
|
|
|
|
(define (encode-items ctx)
|
|
|
(let* ([items (countable->list array)]
|
|
|
[item-count (length items)]
|
|
|
[max-items (if (number? len) len item-count)])
|
|
|
(unless (= item-count max-items)
|
|
|
(raise-argument-error 'Array:encode (format "list or countable with ~a items" max-items) items))
|
|
|
(for ([item (in-list items)])
|
|
|
(send type encode port item ctx))))
|
|
|
|
|
|
(cond
|
|
|
[(NumberT? len) (define ctx (mhash 'pointers null
|
|
|
'startOffset (pos port)
|
|
|
'parent parent))
|
|
|
(ref-set! ctx 'pointerOffset (+ (pos port) (size array ctx)))
|
|
|
(send len encode port (length array)) ; encode length at front
|
|
|
(encode-items ctx)
|
|
|
(for ([ptr (in-list (· ctx pointers))]) ; encode pointer data at end
|
|
|
(send (· ptr type) encode port (· ptr val)))]
|
|
|
[else (encode-items parent)])))
|
|
|
|
|
|
(define-procedures (Array Array? +Array) (ArrayT ArrayT? +ArrayT))
|
|
|
(define-procedures (array% array? array) (ArrayT ArrayT? +ArrayT))
|
|
|
|
|
|
(test-module
|
|
|
(check-equal? (decode (+Array uint16be 3) #"ABCDEF") '(16706 17220 17734))
|
|
|
(check-equal? (encode (+Array uint16be 3) '(16706 17220 17734) #f) #"ABCDEF")
|
|
|
(check-equal? (size (+Array uint16be) '(1 2 3)) 6)
|
|
|
(check-equal? (size (+Array doublebe) '(1 2 3 4 5)) 40))
|